pantsbuild / pants / 19015773527

from __future__ import annotations

import logging

from collections.abc import Mapping

from dataclasses import dataclass

from pants.backend.docker.subsystems.docker_options import DockerOptions

from pants.backend.docker.util_rules.docker_build_args import (

from pants.backend.docker.utils import KeyValueSequenceUtil

from pants.core.util_rules.env_vars import environment_vars_subset

from pants.engine.env_vars import EnvironmentVars, EnvironmentVarsRequest

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

from pants.engine.target import Target

logger = logging.getLogger(__name__)

class DockerBuildEnvironmentError(ValueError):

    @classmethod

    def from_key_error(cls, e: KeyError) -> DockerBuildEnvironmentError:

        return cls(

@dataclass(frozen=True)

class DockerBuildEnvironment:

    environment: EnvironmentVars

    @classmethod

    def create(

        return cls(EnvironmentVars(env))

    def __getitem__(self, key: str) -> str:

        try:

            return self.environment[key]

        except KeyError as e:

            raise DockerBuildEnvironmentError.from_key_error(e) from e

    def get(self, key: str, default: str | None = None) -> str:

        if default is None:

            return self[key]

        return self.environment.get(key, default)

@dataclass(frozen=True)

class DockerBuildEnvironmentRequest:

    target: Target

@rule

async def docker_build_environment_vars(

    build_args = await docker_build_args(DockerBuildArgsRequest(request.target), **implicitly())

    env_vars = KeyValueSequenceUtil.from_strings(

    env = await environment_vars_subset(EnvironmentVarsRequest(tuple(env_vars)), **implicitly())

    return DockerBuildEnvironment.create(env)

def rules():

    return collect_rules()