forked from MTSR/mapserver
46 lines
907 B
Go
46 lines
907 B
Go
package web
|
|
|
|
import (
|
|
"mapserver/app"
|
|
"mapserver/public"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
type MediaHandler struct {
|
|
ctx *app.App
|
|
}
|
|
|
|
func (h *MediaHandler) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
|
|
str := strings.TrimPrefix(req.URL.Path, "/api/media/")
|
|
parts := strings.Split(str, "/")
|
|
if len(parts) != 1 {
|
|
resp.WriteHeader(500)
|
|
resp.Write([]byte("wrong number of arguments"))
|
|
return
|
|
}
|
|
|
|
filename := parts[0]
|
|
fallback, hasfallback := req.URL.Query()["fallback"]
|
|
|
|
content := h.ctx.MediaRepo[filename]
|
|
|
|
if content == nil && hasfallback && len(fallback) > 0 {
|
|
var err error
|
|
content, err = public.Files.ReadFile("pics/" + fallback[0])
|
|
if err != nil {
|
|
resp.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
|
|
if content != nil {
|
|
resp.Write(content)
|
|
resp.Header().Add("content-type", "image/png")
|
|
return
|
|
}
|
|
|
|
resp.WriteHeader(404)
|
|
resp.Write([]byte(filename))
|
|
}
|