111 lines
2.9 KiB
Go
111 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"ponyo.epfl.ch/gitlab/alexandre.tuleu/narco"
|
|
|
|
bleve_http "github.com/blevesearch/bleve/http"
|
|
"github.com/codemodus/chain"
|
|
"github.com/julienschmidt/httprouter"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
var rxExt = regexp.MustCompile(`[0-9]+`)
|
|
|
|
func (a *appData) buildRouter() http.Handler {
|
|
router := httprouter.New()
|
|
|
|
logger := narco.NewLogger()
|
|
f, err := os.Create(filepath.Join("tmp", "log", "access.log"))
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
logger.SetOutput(f)
|
|
recoverer := narco.NewRecoverer()
|
|
ctx := narco.WithErrorFormatter(context.Background(), narco.BasicHTMLErrorFormatter)
|
|
ch := chain.New(logger.Wrap(), recoverer.Wrap()).SetContext(ctx)
|
|
|
|
bleve_http.RegisterIndexName("album", a.index)
|
|
searchHandler := bleve_http.NewSearchHandler("album")
|
|
|
|
router.POST("/api/search", narco.EndChain(ch,
|
|
narco.HandlerFunc(func(_ context.Context, w http.ResponseWriter, req *http.Request, _ httprouter.Params) {
|
|
searchHandler.ServeHTTP(w, req)
|
|
})))
|
|
|
|
router.GET("/api/recents", narco.EndChain(ch,
|
|
narco.HandlerFunc(func(ctx context.Context, w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
|
|
albums, err := a.db.ByPurchaseDate()
|
|
|
|
if err != nil {
|
|
narco.Error(ctx, w, err, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
enc := json.NewEncoder(w)
|
|
err = enc.Encode(albums)
|
|
if err != nil {
|
|
narco.Error(ctx, w, err, http.StatusInternalServerError)
|
|
}
|
|
})))
|
|
|
|
router.GET("/api/albums/:id", narco.EndChain(ch,
|
|
narco.HandlerFunc(func(ctx context.Context, w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
var id uint64
|
|
var err error
|
|
idStr := ps.ByName("id")
|
|
if len(idStr) > 0 {
|
|
id, err = strconv.ParseUint(idStr, 10, 64)
|
|
}
|
|
|
|
if len(idStr) == 0 || err != nil {
|
|
narco.Error(ctx, w, http.StatusText(http.StatusNotFound), http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
albumUnsafe, err := a.db.Get(AlbumID(id))
|
|
if err != nil {
|
|
narco.Error(ctx, w, err, http.StatusInternalServerError)
|
|
}
|
|
|
|
album := *albumUnsafe
|
|
ext := path.Ext(album.CoverURL)
|
|
ext = strings.ToLower(rxExt.ReplaceAllString(ext, ""))
|
|
album.CoverURL = fmt.Sprintf("/covers/%d%s", album.ID, ext)
|
|
|
|
enc := json.NewEncoder(w)
|
|
if err := enc.Encode(album); err != nil {
|
|
narco.Error(ctx, w, err, http.StatusInternalServerError)
|
|
}
|
|
})))
|
|
|
|
dirs := []string{"css", "js", "img"}
|
|
for _, d := range dirs {
|
|
router.ServeFiles(path.Join("/", d, "/*filepath"), http.Dir(filepath.Join("static", d)))
|
|
}
|
|
|
|
router.GET("/", narco.EndChain(ch, narco.HandlerFunc(
|
|
func(ctx context.Context, w http.ResponseWriter, req *http.Request, _ httprouter.Params) {
|
|
f, err := os.Open(filepath.Join("static", "index.html"))
|
|
if err != nil {
|
|
narco.Error(ctx, w, err, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
defer closeOrPanic(f, filepath.Join("static", "index.html"))
|
|
|
|
io.Copy(w, f)
|
|
})))
|
|
|
|
return router
|
|
}
|