mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-14 13:00:10 -06:00
fe5390c068
If GOBIN is set, it will be used as the install path instead of the first item from GOPATH followed by "/bin". If it is used, commands will not be isolated between different repos.
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import contextlib
|
|
import os.path
|
|
import sys
|
|
|
|
import pre_commit.constants as C
|
|
from pre_commit import git
|
|
from pre_commit.envcontext import envcontext
|
|
from pre_commit.envcontext import UNSET
|
|
from pre_commit.envcontext import Var
|
|
from pre_commit.languages import helpers
|
|
from pre_commit.util import clean_path_on_failure
|
|
from pre_commit.util import cmd_output
|
|
from pre_commit.util import rmtree
|
|
|
|
|
|
ENVIRONMENT_DIR = 'golangenv'
|
|
get_default_version = helpers.basic_get_default_version
|
|
healthy = helpers.basic_healthy
|
|
|
|
|
|
def get_env_patch(venv):
|
|
return (
|
|
('GOBIN', UNSET),
|
|
('PATH', (os.path.join(venv, 'bin'), os.pathsep, Var('PATH'))),
|
|
)
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def in_env(prefix):
|
|
envdir = prefix.path(
|
|
helpers.environment_dir(ENVIRONMENT_DIR, C.DEFAULT),
|
|
)
|
|
with envcontext(get_env_patch(envdir)):
|
|
yield
|
|
|
|
|
|
def guess_go_dir(remote_url):
|
|
if remote_url.endswith('.git'):
|
|
remote_url = remote_url[:-1 * len('.git')]
|
|
looks_like_url = (
|
|
not remote_url.startswith('file://') and
|
|
('//' in remote_url or '@' in remote_url)
|
|
)
|
|
remote_url = remote_url.replace(':', '/')
|
|
if looks_like_url:
|
|
_, _, remote_url = remote_url.rpartition('//')
|
|
_, _, remote_url = remote_url.rpartition('@')
|
|
return remote_url
|
|
else:
|
|
return 'unknown_src_dir'
|
|
|
|
|
|
def install_environment(prefix, version, additional_dependencies):
|
|
helpers.assert_version_default('golang', version)
|
|
directory = prefix.path(
|
|
helpers.environment_dir(ENVIRONMENT_DIR, C.DEFAULT),
|
|
)
|
|
|
|
with clean_path_on_failure(directory):
|
|
remote = git.get_remote_url(prefix.prefix_dir)
|
|
repo_src_dir = os.path.join(directory, 'src', guess_go_dir(remote))
|
|
|
|
# Clone into the goenv we'll create
|
|
helpers.run_setup_cmd(prefix, ('git', 'clone', '.', repo_src_dir))
|
|
|
|
if sys.platform == 'cygwin': # pragma: no cover
|
|
_, gopath, _ = cmd_output('cygpath', '-w', directory)
|
|
gopath = gopath.strip()
|
|
else:
|
|
gopath = directory
|
|
env = dict(os.environ, GOPATH=gopath)
|
|
env.pop('GOBIN', None)
|
|
cmd_output('go', 'get', './...', cwd=repo_src_dir, env=env)
|
|
for dependency in additional_dependencies:
|
|
cmd_output('go', 'get', dependency, cwd=repo_src_dir, env=env)
|
|
# Same some disk space, we don't need these after installation
|
|
rmtree(prefix.path(directory, 'src'))
|
|
pkgdir = prefix.path(directory, 'pkg')
|
|
if os.path.exists(pkgdir): # pragma: no cover (go<1.10)
|
|
rmtree(pkgdir)
|
|
|
|
|
|
def run_hook(hook, file_args):
|
|
with in_env(hook.prefix):
|
|
return helpers.run_xargs(hook, helpers.to_cmd(hook), file_args)
|