/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/merge.py

  • Committer: Martin Pool
  • Date: 2005-08-29 01:25:00 UTC
  • Revision ID: mbp@sourcefrog.net-20050829012500-36f2d20363be4a53
* move bzr-specific code from testsweet into bzrlib.selftest

* logging from within test suites is now done using python logging, so
  the regular .bzr.log is not cluttered and the results can be easily
  seen if the test fails

* don't capture stdout/stderr while running tests, instead let it leak
  through so that we can see places where the library is doing its own
  output and should be fixed.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/merge.py
1
 
from merge_core import merge_flex, ApplyMerge3, BackupBeforeChange
2
 
from changeset import generate_changeset, ExceptionConflictHandler
3
 
from changeset import Inventory, Diff3Merge
4
 
from bzrlib import find_branch
5
 
import bzrlib.osutils
6
 
from bzrlib.errors import BzrCommandError
7
 
from bzrlib.delta import compare_trees
8
 
from trace import mutter, warning
 
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
 
9
18
import os.path
10
19
import tempfile
11
20
import shutil
12
21
import errno
13
22
 
14
 
class UnrelatedBranches(BzrCommandError):
15
 
    def __init__(self):
16
 
        msg = "Branches have no common ancestor, and no base revision"\
17
 
            " specified."
18
 
        BzrCommandError.__init__(self, msg)
19
 
 
 
23
import bzrlib.osutils
 
24
import bzrlib.revision
 
25
from bzrlib.merge_core import merge_flex, ApplyMerge3, BackupBeforeChange
 
26
from bzrlib.changeset import generate_changeset, ExceptionConflictHandler
 
27
from bzrlib.changeset import Inventory, Diff3Merge
 
28
from bzrlib.branch import find_branch
 
29
from bzrlib.errors import BzrCommandError, UnrelatedBranches
 
30
from bzrlib.delta import compare_trees
 
31
from bzrlib.trace import mutter, warning
 
32
from bzrlib.fetch import greedy_fetch
 
33
from bzrlib.revision import is_ancestor
 
34
 
 
35
# comments from abentley on irc: merge happens in two stages, each
 
36
# of which generates a changeset object
 
37
 
 
38
# stage 1: generate OLD->OTHER,
 
39
# stage 2: use MINE and OLD->OTHER to generate MINE -> RESULT
20
40
 
21
41
class MergeConflictHandler(ExceptionConflictHandler):
22
 
    """Handle conflicts encountered while merging"""
 
42
    """Handle conflicts encountered while merging.
 
43
 
 
44
    This subclasses ExceptionConflictHandler, so that any types of
 
45
    conflict that are not explicitly handled cause an exception and
 
46
    terminate the merge.
 
47
    """
23
48
    def __init__(self, dir, ignore_zero=False):
24
49
        ExceptionConflictHandler.__init__(self, dir)
25
50
        self.conflicts = 0
36
61
            d_file.write(line)
37
62
        os.chmod(dest, 0777 & os.stat(source).st_mode)
38
63
 
 
64
    def dump(self, lines, dest):
 
65
        """Copy the text and mode of a file
 
66
        :param source: The path of the file to copy
 
67
        :param dest: The distination file to create
 
68
        """
 
69
        d_file = file(dest, "wb")
 
70
        for line in lines:
 
71
            d_file.write(line)
 
72
 
39
73
    def add_suffix(self, name, suffix, last_new_name=None):
40
74
        """Rename a file to append a suffix.  If the new name exists, the
41
75
        suffix is added repeatedly until a non-existant name is found
60
94
        self.conflicts += 1
61
95
        
62
96
 
63
 
    def merge_conflict(self, new_file, this_path, base_path, other_path):
 
97
    def merge_conflict(self, new_file, this_path, base_lines, other_lines):
64
98
        """
65
99
        Handle diff3 conflicts by producing a .THIS, .BASE and .OTHER.  The
66
100
        main file will be a version with diff3 conflicts.
70
104
        :param other_path: Path to the file text for the OTHER tree
71
105
        """
72
106
        self.add_suffix(this_path, ".THIS")
73
 
        self.copy(base_path, this_path+".BASE")
74
 
        self.copy(other_path, this_path+".OTHER")
 
107
        self.dump(base_lines, this_path+".BASE")
 
108
        self.dump(other_lines, this_path+".OTHER")
75
109
        os.rename(new_file, this_path)
76
110
        self.conflict("Diff3 conflict encountered in %s" % this_path)
77
111
 
 
112
    def new_contents_conflict(self, filename, other_contents):
 
113
        """Conflicting contents for newly added file."""
 
114
        self.copy(other_contents, filename + ".OTHER")
 
115
        self.conflict("Conflict in newly added file %s" % filename)
 
116
    
 
117
 
78
118
    def target_exists(self, entry, target, old_path):
79
119
        """Handle the case when the target file or dir exists"""
80
120
        moved_path = self.add_suffix(target, ".moved")
90
130
        if not self.ignore_zero:
91
131
            print "%d conflicts encountered.\n" % self.conflicts
92
132
            
93
 
class SourceFile(object):
94
 
    def __init__(self, path, id, present=None, isdir=None):
95
 
        self.path = path
96
 
        self.id = id
97
 
        self.present = present
98
 
        self.isdir = isdir
99
 
        self.interesting = True
100
 
 
101
 
    def __repr__(self):
102
 
        return "SourceFile(%s, %s)" % (self.path, self.id)
103
 
 
104
 
def get_tree(treespec, temp_root, label):
 
133
def get_tree(treespec, temp_root, label, local_branch=None):
105
134
    location, revno = treespec
106
135
    branch = find_branch(location)
107
136
    if revno is None:
 
137
        revision = None
 
138
    elif revno == -1:
 
139
        revision = branch.last_patch()
 
140
    else:
 
141
        revision = branch.lookup_revision(revno)
 
142
    return branch, get_revid_tree(branch, revision, temp_root, label,
 
143
                                  local_branch)
 
144
 
 
145
def get_revid_tree(branch, revision, temp_root, label, local_branch):
 
146
    if revision is None:
108
147
        base_tree = branch.working_tree()
109
 
    elif revno == -1:
110
 
        base_tree = branch.basis_tree()
111
148
    else:
112
 
        base_tree = branch.revision_tree(branch.lookup_revision(revno))
 
149
        if local_branch is not None:
 
150
            greedy_fetch(local_branch, branch, revision)
 
151
            base_tree = local_branch.revision_tree(revision)
 
152
        else:
 
153
            base_tree = branch.revision_tree(revision)
113
154
    temp_path = os.path.join(temp_root, label)
114
155
    os.mkdir(temp_path)
115
 
    return branch, MergeTree(base_tree, temp_path)
116
 
 
117
 
 
118
 
def abspath(tree, file_id):
119
 
    path = tree.inventory.id2path(file_id)
120
 
    if path == "":
121
 
        return "./."
122
 
    return "./" + path
 
156
    return MergeTree(base_tree, temp_path)
 
157
 
123
158
 
124
159
def file_exists(tree, file_id):
125
160
    return tree.has_filename(tree.id2path(file_id))
126
161
    
127
 
def inventory_map(tree):
128
 
    inventory = {}
129
 
    for file_id in tree.inventory:
130
 
        path = abspath(tree, file_id)
131
 
        inventory[path] = SourceFile(path, file_id)
132
 
    return inventory
133
 
 
134
162
 
135
163
class MergeTree(object):
136
164
    def __init__(self, tree, tempdir):
139
167
            self.root = tree.basedir
140
168
        else:
141
169
            self.root = None
142
 
        self.inventory = inventory_map(tree)
143
170
        self.tree = tree
144
171
        self.tempdir = tempdir
145
172
        os.mkdir(os.path.join(self.tempdir, "texts"))
146
173
        self.cached = {}
147
174
 
 
175
    def __iter__(self):
 
176
        return self.tree.__iter__()
 
177
 
 
178
    def __contains__(self, file_id):
 
179
        return file_id in self.tree
 
180
 
 
181
    def get_file(self, file_id):
 
182
        return self.tree.get_file(file_id)
 
183
 
 
184
    def get_file_sha1(self, id):
 
185
        return self.tree.get_file_sha1(id)
 
186
 
 
187
    def id2path(self, file_id):
 
188
        return self.tree.id2path(file_id)
 
189
 
 
190
    def has_id(self, file_id):
 
191
        return self.tree.has_id(file_id)
 
192
 
 
193
    def has_or_had_id(self, file_id):
 
194
        if file_id == self.tree.inventory.root.file_id:
 
195
            return True
 
196
        return self.tree.inventory.has_id(file_id)
 
197
 
148
198
    def readonly_path(self, id):
149
199
        if id not in self.tree:
150
200
            return None
178
228
    check_clean
179
229
        If true, this_dir must have no uncommitted changes before the
180
230
        merge begins.
 
231
    all available ancestors of other_revision and base_revision are
 
232
    automatically pulled into the branch.
181
233
    """
 
234
    from bzrlib.revision import common_ancestor, MultipleRevisionSources
 
235
    from bzrlib.errors import NoSuchRevision
182
236
    tempdir = tempfile.mkdtemp(prefix="bzr-")
183
237
    try:
184
238
        if this_dir is None:
185
239
            this_dir = '.'
186
240
        this_branch = find_branch(this_dir)
 
241
        this_rev_id = this_branch.last_patch()
 
242
        if this_rev_id is None:
 
243
            raise BzrCommandError("This branch has no commits")
187
244
        if check_clean:
188
245
            changes = compare_trees(this_branch.working_tree(), 
189
246
                                    this_branch.basis_tree(), False)
190
247
            if changes.has_changed():
191
248
                raise BzrCommandError("Working tree has uncommitted changes.")
192
 
        other_branch, other_tree = get_tree(other_revision, tempdir, "other")
 
249
        other_branch, other_tree = get_tree(other_revision, tempdir, "other",
 
250
                                            this_branch)
 
251
        if other_revision[1] == -1:
 
252
            other_rev_id = other_branch.last_patch()
 
253
            other_basis = other_rev_id
 
254
        elif other_revision[1] is not None:
 
255
            other_rev_id = other_branch.lookup_revision(other_revision[1])
 
256
            other_basis = other_rev_id
 
257
        else:
 
258
            other_rev_id = None
 
259
            other_basis = other_branch.last_patch()
193
260
        if base_revision == [None, None]:
194
261
            if other_revision[1] == -1:
195
262
                o_revno = None
196
263
            else:
197
264
                o_revno = other_revision[1]
198
 
            base_revno = this_branch.common_ancestor(other_branch, 
199
 
                                                     other_revno=o_revno)[0]
200
 
            if base_revno is None:
201
265
                raise UnrelatedBranches()
202
 
            base_revision = ['.', base_revno]
203
 
        base_branch, base_tree = get_tree(base_revision, tempdir, "base")
 
266
            try:
 
267
                base_revision = this_branch.get_revision(base_rev_id)
 
268
                base_branch = this_branch
 
269
            except NoSuchRevision:
 
270
                base_branch = other_branch
 
271
            base_tree = get_revid_tree(base_branch, base_rev_id, tempdir, 
 
272
                                       "base")
 
273
            base_is_ancestor = True
 
274
        else:
 
275
            base_branch, base_tree = get_tree(base_revision, tempdir, "base")
 
276
            if base_revision[1] == -1:
 
277
                base_rev_id = base_branch.last_patch()
 
278
            elif base_revision[1] is None:
 
279
                base_rev_id = None
 
280
            else:
 
281
                base_rev_id = base_branch.lookup_revision(base_revision[1])
 
282
            if base_rev_id is not None:
 
283
                base_is_ancestor = is_ancestor(this_rev_id, base_rev_id, 
 
284
                                               MultipleRevisionSources(this_branch, 
 
285
                                                                       base_branch))
 
286
            else:
 
287
                base_is_ancestor = False
204
288
        if file_list is None:
205
289
            interesting_ids = None
206
290
        else:
220
304
        merge_inner(this_branch, other_tree, base_tree, tempdir, 
221
305
                    ignore_zero=ignore_zero, backup_files=backup_files, 
222
306
                    merge_type=merge_type, interesting_ids=interesting_ids)
 
307
        if base_is_ancestor and other_rev_id is not None:
 
308
            this_branch.add_pending_merge(other_rev_id)
223
309
    finally:
224
310
        shutil.rmtree(tempdir)
225
311
 
232
318
             source_file.interesting = source_file.id in interesting_ids
233
319
 
234
320
 
235
 
def set_optimized(tree_a, tree_b, inventory_a, inventory_b):
236
 
    """Mark files that have changed texts as interesting
237
 
    """
238
 
    for file_id in tree_a.tree.inventory:
239
 
        if file_id not in tree_b.tree.inventory:
240
 
            continue
241
 
        entry_a = tree_a.tree.inventory[file_id]
242
 
        entry_b = tree_b.tree.inventory[file_id]
243
 
        if (entry_a.kind, entry_b.kind) != ("file", "file"):
244
 
            continue
245
 
        if None in (entry_a.text_id, entry_b.text_id):
246
 
            continue
247
 
        if entry_a.text_id != entry_b.text_id:
248
 
            continue
249
 
        inventory_a[abspath(tree_a.tree, file_id)].interesting = False
250
 
        inventory_b[abspath(tree_b.tree, file_id)].interesting = False
251
 
 
252
 
 
253
 
def generate_cset_optimized(tree_a, tree_b, inventory_a, inventory_b,
254
 
                            interesting_ids=None):
255
 
    """Generate a changeset, with preprocessing to select interesting files.
256
 
    using the text_id to mark really-changed files.
257
 
    This permits blazing comparisons when text_ids are present.  It also
258
 
    disables metadata comparison for files with identical texts.
 
321
def generate_cset_optimized(tree_a, tree_b, interesting_ids=None):
 
322
    """Generate a changeset.  If interesting_ids is supplied, only changes
 
323
    to those files will be shown.  Metadata changes are stripped.
259
324
    """ 
260
 
    if interesting_ids is None:
261
 
        set_optimized(tree_a, tree_b, inventory_a, inventory_b)
262
 
    else:
263
 
        set_interesting(inventory_a, inventory_b, interesting_ids)
264
 
    cset =  generate_changeset(tree_a, tree_b, inventory_a, inventory_b)
 
325
    cset =  generate_changeset(tree_a, tree_b, interesting_ids)
265
326
    for entry in cset.entries.itervalues():
266
327
        entry.metadata_change = None
267
328
    return cset
271
332
                ignore_zero=False, merge_type=ApplyMerge3, backup_files=False,
272
333
                interesting_ids=None):
273
334
 
274
 
    def merge_factory(base_file, other_file):
275
 
        contents_change = merge_type(base_file, other_file)
 
335
    def merge_factory(file_id, base, other):
 
336
        contents_change = merge_type(file_id, base, other)
276
337
        if backup_files:
277
338
            contents_change = BackupBeforeChange(contents_change)
278
339
        return contents_change
279
 
    
280
 
    def generate_cset(tree_a, tree_b, inventory_a, inventory_b):
281
 
        return generate_cset_optimized(tree_a, tree_b, inventory_a, inventory_b,
282
 
                                       interesting_ids)
283
340
 
284
341
    this_tree = get_tree((this_branch.base, None), tempdir, "this")[1]
285
342
 
286
343
    def get_inventory(tree):
287
 
        return tree.inventory
 
344
        return tree.tree.inventory
288
345
 
289
346
    inv_changes = merge_flex(this_tree, base_tree, other_tree,
290
 
                             generate_cset, get_inventory,
 
347
                             generate_cset_optimized, get_inventory,
291
348
                             MergeConflictHandler(base_tree.root,
292
349
                                                  ignore_zero=ignore_zero),
293
 
                             merge_factory=merge_factory)
 
350
                             merge_factory=merge_factory, 
 
351
                             interesting_ids=interesting_ids)
294
352
 
295
353
    adjust_ids = []
296
354
    for id, path in inv_changes.iteritems():
298
356
            if path == '.':
299
357
                path = ''
300
358
            else:
301
 
                assert path.startswith('./')
 
359
                assert path.startswith('./'), "path is %s" % path
302
360
            path = path[2:]
303
361
        adjust_ids.append((path, id))
304
 
    this_branch.set_inventory(regen_inventory(this_branch, this_tree.root, adjust_ids))
 
362
    if len(adjust_ids) > 0:
 
363
        this_branch.set_inventory(regen_inventory(this_branch, this_tree.root,
 
364
                                                  adjust_ids))
305
365
 
306
366
 
307
367
def regen_inventory(this_branch, root, new_entries):
308
368
    old_entries = this_branch.read_working_inventory()
309
369
    new_inventory = {}
310
370
    by_path = {}
 
371
    new_entries_map = {} 
 
372
    for path, file_id in new_entries:
 
373
        if path is None:
 
374
            continue
 
375
        new_entries_map[file_id] = path
 
376
 
 
377
    def id2path(file_id):
 
378
        path = new_entries_map.get(file_id)
 
379
        if path is not None:
 
380
            return path
 
381
        entry = old_entries[file_id]
 
382
        if entry.parent_id is None:
 
383
            return entry.name
 
384
        return os.path.join(id2path(entry.parent_id), entry.name)
 
385
        
311
386
    for file_id in old_entries:
312
387
        entry = old_entries[file_id]
313
 
        path = old_entries.id2path(file_id)
 
388
        path = id2path(file_id)
314
389
        new_inventory[file_id] = (path, file_id, entry.parent_id, entry.kind)
315
390
        by_path[path] = file_id
316
391