pantsbuild / pants / 19015773527

from __future__ import annotations

import os

from collections.abc import Iterable

from dataclasses import dataclass

from pants.backend.java.subsystems.javac import JavacSubsystem

from pants.backend.java.target_types import (

from pants.core.goals.tailor import (

from pants.engine.intrinsics import path_globs_to_paths

from pants.engine.rules import collect_rules, rule

from pants.engine.target import Target

from pants.engine.unions import UnionRule

from pants.source.filespec import FilespecMatcher

from pants.util.dirutil import group_by_dir

from pants.util.logging import LogLevel

@dataclass(frozen=True)

class PutativeJavaTargetsRequest(PutativeTargetsRequest):

    pass

def classify_source_files(paths: Iterable[str]) -> dict[type[Target], set[str]]:

    tests_filespec_matcher = FilespecMatcher(JavaTestsGeneratorSourcesField.default, ())

    test_filenames = set(tests_filespec_matcher.matches([os.path.basename(path) for path in paths]))

    test_files = {path for path in paths if os.path.basename(path) in test_filenames}

    sources_files = set(paths) - test_files

    return {JunitTestsGeneratorTarget: test_files, JavaSourcesGeneratorTarget: sources_files}

@rule(level=LogLevel.DEBUG, desc="Determine candidate Java targets to create")

async def find_putative_targets(

    putative_targets = []

    if javac.tailor_source_targets:

        all_java_files_globs = req.path_globs("*.java")

        all_java_files = await path_globs_to_paths(all_java_files_globs)

        unowned_java_files = set(all_java_files.files) - set(all_owned_sources)

        classified_unowned_java_files = classify_source_files(unowned_java_files)

        for tgt_type, paths in classified_unowned_java_files.items():

            for dirname, filenames in group_by_dir(paths).items():

                name = "tests" if tgt_type == JunitTestsGeneratorTarget else None

                putative_targets.append(

    return PutativeTargets(putative_targets)

def rules():

    return [