~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/conflicts.py

  • Committer: Martin Pool
  • Date: 2005-05-11 06:05:28 UTC
  • Revision ID: mbp@sourcefrog.net-20050511060528-b2b35cafa4e1785d
- Interpret arguments to bzr status
- Handle giving a non-default branch to bzr status
- bzr status docs

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007, 2009, 2010 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
# TODO: 'bzr resolve' should accept a directory name and work from that
18
 
# point down
19
 
 
20
 
import os
21
 
import re
22
 
 
23
 
from bzrlib.lazy_import import lazy_import
24
 
lazy_import(globals(), """
25
 
import errno
26
 
 
27
 
from bzrlib import (
28
 
    builtins,
29
 
    cleanup,
30
 
    commands,
31
 
    errors,
32
 
    osutils,
33
 
    rio,
34
 
    trace,
35
 
    transform,
36
 
    workingtree,
37
 
    )
38
 
""")
39
 
from bzrlib import (
40
 
    option,
41
 
    registry,
42
 
    )
43
 
 
44
 
 
45
 
CONFLICT_SUFFIXES = ('.THIS', '.BASE', '.OTHER')
46
 
 
47
 
 
48
 
class cmd_conflicts(commands.Command):
49
 
    """List files with conflicts.
50
 
 
51
 
    Merge will do its best to combine the changes in two branches, but there
52
 
    are some kinds of problems only a human can fix.  When it encounters those,
53
 
    it will mark a conflict.  A conflict means that you need to fix something,
54
 
    before you should commit.
55
 
 
56
 
    Conflicts normally are listed as short, human-readable messages.  If --text
57
 
    is supplied, the pathnames of files with text conflicts are listed,
58
 
    instead.  (This is useful for editing all files with text conflicts.)
59
 
 
60
 
    Use bzr resolve when you have fixed a problem.
61
 
    """
62
 
    takes_options = [
63
 
            option.Option('text',
64
 
                          help='List paths of files with text conflicts.'),
65
 
        ]
66
 
    _see_also = ['resolve', 'conflict-types']
67
 
 
68
 
    def run(self, text=False):
69
 
        wt = workingtree.WorkingTree.open_containing(u'.')[0]
70
 
        for conflict in wt.conflicts():
71
 
            if text:
72
 
                if conflict.typestring != 'text conflict':
73
 
                    continue
74
 
                self.outf.write(conflict.path + '\n')
75
 
            else:
76
 
                self.outf.write(str(conflict) + '\n')
77
 
 
78
 
 
79
 
resolve_action_registry = registry.Registry()
80
 
 
81
 
 
82
 
resolve_action_registry.register(
83
 
    'done', 'done', 'Marks the conflict as resolved' )
84
 
resolve_action_registry.register(
85
 
    'take-this', 'take_this',
86
 
    'Resolve the conflict preserving the version in the working tree' )
87
 
resolve_action_registry.register(
88
 
    'take-other', 'take_other',
89
 
    'Resolve the conflict taking the merged version into account' )
90
 
resolve_action_registry.default_key = 'done'
91
 
 
92
 
class ResolveActionOption(option.RegistryOption):
93
 
 
94
 
    def __init__(self):
95
 
        super(ResolveActionOption, self).__init__(
96
 
            'action', 'How to resolve the conflict.',
97
 
            value_switches=True,
98
 
            registry=resolve_action_registry)
99
 
 
100
 
 
101
 
class cmd_resolve(commands.Command):
102
 
    """Mark a conflict as resolved.
103
 
 
104
 
    Merge will do its best to combine the changes in two branches, but there
105
 
    are some kinds of problems only a human can fix.  When it encounters those,
106
 
    it will mark a conflict.  A conflict means that you need to fix something,
107
 
    before you should commit.
108
 
 
109
 
    Once you have fixed a problem, use "bzr resolve" to automatically mark
110
 
    text conflicts as fixed, "bzr resolve FILE" to mark a specific conflict as
111
 
    resolved, or "bzr resolve --all" to mark all conflicts as resolved.
112
 
    """
113
 
    aliases = ['resolved']
114
 
    takes_args = ['file*']
115
 
    takes_options = [
116
 
            option.Option('all', help='Resolve all conflicts in this tree.'),
117
 
            ResolveActionOption(),
118
 
            ]
119
 
    _see_also = ['conflicts']
120
 
    def run(self, file_list=None, all=False, action=None):
121
 
        if all:
122
 
            if file_list:
123
 
                raise errors.BzrCommandError("If --all is specified,"
124
 
                                             " no FILE may be provided")
125
 
            tree = workingtree.WorkingTree.open_containing('.')[0]
126
 
            if action is None:
127
 
                action = 'done'
128
 
        else:
129
 
            tree, file_list = builtins.tree_files(file_list)
130
 
            if file_list is None:
131
 
                if action is None:
132
 
                    # FIXME: There is a special case here related to the option
133
 
                    # handling that could be clearer and easier to discover by
134
 
                    # providing an --auto action (bug #344013 and #383396) and
135
 
                    # make it mandatory instead of implicit and active only
136
 
                    # when no file_list is provided -- vila 091229
137
 
                    action = 'auto'
138
 
            else:
139
 
                if action is None:
140
 
                    action = 'done'
141
 
        if action == 'auto':
142
 
            if file_list is None:
143
 
                un_resolved, resolved = tree.auto_resolve()
144
 
                if len(un_resolved) > 0:
145
 
                    trace.note('%d conflict(s) auto-resolved.', len(resolved))
146
 
                    trace.note('Remaining conflicts:')
147
 
                    for conflict in un_resolved:
148
 
                        trace.note(conflict)
149
 
                    return 1
150
 
                else:
151
 
                    trace.note('All conflicts resolved.')
152
 
                    return 0
153
 
            else:
154
 
                # FIXME: This can never occur but the block above needs some
155
 
                # refactoring to transfer tree.auto_resolve() to
156
 
                # conflict.auto(tree) --vila 091242
157
 
                pass
158
 
        else:
159
 
            resolve(tree, file_list, action=action)
160
 
 
161
 
 
162
 
def resolve(tree, paths=None, ignore_misses=False, recursive=False,
163
 
            action='done'):
164
 
    """Resolve some or all of the conflicts in a working tree.
165
 
 
166
 
    :param paths: If None, resolve all conflicts.  Otherwise, select only
167
 
        specified conflicts.
168
 
    :param recursive: If True, then elements of paths which are directories
169
 
        have all their children resolved, etc.  When invoked as part of
170
 
        recursive commands like revert, this should be True.  For commands
171
 
        or applications wishing finer-grained control, like the resolve
172
 
        command, this should be False.
173
 
    :param ignore_misses: If False, warnings will be printed if the supplied
174
 
        paths do not have conflicts.
175
 
    :param action: How the conflict should be resolved,
176
 
    """
177
 
    tree.lock_tree_write()
178
 
    try:
179
 
        tree_conflicts = tree.conflicts()
180
 
        if paths is None:
181
 
            new_conflicts = ConflictList()
182
 
            to_process = tree_conflicts
183
 
        else:
184
 
            new_conflicts, to_process = tree_conflicts.select_conflicts(
185
 
                tree, paths, ignore_misses, recursive)
186
 
        for conflict in to_process:
187
 
            try:
188
 
                conflict._do(action, tree)
189
 
                conflict.cleanup(tree)
190
 
            except NotImplementedError:
191
 
                new_conflicts.append(conflict)
192
 
        try:
193
 
            tree.set_conflicts(new_conflicts)
194
 
        except errors.UnsupportedOperation:
195
 
            pass
196
 
    finally:
197
 
        tree.unlock()
198
 
 
199
 
 
200
 
def restore(filename):
201
 
    """Restore a conflicted file to the state it was in before merging.
202
 
 
203
 
    Only text restoration is supported at present.
204
 
    """
205
 
    conflicted = False
206
 
    try:
207
 
        osutils.rename(filename + ".THIS", filename)
208
 
        conflicted = True
209
 
    except OSError, e:
210
 
        if e.errno != errno.ENOENT:
211
 
            raise
212
 
    try:
213
 
        os.unlink(filename + ".BASE")
214
 
        conflicted = True
215
 
    except OSError, e:
216
 
        if e.errno != errno.ENOENT:
217
 
            raise
218
 
    try:
219
 
        os.unlink(filename + ".OTHER")
220
 
        conflicted = True
221
 
    except OSError, e:
222
 
        if e.errno != errno.ENOENT:
223
 
            raise
224
 
    if not conflicted:
225
 
        raise errors.NotConflicted(filename)
226
 
 
227
 
 
228
 
class ConflictList(object):
229
 
    """List of conflicts.
230
 
 
231
 
    Typically obtained from WorkingTree.conflicts()
232
 
 
233
 
    Can be instantiated from stanzas or from Conflict subclasses.
234
 
    """
235
 
 
236
 
    def __init__(self, conflicts=None):
237
 
        object.__init__(self)
238
 
        if conflicts is None:
239
 
            self.__list = []
240
 
        else:
241
 
            self.__list = conflicts
242
 
 
243
 
    def is_empty(self):
244
 
        return len(self.__list) == 0
245
 
 
246
 
    def __len__(self):
247
 
        return len(self.__list)
248
 
 
249
 
    def __iter__(self):
250
 
        return iter(self.__list)
251
 
 
252
 
    def __getitem__(self, key):
253
 
        return self.__list[key]
254
 
 
255
 
    def append(self, conflict):
256
 
        return self.__list.append(conflict)
257
 
 
258
 
    def __eq__(self, other_list):
259
 
        return list(self) == list(other_list)
260
 
 
261
 
    def __ne__(self, other_list):
262
 
        return not (self == other_list)
263
 
 
264
 
    def __repr__(self):
265
 
        return "ConflictList(%r)" % self.__list
266
 
 
267
 
    @staticmethod
268
 
    def from_stanzas(stanzas):
269
 
        """Produce a new ConflictList from an iterable of stanzas"""
270
 
        conflicts = ConflictList()
271
 
        for stanza in stanzas:
272
 
            conflicts.append(Conflict.factory(**stanza.as_dict()))
273
 
        return conflicts
274
 
 
275
 
    def to_stanzas(self):
276
 
        """Generator of stanzas"""
277
 
        for conflict in self:
278
 
            yield conflict.as_stanza()
279
 
 
280
 
    def to_strings(self):
281
 
        """Generate strings for the provided conflicts"""
282
 
        for conflict in self:
283
 
            yield str(conflict)
284
 
 
285
 
    def remove_files(self, tree):
286
 
        """Remove the THIS, BASE and OTHER files for listed conflicts"""
287
 
        for conflict in self:
288
 
            if not conflict.has_files:
289
 
                continue
290
 
            conflict.cleanup(tree)
291
 
 
292
 
    def select_conflicts(self, tree, paths, ignore_misses=False,
293
 
                         recurse=False):
294
 
        """Select the conflicts associated with paths in a tree.
295
 
 
296
 
        File-ids are also used for this.
297
 
        :return: a pair of ConflictLists: (not_selected, selected)
298
 
        """
299
 
        path_set = set(paths)
300
 
        ids = {}
301
 
        selected_paths = set()
302
 
        new_conflicts = ConflictList()
303
 
        selected_conflicts = ConflictList()
304
 
        for path in paths:
305
 
            file_id = tree.path2id(path)
306
 
            if file_id is not None:
307
 
                ids[file_id] = path
308
 
 
309
 
        for conflict in self:
310
 
            selected = False
311
 
            for key in ('path', 'conflict_path'):
312
 
                cpath = getattr(conflict, key, None)
313
 
                if cpath is None:
314
 
                    continue
315
 
                if cpath in path_set:
316
 
                    selected = True
317
 
                    selected_paths.add(cpath)
318
 
                if recurse:
319
 
                    if osutils.is_inside_any(path_set, cpath):
320
 
                        selected = True
321
 
                        selected_paths.add(cpath)
322
 
 
323
 
            for key in ('file_id', 'conflict_file_id'):
324
 
                cfile_id = getattr(conflict, key, None)
325
 
                if cfile_id is None:
326
 
                    continue
327
 
                try:
328
 
                    cpath = ids[cfile_id]
329
 
                except KeyError:
330
 
                    continue
331
 
                selected = True
332
 
                selected_paths.add(cpath)
333
 
            if selected:
334
 
                selected_conflicts.append(conflict)
335
 
            else:
336
 
                new_conflicts.append(conflict)
337
 
        if ignore_misses is not True:
338
 
            for path in [p for p in paths if p not in selected_paths]:
339
 
                if not os.path.exists(tree.abspath(path)):
340
 
                    print "%s does not exist" % path
341
 
                else:
342
 
                    print "%s is not conflicted" % path
343
 
        return new_conflicts, selected_conflicts
344
 
 
345
 
 
346
 
class Conflict(object):
347
 
    """Base class for all types of conflict"""
348
 
 
349
 
    # FIXME: cleanup should take care of that ? -- vila 091229
350
 
    has_files = False
351
 
 
352
 
    def __init__(self, path, file_id=None):
353
 
        self.path = path
354
 
        # warn turned off, because the factory blindly transfers the Stanza
355
 
        # values to __init__ and Stanza is purely a Unicode api.
356
 
        self.file_id = osutils.safe_file_id(file_id, warn=False)
357
 
 
358
 
    def as_stanza(self):
359
 
        s = rio.Stanza(type=self.typestring, path=self.path)
360
 
        if self.file_id is not None:
361
 
            # Stanza requires Unicode apis
362
 
            s.add('file_id', self.file_id.decode('utf8'))
363
 
        return s
364
 
 
365
 
    def _cmp_list(self):
366
 
        return [type(self), self.path, self.file_id]
367
 
 
368
 
    def __cmp__(self, other):
369
 
        if getattr(other, "_cmp_list", None) is None:
370
 
            return -1
371
 
        return cmp(self._cmp_list(), other._cmp_list())
372
 
 
373
 
    def __hash__(self):
374
 
        return hash((type(self), self.path, self.file_id))
375
 
 
376
 
    def __eq__(self, other):
377
 
        return self.__cmp__(other) == 0
378
 
 
379
 
    def __ne__(self, other):
380
 
        return not self.__eq__(other)
381
 
 
382
 
    def __str__(self):
383
 
        return self.format % self.__dict__
384
 
 
385
 
    def __repr__(self):
386
 
        rdict = dict(self.__dict__)
387
 
        rdict['class'] = self.__class__.__name__
388
 
        return self.rformat % rdict
389
 
 
390
 
    @staticmethod
391
 
    def factory(type, **kwargs):
392
 
        global ctype
393
 
        return ctype[type](**kwargs)
394
 
 
395
 
    @staticmethod
396
 
    def sort_key(conflict):
397
 
        if conflict.path is not None:
398
 
            return conflict.path, conflict.typestring
399
 
        elif getattr(conflict, "conflict_path", None) is not None:
400
 
            return conflict.conflict_path, conflict.typestring
401
 
        else:
402
 
            return None, conflict.typestring
403
 
 
404
 
    def _do(self, action, tree):
405
 
        """Apply the specified action to the conflict.
406
 
 
407
 
        :param action: The method name to call.
408
 
 
409
 
        :param tree: The tree passed as a parameter to the method.
410
 
        """
411
 
        meth = getattr(self, 'action_%s' % action, None)
412
 
        if meth is None:
413
 
            raise NotImplementedError(self.__class__.__name__ + '.' + action)
414
 
        meth(tree)
415
 
 
416
 
    def associated_filenames(self):
417
 
        """The names of the files generated to help resolve the conflict."""
418
 
        raise NotImplementedError(self.associated_filenames)
419
 
 
420
 
    def cleanup(self, tree):
421
 
        for fname in self.associated_filenames():
422
 
            try:
423
 
                osutils.delete_any(tree.abspath(fname))
424
 
            except OSError, e:
425
 
                if e.errno != errno.ENOENT:
426
 
                    raise
427
 
 
428
 
    def action_done(self, tree):
429
 
        """Mark the conflict as solved once it has been handled."""
430
 
        # This method does nothing but simplifies the design of upper levels.
431
 
        pass
432
 
 
433
 
    def action_take_this(self, tree):
434
 
        raise NotImplementedError(self.action_take_this)
435
 
 
436
 
    def action_take_other(self, tree):
437
 
        raise NotImplementedError(self.action_take_other)
438
 
 
439
 
    def _resolve_with_cleanups(self, tree, *args, **kwargs):
440
 
        tt = transform.TreeTransform(tree)
441
 
        op = cleanup.OperationWithCleanups(self._resolve)
442
 
        op.add_cleanup(tt.finalize)
443
 
        op.run_simple(tt, *args, **kwargs)
444
 
 
445
 
 
446
 
class PathConflict(Conflict):
447
 
    """A conflict was encountered merging file paths"""
448
 
 
449
 
    typestring = 'path conflict'
450
 
 
451
 
    format = 'Path conflict: %(path)s / %(conflict_path)s'
452
 
 
453
 
    rformat = '%(class)s(%(path)r, %(conflict_path)r, %(file_id)r)'
454
 
 
455
 
    def __init__(self, path, conflict_path=None, file_id=None):
456
 
        Conflict.__init__(self, path, file_id)
457
 
        self.conflict_path = conflict_path
458
 
 
459
 
    def as_stanza(self):
460
 
        s = Conflict.as_stanza(self)
461
 
        if self.conflict_path is not None:
462
 
            s.add('conflict_path', self.conflict_path)
463
 
        return s
464
 
 
465
 
    def associated_filenames(self):
466
 
        # No additional files have been generated here
467
 
        return []
468
 
 
469
 
    def _resolve(self, tt, file_id, path, winner):
470
 
        """Resolve the conflict.
471
 
 
472
 
        :param tt: The TreeTransform where the conflict is resolved.
473
 
        :param file_id: The retained file id.
474
 
        :param path: The retained path.
475
 
        :param winner: 'this' or 'other' indicates which side is the winner.
476
 
        """
477
 
        path_to_create = None
478
 
        if winner == 'this':
479
 
            if self.path == '<deleted>':
480
 
                return # Nothing to do
481
 
            if self.conflict_path == '<deleted>':
482
 
                path_to_create = self.path
483
 
                revid = tt._tree.get_parent_ids()[0]
484
 
        elif winner == 'other':
485
 
            if self.conflict_path == '<deleted>':
486
 
                return  # Nothing to do
487
 
            if self.path == '<deleted>':
488
 
                path_to_create = self.conflict_path
489
 
                # FIXME: If there are more than two parents we may need to
490
 
                # iterate. Taking the last parent is the safer bet in the mean
491
 
                # time. -- vila 20100309
492
 
                revid = tt._tree.get_parent_ids()[-1]
493
 
        else:
494
 
            # Programmer error
495
 
            raise AssertionError('bad winner: %r' % (winner,))
496
 
        if path_to_create is not None:
497
 
            tid = tt.trans_id_tree_path(path_to_create)
498
 
            transform.create_from_tree(
499
 
                tt, tt.trans_id_tree_path(path_to_create),
500
 
                self._revision_tree(tt._tree, revid), file_id)
501
 
            tt.version_file(file_id, tid)
502
 
 
503
 
        # Adjust the path for the retained file id
504
 
        tid = tt.trans_id_file_id(file_id)
505
 
        parent_tid = tt.get_tree_parent(tid)
506
 
        tt.adjust_path(path, parent_tid, tid)
507
 
        tt.apply()
508
 
 
509
 
    def _revision_tree(self, tree, revid):
510
 
        return tree.branch.repository.revision_tree(revid)
511
 
 
512
 
    def _infer_file_id(self, tree):
513
 
        # Prior to bug #531967, file_id wasn't always set, there may still be
514
 
        # conflict files in the wild so we need to cope with them
515
 
        # Establish which path we should use to find back the file-id
516
 
        possible_paths = []
517
 
        for p in (self.path, self.conflict_path):
518
 
            if p == '<deleted>':
519
 
                # special hard-coded path 
520
 
                continue
521
 
            if p is not None:
522
 
                possible_paths.append(p)
523
 
        # Search the file-id in the parents with any path available
524
 
        file_id = None
525
 
        for revid in tree.get_parent_ids():
526
 
            revtree = self._revision_tree(tree, revid)
527
 
            for p in possible_paths:
528
 
                file_id = revtree.path2id(p)
529
 
                if file_id is not None:
530
 
                    return revtree, file_id
531
 
        return None, None
532
 
 
533
 
    def action_take_this(self, tree):
534
 
        if self.file_id is not None:
535
 
            self._resolve_with_cleanups(tree, self.file_id, self.path,
536
 
                                        winner='this')
537
 
        else:
538
 
            # Prior to bug #531967 we need to find back the file_id and restore
539
 
            # the content from there
540
 
            revtree, file_id = self._infer_file_id(tree)
541
 
            tree.revert([revtree.id2path(file_id)],
542
 
                        old_tree=revtree, backups=False)
543
 
 
544
 
    def action_take_other(self, tree):
545
 
        if self.file_id is not None:
546
 
            self._resolve_with_cleanups(tree, self.file_id,
547
 
                                        self.conflict_path,
548
 
                                        winner='other')
549
 
        else:
550
 
            # Prior to bug #531967 we need to find back the file_id and restore
551
 
            # the content from there
552
 
            revtree, file_id = self._infer_file_id(tree)
553
 
            tree.revert([revtree.id2path(file_id)],
554
 
                        old_tree=revtree, backups=False)
555
 
 
556
 
 
557
 
class ContentsConflict(PathConflict):
558
 
    """The files are of different types (or both binary), or not present"""
559
 
 
560
 
    has_files = True
561
 
 
562
 
    typestring = 'contents conflict'
563
 
 
564
 
    format = 'Contents conflict in %(path)s'
565
 
 
566
 
    def associated_filenames(self):
567
 
        return [self.path + suffix for suffix in ('.BASE', '.OTHER')]
568
 
 
569
 
    def _resolve(self, tt, suffix_to_remove):
570
 
        """Resolve the conflict.
571
 
 
572
 
        :param tt: The TreeTransform where the conflict is resolved.
573
 
        :param suffix_to_remove: Either 'THIS' or 'OTHER'
574
 
 
575
 
        The resolution is symmetric, when taking THIS, OTHER is deleted and
576
 
        item.THIS is renamed into item and vice-versa.
577
 
        """
578
 
        try:
579
 
            # Delete 'item.THIS' or 'item.OTHER' depending on
580
 
            # suffix_to_remove
581
 
            tt.delete_contents(
582
 
                tt.trans_id_tree_path(self.path + '.' + suffix_to_remove))
583
 
        except errors.NoSuchFile:
584
 
            # There are valid cases where 'item.suffix_to_remove' either
585
 
            # never existed or was already deleted (including the case
586
 
            # where the user deleted it)
587
 
            pass
588
 
        # Rename 'item.suffix_to_remove' (note that if
589
 
        # 'item.suffix_to_remove' has been deleted, this is a no-op)
590
 
        this_tid = tt.trans_id_file_id(self.file_id)
591
 
        parent_tid = tt.get_tree_parent(this_tid)
592
 
        tt.adjust_path(self.path, parent_tid, this_tid)
593
 
        tt.apply()
594
 
 
595
 
    def action_take_this(self, tree):
596
 
        self._resolve_with_cleanups(tree, 'OTHER')
597
 
 
598
 
    def action_take_other(self, tree):
599
 
        self._resolve_with_cleanups(tree, 'THIS')
600
 
 
601
 
 
602
 
# FIXME: TextConflict is about a single file-id, there never is a conflict_path
603
 
# attribute so we shouldn't inherit from PathConflict but simply from Conflict
604
 
 
605
 
# TODO: There should be a base revid attribute to better inform the user about
606
 
# how the conflicts were generated.
607
 
class TextConflict(PathConflict):
608
 
    """The merge algorithm could not resolve all differences encountered."""
609
 
 
610
 
    has_files = True
611
 
 
612
 
    typestring = 'text conflict'
613
 
 
614
 
    format = 'Text conflict in %(path)s'
615
 
 
616
 
    def associated_filenames(self):
617
 
        return [self.path + suffix for suffix in CONFLICT_SUFFIXES]
618
 
 
619
 
 
620
 
class HandledConflict(Conflict):
621
 
    """A path problem that has been provisionally resolved.
622
 
    This is intended to be a base class.
623
 
    """
624
 
 
625
 
    rformat = "%(class)s(%(action)r, %(path)r, %(file_id)r)"
626
 
 
627
 
    def __init__(self, action, path, file_id=None):
628
 
        Conflict.__init__(self, path, file_id)
629
 
        self.action = action
630
 
 
631
 
    def _cmp_list(self):
632
 
        return Conflict._cmp_list(self) + [self.action]
633
 
 
634
 
    def as_stanza(self):
635
 
        s = Conflict.as_stanza(self)
636
 
        s.add('action', self.action)
637
 
        return s
638
 
 
639
 
    def associated_filenames(self):
640
 
        # Nothing has been generated here
641
 
        return []
642
 
 
643
 
 
644
 
class HandledPathConflict(HandledConflict):
645
 
    """A provisionally-resolved path problem involving two paths.
646
 
    This is intended to be a base class.
647
 
    """
648
 
 
649
 
    rformat = "%(class)s(%(action)r, %(path)r, %(conflict_path)r,"\
650
 
        " %(file_id)r, %(conflict_file_id)r)"
651
 
 
652
 
    def __init__(self, action, path, conflict_path, file_id=None,
653
 
                 conflict_file_id=None):
654
 
        HandledConflict.__init__(self, action, path, file_id)
655
 
        self.conflict_path = conflict_path
656
 
        # warn turned off, because the factory blindly transfers the Stanza
657
 
        # values to __init__.
658
 
        self.conflict_file_id = osutils.safe_file_id(conflict_file_id,
659
 
                                                     warn=False)
660
 
 
661
 
    def _cmp_list(self):
662
 
        return HandledConflict._cmp_list(self) + [self.conflict_path,
663
 
                                                  self.conflict_file_id]
664
 
 
665
 
    def as_stanza(self):
666
 
        s = HandledConflict.as_stanza(self)
667
 
        s.add('conflict_path', self.conflict_path)
668
 
        if self.conflict_file_id is not None:
669
 
            s.add('conflict_file_id', self.conflict_file_id.decode('utf8'))
670
 
 
671
 
        return s
672
 
 
673
 
 
674
 
class DuplicateID(HandledPathConflict):
675
 
    """Two files want the same file_id."""
676
 
 
677
 
    typestring = 'duplicate id'
678
 
 
679
 
    format = 'Conflict adding id to %(conflict_path)s.  %(action)s %(path)s.'
680
 
 
681
 
 
682
 
class DuplicateEntry(HandledPathConflict):
683
 
    """Two directory entries want to have the same name."""
684
 
 
685
 
    typestring = 'duplicate'
686
 
 
687
 
    format = 'Conflict adding file %(conflict_path)s.  %(action)s %(path)s.'
688
 
 
689
 
    def action_take_this(self, tree):
690
 
        tree.remove([self.conflict_path], force=True, keep_files=False)
691
 
        tree.rename_one(self.path, self.conflict_path)
692
 
 
693
 
    def action_take_other(self, tree):
694
 
        tree.remove([self.path], force=True, keep_files=False)
695
 
 
696
 
 
697
 
class ParentLoop(HandledPathConflict):
698
 
    """An attempt to create an infinitely-looping directory structure.
699
 
    This is rare, but can be produced like so:
700
 
 
701
 
    tree A:
702
 
      mv foo bar
703
 
    tree B:
704
 
      mv bar foo
705
 
    merge A and B
706
 
    """
707
 
 
708
 
    typestring = 'parent loop'
709
 
 
710
 
    format = 'Conflict moving %(path)s into %(conflict_path)s. %(action)s.'
711
 
 
712
 
    def action_take_this(self, tree):
713
 
        # just acccept bzr proposal
714
 
        pass
715
 
 
716
 
    def action_take_other(self, tree):
717
 
        # FIXME: We shouldn't have to manipulate so many paths here (and there
718
 
        # is probably a bug or two...)
719
 
        base_path = osutils.basename(self.path)
720
 
        conflict_base_path = osutils.basename(self.conflict_path)
721
 
        tt = transform.TreeTransform(tree)
722
 
        try:
723
 
            p_tid = tt.trans_id_file_id(self.file_id)
724
 
            parent_tid = tt.get_tree_parent(p_tid)
725
 
            cp_tid = tt.trans_id_file_id(self.conflict_file_id)
726
 
            cparent_tid = tt.get_tree_parent(cp_tid)
727
 
            tt.adjust_path(base_path, cparent_tid, cp_tid)
728
 
            tt.adjust_path(conflict_base_path, parent_tid, p_tid)
729
 
            tt.apply()
730
 
        finally:
731
 
            tt.finalize()
732
 
 
733
 
 
734
 
class UnversionedParent(HandledConflict):
735
 
    """An attempt to version a file whose parent directory is not versioned.
736
 
    Typically, the result of a merge where one tree unversioned the directory
737
 
    and the other added a versioned file to it.
738
 
    """
739
 
 
740
 
    typestring = 'unversioned parent'
741
 
 
742
 
    format = 'Conflict because %(path)s is not versioned, but has versioned'\
743
 
             ' children.  %(action)s.'
744
 
 
745
 
    # FIXME: We silently do nothing to make tests pass, but most probably the
746
 
    # conflict shouldn't exist (the long story is that the conflict is
747
 
    # generated with another one that can be resolved properly) -- vila 091224
748
 
    def action_take_this(self, tree):
749
 
        pass
750
 
 
751
 
    def action_take_other(self, tree):
752
 
        pass
753
 
 
754
 
 
755
 
class MissingParent(HandledConflict):
756
 
    """An attempt to add files to a directory that is not present.
757
 
    Typically, the result of a merge where THIS deleted the directory and
758
 
    the OTHER added a file to it.
759
 
    See also: DeletingParent (same situation, THIS and OTHER reversed)
760
 
    """
761
 
 
762
 
    typestring = 'missing parent'
763
 
 
764
 
    format = 'Conflict adding files to %(path)s.  %(action)s.'
765
 
 
766
 
    def action_take_this(self, tree):
767
 
        tree.remove([self.path], force=True, keep_files=False)
768
 
 
769
 
    def action_take_other(self, tree):
770
 
        # just acccept bzr proposal
771
 
        pass
772
 
 
773
 
 
774
 
class DeletingParent(HandledConflict):
775
 
    """An attempt to add files to a directory that is not present.
776
 
    Typically, the result of a merge where one OTHER deleted the directory and
777
 
    the THIS added a file to it.
778
 
    """
779
 
 
780
 
    typestring = 'deleting parent'
781
 
 
782
 
    format = "Conflict: can't delete %(path)s because it is not empty.  "\
783
 
             "%(action)s."
784
 
 
785
 
    # FIXME: It's a bit strange that the default action is not coherent with
786
 
    # MissingParent from the *user* pov.
787
 
 
788
 
    def action_take_this(self, tree):
789
 
        # just acccept bzr proposal
790
 
        pass
791
 
 
792
 
    def action_take_other(self, tree):
793
 
        tree.remove([self.path], force=True, keep_files=False)
794
 
 
795
 
 
796
 
class NonDirectoryParent(HandledConflict):
797
 
    """An attempt to add files to a directory that is not a directory or
798
 
    an attempt to change the kind of a directory with files.
799
 
    """
800
 
 
801
 
    typestring = 'non-directory parent'
802
 
 
803
 
    format = "Conflict: %(path)s is not a directory, but has files in it."\
804
 
             "  %(action)s."
805
 
 
806
 
    # FIXME: .OTHER should be used instead of .new when the conflict is created
807
 
 
808
 
    def action_take_this(self, tree):
809
 
        # FIXME: we should preserve that path when the conflict is generated !
810
 
        if self.path.endswith('.new'):
811
 
            conflict_path = self.path[:-(len('.new'))]
812
 
            tree.remove([self.path], force=True, keep_files=False)
813
 
            tree.add(conflict_path)
814
 
        else:
815
 
            raise NotImplementedError(self.action_take_this)
816
 
 
817
 
    def action_take_other(self, tree):
818
 
        # FIXME: we should preserve that path when the conflict is generated !
819
 
        if self.path.endswith('.new'):
820
 
            conflict_path = self.path[:-(len('.new'))]
821
 
            tree.remove([conflict_path], force=True, keep_files=False)
822
 
            tree.rename_one(self.path, conflict_path)
823
 
        else:
824
 
            raise NotImplementedError(self.action_take_other)
825
 
 
826
 
 
827
 
ctype = {}
828
 
 
829
 
 
830
 
def register_types(*conflict_types):
831
 
    """Register a Conflict subclass for serialization purposes"""
832
 
    global ctype
833
 
    for conflict_type in conflict_types:
834
 
        ctype[conflict_type.typestring] = conflict_type
835
 
 
836
 
register_types(ContentsConflict, TextConflict, PathConflict, DuplicateID,
837
 
               DuplicateEntry, ParentLoop, UnversionedParent, MissingParent,
838
 
               DeletingParent, NonDirectoryParent)