mirror of
https://codeberg.org/shroff/phylum.git
synced 2026-02-05 11:09:52 -06:00
120 lines
3.7 KiB
Go
120 lines
3.7 KiB
Go
package serve
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/fvbock/endless"
|
|
"github.com/gin-contrib/static"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/shroff/phylum/server/internal/api/publink"
|
|
apiv1 "github.com/shroff/phylum/server/internal/api/v1"
|
|
"github.com/shroff/phylum/server/internal/api/webdav"
|
|
"github.com/shroff/phylum/server/internal/core/fs"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
const trashRetainDuration = 30 * 24 * time.Hour
|
|
|
|
func SetupCommand() *cobra.Command {
|
|
var cmd = &cobra.Command{
|
|
Use: "serve",
|
|
Short: "Run the server",
|
|
GroupID: "server",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
config := viper.Sub("server")
|
|
|
|
flags := cmd.Flags()
|
|
// TODO: Hack. flag bindings don't work in viper for nested keys
|
|
config.BindPFlag("host", flags.Lookup("host"))
|
|
config.BindPFlag("port", flags.Lookup("port"))
|
|
config.BindPFlag("cors.enabled", flags.Lookup("cors-enabled"))
|
|
config.BindPFlag("cors.origins", flags.Lookup("cors-origins"))
|
|
config.BindPFlag("log_body", flags.Lookup("log-body"))
|
|
config.BindPFlag("web_app_src", flags.Lookup("web-app-src"))
|
|
config.BindPFlag("publink_path", flags.Lookup("publink-path"))
|
|
config.BindPFlag("webdav_path", flags.Lookup("webdav-path"))
|
|
|
|
if !viper.GetBool("debug") {
|
|
gin.SetMode(gin.ReleaseMode)
|
|
}
|
|
engine := createEngine(config)
|
|
|
|
webdav.SetupHandler(engine.Group(config.GetString("webdav_path")))
|
|
apiv1.Setup(engine.Group("/api/v1"))
|
|
publink.Setup(engine.Group(config.GetString("publink_path")))
|
|
|
|
webAppSrcDir := config.GetString("web_app_src")
|
|
if webAppSrcDir != "" {
|
|
setupWebApp(engine, webAppSrcDir)
|
|
}
|
|
setupTrashCompactor()
|
|
|
|
listen := config.GetString("host") + ":" + config.GetString("port")
|
|
server := endless.NewServer(listen, engine)
|
|
server.BeforeBegin = func(addr string) {
|
|
logrus.Info("Listening on " + addr)
|
|
}
|
|
|
|
if err := server.ListenAndServe(); err != nil {
|
|
logrus.Fatal(err.Error())
|
|
}
|
|
},
|
|
}
|
|
flags := cmd.Flags()
|
|
|
|
flags.String("host", "", "Server Host")
|
|
flags.String("port", "2448", "Server Port")
|
|
flags.Bool("cors-enabled", false, "Enable CORS")
|
|
flags.StringSlice("cors-origins", []string{"*"}, "CORS origins")
|
|
flags.Bool("log-body", false, "Log Response Body (Must be used with --debug)")
|
|
flags.String("web-app-src", "web", "Web App Source Directory")
|
|
flags.String("publink-path", "/publink", "Public Share path prefix")
|
|
flags.String("webdav-path", "/webdav", "WebDAV path prefix")
|
|
|
|
// Must be specified otherwise env variables won't be parsed
|
|
viper.SetDefault("server.host", "")
|
|
viper.SetDefault("server.port", "2448")
|
|
viper.SetDefault("server.log_body", false)
|
|
viper.SetDefault("server.cors.enabled", false)
|
|
viper.SetDefault("server.cors.origins", []string{"*"})
|
|
viper.SetDefault("server.web_app_src", "web")
|
|
viper.SetDefault("server.publink_path", "/publink")
|
|
viper.SetDefault("server.webdav_path", "/webdav")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func setupWebApp(r gin.IRoutes, webAppSrcDir string) {
|
|
fs := static.LocalFile(webAppSrcDir, false)
|
|
fileserver := http.FileServer(fs)
|
|
indexFilePath := path.Join(webAppSrcDir, "index.html")
|
|
|
|
staticFileHandler := func(c *gin.Context) {
|
|
if fs.Exists("/", c.Request.URL.Path) {
|
|
fileserver.ServeHTTP(c.Writer, c.Request)
|
|
} else {
|
|
http.ServeFile(c.Writer, c.Request, indexFilePath)
|
|
}
|
|
}
|
|
r.Use(func(c *gin.Context) {
|
|
c.Writer.Header().Set("Cross-Origin-Embedder-Policy", "credentialless")
|
|
c.Writer.Header().Set("Cross-Origin-Opener-Policy", "same-origin")
|
|
}, staticFileHandler)
|
|
}
|
|
|
|
func setupTrashCompactor() {
|
|
ticker := time.NewTimer(24 * time.Hour)
|
|
go func() {
|
|
for {
|
|
<-ticker.C
|
|
fs.TrashCompact(context.Background(), trashRetainDuration)
|
|
}
|
|
}()
|
|
fs.TrashCompact(context.Background(), trashRetainDuration)
|
|
}
|