141 lines
3.9 KiB
Go
141 lines
3.9 KiB
Go
//nolint:exhaustivestruct,gochecknoglobals,gochecknoinits
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/fsnotify/fsnotify"
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"go.xsfx.dev/schnutibox/internal/config"
|
|
"go.xsfx.dev/schnutibox/pkg/prepare"
|
|
)
|
|
|
|
var cfgFile string
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "schnutibox",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if err := cmd.Usage(); err != nil {
|
|
log.Error().Msg(err.Error())
|
|
}
|
|
},
|
|
}
|
|
|
|
// init initializes the command line interface.
|
|
func init() {
|
|
// Run.
|
|
rootCmd.AddCommand(runCmd)
|
|
runCmd.Flags().StringVarP(&cfgFile, "config", "c", "", "config file")
|
|
|
|
if err := runCmd.MarkFlagRequired("config"); err != nil {
|
|
log.Fatal().Err(err).Msg("missing flag")
|
|
}
|
|
|
|
// Prepare.
|
|
rootCmd.AddCommand(prepareCmd)
|
|
prepareCmd.Flags().BoolVar(&prepare.Cfg.ReadOnly, "read-only", false, "Setup read-only system")
|
|
prepareCmd.Flags().StringVarP(&prepare.Cfg.System, "system", "s", "raspbian", "Which kind of system to prepare")
|
|
prepareCmd.Flags().StringVar(&prepare.Cfg.SpotifyUsername, "spotify-username", "", "Spotify username")
|
|
prepareCmd.Flags().StringVar(&prepare.Cfg.SpotifyPassword, "spotify-password", "", "Spotify password")
|
|
prepareCmd.Flags().StringVar(&prepare.Cfg.SpotifyClientID, "spotify-client-id", "", "Spotify client ID")
|
|
prepareCmd.Flags().StringVar(&prepare.Cfg.SpotifyClientSecret, "spotify-client-secret", "", "Spotify client secret")
|
|
prepareCmd.Flags().StringVar(&prepare.Cfg.RFIDReader, "rfid-reader", "/dev/hidraw0", "dev path of rfid reader")
|
|
prepareCmd.Flags().StringVar(&prepare.Cfg.StopID, "stop-id", "", "ID of stop tag")
|
|
|
|
// Version.
|
|
rootCmd.AddCommand(versionCmd)
|
|
|
|
// Web.
|
|
rootCmd.AddCommand(webCmd)
|
|
webCmd.Flags().StringVarP(&cfgFile, "config", "c", "", "config file")
|
|
|
|
if err := webCmd.MarkFlagRequired("config"); err != nil {
|
|
log.Fatal().Err(err).Msg("missing flag")
|
|
}
|
|
}
|
|
|
|
// initConfig loads the config file.
|
|
// fatal defines if config parsing should end in a fatal error or not.
|
|
func initConfig(fatal bool) {
|
|
logger := log.With().Str("config", cfgFile).Logger()
|
|
|
|
// Defaults.
|
|
viper.SetDefault("box.hostname", "localhost")
|
|
viper.SetDefault("box.port", 9999)
|
|
viper.SetDefault("box.grpc", 9998)
|
|
viper.SetDefault("mpd.hostname", "localhost")
|
|
viper.SetDefault("mpd.port", 6600)
|
|
viper.SetDefault("reader.dev", "/dev/hidraw0")
|
|
|
|
// Environment handling.
|
|
viper.SetEnvPrefix("SCHNUTIBOX")
|
|
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
|
|
viper.AutomaticEnv()
|
|
|
|
// Flags.
|
|
if err := viper.BindPFlag("reader.dev", prepareCmd.Flags().Lookup("rfid-reader")); err != nil {
|
|
logger.Fatal().Err(err).Msg("could not bind flag")
|
|
}
|
|
|
|
// Parse config file.
|
|
if cfgFile != "" {
|
|
viper.SetConfigFile(cfgFile)
|
|
parseConfig(logger, fatal)
|
|
} else {
|
|
logger.Fatal().Msg("missing config file")
|
|
}
|
|
|
|
viper.WatchConfig()
|
|
viper.OnConfigChange(func(e fsnotify.Event) {
|
|
logger.Info().Msg("config file changed")
|
|
parseConfig(logger, false)
|
|
})
|
|
}
|
|
|
|
// parseConfig parses the config and does some tests if required fields are there.
|
|
// Its also possible to decide if parsing should end up in a fatal or just an error.
|
|
func parseConfig(logger zerolog.Logger, fatal bool) {
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
if fatal {
|
|
logger.Fatal().Err(err).Msg("error loading config file")
|
|
}
|
|
|
|
logger.Error().Err(err).Msg("error loading config file")
|
|
|
|
return
|
|
}
|
|
|
|
if err := viper.Unmarshal(&config.Cfg); err != nil {
|
|
if fatal {
|
|
logger.Fatal().Err(err).Msg("could not unmarshal config")
|
|
}
|
|
|
|
logger.Error().Err(err).Msg("could not unmarshal config")
|
|
|
|
return
|
|
}
|
|
|
|
if err := config.Cfg.Require(); err != nil {
|
|
if fatal {
|
|
logger.Fatal().Err(err).Msg("missing config parts")
|
|
}
|
|
|
|
logger.Error().Err(err).Msg("missing config parts")
|
|
|
|
return
|
|
}
|
|
}
|
|
|
|
// Execute executes the commandline interface.
|
|
func Execute() {
|
|
if err := rootCmd.Execute(); err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|