Try harder to dig git-svn revision.

build/util/lastchange.py

 #!/usr/bin/env python
-# Copyright (c) 2010 The Chromium Authors. All rights reserved.
+# Copyright (c) 2011 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """
 lastchange.py -- Chromium revision fetching utility.
 """
 
+import re
 import optparse
 import os
 import subprocess
 import sys
 
 class VersionInfo(object):
   def __init__(self, url, root, revision):
     self.url = url
     self.root = root
     self.revision = revision
 
-def FetchGitRevision(directory):
-  """
-  Fetch the Git hash for the a given directory.
-
-  Errors are swallowed.
-
-  Returns:
-    a VersionInfo object or None on error.
-  """
-  # Force shell usage under cygwin & win32. This is a workaround for
-  # mysterious loss of cwd while invoking cygwin's git.
-  # We can't just pass shell=True to Popen, as under win32 this will
-  # cause CMD to be used, while we explicitly want a cygwin shell.
-  command = ['git', 'rev-parse', 'HEAD']
-  if sys.platform in ('cygwin', 'win32'):
-    command = ['sh', '-c', ' '.join(command)]
-  try:
-    proc = subprocess.Popen(command,
-                            stdout=subprocess.PIPE,
-                            stderr=subprocess.PIPE,
-                            cwd=directory)
-  except OSError:
-    return None
-  output = proc.communicate()[0].strip()
-  if proc.returncode == 0 and output:
-    return VersionInfo('git', 'git', output[:7])
-  return None
-
 
 def FetchSVNRevision(directory):
   """
-  Fetch the Subversion branch and revision for the a given directory.
+  Fetch the Subversion branch and revision for a given directory.
 
   Errors are swallowed.
 
   Returns:
     a VersionInfo object or None on error.
   """
   try:
     proc = subprocess.Popen(['svn', 'info'],
                             stdout=subprocess.PIPE,
                             stderr=subprocess.PIPE,
@@ skipping 16 matching lines @@
   try:
     url = attrs['URL']
     root = attrs['Repository Root']
     revision = attrs['Revision']
   except KeyError:
     return None
 
   return VersionInfo(url, root, revision)
 
 
+def RunGitCommand(directory, command, agent):

[evanm, 2011/02/25 19:06:59]

Can you add docs for this like the other functions?

[Denis Lagno, 2011/03/01 12:33:53]

Done.

+  command.insert(0, 'git');

[evanm, 2011/02/25 19:06:59]

Rather than modifying the caller's array, it might be better to do
  command = ['git'] + command

I doubt it matters for this code but it's good style.

[Denis Lagno, 2011/03/01 12:33:53]

Done.

+  # Force shell usage under cygwin & win32. This is a workaround for
+  # mysterious loss of cwd while invoking cygwin's git.
+  # We can't just pass shell=True to Popen, as under win32 this will
+  # cause CMD to be used, while we explicitly want a cygwin shell.
+  if sys.platform in ('cygwin', 'win32'):
+    command = ['sh', '-c', ' '.join(command)]
+  try:
+    proc = subprocess.Popen(command,
+                            stdout=subprocess.PIPE,
+                            stderr=subprocess.PIPE,
+                            cwd=directory)
+    return agent(proc)

[evanm, 2011/02/25 19:06:59]

Why do this continuation-passing style rather than just "return proc"?  It seems
that would make the control flow clearer.

[Denis Lagno, 2011/03/01 12:33:53]

Done.

+  except OSError:
+    return None
+
+
+def FetchGitRevision(directory):
+  """
+  Fetch the Git hash for a given directory.
+
+  Errors are swallowed.
+
+  Returns:
+    a VersionInfo object or None on error.
+  """
+  git_command = ['rev-parse', 'HEAD']
+  def agent(proc):
+    output = proc.communicate()[0].strip()
+    if proc.returncode == 0 and output:
+      return VersionInfo('git', 'git', output[:7])
+    return None
+  return RunGitCommand(directory, git_command, agent)
+
+
+def IsGitSVN(directory):
+  """
+  Checks whether git-svn has been set up.
+
+  Errors are swallowed.
+
+  Returns:
+    whether git-svn has been set up.
+  """
+  # To test whether git-svn has been set up, query the config for any
+  # svn-related configuration.  This command exits with an error code
+  # if there aren't any matches, so ignore its output.
+  git_command = ['config', '--get-regexp', '^svn']
+  return RunGitCommand(directory, git_command, lambda proc: (proc.wait() == 0))
+
+
+def FetchGitSVNURL(directory):
+  """
+  Fetch URL of SVN repository bound to git.
+
+  Errors are swallowed.
+
+  Returns:
+    SVN URL.
+  """
+  git_command = ['svn', 'info', '--url']

[evanm, 2011/02/25 19:06:59]

This will hang the build if the user has a git clone of the git-svn repo without
having git-svn set up.  This is the setup of many non-committers.

[Denis Lagno, 2011/03/01 12:33:53]

this function is never called if IsGitSvn() returned false.
Added explicit check.

+  def agent(proc):
+    output = proc.communicate()[0].strip()
+    if proc.returncode == 0:
+      return output
+    return None
+  return RunGitCommand(directory, git_command, agent)
+
+
+def LookupGitSVNRevision(directory, depth):
+  """
+  Fetch the Git-SVN identifier for the local tree.
+  Parses first |depth| commit messages.
+
+  Errors are swallowed.
+  """
+  if not IsGitSVN(directory):
+    return None
+  git_re = re.compile('^\s*git-svn-id:\s+(\S+)@(\d+)', re.M)
+  git_command = ['log', '-' + str(depth)]
+  def agent(proc):
+    for line in proc.stdout:
+      match = git_re.search(line)
+      if match:
+        id = match.group(2)
+        if id:
+          proc.stdout.close()  # Cut pipe for fast exit.
+          return id
+    return None
+  return RunGitCommand(directory, git_command, agent)
+
+
+def IsGitSVNDirty(directory):
+  """
+  Checks whether our git-svn tree contains clean trunk or some branch.
+
+  Errors are swallowed.
+  """
+  # For git branches the last commit message is either
+  # some local commit or a merge.
+  return LookupGitSVNRevision(directory, 1) is None
+
+
+def FetchGitSVNRevision(directory):
+  """
+  Fetch the Git-SVN identifier for the local tree.
+
+  Errors are swallowed.
+  """
+  # We assume that at least first 999 commit messages contain svn evidence.
+  revision = LookupGitSVNRevision(directory, 999)
+  if not revision:
+    return None
+  if IsGitSVNDirty(directory):
+    revision = revision + '-dirty'
+  return VersionInfo(FetchGitSVNURL(directory), 'git-svn', revision)
+
+
 def FetchVersionInfo(default_lastchange, directory=None):
   """
   Returns the last change (in the form of a branch, revision tuple),
   from some appropriate revision control system.
   """
-  version_info = FetchSVNRevision(directory) or FetchGitRevision(directory)
+  version_info = (FetchSVNRevision(directory) or
+      FetchGitSVNRevision(directory) or FetchGitRevision(directory))
   if not version_info:
     if default_lastchange and os.path.exists(default_lastchange):
       revision = open(default_lastchange, 'r').read().strip()
       version_info = VersionInfo(None, None, revision)
     else:
       version_info = VersionInfo('unknown', '', '0')
   return version_info
 
 
 def WriteIfChanged(file_name, contents):
@@ skipping 44 matching lines @@
     if out_file:
       WriteIfChanged(out_file, contents)
     else:
       sys.stdout.write(contents)
 
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(main())