Mention the author map file in the readme
[python/fast-export.git] / hg-reset.py
1 #!/usr/bin/env python
2
3 # Copyright (c) 2007, 2008 Rocco Rutte <pdmef@gmx.net> and others.
4 # License: GPLv2
5
6 from mercurial import repo,hg,cmdutil,util,ui,revlog,node
7 from hg2git import setup_repo,load_cache,get_changeset,get_git_sha1
8 from optparse import OptionParser
9 import sys
10
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}
25
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]
35
36   return [(repo.changelog.node(r),str(r)) for r in heads]
37
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
55
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
71
72 def mangle_mark(mark):
73   return str(int(mark)-1)
74
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)
80
81   parser=OptionParser()
82
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")
93
94   (options,args)=parser.parse_args()
95
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')
101
102   heads_cache=load_cache(options.headsfile)
103   marks_cache=load_cache(options.marksfile,mangle_mark)
104   state_cache=load_cache(options.statusfile)
105
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)
110
111   ui,repo=setup_repo(options.repourl)
112
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)
115
116   print "Possibly stale branches:"
117   map(lambda b: sys.stdout.write('\t%s\n' % b),stale.keys())
118
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)
121
122   print "Unchanged branches:"
123   map(lambda b: sys.stdout.write('\t%s (r%s)\n' % (b[0],b[2])),unchanged)
124
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)
127
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)
130
131   print "Reset ':tip' in '%s' to '%d'" % (options.statusfile,options.revision)