hg2git.py

   1 #!/usr/bin/env python
   2
   3 # Copyright (c) 2007, 2008 Rocco Rutte <pdmef@gmx.net> and others.
   4 # License: MIT <http://www.opensource.org/licenses/mit-license.php>
   5
   6 from mercurial import hg,ui
   7 import re
   8 import os
   9 import sys
  10 from subprocess import PIPE,Popen
  11
  12 # default git branch name
  13 cfg_master='master'
  14 # default origin name
  15 origin_name=''
  16 # default email address when unknown
  17 unknown_addr='unknown'
  18 split_name_re = re.compile(r'^((?:[^<]|<at>)*?)(?:<(?!at>)(?:(?:([^>]*)|(?:(.*)>(.*)))))?$', re.S|re.I)
  19 git_crud = '\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f' \
  20            '\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f' \
  21            ' .,:;<>"\\' "'"
  22 git_delch_re = re.compile(r'[<>\n]+', re.S)
  23 spelled_at_re = re.compile(r'^(.*?)(?:@|(?:(?:^|\s)<at>(?:\s|$)))', re.I)
  24
  25 def set_default_branch(name):
  26   global cfg_master
  27   cfg_master = name
  28
  29 def set_origin_name(name):
  30   global origin_name
  31   origin_name = name
  32
  33 def setup_repo(url):
  34   try:
  35     myui=ui.ui(interactive=False)
  36   except TypeError:
  37     myui=ui.ui()
  38     myui.setconfig('ui', 'interactive', 'off')
  39   return myui,hg.repository(myui,url)
  40
  41 # Git strips "crud" characters off both the beginning and end of the user's name
  42 # and the user's email, then deletes any remaining '<', '>' and '\n' characters
  43 # before combining the user name with the user email surrounded by '<' and '>'.
  44 # This function provides the crud-stripping and deletion operation that is used
  45 # on both the name and email.
  46 def gitname(name):
  47   name = name.strip(git_crud)
  48   return git_delch_re.sub('', name)
  49
  50 def set_unknown_addr(addr):
  51   global unknown_addr
  52   ans=False
  53   if addr!=None:
  54     addr=gitname(addr)
  55     if addr!='':
  56       unknown_addr=addr
  57       ans=True
  58   return ans
  59
  60 # Split the combined name and email input into a separate name and email and
  61 # apply Git's rules to each part.  The idea is to use anything to the left of
  62 # the first '<' and to the right of the last '>' as the name.  Anything between
  63 # the first '<' and the last '>' is treated as the email.  If there is no '<'
  64 # but the name contains '@' (which may be spelled out) then treat the entire
  65 # thing as an email with no name.  If the detected name is empty then anything
  66 # up to the first '@' (which may be spelled out) in the email is used for the
  67 # name.  Failing that the entire email is used for the name.
  68 def split_name_email(combined):
  69   name = ''
  70   email,rawemail = '',''
  71   match = split_name_re.match(combined)
  72   if match:
  73     left,rest,mid,right = match.groups()
  74     if rest != None:
  75       name = gitname(left)
  76       rawemail = rest
  77       email = gitname(rawemail)
  78     elif mid != None:
  79       name = gitname(left.rstrip() + ' ' + right.lstrip())
  80       rawemail = mid
  81       email = gitname(rawemail)
  82     else:
  83       name = gitname(left)
  84   if email == '' and spelled_at_re.match(left):
  85     rawemail = left
  86     email = name
  87     name = ''
  88   if name == '':
  89     at = spelled_at_re.match(rawemail)
  90     if at:
  91       name = gitname(at.group(1))
  92     if name == '':
  93       name = email
  94     # We do this test to be compatible with the previous behavior of hg2git.py
  95     # When it's given a <email> without any name and email does not contain '@'
  96     # then it sets the email to the unknown address
  97     if (len(left) < 2 or left[-1] != ' ') and not at:
  98       email = ''
  99   return [name, email]
 100
 101 def fixup_user(user,authors):
 102   user=user.strip("\"")
 103   if authors!=None:
 104     # if we have an authors table, try to get mapping
 105     # by defaulting to the current value of 'user'
 106     user=authors.get(user,user)
 107   name,mail=split_name_email(user)
 108   if mail == '':
 109     # If we don't have an email address replace it with unknown_addr.
 110     mail = unknown_addr
 111   if name == '':
 112     # Git does not like an empty name either -- split_name_email can only
 113     # return an empty name if it also returns an empty email.  This probably
 114     # will never happen since the input would have to be empty or only "crud"
 115     # characters, but check just to be safe.
 116     name = '-'
 117   return '%s <%s>' % (name,mail)
 118
 119 def get_branch(name):
 120   # 'HEAD' is the result of a bug in mutt's cvs->hg conversion,
 121   # other CVS imports may need it, too
 122   if name=='HEAD' or name=='default' or name=='':
 123     name=cfg_master
 124   if origin_name:
 125     return origin_name + '/' + name
 126   return name
 127
 128 def get_changeset(ui,repo,revision,authors={}):
 129   node=repo.lookup(revision)
 130   (manifest,user,(time,timezone),files,desc,extra)=repo.changelog.read(node)
 131   tz="%+03d%02d" % (-timezone / 3600, ((-timezone % 3600) / 60))
 132   branch=get_branch(extra.get('branch','master'))
 133   return (node,manifest,fixup_user(user,authors),(time,tz),files,desc,branch,extra)
 134
 135 def mangle_key(key):
 136   return key
 137
 138 def load_cache(filename,get_key=mangle_key):
 139   cache={}
 140   if not os.path.exists(filename):
 141     return cache
 142   f=open(filename,'r')
 143   l=0
 144   for line in f.readlines():
 145     l+=1
 146     fields=line.split(' ')
 147     if fields==None or not len(fields)==2 or fields[0][0]!=':':
 148       sys.stderr.write('Invalid file format in [%s], line %d\n' % (filename,l))
 149       continue
 150     # put key:value in cache, key without ^:
 151     cache[get_key(fields[0][1:])]=fields[1].split('\n')[0]
 152   f.close()
 153   return cache
 154
 155 def save_cache(filename,cache):
 156   f=open(filename,'w+')
 157   map(lambda x: f.write(':%s %s\n' % (str(x),str(cache.get(x)))),cache.keys())
 158   f.close()
 159
 160 def get_git_sha1(name,type='heads'):
 161   try:
 162     # use git-rev-parse to support packed refs
 163     ref="refs/%s/%s" % (type,name)
 164     proc = Popen(["git", "rev-parse", "--verify", "--quiet", ref], stdout=PIPE)
 165     l = proc.communicate()[0]
 166     if l == None or len(l) == 0:
 167       return None
 168     return l[0:40]
 169   except:
 170     return None