Merge pull request #6 from aried3r/master
[fast-export.git] / hg-reset.py
blob000ed6d5cbe4a7a6353377295c8b9c6361dcbb48
1 #!/usr/bin/env python
3 # Copyright (c) 2007, 2008 Rocco Rutte <pdmef@gmx.net> and others.
4 # License: GPLv2
6 from mercurial import node
7 from hg2git import setup_repo,load_cache,get_changeset,get_git_sha1
8 from optparse import OptionParser
9 import sys
11 def heads(ui,repo,start=None,stop=None,max=None):
12 # this is copied from mercurial/revlog.py and differs only in
13 # accepting a max argument for xrange(startrev+1,...) defaulting
14 # to the original repo.changelog.count()
15 if start is None:
16 start = node.nullid
17 if stop is None:
18 stop = []
19 if max is None:
20 max = repo.changelog.count()
21 stoprevs = dict.fromkeys([repo.changelog.rev(n) for n in stop])
22 startrev = repo.changelog.rev(start)
23 reachable = {startrev: 1}
24 heads = {startrev: 1}
26 parentrevs = repo.changelog.parentrevs
27 for r in xrange(startrev + 1, max):
28 for p in parentrevs(r):
29 if p in reachable:
30 if r not in stoprevs:
31 reachable[r] = 1
32 heads[r] = 1
33 if p in heads and p not in stoprevs:
34 del heads[p]
36 return [(repo.changelog.node(r),str(r)) for r in heads]
38 def get_branches(ui,repo,heads_cache,marks_cache,mapping_cache,max):
39 h=heads(ui,repo,max=max)
40 stale=dict.fromkeys(heads_cache)
41 changed=[]
42 unchanged=[]
43 for node,rev in h:
44 _,_,user,(_,_),_,desc,branch,_=get_changeset(ui,repo,rev)
45 del stale[branch]
46 git_sha1=get_git_sha1(branch)
47 cache_sha1=marks_cache.get(str(int(rev)+1))
48 if git_sha1!=None and git_sha1==cache_sha1:
49 unchanged.append([branch,cache_sha1,rev,desc.split('\n')[0],user])
50 else:
51 changed.append([branch,cache_sha1,rev,desc.split('\n')[0],user])
52 changed.sort()
53 unchanged.sort()
54 return stale,changed,unchanged
56 def get_tags(ui,repo,marks_cache,mapping_cache,max):
57 l=repo.tagslist()
58 good,bad=[],[]
59 for tag,node in l:
60 if tag=='tip': continue
61 rev=int(mapping_cache[node.encode('hex_codec')])
62 cache_sha1=marks_cache.get(str(int(rev)+1))
63 _,_,user,(_,_),_,desc,branch,_=get_changeset(ui,repo,rev)
64 if int(rev)>int(max):
65 bad.append([tag,branch,cache_sha1,rev,desc.split('\n')[0],user])
66 else:
67 good.append([tag,branch,cache_sha1,rev,desc.split('\n')[0],user])
68 good.sort()
69 bad.sort()
70 return good,bad
72 def mangle_mark(mark):
73 return str(int(mark)-1)
75 if __name__=='__main__':
76 def bail(parser,opt):
77 sys.stderr.write('Error: No option %s given\n' % opt)
78 parser.print_help()
79 sys.exit(2)
81 parser=OptionParser()
83 parser.add_option("--marks",dest="marksfile",
84 help="File to read git-fast-import's marks from")
85 parser.add_option("--heads",dest="headsfile",
86 help="File to read last run's git heads from")
87 parser.add_option("--status",dest="statusfile",
88 help="File to read status from")
89 parser.add_option("-r","--repo",dest="repourl",
90 help="URL of repo to import")
91 parser.add_option("-R","--revision",type=int,dest="revision",
92 help="Revision to reset to")
94 (options,args)=parser.parse_args()
96 if options.marksfile==None: bail(parser,'--marks option')
97 if options.headsfile==None: bail(parser,'--heads option')
98 if options.statusfile==None: bail(parser,'--status option')
99 if options.repourl==None: bail(parser,'--repo option')
100 if options.revision==None: bail(parser,'-R/--revision')
102 heads_cache=load_cache(options.headsfile)
103 marks_cache=load_cache(options.marksfile,mangle_mark)
104 state_cache=load_cache(options.statusfile)
106 l=int(state_cache.get('tip',options.revision))
107 if options.revision+1>l:
108 sys.stderr.write('Revision is beyond last revision imported: %d>%d\n' % (options.revision,l))
109 sys.exit(1)
111 ui,repo=setup_repo(options.repourl)
113 stale,changed,unchanged=get_branches(ui,repo,heads_cache,marks_cache,mapping_cache,options.revision+1)
114 good,bad=get_tags(ui,repo,marks_cache,mapping_cache,options.revision+1)
116 print "Possibly stale branches:"
117 map(lambda b: sys.stdout.write('\t%s\n' % b),stale.keys())
119 print "Possibly stale tags:"
120 map(lambda b: sys.stdout.write('\t%s on %s (r%s)\n' % (b[0],b[1],b[3])),bad)
122 print "Unchanged branches:"
123 map(lambda b: sys.stdout.write('\t%s (r%s)\n' % (b[0],b[2])),unchanged)
125 print "Unchanged tags:"
126 map(lambda b: sys.stdout.write('\t%s on %s (r%s)\n' % (b[0],b[1],b[3])),good)
128 print "Reset branches in '%s' to:" % options.headsfile
129 map(lambda b: sys.stdout.write('\t:%s %s\n\t\t(r%s: %s: %s)\n' % (b[0],b[1],b[2],b[4],b[3])),changed)
131 print "Reset ':tip' in '%s' to '%d'" % (options.statusfile,options.revision)