mirror of
https://github.com/safedep/pmg.git
synced 2026-08-03 07:24:09 +02:00
Sahil BansalandGitHub
fd7d83704f
* follow proper consistent naming in pypi packagemanager * follow proper consistent naming in pypi packagemanager - 2 * feat: add specialized command parsers to handle pip and uv command formats * add uv support & modify extractor to be more robust * add uv alias * refactor var name & add error handling * update readme & add support for `uv pip sync` cmd
55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
package extractor
|
|
|
|
import (
|
|
packagev1 "buf.build/gen/go/safedep/api/protocolbuffers/go/safedep/messages/package/v1"
|
|
)
|
|
|
|
// PackageManagerExtractor defines the interface for package-manager-specific extractors
|
|
type PackageManagerExtractor interface {
|
|
// Returns the list of supported lockfiles by the package manager
|
|
GetSupportedFiles() []string
|
|
|
|
// Returns the package manager ecosysetm
|
|
GetEcosystem() packagev1.Ecosystem
|
|
|
|
// Returns the package manager name
|
|
GetPackageManager() PackageManagerName
|
|
|
|
// Extracts the packages from the lockfile
|
|
Extract(lockfilePath, scanDir string) ([]*packagev1.PackageVersion, error)
|
|
}
|
|
|
|
type PackageManagerName string
|
|
|
|
const (
|
|
Npm PackageManagerName = "npm"
|
|
Pnpm PackageManagerName = "pnpm"
|
|
Pip PackageManagerName = "pip"
|
|
Bun PackageManagerName = "bun"
|
|
Uv PackageManagerName = "uv"
|
|
)
|
|
|
|
type ExtractorManager struct {
|
|
extractors map[PackageManagerName]PackageManagerExtractor
|
|
}
|
|
|
|
func NewExtractorManager() *ExtractorManager {
|
|
return &ExtractorManager{
|
|
extractors: map[PackageManagerName]PackageManagerExtractor{
|
|
Npm: &NpmExtractor{},
|
|
Pnpm: &PnpmExtractor{},
|
|
Pip: &PipExtractor{},
|
|
Bun: &BunExtractor{},
|
|
Uv: &UvExtractor{},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (e *ExtractorManager) GetExtractorForPackageManager(pmn PackageManagerName) PackageManagerExtractor {
|
|
return e.extractors[pmn]
|
|
}
|
|
|
|
func (e *ExtractorManager) GetSupportedFilesForPackageManager(pmn PackageManagerName) []string {
|
|
return e.extractors[pmn].GetSupportedFiles()
|
|
}
|