/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-09-01 06:32:52 UTC
  • Revision ID: mbp@sourcefrog.net-20050901063252-00fc761bf1076759
- make target to build emacs TAGS file

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