~bzr-pqm/bzr/bzr.dev

1110 by Martin Pool
- merge aaron's merge improvements:
1
# Copyright (C) 2005 Canonical Ltd
2
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
12
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
17
1185.1.2 by Martin Pool
- merge various windows and other fixes from Ollie Rutherfurd
18
import os
1113 by Martin Pool
- fix is_ancestor import problem in merge
19
import tempfile
20
import shutil
21
import errno
22
23
import bzrlib.osutils
24
import bzrlib.revision
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
25
from bzrlib.merge_core import merge_flex, ApplyMerge3, BackupBeforeChange
26
from bzrlib.changeset import generate_changeset, ExceptionConflictHandler
1185.10.8 by Aaron Bentley
Conflict handling where OTHER is deleted
27
from bzrlib.changeset import Inventory, Diff3Merge, ReplaceContents
1185.2.11 by Lalo Martins
killed all uses of find_branch() if favor of the new constructors.
28
from bzrlib.branch import Branch
974.1.80 by Aaron Bentley
Improved merge error handling and testing
29
from bzrlib.errors import BzrCommandError, UnrelatedBranches, NoCommonAncestor
30
from bzrlib.errors import NoCommits
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
31
from bzrlib.delta import compare_trees
1113 by Martin Pool
- fix is_ancestor import problem in merge
32
from bzrlib.trace import mutter, warning
1390 by Robert Collins
pair programming worx... merge integration and weave
33
from bzrlib.fetch import greedy_fetch, fetch
1116 by Martin Pool
- fix a few errors in new merge code
34
from bzrlib.revision import is_ancestor
1185.1.41 by Robert Collins
massive patch from Alexander Belchenko - many PEP8 fixes, removes unused function uuid
35
from bzrlib.osutils import rename
1390 by Robert Collins
pair programming worx... merge integration and weave
36
from bzrlib.revision import common_ancestor, MultipleRevisionSources
37
from bzrlib.errors import NoSuchRevision
1185.1.41 by Robert Collins
massive patch from Alexander Belchenko - many PEP8 fixes, removes unused function uuid
38
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
39
40
# comments from abentley on irc: merge happens in two stages, each
41
# of which generates a changeset object
42
43
# stage 1: generate OLD->OTHER,
44
# stage 2: use MINE and OLD->OTHER to generate MINE -> RESULT
45
493 by Martin Pool
- Merge aaron's merge command
46
class MergeConflictHandler(ExceptionConflictHandler):
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
47
    """Handle conflicts encountered while merging.
48
49
    This subclasses ExceptionConflictHandler, so that any types of
50
    conflict that are not explicitly handled cause an exception and
51
    terminate the merge.
52
    """
1185.10.10 by Aaron Bentley
Handled modified files missing from THIS
53
    def __init__(self, this_tree, base_tree, other_tree, ignore_zero=False):
974.1.83 by Aaron Bentley
Removed unused dir parameter from ExceptionConflictHandler
54
        ExceptionConflictHandler.__init__(self)
622 by Martin Pool
Updated merge patch from Aaron
55
        self.conflicts = 0
56
        self.ignore_zero = ignore_zero
1185.10.10 by Aaron Bentley
Handled modified files missing from THIS
57
        self.this_tree = this_tree
58
        self.base_tree = base_tree
59
        self.other_tree = other_tree
622 by Martin Pool
Updated merge patch from Aaron
60
493 by Martin Pool
- Merge aaron's merge command
61
    def copy(self, source, dest):
62
        """Copy the text and mode of a file
63
        :param source: The path of the file to copy
64
        :param dest: The distination file to create
65
        """
66
        s_file = file(source, "rb")
67
        d_file = file(dest, "wb")
68
        for line in s_file:
69
            d_file.write(line)
70
        os.chmod(dest, 0777 & os.stat(source).st_mode)
71
974.1.23 by Aaron Bentley
Avoided unnecessary temp files
72
    def dump(self, lines, dest):
73
        """Copy the text and mode of a file
74
        :param source: The path of the file to copy
75
        :param dest: The distination file to create
76
        """
77
        d_file = file(dest, "wb")
78
        for line in lines:
79
            d_file.write(line)
80
493 by Martin Pool
- Merge aaron's merge command
81
    def add_suffix(self, name, suffix, last_new_name=None):
82
        """Rename a file to append a suffix.  If the new name exists, the
83
        suffix is added repeatedly until a non-existant name is found
84
85
        :param name: The path of the file
86
        :param suffix: The suffix to append
87
        :param last_new_name: (used for recursive calls) the last name tried
88
        """
89
        if last_new_name is None:
90
            last_new_name = name
91
        new_name = last_new_name+suffix
92
        try:
1185.1.40 by Robert Collins
Merge what applied of Alexander Belchenko's win32 patch.
93
            rename(name, new_name)
622 by Martin Pool
Updated merge patch from Aaron
94
            return new_name
493 by Martin Pool
- Merge aaron's merge command
95
        except OSError, e:
96
            if e.errno != errno.EEXIST and e.errno != errno.ENOTEMPTY:
97
                raise
622 by Martin Pool
Updated merge patch from Aaron
98
            return self.add_suffix(name, suffix, last_new_name=new_name)
99
100
    def conflict(self, text):
101
        warning(text)
102
        self.conflicts += 1
103
        
493 by Martin Pool
- Merge aaron's merge command
104
974.1.23 by Aaron Bentley
Avoided unnecessary temp files
105
    def merge_conflict(self, new_file, this_path, base_lines, other_lines):
493 by Martin Pool
- Merge aaron's merge command
106
        """
107
        Handle diff3 conflicts by producing a .THIS, .BASE and .OTHER.  The
108
        main file will be a version with diff3 conflicts.
109
        :param new_file: Path to the output file with diff3 markers
110
        :param this_path: Path to the file text for the THIS tree
111
        :param base_path: Path to the file text for the BASE tree
112
        :param other_path: Path to the file text for the OTHER tree
113
        """
114
        self.add_suffix(this_path, ".THIS")
974.1.23 by Aaron Bentley
Avoided unnecessary temp files
115
        self.dump(base_lines, this_path+".BASE")
116
        self.dump(other_lines, this_path+".OTHER")
1185.1.40 by Robert Collins
Merge what applied of Alexander Belchenko's win32 patch.
117
        rename(new_file, this_path)
622 by Martin Pool
Updated merge patch from Aaron
118
        self.conflict("Diff3 conflict encountered in %s" % this_path)
493 by Martin Pool
- Merge aaron's merge command
119
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
120
    def new_contents_conflict(self, filename, other_contents):
121
        """Conflicting contents for newly added file."""
122
        self.copy(other_contents, filename + ".OTHER")
123
        self.conflict("Conflict in newly added file %s" % filename)
124
    
125
493 by Martin Pool
- Merge aaron's merge command
126
    def target_exists(self, entry, target, old_path):
127
        """Handle the case when the target file or dir exists"""
622 by Martin Pool
Updated merge patch from Aaron
128
        moved_path = self.add_suffix(target, ".moved")
129
        self.conflict("Moved existing %s to %s" % (target, moved_path))
130
850 by Martin Pool
- Merge merge updates from aaron
131
    def rmdir_non_empty(self, filename):
132
        """Handle the case where the dir to be removed still has contents"""
133
        self.conflict("Directory %s not removed because it is not empty"\
134
            % filename)
135
        return "skip"
136
1185.10.8 by Aaron Bentley
Conflict handling where OTHER is deleted
137
    def rem_contents_conflict(self, filename, this_contents, base_contents):
138
        base_contents(filename+".BASE", self, False)
139
        this_contents(filename+".THIS", self, False)
1185.10.9 by Aaron Bentley
Added conflict message
140
        self.conflict("Other branch deleted locally modified file %s" %
141
                      filename)
1185.10.8 by Aaron Bentley
Conflict handling where OTHER is deleted
142
        return ReplaceContents(this_contents, None)
143
1185.10.10 by Aaron Bentley
Handled modified files missing from THIS
144
    def abs_this_path(self, file_id):
145
        """Return the absolute path for a file_id in the this tree."""
146
        relpath = self.this_tree.id2path(file_id)
147
        return self.this_tree.tree.abspath(relpath)
148
149
    def add_missing_parents(self, file_id, tree):
150
        """If some of the parents for file_id are missing, add them."""
151
        entry = tree.tree.inventory[file_id]
152
        if entry.parent_id not in self.this_tree:
153
            return self.create_all_missing(entry.parent_id, tree)
154
        else:
155
            return self.abs_this_path(entry.parent_id)
156
157
    def create_all_missing(self, file_id, tree):
158
        """Add contents for a file_id and all its parents to a tree."""
159
        entry = tree.tree.inventory[file_id]
160
        if entry.parent_id is not None and entry.parent_id not in self.this_tree:
161
            abspath = self.create_all_missing(entry.parent_id, tree)
162
        else:
163
            abspath = self.abs_this_path(entry.parent_id)
164
        entry_path = os.path.join(abspath, entry.name)
165
        if not os.path.isdir(entry_path):
166
            self.create(file_id, entry_path, tree)
167
        return entry_path
168
169
    def create(self, file_id, path, tree, reverse=False):
170
        """Uses tree data to create a filesystem object for the file_id"""
171
        from merge_core import get_id_contents
172
        get_id_contents(file_id, tree)(path, self, reverse)
173
174
    def missing_for_merge(self, file_id, other_path):
175
        """The file_id doesn't exist in THIS, but does in OTHER and BASE"""
176
        self.conflict("Other branch modified locally deleted file %s" %
177
                      other_path)
178
        parent_dir = self.add_missing_parents(file_id, self.other_tree)
179
        stem = os.path.join(parent_dir, os.path.basename(other_path))
180
        self.create(file_id, stem+".OTHER", self.other_tree)
181
        self.create(file_id, stem+".BASE", self.base_tree)
182
622 by Martin Pool
Updated merge patch from Aaron
183
    def finalize(self):
184
        if not self.ignore_zero:
185
            print "%d conflicts encountered.\n" % self.conflicts
493 by Martin Pool
- Merge aaron's merge command
186
            
974.1.32 by aaron.bentley at utoronto
Made merge do greedy fetching.
187
def get_tree(treespec, temp_root, label, local_branch=None):
622 by Martin Pool
Updated merge patch from Aaron
188
    location, revno = treespec
1185.8.5 by Aaron Bentley
Fixed non-tree-root bug in branch, revert, merge
189
    branch = Branch.open_containing(location)
493 by Martin Pool
- Merge aaron's merge command
190
    if revno is None:
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
191
        revision = None
192
    elif revno == -1:
1241 by Martin Pool
- rename last_patch to last_revision
193
        revision = branch.last_revision()
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
194
    else:
1185.2.7 by Lalo Martins
getting rid of branch.lookup_revision()
195
        revision = branch.get_rev_id(revno)
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
196
    return branch, get_revid_tree(branch, revision, temp_root, label,
197
                                  local_branch)
198
199
def get_revid_tree(branch, revision, temp_root, label, local_branch):
200
    if revision is None:
493 by Martin Pool
- Merge aaron's merge command
201
        base_tree = branch.working_tree()
202
    else:
974.1.32 by aaron.bentley at utoronto
Made merge do greedy fetching.
203
        if local_branch is not None:
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
204
            greedy_fetch(local_branch, branch, revision)
974.1.32 by aaron.bentley at utoronto
Made merge do greedy fetching.
205
            base_tree = local_branch.revision_tree(revision)
206
        else:
207
            base_tree = branch.revision_tree(revision)
493 by Martin Pool
- Merge aaron's merge command
208
    temp_path = os.path.join(temp_root, label)
209
    os.mkdir(temp_path)
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
210
    return MergeTree(base_tree, temp_path)
493 by Martin Pool
- Merge aaron's merge command
211
212
213
def file_exists(tree, file_id):
214
    return tree.has_filename(tree.id2path(file_id))
215
    
216
217
class MergeTree(object):
218
    def __init__(self, tree, tempdir):
219
        object.__init__(self)
220
        if hasattr(tree, "basedir"):
221
            self.root = tree.basedir
222
        else:
223
            self.root = None
224
        self.tree = tree
225
        self.tempdir = tempdir
226
        os.mkdir(os.path.join(self.tempdir, "texts"))
227
        self.cached = {}
228
974.1.19 by Aaron Bentley
Removed MergeTree.inventory
229
    def __iter__(self):
230
        return self.tree.__iter__()
231
974.1.12 by aaron.bentley at utoronto
Switched from text-id to hashcache for merge optimization
232
    def __contains__(self, file_id):
974.1.14 by aaron.bentley at utoronto
Fixed bugs in merge optimization
233
        return file_id in self.tree
974.1.12 by aaron.bentley at utoronto
Switched from text-id to hashcache for merge optimization
234
974.1.23 by Aaron Bentley
Avoided unnecessary temp files
235
    def get_file(self, file_id):
236
        return self.tree.get_file(file_id)
237
974.1.12 by aaron.bentley at utoronto
Switched from text-id to hashcache for merge optimization
238
    def get_file_sha1(self, id):
239
        return self.tree.get_file_sha1(id)
240
974.1.18 by Aaron Bentley
Stopped using SourceFile in inventory
241
    def id2path(self, file_id):
242
        return self.tree.id2path(file_id)
243
244
    def has_id(self, file_id):
245
        return self.tree.has_id(file_id)
246
974.1.47 by Aaron Bentley
Merged changes from the merge4 branch
247
    def has_or_had_id(self, file_id):
248
        if file_id == self.tree.inventory.root.file_id:
249
            return True
250
        return self.tree.inventory.has_id(file_id)
251
974.1.50 by aaron.bentley at utoronto
Merge of robertc@robertcollins.net-20050826013321-52eee1f1da679ee9
252
    def has_or_had_id(self, file_id):
253
        if file_id == self.tree.inventory.root.file_id:
254
            return True
255
        return self.tree.inventory.has_id(file_id)
256
493 by Martin Pool
- Merge aaron's merge command
257
    def readonly_path(self, id):
850 by Martin Pool
- Merge merge updates from aaron
258
        if id not in self.tree:
259
            return None
493 by Martin Pool
- Merge aaron's merge command
260
        if self.root is not None:
261
            return self.tree.abspath(self.tree.id2path(id))
262
        else:
263
            if self.tree.inventory[id].kind in ("directory", "root_directory"):
264
                return self.tempdir
265
            if not self.cached.has_key(id):
266
                path = os.path.join(self.tempdir, "texts", id)
267
                outfile = file(path, "wb")
268
                outfile.write(self.tree.get_file(id).read())
269
                assert(os.path.exists(path))
270
                self.cached[id] = path
271
            return self.cached[id]
272
628 by Martin Pool
- merge aaron's updated merge/pull code
273
274
275
def merge(other_revision, base_revision,
276
          check_clean=True, ignore_zero=False,
974.1.10 by aaron.bentley at utoronto
Added file-selection to merge-revert
277
          this_dir=None, backup_files=False, merge_type=ApplyMerge3,
278
          file_list=None):
628 by Martin Pool
- merge aaron's updated merge/pull code
279
    """Merge changes into a tree.
280
281
    base_revision
1092.1.38 by Robert Collins
make a default merge choose a sane base with branch.common_ancestor
282
        tuple(path, revision) Base for three-way merge.
628 by Martin Pool
- merge aaron's updated merge/pull code
283
    other_revision
1092.1.38 by Robert Collins
make a default merge choose a sane base with branch.common_ancestor
284
        tuple(path, revision) Other revision for three-way merge.
628 by Martin Pool
- merge aaron's updated merge/pull code
285
    this_dir
286
        Directory to merge changes into; '.' by default.
287
    check_clean
288
        If true, this_dir must have no uncommitted changes before the
289
        merge begins.
1390 by Robert Collins
pair programming worx... merge integration and weave
290
291
    All available ancestors of other_revision and base_revision are
974.1.32 by aaron.bentley at utoronto
Made merge do greedy fetching.
292
    automatically pulled into the branch.
628 by Martin Pool
- merge aaron's updated merge/pull code
293
    """
493 by Martin Pool
- Merge aaron's merge command
294
    tempdir = tempfile.mkdtemp(prefix="bzr-")
295
    try:
628 by Martin Pool
- merge aaron's updated merge/pull code
296
        if this_dir is None:
297
            this_dir = '.'
1185.2.11 by Lalo Martins
killed all uses of find_branch() if favor of the new constructors.
298
        this_branch = Branch.open_containing(this_dir)
1241 by Martin Pool
- rename last_patch to last_revision
299
        this_rev_id = this_branch.last_revision()
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
300
        if this_rev_id is None:
301
            raise BzrCommandError("This branch has no commits")
628 by Martin Pool
- merge aaron's updated merge/pull code
302
        if check_clean:
622 by Martin Pool
Updated merge patch from Aaron
303
            changes = compare_trees(this_branch.working_tree(), 
304
                                    this_branch.basis_tree(), False)
305
            if changes.has_changed():
306
                raise BzrCommandError("Working tree has uncommitted changes.")
974.1.32 by aaron.bentley at utoronto
Made merge do greedy fetching.
307
        other_branch, other_tree = get_tree(other_revision, tempdir, "other",
308
                                            this_branch)
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
309
        if other_revision[1] == -1:
1241 by Martin Pool
- rename last_patch to last_revision
310
            other_rev_id = other_branch.last_revision()
974.1.80 by Aaron Bentley
Improved merge error handling and testing
311
            if other_rev_id is None:
312
                raise NoCommits(other_branch)
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
313
            other_basis = other_rev_id
314
        elif other_revision[1] is not None:
1185.2.7 by Lalo Martins
getting rid of branch.lookup_revision()
315
            other_rev_id = other_branch.get_rev_id(other_revision[1])
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
316
            other_basis = other_rev_id
317
        else:
318
            other_rev_id = None
1241 by Martin Pool
- rename last_patch to last_revision
319
            other_basis = other_branch.last_revision()
974.1.80 by Aaron Bentley
Improved merge error handling and testing
320
            if other_basis is None:
321
                raise NoCommits(other_branch)
622 by Martin Pool
Updated merge patch from Aaron
322
        if base_revision == [None, None]:
974.1.80 by Aaron Bentley
Improved merge error handling and testing
323
            try:
324
                base_rev_id = common_ancestor(this_rev_id, other_basis, 
325
                                              this_branch)
326
            except NoCommonAncestor:
622 by Martin Pool
Updated merge patch from Aaron
327
                raise UnrelatedBranches()
974.1.37 by aaron.bentley at utoronto
pending merges, common ancestor work properly
328
            base_tree = get_revid_tree(this_branch, base_rev_id, tempdir, 
329
                                       "base", None)
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
330
            base_is_ancestor = True
331
        else:
332
            base_branch, base_tree = get_tree(base_revision, tempdir, "base")
333
            if base_revision[1] == -1:
1241 by Martin Pool
- rename last_patch to last_revision
334
                base_rev_id = base_branch.last_revision()
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
335
            elif base_revision[1] is None:
974.2.7 by aaron.bentley at utoronto
Merged from bzr.24
336
                base_rev_id = None
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
337
            else:
1185.2.7 by Lalo Martins
getting rid of branch.lookup_revision()
338
                base_rev_id = base_branch.get_rev_id(base_revision[1])
1390 by Robert Collins
pair programming worx... merge integration and weave
339
            fetch(from_branch=base_branch, to_branch=this_branch)
974.1.88 by Aaron Bentley
Set a pending_merge if the merge base is forced to revno 0
340
            base_is_ancestor = is_ancestor(this_rev_id, base_rev_id,
1390 by Robert Collins
pair programming worx... merge integration and weave
341
                                           this_branch)
974.1.10 by aaron.bentley at utoronto
Added file-selection to merge-revert
342
        if file_list is None:
343
            interesting_ids = None
344
        else:
345
            interesting_ids = set()
346
            this_tree = this_branch.working_tree()
347
            for fname in file_list:
348
                path = this_branch.relpath(fname)
349
                found_id = False
350
                for tree in (this_tree, base_tree.tree, other_tree.tree):
351
                    file_id = tree.inventory.path2id(path)
352
                    if file_id is not None:
353
                        interesting_ids.add(file_id)
354
                        found_id = True
355
                if not found_id:
356
                    raise BzrCommandError("%s is not a source file in any"
357
                                          " tree." % fname)
622 by Martin Pool
Updated merge patch from Aaron
358
        merge_inner(this_branch, other_tree, base_tree, tempdir, 
974.1.9 by Aaron Bentley
Added merge-type parameter to merge.
359
                    ignore_zero=ignore_zero, backup_files=backup_files, 
974.1.10 by aaron.bentley at utoronto
Added file-selection to merge-revert
360
                    merge_type=merge_type, interesting_ids=interesting_ids)
974.1.55 by aaron.bentley at utoronto
Fixed bug causing previous revision to be listed as a pending merge on pull
361
        if base_is_ancestor and other_rev_id is not None\
362
            and other_rev_id not in this_branch.revision_history():
974.1.36 by aaron.bentley at utoronto
Committed it even though the test case doesn't work
363
            this_branch.add_pending_merge(other_rev_id)
493 by Martin Pool
- Merge aaron's merge command
364
    finally:
365
        shutil.rmtree(tempdir)
366
367
974.1.10 by aaron.bentley at utoronto
Added file-selection to merge-revert
368
def set_interesting(inventory_a, inventory_b, interesting_ids):
369
    """Mark files whose ids are in interesting_ids as interesting
370
    """
371
    for inventory in (inventory_a, inventory_b):
372
        for path, source_file in inventory.iteritems():
373
             source_file.interesting = source_file.id in interesting_ids
374
375
974.1.19 by Aaron Bentley
Removed MergeTree.inventory
376
def generate_cset_optimized(tree_a, tree_b, interesting_ids=None):
974.1.12 by aaron.bentley at utoronto
Switched from text-id to hashcache for merge optimization
377
    """Generate a changeset.  If interesting_ids is supplied, only changes
378
    to those files will be shown.  Metadata changes are stripped.
974.1.10 by aaron.bentley at utoronto
Added file-selection to merge-revert
379
    """ 
974.1.19 by Aaron Bentley
Removed MergeTree.inventory
380
    cset =  generate_changeset(tree_a, tree_b, interesting_ids)
493 by Martin Pool
- Merge aaron's merge command
381
    for entry in cset.entries.itervalues():
382
        entry.metadata_change = None
383
    return cset
384
385
622 by Martin Pool
Updated merge patch from Aaron
386
def merge_inner(this_branch, other_tree, base_tree, tempdir, 
974.1.10 by aaron.bentley at utoronto
Added file-selection to merge-revert
387
                ignore_zero=False, merge_type=ApplyMerge3, backup_files=False,
388
                interesting_ids=None):
974.1.4 by Aaron Bentley
Implemented merge3 as the default text merge
389
974.1.23 by Aaron Bentley
Avoided unnecessary temp files
390
    def merge_factory(file_id, base, other):
391
        contents_change = merge_type(file_id, base, other)
974.1.8 by Aaron Bentley
Added default backups for merge-revert
392
        if backup_files:
393
            contents_change = BackupBeforeChange(contents_change)
394
        return contents_change
974.1.4 by Aaron Bentley
Implemented merge3 as the default text merge
395
628 by Martin Pool
- merge aaron's updated merge/pull code
396
    this_tree = get_tree((this_branch.base, None), tempdir, "this")[1]
493 by Martin Pool
- Merge aaron's merge command
397
398
    def get_inventory(tree):
974.1.19 by Aaron Bentley
Removed MergeTree.inventory
399
        return tree.tree.inventory
493 by Martin Pool
- Merge aaron's merge command
400
401
    inv_changes = merge_flex(this_tree, base_tree, other_tree,
974.1.85 by Aaron Bentley
Reverted incomplete conflict handling
402
                             generate_cset_optimized, get_inventory,
1185.10.10 by Aaron Bentley
Handled modified files missing from THIS
403
                             MergeConflictHandler(this_tree, base_tree,
404
                             other_tree, ignore_zero=ignore_zero),
974.1.17 by Aaron Bentley
Switched to using a set of interesting file_ids instead of SourceFile attribute
405
                             merge_factory=merge_factory, 
406
                             interesting_ids=interesting_ids)
493 by Martin Pool
- Merge aaron's merge command
407
408
    adjust_ids = []
409
    for id, path in inv_changes.iteritems():
410
        if path is not None:
411
            if path == '.':
412
                path = ''
413
            else:
1185.1.2 by Martin Pool
- merge various windows and other fixes from Ollie Rutherfurd
414
                assert path.startswith('.' + os.sep), "path is %s" % path
493 by Martin Pool
- Merge aaron's merge command
415
            path = path[2:]
416
        adjust_ids.append((path, id))
974.1.13 by aaron.bentley at utoronto
Avoid rewriting inventory when not necessary
417
    if len(adjust_ids) > 0:
418
        this_branch.set_inventory(regen_inventory(this_branch, this_tree.root,
419
                                                  adjust_ids))
493 by Martin Pool
- Merge aaron's merge command
420
421
422
def regen_inventory(this_branch, root, new_entries):
423
    old_entries = this_branch.read_working_inventory()
424
    new_inventory = {}
425
    by_path = {}
974.1.21 by aaron.bentley at utoronto
Handled path generation properly
426
    new_entries_map = {} 
427
    for path, file_id in new_entries:
428
        if path is None:
429
            continue
430
        new_entries_map[file_id] = path
431
432
    def id2path(file_id):
433
        path = new_entries_map.get(file_id)
434
        if path is not None:
435
            return path
436
        entry = old_entries[file_id]
437
        if entry.parent_id is None:
438
            return entry.name
439
        return os.path.join(id2path(entry.parent_id), entry.name)
440
        
493 by Martin Pool
- Merge aaron's merge command
441
    for file_id in old_entries:
442
        entry = old_entries[file_id]
974.1.21 by aaron.bentley at utoronto
Handled path generation properly
443
        path = id2path(file_id)
493 by Martin Pool
- Merge aaron's merge command
444
        new_inventory[file_id] = (path, file_id, entry.parent_id, entry.kind)
445
        by_path[path] = file_id
446
    
447
    deletions = 0
448
    insertions = 0
449
    new_path_list = []
450
    for path, file_id in new_entries:
451
        if path is None:
452
            del new_inventory[file_id]
453
            deletions += 1
454
        else:
455
            new_path_list.append((path, file_id))
456
            if file_id not in old_entries:
457
                insertions += 1
458
    # Ensure no file is added before its parent
459
    new_path_list.sort()
460
    for path, file_id in new_path_list:
461
        if path == '':
462
            parent = None
463
        else:
464
            parent = by_path[os.path.dirname(path)]
465
        kind = bzrlib.osutils.file_kind(os.path.join(root, path))
466
        new_inventory[file_id] = (path, file_id, parent, kind)
467
        by_path[path] = file_id 
468
469
    # Get a list in insertion order
470
    new_inventory_list = new_inventory.values()
471
    mutter ("""Inventory regeneration:
472
old length: %i insertions: %i deletions: %i new_length: %i"""\
473
        % (len(old_entries), insertions, deletions, len(new_inventory_list)))
474
    assert len(new_inventory_list) == len(old_entries) + insertions - deletions
475
    new_inventory_list.sort()
476
    return new_inventory_list
974.1.9 by Aaron Bentley
Added merge-type parameter to merge.
477
478
merge_types = {     "merge3": (ApplyMerge3, "Native diff3-style merge"), 
479
                     "diff3": (Diff3Merge,  "Merge using external diff3")
480
              }
481