vbatts/homebox
1
0
Fork 0
forked from mirrors/homebox
Code Pull requests Activity

encapsulate notFoundHandler

Browse source
This commit is contained in:
Hayden 2022-09-03 19:10:42 -08:00
parent 5bf7efe87e
commit 61483c3ea4
1 changed files with 32 additions and 49 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
backend/app/api/routes.go
View file

@ -5,7 +5,6 @@ import (
"errors"
"fmt"
"io"
"io/fs"
"mime"
"net/http"
"path"
@ -26,14 +25,11 @@ const prefix = "/api"
// registerRoutes registers all the routes for the API
func (a *app) newRouter(repos *repo.AllRepos) *chi.Mux {
registerMimes()
r := chi.NewRouter()
a.setGlobalMiddleware(r)
// =========================================================================
// Base Routes
DumpEmbedContents()
r.Get("/swagger/*", httpSwagger.Handler(
httpSwagger.URL(fmt.Sprintf("%s://%s/swagger/doc.json", a.conf.Swagger.Scheme, a.conf.Swagger.Host)),
))
@ -78,8 +74,7 @@ func (a *app) newRouter(repos *repo.AllRepos) *chi.Mux {
})
}
r.NotFound(NotFoundHandler)
r.NotFound(notFoundHandler())
return r
}
@ -106,7 +101,7 @@ func (a *app) LogRoutes(r *chi.Mux) {
var ErrDir = errors.New("path is dir")
func init() {
func registerMimes() {
err := mime.AddExtensionType(".js", "application/javascript")
if err != nil {
panic(err)
@ -118,7 +113,8 @@ func init() {
}
}
func tryRead(fs embed.FS, prefix, requestedPath string, w http.ResponseWriter) error {
func notFoundHandler() http.HandlerFunc {
tryRead := func(fs embed.FS, prefix, requestedPath string, w http.ResponseWriter) error {
f, err := fs.Open(path.Join(prefix, requestedPath))
if err != nil {
return err
@ -136,7 +132,7 @@ func tryRead(fs embed.FS, prefix, requestedPath string, w http.ResponseWriter) e
return err
}
func NotFoundHandler(w http.ResponseWriter, r *http.Request) {
return func(w http.ResponseWriter, r *http.Request) {
err := tryRead(public, "public", r.URL.Path, w)
if err == nil {
return
@ -149,17 +145,4 @@ func NotFoundHandler(w http.ResponseWriter, r *http.Request) {
panic(err)
}
}
func DumpEmbedContents() {
// recursively prints all contents in the embed.FS
err := fs.WalkDir(public, ".", func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
fmt.Println(path)
return nil
})
if err != nil {
panic(err)
}
}