Merge branch 'master' of git://repo.or.cz/git-gui
[git] / contrib / hg-to-git / hg-to-git.py
1 #! /usr/bin/python
2
3 """ hg-to-svn.py - A Mercurial to GIT converter
4
5     Copyright (C)2007 Stelian Pop <stelian@popies.net>
6
7     This program is free software; you can redistribute it and/or modify
8     it under the terms of the GNU General Public License as published by
9     the Free Software Foundation; either version 2, or (at your option)
10     any later version.
11
12     This program is distributed in the hope that it will be useful,
13     but WITHOUT ANY WARRANTY; without even the implied warranty of
14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15     GNU General Public License for more details.
16
17     You should have received a copy of the GNU General Public License
18     along with this program; if not, write to the Free Software
19     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 """
21
22 import os, os.path, sys
23 import tempfile, popen2, pickle, getopt
24 import re
25
26 # Maps hg version -> git version
27 hgvers = {}
28 # List of children for each hg revision
29 hgchildren = {}
30 # Current branch for each hg revision
31 hgbranch = {}
32
33 #------------------------------------------------------------------------------
34
35 def usage():
36
37         print """\
38 %s: [OPTIONS] <hgprj>
39
40 options:
41     -s, --gitstate=FILE: name of the state to be saved/read
42                          for incrementals
43
44 required:
45     hgprj:  name of the HG project to import (directory)
46 """ % sys.argv[0]
47
48 #------------------------------------------------------------------------------
49
50 def getgitenv(user, date):
51     env = ''
52     elems = re.compile('(.*?)\s+<(.*)>').match(user)
53     if elems:
54         env += 'export GIT_AUTHOR_NAME="%s" ;' % elems.group(1)
55         env += 'export GIT_COMMITER_NAME="%s" ;' % elems.group(1)
56         env += 'export GIT_AUTHOR_EMAIL="%s" ;' % elems.group(2)
57         env += 'export GIT_COMMITER_EMAIL="%s" ;' % elems.group(2)
58     else:
59         env += 'export GIT_AUTHOR_NAME="%s" ;' % user
60         env += 'export GIT_COMMITER_NAME="%s" ;' % user
61         env += 'export GIT_AUTHOR_EMAIL= ;'
62         env += 'export GIT_COMMITER_EMAIL= ;'
63
64     env += 'export GIT_AUTHOR_DATE="%s" ;' % date
65     env += 'export GIT_COMMITTER_DATE="%s" ;' % date
66     return env
67
68 #------------------------------------------------------------------------------
69
70 state = ''
71
72 try:
73     opts, args = getopt.getopt(sys.argv[1:], 's:t:', ['gitstate=', 'tempdir='])
74     for o, a in opts:
75         if o in ('-s', '--gitstate'):
76             state = a
77             state = os.path.abspath(state)
78
79     if len(args) != 1:
80         raise('params')
81 except:
82     usage()
83     sys.exit(1)
84
85 hgprj = args[0]
86 os.chdir(hgprj)
87
88 if state:
89     if os.path.exists(state):
90         print 'State does exist, reading'
91         f = open(state, 'r')
92         hgvers = pickle.load(f)
93     else:
94         print 'State does not exist, first run'
95
96 tip = os.popen('hg tip | head -1 | cut -f 2 -d :').read().strip()
97 print 'tip is', tip
98
99 # Calculate the branches
100 print 'analysing the branches...'
101 hgchildren["0"] = ()
102 hgbranch["0"] = "master"
103 for cset in range(1, int(tip) + 1):
104     hgchildren[str(cset)] = ()
105     prnts = os.popen('hg log -r %d | grep ^parent: | cut -f 2 -d :' % cset).readlines()
106     if len(prnts) > 0:
107         parent = prnts[0].strip()
108     else:
109         parent = str(cset - 1)
110     hgchildren[parent] += ( str(cset), )
111     if len(prnts) > 1:
112         mparent = prnts[1].strip()
113         hgchildren[mparent] += ( str(cset), )
114     else:
115         mparent = None
116
117     if mparent:
118         # For merge changesets, take either one, preferably the 'master' branch
119         if hgbranch[mparent] == 'master':
120             hgbranch[str(cset)] = 'master'
121         else:
122             hgbranch[str(cset)] = hgbranch[parent]
123     else:
124         # Normal changesets
125         # For first children, take the parent branch, for the others create a new branch
126         if hgchildren[parent][0] == str(cset):
127             hgbranch[str(cset)] = hgbranch[parent]
128         else:
129             hgbranch[str(cset)] = "branch-" + str(cset)
130
131 if not hgvers.has_key("0"):
132     print 'creating repository'
133     os.system('git-init-db')
134
135 # loop through every hg changeset
136 for cset in range(int(tip) + 1):
137
138     # incremental, already seen
139     if hgvers.has_key(str(cset)):
140         continue
141
142     # get info
143     prnts = os.popen('hg log -r %d | grep ^parent: | cut -f 2 -d :' % cset).readlines()
144     if len(prnts) > 0:
145         parent = prnts[0].strip()
146     else:
147         parent = str(cset - 1)
148     if len(prnts) > 1:
149         mparent = prnts[1].strip()
150     else:
151         mparent = None
152
153     (fdcomment, filecomment) = tempfile.mkstemp()
154     csetcomment = os.popen('hg log -r %d -v | grep -v ^changeset: | grep -v ^parent: | grep -v ^user: | grep -v ^date | grep -v ^files: | grep -v ^description: | grep -v ^tag:' % cset).read().strip()
155     os.write(fdcomment, csetcomment)
156     os.close(fdcomment)
157
158     date = os.popen('hg log -r %d | grep ^date: | cut -f 2- -d :' % cset).read().strip()
159
160     tag = os.popen('hg log -r %d | grep ^tag: | cut -f 2- -d :' % cset).read().strip()
161
162     user = os.popen('hg log -r %d | grep ^user: | cut -f 2- -d :' % cset).read().strip()
163
164     print '-----------------------------------------'
165     print 'cset:', cset
166     print 'branch:', hgbranch[str(cset)]
167     print 'user:', user
168     print 'date:', date
169     print 'comment:', csetcomment
170     print 'parent:', parent
171     if mparent:
172         print 'mparent:', mparent
173     if tag:
174         print 'tag:', tag
175     print '-----------------------------------------'
176
177     # checkout the parent if necessary
178     if cset != 0:
179         if hgbranch[str(cset)] == "branch-" + str(cset):
180             print 'creating new branch', hgbranch[str(cset)]
181             os.system('git-checkout -b %s %s' % (hgbranch[str(cset)], hgvers[parent]))
182         else:
183             print 'checking out branch', hgbranch[str(cset)]
184             os.system('git-checkout %s' % hgbranch[str(cset)])
185
186     # merge
187     if mparent:
188         if hgbranch[parent] == hgbranch[str(cset)]:
189             otherbranch = hgbranch[mparent]
190         else:
191             otherbranch = hgbranch[parent]
192         print 'merging', otherbranch, 'into', hgbranch[str(cset)]
193         os.system(getgitenv(user, date) + 'git-merge --no-commit -s ours "" %s %s' % (hgbranch[str(cset)], otherbranch))
194
195     # remove everything except .git and .hg directories
196     os.system('find . \( -path "./.hg" -o -path "./.git" \) -prune -o ! -name "." -print | xargs rm -rf')
197
198     # repopulate with checkouted files
199     os.system('hg update -C %d' % cset)
200
201     # add new files
202     os.system('git-ls-files -x .hg --others | git-update-index --add --stdin')
203     # delete removed files
204     os.system('git-ls-files -x .hg --deleted | git-update-index --remove --stdin')
205
206     # commit
207     os.system(getgitenv(user, date) + 'git-commit -a -F %s' % filecomment)
208     os.unlink(filecomment)
209
210     # tag
211     if tag and tag != 'tip':
212         os.system(getgitenv(user, date) + 'git-tag %s' % tag)
213
214     # delete branch if not used anymore...
215     if mparent and len(hgchildren[str(cset)]):
216         print "Deleting unused branch:", otherbranch
217         os.system('git-branch -d %s' % otherbranch)
218
219     # retrieve and record the version
220     vvv = os.popen('git-show | head -1').read()
221     vvv = vvv[vvv.index(' ') + 1 : ].strip()
222     print 'record', cset, '->', vvv
223     hgvers[str(cset)] = vvv
224
225 os.system('git-repack -a -d')
226
227 # write the state for incrementals
228 if state:
229     print 'Writing state'
230     f = open(state, 'w')
231     pickle.dump(hgvers, f)
232
233 # vim: et ts=8 sw=4 sts=4