forked from mirrors/homebox
encapsulate notFoundHandler
This commit is contained in:
parent
5bf7efe87e
commit
61483c3ea4
1 changed files with 32 additions and 49 deletions
|
@ -5,7 +5,6 @@ import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"io/fs"
|
|
||||||
"mime"
|
"mime"
|
||||||
"net/http"
|
"net/http"
|
||||||
"path"
|
"path"
|
||||||
|
@ -26,14 +25,11 @@ const prefix = "/api"
|
||||||
|
|
||||||
// registerRoutes registers all the routes for the API
|
// registerRoutes registers all the routes for the API
|
||||||
func (a *app) newRouter(repos *repo.AllRepos) *chi.Mux {
|
func (a *app) newRouter(repos *repo.AllRepos) *chi.Mux {
|
||||||
|
registerMimes()
|
||||||
|
|
||||||
r := chi.NewRouter()
|
r := chi.NewRouter()
|
||||||
a.setGlobalMiddleware(r)
|
a.setGlobalMiddleware(r)
|
||||||
|
|
||||||
// =========================================================================
|
|
||||||
// Base Routes
|
|
||||||
|
|
||||||
DumpEmbedContents()
|
|
||||||
|
|
||||||
r.Get("/swagger/*", httpSwagger.Handler(
|
r.Get("/swagger/*", httpSwagger.Handler(
|
||||||
httpSwagger.URL(fmt.Sprintf("%s://%s/swagger/doc.json", a.conf.Swagger.Scheme, a.conf.Swagger.Host)),
|
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
|
return r
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -106,7 +101,7 @@ func (a *app) LogRoutes(r *chi.Mux) {
|
||||||
|
|
||||||
var ErrDir = errors.New("path is dir")
|
var ErrDir = errors.New("path is dir")
|
||||||
|
|
||||||
func init() {
|
func registerMimes() {
|
||||||
err := mime.AddExtensionType(".js", "application/javascript")
|
err := mime.AddExtensionType(".js", "application/javascript")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
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))
|
f, err := fs.Open(path.Join(prefix, requestedPath))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
|
@ -136,7 +132,7 @@ func tryRead(fs embed.FS, prefix, requestedPath string, w http.ResponseWriter) e
|
||||||
return err
|
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)
|
err := tryRead(public, "public", r.URL.Path, w)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
return
|
return
|
||||||
|
@ -149,17 +145,4 @@ func NotFoundHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
panic(err)
|
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)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue