pantsbuild / pants / 19015773527

from dataclasses import dataclass

from pants.backend.shell.lint.shfmt.skip_field import SkipShfmtField

from pants.backend.shell.lint.shfmt.subsystem import Shfmt

from pants.backend.shell.target_types import ShellSourceField

from pants.core.goals.fmt import FmtResult, FmtTargetsRequest

from pants.core.goals.resolves import ExportableTool

from pants.core.util_rules.config_files import find_config_file

from pants.core.util_rules.external_tool import download_external_tool

from pants.core.util_rules.partitions import PartitionerType

from pants.engine.fs import MergeDigests

from pants.engine.intrinsics import merge_digests

from pants.engine.platform import Platform

from pants.engine.process import Process, execute_process_or_raise

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

from pants.engine.target import FieldSet, Target

from pants.engine.unions import UnionRule

from pants.util.logging import LogLevel

from pants.util.strutil import pluralize

@dataclass(frozen=True)

class ShfmtFieldSet(FieldSet):

    required_fields = (ShellSourceField,)

    @classmethod

    def opt_out(cls, tgt: Target) -> bool:

        return tgt.get(SkipShfmtField).value

class ShfmtRequest(FmtTargetsRequest):

    field_set_type = ShfmtFieldSet

    tool_subsystem = Shfmt  # type: ignore[assignment]

    partitioner_type = PartitionerType.DEFAULT_SINGLE_PARTITION

@rule(desc="Format with shfmt", level=LogLevel.DEBUG)

async def shfmt_fmt(request: ShfmtRequest.Batch, shfmt: Shfmt, platform: Platform) -> FmtResult:

    download_shfmt_get = download_external_tool(shfmt.get_request(platform))

    config_files_get = find_config_file(shfmt.config_request(request.snapshot.dirs))

    downloaded_shfmt, config_files = await concurrently(download_shfmt_get, config_files_get)

    input_digest = await merge_digests(

    result = await execute_process_or_raise(

    return await FmtResult.create(request, result)

def rules():

    return [