package ctrl
import (
. "github.com/mickael-kerjean/filestash/server/common"
"fmt"
"io"
"text/template"
"net/http"
"os"
"path/filepath"
"strings"
)
func StaticHandler(_path string) func(App, http.ResponseWriter, *http.Request) {
return func(ctx App, res http.ResponseWriter, req *http.Request) {
var base string = GetAbsolutePath(_path)
var srcPath string
if srcPath = JoinPath(base, req.URL.Path); srcPath == base {
http.NotFound(res, req)
return
}
ServeFile(res, req, srcPath)
}
}
func IndexHandler(_path string) func(App, http.ResponseWriter, *http.Request) {
return func(ctx App, res http.ResponseWriter, req *http.Request) {
if req.URL.String() != URL_SETUP && Config.Get("auth.admin").String() == "" {
http.Redirect(res, req, URL_SETUP, http.StatusTemporaryRedirect)
return
}
if ua := req.Header.Get("User-Agent"); strings.Contains(ua, "MSIE ") {
res.WriteHeader(http.StatusBadRequest)
res.Write([]byte(`
Internet explorer is not yet supported
To provide the best possible experience for everyone else, we don't support IE at this time.
Use either Chromium, Firefox or Chrome
`))
return
}
srcPath := GetAbsolutePath(_path)
ServeFile(res, req, srcPath)
}
}
func AboutHandler(ctx App, res http.ResponseWriter, req *http.Request) {
page := `
{{index .App 0}}
({{index .App 1}} - {{index .App 2}})
`
t, _ := template.New("about").Parse(page)
t.Execute(res, struct {
App []string
}{ []string{
"Filestash " + APP_VERSION + "." + BUILD_NUMBER,
hashFile(filepath.Join(GetCurrentDir(), "/filestash"), 6),
hashFile(filepath.Join(GetCurrentDir(), CONFIG_PATH, "config.json"), 6),
}})
}
func hashFile (path string, n int) string {
f, err := os.OpenFile(path, os.O_RDONLY, os.ModePerm)
if err != nil {
return ""
}
defer f.Close()
stat, err := f.Stat()
if err != nil {
return ""
}
return QuickHash(fmt.Sprintf("%s %d %d %s", path, stat.Size(), stat.Mode(), stat.ModTime()), n)
}
func ServeFile(res http.ResponseWriter, req *http.Request, filePath string) {
zFilePath := filePath + ".gz"
etagNormal := hashFile(filePath, 10)
etagGzip := hashFile(zFilePath, 10)
if req.Header.Get("If-None-Match") != "" {
browserTag := req.Header.Get("If-None-Match")
if browserTag == etagNormal {
res.WriteHeader(http.StatusNotModified)
return
} else if browserTag == etagGzip {
res.WriteHeader(http.StatusNotModified)
return
}
}
head := res.Header()
if strings.Contains(req.Header.Get("Accept-Encoding"), "gzip") {
if file, err := os.OpenFile(zFilePath, os.O_RDONLY, os.ModePerm); err == nil {
head.Set("Content-Encoding", "gzip")
head.Set("Etag", etagGzip)
io.Copy(res, file)
file.Close()
return
}
}
file, err := os.OpenFile(filePath, os.O_RDONLY, os.ModePerm)
if err != nil {
http.NotFound(res, req)
return
}
head.Set("Etag", etagNormal)
io.Copy(res, file)
file.Close()
}