~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/controldir.py

  • Committer: Wouter van Heyst
  • Date: 2012-01-25 21:13:15 UTC
  • mto: (6437.3.27 2.5)
  • mto: This revision was merged to the branch mainline in revision 6451.
  • Revision ID: larstiq@larstiq.dyndns.org-20120125211315-ch74doua37bqet0i
mechanically replace file().write() pattern with a with-keyword version

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2010, 2011 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
"""ControlDir is the basic control directory class.
 
18
 
 
19
The ControlDir class is the base for the control directory used
 
20
by all bzr and foreign formats. For the ".bzr" implementation,
 
21
see bzrlib.bzrdir.BzrDir.
 
22
 
 
23
"""
 
24
 
 
25
from __future__ import absolute_import
 
26
 
 
27
from bzrlib.lazy_import import lazy_import
 
28
lazy_import(globals(), """
 
29
import textwrap
 
30
 
 
31
from bzrlib import (
 
32
    errors,
 
33
    hooks,
 
34
    revision as _mod_revision,
 
35
    transport as _mod_transport,
 
36
    trace,
 
37
    ui,
 
38
    urlutils,
 
39
    )
 
40
from bzrlib.transport import local
 
41
from bzrlib.push import (
 
42
    PushResult,
 
43
    )
 
44
 
 
45
from bzrlib.i18n import gettext
 
46
""")
 
47
 
 
48
from bzrlib import registry
 
49
 
 
50
 
 
51
class ControlComponent(object):
 
52
    """Abstract base class for control directory components.
 
53
 
 
54
    This provides interfaces that are common across controldirs,
 
55
    repositories, branches, and workingtree control directories.
 
56
 
 
57
    They all expose two urls and transports: the *user* URL is the
 
58
    one that stops above the control directory (eg .bzr) and that
 
59
    should normally be used in messages, and the *control* URL is
 
60
    under that in eg .bzr/checkout and is used to read the control
 
61
    files.
 
62
 
 
63
    This can be used as a mixin and is intended to fit with
 
64
    foreign formats.
 
65
    """
 
66
 
 
67
    @property
 
68
    def control_transport(self):
 
69
        raise NotImplementedError
 
70
 
 
71
    @property
 
72
    def control_url(self):
 
73
        return self.control_transport.base
 
74
 
 
75
    @property
 
76
    def user_transport(self):
 
77
        raise NotImplementedError
 
78
 
 
79
    @property
 
80
    def user_url(self):
 
81
        return self.user_transport.base
 
82
 
 
83
 
 
84
class ControlDir(ControlComponent):
 
85
    """A control directory.
 
86
 
 
87
    While this represents a generic control directory, there are a few
 
88
    features that are present in this interface that are currently only
 
89
    supported by one of its implementations, BzrDir.
 
90
 
 
91
    These features (bound branches, stacked branches) are currently only
 
92
    supported by Bazaar, but could be supported by other version control
 
93
    systems as well. Implementations are required to raise the appropriate
 
94
    exceptions when an operation is requested that is not supported.
 
95
 
 
96
    This also makes life easier for API users who can rely on the
 
97
    implementation always allowing a particular feature to be requested but
 
98
    raising an exception when it is not supported, rather than requiring the
 
99
    API users to check for magic attributes to see what features are supported.
 
100
    """
 
101
 
 
102
    def can_convert_format(self):
 
103
        """Return true if this controldir is one whose format we can convert
 
104
        from."""
 
105
        return True
 
106
 
 
107
    def list_branches(self):
 
108
        """Return a sequence of all branches local to this control directory.
 
109
 
 
110
        """
 
111
        return self.get_branches().values()
 
112
 
 
113
    def get_branches(self):
 
114
        """Get all branches in this control directory, as a dictionary.
 
115
        
 
116
        :return: Dictionary mapping branch names to instances.
 
117
        """
 
118
        try:
 
119
           return { "": self.open_branch() }
 
120
        except (errors.NotBranchError, errors.NoRepositoryPresent):
 
121
           return {}
 
122
 
 
123
    def is_control_filename(self, filename):
 
124
        """True if filename is the name of a path which is reserved for
 
125
        controldirs.
 
126
 
 
127
        :param filename: A filename within the root transport of this
 
128
            controldir.
 
129
 
 
130
        This is true IF and ONLY IF the filename is part of the namespace reserved
 
131
        for bzr control dirs. Currently this is the '.bzr' directory in the root
 
132
        of the root_transport. it is expected that plugins will need to extend
 
133
        this in the future - for instance to make bzr talk with svn working
 
134
        trees.
 
135
        """
 
136
        raise NotImplementedError(self.is_control_filename)
 
137
 
 
138
    def needs_format_conversion(self, format=None):
 
139
        """Return true if this controldir needs convert_format run on it.
 
140
 
 
141
        For instance, if the repository format is out of date but the
 
142
        branch and working tree are not, this should return True.
 
143
 
 
144
        :param format: Optional parameter indicating a specific desired
 
145
                       format we plan to arrive at.
 
146
        """
 
147
        raise NotImplementedError(self.needs_format_conversion)
 
148
 
 
149
    def create_repository(self, shared=False):
 
150
        """Create a new repository in this control directory.
 
151
 
 
152
        :param shared: If a shared repository should be created
 
153
        :return: The newly created repository
 
154
        """
 
155
        raise NotImplementedError(self.create_repository)
 
156
 
 
157
    def destroy_repository(self):
 
158
        """Destroy the repository in this ControlDir."""
 
159
        raise NotImplementedError(self.destroy_repository)
 
160
 
 
161
    def create_branch(self, name=None, repository=None,
 
162
                      append_revisions_only=None):
 
163
        """Create a branch in this ControlDir.
 
164
 
 
165
        :param name: Name of the colocated branch to create, None for
 
166
            the default branch.
 
167
        :param append_revisions_only: Whether this branch should only allow
 
168
            appending new revisions to its history.
 
169
 
 
170
        The controldirs format will control what branch format is created.
 
171
        For more control see BranchFormatXX.create(a_controldir).
 
172
        """
 
173
        raise NotImplementedError(self.create_branch)
 
174
 
 
175
    def destroy_branch(self, name=None):
 
176
        """Destroy a branch in this ControlDir.
 
177
 
 
178
        :param name: Name of the branch to destroy, None for the default 
 
179
            branch.
 
180
        """
 
181
        raise NotImplementedError(self.destroy_branch)
 
182
 
 
183
    def create_workingtree(self, revision_id=None, from_branch=None,
 
184
        accelerator_tree=None, hardlink=False):
 
185
        """Create a working tree at this ControlDir.
 
186
 
 
187
        :param revision_id: create it as of this revision id.
 
188
        :param from_branch: override controldir branch 
 
189
            (for lightweight checkouts)
 
190
        :param accelerator_tree: A tree which can be used for retrieving file
 
191
            contents more quickly than the revision tree, i.e. a workingtree.
 
192
            The revision tree will be used for cases where accelerator_tree's
 
193
            content is different.
 
194
        """
 
195
        raise NotImplementedError(self.create_workingtree)
 
196
 
 
197
    def destroy_workingtree(self):
 
198
        """Destroy the working tree at this ControlDir.
 
199
 
 
200
        Formats that do not support this may raise UnsupportedOperation.
 
201
        """
 
202
        raise NotImplementedError(self.destroy_workingtree)
 
203
 
 
204
    def destroy_workingtree_metadata(self):
 
205
        """Destroy the control files for the working tree at this ControlDir.
 
206
 
 
207
        The contents of working tree files are not affected.
 
208
        Formats that do not support this may raise UnsupportedOperation.
 
209
        """
 
210
        raise NotImplementedError(self.destroy_workingtree_metadata)
 
211
 
 
212
    def find_branch_format(self, name=None):
 
213
        """Find the branch 'format' for this controldir.
 
214
 
 
215
        This might be a synthetic object for e.g. RemoteBranch and SVN.
 
216
        """
 
217
        raise NotImplementedError(self.find_branch_format)
 
218
 
 
219
    def get_branch_reference(self, name=None):
 
220
        """Return the referenced URL for the branch in this controldir.
 
221
 
 
222
        :param name: Optional colocated branch name
 
223
        :raises NotBranchError: If there is no Branch.
 
224
        :raises NoColocatedBranchSupport: If a branch name was specified
 
225
            but colocated branches are not supported.
 
226
        :return: The URL the branch in this controldir references if it is a
 
227
            reference branch, or None for regular branches.
 
228
        """
 
229
        if name is not None:
 
230
            raise errors.NoColocatedBranchSupport(self)
 
231
        return None
 
232
 
 
233
    def set_branch_reference(self, target_branch, name=None):
 
234
        """Set the referenced URL for the branch in this controldir.
 
235
 
 
236
        :param name: Optional colocated branch name
 
237
        :param target_branch: Branch to reference
 
238
        :raises NoColocatedBranchSupport: If a branch name was specified
 
239
            but colocated branches are not supported.
 
240
        :return: The referencing branch
 
241
        """
 
242
        raise NotImplementedError(self.set_branch_reference)
 
243
 
 
244
    def open_branch(self, name=None, unsupported=False,
 
245
                    ignore_fallbacks=False, possible_transports=None):
 
246
        """Open the branch object at this ControlDir if one is present.
 
247
 
 
248
        :param unsupported: if True, then no longer supported branch formats can
 
249
            still be opened.
 
250
        :param ignore_fallbacks: Whether to open fallback repositories
 
251
        :param possible_transports: Transports to use for opening e.g.
 
252
            fallback repositories.
 
253
        """
 
254
        raise NotImplementedError(self.open_branch)
 
255
 
 
256
    def open_repository(self, _unsupported=False):
 
257
        """Open the repository object at this ControlDir if one is present.
 
258
 
 
259
        This will not follow the Branch object pointer - it's strictly a direct
 
260
        open facility. Most client code should use open_branch().repository to
 
261
        get at a repository.
 
262
 
 
263
        :param _unsupported: a private parameter, not part of the api.
 
264
        """
 
265
        raise NotImplementedError(self.open_repository)
 
266
 
 
267
    def find_repository(self):
 
268
        """Find the repository that should be used.
 
269
 
 
270
        This does not require a branch as we use it to find the repo for
 
271
        new branches as well as to hook existing branches up to their
 
272
        repository.
 
273
        """
 
274
        raise NotImplementedError(self.find_repository)
 
275
 
 
276
    def open_workingtree(self, unsupported=False,
 
277
                         recommend_upgrade=True, from_branch=None):
 
278
        """Open the workingtree object at this ControlDir if one is present.
 
279
 
 
280
        :param recommend_upgrade: Optional keyword parameter, when True (the
 
281
            default), emit through the ui module a recommendation that the user
 
282
            upgrade the working tree when the workingtree being opened is old
 
283
            (but still fully supported).
 
284
        :param from_branch: override controldir branch (for lightweight
 
285
            checkouts)
 
286
        """
 
287
        raise NotImplementedError(self.open_workingtree)
 
288
 
 
289
    def has_branch(self, name=None):
 
290
        """Tell if this controldir contains a branch.
 
291
 
 
292
        Note: if you're going to open the branch, you should just go ahead
 
293
        and try, and not ask permission first.  (This method just opens the
 
294
        branch and discards it, and that's somewhat expensive.)
 
295
        """
 
296
        try:
 
297
            self.open_branch(name, ignore_fallbacks=True)
 
298
            return True
 
299
        except errors.NotBranchError:
 
300
            return False
 
301
 
 
302
    def _get_selected_branch(self):
 
303
        """Return the name of the branch selected by the user.
 
304
 
 
305
        :return: Name of the branch selected by the user, or None.
 
306
        """
 
307
        branch = self.root_transport.get_segment_parameters().get("branch")
 
308
        if branch is None:
 
309
            branch = ""
 
310
        return urlutils.unescape(branch)
 
311
 
 
312
    def has_workingtree(self):
 
313
        """Tell if this controldir contains a working tree.
 
314
 
 
315
        This will still raise an exception if the controldir has a workingtree
 
316
        that is remote & inaccessible.
 
317
 
 
318
        Note: if you're going to open the working tree, you should just go ahead
 
319
        and try, and not ask permission first.  (This method just opens the
 
320
        workingtree and discards it, and that's somewhat expensive.)
 
321
        """
 
322
        try:
 
323
            self.open_workingtree(recommend_upgrade=False)
 
324
            return True
 
325
        except errors.NoWorkingTree:
 
326
            return False
 
327
 
 
328
    def cloning_metadir(self, require_stacking=False):
 
329
        """Produce a metadir suitable for cloning or sprouting with.
 
330
 
 
331
        These operations may produce workingtrees (yes, even though they're
 
332
        "cloning" something that doesn't have a tree), so a viable workingtree
 
333
        format must be selected.
 
334
 
 
335
        :require_stacking: If True, non-stackable formats will be upgraded
 
336
            to similar stackable formats.
 
337
        :returns: a ControlDirFormat with all component formats either set
 
338
            appropriately or set to None if that component should not be
 
339
            created.
 
340
        """
 
341
        raise NotImplementedError(self.cloning_metadir)
 
342
 
 
343
    def checkout_metadir(self):
 
344
        """Produce a metadir suitable for checkouts of this controldir.
 
345
 
 
346
        :returns: A ControlDirFormat with all component formats
 
347
            either set appropriately or set to None if that component
 
348
            should not be created.
 
349
        """
 
350
        return self.cloning_metadir()
 
351
 
 
352
    def sprout(self, url, revision_id=None, force_new_repo=False,
 
353
               recurse='down', possible_transports=None,
 
354
               accelerator_tree=None, hardlink=False, stacked=False,
 
355
               source_branch=None, create_tree_if_local=True):
 
356
        """Create a copy of this controldir prepared for use as a new line of
 
357
        development.
 
358
 
 
359
        If url's last component does not exist, it will be created.
 
360
 
 
361
        Attributes related to the identity of the source branch like
 
362
        branch nickname will be cleaned, a working tree is created
 
363
        whether one existed before or not; and a local branch is always
 
364
        created.
 
365
 
 
366
        :param revision_id: if revision_id is not None, then the clone
 
367
            operation may tune itself to download less data.
 
368
        :param accelerator_tree: A tree which can be used for retrieving file
 
369
            contents more quickly than the revision tree, i.e. a workingtree.
 
370
            The revision tree will be used for cases where accelerator_tree's
 
371
            content is different.
 
372
        :param hardlink: If true, hard-link files from accelerator_tree,
 
373
            where possible.
 
374
        :param stacked: If true, create a stacked branch referring to the
 
375
            location of this control directory.
 
376
        :param create_tree_if_local: If true, a working-tree will be created
 
377
            when working locally.
 
378
        """
 
379
        raise NotImplementedError(self.sprout)
 
380
 
 
381
    def push_branch(self, source, revision_id=None, overwrite=False, 
 
382
        remember=False, create_prefix=False):
 
383
        """Push the source branch into this ControlDir."""
 
384
        br_to = None
 
385
        # If we can open a branch, use its direct repository, otherwise see
 
386
        # if there is a repository without a branch.
 
387
        try:
 
388
            br_to = self.open_branch()
 
389
        except errors.NotBranchError:
 
390
            # Didn't find a branch, can we find a repository?
 
391
            repository_to = self.find_repository()
 
392
        else:
 
393
            # Found a branch, so we must have found a repository
 
394
            repository_to = br_to.repository
 
395
 
 
396
        push_result = PushResult()
 
397
        push_result.source_branch = source
 
398
        if br_to is None:
 
399
            # We have a repository but no branch, copy the revisions, and then
 
400
            # create a branch.
 
401
            if revision_id is None:
 
402
                # No revision supplied by the user, default to the branch
 
403
                # revision
 
404
                revision_id = source.last_revision()
 
405
            repository_to.fetch(source.repository, revision_id=revision_id)
 
406
            br_to = source.clone(self, revision_id=revision_id)
 
407
            if source.get_push_location() is None or remember:
 
408
                source.set_push_location(br_to.base)
 
409
            push_result.stacked_on = None
 
410
            push_result.branch_push_result = None
 
411
            push_result.old_revno = None
 
412
            push_result.old_revid = _mod_revision.NULL_REVISION
 
413
            push_result.target_branch = br_to
 
414
            push_result.master_branch = None
 
415
            push_result.workingtree_updated = False
 
416
        else:
 
417
            # We have successfully opened the branch, remember if necessary:
 
418
            if source.get_push_location() is None or remember:
 
419
                source.set_push_location(br_to.base)
 
420
            try:
 
421
                tree_to = self.open_workingtree()
 
422
            except errors.NotLocalUrl:
 
423
                push_result.branch_push_result = source.push(br_to, 
 
424
                    overwrite, stop_revision=revision_id)
 
425
                push_result.workingtree_updated = False
 
426
            except errors.NoWorkingTree:
 
427
                push_result.branch_push_result = source.push(br_to,
 
428
                    overwrite, stop_revision=revision_id)
 
429
                push_result.workingtree_updated = None # Not applicable
 
430
            else:
 
431
                tree_to.lock_write()
 
432
                try:
 
433
                    push_result.branch_push_result = source.push(
 
434
                        tree_to.branch, overwrite, stop_revision=revision_id)
 
435
                    tree_to.update()
 
436
                finally:
 
437
                    tree_to.unlock()
 
438
                push_result.workingtree_updated = True
 
439
            push_result.old_revno = push_result.branch_push_result.old_revno
 
440
            push_result.old_revid = push_result.branch_push_result.old_revid
 
441
            push_result.target_branch = \
 
442
                push_result.branch_push_result.target_branch
 
443
        return push_result
 
444
 
 
445
    def _get_tree_branch(self, name=None):
 
446
        """Return the branch and tree, if any, for this controldir.
 
447
 
 
448
        :param name: Name of colocated branch to open.
 
449
 
 
450
        Return None for tree if not present or inaccessible.
 
451
        Raise NotBranchError if no branch is present.
 
452
        :return: (tree, branch)
 
453
        """
 
454
        try:
 
455
            tree = self.open_workingtree()
 
456
        except (errors.NoWorkingTree, errors.NotLocalUrl):
 
457
            tree = None
 
458
            branch = self.open_branch(name=name)
 
459
        else:
 
460
            if name is not None:
 
461
                branch = self.open_branch(name=name)
 
462
            else:
 
463
                branch = tree.branch
 
464
        return tree, branch
 
465
 
 
466
    def get_config(self):
 
467
        """Get configuration for this ControlDir."""
 
468
        raise NotImplementedError(self.get_config)
 
469
 
 
470
    def check_conversion_target(self, target_format):
 
471
        """Check that a controldir as a whole can be converted to a new format."""
 
472
        raise NotImplementedError(self.check_conversion_target)
 
473
 
 
474
    def clone(self, url, revision_id=None, force_new_repo=False,
 
475
              preserve_stacking=False):
 
476
        """Clone this controldir and its contents to url verbatim.
 
477
 
 
478
        :param url: The url create the clone at.  If url's last component does
 
479
            not exist, it will be created.
 
480
        :param revision_id: The tip revision-id to use for any branch or
 
481
            working tree.  If not None, then the clone operation may tune
 
482
            itself to download less data.
 
483
        :param force_new_repo: Do not use a shared repository for the target
 
484
                               even if one is available.
 
485
        :param preserve_stacking: When cloning a stacked branch, stack the
 
486
            new branch on top of the other branch's stacked-on branch.
 
487
        """
 
488
        return self.clone_on_transport(_mod_transport.get_transport(url),
 
489
                                       revision_id=revision_id,
 
490
                                       force_new_repo=force_new_repo,
 
491
                                       preserve_stacking=preserve_stacking)
 
492
 
 
493
    def clone_on_transport(self, transport, revision_id=None,
 
494
        force_new_repo=False, preserve_stacking=False, stacked_on=None,
 
495
        create_prefix=False, use_existing_dir=True, no_tree=False):
 
496
        """Clone this controldir and its contents to transport verbatim.
 
497
 
 
498
        :param transport: The transport for the location to produce the clone
 
499
            at.  If the target directory does not exist, it will be created.
 
500
        :param revision_id: The tip revision-id to use for any branch or
 
501
            working tree.  If not None, then the clone operation may tune
 
502
            itself to download less data.
 
503
        :param force_new_repo: Do not use a shared repository for the target,
 
504
                               even if one is available.
 
505
        :param preserve_stacking: When cloning a stacked branch, stack the
 
506
            new branch on top of the other branch's stacked-on branch.
 
507
        :param create_prefix: Create any missing directories leading up to
 
508
            to_transport.
 
509
        :param use_existing_dir: Use an existing directory if one exists.
 
510
        :param no_tree: If set to true prevents creation of a working tree.
 
511
        """
 
512
        raise NotImplementedError(self.clone_on_transport)
 
513
 
 
514
    @classmethod
 
515
    def find_bzrdirs(klass, transport, evaluate=None, list_current=None):
 
516
        """Find control dirs recursively from current location.
 
517
 
 
518
        This is intended primarily as a building block for more sophisticated
 
519
        functionality, like finding trees under a directory, or finding
 
520
        branches that use a given repository.
 
521
 
 
522
        :param evaluate: An optional callable that yields recurse, value,
 
523
            where recurse controls whether this controldir is recursed into
 
524
            and value is the value to yield.  By default, all bzrdirs
 
525
            are recursed into, and the return value is the controldir.
 
526
        :param list_current: if supplied, use this function to list the current
 
527
            directory, instead of Transport.list_dir
 
528
        :return: a generator of found bzrdirs, or whatever evaluate returns.
 
529
        """
 
530
        if list_current is None:
 
531
            def list_current(transport):
 
532
                return transport.list_dir('')
 
533
        if evaluate is None:
 
534
            def evaluate(controldir):
 
535
                return True, controldir
 
536
 
 
537
        pending = [transport]
 
538
        while len(pending) > 0:
 
539
            current_transport = pending.pop()
 
540
            recurse = True
 
541
            try:
 
542
                controldir = klass.open_from_transport(current_transport)
 
543
            except (errors.NotBranchError, errors.PermissionDenied):
 
544
                pass
 
545
            else:
 
546
                recurse, value = evaluate(controldir)
 
547
                yield value
 
548
            try:
 
549
                subdirs = list_current(current_transport)
 
550
            except (errors.NoSuchFile, errors.PermissionDenied):
 
551
                continue
 
552
            if recurse:
 
553
                for subdir in sorted(subdirs, reverse=True):
 
554
                    pending.append(current_transport.clone(subdir))
 
555
 
 
556
    @classmethod
 
557
    def find_branches(klass, transport):
 
558
        """Find all branches under a transport.
 
559
 
 
560
        This will find all branches below the transport, including branches
 
561
        inside other branches.  Where possible, it will use
 
562
        Repository.find_branches.
 
563
 
 
564
        To list all the branches that use a particular Repository, see
 
565
        Repository.find_branches
 
566
        """
 
567
        def evaluate(controldir):
 
568
            try:
 
569
                repository = controldir.open_repository()
 
570
            except errors.NoRepositoryPresent:
 
571
                pass
 
572
            else:
 
573
                return False, ([], repository)
 
574
            return True, (controldir.list_branches(), None)
 
575
        ret = []
 
576
        for branches, repo in klass.find_bzrdirs(
 
577
                transport, evaluate=evaluate):
 
578
            if repo is not None:
 
579
                ret.extend(repo.find_branches())
 
580
            if branches is not None:
 
581
                ret.extend(branches)
 
582
        return ret
 
583
 
 
584
    @classmethod
 
585
    def create_branch_and_repo(klass, base, force_new_repo=False, format=None):
 
586
        """Create a new ControlDir, Branch and Repository at the url 'base'.
 
587
 
 
588
        This will use the current default ControlDirFormat unless one is
 
589
        specified, and use whatever
 
590
        repository format that that uses via controldir.create_branch and
 
591
        create_repository. If a shared repository is available that is used
 
592
        preferentially.
 
593
 
 
594
        The created Branch object is returned.
 
595
 
 
596
        :param base: The URL to create the branch at.
 
597
        :param force_new_repo: If True a new repository is always created.
 
598
        :param format: If supplied, the format of branch to create.  If not
 
599
            supplied, the default is used.
 
600
        """
 
601
        controldir = klass.create(base, format)
 
602
        controldir._find_or_create_repository(force_new_repo)
 
603
        return controldir.create_branch()
 
604
 
 
605
    @classmethod
 
606
    def create_branch_convenience(klass, base, force_new_repo=False,
 
607
                                  force_new_tree=None, format=None,
 
608
                                  possible_transports=None):
 
609
        """Create a new ControlDir, Branch and Repository at the url 'base'.
 
610
 
 
611
        This is a convenience function - it will use an existing repository
 
612
        if possible, can be told explicitly whether to create a working tree or
 
613
        not.
 
614
 
 
615
        This will use the current default ControlDirFormat unless one is
 
616
        specified, and use whatever
 
617
        repository format that that uses via ControlDir.create_branch and
 
618
        create_repository. If a shared repository is available that is used
 
619
        preferentially. Whatever repository is used, its tree creation policy
 
620
        is followed.
 
621
 
 
622
        The created Branch object is returned.
 
623
        If a working tree cannot be made due to base not being a file:// url,
 
624
        no error is raised unless force_new_tree is True, in which case no
 
625
        data is created on disk and NotLocalUrl is raised.
 
626
 
 
627
        :param base: The URL to create the branch at.
 
628
        :param force_new_repo: If True a new repository is always created.
 
629
        :param force_new_tree: If True or False force creation of a tree or
 
630
                               prevent such creation respectively.
 
631
        :param format: Override for the controldir format to create.
 
632
        :param possible_transports: An optional reusable transports list.
 
633
        """
 
634
        if force_new_tree:
 
635
            # check for non local urls
 
636
            t = _mod_transport.get_transport(base, possible_transports)
 
637
            if not isinstance(t, local.LocalTransport):
 
638
                raise errors.NotLocalUrl(base)
 
639
        controldir = klass.create(base, format, possible_transports)
 
640
        repo = controldir._find_or_create_repository(force_new_repo)
 
641
        result = controldir.create_branch()
 
642
        if force_new_tree or (repo.make_working_trees() and
 
643
                              force_new_tree is None):
 
644
            try:
 
645
                controldir.create_workingtree()
 
646
            except errors.NotLocalUrl:
 
647
                pass
 
648
        return result
 
649
 
 
650
    @classmethod
 
651
    def create_standalone_workingtree(klass, base, format=None):
 
652
        """Create a new ControlDir, WorkingTree, Branch and Repository at 'base'.
 
653
 
 
654
        'base' must be a local path or a file:// url.
 
655
 
 
656
        This will use the current default ControlDirFormat unless one is
 
657
        specified, and use whatever
 
658
        repository format that that uses for bzrdirformat.create_workingtree,
 
659
        create_branch and create_repository.
 
660
 
 
661
        :param format: Override for the controldir format to create.
 
662
        :return: The WorkingTree object.
 
663
        """
 
664
        t = _mod_transport.get_transport(base)
 
665
        if not isinstance(t, local.LocalTransport):
 
666
            raise errors.NotLocalUrl(base)
 
667
        controldir = klass.create_branch_and_repo(base,
 
668
                                               force_new_repo=True,
 
669
                                               format=format).bzrdir
 
670
        return controldir.create_workingtree()
 
671
 
 
672
    @classmethod
 
673
    def open_unsupported(klass, base):
 
674
        """Open a branch which is not supported."""
 
675
        return klass.open(base, _unsupported=True)
 
676
 
 
677
    @classmethod
 
678
    def open(klass, base, possible_transports=None, probers=None,
 
679
             _unsupported=False):
 
680
        """Open an existing controldir, rooted at 'base' (url).
 
681
 
 
682
        :param _unsupported: a private parameter to the ControlDir class.
 
683
        """
 
684
        t = _mod_transport.get_transport(base, possible_transports)
 
685
        return klass.open_from_transport(t, probers=probers,
 
686
                _unsupported=_unsupported)
 
687
 
 
688
    @classmethod
 
689
    def open_from_transport(klass, transport, _unsupported=False,
 
690
                            probers=None):
 
691
        """Open a controldir within a particular directory.
 
692
 
 
693
        :param transport: Transport containing the controldir.
 
694
        :param _unsupported: private.
 
695
        """
 
696
        for hook in klass.hooks['pre_open']:
 
697
            hook(transport)
 
698
        # Keep initial base since 'transport' may be modified while following
 
699
        # the redirections.
 
700
        base = transport.base
 
701
        def find_format(transport):
 
702
            return transport, ControlDirFormat.find_format(transport,
 
703
                probers=probers)
 
704
 
 
705
        def redirected(transport, e, redirection_notice):
 
706
            redirected_transport = transport._redirected_to(e.source, e.target)
 
707
            if redirected_transport is None:
 
708
                raise errors.NotBranchError(base)
 
709
            trace.note(gettext('{0} is{1} redirected to {2}').format(
 
710
                 transport.base, e.permanently, redirected_transport.base))
 
711
            return redirected_transport
 
712
 
 
713
        try:
 
714
            transport, format = _mod_transport.do_catching_redirections(
 
715
                find_format, transport, redirected)
 
716
        except errors.TooManyRedirections:
 
717
            raise errors.NotBranchError(base)
 
718
 
 
719
        format.check_support_status(_unsupported)
 
720
        return format.open(transport, _found=True)
 
721
 
 
722
    @classmethod
 
723
    def open_containing(klass, url, possible_transports=None):
 
724
        """Open an existing branch which contains url.
 
725
 
 
726
        :param url: url to search from.
 
727
 
 
728
        See open_containing_from_transport for more detail.
 
729
        """
 
730
        transport = _mod_transport.get_transport(url, possible_transports)
 
731
        return klass.open_containing_from_transport(transport)
 
732
 
 
733
    @classmethod
 
734
    def open_containing_from_transport(klass, a_transport):
 
735
        """Open an existing branch which contains a_transport.base.
 
736
 
 
737
        This probes for a branch at a_transport, and searches upwards from there.
 
738
 
 
739
        Basically we keep looking up until we find the control directory or
 
740
        run into the root.  If there isn't one, raises NotBranchError.
 
741
        If there is one and it is either an unrecognised format or an unsupported
 
742
        format, UnknownFormatError or UnsupportedFormatError are raised.
 
743
        If there is one, it is returned, along with the unused portion of url.
 
744
 
 
745
        :return: The ControlDir that contains the path, and a Unicode path
 
746
                for the rest of the URL.
 
747
        """
 
748
        # this gets the normalised url back. I.e. '.' -> the full path.
 
749
        url = a_transport.base
 
750
        while True:
 
751
            try:
 
752
                result = klass.open_from_transport(a_transport)
 
753
                return result, urlutils.unescape(a_transport.relpath(url))
 
754
            except errors.NotBranchError, e:
 
755
                pass
 
756
            except errors.PermissionDenied:
 
757
                pass
 
758
            try:
 
759
                new_t = a_transport.clone('..')
 
760
            except errors.InvalidURLJoin:
 
761
                # reached the root, whatever that may be
 
762
                raise errors.NotBranchError(path=url)
 
763
            if new_t.base == a_transport.base:
 
764
                # reached the root, whatever that may be
 
765
                raise errors.NotBranchError(path=url)
 
766
            a_transport = new_t
 
767
 
 
768
    @classmethod
 
769
    def open_tree_or_branch(klass, location):
 
770
        """Return the branch and working tree at a location.
 
771
 
 
772
        If there is no tree at the location, tree will be None.
 
773
        If there is no branch at the location, an exception will be
 
774
        raised
 
775
        :return: (tree, branch)
 
776
        """
 
777
        controldir = klass.open(location)
 
778
        return controldir._get_tree_branch()
 
779
 
 
780
    @classmethod
 
781
    def open_containing_tree_or_branch(klass, location):
 
782
        """Return the branch and working tree contained by a location.
 
783
 
 
784
        Returns (tree, branch, relpath).
 
785
        If there is no tree at containing the location, tree will be None.
 
786
        If there is no branch containing the location, an exception will be
 
787
        raised
 
788
        relpath is the portion of the path that is contained by the branch.
 
789
        """
 
790
        controldir, relpath = klass.open_containing(location)
 
791
        tree, branch = controldir._get_tree_branch()
 
792
        return tree, branch, relpath
 
793
 
 
794
    @classmethod
 
795
    def open_containing_tree_branch_or_repository(klass, location):
 
796
        """Return the working tree, branch and repo contained by a location.
 
797
 
 
798
        Returns (tree, branch, repository, relpath).
 
799
        If there is no tree containing the location, tree will be None.
 
800
        If there is no branch containing the location, branch will be None.
 
801
        If there is no repository containing the location, repository will be
 
802
        None.
 
803
        relpath is the portion of the path that is contained by the innermost
 
804
        ControlDir.
 
805
 
 
806
        If no tree, branch or repository is found, a NotBranchError is raised.
 
807
        """
 
808
        controldir, relpath = klass.open_containing(location)
 
809
        try:
 
810
            tree, branch = controldir._get_tree_branch()
 
811
        except errors.NotBranchError:
 
812
            try:
 
813
                repo = controldir.find_repository()
 
814
                return None, None, repo, relpath
 
815
            except (errors.NoRepositoryPresent):
 
816
                raise errors.NotBranchError(location)
 
817
        return tree, branch, branch.repository, relpath
 
818
 
 
819
    @classmethod
 
820
    def create(klass, base, format=None, possible_transports=None):
 
821
        """Create a new ControlDir at the url 'base'.
 
822
 
 
823
        :param format: If supplied, the format of branch to create.  If not
 
824
            supplied, the default is used.
 
825
        :param possible_transports: If supplied, a list of transports that
 
826
            can be reused to share a remote connection.
 
827
        """
 
828
        if klass is not ControlDir:
 
829
            raise AssertionError("ControlDir.create always creates the"
 
830
                "default format, not one of %r" % klass)
 
831
        t = _mod_transport.get_transport(base, possible_transports)
 
832
        t.ensure_base()
 
833
        if format is None:
 
834
            format = ControlDirFormat.get_default_format()
 
835
        return format.initialize_on_transport(t)
 
836
 
 
837
 
 
838
class ControlDirHooks(hooks.Hooks):
 
839
    """Hooks for ControlDir operations."""
 
840
 
 
841
    def __init__(self):
 
842
        """Create the default hooks."""
 
843
        hooks.Hooks.__init__(self, "bzrlib.controldir", "ControlDir.hooks")
 
844
        self.add_hook('pre_open',
 
845
            "Invoked before attempting to open a ControlDir with the transport "
 
846
            "that the open will use.", (1, 14))
 
847
        self.add_hook('post_repo_init',
 
848
            "Invoked after a repository has been initialized. "
 
849
            "post_repo_init is called with a "
 
850
            "bzrlib.controldir.RepoInitHookParams.",
 
851
            (2, 2))
 
852
 
 
853
# install the default hooks
 
854
ControlDir.hooks = ControlDirHooks()
 
855
 
 
856
 
 
857
class ControlComponentFormat(object):
 
858
    """A component that can live inside of a control directory."""
 
859
 
 
860
    upgrade_recommended = False
 
861
 
 
862
    def get_format_description(self):
 
863
        """Return the short description for this format."""
 
864
        raise NotImplementedError(self.get_format_description)
 
865
 
 
866
    def is_supported(self):
 
867
        """Is this format supported?
 
868
 
 
869
        Supported formats must be initializable and openable.
 
870
        Unsupported formats may not support initialization or committing or
 
871
        some other features depending on the reason for not being supported.
 
872
        """
 
873
        return True
 
874
 
 
875
    def check_support_status(self, allow_unsupported, recommend_upgrade=True,
 
876
        basedir=None):
 
877
        """Give an error or warning on old formats.
 
878
 
 
879
        :param allow_unsupported: If true, allow opening
 
880
            formats that are strongly deprecated, and which may
 
881
            have limited functionality.
 
882
 
 
883
        :param recommend_upgrade: If true (default), warn
 
884
            the user through the ui object that they may wish
 
885
            to upgrade the object.
 
886
        """
 
887
        if not allow_unsupported and not self.is_supported():
 
888
            # see open_downlevel to open legacy branches.
 
889
            raise errors.UnsupportedFormatError(format=self)
 
890
        if recommend_upgrade and self.upgrade_recommended:
 
891
            ui.ui_factory.recommend_upgrade(
 
892
                self.get_format_description(), basedir)
 
893
 
 
894
    @classmethod
 
895
    def get_format_string(cls):
 
896
        raise NotImplementedError(cls.get_format_string)
 
897
 
 
898
 
 
899
class ControlComponentFormatRegistry(registry.FormatRegistry):
 
900
    """A registry for control components (branch, workingtree, repository)."""
 
901
 
 
902
    def __init__(self, other_registry=None):
 
903
        super(ControlComponentFormatRegistry, self).__init__(other_registry)
 
904
        self._extra_formats = []
 
905
 
 
906
    def register(self, format):
 
907
        """Register a new format."""
 
908
        super(ControlComponentFormatRegistry, self).register(
 
909
            format.get_format_string(), format)
 
910
 
 
911
    def remove(self, format):
 
912
        """Remove a registered format."""
 
913
        super(ControlComponentFormatRegistry, self).remove(
 
914
            format.get_format_string())
 
915
 
 
916
    def register_extra(self, format):
 
917
        """Register a format that can not be used in a metadir.
 
918
 
 
919
        This is mainly useful to allow custom repository formats, such as older
 
920
        Bazaar formats and foreign formats, to be tested.
 
921
        """
 
922
        self._extra_formats.append(registry._ObjectGetter(format))
 
923
 
 
924
    def remove_extra(self, format):
 
925
        """Remove an extra format.
 
926
        """
 
927
        self._extra_formats.remove(registry._ObjectGetter(format))
 
928
 
 
929
    def register_extra_lazy(self, module_name, member_name):
 
930
        """Register a format lazily.
 
931
        """
 
932
        self._extra_formats.append(
 
933
            registry._LazyObjectGetter(module_name, member_name))
 
934
 
 
935
    def _get_extra(self):
 
936
        """Return all "extra" formats, not usable in meta directories."""
 
937
        result = []
 
938
        for getter in self._extra_formats:
 
939
            f = getter.get_obj()
 
940
            if callable(f):
 
941
                f = f()
 
942
            result.append(f)
 
943
        return result
 
944
 
 
945
    def _get_all(self):
 
946
        """Return all formats, even those not usable in metadirs.
 
947
        """
 
948
        result = []
 
949
        for name in self.keys():
 
950
            fmt = self.get(name)
 
951
            if callable(fmt):
 
952
                fmt = fmt()
 
953
            result.append(fmt)
 
954
        return result + self._get_extra()
 
955
 
 
956
    def _get_all_modules(self):
 
957
        """Return a set of the modules providing objects."""
 
958
        modules = set()
 
959
        for name in self.keys():
 
960
            modules.add(self._get_module(name))
 
961
        for getter in self._extra_formats:
 
962
            modules.add(getter.get_module())
 
963
        return modules
 
964
 
 
965
 
 
966
class Converter(object):
 
967
    """Converts a disk format object from one format to another."""
 
968
 
 
969
    def convert(self, to_convert, pb):
 
970
        """Perform the conversion of to_convert, giving feedback via pb.
 
971
 
 
972
        :param to_convert: The disk object to convert.
 
973
        :param pb: a progress bar to use for progress information.
 
974
        """
 
975
 
 
976
    def step(self, message):
 
977
        """Update the pb by a step."""
 
978
        self.count +=1
 
979
        self.pb.update(message, self.count, self.total)
 
980
 
 
981
 
 
982
class ControlDirFormat(object):
 
983
    """An encapsulation of the initialization and open routines for a format.
 
984
 
 
985
    Formats provide three things:
 
986
     * An initialization routine,
 
987
     * a format string,
 
988
     * an open routine.
 
989
 
 
990
    Formats are placed in a dict by their format string for reference
 
991
    during controldir opening. These should be subclasses of ControlDirFormat
 
992
    for consistency.
 
993
 
 
994
    Once a format is deprecated, just deprecate the initialize and open
 
995
    methods on the format class. Do not deprecate the object, as the
 
996
    object will be created every system load.
 
997
 
 
998
    :cvar colocated_branches: Whether this formats supports colocated branches.
 
999
    :cvar supports_workingtrees: This control directory can co-exist with a
 
1000
        working tree.
 
1001
    """
 
1002
 
 
1003
    _default_format = None
 
1004
    """The default format used for new control directories."""
 
1005
 
 
1006
    _server_probers = []
 
1007
    """The registered server format probers, e.g. RemoteBzrProber.
 
1008
 
 
1009
    This is a list of Prober-derived classes.
 
1010
    """
 
1011
 
 
1012
    _probers = []
 
1013
    """The registered format probers, e.g. BzrProber.
 
1014
 
 
1015
    This is a list of Prober-derived classes.
 
1016
    """
 
1017
 
 
1018
    colocated_branches = False
 
1019
    """Whether co-located branches are supported for this control dir format.
 
1020
    """
 
1021
 
 
1022
    supports_workingtrees = True
 
1023
    """Whether working trees can exist in control directories of this format.
 
1024
    """
 
1025
 
 
1026
    fixed_components = False
 
1027
    """Whether components can not change format independent of the control dir.
 
1028
    """
 
1029
 
 
1030
    upgrade_recommended = False
 
1031
    """Whether an upgrade from this format is recommended."""
 
1032
 
 
1033
    def get_format_description(self):
 
1034
        """Return the short description for this format."""
 
1035
        raise NotImplementedError(self.get_format_description)
 
1036
 
 
1037
    def get_converter(self, format=None):
 
1038
        """Return the converter to use to convert controldirs needing converts.
 
1039
 
 
1040
        This returns a bzrlib.controldir.Converter object.
 
1041
 
 
1042
        This should return the best upgrader to step this format towards the
 
1043
        current default format. In the case of plugins we can/should provide
 
1044
        some means for them to extend the range of returnable converters.
 
1045
 
 
1046
        :param format: Optional format to override the default format of the
 
1047
                       library.
 
1048
        """
 
1049
        raise NotImplementedError(self.get_converter)
 
1050
 
 
1051
    def is_supported(self):
 
1052
        """Is this format supported?
 
1053
 
 
1054
        Supported formats must be openable.
 
1055
        Unsupported formats may not support initialization or committing or
 
1056
        some other features depending on the reason for not being supported.
 
1057
        """
 
1058
        return True
 
1059
 
 
1060
    def is_initializable(self):
 
1061
        """Whether new control directories of this format can be initialized.
 
1062
        """
 
1063
        return self.is_supported()
 
1064
 
 
1065
    def check_support_status(self, allow_unsupported, recommend_upgrade=True,
 
1066
        basedir=None):
 
1067
        """Give an error or warning on old formats.
 
1068
 
 
1069
        :param allow_unsupported: If true, allow opening
 
1070
            formats that are strongly deprecated, and which may
 
1071
            have limited functionality.
 
1072
 
 
1073
        :param recommend_upgrade: If true (default), warn
 
1074
            the user through the ui object that they may wish
 
1075
            to upgrade the object.
 
1076
        """
 
1077
        if not allow_unsupported and not self.is_supported():
 
1078
            # see open_downlevel to open legacy branches.
 
1079
            raise errors.UnsupportedFormatError(format=self)
 
1080
        if recommend_upgrade and self.upgrade_recommended:
 
1081
            ui.ui_factory.recommend_upgrade(
 
1082
                self.get_format_description(), basedir)
 
1083
 
 
1084
    def same_model(self, target_format):
 
1085
        return (self.repository_format.rich_root_data ==
 
1086
            target_format.rich_root_data)
 
1087
 
 
1088
    @classmethod
 
1089
    def register_format(klass, format):
 
1090
        """Register a format that does not use '.bzr' for its control dir.
 
1091
 
 
1092
        """
 
1093
        raise errors.BzrError("ControlDirFormat.register_format() has been "
 
1094
            "removed in Bazaar 2.4. Please upgrade your plugins.")
 
1095
 
 
1096
    @classmethod
 
1097
    def register_prober(klass, prober):
 
1098
        """Register a prober that can look for a control dir.
 
1099
 
 
1100
        """
 
1101
        klass._probers.append(prober)
 
1102
 
 
1103
    @classmethod
 
1104
    def unregister_prober(klass, prober):
 
1105
        """Unregister a prober.
 
1106
 
 
1107
        """
 
1108
        klass._probers.remove(prober)
 
1109
 
 
1110
    @classmethod
 
1111
    def register_server_prober(klass, prober):
 
1112
        """Register a control format prober for client-server environments.
 
1113
 
 
1114
        These probers will be used before ones registered with
 
1115
        register_prober.  This gives implementations that decide to the
 
1116
        chance to grab it before anything looks at the contents of the format
 
1117
        file.
 
1118
        """
 
1119
        klass._server_probers.append(prober)
 
1120
 
 
1121
    def __str__(self):
 
1122
        # Trim the newline
 
1123
        return self.get_format_description().rstrip()
 
1124
 
 
1125
    @classmethod
 
1126
    def all_probers(klass):
 
1127
        return klass._server_probers + klass._probers
 
1128
 
 
1129
    @classmethod
 
1130
    def known_formats(klass):
 
1131
        """Return all the known formats.
 
1132
        """
 
1133
        result = set()
 
1134
        for prober_kls in klass.all_probers():
 
1135
            result.update(prober_kls.known_formats())
 
1136
        return result
 
1137
 
 
1138
    @classmethod
 
1139
    def find_format(klass, transport, probers=None):
 
1140
        """Return the format present at transport."""
 
1141
        if probers is None:
 
1142
            probers = klass.all_probers()
 
1143
        for prober_kls in probers:
 
1144
            prober = prober_kls()
 
1145
            try:
 
1146
                return prober.probe_transport(transport)
 
1147
            except errors.NotBranchError:
 
1148
                # this format does not find a control dir here.
 
1149
                pass
 
1150
        raise errors.NotBranchError(path=transport.base)
 
1151
 
 
1152
    def initialize(self, url, possible_transports=None):
 
1153
        """Create a control dir at this url and return an opened copy.
 
1154
 
 
1155
        While not deprecated, this method is very specific and its use will
 
1156
        lead to many round trips to setup a working environment. See
 
1157
        initialize_on_transport_ex for a [nearly] all-in-one method.
 
1158
 
 
1159
        Subclasses should typically override initialize_on_transport
 
1160
        instead of this method.
 
1161
        """
 
1162
        return self.initialize_on_transport(
 
1163
            _mod_transport.get_transport(url, possible_transports))
 
1164
 
 
1165
    def initialize_on_transport(self, transport):
 
1166
        """Initialize a new controldir in the base directory of a Transport."""
 
1167
        raise NotImplementedError(self.initialize_on_transport)
 
1168
 
 
1169
    def initialize_on_transport_ex(self, transport, use_existing_dir=False,
 
1170
        create_prefix=False, force_new_repo=False, stacked_on=None,
 
1171
        stack_on_pwd=None, repo_format_name=None, make_working_trees=None,
 
1172
        shared_repo=False, vfs_only=False):
 
1173
        """Create this format on transport.
 
1174
 
 
1175
        The directory to initialize will be created.
 
1176
 
 
1177
        :param force_new_repo: Do not use a shared repository for the target,
 
1178
                               even if one is available.
 
1179
        :param create_prefix: Create any missing directories leading up to
 
1180
            to_transport.
 
1181
        :param use_existing_dir: Use an existing directory if one exists.
 
1182
        :param stacked_on: A url to stack any created branch on, None to follow
 
1183
            any target stacking policy.
 
1184
        :param stack_on_pwd: If stack_on is relative, the location it is
 
1185
            relative to.
 
1186
        :param repo_format_name: If non-None, a repository will be
 
1187
            made-or-found. Should none be found, or if force_new_repo is True
 
1188
            the repo_format_name is used to select the format of repository to
 
1189
            create.
 
1190
        :param make_working_trees: Control the setting of make_working_trees
 
1191
            for a new shared repository when one is made. None to use whatever
 
1192
            default the format has.
 
1193
        :param shared_repo: Control whether made repositories are shared or
 
1194
            not.
 
1195
        :param vfs_only: If True do not attempt to use a smart server
 
1196
        :return: repo, controldir, require_stacking, repository_policy. repo is
 
1197
            None if none was created or found, controldir is always valid.
 
1198
            require_stacking is the result of examining the stacked_on
 
1199
            parameter and any stacking policy found for the target.
 
1200
        """
 
1201
        raise NotImplementedError(self.initialize_on_transport_ex)
 
1202
 
 
1203
    def network_name(self):
 
1204
        """A simple byte string uniquely identifying this format for RPC calls.
 
1205
 
 
1206
        Bzr control formats use this disk format string to identify the format
 
1207
        over the wire. Its possible that other control formats have more
 
1208
        complex detection requirements, so we permit them to use any unique and
 
1209
        immutable string they desire.
 
1210
        """
 
1211
        raise NotImplementedError(self.network_name)
 
1212
 
 
1213
    def open(self, transport, _found=False):
 
1214
        """Return an instance of this format for the dir transport points at.
 
1215
        """
 
1216
        raise NotImplementedError(self.open)
 
1217
 
 
1218
    @classmethod
 
1219
    def _set_default_format(klass, format):
 
1220
        """Set default format (for testing behavior of defaults only)"""
 
1221
        klass._default_format = format
 
1222
 
 
1223
    @classmethod
 
1224
    def get_default_format(klass):
 
1225
        """Return the current default format."""
 
1226
        return klass._default_format
 
1227
 
 
1228
    def supports_transport(self, transport):
 
1229
        """Check if this format can be opened over a particular transport.
 
1230
        """
 
1231
        raise NotImplementedError(self.supports_transport)
 
1232
 
 
1233
 
 
1234
class Prober(object):
 
1235
    """Abstract class that can be used to detect a particular kind of
 
1236
    control directory.
 
1237
 
 
1238
    At the moment this just contains a single method to probe a particular
 
1239
    transport, but it may be extended in the future to e.g. avoid
 
1240
    multiple levels of probing for Subversion repositories.
 
1241
 
 
1242
    See BzrProber and RemoteBzrProber in bzrlib.bzrdir for the
 
1243
    probers that detect .bzr/ directories and Bazaar smart servers,
 
1244
    respectively.
 
1245
 
 
1246
    Probers should be registered using the register_server_prober or
 
1247
    register_prober methods on ControlDirFormat.
 
1248
    """
 
1249
 
 
1250
    def probe_transport(self, transport):
 
1251
        """Return the controldir style format present in a directory.
 
1252
 
 
1253
        :raise UnknownFormatError: If a control dir was found but is
 
1254
            in an unknown format.
 
1255
        :raise NotBranchError: If no control directory was found.
 
1256
        :return: A ControlDirFormat instance.
 
1257
        """
 
1258
        raise NotImplementedError(self.probe_transport)
 
1259
 
 
1260
    @classmethod
 
1261
    def known_formats(klass):
 
1262
        """Return the control dir formats known by this prober.
 
1263
 
 
1264
        Multiple probers can return the same formats, so this should
 
1265
        return a set.
 
1266
 
 
1267
        :return: A set of known formats.
 
1268
        """
 
1269
        raise NotImplementedError(klass.known_formats)
 
1270
 
 
1271
 
 
1272
class ControlDirFormatInfo(object):
 
1273
 
 
1274
    def __init__(self, native, deprecated, hidden, experimental):
 
1275
        self.deprecated = deprecated
 
1276
        self.native = native
 
1277
        self.hidden = hidden
 
1278
        self.experimental = experimental
 
1279
 
 
1280
 
 
1281
class ControlDirFormatRegistry(registry.Registry):
 
1282
    """Registry of user-selectable ControlDir subformats.
 
1283
 
 
1284
    Differs from ControlDirFormat._formats in that it provides sub-formats,
 
1285
    e.g. BzrDirMeta1 with weave repository.  Also, it's more user-oriented.
 
1286
    """
 
1287
 
 
1288
    def __init__(self):
 
1289
        """Create a ControlDirFormatRegistry."""
 
1290
        self._aliases = set()
 
1291
        self._registration_order = list()
 
1292
        super(ControlDirFormatRegistry, self).__init__()
 
1293
 
 
1294
    def aliases(self):
 
1295
        """Return a set of the format names which are aliases."""
 
1296
        return frozenset(self._aliases)
 
1297
 
 
1298
    def register(self, key, factory, help, native=True, deprecated=False,
 
1299
                 hidden=False, experimental=False, alias=False):
 
1300
        """Register a ControlDirFormat factory.
 
1301
 
 
1302
        The factory must be a callable that takes one parameter: the key.
 
1303
        It must produce an instance of the ControlDirFormat when called.
 
1304
 
 
1305
        This function mainly exists to prevent the info object from being
 
1306
        supplied directly.
 
1307
        """
 
1308
        registry.Registry.register(self, key, factory, help,
 
1309
            ControlDirFormatInfo(native, deprecated, hidden, experimental))
 
1310
        if alias:
 
1311
            self._aliases.add(key)
 
1312
        self._registration_order.append(key)
 
1313
 
 
1314
    def register_lazy(self, key, module_name, member_name, help, native=True,
 
1315
        deprecated=False, hidden=False, experimental=False, alias=False):
 
1316
        registry.Registry.register_lazy(self, key, module_name, member_name,
 
1317
            help, ControlDirFormatInfo(native, deprecated, hidden, experimental))
 
1318
        if alias:
 
1319
            self._aliases.add(key)
 
1320
        self._registration_order.append(key)
 
1321
 
 
1322
    def set_default(self, key):
 
1323
        """Set the 'default' key to be a clone of the supplied key.
 
1324
 
 
1325
        This method must be called once and only once.
 
1326
        """
 
1327
        registry.Registry.register(self, 'default', self.get(key),
 
1328
            self.get_help(key), info=self.get_info(key))
 
1329
        self._aliases.add('default')
 
1330
 
 
1331
    def set_default_repository(self, key):
 
1332
        """Set the FormatRegistry default and Repository default.
 
1333
 
 
1334
        This is a transitional method while Repository.set_default_format
 
1335
        is deprecated.
 
1336
        """
 
1337
        if 'default' in self:
 
1338
            self.remove('default')
 
1339
        self.set_default(key)
 
1340
        format = self.get('default')()
 
1341
 
 
1342
    def make_bzrdir(self, key):
 
1343
        return self.get(key)()
 
1344
 
 
1345
    def help_topic(self, topic):
 
1346
        output = ""
 
1347
        default_realkey = None
 
1348
        default_help = self.get_help('default')
 
1349
        help_pairs = []
 
1350
        for key in self._registration_order:
 
1351
            if key == 'default':
 
1352
                continue
 
1353
            help = self.get_help(key)
 
1354
            if help == default_help:
 
1355
                default_realkey = key
 
1356
            else:
 
1357
                help_pairs.append((key, help))
 
1358
 
 
1359
        def wrapped(key, help, info):
 
1360
            if info.native:
 
1361
                help = '(native) ' + help
 
1362
            return ':%s:\n%s\n\n' % (key,
 
1363
                textwrap.fill(help, initial_indent='    ',
 
1364
                    subsequent_indent='    ',
 
1365
                    break_long_words=False))
 
1366
        if default_realkey is not None:
 
1367
            output += wrapped(default_realkey, '(default) %s' % default_help,
 
1368
                              self.get_info('default'))
 
1369
        deprecated_pairs = []
 
1370
        experimental_pairs = []
 
1371
        for key, help in help_pairs:
 
1372
            info = self.get_info(key)
 
1373
            if info.hidden:
 
1374
                continue
 
1375
            elif info.deprecated:
 
1376
                deprecated_pairs.append((key, help))
 
1377
            elif info.experimental:
 
1378
                experimental_pairs.append((key, help))
 
1379
            else:
 
1380
                output += wrapped(key, help, info)
 
1381
        output += "\nSee :doc:`formats-help` for more about storage formats."
 
1382
        other_output = ""
 
1383
        if len(experimental_pairs) > 0:
 
1384
            other_output += "Experimental formats are shown below.\n\n"
 
1385
            for key, help in experimental_pairs:
 
1386
                info = self.get_info(key)
 
1387
                other_output += wrapped(key, help, info)
 
1388
        else:
 
1389
            other_output += \
 
1390
                "No experimental formats are available.\n\n"
 
1391
        if len(deprecated_pairs) > 0:
 
1392
            other_output += "\nDeprecated formats are shown below.\n\n"
 
1393
            for key, help in deprecated_pairs:
 
1394
                info = self.get_info(key)
 
1395
                other_output += wrapped(key, help, info)
 
1396
        else:
 
1397
            other_output += \
 
1398
                "\nNo deprecated formats are available.\n\n"
 
1399
        other_output += \
 
1400
                "\nSee :doc:`formats-help` for more about storage formats."
 
1401
 
 
1402
        if topic == 'other-formats':
 
1403
            return other_output
 
1404
        else:
 
1405
            return output
 
1406
 
 
1407
 
 
1408
class RepoInitHookParams(object):
 
1409
    """Object holding parameters passed to `*_repo_init` hooks.
 
1410
 
 
1411
    There are 4 fields that hooks may wish to access:
 
1412
 
 
1413
    :ivar repository: Repository created
 
1414
    :ivar format: Repository format
 
1415
    :ivar bzrdir: The controldir for the repository
 
1416
    :ivar shared: The repository is shared
 
1417
    """
 
1418
 
 
1419
    def __init__(self, repository, format, controldir, shared):
 
1420
        """Create a group of RepoInitHook parameters.
 
1421
 
 
1422
        :param repository: Repository created
 
1423
        :param format: Repository format
 
1424
        :param controldir: The controldir for the repository
 
1425
        :param shared: The repository is shared
 
1426
        """
 
1427
        self.repository = repository
 
1428
        self.format = format
 
1429
        self.bzrdir = controldir
 
1430
        self.shared = shared
 
1431
 
 
1432
    def __eq__(self, other):
 
1433
        return self.__dict__ == other.__dict__
 
1434
 
 
1435
    def __repr__(self):
 
1436
        if self.repository:
 
1437
            return "<%s for %s>" % (self.__class__.__name__,
 
1438
                self.repository)
 
1439
        else:
 
1440
            return "<%s for %s>" % (self.__class__.__name__,
 
1441
                self.bzrdir)
 
1442
 
 
1443
 
 
1444
# Please register new formats after old formats so that formats
 
1445
# appear in chronological order and format descriptions can build
 
1446
# on previous ones.
 
1447
format_registry = ControlDirFormatRegistry()
 
1448
 
 
1449
network_format_registry = registry.FormatRegistry()
 
1450
"""Registry of formats indexed by their network name.
 
1451
 
 
1452
The network name for a ControlDirFormat is an identifier that can be used when
 
1453
referring to formats with smart server operations. See
 
1454
ControlDirFormat.network_name() for more detail.
 
1455
"""