pantsbuild / pants / 24055979590

from __future__ import annotations

import xml.etree.ElementTree as ET

from dataclasses import dataclass

from pants.engine.fs import Digest

from pants.engine.intrinsics import get_digest_contents

from pants.engine.rules import collect_rules, rule

from pants.jvm.resolve.coordinate import Coordinate

@dataclass(frozen=True)

class AnalysePomRequest:

    pom_digest: Digest

@dataclass(frozen=True)

class PomReport:

    dependencies: tuple[Coordinate, ...]

_MAVEN_NAMESPACE = "http://maven.apache.org/POM/4.0.0"

_MAVEN_NAMESPACE_MAP = {"mvn": _MAVEN_NAMESPACE}

@rule(desc="Analysing Maven POM file")

async def analyse_pom(request: AnalysePomRequest) -> PomReport:

    contents = await get_digest_contents(request.pom_digest)

    assert len(contents) == 1

    root = ET.fromstring(contents[0].content)

    def maybe_lookup_property(txt: str | None) -> str | None:

        if not txt:

            return None

        if txt.startswith("${") and txt.endswith("}"):

            prop = txt[2:-1]

            prop_value = root.findtext(

            if prop_value:

                return prop_value

        return txt

    coordinates = []

    for dep in root.findall(".//mvn:dependency", namespaces=_MAVEN_NAMESPACE_MAP):

        scope = maybe_lookup_property(dep.findtext("mvn:scope", namespaces=_MAVEN_NAMESPACE_MAP))

        if scope and scope == "test":

            continue

        coord_dict = {

        coordinates.append(Coordinate.from_json_dict(coord_dict))

    return PomReport(tuple(coordinates))

def rules():

    return collect_rules()