65280224fa
I don't love monolithic files, this simply moves things around to submodules without changing any logic. Differential Revision: https://phabricator.services.mozilla.com/D228091
50 lines
1.5 KiB
Python
50 lines
1.5 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/.
|
|
|
|
import subprocess
|
|
from pathlib import Path
|
|
from typing import Union
|
|
|
|
from mozversioncontrol.repo.base import get_tool_path
|
|
|
|
# The logic here is far from robust. Improvements are welcome.
|
|
|
|
|
|
def update_git_repo(repo: str, path: Union[str, Path]):
|
|
"""Ensure a git repository exists at a path and is up to date."""
|
|
git = get_tool_path("git")
|
|
path = Path(path)
|
|
if path.exists():
|
|
subprocess.check_call([git, "pull", repo], cwd=str(path))
|
|
else:
|
|
subprocess.check_call([git, "clone", repo, str(path)])
|
|
|
|
|
|
def update_mercurial_repo(
|
|
repo: str,
|
|
path: Union[str, Path],
|
|
revision="default",
|
|
hostfingerprints=None,
|
|
global_args=None,
|
|
):
|
|
"""Ensure a HG repository exists at a path and is up to date."""
|
|
hostfingerprints = hostfingerprints or {}
|
|
|
|
hg = get_tool_path("hg")
|
|
path = Path(path)
|
|
|
|
args = [hg]
|
|
if global_args:
|
|
args.extend(global_args)
|
|
|
|
for host, fingerprint in sorted(hostfingerprints.items()):
|
|
args.extend(["--config", "hostfingerprints.%s=%s" % (host, fingerprint)])
|
|
|
|
if path.exists():
|
|
subprocess.check_call(args + ["pull", repo], cwd=str(path))
|
|
else:
|
|
subprocess.check_call(args + ["clone", repo, str(path)])
|
|
|
|
subprocess.check_call([hg, "update", "-r", revision], cwd=str(path))
|