mirror of
https://github.com/mickael-kerjean/filestash
synced 2025-12-16 05:18:38 +01:00
96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"mime"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
"net/url"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
CHROOT = "/"
|
|
ENDPOINT = "127.0.0.1:8000"
|
|
)
|
|
|
|
func main() {
|
|
fmt.Printf("starting server http://%s ...\n", ENDPOINT)
|
|
err := http.ListenAndServe(ENDPOINT, http.HandlerFunc(serveStatic))
|
|
if err != nil {
|
|
fmt.Printf("Oops %s\n", err.Error())
|
|
}
|
|
}
|
|
|
|
func serveStatic(w http.ResponseWriter, r *http.Request) {
|
|
if strings.HasPrefix(r.URL.Path, "/admin/api/") ||
|
|
strings.HasPrefix(r.URL.Path, "/api/") ||
|
|
strings.HasPrefix(r.URL.Path, "/about") ||
|
|
strings.HasPrefix(r.URL.Path, "/overrides/") ||
|
|
strings.HasPrefix(r.URL.Path, "/sw_cache.js") {
|
|
u, _ := url.Parse("http://127.0.0.1:8334")
|
|
httputil.NewSingleHostReverseProxy(u).ServeHTTP(w, r)
|
|
return
|
|
}
|
|
var (
|
|
fs http.Dir
|
|
f http.File
|
|
err error
|
|
)
|
|
defer func() {
|
|
now := time.Now().Format("2006-01-02 15:04:05")
|
|
if err != nil {
|
|
fmt.Printf("%s HTTP %s %s err[%s]\n", now, r.Method, r.URL.Path, err.Error())
|
|
return
|
|
}
|
|
fmt.Printf("%s HTTP %s %s\n", now, r.Method, r.URL.Path)
|
|
}()
|
|
fs = http.Dir(".")
|
|
if strings.HasSuffix(r.URL.Path, "/") && strings.HasPrefix(r.URL.Path, "/assets/") {
|
|
http.FileServer(fs).ServeHTTP(w, r)
|
|
return
|
|
}
|
|
if r.URL.Path == "/" {
|
|
r.URL.Path = CHROOT + "index.html"
|
|
}
|
|
f, err = openFile(fs, r, w)
|
|
if err != nil {
|
|
r.URL.Path = CHROOT + "index.html"
|
|
f, err = openFile(fs, r, w)
|
|
}
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
w.Write([]byte("Oops"))
|
|
return
|
|
}
|
|
// TODO: etag
|
|
w.Header().Set("Content-Type", mime.TypeByExtension(filepath.Ext(r.URL.Path)))
|
|
io.Copy(w, f)
|
|
f.Close()
|
|
}
|
|
|
|
func openFile(fs http.Dir, r *http.Request, w http.ResponseWriter) (http.File, error) {
|
|
encs := [][]string{}
|
|
if strings.Contains(r.Header.Get("Accept-Encoding"), "br") {
|
|
encs = append(encs, []string{".br", "br"})
|
|
}
|
|
if strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
|
|
encs = append(encs, []string{".gz", "gzip"})
|
|
}
|
|
encs = append(encs, []string{"", ""})
|
|
|
|
for _, enc := range encs {
|
|
f, err := fs.Open(r.URL.Path + enc[0])
|
|
if err == nil {
|
|
if enc[1] != "" {
|
|
w.Header().Set("Content-Encoding", enc[1])
|
|
}
|
|
return f, nil
|
|
}
|
|
}
|
|
return nil, errors.New("not found")
|
|
}
|