mirror of
https://github.com/muety/wakapi.git
synced 2023-08-10 21:12:56 +03:00
feat: serve swagger ui
fix: forbid to browse file system index
This commit is contained in:
32
middlewares/filetype.go
Normal file
32
middlewares/filetype.go
Normal file
@@ -0,0 +1,32 @@
|
||||
package middlewares
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type SuffixFilterMiddleware struct {
|
||||
handler http.Handler
|
||||
filterTypes []string
|
||||
}
|
||||
|
||||
func NewFileTypeFilterMiddleware(filter []string) func(http.Handler) http.Handler {
|
||||
return func(h http.Handler) http.Handler {
|
||||
return &SuffixFilterMiddleware{
|
||||
handler: h,
|
||||
filterTypes: filter,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (f *SuffixFilterMiddleware) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
path := strings.ToLower(r.URL.Path)
|
||||
for _, t := range f.filterTypes {
|
||||
if strings.HasSuffix(path, strings.ToLower(t)) {
|
||||
w.WriteHeader(http.StatusForbidden)
|
||||
w.Write([]byte("403 forbidden"))
|
||||
return
|
||||
}
|
||||
}
|
||||
f.handler.ServeHTTP(w, r)
|
||||
}
|
Reference in New Issue
Block a user