mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-13 20:40:08 -06:00
115 lines
3.6 KiB
Python
115 lines
3.6 KiB
Python
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import logging
|
|
import sys
|
|
|
|
from aspy.yaml import ordered_dump
|
|
from aspy.yaml import ordered_load
|
|
|
|
import pre_commit.constants as C
|
|
from pre_commit.clientlib.validate_config import CONFIG_JSON_SCHEMA
|
|
from pre_commit.clientlib.validate_config import is_local_hooks
|
|
from pre_commit.clientlib.validate_config import load_config
|
|
from pre_commit.jsonschema_extensions import remove_defaults
|
|
from pre_commit.logging_handler import LoggingHandler
|
|
from pre_commit.ordereddict import OrderedDict
|
|
from pre_commit.repository import Repository
|
|
from pre_commit.util import cmd_output
|
|
from pre_commit.util import cwd
|
|
|
|
|
|
logger = logging.getLogger('pre_commit')
|
|
|
|
|
|
class RepositoryCannotBeUpdatedError(RuntimeError):
|
|
pass
|
|
|
|
|
|
def _update_repository(repo_config, runner):
|
|
"""Updates a repository to the tip of `master`. If the repository cannot
|
|
be updated because a hook that is configured does not exist in `master`,
|
|
this raises a RepositoryCannotBeUpdatedError
|
|
|
|
Args:
|
|
repo_config - A config for a repository
|
|
"""
|
|
repo = Repository.create(repo_config, runner.store)
|
|
|
|
with cwd(repo.repo_path_getter.repo_path):
|
|
cmd_output('git', 'fetch')
|
|
head_sha = cmd_output('git', 'rev-parse', 'origin/master')[1].strip()
|
|
|
|
# Don't bother trying to update if our sha is the same
|
|
if head_sha == repo_config['sha']:
|
|
return repo_config
|
|
|
|
# Construct a new config with the head sha
|
|
new_config = OrderedDict(repo_config)
|
|
new_config['sha'] = head_sha
|
|
new_repo = Repository.create(new_config, runner.store)
|
|
|
|
# See if any of our hooks were deleted with the new commits
|
|
hooks = set(hook_id for hook_id, _ in repo.hooks)
|
|
hooks_missing = hooks - (hooks & set(new_repo.manifest.hooks.keys()))
|
|
if hooks_missing:
|
|
raise RepositoryCannotBeUpdatedError(
|
|
'Cannot update because the tip of master is missing these hooks:\n'
|
|
'{0}'.format(', '.join(sorted(hooks_missing)))
|
|
)
|
|
|
|
return new_config
|
|
|
|
|
|
def autoupdate(runner):
|
|
"""Auto-update the pre-commit config to the latest versions of repos."""
|
|
# Set up our logging handler
|
|
logger.addHandler(LoggingHandler(False))
|
|
logger.setLevel(logging.WARNING)
|
|
|
|
retv = 0
|
|
output_configs = []
|
|
changed = False
|
|
|
|
input_configs = load_config(
|
|
runner.config_file_path,
|
|
load_strategy=ordered_load,
|
|
)
|
|
|
|
for repo_config in input_configs:
|
|
if is_local_hooks(repo_config):
|
|
output_configs.append(repo_config)
|
|
continue
|
|
sys.stdout.write('Updating {0}...'.format(repo_config['repo']))
|
|
sys.stdout.flush()
|
|
try:
|
|
new_repo_config = _update_repository(repo_config, runner)
|
|
except RepositoryCannotBeUpdatedError as error:
|
|
print(error.args[0])
|
|
output_configs.append(repo_config)
|
|
retv = 1
|
|
continue
|
|
|
|
if new_repo_config['sha'] != repo_config['sha']:
|
|
changed = True
|
|
print(
|
|
'updating {0} -> {1}.'.format(
|
|
repo_config['sha'], new_repo_config['sha'],
|
|
)
|
|
)
|
|
output_configs.append(new_repo_config)
|
|
else:
|
|
print('already up to date.')
|
|
output_configs.append(repo_config)
|
|
|
|
if changed:
|
|
with open(runner.config_file_path, 'w') as config_file:
|
|
config_file.write(
|
|
ordered_dump(
|
|
remove_defaults(output_configs, CONFIG_JSON_SCHEMA),
|
|
**C.YAML_DUMP_KWARGS
|
|
)
|
|
)
|
|
|
|
return retv
|