schnutibox/cmd/root.go

//nolint:exhaustivestruct,gochecknoglobals,gochecknoinits,gomnd
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.
//
// nolint:funlen
func init() {
	// Root.
	rootCmd.PersistentFlags().Bool("pprof", false, "Enables pprof for debugging")

	// 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")
	}

	runCmd.Flags().Bool("ignore-reader", false, "Ignoring that the reader is missing")

	// 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")
	}

	// Timer.
	rootCmd.AddCommand(timerCmd)

	// 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")
	viper.SetDefault("reader.ignore", false)
	viper.SetDefault("pprof", false)

	// Environment handling.
	viper.SetEnvPrefix("SCHNUTIBOX")
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
	viper.AutomaticEnv()

	// Flags.
	if err := viper.BindPFlag("pprof", rootCmd.PersistentFlags().Lookup("pprof")); err != nil {
		log.Fatal().Err(err).Msg("could not bind flag")
	}

	if err := viper.BindPFlag("reader.dev", prepareCmd.Flags().Lookup("rfid-reader")); err != nil {
		log.Fatal().Err(err).Msg("could not bind flag")
	}

	if err := viper.BindPFlag("reader.ignore", runCmd.Flags().Lookup("ignore-reader")); err != nil {
		log.Fatal().Err(err).Msg("could not bind 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()

	// 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)
	}
}