git-cms-cvs-history

#!/usr/bin/env python
from commands import getstatusoutput
from optparse import OptionParser
from sys import exit
from os import environ, chdir
from os.path import exists, join
import re

def format(s, **kwds):
  return s % kwds

def die(s, verbose=None):
  print s
  if verbose:
    print verbose
  exit(1)

def findSymbol(tag, package):
  cmssw_base = environ["CMSSW_BASE"]
  if tag == "HEAD":
    tag = "master"
  if tag != "master":
    branchAttempt = format("refs/remotes/%(package)s/%(package)s-%(tag)s",
                           package=package,
                           tag=tag)
  else:
    branchAttempt = format("refs/remotes/%(package)s/master",
                           package=package,
                           tag=tag)
  tagAttempt = format("refs/tags/%(package)s-%(tag)s",
                      package=package,
                      tag=tag)
  if exists(join(cmssw_base, "src/.git", branchAttempt)):
    tag = branchAttempt
  elif exists(join(cmssw_base, "src/.git", tagAttempt)):
    tag = tagAttempt
  else:
    print branchAttempt
    die("Unable to find symbol %s for package %s." % (tag, realPackage))
  return tag

if __name__ == "__main__":
  parser = OptionParser(usage="git cms-cvs-history diff <cvs-tag>[..<cvs-tag>] SubSystem/Package\n"
                              "                           import <cvs-tag> SubSystem/Package\n"
                              "                           tags SubSystem/Package\n"
                              "                           log SubSystem/Package")
  opts, args = parser.parse_args()
  if len(args) < 1:
    parser.error("`git cms-cvs-history' requires either `import`, `diff', `tags' or `log' sub-command")
  if len(args) == 1:
    parser.error("`git cms-cvs-history' requires %s requires a package" % args[0])
  command = args[0]
   
  if not command in ["diff", "import", "log", "tags"]:
    parser.error("`git cms-cvs-history' requires either `import`, `diff' or `log' sub-command")
  
  cmssw_base = environ.get("CMSSW_BASE", None)
  if not cmssw_base:
    die("Please setup your CMSSW area")

  chdir(join(cmssw_base, "src"))

  err, output = getstatusoutput(format("cd $CMSSW_BASE/src ; git init"))
  if err:
    die("Error while setting up git repository")
  # The last arg should be the package
  realPackage = args[-1]
  package = args[-1].replace("/","-")
  err, output = getstatusoutput("git remote show | grep %s" % package)
  if err:
    err, output = getstatusoutput(format("git remote add %(package)s git@github.com:cms-cvs-history/%(package)s.git",
                                         package=package))
    if err:
      die("Error while adding repository", output)
    
    err, output = getstatusoutput(format("git fetch %(package)s ;"
                                         "git fetch %(package)s --tags", package=package))
    if err:
      err, output2 = getstatusoutput(format("git remote rm %(package)s",
                                           package=package))
      die("Could not fetch info for package package %s." % package, output)

  # Poor man command dispatcher..
  if command == "diff":
    if len(args) != 3:
      parser.error("diff requires <cvs-tag>[..<cvs-tag>] <package>")
    diffArgs = args[1]
    tags = diffArgs.split("..")
    if len(tags) == 1:
      tags.append("HEAD")
    
    tags[0] = findSymbol(tags[0], package)
    tags[1] = findSymbol(tags[1], package)
      
    err, output = getstatusoutput("git diff %s..%s" % (tags[0], tags[1]))
    if err:
      die("Could not create diff", output)
    print output
    exit(0)
  elif command == "tags":
    if len(args) == 1:
      parser.error("Please specify a package")
    err, output = getstatusoutput(format("git tag | grep -e '^%(package)s' | sed -e's/^%(package)s-//'", 
                                         package=package))
    print output
    exit(0)
  elif command == "import":
    tag = "master"
    if len(args) == 3:
      tag = args[1]
    elif len(args) != 2:
      parser.error("`import` takes only <tag> and <package> as arguments.")
    tag = findSymbol(tag, package)
    command = format("mkdir -p $CMSSW_BASE/src/%(realPackage)s &&"
                     "cd $CMSSW_BASE/src &&"
                     "git archive --format=tar %(tag)s | (cd $CMSSW_BASE/src/%(realPackage)s ; tar xf -)",
                     package=package, 
                     tag=tag,
                     realPackage=realPackage)
    err, output = getstatusoutput(command)
    if err:
      die("Error while checking out %s" % realPackage, output)
    exit(0)
  elif command == "log":
    tag = "master"
    if len(args) == 3:
      tag = args[1]
    err, output = getstatusoutput("git log refs/remotes/%s/%s" % (package, tag))
    if err:
      die("Error while doing git log", output)
    print output