mirror of
https://github.com/stashapp/stash.git
synced 2025-12-06 16:34:02 +01:00
7bb35b2b09
* getStringMapString return nil if key not found * Refactor custom routes. Handle /index.html
35 lines
780 B
Go
35 lines
780 B
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/go-chi/chi"
|
|
"github.com/stashapp/stash/internal/manager/config"
|
|
)
|
|
|
|
type customRoutes struct {
|
|
servedFolders config.URLMap
|
|
}
|
|
|
|
func (rs customRoutes) Routes() chi.Router {
|
|
r := chi.NewRouter()
|
|
|
|
r.HandleFunc("/*", func(w http.ResponseWriter, r *http.Request) {
|
|
r.URL.Path = strings.Replace(r.URL.Path, "/custom", "", 1)
|
|
|
|
// http.FileServer redirects to / if the path ends with index.html
|
|
r.URL.Path = strings.TrimSuffix(r.URL.Path, "/index.html")
|
|
|
|
// map the path to the applicable filesystem location
|
|
var dir string
|
|
r.URL.Path, dir = rs.servedFolders.GetFilesystemLocation(r.URL.Path)
|
|
if dir != "" {
|
|
http.FileServer(http.Dir(dir)).ServeHTTP(w, r)
|
|
} else {
|
|
http.NotFound(w, r)
|
|
}
|
|
})
|
|
|
|
return r
|
|
}
|