103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/creasty/defaults"
|
|
"github.com/pelletier/go-toml/v2"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
var ALWAYS_BLACKLISTED_PATHS = []string{
|
|
"/.well-known/acme-challenge/",
|
|
}
|
|
|
|
func NewDefaultConfig() Config {
|
|
config := Config{}
|
|
if err := defaults.Set(&config); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// defaults does not support setting arrays from strings
|
|
config.Server.PagesBranches = []string{"main", "master", "pages"}
|
|
|
|
return config
|
|
}
|
|
|
|
func ReadConfig(ctx *cli.Context) (*Config, error) {
|
|
config := NewDefaultConfig()
|
|
// if config is not given as argument return empty config
|
|
if !ctx.IsSet("config-file") {
|
|
return &config, nil
|
|
}
|
|
|
|
configFile := path.Clean(ctx.String("config-file"))
|
|
|
|
log.Debug().Str("config-file", configFile).Msg("reading config file")
|
|
content, err := os.ReadFile(configFile)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = toml.Unmarshal(content, &config)
|
|
return &config, err
|
|
}
|
|
|
|
func MergeConfig(ctx *cli.Context, config *Config) {
|
|
if ctx.IsSet("log-level") {
|
|
config.LogLevel = ctx.String("log-level")
|
|
}
|
|
|
|
mergeServerConfig(ctx, &config.Server)
|
|
mergeForgeConfig(ctx, &config.Forge)
|
|
}
|
|
|
|
func mergeServerConfig(ctx *cli.Context, config *ServerConfig) {
|
|
if ctx.IsSet("host") {
|
|
config.Host = ctx.String("host")
|
|
}
|
|
if ctx.IsSet("port") {
|
|
config.Port = uint16(ctx.Uint("port"))
|
|
}
|
|
if ctx.IsSet("pages-domain") {
|
|
config.MainDomain = ctx.String("pages-domain")
|
|
}
|
|
if ctx.IsSet("raw-domain") {
|
|
config.RawDomain = ctx.String("raw-domain")
|
|
}
|
|
if ctx.IsSet("pages-branch") {
|
|
config.PagesBranches = ctx.StringSlice("pages-branch")
|
|
}
|
|
if ctx.IsSet("allowed-cors-domains") {
|
|
config.AllowedCorsDomains = ctx.StringSlice("allowed-cors-domains")
|
|
}
|
|
if ctx.IsSet("blacklisted-paths") {
|
|
config.BlacklistedPaths = ctx.StringSlice("blacklisted-paths")
|
|
}
|
|
|
|
// add the paths that should always be blacklisted
|
|
config.BlacklistedPaths = append(config.BlacklistedPaths, ALWAYS_BLACKLISTED_PATHS...)
|
|
}
|
|
|
|
func mergeForgeConfig(ctx *cli.Context, config *ForgeConfig) {
|
|
if ctx.IsSet("forge-root") {
|
|
config.Root = ctx.String("forge-root")
|
|
}
|
|
if ctx.IsSet("forge-api-token") {
|
|
config.Token = ctx.String("forge-api-token")
|
|
}
|
|
if ctx.IsSet("enable-lfs-support") {
|
|
config.LFSEnabled = ctx.Bool("enable-lfs-support")
|
|
}
|
|
if ctx.IsSet("enable-symlink-support") {
|
|
config.FollowSymlinks = ctx.Bool("enable-symlink-support")
|
|
}
|
|
if ctx.IsSet("default-mime-type") {
|
|
config.DefaultMimeType = ctx.String("default-mime-type")
|
|
}
|
|
if ctx.IsSet("forbidden-mime-types") {
|
|
config.ForbiddenMimeTypes = ctx.StringSlice("forbidden-mime-types")
|
|
}
|
|
}
|