hg-fast-export: add option to track remote branches under a custom namespace
[python/fast-export.git] / 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 repo,hg,cmdutil,util,ui,revlog,node
7 import re
8 import os
9 import sys
10
11 # default git branch name
12 cfg_master='master'
13 # default origin name
14 origin_name=''
15 # silly regex to see if user field has email address
16 user_re=re.compile('([^<]+) (<[^>]+>)$')
17 # silly regex to clean out user names
18 user_clean_re=re.compile('^["]([^"]+)["]$')
19
20 def set_default_branch(name):
21   global cfg_master
22   cfg_master = name
23
24 def set_origin_name(name):
25   global origin_name
26   origin_name = name
27
28 def setup_repo(url):
29   myui=ui.ui(interactive=False)
30   return myui,hg.repository(myui,url)
31
32 def fixup_user(user,authors):
33   if authors!=None:
34     # if we have an authors table, try to get mapping
35     # by defaulting to the current value of 'user'
36     user=authors.get(user,user)
37   name,mail,m='','',user_re.match(user)
38   if m==None:
39     # if we don't have 'Name <mail>' syntax, use 'user
40     # <devnull@localhost>' if use contains no at and
41     # 'user <user>' otherwise
42     name=user
43     if '@' not in user:
44       mail='<devnull@localhost>'
45     else:
46       mail='<%s>' % user
47   else:
48     # if we have 'Name <mail>' syntax, everything is fine :)
49     name,mail=m.group(1),m.group(2)
50
51   # remove any silly quoting from username
52   m2=user_clean_re.match(name)
53   if m2!=None:
54     name=m2.group(1)
55   return '%s %s' % (name,mail)
56
57 def get_branch(name):
58   # 'HEAD' is the result of a bug in mutt's cvs->hg conversion,
59   # other CVS imports may need it, too
60   if name=='HEAD' or name=='default' or name=='':
61     name=cfg_master
62   if origin_name:
63     return origin_name + '/' + name
64   return name
65
66 def get_changeset(ui,repo,revision,authors={}):
67   node=repo.lookup(revision)
68   (manifest,user,(time,timezone),files,desc,extra)=repo.changelog.read(node)
69   tz="%+03d%02d" % (-timezone / 3600, ((-timezone % 3600) / 60))
70   branch=get_branch(extra.get('branch','master'))
71   return (node,manifest,fixup_user(user,authors),(time,tz),files,desc,branch,extra)
72
73 def mangle_key(key):
74   return key
75
76 def load_cache(filename,get_key=mangle_key):
77   cache={}
78   if not os.path.exists(filename):
79     return cache
80   f=open(filename,'r')
81   l=0
82   for line in f.readlines():
83     l+=1
84     fields=line.split(' ')
85     if fields==None or not len(fields)==2 or fields[0][0]!=':':
86       sys.stderr.write('Invalid file format in [%s], line %d\n' % (filename,l))
87       continue
88     # put key:value in cache, key without ^:
89     cache[get_key(fields[0][1:])]=fields[1].split('\n')[0]
90   f.close()
91   return cache
92
93 def save_cache(filename,cache):
94   f=open(filename,'w+')
95   map(lambda x: f.write(':%s %s\n' % (str(x),str(cache.get(x)))),cache.keys())
96   f.close()
97
98 def get_git_sha1(name,type='heads'):
99   try:
100     # use git-rev-parse to support packed refs
101     cmd="GIT_DIR='%s' git rev-parse --verify refs/%s/%s 2>/dev/null" % (os.getenv('GIT_DIR','/dev/null'),type,name)
102     p=os.popen(cmd)
103     l=p.readline()
104     p.close()
105     if l == None or len(l) == 0:
106       return None
107     return l[0:40]
108   except IOError:
109     return None