Add Pip3 support (#75)

* add pip3 support

* add pip3 e2e

* chore: rename var
This commit is contained in:
Sahil Bansal
2025-12-03 20:28:45 +05:30
committed by GitHub
parent c47cb30de0
commit 50d7b55445
7 changed files with 122 additions and 2 deletions
+61
View File
@@ -0,0 +1,61 @@
package pypi
import (
"context"
"fmt"
"github.com/safedep/pmg/config"
"github.com/safedep/pmg/internal/analytics"
"github.com/safedep/pmg/internal/flows"
"github.com/safedep/pmg/internal/ui"
"github.com/safedep/pmg/packagemanager"
"github.com/spf13/cobra"
)
func NewPip3Command() *cobra.Command {
return &cobra.Command{
Use: "pip3 [action] [package]",
Short: "Guard pip3 package manager",
DisableFlagParsing: true,
RunE: func(cmd *cobra.Command, args []string) error {
err := executePip3Flow(cmd.Context(), args)
if err != nil {
ui.ErrorExit(err)
}
return nil
},
}
}
func executePip3Flow(ctx context.Context, args []string) error {
analytics.TrackCommandPip3()
packageManager, err := packagemanager.NewPypiPackageManager(packagemanager.DefaultPip3PackageManagerConfig())
if err != nil {
return fmt.Errorf("failed to create pip3 package manager proxy: %w", err)
}
config, err := config.FromContext(ctx)
if err != nil {
return fmt.Errorf("failed to get config: %w", err)
}
parsedCommand, err := packageManager.ParseCommand(args)
if err != nil {
return fmt.Errorf("failed to parse command: %w", err)
}
// Parse the args right here
packageResolverConfig := packagemanager.NewDefaultPypiDependencyResolverConfig()
packageResolverConfig.IncludeTransitiveDependencies = config.Transitive
packageResolverConfig.TransitiveDepth = config.TransitiveDepth
packageResolverConfig.IncludeDevDependencies = config.IncludeDevDependencies
packageResolverConfig.PackageInstallTargets = parsedCommand.InstallTargets
packageResolver, err := packagemanager.NewPypiDependencyResolver(packageResolverConfig)
if err != nil {
return fmt.Errorf("failed to create dependency resolver: %w", err)
}
return flows.Common(packageManager, packageResolver, config).Run(ctx, args, parsedCommand)
}