~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/check.py

  • Committer: Patch Queue Manager
  • Date: 2011-10-14 16:54:26 UTC
  • mfrom: (6216.1.1 remove-this-file)
  • Revision ID: pqm@pqm.ubuntu.com-20111014165426-tjix4e6idryf1r2z
(jelmer) Remove an accidentally committed .THIS file. (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2004, 2005 by Martin Pool
2
 
# Copyright (C) 2005 by Canonical Ltd
3
 
 
 
1
# Copyright (C) 2005, 2006 Canonical Ltd
 
2
#
4
3
# This program is free software; you can redistribute it and/or modify
5
4
# it under the terms of the GNU General Public License as published by
6
5
# the Free Software Foundation; either version 2 of the License, or
7
6
# (at your option) any later version.
8
 
 
 
7
#
9
8
# This program is distributed in the hope that it will be useful,
10
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
11
# GNU General Public License for more details.
13
 
 
 
12
#
14
13
# You should have received a copy of the GNU General Public License
15
14
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
16
 
18
17
# TODO: Check ancestries are correct for every revision: includes
19
18
# every committed so far, and in a reasonable order.
22
21
 
23
22
# TODO: Check for extra files in the control directory.
24
23
 
25
 
# TODO: Check revision, inventory and entry objects have all 
 
24
# TODO: Check revision, inventory and entry objects have all
26
25
# required fields.
27
26
 
28
 
 
29
 
import bzrlib.ui
30
 
from bzrlib.trace import note, warning
31
 
from bzrlib.osutils import rename, sha_string, fingerprint_file
32
 
from bzrlib.trace import mutter
33
 
from bzrlib.errors import BzrCheckError, NoSuchRevision
34
 
from bzrlib.inventory import ROOT_ID
35
 
from bzrlib.branch import gen_root_id
36
 
 
 
27
# TODO: Get every revision in the revision-store even if they're not
 
28
# referenced by history and make sure they're all valid.
 
29
 
 
30
# TODO: Perhaps have a way to record errors other than by raising exceptions;
 
31
# would perhaps be enough to accumulate exception objects in a list without
 
32
# raising them.  If there's more than one exception it'd be good to see them
 
33
# all.
 
34
 
 
35
"""Checking of bzr objects.
 
36
 
 
37
check_refs is a concept used for optimising check. Objects that depend on other
 
38
objects (e.g. tree on repository) can list the objects they would be requesting
 
39
so that when the dependent object is checked, matches can be pulled out and
 
40
evaluated in-line rather than re-reading the same data many times.
 
41
check_refs are tuples (kind, value). Currently defined kinds are:
 
42
 
 
43
* 'trees', where value is a revid and the looked up objects are revision trees.
 
44
* 'lefthand-distance', where value is a revid and the looked up objects are the
 
45
  distance along the lefthand path to NULL for that revid.
 
46
* 'revision-existence', where value is a revid, and the result is True or False
 
47
  indicating that the revision was found/not found.
 
48
"""
 
49
 
 
50
from bzrlib import (
 
51
    errors,
 
52
    ui,
 
53
    )
 
54
from bzrlib.branch import Branch
 
55
from bzrlib.controldir import ControlDir
 
56
from bzrlib.revision import NULL_REVISION
 
57
from bzrlib.trace import note
 
58
from bzrlib.workingtree import WorkingTree
 
59
from bzrlib.i18n import gettext
37
60
 
38
61
class Check(object):
39
 
    """Check a branch"""
40
 
 
41
 
    def __init__(self, branch):
42
 
        self.branch = branch
43
 
        self.checked_text_cnt = 0
 
62
    """Check a repository"""
 
63
 
 
64
    def __init__(self, repository, check_repo=True):
 
65
        self.repository = repository
 
66
 
 
67
    def report_results(self, verbose):
 
68
        raise NotImplementedError(self.report_results)
 
69
 
 
70
 
 
71
class VersionedFileCheck(Check):
 
72
    """Check a versioned file repository"""
 
73
 
 
74
    # The Check object interacts with InventoryEntry.check, etc.
 
75
 
 
76
    def __init__(self, repository, check_repo=True):
 
77
        self.repository = repository
44
78
        self.checked_rev_cnt = 0
45
 
        self.ghosts = []
46
 
        self.repeated_text_cnt = 0
 
79
        self.ghosts = set()
47
80
        self.missing_parent_links = {}
48
81
        self.missing_inventory_sha_cnt = 0
49
82
        self.missing_revision_cnt = 0
50
 
        # maps (file-id, version) -> sha1
51
 
        self.checked_texts = {}
 
83
        self.checked_weaves = set()
 
84
        self.unreferenced_versions = set()
 
85
        self.inconsistent_parents = []
 
86
        self.rich_roots = repository.supports_rich_root()
 
87
        self.text_key_references = {}
 
88
        self.check_repo = check_repo
 
89
        self.other_results = []
 
90
        # Plain text lines to include in the report
 
91
        self._report_items = []
 
92
        # Keys we are looking for; may be large and need spilling to disk.
 
93
        # key->(type(revision/inventory/text/signature/map), sha1, first-referer)
 
94
        self.pending_keys = {}
 
95
        # Ancestors map for all of revisions being checked; while large helper
 
96
        # functions we call would create it anyway, so better to have once and
 
97
        # keep.
 
98
        self.ancestors = {}
52
99
 
53
 
    def check(self):
54
 
        self.branch.lock_read()
 
100
    def check(self, callback_refs=None, check_repo=True):
 
101
        if callback_refs is None:
 
102
            callback_refs = {}
 
103
        self.repository.lock_read()
 
104
        self.progress = ui.ui_factory.nested_progress_bar()
55
105
        try:
56
 
            self.history = self.branch.revision_history()
57
 
            if not len(self.history):
58
 
                # nothing to see here
59
 
                return
60
 
            self.planned_revisions = self.branch.get_ancestry(self.history[-1])
61
 
            self.planned_revisions.remove(None)
62
 
            revno = 0
63
 
    
64
 
            self.progress = bzrlib.ui.ui_factory.progress_bar()
65
 
            while revno < len(self.planned_revisions):
66
 
                rev_id = self.planned_revisions[revno]
67
 
                self.progress.update('checking revision', revno,
68
 
                                     len(self.planned_revisions))
69
 
                revno += 1
70
 
                self.check_one_rev(rev_id)
71
 
            self.progress.clear()
 
106
            self.progress.update(gettext('check'), 0, 4)
 
107
            if self.check_repo:
 
108
                self.progress.update(gettext('checking revisions'), 0)
 
109
                self.check_revisions()
 
110
                self.progress.update(gettext('checking commit contents'), 1)
 
111
                self.repository._check_inventories(self)
 
112
                self.progress.update(gettext('checking file graphs'), 2)
 
113
                # check_weaves is done after the revision scan so that
 
114
                # revision index is known to be valid.
 
115
                self.check_weaves()
 
116
            self.progress.update(gettext('checking branches and trees'), 3)
 
117
            if callback_refs:
 
118
                repo = self.repository
 
119
                # calculate all refs, and callback the objects requesting them.
 
120
                refs = {}
 
121
                wanting_items = set()
 
122
                # Current crude version calculates everything and calls
 
123
                # everything at once. Doing a queue and popping as things are
 
124
                # satisfied would be cheaper on memory [but few people have
 
125
                # huge numbers of working trees today. TODO: fix before
 
126
                # landing].
 
127
                distances = set()
 
128
                existences = set()
 
129
                for ref, wantlist in callback_refs.iteritems():
 
130
                    wanting_items.update(wantlist)
 
131
                    kind, value = ref
 
132
                    if kind == 'trees':
 
133
                        refs[ref] = repo.revision_tree(value)
 
134
                    elif kind == 'lefthand-distance':
 
135
                        distances.add(value)
 
136
                    elif kind == 'revision-existence':
 
137
                        existences.add(value)
 
138
                    else:
 
139
                        raise AssertionError(
 
140
                            'unknown ref kind for ref %s' % ref)
 
141
                node_distances = repo.get_graph().find_lefthand_distances(distances)
 
142
                for key, distance in node_distances.iteritems():
 
143
                    refs[('lefthand-distance', key)] = distance
 
144
                    if key in existences and distance > 0:
 
145
                        refs[('revision-existence', key)] = True
 
146
                        existences.remove(key)
 
147
                parent_map = repo.get_graph().get_parent_map(existences)
 
148
                for key in parent_map:
 
149
                    refs[('revision-existence', key)] = True
 
150
                    existences.remove(key)
 
151
                for key in existences:
 
152
                    refs[('revision-existence', key)] = False
 
153
                for item in wanting_items:
 
154
                    if isinstance(item, WorkingTree):
 
155
                        item._check(refs)
 
156
                    if isinstance(item, Branch):
 
157
                        self.other_results.append(item.check(refs))
72
158
        finally:
73
 
            self.branch.unlock()
 
159
            self.progress.finished()
 
160
            self.repository.unlock()
 
161
 
 
162
    def _check_revisions(self, revisions_iterator):
 
163
        """Check revision objects by decorating a generator.
 
164
 
 
165
        :param revisions_iterator: An iterator of(revid, Revision-or-None).
 
166
        :return: A generator of the contents of revisions_iterator.
 
167
        """
 
168
        self.planned_revisions = set()
 
169
        for revid, revision in revisions_iterator:
 
170
            yield revid, revision
 
171
            self._check_one_rev(revid, revision)
 
172
        # Flatten the revisions we found to guarantee consistent later
 
173
        # iteration.
 
174
        self.planned_revisions = list(self.planned_revisions)
 
175
        # TODO: extract digital signatures as items to callback on too.
 
176
 
 
177
    def check_revisions(self):
 
178
        """Scan revisions, checking data directly available as we go."""
 
179
        revision_iterator = self.repository._iter_revisions(None)
 
180
        revision_iterator = self._check_revisions(revision_iterator)
 
181
        # We read the all revisions here:
 
182
        # - doing this allows later code to depend on the revision index.
 
183
        # - we can fill out existence flags at this point
 
184
        # - we can read the revision inventory sha at this point
 
185
        # - we can check properties and serialisers etc.
 
186
        if not self.repository._format.revision_graph_can_have_wrong_parents:
 
187
            # The check against the index isn't needed.
 
188
            self.revs_with_bad_parents_in_index = None
 
189
            for thing in revision_iterator:
 
190
                pass
 
191
        else:
 
192
            bad_revisions = self.repository._find_inconsistent_revision_parents(
 
193
                revision_iterator)
 
194
            self.revs_with_bad_parents_in_index = list(bad_revisions)
74
195
 
75
196
    def report_results(self, verbose):
76
 
        note('checked branch %s format %d',
77
 
             self.branch.base, 
78
 
             self.branch._branch_format)
 
197
        if self.check_repo:
 
198
            self._report_repo_results(verbose)
 
199
        for result in self.other_results:
 
200
            result.report_results(verbose)
79
201
 
80
 
        note('%6d revisions', self.checked_rev_cnt)
81
 
        note('%6d unique file texts', self.checked_text_cnt)
82
 
        note('%6d repeated file texts', self.repeated_text_cnt)
 
202
    def _report_repo_results(self, verbose):
 
203
        note(gettext('checked repository {0} format {1}').format(
 
204
            self.repository.user_url,
 
205
            self.repository._format))
 
206
        note(gettext('%6d revisions'), self.checked_rev_cnt)
 
207
        note(gettext('%6d file-ids'), len(self.checked_weaves))
 
208
        if verbose:
 
209
            note(gettext('%6d unreferenced text versions'),
 
210
                len(self.unreferenced_versions))
 
211
        if verbose and len(self.unreferenced_versions):
 
212
                for file_id, revision_id in self.unreferenced_versions:
 
213
                    note(gettext('unreferenced version: {{{0}}} in {1}').format(revision_id,
 
214
                        file_id))
83
215
        if self.missing_inventory_sha_cnt:
84
 
            note('%6d revisions are missing inventory_sha1',
 
216
            note(gettext('%6d revisions are missing inventory_sha1'),
85
217
                 self.missing_inventory_sha_cnt)
86
218
        if self.missing_revision_cnt:
87
 
            note('%6d revisions are mentioned but not present',
 
219
            note(gettext('%6d revisions are mentioned but not present'),
88
220
                 self.missing_revision_cnt)
89
221
        if len(self.ghosts):
90
 
            note('%6d ghost revisions', len(self.ghosts))
 
222
            note(gettext('%6d ghost revisions'), len(self.ghosts))
91
223
            if verbose:
92
224
                for ghost in self.ghosts:
93
225
                    note('      %s', ghost)
94
226
        if len(self.missing_parent_links):
95
 
            note('%6d revisions missing parents in ancestry', 
 
227
            note(gettext('%6d revisions missing parents in ancestry'),
96
228
                 len(self.missing_parent_links))
97
229
            if verbose:
98
230
                for link, linkers in self.missing_parent_links.items():
99
 
                    note('      %s should be in the ancestry for:', link)
 
231
                    note(gettext('      %s should be in the ancestry for:'), link)
100
232
                    for linker in linkers:
101
233
                        note('       * %s', linker)
102
 
 
103
 
    def check_one_rev(self, rev_id):
104
 
        """Check one revision.
105
 
 
106
 
        rev_id - the one to check
107
 
 
108
 
        last_rev_id - the previous one on the mainline, if any.
 
234
        if len(self.inconsistent_parents):
 
235
            note(gettext('%6d inconsistent parents'), len(self.inconsistent_parents))
 
236
            if verbose:
 
237
                for info in self.inconsistent_parents:
 
238
                    revision_id, file_id, found_parents, correct_parents = info
 
239
                    note(gettext('      * {0} version {1} has parents {2!r} '
 
240
                         'but should have {3!r}').format(
 
241
                         file_id, revision_id, found_parents,
 
242
                             correct_parents))
 
243
        if self.revs_with_bad_parents_in_index:
 
244
            note(gettext(
 
245
                 '%6d revisions have incorrect parents in the revision index'),
 
246
                 len(self.revs_with_bad_parents_in_index))
 
247
            if verbose:
 
248
                for item in self.revs_with_bad_parents_in_index:
 
249
                    revision_id, index_parents, actual_parents = item
 
250
                    note(gettext(
 
251
                        '       {0} has wrong parents in index: '
 
252
                        '{1!r} should be {2!r}').format(
 
253
                        revision_id, index_parents, actual_parents))
 
254
        for item in self._report_items:
 
255
            note(item)
 
256
 
 
257
    def _check_one_rev(self, rev_id, rev):
 
258
        """Cross-check one revision.
 
259
 
 
260
        :param rev_id: A revision id to check.
 
261
        :param rev: A revision or None to indicate a missing revision.
109
262
        """
110
 
 
111
 
        # mutter('    revision {%s}' % rev_id)
112
 
        branch = self.branch
113
 
        try:
114
 
            rev_history_position = self.history.index(rev_id)
115
 
        except ValueError:
116
 
            rev_history_position = None
117
 
        last_rev_id = None
118
 
        if rev_history_position:
119
 
            rev = branch.get_revision(rev_id)
120
 
            if rev_history_position > 0:
121
 
                last_rev_id = self.history[rev_history_position - 1]
122
 
        else:
123
 
            rev = branch.get_revision(rev_id)
124
 
                
125
263
        if rev.revision_id != rev_id:
126
 
            raise BzrCheckError('wrong internal revision id in revision {%s}'
127
 
                                % rev_id)
128
 
 
129
 
        # check the previous history entry is a parent of this entry
130
 
        if rev.parent_ids:
131
 
            if last_rev_id is None and rev_history_position is not None:
132
 
                # what if the start is a ghost ? i.e. conceptually the 
133
 
                # baz imports
134
 
                raise BzrCheckError("revision {%s} has %d parents, but is the "
135
 
                                    "start of the branch"
136
 
                                    % (rev_id, len(rev.parent_ids)))
137
 
            if last_rev_id is not None:
138
 
                for parent_id in rev.parent_ids:
139
 
                    if parent_id == last_rev_id:
140
 
                        break
141
 
                else:
142
 
                    raise BzrCheckError("previous revision {%s} not listed among "
143
 
                                        "parents of {%s}"
144
 
                                        % (last_rev_id, rev_id))
145
 
            for parent in rev.parent_ids:
146
 
                if not parent in self.planned_revisions:
147
 
                    missing_links = self.missing_parent_links.get(parent, [])
148
 
                    missing_links.append(rev_id)
149
 
                    self.missing_parent_links[parent] = missing_links
150
 
                    # list based so slow, TODO have a planned_revisions list and set.
151
 
                    if self.branch.has_revision(parent):
152
 
                        missing_ancestry = self.branch.get_ancestry(parent)
153
 
                        for missing in missing_ancestry:
154
 
                            if (missing is not None 
155
 
                                and missing not in self.planned_revisions):
156
 
                                self.planned_revisions.append(missing)
157
 
                    else:
158
 
                        self.ghosts.append(rev_id)
159
 
        elif last_rev_id:
160
 
            raise BzrCheckError("revision {%s} has no parents listed "
161
 
                                "but preceded by {%s}"
162
 
                                % (rev_id, last_rev_id))
163
 
 
164
 
        if rev.inventory_sha1:
165
 
            inv_sha1 = branch.get_inventory_sha1(rev_id)
166
 
            if inv_sha1 != rev.inventory_sha1:
167
 
                raise BzrCheckError('Inventory sha1 hash doesn\'t match'
168
 
                    ' value in revision {%s}' % rev_id)
169
 
        else:
170
 
            missing_inventory_sha_cnt += 1
171
 
            mutter("no inventory_sha1 on revision {%s}" % rev_id)
172
 
        self._check_revision_tree(rev_id)
 
264
            self._report_items.append(gettext(
 
265
                'Mismatched internal revid {{{0}}} and index revid {{{1}}}').format(
 
266
                rev.revision_id, rev_id))
 
267
            rev_id = rev.revision_id
 
268
        # Check this revision tree etc, and count as seen when we encounter a
 
269
        # reference to it.
 
270
        self.planned_revisions.add(rev_id)
 
271
        # It is not a ghost
 
272
        self.ghosts.discard(rev_id)
 
273
        # Count all parents as ghosts if we haven't seen them yet.
 
274
        for parent in rev.parent_ids:
 
275
            if not parent in self.planned_revisions:
 
276
                self.ghosts.add(parent)
 
277
        
 
278
        self.ancestors[rev_id] = tuple(rev.parent_ids) or (NULL_REVISION,)
 
279
        self.add_pending_item(rev_id, ('inventories', rev_id), 'inventory',
 
280
            rev.inventory_sha1)
173
281
        self.checked_rev_cnt += 1
174
282
 
175
 
    def _check_revision_tree(self, rev_id):
176
 
        tree = self.branch.revision_tree(rev_id)
177
 
        inv = tree.inventory
178
 
        seen_ids = {}
179
 
        for file_id in inv:
180
 
            if file_id in seen_ids:
181
 
                raise BzrCheckError('duplicated file_id {%s} '
182
 
                                    'in inventory for revision {%s}'
183
 
                                    % (file_id, rev_id))
184
 
            seen_ids[file_id] = True
185
 
        for file_id in inv:
186
 
            ie = inv[file_id]
187
 
            ie.check(self, rev_id, inv, tree)
188
 
        seen_names = {}
189
 
        for path, ie in inv.iter_entries():
190
 
            if path in seen_names:
191
 
                raise BzrCheckError('duplicated path %s '
192
 
                                    'in inventory for revision {%s}'
193
 
                                    % (path, rev_id))
194
 
            seen_names[path] = True
195
 
 
196
 
 
197
 
def check(branch, verbose):
198
 
    """Run consistency checks on a branch."""
199
 
    checker = Check(branch)
200
 
    checker.check()
201
 
    checker.report_results(verbose)
 
283
    def add_pending_item(self, referer, key, kind, sha1):
 
284
        """Add a reference to a sha1 to be cross checked against a key.
 
285
 
 
286
        :param referer: The referer that expects key to have sha1.
 
287
        :param key: A storage key e.g. ('texts', 'foo@bar-20040504-1234')
 
288
        :param kind: revision/inventory/text/map/signature
 
289
        :param sha1: A hex sha1 or None if no sha1 is known.
 
290
        """
 
291
        existing = self.pending_keys.get(key)
 
292
        if existing:
 
293
            if sha1 != existing[1]:
 
294
                self._report_items.append(gettext('Multiple expected sha1s for {0}. {{{1}}}'
 
295
                    ' expects {{{2}}}, {{{3}}} expects {{{4}}}').format(
 
296
                    key, referer, sha1, existing[1], existing[0]))
 
297
        else:
 
298
            self.pending_keys[key] = (kind, sha1, referer)
 
299
 
 
300
    def check_weaves(self):
 
301
        """Check all the weaves we can get our hands on.
 
302
        """
 
303
        weave_ids = []
 
304
        storebar = ui.ui_factory.nested_progress_bar()
 
305
        try:
 
306
            self._check_weaves(storebar)
 
307
        finally:
 
308
            storebar.finished()
 
309
 
 
310
    def _check_weaves(self, storebar):
 
311
        storebar.update('text-index', 0, 2)
 
312
        if self.repository._format.fast_deltas:
 
313
            # We haven't considered every fileid instance so far.
 
314
            weave_checker = self.repository._get_versioned_file_checker(
 
315
                ancestors=self.ancestors)
 
316
        else:
 
317
            weave_checker = self.repository._get_versioned_file_checker(
 
318
                text_key_references=self.text_key_references,
 
319
                ancestors=self.ancestors)
 
320
        storebar.update('file-graph', 1)
 
321
        result = weave_checker.check_file_version_parents(
 
322
            self.repository.texts)
 
323
        self.checked_weaves = weave_checker.file_ids
 
324
        bad_parents, unused_versions = result
 
325
        bad_parents = bad_parents.items()
 
326
        for text_key, (stored_parents, correct_parents) in bad_parents:
 
327
            # XXX not ready for id join/split operations.
 
328
            weave_id = text_key[0]
 
329
            revision_id = text_key[-1]
 
330
            weave_parents = tuple([parent[-1] for parent in stored_parents])
 
331
            correct_parents = tuple([parent[-1] for parent in correct_parents])
 
332
            self.inconsistent_parents.append(
 
333
                (revision_id, weave_id, weave_parents, correct_parents))
 
334
        self.unreferenced_versions.update(unused_versions)
 
335
 
 
336
    def _add_entry_to_text_key_references(self, inv, entry):
 
337
        if not self.rich_roots and entry.name == '':
 
338
            return
 
339
        key = (entry.file_id, entry.revision)
 
340
        self.text_key_references.setdefault(key, False)
 
341
        if entry.revision == inv.revision_id:
 
342
            self.text_key_references[key] = True
 
343
 
 
344
 
 
345
def scan_branch(branch, needed_refs, to_unlock):
 
346
    """Scan a branch for refs.
 
347
 
 
348
    :param branch:  The branch to schedule for checking.
 
349
    :param needed_refs: Refs we are accumulating.
 
350
    :param to_unlock: The unlock list accumulating.
 
351
    """
 
352
    note(gettext("Checking branch at '%s'.") % (branch.base,))
 
353
    branch.lock_read()
 
354
    to_unlock.append(branch)
 
355
    branch_refs = branch._get_check_refs()
 
356
    for ref in branch_refs:
 
357
        reflist = needed_refs.setdefault(ref, [])
 
358
        reflist.append(branch)
 
359
 
 
360
 
 
361
def scan_tree(base_tree, tree, needed_refs, to_unlock):
 
362
    """Scan a tree for refs.
 
363
 
 
364
    :param base_tree: The original tree check opened, used to detect duplicate
 
365
        tree checks.
 
366
    :param tree:  The tree to schedule for checking.
 
367
    :param needed_refs: Refs we are accumulating.
 
368
    :param to_unlock: The unlock list accumulating.
 
369
    """
 
370
    if base_tree is not None and tree.basedir == base_tree.basedir:
 
371
        return
 
372
    note(gettext("Checking working tree at '%s'.") % (tree.basedir,))
 
373
    tree.lock_read()
 
374
    to_unlock.append(tree)
 
375
    tree_refs = tree._get_check_refs()
 
376
    for ref in tree_refs:
 
377
        reflist = needed_refs.setdefault(ref, [])
 
378
        reflist.append(tree)
 
379
 
 
380
 
 
381
def check_dwim(path, verbose, do_branch=False, do_repo=False, do_tree=False):
 
382
    """Check multiple objects.
 
383
 
 
384
    If errors occur they are accumulated and reported as far as possible, and
 
385
    an exception raised at the end of the process.
 
386
    """
 
387
    try:
 
388
        base_tree, branch, repo, relpath = \
 
389
                        ControlDir.open_containing_tree_branch_or_repository(path)
 
390
    except errors.NotBranchError:
 
391
        base_tree = branch = repo = None
 
392
 
 
393
    to_unlock = []
 
394
    needed_refs= {}
 
395
    try:
 
396
        if base_tree is not None:
 
397
            # If the tree is a lightweight checkout we won't see it in
 
398
            # repo.find_branches - add now.
 
399
            if do_tree:
 
400
                scan_tree(None, base_tree, needed_refs, to_unlock)
 
401
            branch = base_tree.branch
 
402
        if branch is not None:
 
403
            # We have a branch
 
404
            if repo is None:
 
405
                # The branch is in a shared repository
 
406
                repo = branch.repository
 
407
        if repo is not None:
 
408
            repo.lock_read()
 
409
            to_unlock.append(repo)
 
410
            branches = repo.find_branches(using=True)
 
411
            saw_tree = False
 
412
            if do_branch or do_tree:
 
413
                for branch in branches:
 
414
                    if do_tree:
 
415
                        try:
 
416
                            tree = branch.bzrdir.open_workingtree()
 
417
                            saw_tree = True
 
418
                        except (errors.NotLocalUrl, errors.NoWorkingTree):
 
419
                            pass
 
420
                        else:
 
421
                            scan_tree(base_tree, tree, needed_refs, to_unlock)
 
422
                    if do_branch:
 
423
                        scan_branch(branch, needed_refs, to_unlock)
 
424
            if do_branch and not branches:
 
425
                note(gettext("No branch found at specified location."))
 
426
            if do_tree and base_tree is None and not saw_tree:
 
427
                note(gettext("No working tree found at specified location."))
 
428
            if do_repo or do_branch or do_tree:
 
429
                if do_repo:
 
430
                    note(gettext("Checking repository at '%s'.")
 
431
                         % (repo.user_url,))
 
432
                result = repo.check(None, callback_refs=needed_refs,
 
433
                    check_repo=do_repo)
 
434
                result.report_results(verbose)
 
435
        else:
 
436
            if do_tree:
 
437
                note(gettext("No working tree found at specified location."))
 
438
            if do_branch:
 
439
                note(gettext("No branch found at specified location."))
 
440
            if do_repo:
 
441
                note(gettext("No repository found at specified location."))
 
442
    finally:
 
443
        for thing in to_unlock:
 
444
            thing.unlock()