pantsbuild / pants / 19015773527

from __future__ import annotations

from collections.abc import Iterable

from dataclasses import dataclass

from enum import Enum

from pants.engine.rules import Rule, collect_rules

from pants.engine.target import (

from pants.engine.unions import UnionRule

CC_HEADER_FILE_EXTENSIONS = (

C_SOURCE_FILE_EXTENSIONS = (".c",)

CXX_SOURCE_FILE_EXTENSIONS = (".cc", ".cpp", ".cxx")

CC_SOURCE_FILE_EXTENSIONS = C_SOURCE_FILE_EXTENSIONS + CXX_SOURCE_FILE_EXTENSIONS

CC_FILE_EXTENSIONS = CC_HEADER_FILE_EXTENSIONS + CC_SOURCE_FILE_EXTENSIONS

class CCLanguage(Enum):

    C = "c"

    CXX = "cxx"

class CCDependenciesField(Dependencies):

    pass

class CCSourceField(SingleSourceField):

    expected_file_extensions = CC_FILE_EXTENSIONS

class CCGeneratorSourcesField(MultipleSourcesField):

    expected_file_extensions = CC_FILE_EXTENSIONS

@dataclass(frozen=True)

class CCFieldSet(FieldSet):

    required_fields = (CCSourceField,)

    sources: CCSourceField

@dataclass(frozen=True)

class CCGeneratorFieldSet(FieldSet):

    required_fields = (CCGeneratorSourcesField,)

    sources: CCGeneratorSourcesField

class CCSourceTarget(Target):

    alias = "cc_source"

    core_fields = (

    help = "A single C/C++ source file or header file."

class CCSourcesGeneratorSourcesField(CCGeneratorSourcesField):

    default = tuple(f"*{ext}" for ext in CC_FILE_EXTENSIONS)

    help = generate_multiple_sources_field_help_message(

class CCSourcesGeneratorTarget(TargetFilesGenerator):

    alias = "cc_sources"

    core_fields = (

    generated_target_cls = CCSourceTarget

    copied_fields = COMMON_TARGET_FIELDS

    moved_fields = (Dependencies,)

    help = "Generate a `cc_source` target for each file in the `sources` field."

def rules() -> Iterable[Rule | UnionRule]:

    return collect_rules()