diff options
Diffstat (limited to 'python/mozversioncontrol/mozversioncontrol/repoupdate.py')
-rw-r--r-- | python/mozversioncontrol/mozversioncontrol/repoupdate.py | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/python/mozversioncontrol/mozversioncontrol/repoupdate.py b/python/mozversioncontrol/mozversioncontrol/repoupdate.py new file mode 100644 index 0000000000..a78de56377 --- /dev/null +++ b/python/mozversioncontrol/mozversioncontrol/repoupdate.py @@ -0,0 +1,49 @@ +# 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 . 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)) |