steamcache2/cmd/root.go

// cmd/root.go
package cmd

import (
	"fmt"
	"os"
	"runtime"
	"s1d3sw1ped/SteamCache2/config"
	"s1d3sw1ped/SteamCache2/steamcache"
	"s1d3sw1ped/SteamCache2/steamcache/logger"
	"s1d3sw1ped/SteamCache2/version"
	"strings"

	"github.com/rs/zerolog"
	"github.com/spf13/cobra"
)

var (
	threads    int
	configPath string

	logLevel  string
	logFormat string
)

var rootCmd = &cobra.Command{
	Use:   "SteamCache2",
	Short: "SteamCache2 is a caching solution for Steam game updates and installations",
	Long: `SteamCache2 is a caching solution designed to optimize the delivery of Steam game updates and installations. 
	It reduces bandwidth usage and speeds up the download process by caching game files locally. 
	This tool is particularly useful for environments with multiple Steam users, such as gaming cafes or households with multiple gamers. 
	By caching game files, SteamCache2 ensures that subsequent downloads of the same files are served from the local cache, 
	significantly improving download times and reducing the load on the internet connection.`,
	Run: func(cmd *cobra.Command, args []string) {
		// Configure logging
		switch logLevel {
		case "debug":
			zerolog.SetGlobalLevel(zerolog.DebugLevel)
		case "error":
			zerolog.SetGlobalLevel(zerolog.ErrorLevel)
		case "info":
			zerolog.SetGlobalLevel(zerolog.InfoLevel)
		default:
			zerolog.SetGlobalLevel(zerolog.InfoLevel) // Default to info level if not specified
		}
		var writer zerolog.ConsoleWriter
		if logFormat == "json" {
			writer = zerolog.ConsoleWriter{Out: os.Stderr, NoColor: true}
		} else {
			writer = zerolog.ConsoleWriter{Out: os.Stderr}
		}
		logger.Logger = zerolog.New(writer).With().Timestamp().Logger()

		logger.Logger.Info().
			Msg("SteamCache2 " + version.Version + " " + version.Date + " starting...")

		// Load configuration
		cfg, err := config.LoadConfig(configPath)
		if err != nil {
			// Check if the error is because the config file doesn't exist
			// The error is wrapped, so we check the error message
			if strings.Contains(err.Error(), "no such file") ||
				strings.Contains(err.Error(), "cannot find the file") ||
				strings.Contains(err.Error(), "The system cannot find the file") {
				logger.Logger.Info().
					Str("config_path", configPath).
					Msg("Config file not found, creating default configuration")

				if err := config.SaveDefaultConfig(configPath); err != nil {
					logger.Logger.Error().
						Err(err).
						Str("config_path", configPath).
						Msg("Failed to create default configuration")
					fmt.Fprintf(os.Stderr, "Error: Failed to create default config at %s: %v\n", configPath, err)
					os.Exit(1)
				}

				logger.Logger.Info().
					Str("config_path", configPath).
					Msg("Default configuration created successfully. Please edit the file and run again.")

				fmt.Printf("Default configuration created at %s\n", configPath)
				fmt.Println("Please edit the configuration file as needed and run the application again.")
				os.Exit(0)
			} else {
				logger.Logger.Error().
					Err(err).
					Str("config_path", configPath).
					Msg("Failed to load configuration")
				fmt.Fprintf(os.Stderr, "Error: Failed to load configuration from %s: %v\n", configPath, err)
				os.Exit(1)
			}
		}

		logger.Logger.Info().
			Str("config_path", configPath).
			Msg("Configuration loaded successfully")

		if runtime.GOMAXPROCS(-1) != threads {
			runtime.GOMAXPROCS(threads)
			logger.Logger.Info().
				Int("threads", threads).
				Msg("Maximum number of threads set")
		}

		sc := steamcache.New(
			cfg.ListenAddress,
			cfg.Cache.Memory.Size,
			cfg.Cache.Disk.Size,
			cfg.Cache.Disk.Path,
			cfg.Upstream,
			cfg.Cache.Memory.GCAlgorithm,
			cfg.Cache.Disk.GCAlgorithm,
		)

		logger.Logger.Info().
			Msg("SteamCache2 " + version.Version + " started on " + cfg.ListenAddress)

		sc.Run()

		logger.Logger.Info().Msg("SteamCache2 stopped")
		os.Exit(0)
	},
}

// Execute adds all child commands to the root command and sets flags appropriately.
// This is called by main.main(). It only needs to happen once to the rootCmd.
func Execute() {
	err := rootCmd.Execute()
	if err != nil {
		os.Exit(1)
	}
}

func init() {
	rootCmd.Flags().StringVarP(&configPath, "config", "c", "config.yaml", "Path to configuration file")

	rootCmd.Flags().IntVarP(&threads, "threads", "t", runtime.GOMAXPROCS(-1), "Number of worker threads to use for processing requests")

	rootCmd.Flags().StringVarP(&logLevel, "log-level", "l", "info", "Logging level: debug, info, error")
	rootCmd.Flags().StringVarP(&logFormat, "log-format", "f", "console", "Logging format: json, console")
}