pmg/extractor/ecosystems.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"
	Pip3   PackageManagerName = "pip3"
	Bun    PackageManagerName = "bun"
	Yarn   PackageManagerName = "yarn"
	Uv     PackageManagerName = "uv"
	Poetry PackageManagerName = "poetry"
)

type ExtractorManager struct {
	extractors map[PackageManagerName]PackageManagerExtractor
}

func NewExtractorManager() *ExtractorManager {
	return &ExtractorManager{
		extractors: map[PackageManagerName]PackageManagerExtractor{
			Npm:    &NpmExtractor{},
			Pnpm:   &PnpmExtractor{},
			Pip:    &PipExtractor{},
			Bun:    &BunExtractor{},
			Yarn:   &YarnExtractor{},
			Uv:     &UvExtractor{},
			Poetry: &PoetryExtractor{},
			Pip3:   &Pip3Extractor{},
		},
	}
}

func (e *ExtractorManager) GetExtractorForPackageManager(pmn PackageManagerName) PackageManagerExtractor {
	return e.extractors[pmn]
}

func (e *ExtractorManager) GetSupportedFilesForPackageManager(pmn PackageManagerName) []string {
	return e.extractors[pmn].GetSupportedFiles()
}