Ivan/commento
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

router_static.go: set Content-Type header

Browse Source
This commit is contained in:
Adhityaa 2018-06-03 20:19:36 +05:30
parent 7e9713b35e
commit efb23778b8
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/router_static.go
View File

@ -8,6 +8,8 @@ import (
"io/ioutil"
"net/http"
"os"
"path"
"mime"
)
func redirectLogin(w http.ResponseWriter, r *http.Request) {
@ -34,11 +36,11 @@ func initStaticRouter(router *mux.Router) error {
for _, file := range files {
sl := string(os.PathSeparator)
path := sl + dir + sl + file.Name()
p := sl + dir + sl + file.Name()
contents, err := ioutil.ReadFile("." + path)
contents, err := ioutil.ReadFile("." + p)
if err != nil {
logger.Errorf("cannot read file %s: %v", path, err)
logger.Errorf("cannot read file %s: %v", p, err)
return err
}
@ -49,7 +51,8 @@ func initStaticRouter(router *mux.Router) error {
asset[p] = prefix + string(contents);
router.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
router.HandleFunc(p, func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", mime.TypeByExtension(path.Ext(r.URL.Path)))
fmt.Fprint(w, asset[r.URL.Path])
})
}

4
api/utils_sanitise.go
View File

@ -26,10 +26,10 @@ func stripDomain(domain string) string {
return noProtocol
}
var path = regexp.MustCompile(`(https?://[^/]*)`)
var pathMatch = regexp.MustCompile(`(https?://[^/]*)`)
func stripPath(url string) string {
strippedPath := path.ReplaceAllString(url, ``)
strippedPath := pathMatch.ReplaceAllString(url, ``)
return strippedPath
}