86 lines
1.4 KiB
Go
86 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
|
|
"git.sp4ke.com/sp4ke/gomark/bookmarks"
|
|
"git.sp4ke.com/sp4ke/gomark/database"
|
|
|
|
"git.sp4ke.com/sp4ke/gum"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type Bookmark = bookmarks.Bookmark
|
|
|
|
func getBookmarks(c *gin.Context) {
|
|
|
|
rows, err := CacheDB.Handle.QueryContext(c, "SELECT URL, metadata, tags FROM bookmarks")
|
|
if err != nil {
|
|
log.Error(err)
|
|
}
|
|
var bookmarks []Bookmark
|
|
|
|
var tags string
|
|
for rows.Next() {
|
|
bookmark := Bookmark{}
|
|
err = rows.Scan(&bookmark.URL, &bookmark.Metadata, &tags)
|
|
if err != nil {
|
|
log.Error(err)
|
|
}
|
|
|
|
bookmark.Tags = strings.Split(tags, database.TagJoinSep)
|
|
//log.Debugf("GET %s", tags)
|
|
//log.Debugf("%v", bookmark)
|
|
|
|
bookmarks = append(bookmarks, bookmark)
|
|
}
|
|
//log.Debugf("%v", bookmarks)
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"bookmarks": bookmarks,
|
|
})
|
|
}
|
|
|
|
type API struct {
|
|
engine *gin.Engine
|
|
router *gin.RouterGroup
|
|
}
|
|
|
|
func (api *API) Shutdown() {}
|
|
|
|
func (api *API) Run(m gum.UnitManager) {
|
|
api.router.GET("/urls", getBookmarks)
|
|
|
|
// Run router
|
|
// TODO: config params for api
|
|
go func() {
|
|
err := api.engine.Run(":4444")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
}()
|
|
|
|
// Wait for stop signal
|
|
<-m.ShouldStop()
|
|
|
|
api.Shutdown()
|
|
m.Done()
|
|
}
|
|
|
|
func NewApi() *API {
|
|
apiLogFile, _ := os.Create(".api.log")
|
|
gin.DefaultWriter = io.MultiWriter(apiLogFile, os.Stdout)
|
|
|
|
api := gin.Default()
|
|
|
|
return &API{
|
|
engine: api,
|
|
router: api.Group("/api"),
|
|
}
|
|
|
|
}
|