91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
|
import contextlib
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
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
|
||
|
|
|
||
|
|
ENVIRONMENT_DIR = 'lua_env'
|
||
|
|
get_default_version = helpers.basic_get_default_version
|
||
|
|
healthy = helpers.basic_healthy
|
||
|
|
|
||
|
|
|
||
|
|
def _get_lua_version() -> str: # pragma: win32 no cover
|
||
|
|
"""Get the Lua version used in file paths."""
|
||
|
|
_, stdout, _ = cmd_output('luarocks', 'config', '--lua-ver')
|
||
|
|
return stdout.strip()
|
||
|
|
|
||
|
|
|
||
|
|
def get_env_patch(d: str) -> PatchesT: # pragma: win32 no cover
|
||
|
|
version = _get_lua_version()
|
||
|
|
so_ext = 'dll' if sys.platform == 'win32' else 'so'
|
||
|
|
return (
|
||
|
|
('PATH', (os.path.join(d, 'bin'), os.pathsep, Var('PATH'))),
|
||
|
|
(
|
||
|
|
'LUA_PATH', (
|
||
|
|
os.path.join(d, 'share', 'lua', version, '?.lua;'),
|
||
|
|
os.path.join(d, 'share', 'lua', version, '?', 'init.lua;;'),
|
||
|
|
),
|
||
|
|
),
|
||
|
|
(
|
||
|
|
'LUA_CPATH',
|
||
|
|
(os.path.join(d, 'lib', 'lua', version, f'?.{so_ext};;'),),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def _envdir(prefix: Prefix) -> str: # pragma: win32 no cover
|
||
|
|
directory = helpers.environment_dir(ENVIRONMENT_DIR, C.DEFAULT)
|
||
|
|
return prefix.path(directory)
|
||
|
|
|
||
|
|
|
||
|
|
@contextlib.contextmanager # pragma: win32 no cover
|
||
|
|
def in_env(prefix: Prefix) -> Generator[None, None, None]:
|
||
|
|
with envcontext(get_env_patch(_envdir(prefix))):
|
||
|
|
yield
|
||
|
|
|
||
|
|
|
||
|
|
def install_environment(
|
||
|
|
prefix: Prefix,
|
||
|
|
version: str,
|
||
|
|
additional_dependencies: Sequence[str],
|
||
|
|
) -> None: # pragma: win32 no cover
|
||
|
|
helpers.assert_version_default('lua', version)
|
||
|
|
|
||
|
|
envdir = _envdir(prefix)
|
||
|
|
with clean_path_on_failure(envdir):
|
||
|
|
with in_env(prefix):
|
||
|
|
# luarocks doesn't bootstrap a tree prior to installing
|
||
|
|
# so ensure the directory exists.
|
||
|
|
os.makedirs(envdir, exist_ok=True)
|
||
|
|
|
||
|
|
# Older luarocks (e.g., 2.4.2) expect the rockspec as an arg
|
||
|
|
for rockspec in prefix.star('.rockspec'):
|
||
|
|
make_cmd = ('luarocks', '--tree', envdir, 'make', rockspec)
|
||
|
|
helpers.run_setup_cmd(prefix, make_cmd)
|
||
|
|
|
||
|
|
# luarocks can't install multiple packages at once
|
||
|
|
# so install them individually.
|
||
|
|
for dependency in additional_dependencies:
|
||
|
|
cmd = ('luarocks', '--tree', envdir, 'install', dependency)
|
||
|
|
helpers.run_setup_cmd(prefix, cmd)
|
||
|
|
|
||
|
|
|
||
|
|
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)
|