pantsbuild / pants / 19015773527

from __future__ import annotations

import logging

from collections import defaultdict

from collections.abc import Mapping

from pants.backend.java.dependency_inference.java_parser import resolve_fallible_result_to_analysis

from pants.backend.java.target_types import JavaSourceField

from pants.core.util_rules.source_files import SourceFilesRequest

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

from pants.engine.target import AllTargets, Targets

from pants.engine.unions import UnionRule

from pants.jvm.dependency_inference import symbol_mapper

from pants.jvm.dependency_inference.artifact_mapper import MutableTrieNode

from pants.jvm.dependency_inference.symbol_mapper import FirstPartyMappingRequest, SymbolMap

from pants.jvm.subsystems import JvmSubsystem

from pants.jvm.target_types import JvmResolveField

from pants.util.logging import LogLevel

logger = logging.getLogger(__name__)

class AllJavaTargets(Targets):

    pass

@rule(desc="Find all Java targets in project", level=LogLevel.DEBUG)

async def find_all_java_targets(tgts: AllTargets) -> AllJavaTargets:

    return AllJavaTargets(tgt for tgt in tgts if tgt.has_field(JavaSourceField))

class FirstPartyJavaTargetsMappingRequest(FirstPartyMappingRequest):

    pass

@rule(desc="Map all first party Java targets to their packages", level=LogLevel.DEBUG)

async def map_first_party_java_targets_to_symbols(

    source_analysis = await concurrently(

    address_and_analysis = zip(

    mapping: Mapping[str, MutableTrieNode] = defaultdict(MutableTrieNode)

    for (address, resolve), analysis in address_and_analysis:

        for top_level_type in analysis.top_level_types:

            mapping[resolve].insert(top_level_type, [address], first_party=True)

    return SymbolMap((resolve, node.frozen()) for resolve, node in mapping.items())

def rules():

    return (