pantsbuild / pants / 24055979590

from __future__ import annotations

from collections.abc import Iterable

from dataclasses import dataclass

from pathlib import PurePath

from pants.backend.python.subsystems.python_tool_base import PythonToolBase

from pants.backend.python.target_types import ConsoleScript, EntryPoint

from pants.backend.python.util_rules.pex import VenvPex, create_venv_pex

from pants.engine.fs import CreateDigest, FileContent

from pants.engine.intrinsics import create_digest

from pants.engine.rules import Rule, collect_rules, implicitly, rule

from pants.util.logging import LogLevel

from pants.util.resources import read_resource

from pants.util.strutil import help_text

_ELFDEPS_PACKAGE = "pants.backend.nfpm.native_libs.elfdeps"

_ANALYZE_SCRIPT = "analyze.py"

_ANALYZE_TOOL = "__pants_elfdeps_analyze.py"

class Elfdeps(PythonToolBase):

    options_scope = "elfdeps"

    help_short = help_text(

    default_main = ConsoleScript("elfdeps")

    default_requirements = ["elfdeps>=0.2.0"]

    default_interpreter_constraints = ["CPython>=3.10,<3.15"]

    register_interpreter_constraints = True

    default_lockfile_resource = (_ELFDEPS_PACKAGE, "elfdeps.lock")

@dataclass(frozen=True)

class ElfdepsAnalyzeWheelsTool:

    pex: VenvPex

@rule(desc=f"Setup elfdeps/{_ANALYZE_SCRIPT}", level=LogLevel.DEBUG)

async def setup_elfdeps_analyze_tool(

    analyze_script = read_resource(_ELFDEPS_PACKAGE, _ANALYZE_SCRIPT)

    if not analyze_script:

        raise ValueError(f"Unable to find source of {_ANALYZE_SCRIPT!r} in {_ELFDEPS_PACKAGE}")

    analyze_script_content = FileContent(

    analyze_script_digest = await create_digest(CreateDigest([analyze_script_content]))

    analyze_pex = await create_venv_pex(

    return ElfdepsAnalyzeWheelsTool(analyze_pex)

def rules() -> Iterable[Rule]:

    return collect_rules()