aquasecurity-trivy/pkg/dependency/parser/python/pyproject/pyproject.go

package pyproject

import (
	"io"

	"github.com/BurntSushi/toml"
	"golang.org/x/xerrors"
)

type PyProject struct {
	Tool Tool `toml:"tool"`
}

type Tool struct {
	Poetry Poetry `toml:"poetry"`
}

type Poetry struct {
	Dependencies map[string]any `toml:"dependencies"`
}

// Parser parses pyproject.toml defined in PEP518.
// https://peps.python.org/pep-0518/
type Parser struct {
}

func NewParser() *Parser {
	return &Parser{}
}

func (p *Parser) Parse(r io.Reader) (map[string]any, error) {
	var conf PyProject
	if _, err := toml.NewDecoder(r).Decode(&conf); err != nil {
		return nil, xerrors.Errorf("toml decode error: %w", err)
	}
	return conf.Tool.Poetry.Dependencies, nil
}