pmg/main.go

package main

import (
	"fmt"
	"os"

	"github.com/safedep/dry/log"
	"github.com/safedep/pmg/cmd/executors"
	"github.com/safedep/pmg/cmd/npm"
	"github.com/safedep/pmg/cmd/pypi"
	"github.com/safedep/pmg/cmd/setup"
	"github.com/safedep/pmg/cmd/version"
	"github.com/safedep/pmg/config"
	"github.com/safedep/pmg/internal/analytics"
	"github.com/safedep/pmg/internal/audit"
	"github.com/safedep/pmg/internal/eventlog"
	"github.com/safedep/pmg/internal/ui"
	appVersion "github.com/safedep/pmg/internal/version"
	"github.com/spf13/cobra"
)

var (
	debug   bool
	silent  bool
	verbose bool
	logFile string
)

func main() {
	cmd := &cobra.Command{
		Use:              "pmg",
		TraverseChildren: true,
		PersistentPreRun: func(cmd *cobra.Command, args []string) {
			// Always set this first because we will override the log
			// level if debug or verbose is set
			if logFile != "" {
				os.Setenv("APP_LOG_FILE", logFile)
				os.Setenv("APP_LOG_LEVEL", "info")
			}

			// Set the log level when debug is enabled
			if debug {
				os.Setenv("APP_LOG_LEVEL", "debug")
			}

			// Skip stdout logging when debugging is not enabled
			if !debug {
				os.Setenv("APP_LOG_SKIP_STDOUT_LOGGER", "true")
			}

			// Apply config-based verbosity first
			switch config.Get().Config.Verbosity {
			case config.VerbositySilent:
				ui.SetVerbosityLevel(ui.VerbosityLevelSilent)
			case config.VerbosityVerbose:
				ui.SetVerbosityLevel(ui.VerbosityLevelVerbose)
			default:
				ui.SetVerbosityLevel(ui.VerbosityLevelNormal)
			}

			// CLI flags override config
			if silent && verbose {
				ui.Fatalf("pmg: --silent and --verbose cannot be used together")
			}

			if silent {
				ui.SetVerbosityLevel(ui.VerbosityLevelSilent)
			} else if verbose {
				ui.SetVerbosityLevel(ui.VerbosityLevelVerbose)
			}

			log.InitZapLogger("pmg", "cli")

			// Initialize event logging (silently fail if it can't be initialized)
			var eventlogErr error
			if logFile != "" {
				// If a custom log file is specified, use it for event logging too
				eventlogErr = eventlog.InitializeWithFile(logFile)
			} else {
				// Otherwise use the default log directory
				eventlogErr = eventlog.Initialize()
			}

			if eventlogErr != nil {
				ui.Fatalf("failed to initialize event logging: %v", eventlogErr)
			}

			if err := audit.Initialize(); err != nil {
				log.Warnf("failed to initialize audit system: %v", err)
			}

			config.FinalizeDependencyCooldownOverride()

			// Parse and validate --sandbox-allow flags after all flags are resolved
			if err := config.FinalizeSandboxAllowOverrides(); err != nil {
				ui.Fatalf("pmg: %v", err)
			}
		},
		RunE: func(cmd *cobra.Command, args []string) error {
			if len(args) == 0 {
				return cmd.Help()
			}

			return fmt.Errorf("pmg: %s is not a valid command", args[0])
		},
	}

	cmd.PersistentFlags().StringVar(&logFile, "log", "", "Log file to write to")
	cmd.PersistentFlags().BoolVar(&silent, "silent", false, "Silent mode for invisible experience")
	cmd.PersistentFlags().BoolVar(&verbose, "verbose", false, "Verbose mode for more information")
	cmd.PersistentFlags().BoolVar(&debug, "debug", false, "Enable debug logging (defaults to stdout)")

	// Apply config flags to the command. This allows for overriding the configuration at runtime
	// using the command line.
	config.ApplyCobraFlags(cmd)

	cmd.AddCommand(npm.NewNpmCommand())
	cmd.AddCommand(npm.NewPnpmCommand())
	cmd.AddCommand(npm.NewBunCommand())
	cmd.AddCommand(npm.NewYarnCommand())
	cmd.AddCommand(executors.NewNpxCommand())
	cmd.AddCommand(executors.NewPnpxCommand())
	cmd.AddCommand(pypi.NewPipCommand())
	cmd.AddCommand(pypi.NewPip3Command())
	cmd.AddCommand(pypi.NewUvCommand())
	cmd.AddCommand(pypi.NewPoetryCommand())
	cmd.AddCommand(version.NewVersionCommand())
	cmd.AddCommand(setup.NewSetupCommand())
	cmd.AddCommand(setup.NewRemoveCommand())

	// Print Banner on --help / -h
	cmd.SetHelpFunc(func(command *cobra.Command, args []string) {
		fmt.Print(ui.GeneratePMGBanner(appVersion.Version, appVersion.Commit))
		fmt.Println(command.UsageString())
	})

	defer analytics.Close()
	defer eventlog.Close()
	defer func() {
		if err := audit.Close(); err != nil {
			log.Warnf("failed to close audit system: %v", err)
		}
	}()

	analytics.TrackCommandRun()
	analytics.TrackCI()

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