Allow pickaxe and diff-filter options to be used by git log.
[git.git] / git-merge-recursive.py
1 #!/usr/bin/python
2 #
3 # Copyright (C) 2005 Fredrik Kuivinen
4 #
5
6 import sys
7 sys.path.append('''@@GIT_PYTHON_PATH@@''')
8
9 import math, random, os, re, signal, tempfile, stat, errno, traceback
10 from heapq import heappush, heappop
11 from sets import Set
12
13 from gitMergeCommon import *
14
15 outputIndent = 0
16 def output(*args):
17     sys.stdout.write('  '*outputIndent)
18     printList(args)
19
20 originalIndexFile = os.environ.get('GIT_INDEX_FILE',
21                                    os.environ.get('GIT_DIR', '.git') + '/index')
22 temporaryIndexFile = os.environ.get('GIT_DIR', '.git') + \
23                      '/merge-recursive-tmp-index'
24 def setupIndex(temporary):
25     try:
26         os.unlink(temporaryIndexFile)
27     except OSError:
28         pass
29     if temporary:
30         newIndex = temporaryIndexFile
31     else:
32         newIndex = originalIndexFile
33     os.environ['GIT_INDEX_FILE'] = newIndex
34
35 # This is a global variable which is used in a number of places but
36 # only written to in the 'merge' function.
37
38 # cacheOnly == True  => Don't leave any non-stage 0 entries in the cache and
39 #                       don't update the working directory.
40 #              False => Leave unmerged entries in the cache and update
41 #                       the working directory.
42
43 cacheOnly = False
44
45 # The entry point to the merge code
46 # ---------------------------------
47
48 def merge(h1, h2, branch1Name, branch2Name, graph, callDepth=0, ancestor=None):
49     '''Merge the commits h1 and h2, return the resulting virtual
50     commit object and a flag indicating the cleaness of the merge.'''
51     assert(isinstance(h1, Commit) and isinstance(h2, Commit))
52
53     global outputIndent
54
55     output('Merging:')
56     output(h1)
57     output(h2)
58     sys.stdout.flush()
59
60     if ancestor:
61         ca = [ancestor]
62     else:
63         assert(isinstance(graph, Graph))
64         ca = getCommonAncestors(graph, h1, h2)
65     output('found', len(ca), 'common ancestor(s):')
66     for x in ca:
67         output(x)
68     sys.stdout.flush()
69
70     mergedCA = ca[0]
71     for h in ca[1:]:
72         outputIndent = callDepth+1
73         [mergedCA, dummy] = merge(mergedCA, h,
74                                   'Temporary merge branch 1',
75                                   'Temporary merge branch 2',
76                                   graph, callDepth+1)
77         outputIndent = callDepth
78         assert(isinstance(mergedCA, Commit))
79
80     global cacheOnly
81     if callDepth == 0:
82         setupIndex(False)
83         cacheOnly = False
84     else:
85         setupIndex(True)
86         runProgram(['git-read-tree', h1.tree()])
87         cacheOnly = True
88
89     [shaRes, clean] = mergeTrees(h1.tree(), h2.tree(), mergedCA.tree(),
90                                  branch1Name, branch2Name)
91
92     if graph and (clean or cacheOnly):
93         res = Commit(None, [h1, h2], tree=shaRes)
94         graph.addNode(res)
95     else:
96         res = None
97
98     return [res, clean]
99
100 getFilesRE = re.compile(r'^([0-7]+) (\S+) ([0-9a-f]{40})\t(.*)$', re.S)
101 def getFilesAndDirs(tree):
102     files = Set()
103     dirs = Set()
104     out = runProgram(['git-ls-tree', '-r', '-z', '-t', tree])
105     for l in out.split('\0'):
106         m = getFilesRE.match(l)
107         if m:
108             if m.group(2) == 'tree':
109                 dirs.add(m.group(4))
110             elif m.group(2) == 'blob':
111                 files.add(m.group(4))
112
113     return [files, dirs]
114
115 # Those two global variables are used in a number of places but only
116 # written to in 'mergeTrees' and 'uniquePath'. They keep track of
117 # every file and directory in the two branches that are about to be
118 # merged.
119 currentFileSet = None
120 currentDirectorySet = None
121
122 def mergeTrees(head, merge, common, branch1Name, branch2Name):
123     '''Merge the trees 'head' and 'merge' with the common ancestor
124     'common'. The name of the head branch is 'branch1Name' and the name of
125     the merge branch is 'branch2Name'. Return a tuple (tree, cleanMerge)
126     where tree is the resulting tree and cleanMerge is True iff the
127     merge was clean.'''
128     
129     assert(isSha(head) and isSha(merge) and isSha(common))
130
131     if common == merge:
132         output('Already uptodate!')
133         return [head, True]
134
135     if cacheOnly:
136         updateArg = '-i'
137     else:
138         updateArg = '-u'
139
140     [out, code] = runProgram(['git-read-tree', updateArg, '-m',
141                                 common, head, merge], returnCode = True)
142     if code != 0:
143         die('git-read-tree:', out)
144
145     [tree, code] = runProgram('git-write-tree', returnCode=True)
146     tree = tree.rstrip()
147     if code != 0:
148         global currentFileSet, currentDirectorySet
149         [currentFileSet, currentDirectorySet] = getFilesAndDirs(head)
150         [filesM, dirsM] = getFilesAndDirs(merge)
151         currentFileSet.union_update(filesM)
152         currentDirectorySet.union_update(dirsM)
153
154         entries = unmergedCacheEntries()
155         renamesHead =  getRenames(head, common, head, merge, entries)
156         renamesMerge = getRenames(merge, common, head, merge, entries)
157
158         cleanMerge = processRenames(renamesHead, renamesMerge,
159                                     branch1Name, branch2Name)
160         for entry in entries:
161             if entry.processed:
162                 continue
163             if not processEntry(entry, branch1Name, branch2Name):
164                 cleanMerge = False
165                 
166         if cleanMerge or cacheOnly:
167             tree = runProgram('git-write-tree').rstrip()
168         else:
169             tree = None
170     else:
171         cleanMerge = True
172
173     return [tree, cleanMerge]
174
175 # Low level file merging, update and removal
176 # ------------------------------------------
177
178 def mergeFile(oPath, oSha, oMode, aPath, aSha, aMode, bPath, bSha, bMode,
179               branch1Name, branch2Name):
180
181     merge = False
182     clean = True
183
184     if stat.S_IFMT(aMode) != stat.S_IFMT(bMode):
185         clean = False
186         if stat.S_ISREG(aMode):
187             mode = aMode
188             sha = aSha
189         else:
190             mode = bMode
191             sha = bSha
192     else:
193         if aSha != oSha and bSha != oSha:
194             merge = True
195
196         if aMode == oMode:
197             mode = bMode
198         else:
199             mode = aMode
200
201         if aSha == oSha:
202             sha = bSha
203         elif bSha == oSha:
204             sha = aSha
205         elif stat.S_ISREG(aMode):
206             assert(stat.S_ISREG(bMode))
207
208             orig = runProgram(['git-unpack-file', oSha]).rstrip()
209             src1 = runProgram(['git-unpack-file', aSha]).rstrip()
210             src2 = runProgram(['git-unpack-file', bSha]).rstrip()
211             try:
212                 [out, code] = runProgram(['merge',
213                                           '-L', branch1Name + '/' + aPath,
214                                           '-L', 'orig/' + oPath,
215                                           '-L', branch2Name + '/' + bPath,
216                                           src1, orig, src2], returnCode=True)
217             except ProgramError, e:
218                 print >>sys.stderr, e
219                 die("Failed to execute 'merge'. merge(1) is used as the "
220                     "file-level merge tool. Is 'merge' in your path?")
221
222             sha = runProgram(['git-hash-object', '-t', 'blob', '-w',
223                               src1]).rstrip()
224
225             os.unlink(orig)
226             os.unlink(src1)
227             os.unlink(src2)
228
229             clean = (code == 0)
230         else:
231             assert(stat.S_ISLNK(aMode) and stat.S_ISLNK(bMode))
232             sha = aSha
233
234             if aSha != bSha:
235                 clean = False
236
237     return [sha, mode, clean, merge]
238
239 def updateFile(clean, sha, mode, path):
240     updateCache = cacheOnly or clean
241     updateWd = not cacheOnly
242
243     return updateFileExt(sha, mode, path, updateCache, updateWd)
244
245 def updateFileExt(sha, mode, path, updateCache, updateWd):
246     if cacheOnly:
247         updateWd = False
248
249     if updateWd:
250         pathComponents = path.split('/')
251         for x in xrange(1, len(pathComponents)):
252             p = '/'.join(pathComponents[0:x])
253
254             try:
255                 createDir = not stat.S_ISDIR(os.lstat(p).st_mode)
256             except OSError:
257                 createDir = True
258             
259             if createDir:
260                 try:
261                     os.mkdir(p)
262                 except OSError, e:
263                     die("Couldn't create directory", p, e.strerror)
264
265         prog = ['git-cat-file', 'blob', sha]
266         if stat.S_ISREG(mode):
267             try:
268                 os.unlink(path)
269             except OSError:
270                 pass
271             if mode & 0100:
272                 mode = 0777
273             else:
274                 mode = 0666
275             fd = os.open(path, os.O_WRONLY | os.O_TRUNC | os.O_CREAT, mode)
276             proc = subprocess.Popen(prog, stdout=fd)
277             proc.wait()
278             os.close(fd)
279         elif stat.S_ISLNK(mode):
280             linkTarget = runProgram(prog)
281             os.symlink(linkTarget, path)
282         else:
283             assert(False)
284
285     if updateWd and updateCache:
286         runProgram(['git-update-index', '--add', '--', path])
287     elif updateCache:
288         runProgram(['git-update-index', '--add', '--cacheinfo',
289                     '0%o' % mode, sha, path])
290
291 def setIndexStages(path,
292                    oSHA1, oMode,
293                    aSHA1, aMode,
294                    bSHA1, bMode,
295                    clear=True):
296     istring = []
297     if clear:
298         istring.append("0 " + ("0" * 40) + "\t" + path + "\0")
299     if oMode:
300         istring.append("%o %s %d\t%s\0" % (oMode, oSHA1, 1, path))
301     if aMode:
302         istring.append("%o %s %d\t%s\0" % (aMode, aSHA1, 2, path))
303     if bMode:
304         istring.append("%o %s %d\t%s\0" % (bMode, bSHA1, 3, path))
305
306     runProgram(['git-update-index', '-z', '--index-info'],
307                input="".join(istring))
308
309 def removeFile(clean, path):
310     updateCache = cacheOnly or clean
311     updateWd = not cacheOnly
312
313     if updateCache:
314         runProgram(['git-update-index', '--force-remove', '--', path])
315
316     if updateWd:
317         try:
318             os.unlink(path)
319         except OSError, e:
320             if e.errno != errno.ENOENT and e.errno != errno.EISDIR:
321                 raise
322         try:
323             os.removedirs(os.path.dirname(path))
324         except OSError:
325             pass
326
327 def uniquePath(path, branch):
328     def fileExists(path):
329         try:
330             os.lstat(path)
331             return True
332         except OSError, e:
333             if e.errno == errno.ENOENT:
334                 return False
335             else:
336                 raise
337
338     branch = branch.replace('/', '_')
339     newPath = path + '~' + branch
340     suffix = 0
341     while newPath in currentFileSet or \
342           newPath in currentDirectorySet  or \
343           fileExists(newPath):
344         suffix += 1
345         newPath = path + '~' + branch + '_' + str(suffix)
346     currentFileSet.add(newPath)
347     return newPath
348
349 # Cache entry management
350 # ----------------------
351
352 class CacheEntry:
353     def __init__(self, path):
354         class Stage:
355             def __init__(self):
356                 self.sha1 = None
357                 self.mode = None
358
359             # Used for debugging only
360             def __str__(self):
361                 if self.mode != None:
362                     m = '0%o' % self.mode
363                 else:
364                     m = 'None'
365
366                 if self.sha1:
367                     sha1 = self.sha1
368                 else:
369                     sha1 = 'None'
370                 return 'sha1: ' + sha1 + ' mode: ' + m
371         
372         self.stages = [Stage(), Stage(), Stage(), Stage()]
373         self.path = path
374         self.processed = False
375
376     def __str__(self):
377         return 'path: ' + self.path + ' stages: ' + repr([str(x) for x in self.stages])
378
379 class CacheEntryContainer:
380     def __init__(self):
381         self.entries = {}
382
383     def add(self, entry):
384         self.entries[entry.path] = entry
385
386     def get(self, path):
387         return self.entries.get(path)
388
389     def __iter__(self):
390         return self.entries.itervalues()
391     
392 unmergedRE = re.compile(r'^([0-7]+) ([0-9a-f]{40}) ([1-3])\t(.*)$', re.S)
393 def unmergedCacheEntries():
394     '''Create a dictionary mapping file names to CacheEntry
395     objects. The dictionary contains one entry for every path with a
396     non-zero stage entry.'''
397
398     lines = runProgram(['git-ls-files', '-z', '--unmerged']).split('\0')
399     lines.pop()
400
401     res = CacheEntryContainer()
402     for l in lines:
403         m = unmergedRE.match(l)
404         if m:
405             mode = int(m.group(1), 8)
406             sha1 = m.group(2)
407             stage = int(m.group(3))
408             path = m.group(4)
409
410             e = res.get(path)
411             if not e:
412                 e = CacheEntry(path)
413                 res.add(e)
414
415             e.stages[stage].mode = mode
416             e.stages[stage].sha1 = sha1
417         else:
418             die('Error: Merge program failed: Unexpected output from',
419                 'git-ls-files:', l)
420     return res
421
422 lsTreeRE = re.compile(r'^([0-7]+) (\S+) ([0-9a-f]{40})\t(.*)\n$', re.S)
423 def getCacheEntry(path, origTree, aTree, bTree):
424     '''Returns a CacheEntry object which doesn't have to correspond to
425     a real cache entry in Git's index.'''
426     
427     def parse(out):
428         if out == '':
429             return [None, None]
430         else:
431             m = lsTreeRE.match(out)
432             if not m:
433                 die('Unexpected output from git-ls-tree:', out)
434             elif m.group(2) == 'blob':
435                 return [m.group(3), int(m.group(1), 8)]
436             else:
437                 return [None, None]
438
439     res = CacheEntry(path)
440
441     [oSha, oMode] = parse(runProgram(['git-ls-tree', origTree, '--', path]))
442     [aSha, aMode] = parse(runProgram(['git-ls-tree', aTree, '--', path]))
443     [bSha, bMode] = parse(runProgram(['git-ls-tree', bTree, '--', path]))
444
445     res.stages[1].sha1 = oSha
446     res.stages[1].mode = oMode
447     res.stages[2].sha1 = aSha
448     res.stages[2].mode = aMode
449     res.stages[3].sha1 = bSha
450     res.stages[3].mode = bMode
451
452     return res
453
454 # Rename detection and handling
455 # -----------------------------
456
457 class RenameEntry:
458     def __init__(self,
459                  src, srcSha, srcMode, srcCacheEntry,
460                  dst, dstSha, dstMode, dstCacheEntry,
461                  score):
462         self.srcName = src
463         self.srcSha = srcSha
464         self.srcMode = srcMode
465         self.srcCacheEntry = srcCacheEntry
466         self.dstName = dst
467         self.dstSha = dstSha
468         self.dstMode = dstMode
469         self.dstCacheEntry = dstCacheEntry
470         self.score = score
471
472         self.processed = False
473
474 class RenameEntryContainer:
475     def __init__(self):
476         self.entriesSrc = {}
477         self.entriesDst = {}
478
479     def add(self, entry):
480         self.entriesSrc[entry.srcName] = entry
481         self.entriesDst[entry.dstName] = entry
482
483     def getSrc(self, path):
484         return self.entriesSrc.get(path)
485
486     def getDst(self, path):
487         return self.entriesDst.get(path)
488
489     def __iter__(self):
490         return self.entriesSrc.itervalues()
491
492 parseDiffRenamesRE = re.compile('^:([0-7]+) ([0-7]+) ([0-9a-f]{40}) ([0-9a-f]{40}) R([0-9]*)$')
493 def getRenames(tree, oTree, aTree, bTree, cacheEntries):
494     '''Get information of all renames which occured between 'oTree' and
495     'tree'. We need the three trees in the merge ('oTree', 'aTree' and
496     'bTree') to be able to associate the correct cache entries with
497     the rename information. 'tree' is always equal to either aTree or bTree.'''
498
499     assert(tree == aTree or tree == bTree)
500     inp = runProgram(['git-diff-tree', '-M', '--diff-filter=R', '-r',
501                       '-z', oTree, tree])
502
503     ret = RenameEntryContainer()
504     try:
505         recs = inp.split("\0")
506         recs.pop() # remove last entry (which is '')
507         it = recs.__iter__()
508         while True:
509             rec = it.next()
510             m = parseDiffRenamesRE.match(rec)
511
512             if not m:
513                 die('Unexpected output from git-diff-tree:', rec)
514
515             srcMode = int(m.group(1), 8)
516             dstMode = int(m.group(2), 8)
517             srcSha = m.group(3)
518             dstSha = m.group(4)
519             score = m.group(5)
520             src = it.next()
521             dst = it.next()
522
523             srcCacheEntry = cacheEntries.get(src)
524             if not srcCacheEntry:
525                 srcCacheEntry = getCacheEntry(src, oTree, aTree, bTree)
526                 cacheEntries.add(srcCacheEntry)
527
528             dstCacheEntry = cacheEntries.get(dst)
529             if not dstCacheEntry:
530                 dstCacheEntry = getCacheEntry(dst, oTree, aTree, bTree)
531                 cacheEntries.add(dstCacheEntry)
532
533             ret.add(RenameEntry(src, srcSha, srcMode, srcCacheEntry,
534                                 dst, dstSha, dstMode, dstCacheEntry,
535                                 score))
536     except StopIteration:
537         pass
538     return ret
539
540 def fmtRename(src, dst):
541     srcPath = src.split('/')
542     dstPath = dst.split('/')
543     path = []
544     endIndex = min(len(srcPath), len(dstPath)) - 1
545     for x in range(0, endIndex):
546         if srcPath[x] == dstPath[x]:
547             path.append(srcPath[x])
548         else:
549             endIndex = x
550             break
551
552     if len(path) > 0:
553         return '/'.join(path) + \
554                '/{' + '/'.join(srcPath[endIndex:]) + ' => ' + \
555                '/'.join(dstPath[endIndex:]) + '}'
556     else:
557         return src + ' => ' + dst
558
559 def processRenames(renamesA, renamesB, branchNameA, branchNameB):
560     srcNames = Set()
561     for x in renamesA:
562         srcNames.add(x.srcName)
563     for x in renamesB:
564         srcNames.add(x.srcName)
565
566     cleanMerge = True
567     for path in srcNames:
568         if renamesA.getSrc(path):
569             renames1 = renamesA
570             renames2 = renamesB
571             branchName1 = branchNameA
572             branchName2 = branchNameB
573         else:
574             renames1 = renamesB
575             renames2 = renamesA
576             branchName1 = branchNameB
577             branchName2 = branchNameA
578         
579         ren1 = renames1.getSrc(path)
580         ren2 = renames2.getSrc(path)
581
582         ren1.dstCacheEntry.processed = True
583         ren1.srcCacheEntry.processed = True
584
585         if ren1.processed:
586             continue
587
588         ren1.processed = True
589
590         if ren2:
591             # Renamed in 1 and renamed in 2
592             assert(ren1.srcName == ren2.srcName)
593             ren2.dstCacheEntry.processed = True
594             ren2.processed = True
595
596             if ren1.dstName != ren2.dstName:
597                 output('CONFLICT (rename/rename): Rename',
598                        fmtRename(path, ren1.dstName), 'in branch', branchName1,
599                        'rename', fmtRename(path, ren2.dstName), 'in',
600                        branchName2)
601                 cleanMerge = False
602
603                 if ren1.dstName in currentDirectorySet:
604                     dstName1 = uniquePath(ren1.dstName, branchName1)
605                     output(ren1.dstName, 'is a directory in', branchName2,
606                            'adding as', dstName1, 'instead.')
607                     removeFile(False, ren1.dstName)
608                 else:
609                     dstName1 = ren1.dstName
610
611                 if ren2.dstName in currentDirectorySet:
612                     dstName2 = uniquePath(ren2.dstName, branchName2)
613                     output(ren2.dstName, 'is a directory in', branchName1,
614                            'adding as', dstName2, 'instead.')
615                     removeFile(False, ren2.dstName)
616                 else:
617                     dstName2 = ren2.dstName
618                 setIndexStages(dstName1,
619                                None, None,
620                                ren1.dstSha, ren1.dstMode,
621                                None, None)
622                 setIndexStages(dstName2,
623                                None, None,
624                                None, None,
625                                ren2.dstSha, ren2.dstMode)
626
627             else:
628                 removeFile(True, ren1.srcName)
629
630                 [resSha, resMode, clean, merge] = \
631                          mergeFile(ren1.srcName, ren1.srcSha, ren1.srcMode,
632                                    ren1.dstName, ren1.dstSha, ren1.dstMode,
633                                    ren2.dstName, ren2.dstSha, ren2.dstMode,
634                                    branchName1, branchName2)
635
636                 if merge or not clean:
637                     output('Renaming', fmtRename(path, ren1.dstName))
638
639                 if merge:
640                     output('Auto-merging', ren1.dstName)
641
642                 if not clean:
643                     output('CONFLICT (content): merge conflict in',
644                            ren1.dstName)
645                     cleanMerge = False
646
647                     if not cacheOnly:
648                         setIndexStages(ren1.dstName,
649                                        ren1.srcSha, ren1.srcMode,
650                                        ren1.dstSha, ren1.dstMode,
651                                        ren2.dstSha, ren2.dstMode)
652
653                 updateFile(clean, resSha, resMode, ren1.dstName)
654         else:
655             removeFile(True, ren1.srcName)
656
657             # Renamed in 1, maybe changed in 2
658             if renamesA == renames1:
659                 stage = 3
660             else:
661                 stage = 2
662                 
663             srcShaOtherBranch  = ren1.srcCacheEntry.stages[stage].sha1
664             srcModeOtherBranch = ren1.srcCacheEntry.stages[stage].mode
665
666             dstShaOtherBranch  = ren1.dstCacheEntry.stages[stage].sha1
667             dstModeOtherBranch = ren1.dstCacheEntry.stages[stage].mode
668
669             tryMerge = False
670             
671             if ren1.dstName in currentDirectorySet:
672                 newPath = uniquePath(ren1.dstName, branchName1)
673                 output('CONFLICT (rename/directory): Rename',
674                        fmtRename(ren1.srcName, ren1.dstName), 'in', branchName1,
675                        'directory', ren1.dstName, 'added in', branchName2)
676                 output('Renaming', ren1.srcName, 'to', newPath, 'instead')
677                 cleanMerge = False
678                 removeFile(False, ren1.dstName)
679                 updateFile(False, ren1.dstSha, ren1.dstMode, newPath)
680             elif srcShaOtherBranch == None:
681                 output('CONFLICT (rename/delete): Rename',
682                        fmtRename(ren1.srcName, ren1.dstName), 'in',
683                        branchName1, 'and deleted in', branchName2)
684                 cleanMerge = False
685                 updateFile(False, ren1.dstSha, ren1.dstMode, ren1.dstName)
686             elif dstShaOtherBranch:
687                 newPath = uniquePath(ren1.dstName, branchName2)
688                 output('CONFLICT (rename/add): Rename',
689                        fmtRename(ren1.srcName, ren1.dstName), 'in',
690                        branchName1 + '.', ren1.dstName, 'added in', branchName2)
691                 output('Adding as', newPath, 'instead')
692                 updateFile(False, dstShaOtherBranch, dstModeOtherBranch, newPath)
693                 cleanMerge = False
694                 tryMerge = True
695             elif renames2.getDst(ren1.dstName):
696                 dst2 = renames2.getDst(ren1.dstName)
697                 newPath1 = uniquePath(ren1.dstName, branchName1)
698                 newPath2 = uniquePath(dst2.dstName, branchName2)
699                 output('CONFLICT (rename/rename): Rename',
700                        fmtRename(ren1.srcName, ren1.dstName), 'in',
701                        branchName1+'. Rename',
702                        fmtRename(dst2.srcName, dst2.dstName), 'in', branchName2)
703                 output('Renaming', ren1.srcName, 'to', newPath1, 'and',
704                        dst2.srcName, 'to', newPath2, 'instead')
705                 removeFile(False, ren1.dstName)
706                 updateFile(False, ren1.dstSha, ren1.dstMode, newPath1)
707                 updateFile(False, dst2.dstSha, dst2.dstMode, newPath2)
708                 dst2.processed = True
709                 cleanMerge = False
710             else:
711                 tryMerge = True
712
713             if tryMerge:
714
715                 oName, oSHA1, oMode = ren1.srcName, ren1.srcSha, ren1.srcMode
716                 aName, bName = ren1.dstName, ren1.srcName
717                 aSHA1, bSHA1 = ren1.dstSha, srcShaOtherBranch
718                 aMode, bMode = ren1.dstMode, srcModeOtherBranch
719                 aBranch, bBranch = branchName1, branchName2
720
721                 if renamesA != renames1:
722                     aName, bName = bName, aName
723                     aSHA1, bSHA1 = bSHA1, aSHA1
724                     aMode, bMode = bMode, aMode
725                     aBranch, bBranch = bBranch, aBranch
726
727                 [resSha, resMode, clean, merge] = \
728                          mergeFile(oName, oSHA1, oMode,
729                                    aName, aSHA1, aMode,
730                                    bName, bSHA1, bMode,
731                                    aBranch, bBranch);
732
733                 if merge or not clean:
734                     output('Renaming', fmtRename(ren1.srcName, ren1.dstName))
735
736                 if merge:
737                     output('Auto-merging', ren1.dstName)
738
739                 if not clean:
740                     output('CONFLICT (rename/modify): Merge conflict in',
741                            ren1.dstName)
742                     cleanMerge = False
743
744                     if not cacheOnly:
745                         setIndexStages(ren1.dstName,
746                                        oSHA1, oMode,
747                                        aSHA1, aMode,
748                                        bSHA1, bMode)
749
750                 updateFile(clean, resSha, resMode, ren1.dstName)
751
752     return cleanMerge
753
754 # Per entry merge function
755 # ------------------------
756
757 def processEntry(entry, branch1Name, branch2Name):
758     '''Merge one cache entry.'''
759
760     debug('processing', entry.path, 'clean cache:', cacheOnly)
761
762     cleanMerge = True
763
764     path = entry.path
765     oSha = entry.stages[1].sha1
766     oMode = entry.stages[1].mode
767     aSha = entry.stages[2].sha1
768     aMode = entry.stages[2].mode
769     bSha = entry.stages[3].sha1
770     bMode = entry.stages[3].mode
771
772     assert(oSha == None or isSha(oSha))
773     assert(aSha == None or isSha(aSha))
774     assert(bSha == None or isSha(bSha))
775
776     assert(oMode == None or type(oMode) is int)
777     assert(aMode == None or type(aMode) is int)
778     assert(bMode == None or type(bMode) is int)
779
780     if (oSha and (not aSha or not bSha)):
781     #
782     # Case A: Deleted in one
783     #
784         if (not aSha     and not bSha) or \
785            (aSha == oSha and not bSha) or \
786            (not aSha     and bSha == oSha):
787     # Deleted in both or deleted in one and unchanged in the other
788             if aSha:
789                 output('Removing', path)
790             removeFile(True, path)
791         else:
792     # Deleted in one and changed in the other
793             cleanMerge = False
794             if not aSha:
795                 output('CONFLICT (delete/modify):', path, 'deleted in',
796                        branch1Name, 'and modified in', branch2Name + '.',
797                        'Version', branch2Name, 'of', path, 'left in tree.')
798                 mode = bMode
799                 sha = bSha
800             else:
801                 output('CONFLICT (modify/delete):', path, 'deleted in',
802                        branch2Name, 'and modified in', branch1Name + '.',
803                        'Version', branch1Name, 'of', path, 'left in tree.')
804                 mode = aMode
805                 sha = aSha
806
807             updateFile(False, sha, mode, path)
808
809     elif (not oSha and aSha     and not bSha) or \
810          (not oSha and not aSha and bSha):
811     #
812     # Case B: Added in one.
813     #
814         if aSha:
815             addBranch = branch1Name
816             otherBranch = branch2Name
817             mode = aMode
818             sha = aSha
819             conf = 'file/directory'
820         else:
821             addBranch = branch2Name
822             otherBranch = branch1Name
823             mode = bMode
824             sha = bSha
825             conf = 'directory/file'
826     
827         if path in currentDirectorySet:
828             cleanMerge = False
829             newPath = uniquePath(path, addBranch)
830             output('CONFLICT (' + conf + '):',
831                    'There is a directory with name', path, 'in',
832                    otherBranch + '. Adding', path, 'as', newPath)
833
834             removeFile(False, path)
835             updateFile(False, sha, mode, newPath)
836         else:
837             output('Adding', path)
838             updateFile(True, sha, mode, path)
839     
840     elif not oSha and aSha and bSha:
841     #
842     # Case C: Added in both (check for same permissions).
843     #
844         if aSha == bSha:
845             if aMode != bMode:
846                 cleanMerge = False
847                 output('CONFLICT: File', path,
848                        'added identically in both branches, but permissions',
849                        'conflict', '0%o' % aMode, '->', '0%o' % bMode)
850                 output('CONFLICT: adding with permission:', '0%o' % aMode)
851
852                 updateFile(False, aSha, aMode, path)
853             else:
854                 # This case is handled by git-read-tree
855                 assert(False)
856         else:
857             cleanMerge = False
858             newPath1 = uniquePath(path, branch1Name)
859             newPath2 = uniquePath(path, branch2Name)
860             output('CONFLICT (add/add): File', path,
861                    'added non-identically in both branches. Adding as',
862                    newPath1, 'and', newPath2, 'instead.')
863             removeFile(False, path)
864             updateFile(False, aSha, aMode, newPath1)
865             updateFile(False, bSha, bMode, newPath2)
866
867     elif oSha and aSha and bSha:
868     #
869     # case D: Modified in both, but differently.
870     #
871         output('Auto-merging', path)
872         [sha, mode, clean, dummy] = \
873               mergeFile(path, oSha, oMode,
874                         path, aSha, aMode,
875                         path, bSha, bMode,
876                         branch1Name, branch2Name)
877         if clean:
878             updateFile(True, sha, mode, path)
879         else:
880             cleanMerge = False
881             output('CONFLICT (content): Merge conflict in', path)
882
883             if cacheOnly:
884                 updateFile(False, sha, mode, path)
885             else:
886                 updateFileExt(sha, mode, path, updateCache=False, updateWd=True)
887     else:
888         die("ERROR: Fatal merge failure, shouldn't happen.")
889
890     return cleanMerge
891
892 def usage():
893     die('Usage:', sys.argv[0], ' <base>... -- <head> <remote>..')
894
895 # main entry point as merge strategy module
896 # The first parameters up to -- are merge bases, and the rest are heads.
897
898 if len(sys.argv) < 4:
899     usage()
900
901 bases = []
902 for nextArg in xrange(1, len(sys.argv)):
903     if sys.argv[nextArg] == '--':
904         if len(sys.argv) != nextArg + 3:
905             die('Not handling anything other than two heads merge.')
906         try:
907             h1 = firstBranch = sys.argv[nextArg + 1]
908             h2 = secondBranch = sys.argv[nextArg + 2]
909         except IndexError:
910             usage()
911         break
912     else:
913         bases.append(sys.argv[nextArg])
914
915 print 'Merging', h1, 'with', h2
916
917 try:
918     h1 = runProgram(['git-rev-parse', '--verify', h1 + '^0']).rstrip()
919     h2 = runProgram(['git-rev-parse', '--verify', h2 + '^0']).rstrip()
920
921     if len(bases) == 1:
922         base = runProgram(['git-rev-parse', '--verify',
923                            bases[0] + '^0']).rstrip()
924         ancestor = Commit(base, None)
925         [dummy, clean] = merge(Commit(h1, None), Commit(h2, None),
926                                firstBranch, secondBranch, None, 0,
927                                ancestor)
928     else:
929         graph = buildGraph([h1, h2])
930         [dummy, clean] = merge(graph.shaMap[h1], graph.shaMap[h2],
931                                firstBranch, secondBranch, graph)
932
933     print ''
934 except:
935     if isinstance(sys.exc_info()[1], SystemExit):
936         raise
937     else:
938         traceback.print_exc(None, sys.stderr)
939         sys.exit(2)
940
941 if clean:
942     sys.exit(0)
943 else:
944     sys.exit(1)