main.go

// package main is the entry point for the cherry-pick tool
package main

import (
	"log/slog"
	"os"

	configcmd "github.com/alan/cherry-picker/cmd/config"
	fetchcmd "github.com/alan/cherry-picker/cmd/fetch"
	"github.com/alan/cherry-picker/cmd/merge"
	"github.com/alan/cherry-picker/cmd/pick"
	"github.com/alan/cherry-picker/cmd/retry"
	"github.com/alan/cherry-picker/cmd/status"
	"github.com/alan/cherry-picker/cmd/summary"
	"github.com/alan/cherry-picker/internal/config"
	"github.com/spf13/cobra"
)

func main() {
	var configFile string
	var logLevel string
	var logFormat string

	rootCmd := &cobra.Command{
		Use:   "cherry-picker",
		Short: "A CLI tool for managing cherry-picks across GitHub repositories",
		Long: `cherry-picker is a CLI tool that helps manage cherry-picking commits
across GitHub repositories using a YAML configuration file to track state.`,
		PersistentPreRun: func(_ *cobra.Command, _ []string) {
			setupLogger(logLevel, logFormat)
		},
	}

	// Add global flags
	rootCmd.PersistentFlags().StringVarP(&configFile, "config", "c", "cherry-picks.yaml", "Configuration file path")
	rootCmd.PersistentFlags().StringVarP(&logLevel, "log-level", "l", "info", "Log level (debug, info, warn, error)")
	rootCmd.PersistentFlags().StringVarP(&logFormat, "log-format", "f", "text", "Log format (text, json)")

	// Create commands with access to the global config file
	rootCmd.AddCommand(configcmd.NewConfigCmd(&configFile, config.LoadConfig, config.SaveConfig))
	rootCmd.AddCommand(fetchcmd.NewFetchCmd(&configFile, config.LoadConfig, config.SaveConfig))
	rootCmd.AddCommand(status.NewStatusCmd(&configFile, config.LoadConfig, config.SaveConfig))
	rootCmd.AddCommand(pick.NewPickCmd(&configFile, config.LoadConfig, config.SaveConfig))
	rootCmd.AddCommand(retry.NewRetryCmd(&configFile, config.LoadConfig, config.SaveConfig))
	rootCmd.AddCommand(merge.NewMergeCmd(&configFile, config.LoadConfig, config.SaveConfig))
	rootCmd.AddCommand(summary.NewSummaryCmd(&configFile, config.LoadConfig))

	if err := rootCmd.Execute(); err != nil {
		os.Exit(1)
	}
}

func setupLogger(level, format string) {
	var logLevel slog.Level
	switch level {
	case "debug":
		logLevel = slog.LevelDebug
	case "info":
		logLevel = slog.LevelInfo
	case "warn":
		logLevel = slog.LevelWarn
	case "error":
		logLevel = slog.LevelError
	default:
		logLevel = slog.LevelInfo
	}

	var handler slog.Handler
	if format == "json" {
		handler = slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: logLevel})
	} else {
		handler = slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: logLevel})
	}

	slog.SetDefault(slog.New(handler))
}