gecko/tools/mercurial/hgsetup/update.py
2015-07-17 10:28:12 -07:00

69 lines
2.1 KiB
Python

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this,
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from __future__ import unicode_literals
import errno
import os
import which
from mozversioncontrol.repoupdate import update_mercurial_repo
from .config import (
HOST_FINGERPRINTS,
)
FINISHED = '''
Your Mercurial recommended extensions are now up to date!
'''.lstrip()
class MercurialUpdater(object):
def __init__(self, state_dir):
self.state_dir = os.path.normpath(state_dir)
self.ext_dir = os.path.join(self.state_dir, 'mercurial', 'extensions')
self.vcs_tools_dir = os.path.join(self.state_dir, 'version-control-tools')
def update_all(self):
try:
os.makedirs(self.ext_dir)
except OSError as e:
if e.errno != errno.EEXIST:
raise
try:
hg = which.which('hg')
except which.WhichError as e:
print(e)
print('Try running |mach bootstrap| to ensure your environment is '
'up to date.')
return 1
if os.path.isdir(self.vcs_tools_dir):
self.update_mercurial_repo(
hg,
'https://hg.mozilla.org/hgcustom/version-control-tools',
self.vcs_tools_dir,
'default',
'Ensuring version-control-tools is up to date...')
print(FINISHED)
return 0
def update_mercurial_repo(self, hg, url, dest, branch, msg):
# We always pass the host fingerprints that we "know" to be canonical
# because the existing config may have outdated fingerprints and this
# may cause Mercurial to abort.
return self._update_repo(hg, url, dest, branch, msg,
update_mercurial_repo, hostfingerprints=HOST_FINGERPRINTS)
def _update_repo(self, binary, url, dest, branch, msg, fn, *args, **kwargs):
print('=' * 80)
print(msg)
try:
fn(binary, url, dest, branch, *args, **kwargs)
finally:
print('=' * 80)
print('')