Andrew's git - gitweb.git/blob - 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
  22import os, os.path, sys
  23import tempfile, popen2, pickle, getopt
  24import re
  25
  26# Maps hg version -> git version
  27hgvers = {}
  28# List of children for each hg revision
  29hgchildren = {}
  30# Current branch for each hg revision
  31hgbranch = {}
  32
  33#------------------------------------------------------------------------------
  34
  35def usage():
  36
  37        print """\
  38%s: [OPTIONS] <hgprj>
  39
  40options:
  41    -s, --gitstate=FILE: name of the state to be saved/read
  42                         for incrementals
  43
  44required:
  45    hgprj:  name of the HG project to import (directory)
  46""" % sys.argv[0]
  47
  48#------------------------------------------------------------------------------
  49
  50def 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
  70state = ''
  71
  72try:
  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')
  81except:
  82    usage()
  83    sys.exit(1)
  84
  85hgprj = args[0]
  86os.chdir(hgprj)
  87
  88if 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
  96tip = os.popen('hg tip | head -1 | cut -f 2 -d :').read().strip()
  97print 'tip is', tip
  98
  99# Calculate the branches
 100print 'analysing the branches...'
 101hgchildren["0"] = ()
 102hgbranch["0"] = "master"
 103for 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
 131if not hgvers.has_key("0"):
 132    print 'creating repository'
 133    os.system('git-init-db')
 134
 135# loop through every hg changeset
 136for 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
 225os.system('git-repack -a -d')
 226
 227# write the state for incrementals
 228if 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