summaryrefslogtreecommitdiffstats
path: root/client.py
diff options
context:
space:
mode:
Diffstat (limited to 'client.py')
-rwxr-xr-xclient.py174
1 files changed, 174 insertions, 0 deletions
diff --git a/client.py b/client.py
new file mode 100755
index 0000000000..ea16f95627
--- /dev/null
+++ b/client.py
@@ -0,0 +1,174 @@
+#!/usr/bin/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/.
+
+HG_EXCLUSIONS = [".hg", ".hgignore", ".hgtags"]
+
+import glob
+import os
+import shutil
+import sys
+from optparse import OptionParser
+from subprocess import check_call
+
+topsrcdir = os.path.dirname(__file__)
+if topsrcdir == "":
+ topsrcdir = "."
+
+
+def check_call_noisy(cmd, *args, **kwargs):
+ print("Executing command:", cmd)
+ check_call(cmd, *args, **kwargs)
+
+
+def do_hg_pull(dir, repository, hg):
+ fulldir = os.path.join(topsrcdir, dir)
+ # clone if the dir doesn't exist, pull if it does
+ if not os.path.exists(fulldir):
+ check_call_noisy([hg, "clone", repository, fulldir])
+ else:
+ cmd = [hg, "pull", "-u", "-R", fulldir]
+ if repository is not None:
+ cmd.append(repository)
+ check_call_noisy(cmd)
+ check_call(
+ [hg, "parent", "-R", fulldir, "--template=Updated to revision {node}.\n"]
+ )
+
+
+def do_hg_replace(dir, repository, tag, exclusions, hg):
+ """
+ Replace the contents of dir with the contents of repository, except for
+ files matching exclusions.
+ """
+ fulldir = os.path.join(topsrcdir, dir)
+ if os.path.exists(fulldir):
+ shutil.rmtree(fulldir)
+
+ assert not os.path.exists(fulldir)
+ check_call_noisy([hg, "clone", "-u", tag, repository, fulldir])
+
+ for thing in exclusions:
+ for excluded in glob.iglob(os.path.join(fulldir, thing)):
+ if os.path.isdir(excluded):
+ shutil.rmtree(excluded)
+ else:
+ os.remove(excluded)
+
+
+def toggle_trailing_blank_line(depname):
+ """If the trailing line is empty, then we'll delete it.
+ Otherwise we'll add a blank line."""
+ lines = open(depname, "rb").readlines()
+ if not lines:
+ print("unexpected short file", file=sys.stderr)
+ return
+
+ if not lines[-1].strip():
+ # trailing line is blank, removing it
+ open(depname, "wb").writelines(lines[:-1])
+ else:
+ # adding blank line
+ open(depname, "ab").write(b"\n")
+
+
+def get_trailing_blank_line_state(depname):
+ lines = open(depname, "r").readlines()
+ if not lines:
+ print("unexpected short file", file=sys.stderr)
+ return "no blank line"
+
+ if not lines[-1].strip():
+ return "has blank line"
+ return "no blank line"
+
+
+def update_nspr_or_nss(tag, depfile, destination, hgpath):
+ destination = destination.rstrip("/")
+ permanent_patch_dir = destination + "/patches"
+ temporary_patch_dir = destination + ".patches"
+ if os.path.exists(temporary_patch_dir):
+ print("please clean up leftover directory " + temporary_patch_dir)
+ sys.exit(2)
+ warn_if_patch_exists(permanent_patch_dir)
+ # protect patch directory from being removed by do_hg_replace
+ if os.path.exists(permanent_patch_dir):
+ shutil.move(permanent_patch_dir, temporary_patch_dir)
+ # now update the destination
+ print("reverting to HG version of %s to get its blank line state" % depfile)
+ check_call_noisy([options.hg, "revert", depfile])
+ old_state = get_trailing_blank_line_state(depfile)
+ print("old state of %s is: %s" % (depfile, old_state))
+ do_hg_replace(destination, hgpath, tag, HG_EXCLUSIONS, options.hg)
+ new_state = get_trailing_blank_line_state(depfile)
+ print("new state of %s is: %s" % (depfile, new_state))
+ if old_state == new_state:
+ print("toggling blank line in: ", depfile)
+ toggle_trailing_blank_line(depfile)
+ tag_file = destination + "/TAG-INFO"
+ with open(tag_file, "w") as f:
+ f.write(tag)
+ # move patch directory back to a subdirectory
+ if os.path.exists(temporary_patch_dir):
+ shutil.move(temporary_patch_dir, permanent_patch_dir)
+
+
+def warn_if_patch_exists(path):
+ # If the given patch directory exists and contains at least one file,
+ # then print warning and wait for the user to acknowledge.
+ if os.path.isdir(path) and os.listdir(path):
+ print("========================================")
+ print("WARNING: At least one patch file exists")
+ print("in directory: " + path)
+ print("You must manually re-apply all patches")
+ print("after this script has completed!")
+ print("========================================")
+ input("Press Enter to continue...")
+ return
+
+
+o = OptionParser(usage="client.py [options] update_nspr tagname | update_nss tagname")
+o.add_option(
+ "--skip-mozilla",
+ dest="skip_mozilla",
+ action="store_true",
+ default=False,
+ help="Obsolete",
+)
+
+o.add_option(
+ "--hg",
+ dest="hg",
+ default=os.environ.get("HG", "hg"),
+ help="The location of the hg binary",
+)
+o.add_option(
+ "--repo", dest="repo", help="the repo to update from (default: upstream repo)"
+)
+
+try:
+ options, args = o.parse_args()
+ action = args[0]
+except IndexError:
+ o.print_help()
+ sys.exit(2)
+
+if action in ("checkout", "co"):
+ print("Warning: client.py checkout is obsolete.", file=sys.stderr)
+ pass
+elif action in ("update_nspr"):
+ (tag,) = args[1:]
+ depfile = "nsprpub/config/prdepend.h"
+ if not options.repo:
+ options.repo = "https://hg.mozilla.org/projects/nspr"
+ update_nspr_or_nss(tag, depfile, "nsprpub", options.repo)
+elif action in ("update_nss"):
+ (tag,) = args[1:]
+ depfile = "security/nss/coreconf/coreconf.dep"
+ if not options.repo:
+ options.repo = "https://hg.mozilla.org/projects/nss"
+ update_nspr_or_nss(tag, depfile, "security/nss", options.repo)
+else:
+ o.print_help()
+ sys.exit(2)