mirror of
https://github.com/safedep/pmg.git
synced 2026-08-03 07:24:09 +02:00
Sahil BansalandGitHub
12aaba8a05
* add poetry command * add poetry.lock extractor & define cmd parser for poetry * feat: support Poetry caret/tilde version syntax * add wildcard constraint support & test cases * readme update * chore: small fixes
117 lines
3.6 KiB
Go
117 lines
3.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/safedep/dry/log"
|
|
"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/ui"
|
|
appVersion "github.com/safedep/pmg/internal/version"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
debug bool
|
|
silent bool
|
|
verbose bool
|
|
logFile string
|
|
globalConfig config.Config
|
|
)
|
|
|
|
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")
|
|
}
|
|
|
|
if silent && verbose {
|
|
fmt.Println("pmg: --silent and --verbose cannot be used together")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if silent {
|
|
ui.SetVerbosityLevel(ui.VerbosityLevelSilent)
|
|
} else if verbose {
|
|
ui.SetVerbosityLevel(ui.VerbosityLevelVerbose)
|
|
}
|
|
|
|
// Check for PMG_INSECURE_INSTALLATION environment variable
|
|
if val := os.Getenv("PMG_INSECURE_INSTALLATION"); val != "" {
|
|
if boolVal, err := strconv.ParseBool(val); err == nil {
|
|
globalConfig.InsecureInstallation = boolVal
|
|
}
|
|
}
|
|
|
|
log.InitZapLogger("pmg", "cli")
|
|
cmd.SetContext(globalConfig.Inject(cmd.Context()))
|
|
},
|
|
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)")
|
|
cmd.PersistentFlags().BoolVar(&globalConfig.Transitive, "transitive", true, "Resolve transitive dependencies")
|
|
cmd.PersistentFlags().IntVar(&globalConfig.TransitiveDepth, "transitive-depth", 5,
|
|
"Maximum depth of transitive dependencies to resolve")
|
|
cmd.PersistentFlags().BoolVar(&globalConfig.IncludeDevDependencies, "include-dev-dependencies", false,
|
|
"Include dev dependencies in the dependency graph (slows down resolution)")
|
|
cmd.PersistentFlags().BoolVar(&globalConfig.DryRun, "dry-run", false, "Dry run skips execution of package manager")
|
|
cmd.PersistentFlags().BoolVar(&globalConfig.Paranoid, "paranoid", false, "Perform active scanning of unknown packages (slow)")
|
|
|
|
cmd.AddCommand(npm.NewNpmCommand())
|
|
cmd.AddCommand(npm.NewPnpmCommand())
|
|
cmd.AddCommand(npm.NewBunCommand())
|
|
cmd.AddCommand(pypi.NewPipCommand())
|
|
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()
|
|
|
|
analytics.TrackCommandRun()
|
|
analytics.TrackCI()
|
|
|
|
if err := cmd.Execute(); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|