mirror of
https://github.com/go-shiori/shiori.git
synced 2025-10-02 09:46:59 +08:00
99 lines
2.9 KiB
Go
99 lines
2.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/go-shiori/shiori/internal/config"
|
|
"github.com/go-shiori/shiori/internal/http"
|
|
"github.com/go-shiori/shiori/internal/model"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/pflag"
|
|
)
|
|
|
|
func newServerCommand() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "server",
|
|
Short: "Starts the Shiori webserver",
|
|
Long: "Serves the Shiori web interface and API.",
|
|
Run: newServerCommandHandler(),
|
|
}
|
|
|
|
cmd.Flags().IntP("port", "p", 8080, "Port used by the server")
|
|
cmd.Flags().StringP("address", "a", "", "Address the server listens to")
|
|
cmd.Flags().StringP("webroot", "r", "/", "Root path that used by server")
|
|
cmd.Flags().Bool("access-log", false, "Print out a non-standard access log")
|
|
cmd.Flags().Bool("serve-web-ui", true, "Serve static files from the webroot path")
|
|
cmd.Flags().String("secret-key", "", "Secret key used for encrypting session data")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func setIfFlagChanged(flagName string, flags *pflag.FlagSet, cfg *config.Config, fn func(cfg *config.Config)) {
|
|
if flags.Changed(flagName) {
|
|
fn(cfg)
|
|
}
|
|
}
|
|
|
|
func newServerCommandHandler() func(cmd *cobra.Command, args []string) {
|
|
return func(cmd *cobra.Command, args []string) {
|
|
ctx := context.Background()
|
|
|
|
// Get flags values
|
|
port, _ := cmd.Flags().GetInt("port")
|
|
address, _ := cmd.Flags().GetString("address")
|
|
rootPath, _ := cmd.Flags().GetString("webroot")
|
|
accessLog, _ := cmd.Flags().GetBool("access-log")
|
|
serveWebUI, _ := cmd.Flags().GetBool("serve-web-ui")
|
|
secretKey, _ := cmd.Flags().GetBytesHex("secret-key")
|
|
|
|
cfg, dependencies := initShiori(ctx, cmd)
|
|
|
|
// Validate root path
|
|
if rootPath == "" {
|
|
rootPath = "/"
|
|
}
|
|
|
|
if !strings.HasPrefix(rootPath, "/") {
|
|
rootPath = "/" + rootPath
|
|
}
|
|
|
|
if !strings.HasSuffix(rootPath, "/") {
|
|
rootPath += "/"
|
|
}
|
|
|
|
// Override configuration from flags if needed
|
|
setIfFlagChanged("port", cmd.Flags(), cfg, func(cfg *config.Config) {
|
|
cfg.Http.Port = port
|
|
})
|
|
setIfFlagChanged("address", cmd.Flags(), cfg, func(cfg *config.Config) {
|
|
cfg.Http.Address = address + ":"
|
|
})
|
|
setIfFlagChanged("webroot", cmd.Flags(), cfg, func(cfg *config.Config) {
|
|
cfg.Http.RootPath = rootPath
|
|
})
|
|
setIfFlagChanged("access-log", cmd.Flags(), cfg, func(cfg *config.Config) {
|
|
cfg.Http.AccessLog = accessLog
|
|
})
|
|
setIfFlagChanged("serve-web-ui", cmd.Flags(), cfg, func(cfg *config.Config) {
|
|
cfg.Http.ServeWebUI = serveWebUI
|
|
})
|
|
setIfFlagChanged("secret-key", cmd.Flags(), cfg, func(cfg *config.Config) {
|
|
cfg.Http.SecretKey = secretKey
|
|
})
|
|
|
|
dependencies.Log.Infof("Starting Shiori v%s", model.BuildVersion)
|
|
|
|
server, err := http.NewHttpServer(dependencies.Log).Setup(cfg, dependencies)
|
|
if err != nil {
|
|
dependencies.Log.WithError(err).Fatal("error setting up server")
|
|
}
|
|
|
|
if err := server.Start(ctx); err != nil {
|
|
dependencies.Log.WithError(err).Fatal("error starting server")
|
|
}
|
|
dependencies.Log.WithField("addr", address).Debug("started http server")
|
|
|
|
server.WaitStop(ctx)
|
|
}
|
|
}
|