mirror of
https://github.com/opencloud-eu/opencloud.git
synced 2026-01-24 13:58:12 -05:00
b6d43661a8
So far Phoenix have been embedded into the binary and we can customize the required config based on flags, env variables and optionally via config file. For now I'm embedding the whole Phonix content, optherwise the all-in-one binary `ocis` will get pretty complicated until we add the generate commands to that repo as well and provide a mechanism to inject the embedding.
105 lines
2.6 KiB
Go
105 lines
2.6 KiB
Go
package command
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/owncloud/ocis-phoenix/pkg/version"
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// Root is the entry point for the ocis-phoenix command.
|
|
func Root() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "ocis-phoenix",
|
|
Short: "Reva service for phoenix",
|
|
Long: ``,
|
|
Version: version.String,
|
|
PersistentPreRun: func(cmd *cobra.Command, args []string) {
|
|
setupLogger()
|
|
setupConfig()
|
|
},
|
|
}
|
|
|
|
cmd.PersistentFlags().String("log-level", "", "Set logging level")
|
|
viper.BindPFlag("log.level", cmd.PersistentFlags().Lookup("log-level"))
|
|
viper.SetDefault("log.level", "info")
|
|
viper.BindEnv("log.level", "PHOENIX_LOG_LEVEL")
|
|
|
|
cmd.PersistentFlags().Bool("log-pretty", false, "Enable pretty logging")
|
|
viper.BindPFlag("log.pretty", cmd.PersistentFlags().Lookup("log-pretty"))
|
|
viper.SetDefault("log.pretty", true)
|
|
viper.BindEnv("log.pretty", "PHOENIX_LOG_PRETTY")
|
|
|
|
cmd.PersistentFlags().Bool("log-color", false, "Enable colored logging")
|
|
viper.BindPFlag("log.color", cmd.PersistentFlags().Lookup("log-color"))
|
|
viper.SetDefault("log.color", true)
|
|
viper.BindEnv("log.color", "PHOENIX_LOG_COLOR")
|
|
|
|
cmd.AddCommand(Server())
|
|
cmd.AddCommand(Health())
|
|
|
|
return cmd
|
|
}
|
|
|
|
// setupLogger prepares the logger.
|
|
func setupLogger() {
|
|
switch strings.ToLower(viper.GetString("log.level")) {
|
|
case "panic":
|
|
zerolog.SetGlobalLevel(zerolog.PanicLevel)
|
|
case "fatal":
|
|
zerolog.SetGlobalLevel(zerolog.FatalLevel)
|
|
case "error":
|
|
zerolog.SetGlobalLevel(zerolog.ErrorLevel)
|
|
case "warn":
|
|
zerolog.SetGlobalLevel(zerolog.WarnLevel)
|
|
case "info":
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
case "debug":
|
|
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
|
default:
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
}
|
|
|
|
if viper.GetBool("log.pretty") {
|
|
log.Logger = log.Output(
|
|
zerolog.ConsoleWriter{
|
|
Out: os.Stderr,
|
|
NoColor: !viper.GetBool("log.color"),
|
|
},
|
|
)
|
|
}
|
|
}
|
|
|
|
// setupConfig prepares the config
|
|
func setupConfig() {
|
|
viper.SetConfigName("phoenix")
|
|
|
|
viper.AddConfigPath("/etc/ocis")
|
|
viper.AddConfigPath("$HOME/.ocis")
|
|
viper.AddConfigPath("./config")
|
|
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
switch err.(type) {
|
|
case viper.ConfigFileNotFoundError:
|
|
log.Debug().
|
|
Msg("Initializing without config file")
|
|
case viper.UnsupportedConfigError:
|
|
log.Fatal().
|
|
Msg("Unsupported config type")
|
|
default:
|
|
if e := log.Debug(); e.Enabled() {
|
|
log.Fatal().
|
|
Err(err).
|
|
Msg("Failed to read config")
|
|
} else {
|
|
log.Fatal().
|
|
Msg("Failed to read config")
|
|
}
|
|
}
|
|
}
|
|
}
|