64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
import contextlib
|
|
import os
|
|
from typing import Generator
|
|
from typing import Sequence
|
|
from typing import Tuple
|
|
|
|
import pre_commit.constants as C
|
|
from pre_commit.envcontext import envcontext
|
|
from pre_commit.envcontext import PatchesT
|
|
from pre_commit.envcontext import Var
|
|
from pre_commit.hook import Hook
|
|
from pre_commit.languages import helpers
|
|
from pre_commit.prefix import Prefix
|
|
from pre_commit.util import clean_path_on_failure
|
|
from pre_commit.util import cmd_output_b
|
|
|
|
ENVIRONMENT_DIR = 'swift_env'
|
|
get_default_version = helpers.basic_get_default_version
|
|
healthy = helpers.basic_healthy
|
|
BUILD_DIR = '.build'
|
|
BUILD_CONFIG = 'release'
|
|
|
|
|
|
def get_env_patch(venv: str) -> PatchesT: # pragma: win32 no cover
|
|
bin_path = os.path.join(venv, BUILD_DIR, BUILD_CONFIG)
|
|
return (('PATH', (bin_path, os.pathsep, Var('PATH'))),)
|
|
|
|
|
|
@contextlib.contextmanager # pragma: win32 no cover
|
|
def in_env(prefix: Prefix) -> Generator[None, None, None]:
|
|
envdir = prefix.path(
|
|
helpers.environment_dir(ENVIRONMENT_DIR, C.DEFAULT),
|
|
)
|
|
with envcontext(get_env_patch(envdir)):
|
|
yield
|
|
|
|
|
|
def install_environment(
|
|
prefix: Prefix, version: str, additional_dependencies: Sequence[str],
|
|
) -> None: # pragma: win32 no cover
|
|
helpers.assert_version_default('swift', version)
|
|
helpers.assert_no_additional_deps('swift', additional_dependencies)
|
|
directory = prefix.path(
|
|
helpers.environment_dir(ENVIRONMENT_DIR, C.DEFAULT),
|
|
)
|
|
|
|
# Build the swift package
|
|
with clean_path_on_failure(directory):
|
|
os.mkdir(directory)
|
|
cmd_output_b(
|
|
'swift', 'build',
|
|
'-C', prefix.prefix_dir,
|
|
'-c', BUILD_CONFIG,
|
|
'--build-path', os.path.join(directory, BUILD_DIR),
|
|
)
|
|
|
|
|
|
def run_hook(
|
|
hook: Hook,
|
|
file_args: Sequence[str],
|
|
color: bool,
|
|
) -> Tuple[int, bytes]: # pragma: win32 no cover
|
|
with in_env(hook.prefix):
|
|
return helpers.run_xargs(hook, hook.cmd, file_args, color=color)
|