pantsbuild / pants / 19015773527

from __future__ import annotations

import logging

from dataclasses import dataclass

from operator import itemgetter

from typing import ClassVar

from pants.backend.go.util_rules.build_opts import GoBuildOptions

from pants.engine.fs import CreateDigest, FileContent

from pants.engine.internals.native_engine import Digest

from pants.engine.intrinsics import create_digest

from pants.engine.rules import collect_rules, rule

from pants.util.frozendict import FrozenDict

logger = logging.getLogger(__name__)

@dataclass(frozen=True)

class ImportConfig:

    digest: Digest

    CONFIG_PATH: ClassVar[str] = "./importcfg"

@dataclass(frozen=True)

class ImportConfigRequest:

    import_paths_to_pkg_a_files: FrozenDict[str, str]

    build_opts: GoBuildOptions

    import_map: FrozenDict[str, str] | None = None

@rule

async def generate_import_config(request: ImportConfigRequest) -> ImportConfig:

    key_fn = itemgetter(0)

    packages = sorted(request.import_paths_to_pkg_a_files.items(), key=key_fn)

    import_map = sorted((request.import_map or {}).items(), key=key_fn)

    lines = [

    content = "\n".join(lines).encode("utf-8")

    result = await create_digest(CreateDigest([FileContent(ImportConfig.CONFIG_PATH, content)]))

    return ImportConfig(result)

def rules():

    return collect_rules()