/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/tree.py

  • Committer: Ian Clatworthy
  • Date: 2008-06-25 07:17:14 UTC
  • mto: (3515.1.1 ianc-integration)
  • mto: This revision was merged to the branch mainline in revision 3516.
  • Revision ID: ian.clatworthy@canonical.com-20080625071714-am1ya6mgo54ikrgv
jam feedback - make patterns a separate help topic

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
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
"""Tree classes, representing directory at point in time.
 
18
"""
 
19
 
 
20
import os
 
21
from collections import deque
 
22
from cStringIO import StringIO
 
23
 
 
24
import bzrlib
 
25
from bzrlib import (
 
26
    conflicts as _mod_conflicts,
 
27
    delta,
 
28
    osutils,
 
29
    revision as _mod_revision,
 
30
    rules,
 
31
    symbol_versioning,
 
32
    )
 
33
from bzrlib.decorators import needs_read_lock
 
34
from bzrlib.errors import BzrError, BzrCheckError
 
35
from bzrlib import errors
 
36
from bzrlib.inventory import Inventory, InventoryFile
 
37
from bzrlib.inter import InterObject
 
38
from bzrlib.osutils import fingerprint_file
 
39
import bzrlib.revision
 
40
from bzrlib.trace import mutter, note
 
41
 
 
42
 
 
43
class Tree(object):
 
44
    """Abstract file tree.
 
45
 
 
46
    There are several subclasses:
 
47
    
 
48
    * `WorkingTree` exists as files on disk editable by the user.
 
49
 
 
50
    * `RevisionTree` is a tree as recorded at some point in the past.
 
51
 
 
52
    Trees contain an `Inventory` object, and also know how to retrieve
 
53
    file texts mentioned in the inventory, either from a working
 
54
    directory or from a store.
 
55
 
 
56
    It is possible for trees to contain files that are not described
 
57
    in their inventory or vice versa; for this use `filenames()`.
 
58
 
 
59
    Trees can be compared, etc, regardless of whether they are working
 
60
    trees or versioned trees.
 
61
    """
 
62
    
 
63
    def changes_from(self, other, want_unchanged=False, specific_files=None,
 
64
        extra_trees=None, require_versioned=False, include_root=False,
 
65
        want_unversioned=False):
 
66
        """Return a TreeDelta of the changes from other to this tree.
 
67
 
 
68
        :param other: A tree to compare with.
 
69
        :param specific_files: An optional list of file paths to restrict the
 
70
            comparison to. When mapping filenames to ids, all matches in all
 
71
            trees (including optional extra_trees) are used, and all children of
 
72
            matched directories are included.
 
73
        :param want_unchanged: An optional boolean requesting the inclusion of
 
74
            unchanged entries in the result.
 
75
        :param extra_trees: An optional list of additional trees to use when
 
76
            mapping the contents of specific_files (paths) to file_ids.
 
77
        :param require_versioned: An optional boolean (defaults to False). When
 
78
            supplied and True all the 'specific_files' must be versioned, or
 
79
            a PathsNotVersionedError will be thrown.
 
80
        :param want_unversioned: Scan for unversioned paths.
 
81
 
 
82
        The comparison will be performed by an InterTree object looked up on 
 
83
        self and other.
 
84
        """
 
85
        # Martin observes that Tree.changes_from returns a TreeDelta and this
 
86
        # may confuse people, because the class name of the returned object is
 
87
        # a synonym of the object referenced in the method name.
 
88
        return InterTree.get(other, self).compare(
 
89
            want_unchanged=want_unchanged,
 
90
            specific_files=specific_files,
 
91
            extra_trees=extra_trees,
 
92
            require_versioned=require_versioned,
 
93
            include_root=include_root,
 
94
            want_unversioned=want_unversioned,
 
95
            )
 
96
 
 
97
    @symbol_versioning.deprecated_method(symbol_versioning.one_three)
 
98
    def _iter_changes(self, *args, **kwargs):
 
99
        return self.iter_changes(*args, **kwargs)
 
100
 
 
101
    def iter_changes(self, from_tree, include_unchanged=False,
 
102
                     specific_files=None, pb=None, extra_trees=None,
 
103
                     require_versioned=True, want_unversioned=False):
 
104
        intertree = InterTree.get(from_tree, self)
 
105
        return intertree.iter_changes(include_unchanged, specific_files, pb,
 
106
            extra_trees, require_versioned, want_unversioned=want_unversioned)
 
107
    
 
108
    def conflicts(self):
 
109
        """Get a list of the conflicts in the tree.
 
110
 
 
111
        Each conflict is an instance of bzrlib.conflicts.Conflict.
 
112
        """
 
113
        return _mod_conflicts.ConflictList()
 
114
 
 
115
    def extras(self):
 
116
        """For trees that can have unversioned files, return all such paths."""
 
117
        return []
 
118
 
 
119
    def get_parent_ids(self):
 
120
        """Get the parent ids for this tree. 
 
121
 
 
122
        :return: a list of parent ids. [] is returned to indicate
 
123
        a tree with no parents.
 
124
        :raises: BzrError if the parents are not known.
 
125
        """
 
126
        raise NotImplementedError(self.get_parent_ids)
 
127
    
 
128
    def has_filename(self, filename):
 
129
        """True if the tree has given filename."""
 
130
        raise NotImplementedError(self.has_filename)
 
131
 
 
132
    def has_id(self, file_id):
 
133
        return self.inventory.has_id(file_id)
 
134
 
 
135
    __contains__ = has_id
 
136
 
 
137
    def has_or_had_id(self, file_id):
 
138
        if file_id == self.inventory.root.file_id:
 
139
            return True
 
140
        return self.inventory.has_id(file_id)
 
141
 
 
142
    def is_ignored(self, filename):
 
143
        """Check whether the filename is ignored by this tree.
 
144
 
 
145
        :param filename: The relative filename within the tree.
 
146
        :return: True if the filename is ignored.
 
147
        """
 
148
        return False
 
149
 
 
150
    def __iter__(self):
 
151
        return iter(self.inventory)
 
152
 
 
153
    def all_file_ids(self):
 
154
        """Iterate through all file ids, including ids for missing files."""
 
155
        return set(self.inventory)
 
156
 
 
157
    def id2path(self, file_id):
 
158
        """Return the path for a file id.
 
159
 
 
160
        :raises NoSuchId:
 
161
        """
 
162
        return self.inventory.id2path(file_id)
 
163
 
 
164
    def is_control_filename(self, filename):
 
165
        """True if filename is the name of a control file in this tree.
 
166
        
 
167
        :param filename: A filename within the tree. This is a relative path
 
168
        from the root of this tree.
 
169
 
 
170
        This is true IF and ONLY IF the filename is part of the meta data
 
171
        that bzr controls in this tree. I.E. a random .bzr directory placed
 
172
        on disk will not be a control file for this tree.
 
173
        """
 
174
        return self.bzrdir.is_control_filename(filename)
 
175
 
 
176
    @needs_read_lock
 
177
    def iter_entries_by_dir(self, specific_file_ids=None):
 
178
        """Walk the tree in 'by_dir' order.
 
179
 
 
180
        This will yield each entry in the tree as a (path, entry) tuple. The
 
181
        order that they are yielded is: the contents of a directory are 
 
182
        preceeded by the parent of a directory, and all the contents of a 
 
183
        directory are grouped together.
 
184
        """
 
185
        return self.inventory.iter_entries_by_dir(
 
186
            specific_file_ids=specific_file_ids)
 
187
 
 
188
    def iter_references(self):
 
189
        for path, entry in self.iter_entries_by_dir():
 
190
            if entry.kind == 'tree-reference':
 
191
                yield path, entry.file_id
 
192
 
 
193
    def kind(self, file_id):
 
194
        raise NotImplementedError("Tree subclass %s must implement kind"
 
195
            % self.__class__.__name__)
 
196
 
 
197
    def stored_kind(self, file_id):
 
198
        """File kind stored for this file_id.
 
199
 
 
200
        May not match kind on disk for working trees.  Always available
 
201
        for versioned files, even when the file itself is missing.
 
202
        """
 
203
        return self.kind(file_id)
 
204
 
 
205
    def path_content_summary(self, path):
 
206
        """Get a summary of the information about path.
 
207
        
 
208
        :param path: A relative path within the tree.
 
209
        :return: A tuple containing kind, size, exec, sha1-or-link.
 
210
            Kind is always present (see tree.kind()).
 
211
            size is present if kind is file, None otherwise.
 
212
            exec is None unless kind is file and the platform supports the 'x'
 
213
                bit.
 
214
            sha1-or-link is the link target if kind is symlink, or the sha1 if
 
215
                it can be obtained without reading the file.
 
216
        """
 
217
        raise NotImplementedError(self.path_content_summary)
 
218
 
 
219
    def get_reference_revision(self, file_id, path=None):
 
220
        raise NotImplementedError("Tree subclass %s must implement "
 
221
                                  "get_reference_revision"
 
222
            % self.__class__.__name__)
 
223
 
 
224
    def _comparison_data(self, entry, path):
 
225
        """Return a tuple of kind, executable, stat_value for a file.
 
226
 
 
227
        entry may be None if there is no inventory entry for the file, but
 
228
        path must always be supplied.
 
229
 
 
230
        kind is None if there is no file present (even if an inventory id is
 
231
        present).  executable is False for non-file entries.
 
232
        """
 
233
        raise NotImplementedError(self._comparison_data)
 
234
 
 
235
    def _file_size(self, entry, stat_value):
 
236
        raise NotImplementedError(self._file_size)
 
237
 
 
238
    def _get_inventory(self):
 
239
        return self._inventory
 
240
    
 
241
    def get_file(self, file_id, path=None):
 
242
        """Return a file object for the file file_id in the tree.
 
243
        
 
244
        If both file_id and path are defined, it is implementation defined as
 
245
        to which one is used.
 
246
        """
 
247
        raise NotImplementedError(self.get_file)
 
248
 
 
249
    def get_file_mtime(self, file_id, path=None):
 
250
        """Return the modification time for a file.
 
251
 
 
252
        :param file_id: The handle for this file.
 
253
        :param path: The path that this file can be found at.
 
254
            These must point to the same object.
 
255
        """
 
256
        raise NotImplementedError(self.get_file_mtime)
 
257
 
 
258
    def get_file_size(self, file_id):
 
259
        """Return the size of a file in bytes.
 
260
 
 
261
        This applies only to regular files.  If invoked on directories or
 
262
        symlinks, it will return None.
 
263
        :param file_id: The file-id of the file
 
264
        """
 
265
        raise NotImplementedError(self.get_file_size)
 
266
 
 
267
    def get_file_by_path(self, path):
 
268
        return self.get_file(self._inventory.path2id(path), path)
 
269
 
 
270
    def iter_files_bytes(self, desired_files):
 
271
        """Iterate through file contents.
 
272
 
 
273
        Files will not necessarily be returned in the order they occur in
 
274
        desired_files.  No specific order is guaranteed.
 
275
 
 
276
        Yields pairs of identifier, bytes_iterator.  identifier is an opaque
 
277
        value supplied by the caller as part of desired_files.  It should
 
278
        uniquely identify the file version in the caller's context.  (Examples:
 
279
        an index number or a TreeTransform trans_id.)
 
280
 
 
281
        bytes_iterator is an iterable of bytestrings for the file.  The
 
282
        kind of iterable and length of the bytestrings are unspecified, but for
 
283
        this implementation, it is a tuple containing a single bytestring with
 
284
        the complete text of the file.
 
285
 
 
286
        :param desired_files: a list of (file_id, identifier) pairs
 
287
        """
 
288
        for file_id, identifier in desired_files:
 
289
            # We wrap the string in a tuple so that we can return an iterable
 
290
            # of bytestrings.  (Technically, a bytestring is also an iterable
 
291
            # of bytestrings, but iterating through each character is not
 
292
            # performant.)
 
293
            cur_file = (self.get_file_text(file_id),)
 
294
            yield identifier, cur_file
 
295
 
 
296
    def get_symlink_target(self, file_id):
 
297
        """Get the target for a given file_id.
 
298
 
 
299
        It is assumed that the caller already knows that file_id is referencing
 
300
        a symlink.
 
301
        :param file_id: Handle for the symlink entry.
 
302
        :return: The path the symlink points to.
 
303
        """
 
304
        raise NotImplementedError(self.get_symlink_target)
 
305
 
 
306
    def get_root_id(self):
 
307
        """Return the file_id for the root of this tree."""
 
308
        raise NotImplementedError(self.get_root_id)
 
309
 
 
310
    def annotate_iter(self, file_id,
 
311
                      default_revision=_mod_revision.CURRENT_REVISION):
 
312
        """Return an iterator of revision_id, line tuples.
 
313
 
 
314
        For working trees (and mutable trees in general), the special
 
315
        revision_id 'current:' will be used for lines that are new in this
 
316
        tree, e.g. uncommitted changes.
 
317
        :param file_id: The file to produce an annotated version from
 
318
        :param default_revision: For lines that don't match a basis, mark them
 
319
            with this revision id. Not all implementations will make use of
 
320
            this value.
 
321
        """
 
322
        raise NotImplementedError(self.annotate_iter)
 
323
 
 
324
    def _get_plan_merge_data(self, file_id, other, base):
 
325
        from bzrlib import merge, versionedfile
 
326
        vf = versionedfile._PlanMergeVersionedFile(file_id)
 
327
        last_revision_a = self._get_file_revision(file_id, vf, 'this:')
 
328
        last_revision_b = other._get_file_revision(file_id, vf, 'other:')
 
329
        if base is None:
 
330
            last_revision_base = None
 
331
        else:
 
332
            last_revision_base = base._get_file_revision(file_id, vf, 'base:')
 
333
        return vf, last_revision_a, last_revision_b, last_revision_base
 
334
 
 
335
    def plan_file_merge(self, file_id, other, base=None):
 
336
        """Generate a merge plan based on annotations.
 
337
 
 
338
        If the file contains uncommitted changes in this tree, they will be
 
339
        attributed to the 'current:' pseudo-revision.  If the file contains
 
340
        uncommitted changes in the other tree, they will be assigned to the
 
341
        'other:' pseudo-revision.
 
342
        """
 
343
        data = self._get_plan_merge_data(file_id, other, base)
 
344
        vf, last_revision_a, last_revision_b, last_revision_base = data
 
345
        return vf.plan_merge(last_revision_a, last_revision_b,
 
346
                             last_revision_base)
 
347
 
 
348
    def plan_file_lca_merge(self, file_id, other, base=None):
 
349
        """Generate a merge plan based lca-newness.
 
350
 
 
351
        If the file contains uncommitted changes in this tree, they will be
 
352
        attributed to the 'current:' pseudo-revision.  If the file contains
 
353
        uncommitted changes in the other tree, they will be assigned to the
 
354
        'other:' pseudo-revision.
 
355
        """
 
356
        data = self._get_plan_merge_data(file_id, other, base)
 
357
        vf, last_revision_a, last_revision_b, last_revision_base = data
 
358
        return vf.plan_lca_merge(last_revision_a, last_revision_b,
 
359
                                 last_revision_base)
 
360
 
 
361
    def _get_file_revision(self, file_id, vf, tree_revision):
 
362
        def file_revision(revision_tree):
 
363
            revision_tree.lock_read()
 
364
            try:
 
365
                return revision_tree.inventory[file_id].revision
 
366
            finally:
 
367
                revision_tree.unlock()
 
368
 
 
369
        def iter_parent_trees():
 
370
            for revision_id in self.get_parent_ids():
 
371
                try:
 
372
                    yield self.revision_tree(revision_id)
 
373
                except:
 
374
                    yield self.repository.revision_tree(revision_id)
 
375
 
 
376
        if getattr(self, '_get_weave', None) is None:
 
377
            last_revision = tree_revision
 
378
            parent_revisions = [file_revision(t) for t in iter_parent_trees()]
 
379
            vf.add_lines(last_revision, parent_revisions,
 
380
                         self.get_file(file_id).readlines())
 
381
            repo = self.branch.repository
 
382
            transaction = repo.get_transaction()
 
383
            base_vf = repo.weave_store.get_weave(file_id, transaction)
 
384
        else:
 
385
            last_revision = file_revision(self)
 
386
            base_vf = self._get_weave(file_id)
 
387
        vf.fallback_versionedfiles.append(base_vf)
 
388
        return last_revision
 
389
 
 
390
    inventory = property(_get_inventory,
 
391
                         doc="Inventory of this Tree")
 
392
 
 
393
    def _check_retrieved(self, ie, f):
 
394
        if not __debug__:
 
395
            return  
 
396
        fp = fingerprint_file(f)
 
397
        f.seek(0)
 
398
        
 
399
        if ie.text_size is not None:
 
400
            if ie.text_size != fp['size']:
 
401
                raise BzrError("mismatched size for file %r in %r" % (ie.file_id, self._store),
 
402
                        ["inventory expects %d bytes" % ie.text_size,
 
403
                         "file is actually %d bytes" % fp['size'],
 
404
                         "store is probably damaged/corrupt"])
 
405
 
 
406
        if ie.text_sha1 != fp['sha1']:
 
407
            raise BzrError("wrong SHA-1 for file %r in %r" % (ie.file_id, self._store),
 
408
                    ["inventory expects %s" % ie.text_sha1,
 
409
                     "file is actually %s" % fp['sha1'],
 
410
                     "store is probably damaged/corrupt"])
 
411
 
 
412
    @needs_read_lock
 
413
    def path2id(self, path):
 
414
        """Return the id for path in this tree."""
 
415
        return self._inventory.path2id(path)
 
416
 
 
417
    def paths2ids(self, paths, trees=[], require_versioned=True):
 
418
        """Return all the ids that can be reached by walking from paths.
 
419
        
 
420
        Each path is looked up in this tree and any extras provided in
 
421
        trees, and this is repeated recursively: the children in an extra tree
 
422
        of a directory that has been renamed under a provided path in this tree
 
423
        are all returned, even if none exist under a provided path in this
 
424
        tree, and vice versa.
 
425
 
 
426
        :param paths: An iterable of paths to start converting to ids from.
 
427
            Alternatively, if paths is None, no ids should be calculated and None
 
428
            will be returned. This is offered to make calling the api unconditional
 
429
            for code that *might* take a list of files.
 
430
        :param trees: Additional trees to consider.
 
431
        :param require_versioned: If False, do not raise NotVersionedError if
 
432
            an element of paths is not versioned in this tree and all of trees.
 
433
        """
 
434
        return find_ids_across_trees(paths, [self] + list(trees), require_versioned)
 
435
 
 
436
    def print_file(self, file_id):
 
437
        """Print file with id `file_id` to stdout."""
 
438
        import sys
 
439
        sys.stdout.write(self.get_file_text(file_id))
 
440
 
 
441
    def lock_read(self):
 
442
        pass
 
443
 
 
444
    def revision_tree(self, revision_id):
 
445
        """Obtain a revision tree for the revision revision_id.
 
446
 
 
447
        The intention of this method is to allow access to possibly cached
 
448
        tree data. Implementors of this method should raise NoSuchRevision if
 
449
        the tree is not locally available, even if they could obtain the 
 
450
        tree via a repository or some other means. Callers are responsible 
 
451
        for finding the ultimate source for a revision tree.
 
452
 
 
453
        :param revision_id: The revision_id of the requested tree.
 
454
        :return: A Tree.
 
455
        :raises: NoSuchRevision if the tree cannot be obtained.
 
456
        """
 
457
        raise errors.NoSuchRevisionInTree(self, revision_id)
 
458
 
 
459
    def unknowns(self):
 
460
        """What files are present in this tree and unknown.
 
461
        
 
462
        :return: an iterator over the unknown files.
 
463
        """
 
464
        return iter([])
 
465
 
 
466
    def unlock(self):
 
467
        pass
 
468
 
 
469
    def filter_unversioned_files(self, paths):
 
470
        """Filter out paths that are versioned.
 
471
 
 
472
        :return: set of paths.
 
473
        """
 
474
        # NB: we specifically *don't* call self.has_filename, because for
 
475
        # WorkingTrees that can indicate files that exist on disk but that 
 
476
        # are not versioned.
 
477
        pred = self.inventory.has_filename
 
478
        return set((p for p in paths if not pred(p)))
 
479
 
 
480
    def walkdirs(self, prefix=""):
 
481
        """Walk the contents of this tree from path down.
 
482
 
 
483
        This yields all the data about the contents of a directory at a time.
 
484
        After each directory has been yielded, if the caller has mutated the
 
485
        list to exclude some directories, they are then not descended into.
 
486
        
 
487
        The data yielded is of the form:
 
488
        ((directory-relpath, directory-path-from-root, directory-fileid),
 
489
        [(relpath, basename, kind, lstat, path_from_tree_root, file_id, 
 
490
          versioned_kind), ...]),
 
491
         - directory-relpath is the containing dirs relpath from prefix
 
492
         - directory-path-from-root is the containing dirs path from /
 
493
         - directory-fileid is the id of the directory if it is versioned.
 
494
         - relpath is the relative path within the subtree being walked.
 
495
         - basename is the basename
 
496
         - kind is the kind of the file now. If unknonwn then the file is not
 
497
           present within the tree - but it may be recorded as versioned. See
 
498
           versioned_kind.
 
499
         - lstat is the stat data *if* the file was statted.
 
500
         - path_from_tree_root is the path from the root of the tree.
 
501
         - file_id is the file_id if the entry is versioned.
 
502
         - versioned_kind is the kind of the file as last recorded in the 
 
503
           versioning system. If 'unknown' the file is not versioned.
 
504
        One of 'kind' and 'versioned_kind' must not be 'unknown'.
 
505
 
 
506
        :param prefix: Start walking from prefix within the tree rather than
 
507
        at the root. This allows one to walk a subtree but get paths that are
 
508
        relative to a tree rooted higher up.
 
509
        :return: an iterator over the directory data.
 
510
        """
 
511
        raise NotImplementedError(self.walkdirs)
 
512
 
 
513
    def iter_search_rules(self, path_names, pref_names=None,
 
514
        _default_searcher=rules._per_user_searcher):
 
515
        """Find the preferences for filenames in a tree.
 
516
 
 
517
        :param path_names: an iterable of paths to find attributes for.
 
518
          Paths are given relative to the root of the tree.
 
519
        :param pref_names: the list of preferences to lookup - None for all
 
520
        :param _default_searcher: private parameter to assist testing - don't use
 
521
        :return: an iterator of tuple sequences, one per path-name.
 
522
          See _RulesSearcher.get_items for details on the tuple sequence.
 
523
        """
 
524
        searcher = self._get_rules_searcher(_default_searcher)
 
525
        if searcher is not None:
 
526
            for path in path_names:
 
527
                yield searcher.get_items(path, pref_names)
 
528
 
 
529
    @needs_read_lock
 
530
    def _get_rules_searcher(self, default_searcher):
 
531
        """Get the RulesSearcher for this tree given the default one."""
 
532
        searcher = default_searcher
 
533
        file_id = self.path2id(rules.RULES_TREE_FILENAME)
 
534
        if file_id is not None:
 
535
            ini_file = self.get_file(file_id)
 
536
            searcher = rules._StackedRulesSearcher(
 
537
                [rules._IniBasedRulesSearcher(ini_file), default_searcher])
 
538
        return searcher
 
539
 
 
540
 
 
541
class EmptyTree(Tree):
 
542
 
 
543
    def __init__(self):
 
544
        self._inventory = Inventory(root_id=None)
 
545
        symbol_versioning.warn('EmptyTree is deprecated as of bzr 0.9 please'
 
546
                               ' use repository.revision_tree instead.',
 
547
                               DeprecationWarning, stacklevel=2)
 
548
 
 
549
    def get_parent_ids(self):
 
550
        return []
 
551
 
 
552
    def get_symlink_target(self, file_id):
 
553
        return None
 
554
 
 
555
    def has_filename(self, filename):
 
556
        return False
 
557
 
 
558
    def kind(self, file_id):
 
559
        return "directory"
 
560
 
 
561
    def list_files(self, include_root=False):
 
562
        return iter([])
 
563
    
 
564
    def __contains__(self, file_id):
 
565
        return (file_id in self._inventory)
 
566
 
 
567
    def get_file_sha1(self, file_id, path=None, stat_value=None):
 
568
        return None
 
569
 
 
570
 
 
571
######################################################################
 
572
# diff
 
573
 
 
574
# TODO: Merge these two functions into a single one that can operate
 
575
# on either a whole tree or a set of files.
 
576
 
 
577
# TODO: Return the diff in order by filename, not by category or in
 
578
# random order.  Can probably be done by lock-stepping through the
 
579
# filenames from both trees.
 
580
 
 
581
 
 
582
def file_status(filename, old_tree, new_tree):
 
583
    """Return single-letter status, old and new names for a file.
 
584
 
 
585
    The complexity here is in deciding how to represent renames;
 
586
    many complex cases are possible.
 
587
    """
 
588
    old_inv = old_tree.inventory
 
589
    new_inv = new_tree.inventory
 
590
    new_id = new_inv.path2id(filename)
 
591
    old_id = old_inv.path2id(filename)
 
592
 
 
593
    if not new_id and not old_id:
 
594
        # easy: doesn't exist in either; not versioned at all
 
595
        if new_tree.is_ignored(filename):
 
596
            return 'I', None, None
 
597
        else:
 
598
            return '?', None, None
 
599
    elif new_id:
 
600
        # There is now a file of this name, great.
 
601
        pass
 
602
    else:
 
603
        # There is no longer a file of this name, but we can describe
 
604
        # what happened to the file that used to have
 
605
        # this name.  There are two possibilities: either it was
 
606
        # deleted entirely, or renamed.
 
607
        if new_inv.has_id(old_id):
 
608
            return 'X', old_inv.id2path(old_id), new_inv.id2path(old_id)
 
609
        else:
 
610
            return 'D', old_inv.id2path(old_id), None
 
611
 
 
612
    # if the file_id is new in this revision, it is added
 
613
    if new_id and not old_inv.has_id(new_id):
 
614
        return 'A'
 
615
 
 
616
    # if there used to be a file of this name, but that ID has now
 
617
    # disappeared, it is deleted
 
618
    if old_id and not new_inv.has_id(old_id):
 
619
        return 'D'
 
620
 
 
621
    return 'wtf?'
 
622
 
 
623
    
 
624
 
 
625
def find_renames(old_inv, new_inv):
 
626
    for file_id in old_inv:
 
627
        if file_id not in new_inv:
 
628
            continue
 
629
        old_name = old_inv.id2path(file_id)
 
630
        new_name = new_inv.id2path(file_id)
 
631
        if old_name != new_name:
 
632
            yield (old_name, new_name)
 
633
            
 
634
 
 
635
def find_ids_across_trees(filenames, trees, require_versioned=True):
 
636
    """Find the ids corresponding to specified filenames.
 
637
    
 
638
    All matches in all trees will be used, and all children of matched
 
639
    directories will be used.
 
640
 
 
641
    :param filenames: The filenames to find file_ids for (if None, returns
 
642
        None)
 
643
    :param trees: The trees to find file_ids within
 
644
    :param require_versioned: if true, all specified filenames must occur in
 
645
    at least one tree.
 
646
    :return: a set of file ids for the specified filenames and their children.
 
647
    """
 
648
    if not filenames:
 
649
        return None
 
650
    specified_path_ids = _find_ids_across_trees(filenames, trees,
 
651
        require_versioned)
 
652
    return _find_children_across_trees(specified_path_ids, trees)
 
653
 
 
654
 
 
655
def _find_ids_across_trees(filenames, trees, require_versioned):
 
656
    """Find the ids corresponding to specified filenames.
 
657
    
 
658
    All matches in all trees will be used, but subdirectories are not scanned.
 
659
 
 
660
    :param filenames: The filenames to find file_ids for
 
661
    :param trees: The trees to find file_ids within
 
662
    :param require_versioned: if true, all specified filenames must occur in
 
663
        at least one tree.
 
664
    :return: a set of file ids for the specified filenames
 
665
    """
 
666
    not_versioned = []
 
667
    interesting_ids = set()
 
668
    for tree_path in filenames:
 
669
        not_found = True
 
670
        for tree in trees:
 
671
            file_id = tree.path2id(tree_path)
 
672
            if file_id is not None:
 
673
                interesting_ids.add(file_id)
 
674
                not_found = False
 
675
        if not_found:
 
676
            not_versioned.append(tree_path)
 
677
    if len(not_versioned) > 0 and require_versioned:
 
678
        raise errors.PathsNotVersionedError(not_versioned)
 
679
    return interesting_ids
 
680
 
 
681
 
 
682
def _find_children_across_trees(specified_ids, trees):
 
683
    """Return a set including specified ids and their children.
 
684
    
 
685
    All matches in all trees will be used.
 
686
 
 
687
    :param trees: The trees to find file_ids within
 
688
    :return: a set containing all specified ids and their children 
 
689
    """
 
690
    interesting_ids = set(specified_ids)
 
691
    pending = interesting_ids
 
692
    # now handle children of interesting ids
 
693
    # we loop so that we handle all children of each id in both trees
 
694
    while len(pending) > 0:
 
695
        new_pending = set()
 
696
        for file_id in pending:
 
697
            for tree in trees:
 
698
                if not tree.has_id(file_id):
 
699
                    continue
 
700
                entry = tree.inventory[file_id]
 
701
                for child in getattr(entry, 'children', {}).itervalues():
 
702
                    if child.file_id not in interesting_ids:
 
703
                        new_pending.add(child.file_id)
 
704
        interesting_ids.update(new_pending)
 
705
        pending = new_pending
 
706
    return interesting_ids
 
707
 
 
708
 
 
709
class InterTree(InterObject):
 
710
    """This class represents operations taking place between two Trees.
 
711
 
 
712
    Its instances have methods like 'compare' and contain references to the
 
713
    source and target trees these operations are to be carried out on.
 
714
 
 
715
    Clients of bzrlib should not need to use InterTree directly, rather they
 
716
    should use the convenience methods on Tree such as 'Tree.compare()' which
 
717
    will pass through to InterTree as appropriate.
 
718
    """
 
719
 
 
720
    _optimisers = []
 
721
 
 
722
    @needs_read_lock
 
723
    def compare(self, want_unchanged=False, specific_files=None,
 
724
        extra_trees=None, require_versioned=False, include_root=False,
 
725
        want_unversioned=False):
 
726
        """Return the changes from source to target.
 
727
 
 
728
        :return: A TreeDelta.
 
729
        :param specific_files: An optional list of file paths to restrict the
 
730
            comparison to. When mapping filenames to ids, all matches in all
 
731
            trees (including optional extra_trees) are used, and all children of
 
732
            matched directories are included.
 
733
        :param want_unchanged: An optional boolean requesting the inclusion of
 
734
            unchanged entries in the result.
 
735
        :param extra_trees: An optional list of additional trees to use when
 
736
            mapping the contents of specific_files (paths) to file_ids.
 
737
        :param require_versioned: An optional boolean (defaults to False). When
 
738
            supplied and True all the 'specific_files' must be versioned, or
 
739
            a PathsNotVersionedError will be thrown.
 
740
        :param want_unversioned: Scan for unversioned paths.
 
741
        """
 
742
        # NB: show_status depends on being able to pass in non-versioned files
 
743
        # and report them as unknown
 
744
        trees = (self.source,)
 
745
        if extra_trees is not None:
 
746
            trees = trees + tuple(extra_trees)
 
747
        # target is usually the newer tree:
 
748
        specific_file_ids = self.target.paths2ids(specific_files, trees,
 
749
            require_versioned=require_versioned)
 
750
        if specific_files and not specific_file_ids:
 
751
            # All files are unversioned, so just return an empty delta
 
752
            # _compare_trees would think we want a complete delta
 
753
            result = delta.TreeDelta()
 
754
            fake_entry = InventoryFile('unused', 'unused', 'unused')
 
755
            result.unversioned = [(path, None,
 
756
                self.target._comparison_data(fake_entry, path)[0]) for path in
 
757
                specific_files]
 
758
            return result
 
759
        return delta._compare_trees(self.source, self.target, want_unchanged,
 
760
            specific_files, include_root, extra_trees=extra_trees,
 
761
            require_versioned=require_versioned,
 
762
            want_unversioned=want_unversioned)
 
763
 
 
764
    def iter_changes(self, include_unchanged=False,
 
765
                      specific_files=None, pb=None, extra_trees=[],
 
766
                      require_versioned=True, want_unversioned=False):
 
767
        """Generate an iterator of changes between trees.
 
768
 
 
769
        A tuple is returned:
 
770
        (file_id, (path_in_source, path_in_target),
 
771
         changed_content, versioned, parent, name, kind,
 
772
         executable)
 
773
 
 
774
        Changed_content is True if the file's content has changed.  This
 
775
        includes changes to its kind, and to a symlink's target.
 
776
 
 
777
        versioned, parent, name, kind, executable are tuples of (from, to).
 
778
        If a file is missing in a tree, its kind is None.
 
779
 
 
780
        Iteration is done in parent-to-child order, relative to the target
 
781
        tree.
 
782
 
 
783
        There is no guarantee that all paths are in sorted order: the
 
784
        requirement to expand the search due to renames may result in children
 
785
        that should be found early being found late in the search, after
 
786
        lexically later results have been returned.
 
787
        :param require_versioned: Raise errors.PathsNotVersionedError if a
 
788
            path in the specific_files list is not versioned in one of
 
789
            source, target or extra_trees.
 
790
        :param want_unversioned: Should unversioned files be returned in the
 
791
            output. An unversioned file is defined as one with (False, False)
 
792
            for the versioned pair.
 
793
        """
 
794
        result = []
 
795
        lookup_trees = [self.source]
 
796
        if extra_trees:
 
797
             lookup_trees.extend(extra_trees)
 
798
        if specific_files == []:
 
799
            specific_file_ids = []
 
800
        else:
 
801
            specific_file_ids = self.target.paths2ids(specific_files,
 
802
                lookup_trees, require_versioned=require_versioned)
 
803
        if want_unversioned:
 
804
            all_unversioned = sorted([(p.split('/'), p) for p in
 
805
                                     self.target.extras()
 
806
                if specific_files is None or
 
807
                    osutils.is_inside_any(specific_files, p)])
 
808
            all_unversioned = deque(all_unversioned)
 
809
        else:
 
810
            all_unversioned = deque()
 
811
        to_paths = {}
 
812
        from_entries_by_dir = list(self.source.inventory.iter_entries_by_dir(
 
813
            specific_file_ids=specific_file_ids))
 
814
        from_data = dict((e.file_id, (p, e)) for p, e in from_entries_by_dir)
 
815
        to_entries_by_dir = list(self.target.inventory.iter_entries_by_dir(
 
816
            specific_file_ids=specific_file_ids))
 
817
        num_entries = len(from_entries_by_dir) + len(to_entries_by_dir)
 
818
        entry_count = 0
 
819
        # the unversioned path lookup only occurs on real trees - where there 
 
820
        # can be extras. So the fake_entry is solely used to look up
 
821
        # executable it values when execute is not supported.
 
822
        fake_entry = InventoryFile('unused', 'unused', 'unused')
 
823
        for to_path, to_entry in to_entries_by_dir:
 
824
            while all_unversioned and all_unversioned[0][0] < to_path.split('/'):
 
825
                unversioned_path = all_unversioned.popleft()
 
826
                to_kind, to_executable, to_stat = \
 
827
                    self.target._comparison_data(fake_entry, unversioned_path[1])
 
828
                yield (None, (None, unversioned_path[1]), True, (False, False),
 
829
                    (None, None),
 
830
                    (None, unversioned_path[0][-1]),
 
831
                    (None, to_kind),
 
832
                    (None, to_executable))
 
833
            file_id = to_entry.file_id
 
834
            to_paths[file_id] = to_path
 
835
            entry_count += 1
 
836
            changed_content = False
 
837
            from_path, from_entry = from_data.get(file_id, (None, None))
 
838
            from_versioned = (from_entry is not None)
 
839
            if from_entry is not None:
 
840
                from_versioned = True
 
841
                from_name = from_entry.name
 
842
                from_parent = from_entry.parent_id
 
843
                from_kind, from_executable, from_stat = \
 
844
                    self.source._comparison_data(from_entry, from_path)
 
845
                entry_count += 1
 
846
            else:
 
847
                from_versioned = False
 
848
                from_kind = None
 
849
                from_parent = None
 
850
                from_name = None
 
851
                from_executable = None
 
852
            versioned = (from_versioned, True)
 
853
            to_kind, to_executable, to_stat = \
 
854
                self.target._comparison_data(to_entry, to_path)
 
855
            kind = (from_kind, to_kind)
 
856
            if kind[0] != kind[1]:
 
857
                changed_content = True
 
858
            elif from_kind == 'file':
 
859
                from_size = self.source._file_size(from_entry, from_stat)
 
860
                to_size = self.target._file_size(to_entry, to_stat)
 
861
                if from_size != to_size:
 
862
                    changed_content = True
 
863
                elif (self.source.get_file_sha1(file_id, from_path, from_stat) !=
 
864
                    self.target.get_file_sha1(file_id, to_path, to_stat)):
 
865
                    changed_content = True
 
866
            elif from_kind == 'symlink':
 
867
                if (self.source.get_symlink_target(file_id) !=
 
868
                    self.target.get_symlink_target(file_id)):
 
869
                    changed_content = True
 
870
                elif from_kind == 'tree-reference':
 
871
                    if (self.source.get_reference_revision(file_id, from_path)
 
872
                        != self.target.get_reference_revision(file_id, to_path)):
 
873
                        changed_content = True 
 
874
            parent = (from_parent, to_entry.parent_id)
 
875
            name = (from_name, to_entry.name)
 
876
            executable = (from_executable, to_executable)
 
877
            if pb is not None:
 
878
                pb.update('comparing files', entry_count, num_entries)
 
879
            if (changed_content is not False or versioned[0] != versioned[1]
 
880
                or parent[0] != parent[1] or name[0] != name[1] or 
 
881
                executable[0] != executable[1] or include_unchanged):
 
882
                yield (file_id, (from_path, to_path), changed_content,
 
883
                    versioned, parent, name, kind, executable)
 
884
 
 
885
        while all_unversioned:
 
886
            # yield any trailing unversioned paths
 
887
            unversioned_path = all_unversioned.popleft()
 
888
            to_kind, to_executable, to_stat = \
 
889
                self.target._comparison_data(fake_entry, unversioned_path[1])
 
890
            yield (None, (None, unversioned_path[1]), True, (False, False),
 
891
                (None, None),
 
892
                (None, unversioned_path[0][-1]),
 
893
                (None, to_kind),
 
894
                (None, to_executable))
 
895
 
 
896
        def get_to_path(to_entry):
 
897
            if to_entry.parent_id is None:
 
898
                to_path = '' # the root
 
899
            else:
 
900
                if to_entry.parent_id not in to_paths:
 
901
                    # recurse up
 
902
                    return get_to_path(self.target.inventory[to_entry.parent_id])
 
903
                to_path = osutils.pathjoin(to_paths[to_entry.parent_id],
 
904
                                           to_entry.name)
 
905
            to_paths[to_entry.file_id] = to_path
 
906
            return to_path
 
907
 
 
908
        for path, from_entry in from_entries_by_dir:
 
909
            file_id = from_entry.file_id
 
910
            if file_id in to_paths:
 
911
                # already returned
 
912
                continue
 
913
            if not file_id in self.target.inventory:
 
914
                # common case - paths we have not emitted are not present in
 
915
                # target.
 
916
                to_path = None
 
917
            else:
 
918
                to_path = get_to_path(self.target.inventory[file_id])
 
919
            entry_count += 1
 
920
            if pb is not None:
 
921
                pb.update('comparing files', entry_count, num_entries)
 
922
            versioned = (True, False)
 
923
            parent = (from_entry.parent_id, None)
 
924
            name = (from_entry.name, None)
 
925
            from_kind, from_executable, stat_value = \
 
926
                self.source._comparison_data(from_entry, path)
 
927
            kind = (from_kind, None)
 
928
            executable = (from_executable, None)
 
929
            changed_content = True
 
930
            # the parent's path is necessarily known at this point.
 
931
            yield(file_id, (path, to_path), changed_content, versioned, parent,
 
932
                  name, kind, executable)