src/v8/tools/push-to-trunk/git_recipes.py

   1 #!/usr/bin/env python
   2 # Copyright 2014 the V8 project authors. All rights reserved.
   3 # Redistribution and use in source and binary forms, with or without
   4 # modification, are permitted provided that the following conditions are
   5 # met:
   6 #
   7 #     * Redistributions of source code must retain the above copyright
   8 #       notice, this list of conditions and the following disclaimer.
   9 #     * Redistributions in binary form must reproduce the above
  10 #       copyright notice, this list of conditions and the following
  11 #       disclaimer in the documentation and/or other materials provided
  12 #       with the distribution.
  13 #     * Neither the name of Google Inc. nor the names of its
  14 #       contributors may be used to endorse or promote products derived
  15 #       from this software without specific prior written permission.
  16 #
  17 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  18 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  19 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  20 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  21 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  22 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  23 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  27 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28
  29 import re
  30
  31
  32 class GitFailedException(Exception):
  33   pass
  34
  35
  36 def Strip(f):
  37   def new_f(*args, **kwargs):
  38     return f(*args, **kwargs).strip()
  39   return new_f
  40
  41
  42 def MakeArgs(l):
  43   """['-a', '', 'abc', ''] -> '-a abc'"""
  44   return " ".join(filter(None, l))
  45
  46
  47 def Quoted(s):
  48   return "\"%s\"" % s
  49
  50
  51 class GitRecipesMixin(object):
  52   def GitIsWorkdirClean(self):
  53     return self.Git("status -s -uno").strip() == ""
  54
  55   @Strip
  56   def GitBranch(self):
  57     return self.Git("branch")
  58
  59   def GitCreateBranch(self, name, branch=""):
  60     assert name
  61     self.Git(MakeArgs(["checkout -b", name, branch]))
  62
  63   def GitDeleteBranch(self, name):
  64     assert name
  65     self.Git(MakeArgs(["branch -D", name]))
  66
  67   def GitReset(self, name):
  68     assert name
  69     self.Git(MakeArgs(["reset --hard", name]))
  70
  71   def GitRemotes(self):
  72     return map(str.strip, self.Git(MakeArgs(["branch -r"])).splitlines())
  73
  74   def GitCheckout(self, name):
  75     assert name
  76     self.Git(MakeArgs(["checkout -f", name]))
  77
  78   def GitCheckoutFile(self, name, branch_or_hash):
  79     assert name
  80     assert branch_or_hash
  81     self.Git(MakeArgs(["checkout -f", branch_or_hash, "--", name]))
  82
  83   def GitCheckoutFileSafe(self, name, branch_or_hash):
  84     try:
  85       self.GitCheckoutFile(name, branch_or_hash)
  86     except GitFailedException:  # pragma: no cover
  87       # The file doesn't exist in that revision.
  88       return False
  89     return True
  90
  91   def GitChangedFiles(self, git_hash):
  92     assert git_hash
  93     try:
  94       files = self.Git(MakeArgs(["diff --name-only",
  95                                  git_hash,
  96                                  "%s^" % git_hash]))
  97       return map(str.strip, files.splitlines())
  98     except GitFailedException:  # pragma: no cover
  99       # Git fails using "^" at branch roots.
 100       return []
 101
 102
 103   @Strip
 104   def GitCurrentBranch(self):
 105     for line in self.Git("status -s -b -uno").strip().splitlines():
 106       match = re.match(r"^## (.+)", line)
 107       if match: return match.group(1)
 108     raise Exception("Couldn't find curent branch.")  # pragma: no cover
 109
 110   @Strip
 111   def GitLog(self, n=0, format="", grep="", git_hash="", parent_hash="",
 112              branch="", reverse=False):
 113     assert not (git_hash and parent_hash)
 114     args = ["log"]
 115     if n > 0:
 116       args.append("-%d" % n)
 117     if format:
 118       args.append("--format=%s" % format)
 119     if grep:
 120       args.append("--grep=\"%s\"" % grep.replace("\"", "\\\""))
 121     if reverse:
 122       args.append("--reverse")
 123     if git_hash:
 124       args.append(git_hash)
 125     if parent_hash:
 126       args.append("%s^" % parent_hash)
 127     args.append(branch)
 128     return self.Git(MakeArgs(args))
 129
 130   def GitGetPatch(self, git_hash):
 131     assert git_hash
 132     return self.Git(MakeArgs(["log", "-1", "-p", git_hash]))
 133
 134   # TODO(machenbach): Unused? Remove.
 135   def GitAdd(self, name):
 136     assert name
 137     self.Git(MakeArgs(["add", Quoted(name)]))
 138
 139   def GitApplyPatch(self, patch_file, reverse=False):
 140     assert patch_file
 141     args = ["apply --index --reject"]
 142     if reverse:
 143       args.append("--reverse")
 144     args.append(Quoted(patch_file))
 145     self.Git(MakeArgs(args))
 146
 147   def GitUpload(self, reviewer="", author="", force=False):
 148     args = ["cl upload --send-mail"]
 149     if author:
 150       args += ["--email", Quoted(author)]
 151     if reviewer:
 152       args += ["-r", Quoted(reviewer)]
 153     if force:
 154       args.append("-f")
 155     # TODO(machenbach): Check output in forced mode. Verify that all required
 156     # base files were uploaded, if not retry.
 157     self.Git(MakeArgs(args), pipe=False)
 158
 159   def GitCommit(self, message="", file_name=""):
 160     assert message or file_name
 161     args = ["commit"]
 162     if file_name:
 163       args += ["-aF", Quoted(file_name)]
 164     if message:
 165       args += ["-am", Quoted(message)]
 166     self.Git(MakeArgs(args))
 167
 168   def GitPresubmit(self):
 169     self.Git("cl presubmit", "PRESUBMIT_TREE_CHECK=\"skip\"")
 170
 171   def GitDCommit(self):
 172     self.Git("cl dcommit -f --bypass-hooks", retry_on=lambda x: x is None)
 173
 174   def GitDiff(self, loc1, loc2):
 175     return self.Git(MakeArgs(["diff", loc1, loc2]))
 176
 177   def GitPull(self):
 178     self.Git("pull")
 179
 180   def GitSVNFetch(self):
 181     self.Git("svn fetch")
 182
 183   # TODO(machenbach): Unused? Remove.
 184   @Strip
 185   def GitSVNLog(self):
 186     return self.Git("svn log -1 --oneline")
 187
 188   @Strip
 189   def GitSVNFindGitHash(self, revision, branch=""):
 190     assert revision
 191     return self.Git(MakeArgs(["svn find-rev", "r%s" % revision, branch]))
 192
 193   @Strip
 194   def GitSVNFindSVNRev(self, git_hash, branch=""):
 195     return self.Git(MakeArgs(["svn find-rev", git_hash, branch]))
 196
 197   def GitSVNDCommit(self):
 198     return self.Git("svn dcommit 2>&1", retry_on=lambda x: x is None)
 199
 200   def GitSVNTag(self, version):
 201     self.Git(("svn tag %s -m \"Tagging version %s\"" % (version, version)),
 202              retry_on=lambda x: x is None)