mirror of
https://github.com/mickael-kerjean/filestash.git
synced 2025-11-01 02:43:35 +08:00
278 lines
8.3 KiB
Go
278 lines
8.3 KiB
Go
package ctrl
|
|
|
|
import (
|
|
"embed"
|
|
"fmt"
|
|
. "github.com/mickael-kerjean/filestash/server/common"
|
|
"io"
|
|
"io/fs"
|
|
"net/http"
|
|
URL "net/url"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
"text/template"
|
|
)
|
|
|
|
var (
|
|
WWWDir fs.FS
|
|
//go:embed static/www
|
|
WWWEmbed embed.FS
|
|
|
|
//go:embed static/404.html
|
|
HtmlPage404 []byte
|
|
)
|
|
|
|
func init() {
|
|
WWWDir = os.DirFS(GetAbsolutePath("../"))
|
|
}
|
|
|
|
func StaticHandler(_path string) func(*App, http.ResponseWriter, *http.Request) {
|
|
return func(ctx *App, res http.ResponseWriter, req *http.Request) {
|
|
var chroot string = GetAbsolutePath(_path)
|
|
if srcPath := JoinPath(chroot, req.URL.Path); strings.HasPrefix(srcPath, chroot) == false {
|
|
http.NotFound(res, req)
|
|
return
|
|
}
|
|
ServeFile(res, req, JoinPath(_path, req.URL.Path))
|
|
}
|
|
}
|
|
|
|
func IndexHandler(ctx *App, res http.ResponseWriter, req *http.Request) {
|
|
urlObj, err := URL.Parse(req.URL.String())
|
|
if err != nil {
|
|
NotFoundHandler(ctx, res, req)
|
|
return
|
|
}
|
|
url := urlObj.Path
|
|
|
|
if url != URL_SETUP && Config.Get("auth.admin").String() == "" {
|
|
http.Redirect(res, req, URL_SETUP, http.StatusTemporaryRedirect)
|
|
return
|
|
} else if url != "/" && strings.HasPrefix(url, "/s/") == false &&
|
|
strings.HasPrefix(url, "/view/") == false && strings.HasPrefix(url, "/files/") == false &&
|
|
url != "/login" && url != "/logout" && strings.HasPrefix(url, "/admin") == false && strings.HasPrefix(url, "/tags") == false {
|
|
NotFoundHandler(ctx, res, req)
|
|
return
|
|
}
|
|
ua := req.Header.Get("User-Agent")
|
|
if strings.Contains(ua, "MSIE ") || strings.Contains(ua, "Trident/") || strings.Contains(ua, "Edge/") {
|
|
// Microsoft is behaving on many occasion differently than Firefox / Chrome.
|
|
// I have neither the time / motivation for it to work properly
|
|
res.WriteHeader(http.StatusBadRequest)
|
|
res.Write([]byte(
|
|
Page(`
|
|
<h1>Internet explorer is not supported</h1>
|
|
<p>
|
|
We don't support IE / Edge at this time
|
|
<br>
|
|
Please use either Chromium, Firefox or Chrome
|
|
</p>
|
|
`)))
|
|
return
|
|
}
|
|
ServeFile(res, req, "/index.html")
|
|
}
|
|
|
|
func NotFoundHandler(ctx *App, res http.ResponseWriter, req *http.Request) {
|
|
if strings.Contains(req.Header.Get("accept"), "text/html") {
|
|
res.WriteHeader(http.StatusNotFound)
|
|
res.Write(HtmlPage404)
|
|
return
|
|
}
|
|
SendErrorResult(res, ErrNotFound)
|
|
}
|
|
|
|
var listOfPlugins map[string][]string = map[string][]string{
|
|
"oss": []string{},
|
|
"enterprise": []string{},
|
|
"custom": []string{},
|
|
}
|
|
|
|
func AboutHandler(ctx *App, res http.ResponseWriter, req *http.Request) {
|
|
t, _ := template.
|
|
New("about").
|
|
Funcs(map[string]interface{}{
|
|
"renderPlugin": func(lstr string, commit string) string {
|
|
if len(lstr) == 0 {
|
|
return "N/A"
|
|
} else if commit == "" {
|
|
return lstr
|
|
}
|
|
list := strings.Split(lstr, " ")
|
|
for i, _ := range list {
|
|
list[i] = `<a href="https://github.com/mickael-kerjean/filestash/tree/` + commit +
|
|
`/server/plugin/` + list[i] + `" target="_blank">` + list[i] + `</a>`
|
|
}
|
|
return strings.Join(list, " ")
|
|
},
|
|
}).
|
|
Parse(Page(`
|
|
<h1> {{ .Version }} </h1>
|
|
<table>
|
|
<tr> <td style="width:150px;"> Commit hash </td> <td> <a href="https://github.com/mickael-kerjean/filestash/tree/{{ .CommitHash }}">{{ .CommitHash }}</a> </td> </tr>
|
|
<tr> <td> Binary hash </td> <td> {{ index .Checksum 0}} </td> </tr>
|
|
<tr> <td> Config hash </td> <td> {{ index .Checksum 1}} </td> </tr>
|
|
<tr> <td> License </td> <td> {{ .License }} </td> </tr>
|
|
<tr>
|
|
<td> Plugins </td>
|
|
<td>
|
|
{{ $oss := (index .Plugins 0) }}
|
|
{{ $enterprise := (index .Plugins 1) }}
|
|
{{ $custom := (index .Plugins 2) }}
|
|
STANDARD[<span class="small">{{ renderPlugin (index .Plugins 0) .CommitHash }}</span>]
|
|
<br/>
|
|
EXTENDED[<span class="small">{{ renderPlugin (index .Plugins 1) "" }}</span>]
|
|
<br/>
|
|
CUSTOM[<span class="small">{{ renderPlugin (index .Plugins 2) "" }}</span>]
|
|
</td>
|
|
</tr>
|
|
</table>
|
|
|
|
<style>
|
|
table { margin: 0 auto; font-family: monospace; opacity: 0.8; max-width: 1000px; width: 95%;}
|
|
table td { text-align: right; padding-left: 10px; vertical-align: top; }
|
|
table td span.small { font-size:0.8rem; }
|
|
table a { color: inherit; text-decoration: none; }
|
|
</style>
|
|
`))
|
|
hashFileContent := func(path string, n int) string {
|
|
f, err := os.OpenFile(path, os.O_RDONLY, os.ModePerm)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
defer f.Close()
|
|
return HashStream(f, n)
|
|
}
|
|
t.Execute(res, struct {
|
|
Version string
|
|
CommitHash string
|
|
Checksum []string
|
|
License string
|
|
Plugins []string
|
|
}{
|
|
Version: fmt.Sprintf("Filestash %s.%s", APP_VERSION, BUILD_DATE),
|
|
CommitHash: BUILD_REF,
|
|
Checksum: []string{
|
|
hashFileContent(GetAbsolutePath("filestash"), 0),
|
|
hashFileContent(GetAbsolutePath(CONFIG_PATH, "config.json"), 0),
|
|
},
|
|
License: strings.ToUpper(LICENSE),
|
|
Plugins: []string{
|
|
strings.Join(listOfPlugins["oss"], " "),
|
|
strings.Join(listOfPlugins["enterprise"], " "),
|
|
strings.Join(listOfPlugins["custom"], " "),
|
|
},
|
|
})
|
|
}
|
|
|
|
func ManifestHandler(ctx *App, res http.ResponseWriter, req *http.Request) {
|
|
res.WriteHeader(http.StatusFound)
|
|
res.Write([]byte(fmt.Sprintf(`{
|
|
"name": "%s",
|
|
"short_name": "%s",
|
|
"icons": [
|
|
{
|
|
"src": "/assets/logo/android-chrome-192x192.png",
|
|
"type": "image/png",
|
|
"sizes": "192x192"
|
|
},
|
|
{
|
|
"src": "/assets/logo/android-chrome-512x512.png",
|
|
"type": "image/png",
|
|
"sizes": "512x512"
|
|
}
|
|
],
|
|
"theme_color": "#f2f3f5",
|
|
"background_color": "#f2f3f5",
|
|
"orientation": "any",
|
|
"display": "standalone",
|
|
"start_url": "/"
|
|
}`, Config.Get("general.name"), Config.Get("general.name"))))
|
|
}
|
|
|
|
func RobotsHandler(ctx *App, res http.ResponseWriter, req *http.Request) {
|
|
res.Write([]byte(""))
|
|
}
|
|
|
|
func InitPluginList(code []byte) {
|
|
listOfPackages := regexp.MustCompile(`github.com/mickael-kerjean/([^\"]+)`).FindAllString(string(code), -1)
|
|
for _, packageName := range listOfPackages {
|
|
if packageName == "github.com/mickael-kerjean/filestash/server/common" {
|
|
continue
|
|
}
|
|
|
|
if strings.HasPrefix(packageName, "github.com/mickael-kerjean/filestash/server/plugin/") {
|
|
listOfPlugins["oss"] = append(
|
|
listOfPlugins["oss"],
|
|
strings.TrimPrefix(packageName, "github.com/mickael-kerjean/filestash/server/plugin/"),
|
|
)
|
|
} else if strings.HasPrefix(packageName, "github.com/mickael-kerjean/filestash/filestash-enterprise/plugins/") {
|
|
listOfPlugins["enterprise"] = append(
|
|
listOfPlugins["enterprise"],
|
|
strings.TrimPrefix(packageName, "github.com/mickael-kerjean/filestash/filestash-enterprise/plugins/"),
|
|
)
|
|
} else if strings.HasPrefix(packageName, "github.com/mickael-kerjean/filestash/filestash-enterprise/customers/") {
|
|
listOfPlugins["custom"] = append(
|
|
listOfPlugins["custom"],
|
|
strings.TrimPrefix(packageName, "github.com/mickael-kerjean/filestash/filestash-enterprise/customers/"),
|
|
)
|
|
} else {
|
|
listOfPlugins["custom"] = append(
|
|
listOfPlugins["custom"],
|
|
packageName,
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
func CustomCssHandler(ctx *App, res http.ResponseWriter, req *http.Request) {
|
|
res.Header().Set("Content-Type", "text/css")
|
|
io.WriteString(res, Hooks.Get.CSS())
|
|
io.WriteString(res, Config.Get("general.custom_css").String())
|
|
}
|
|
|
|
func ServeFile(res http.ResponseWriter, req *http.Request, filePath string) {
|
|
staticConfig := []struct {
|
|
ContentType string
|
|
FileExt string
|
|
}{
|
|
{"br", ".br"},
|
|
{"gzip", ".gz"},
|
|
{"", ""},
|
|
}
|
|
|
|
head := res.Header()
|
|
acceptEncoding := req.Header.Get("Accept-Encoding")
|
|
for _, cfg := range staticConfig {
|
|
if strings.Contains(acceptEncoding, cfg.ContentType) == false {
|
|
continue
|
|
}
|
|
curPath := filePath + cfg.FileExt
|
|
file, err := WWWEmbed.Open("static/www" + curPath)
|
|
if env := os.Getenv("NODE_ENV"); env == "development" {
|
|
file, err = WWWDir.Open("server/ctrl/static/www" + curPath)
|
|
}
|
|
if err != nil {
|
|
continue
|
|
} else if stat, err := file.Stat(); err == nil {
|
|
etag := QuickHash(fmt.Sprintf(
|
|
"%s %d %d %s",
|
|
curPath, stat.Size(), stat.Mode(), stat.ModTime()), 10,
|
|
)
|
|
if etag == req.Header.Get("If-None-Match") {
|
|
res.WriteHeader(http.StatusNotModified)
|
|
return
|
|
}
|
|
head.Set("Etag", etag)
|
|
}
|
|
if cfg.ContentType != "" {
|
|
head.Set("Content-Encoding", cfg.ContentType)
|
|
}
|
|
io.Copy(res, file)
|
|
file.Close()
|
|
return
|
|
}
|
|
http.NotFound(res, req)
|
|
}
|