mirror of https://github.com/status-im/fathom.git
77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/gobuffalo/packr"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// Handler is our custom HTTP handler with error returns
|
|
type Handler func(w http.ResponseWriter, r *http.Request) error
|
|
|
|
type envelope struct {
|
|
Data interface{} `json:",omitempty"`
|
|
Error interface{} `json:",omitempty"`
|
|
}
|
|
|
|
func (h Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
if err := h(w, r); err != nil {
|
|
HandleError(w, r, err)
|
|
}
|
|
}
|
|
|
|
// HandlerFunc takes a custom Handler func and converts it to http.HandlerFunc
|
|
func HandlerFunc(fn Handler) http.HandlerFunc {
|
|
return http.HandlerFunc(Handler(fn).ServeHTTP)
|
|
}
|
|
|
|
// HandleError handles errors
|
|
func HandleError(w http.ResponseWriter, r *http.Request, err error) {
|
|
log.WithFields(log.Fields{
|
|
"request": r.Method + " " + r.RequestURI,
|
|
"error": err,
|
|
}).Error("error handling request")
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.Write([]byte("false"))
|
|
}
|
|
|
|
func respond(w http.ResponseWriter, d interface{}) error {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
err := json.NewEncoder(w).Encode(d)
|
|
return err
|
|
}
|
|
|
|
func serveFileHandler(box *packr.Box, filename string) http.Handler {
|
|
return HandlerFunc(serveFile(box, filename))
|
|
}
|
|
|
|
func serveFile(box *packr.Box, filename string) Handler {
|
|
return func(w http.ResponseWriter, r *http.Request) error {
|
|
f, err := box.Open(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
d, err := f.Stat()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
http.ServeContent(w, r, filename, d.ModTime(), f)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func NotFoundHandler(box *packr.Box) http.Handler {
|
|
return HandlerFunc(func(w http.ResponseWriter, r *http.Request) error {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
w.Write(box.Bytes("404.html"))
|
|
return nil
|
|
})
|
|
}
|