Rename service package to command

2019-09-04 10:18:30 +02:00
parent 07a356c649
commit 66facde60a
5 changed files with 14 additions and 14 deletions
@@ -0,0 +1,102 @@
+package command
+
+import (
+	"os"
+	"strings"
+
+	"github.com/owncloud/reva-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 reva-phoenix command.
+func Root() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "reva-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
+}
+
+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"),
+			},
+		)
+	}
+}
+
+func setupConfig() {
+	viper.SetConfigName("phoenix")
+
+	viper.AddConfigPath("/etc/reva")
+	viper.AddConfigPath("$HOME/.reva")
+	viper.AddConfigPath("./config")
+
+	if err := viper.ReadInConfig(); err != nil {
+		switch err.(type) {
+		case viper.ConfigFileNotFoundError:
+			log.Debug().
+				Msg("Continue without config")
+		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")
+			}
+		}
+	}
+}