/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/controldir.py

  • Committer: Vincent Ladeuil
  • Date: 2011-02-11 16:15:39 UTC
  • mto: This revision was merged to the branch mainline in revision 5661.
  • Revision ID: v.ladeuil+lp@free.fr-20110211161539-26o5a28ihyemvuzg
Fix pqm failure on python2.4.

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 bzrlib.lazy_import import lazy_import
 
26
lazy_import(globals(), """
 
27
import textwrap
 
28
 
 
29
from bzrlib import (
 
30
    cleanup,
 
31
    errors,
 
32
    fetch,
 
33
    graph,
 
34
    revision as _mod_revision,
 
35
    transport as _mod_transport,
 
36
    urlutils,
 
37
    )
 
38
from bzrlib.push import (
 
39
    PushResult,
 
40
    )
 
41
from bzrlib.trace import (
 
42
    mutter,
 
43
    )
 
44
from bzrlib.transport import (
 
45
    local,
 
46
    )
 
47
 
 
48
""")
 
49
 
 
50
from bzrlib import registry
 
51
 
 
52
 
 
53
class ControlComponent(object):
 
54
    """Abstract base class for control directory components.
 
55
 
 
56
    This provides interfaces that are common across controldirs,
 
57
    repositories, branches, and workingtree control directories.
 
58
 
 
59
    They all expose two urls and transports: the *user* URL is the
 
60
    one that stops above the control directory (eg .bzr) and that
 
61
    should normally be used in messages, and the *control* URL is
 
62
    under that in eg .bzr/checkout and is used to read the control
 
63
    files.
 
64
 
 
65
    This can be used as a mixin and is intended to fit with
 
66
    foreign formats.
 
67
    """
 
68
 
 
69
    @property
 
70
    def control_transport(self):
 
71
        raise NotImplementedError
 
72
 
 
73
    @property
 
74
    def control_url(self):
 
75
        return self.control_transport.base
 
76
 
 
77
    @property
 
78
    def user_transport(self):
 
79
        raise NotImplementedError
 
80
 
 
81
    @property
 
82
    def user_url(self):
 
83
        return self.user_transport.base
 
84
 
 
85
 
 
86
class ControlDir(ControlComponent):
 
87
    """A control directory.
 
88
 
 
89
    While this represents a generic control directory, there are a few
 
90
    features that are present in this interface that are currently only
 
91
    supported by one of its implementations, BzrDir.
 
92
 
 
93
    These features (bound branches, stacked branches) are currently only
 
94
    supported by Bazaar, but could be supported by other version control
 
95
    systems as well. Implementations are required to raise the appropriate
 
96
    exceptions when an operation is requested that is not supported.
 
97
 
 
98
    This also makes life easier for API users who can rely on the
 
99
    implementation always allowing a particular feature to be requested but
 
100
    raising an exception when it is not supported, rather than requiring the
 
101
    API users to check for magic attributes to see what features are supported.
 
102
    """
 
103
 
 
104
    def can_convert_format(self):
 
105
        """Return true if this controldir is one whose format we can convert
 
106
        from."""
 
107
        return True
 
108
 
 
109
    def list_branches(self):
 
110
        """Return a sequence of all branches local to this control directory.
 
111
 
 
112
        """
 
113
        try:
 
114
            return [self.open_branch()]
 
115
        except (errors.NotBranchError, errors.NoRepositoryPresent):
 
116
            return []
 
117
 
 
118
    def is_control_filename(self, filename):
 
119
        """True if filename is the name of a path which is reserved for
 
120
        controldirs.
 
121
 
 
122
        :param filename: A filename within the root transport of this
 
123
            controldir.
 
124
 
 
125
        This is true IF and ONLY IF the filename is part of the namespace reserved
 
126
        for bzr control dirs. Currently this is the '.bzr' directory in the root
 
127
        of the root_transport. it is expected that plugins will need to extend
 
128
        this in the future - for instance to make bzr talk with svn working
 
129
        trees.
 
130
        """
 
131
        raise NotImplementedError(self.is_control_filename)
 
132
 
 
133
    def needs_format_conversion(self, format=None):
 
134
        """Return true if this controldir needs convert_format run on it.
 
135
 
 
136
        For instance, if the repository format is out of date but the
 
137
        branch and working tree are not, this should return True.
 
138
 
 
139
        :param format: Optional parameter indicating a specific desired
 
140
                       format we plan to arrive at.
 
141
        """
 
142
        raise NotImplementedError(self.needs_format_conversion)
 
143
 
 
144
    def destroy_repository(self):
 
145
        """Destroy the repository in this ControlDir."""
 
146
        raise NotImplementedError(self.destroy_repository)
 
147
 
 
148
    def create_branch(self, name=None, repository=None):
 
149
        """Create a branch in this ControlDir.
 
150
 
 
151
        :param name: Name of the colocated branch to create, None for
 
152
            the default branch.
 
153
 
 
154
        The controldirs format will control what branch format is created.
 
155
        For more control see BranchFormatXX.create(a_controldir).
 
156
        """
 
157
        raise NotImplementedError(self.create_branch)
 
158
 
 
159
    def destroy_branch(self, name=None):
 
160
        """Destroy a branch in this ControlDir.
 
161
 
 
162
        :param name: Name of the branch to destroy, None for the default 
 
163
            branch.
 
164
        """
 
165
        raise NotImplementedError(self.destroy_branch)
 
166
 
 
167
    def create_workingtree(self, revision_id=None, from_branch=None,
 
168
        accelerator_tree=None, hardlink=False):
 
169
        """Create a working tree at this ControlDir.
 
170
 
 
171
        :param revision_id: create it as of this revision id.
 
172
        :param from_branch: override controldir branch 
 
173
            (for lightweight checkouts)
 
174
        :param accelerator_tree: A tree which can be used for retrieving file
 
175
            contents more quickly than the revision tree, i.e. a workingtree.
 
176
            The revision tree will be used for cases where accelerator_tree's
 
177
            content is different.
 
178
        """
 
179
        raise NotImplementedError(self.create_workingtree)
 
180
 
 
181
    def destroy_workingtree(self):
 
182
        """Destroy the working tree at this ControlDir.
 
183
 
 
184
        Formats that do not support this may raise UnsupportedOperation.
 
185
        """
 
186
        raise NotImplementedError(self.destroy_workingtree)
 
187
 
 
188
    def destroy_workingtree_metadata(self):
 
189
        """Destroy the control files for the working tree at this ControlDir.
 
190
 
 
191
        The contents of working tree files are not affected.
 
192
        Formats that do not support this may raise UnsupportedOperation.
 
193
        """
 
194
        raise NotImplementedError(self.destroy_workingtree_metadata)
 
195
 
 
196
    def get_branch_reference(self, name=None):
 
197
        """Return the referenced URL for the branch in this controldir.
 
198
 
 
199
        :param name: Optional colocated branch name
 
200
        :raises NotBranchError: If there is no Branch.
 
201
        :raises NoColocatedBranchSupport: If a branch name was specified
 
202
            but colocated branches are not supported.
 
203
        :return: The URL the branch in this controldir references if it is a
 
204
            reference branch, or None for regular branches.
 
205
        """
 
206
        if name is not None:
 
207
            raise errors.NoColocatedBranchSupport(self)
 
208
        return None
 
209
 
 
210
    def get_branch_transport(self, branch_format, name=None):
 
211
        """Get the transport for use by branch format in this ControlDir.
 
212
 
 
213
        Note that bzr dirs that do not support format strings will raise
 
214
        IncompatibleFormat if the branch format they are given has
 
215
        a format string, and vice versa.
 
216
 
 
217
        If branch_format is None, the transport is returned with no
 
218
        checking. If it is not None, then the returned transport is
 
219
        guaranteed to point to an existing directory ready for use.
 
220
        """
 
221
        raise NotImplementedError(self.get_branch_transport)
 
222
 
 
223
    def get_repository_transport(self, repository_format):
 
224
        """Get the transport for use by repository format in this ControlDir.
 
225
 
 
226
        Note that bzr dirs that do not support format strings will raise
 
227
        IncompatibleFormat if the repository format they are given has
 
228
        a format string, and vice versa.
 
229
 
 
230
        If repository_format is None, the transport is returned with no
 
231
        checking. If it is not None, then the returned transport is
 
232
        guaranteed to point to an existing directory ready for use.
 
233
        """
 
234
        raise NotImplementedError(self.get_repository_transport)
 
235
 
 
236
    def get_workingtree_transport(self, tree_format):
 
237
        """Get the transport for use by workingtree format in this ControlDir.
 
238
 
 
239
        Note that bzr dirs that do not support format strings will raise
 
240
        IncompatibleFormat if the workingtree format they are given has a
 
241
        format string, and vice versa.
 
242
 
 
243
        If workingtree_format is None, the transport is returned with no
 
244
        checking. If it is not None, then the returned transport is
 
245
        guaranteed to point to an existing directory ready for use.
 
246
        """
 
247
        raise NotImplementedError(self.get_workingtree_transport)
 
248
 
 
249
    def open_branch(self, name=None, unsupported=False,
 
250
                    ignore_fallbacks=False):
 
251
        """Open the branch object at this ControlDir if one is present.
 
252
 
 
253
        If unsupported is True, then no longer supported branch formats can
 
254
        still be opened.
 
255
 
 
256
        TODO: static convenience version of this?
 
257
        """
 
258
        raise NotImplementedError(self.open_branch)
 
259
 
 
260
    def open_repository(self, _unsupported=False):
 
261
        """Open the repository object at this ControlDir if one is present.
 
262
 
 
263
        This will not follow the Branch object pointer - it's strictly a direct
 
264
        open facility. Most client code should use open_branch().repository to
 
265
        get at a repository.
 
266
 
 
267
        :param _unsupported: a private parameter, not part of the api.
 
268
        TODO: static convenience version of this?
 
269
        """
 
270
        raise NotImplementedError(self.open_repository)
 
271
 
 
272
    def find_repository(self):
 
273
        """Find the repository that should be used.
 
274
 
 
275
        This does not require a branch as we use it to find the repo for
 
276
        new branches as well as to hook existing branches up to their
 
277
        repository.
 
278
        """
 
279
        raise NotImplementedError(self.find_repository)
 
280
 
 
281
    def open_workingtree(self, _unsupported=False,
 
282
                         recommend_upgrade=True, from_branch=None):
 
283
        """Open the workingtree object at this ControlDir if one is present.
 
284
 
 
285
        :param recommend_upgrade: Optional keyword parameter, when True (the
 
286
            default), emit through the ui module a recommendation that the user
 
287
            upgrade the working tree when the workingtree being opened is old
 
288
            (but still fully supported).
 
289
        :param from_branch: override controldir branch (for lightweight
 
290
            checkouts)
 
291
        """
 
292
        raise NotImplementedError(self.open_workingtree)
 
293
 
 
294
    def has_branch(self, name=None):
 
295
        """Tell if this controldir contains a branch.
 
296
 
 
297
        Note: if you're going to open the branch, you should just go ahead
 
298
        and try, and not ask permission first.  (This method just opens the
 
299
        branch and discards it, and that's somewhat expensive.)
 
300
        """
 
301
        try:
 
302
            self.open_branch(name)
 
303
            return True
 
304
        except errors.NotBranchError:
 
305
            return False
 
306
 
 
307
    def has_workingtree(self):
 
308
        """Tell if this controldir contains a working tree.
 
309
 
 
310
        This will still raise an exception if the controldir has a workingtree
 
311
        that is remote & inaccessible.
 
312
 
 
313
        Note: if you're going to open the working tree, you should just go ahead
 
314
        and try, and not ask permission first.  (This method just opens the
 
315
        workingtree and discards it, and that's somewhat expensive.)
 
316
        """
 
317
        try:
 
318
            self.open_workingtree(recommend_upgrade=False)
 
319
            return True
 
320
        except errors.NoWorkingTree:
 
321
            return False
 
322
 
 
323
    def cloning_metadir(self, require_stacking=False):
 
324
        """Produce a metadir suitable for cloning or sprouting with.
 
325
 
 
326
        These operations may produce workingtrees (yes, even though they're
 
327
        "cloning" something that doesn't have a tree), so a viable workingtree
 
328
        format must be selected.
 
329
 
 
330
        :require_stacking: If True, non-stackable formats will be upgraded
 
331
            to similar stackable formats.
 
332
        :returns: a ControlDirFormat with all component formats either set
 
333
            appropriately or set to None if that component should not be
 
334
            created.
 
335
        """
 
336
        raise NotImplementedError(self.cloning_metadir)
 
337
 
 
338
    def checkout_metadir(self):
 
339
        """Produce a metadir suitable for checkouts of this controldir."""
 
340
        return self.cloning_metadir()
 
341
 
 
342
    def sprout(self, url, revision_id=None, force_new_repo=False,
 
343
               recurse='down', possible_transports=None,
 
344
               accelerator_tree=None, hardlink=False, stacked=False,
 
345
               source_branch=None, create_tree_if_local=True):
 
346
        """Create a copy of this controldir prepared for use as a new line of
 
347
        development.
 
348
 
 
349
        If url's last component does not exist, it will be created.
 
350
 
 
351
        Attributes related to the identity of the source branch like
 
352
        branch nickname will be cleaned, a working tree is created
 
353
        whether one existed before or not; and a local branch is always
 
354
        created.
 
355
 
 
356
        if revision_id is not None, then the clone operation may tune
 
357
            itself to download less data.
 
358
        :param accelerator_tree: A tree which can be used for retrieving file
 
359
            contents more quickly than the revision tree, i.e. a workingtree.
 
360
            The revision tree will be used for cases where accelerator_tree's
 
361
            content is different.
 
362
        :param hardlink: If true, hard-link files from accelerator_tree,
 
363
            where possible.
 
364
        :param stacked: If true, create a stacked branch referring to the
 
365
            location of this control directory.
 
366
        :param create_tree_if_local: If true, a working-tree will be created
 
367
            when working locally.
 
368
        """
 
369
        operation = cleanup.OperationWithCleanups(self._sprout)
 
370
        return operation.run(url, revision_id=revision_id,
 
371
            force_new_repo=force_new_repo, recurse=recurse,
 
372
            possible_transports=possible_transports,
 
373
            accelerator_tree=accelerator_tree, hardlink=hardlink,
 
374
            stacked=stacked, source_branch=source_branch,
 
375
            create_tree_if_local=create_tree_if_local)
 
376
 
 
377
    def _sprout(self, op, url, revision_id=None, force_new_repo=False,
 
378
               recurse='down', possible_transports=None,
 
379
               accelerator_tree=None, hardlink=False, stacked=False,
 
380
               source_branch=None, create_tree_if_local=True):
 
381
        add_cleanup = op.add_cleanup
 
382
        fetch_spec_factory = fetch.FetchSpecFactory()
 
383
        if revision_id is not None:
 
384
            fetch_spec_factory.add_revision_ids([revision_id])
 
385
            fetch_spec_factory.source_branch_stop_revision_id = revision_id
 
386
        target_transport = _mod_transport.get_transport(url,
 
387
            possible_transports)
 
388
        target_transport.ensure_base()
 
389
        cloning_format = self.cloning_metadir(stacked)
 
390
        # Create/update the result branch
 
391
        result = cloning_format.initialize_on_transport(target_transport)
 
392
        source_branch, source_repository = self._find_source_repo(
 
393
            add_cleanup, source_branch)
 
394
        fetch_spec_factory.source_branch = source_branch
 
395
        # if a stacked branch wasn't requested, we don't create one
 
396
        # even if the origin was stacked
 
397
        if stacked and source_branch is not None:
 
398
            stacked_branch_url = self.root_transport.base
 
399
        else:
 
400
            stacked_branch_url = None
 
401
        repository_policy = result.determine_repository_policy(
 
402
            force_new_repo, stacked_branch_url, require_stacking=stacked)
 
403
        result_repo, is_new_repo = repository_policy.acquire_repository()
 
404
        add_cleanup(result_repo.lock_write().unlock)
 
405
        fetch_spec_factory.source_repo = source_repository
 
406
        fetch_spec_factory.target_repo = result_repo
 
407
        if stacked or (len(result_repo._fallback_repositories) != 0):
 
408
            target_repo_kind = fetch.TargetRepoKinds.STACKED
 
409
        elif is_new_repo:
 
410
            target_repo_kind = fetch.TargetRepoKinds.EMPTY
 
411
        else:
 
412
            target_repo_kind = fetch.TargetRepoKinds.PREEXISTING
 
413
        fetch_spec_factory.target_repo_kind = target_repo_kind
 
414
        if source_repository is not None:
 
415
            fetch_spec = fetch_spec_factory.make_fetch_spec()
 
416
            result_repo.fetch(source_repository, fetch_spec=fetch_spec)
 
417
 
 
418
        if source_branch is None:
 
419
            # this is for sprouting a controldir without a branch; is that
 
420
            # actually useful?
 
421
            # Not especially, but it's part of the contract.
 
422
            result_branch = result.create_branch()
 
423
        else:
 
424
            result_branch = source_branch.sprout(result,
 
425
                revision_id=revision_id, repository_policy=repository_policy,
 
426
                repository=result_repo)
 
427
        mutter("created new branch %r" % (result_branch,))
 
428
 
 
429
        # Create/update the result working tree
 
430
        if (create_tree_if_local and
 
431
            isinstance(target_transport, local.LocalTransport) and
 
432
            (result_repo is None or result_repo.make_working_trees())):
 
433
            wt = result.create_workingtree(accelerator_tree=accelerator_tree,
 
434
                hardlink=hardlink, from_branch=result_branch)
 
435
            wt.lock_write()
 
436
            try:
 
437
                if wt.path2id('') is None:
 
438
                    try:
 
439
                        wt.set_root_id(self.open_workingtree.get_root_id())
 
440
                    except errors.NoWorkingTree:
 
441
                        pass
 
442
            finally:
 
443
                wt.unlock()
 
444
        else:
 
445
            wt = None
 
446
        if recurse == 'down':
 
447
            basis = None
 
448
            if wt is not None:
 
449
                basis = wt.basis_tree()
 
450
            elif result_branch is not None:
 
451
                basis = result_branch.basis_tree()
 
452
            elif source_branch is not None:
 
453
                basis = source_branch.basis_tree()
 
454
            if basis is not None:
 
455
                add_cleanup(basis.lock_read().unlock)
 
456
                subtrees = basis.iter_references()
 
457
            else:
 
458
                subtrees = []
 
459
            for path, file_id in subtrees:
 
460
                target = urlutils.join(url, urlutils.escape(path))
 
461
                sublocation = source_branch.reference_parent(file_id, path)
 
462
                sublocation.bzrdir.sprout(target,
 
463
                    basis.get_reference_revision(file_id, path),
 
464
                    force_new_repo=force_new_repo, recurse=recurse,
 
465
                    stacked=stacked)
 
466
        return result
 
467
 
 
468
    def _find_source_repo(self, add_cleanup, source_branch):
 
469
        """Find the source branch and repo for a sprout operation.
 
470
        
 
471
        This is helper intended for use by _sprout.
 
472
 
 
473
        :returns: (source_branch, source_repository).  Either or both may be
 
474
            None.  If not None, they will be read-locked (and their unlock(s)
 
475
            scheduled via the add_cleanup param).
 
476
        """
 
477
        if source_branch is not None:
 
478
            add_cleanup(source_branch.lock_read().unlock)
 
479
            return source_branch, source_branch.repository
 
480
        try:
 
481
            source_branch = self.open_branch()
 
482
            source_repository = source_branch.repository
 
483
        except errors.NotBranchError:
 
484
            source_branch = None
 
485
            try:
 
486
                source_repository = self.open_repository()
 
487
            except errors.NoRepositoryPresent:
 
488
                source_repository = None
 
489
            else:
 
490
                add_cleanup(source_repository.lock_read().unlock)
 
491
        else:
 
492
            add_cleanup(source_branch.lock_read().unlock)
 
493
        return source_branch, source_repository
 
494
 
 
495
    def push_branch(self, source, revision_id=None, overwrite=False, 
 
496
        remember=False, create_prefix=False):
 
497
        """Push the source branch into this ControlDir."""
 
498
        br_to = None
 
499
        # If we can open a branch, use its direct repository, otherwise see
 
500
        # if there is a repository without a branch.
 
501
        try:
 
502
            br_to = self.open_branch()
 
503
        except errors.NotBranchError:
 
504
            # Didn't find a branch, can we find a repository?
 
505
            repository_to = self.find_repository()
 
506
        else:
 
507
            # Found a branch, so we must have found a repository
 
508
            repository_to = br_to.repository
 
509
 
 
510
        push_result = PushResult()
 
511
        push_result.source_branch = source
 
512
        if br_to is None:
 
513
            # We have a repository but no branch, copy the revisions, and then
 
514
            # create a branch.
 
515
            repository_to.fetch(source.repository, revision_id=revision_id)
 
516
            br_to = source.clone(self, revision_id=revision_id)
 
517
            if source.get_push_location() is None or remember:
 
518
                source.set_push_location(br_to.base)
 
519
            push_result.stacked_on = None
 
520
            push_result.branch_push_result = None
 
521
            push_result.old_revno = None
 
522
            push_result.old_revid = _mod_revision.NULL_REVISION
 
523
            push_result.target_branch = br_to
 
524
            push_result.master_branch = None
 
525
            push_result.workingtree_updated = False
 
526
        else:
 
527
            # We have successfully opened the branch, remember if necessary:
 
528
            if source.get_push_location() is None or remember:
 
529
                source.set_push_location(br_to.base)
 
530
            try:
 
531
                tree_to = self.open_workingtree()
 
532
            except errors.NotLocalUrl:
 
533
                push_result.branch_push_result = source.push(br_to, 
 
534
                    overwrite, stop_revision=revision_id)
 
535
                push_result.workingtree_updated = False
 
536
            except errors.NoWorkingTree:
 
537
                push_result.branch_push_result = source.push(br_to,
 
538
                    overwrite, stop_revision=revision_id)
 
539
                push_result.workingtree_updated = None # Not applicable
 
540
            else:
 
541
                tree_to.lock_write()
 
542
                try:
 
543
                    push_result.branch_push_result = source.push(
 
544
                        tree_to.branch, overwrite, stop_revision=revision_id)
 
545
                    tree_to.update()
 
546
                finally:
 
547
                    tree_to.unlock()
 
548
                push_result.workingtree_updated = True
 
549
            push_result.old_revno = push_result.branch_push_result.old_revno
 
550
            push_result.old_revid = push_result.branch_push_result.old_revid
 
551
            push_result.target_branch = \
 
552
                push_result.branch_push_result.target_branch
 
553
        return push_result
 
554
 
 
555
    def _get_tree_branch(self, name=None):
 
556
        """Return the branch and tree, if any, for this bzrdir.
 
557
 
 
558
        :param name: Name of colocated branch to open.
 
559
 
 
560
        Return None for tree if not present or inaccessible.
 
561
        Raise NotBranchError if no branch is present.
 
562
        :return: (tree, branch)
 
563
        """
 
564
        try:
 
565
            tree = self.open_workingtree()
 
566
        except (errors.NoWorkingTree, errors.NotLocalUrl):
 
567
            tree = None
 
568
            branch = self.open_branch(name=name)
 
569
        else:
 
570
            if name is not None:
 
571
                branch = self.open_branch(name=name)
 
572
            else:
 
573
                branch = tree.branch
 
574
        return tree, branch
 
575
 
 
576
    def get_config(self):
 
577
        """Get configuration for this ControlDir."""
 
578
        raise NotImplementedError(self.get_config)
 
579
 
 
580
    def check_conversion_target(self, target_format):
 
581
        """Check that a bzrdir as a whole can be converted to a new format."""
 
582
        raise NotImplementedError(self.check_conversion_target)
 
583
 
 
584
    def clone(self, url, revision_id=None, force_new_repo=False,
 
585
              preserve_stacking=False):
 
586
        """Clone this bzrdir and its contents to url verbatim.
 
587
 
 
588
        :param url: The url create the clone at.  If url's last component does
 
589
            not exist, it will be created.
 
590
        :param revision_id: The tip revision-id to use for any branch or
 
591
            working tree.  If not None, then the clone operation may tune
 
592
            itself to download less data.
 
593
        :param force_new_repo: Do not use a shared repository for the target
 
594
                               even if one is available.
 
595
        :param preserve_stacking: When cloning a stacked branch, stack the
 
596
            new branch on top of the other branch's stacked-on branch.
 
597
        """
 
598
        return self.clone_on_transport(_mod_transport.get_transport(url),
 
599
                                       revision_id=revision_id,
 
600
                                       force_new_repo=force_new_repo,
 
601
                                       preserve_stacking=preserve_stacking)
 
602
 
 
603
    def clone_on_transport(self, transport, revision_id=None,
 
604
        force_new_repo=False, preserve_stacking=False, stacked_on=None,
 
605
        create_prefix=False, use_existing_dir=True):
 
606
        """Clone this bzrdir and its contents to transport verbatim.
 
607
 
 
608
        :param transport: The transport for the location to produce the clone
 
609
            at.  If the target directory does not exist, it will be created.
 
610
        :param revision_id: The tip revision-id to use for any branch or
 
611
            working tree.  If not None, then the clone operation may tune
 
612
            itself to download less data.
 
613
        :param force_new_repo: Do not use a shared repository for the target,
 
614
                               even if one is available.
 
615
        :param preserve_stacking: When cloning a stacked branch, stack the
 
616
            new branch on top of the other branch's stacked-on branch.
 
617
        :param create_prefix: Create any missing directories leading up to
 
618
            to_transport.
 
619
        :param use_existing_dir: Use an existing directory if one exists.
 
620
        """
 
621
        raise NotImplementedError(self.clone_on_transport)
 
622
 
 
623
 
 
624
class ControlDirFormat(object):
 
625
    """An encapsulation of the initialization and open routines for a format.
 
626
 
 
627
    Formats provide three things:
 
628
     * An initialization routine,
 
629
     * a format string,
 
630
     * an open routine.
 
631
 
 
632
    Formats are placed in a dict by their format string for reference
 
633
    during controldir opening. These should be subclasses of ControlDirFormat
 
634
    for consistency.
 
635
 
 
636
    Once a format is deprecated, just deprecate the initialize and open
 
637
    methods on the format class. Do not deprecate the object, as the
 
638
    object will be created every system load.
 
639
 
 
640
    :cvar colocated_branches: Whether this formats supports colocated branches.
 
641
    :cvar supports_workingtrees: This control directory can co-exist with a
 
642
        working tree.
 
643
    """
 
644
 
 
645
    _default_format = None
 
646
    """The default format used for new control directories."""
 
647
 
 
648
    _formats = []
 
649
    """The registered control formats - .bzr, ....
 
650
 
 
651
    This is a list of ControlDirFormat objects.
 
652
    """
 
653
 
 
654
    _server_probers = []
 
655
    """The registered server format probers, e.g. RemoteBzrProber.
 
656
 
 
657
    This is a list of Prober-derived classes.
 
658
    """
 
659
 
 
660
    _probers = []
 
661
    """The registered format probers, e.g. BzrProber.
 
662
 
 
663
    This is a list of Prober-derived classes.
 
664
    """
 
665
 
 
666
    colocated_branches = False
 
667
    """Whether co-located branches are supported for this control dir format.
 
668
    """
 
669
 
 
670
    supports_workingtrees = True
 
671
 
 
672
    def get_format_description(self):
 
673
        """Return the short description for this format."""
 
674
        raise NotImplementedError(self.get_format_description)
 
675
 
 
676
    def get_converter(self, format=None):
 
677
        """Return the converter to use to convert controldirs needing converts.
 
678
 
 
679
        This returns a bzrlib.controldir.Converter object.
 
680
 
 
681
        This should return the best upgrader to step this format towards the
 
682
        current default format. In the case of plugins we can/should provide
 
683
        some means for them to extend the range of returnable converters.
 
684
 
 
685
        :param format: Optional format to override the default format of the
 
686
                       library.
 
687
        """
 
688
        raise NotImplementedError(self.get_converter)
 
689
 
 
690
    def is_supported(self):
 
691
        """Is this format supported?
 
692
 
 
693
        Supported formats must be initializable and openable.
 
694
        Unsupported formats may not support initialization or committing or
 
695
        some other features depending on the reason for not being supported.
 
696
        """
 
697
        return True
 
698
 
 
699
    def same_model(self, target_format):
 
700
        return (self.repository_format.rich_root_data ==
 
701
            target_format.rich_root_data)
 
702
 
 
703
    @classmethod
 
704
    def register_format(klass, format):
 
705
        """Register a format that does not use '.bzr' for its control dir.
 
706
 
 
707
        """
 
708
        klass._formats.append(format)
 
709
 
 
710
    @classmethod
 
711
    def register_prober(klass, prober):
 
712
        """Register a prober that can look for a control dir.
 
713
 
 
714
        """
 
715
        klass._probers.append(prober)
 
716
 
 
717
    @classmethod
 
718
    def unregister_prober(klass, prober):
 
719
        """Unregister a prober.
 
720
 
 
721
        """
 
722
        klass._probers.remove(prober)
 
723
 
 
724
    @classmethod
 
725
    def register_server_prober(klass, prober):
 
726
        """Register a control format prober for client-server environments.
 
727
 
 
728
        These probers will be used before ones registered with
 
729
        register_prober.  This gives implementations that decide to the
 
730
        chance to grab it before anything looks at the contents of the format
 
731
        file.
 
732
        """
 
733
        klass._server_probers.append(prober)
 
734
 
 
735
    def __str__(self):
 
736
        # Trim the newline
 
737
        return self.get_format_description().rstrip()
 
738
 
 
739
    @classmethod
 
740
    def unregister_format(klass, format):
 
741
        klass._formats.remove(format)
 
742
 
 
743
    @classmethod
 
744
    def known_formats(klass):
 
745
        """Return all the known formats.
 
746
        """
 
747
        return set(klass._formats)
 
748
 
 
749
    @classmethod
 
750
    def find_format(klass, transport, _server_formats=True):
 
751
        """Return the format present at transport."""
 
752
        if _server_formats:
 
753
            _probers = klass._server_probers + klass._probers
 
754
        else:
 
755
            _probers = klass._probers
 
756
        for prober_kls in _probers:
 
757
            prober = prober_kls()
 
758
            try:
 
759
                return prober.probe_transport(transport)
 
760
            except errors.NotBranchError:
 
761
                # this format does not find a control dir here.
 
762
                pass
 
763
        raise errors.NotBranchError(path=transport.base)
 
764
 
 
765
    def initialize(self, url, possible_transports=None):
 
766
        """Create a control dir at this url and return an opened copy.
 
767
 
 
768
        While not deprecated, this method is very specific and its use will
 
769
        lead to many round trips to setup a working environment. See
 
770
        initialize_on_transport_ex for a [nearly] all-in-one method.
 
771
 
 
772
        Subclasses should typically override initialize_on_transport
 
773
        instead of this method.
 
774
        """
 
775
        return self.initialize_on_transport(
 
776
            _mod_transport.get_transport(url, possible_transports))
 
777
 
 
778
    def initialize_on_transport(self, transport):
 
779
        """Initialize a new controldir in the base directory of a Transport."""
 
780
        raise NotImplementedError(self.initialize_on_transport)
 
781
 
 
782
    def initialize_on_transport_ex(self, transport, use_existing_dir=False,
 
783
        create_prefix=False, force_new_repo=False, stacked_on=None,
 
784
        stack_on_pwd=None, repo_format_name=None, make_working_trees=None,
 
785
        shared_repo=False, vfs_only=False):
 
786
        """Create this format on transport.
 
787
 
 
788
        The directory to initialize will be created.
 
789
 
 
790
        :param force_new_repo: Do not use a shared repository for the target,
 
791
                               even if one is available.
 
792
        :param create_prefix: Create any missing directories leading up to
 
793
            to_transport.
 
794
        :param use_existing_dir: Use an existing directory if one exists.
 
795
        :param stacked_on: A url to stack any created branch on, None to follow
 
796
            any target stacking policy.
 
797
        :param stack_on_pwd: If stack_on is relative, the location it is
 
798
            relative to.
 
799
        :param repo_format_name: If non-None, a repository will be
 
800
            made-or-found. Should none be found, or if force_new_repo is True
 
801
            the repo_format_name is used to select the format of repository to
 
802
            create.
 
803
        :param make_working_trees: Control the setting of make_working_trees
 
804
            for a new shared repository when one is made. None to use whatever
 
805
            default the format has.
 
806
        :param shared_repo: Control whether made repositories are shared or
 
807
            not.
 
808
        :param vfs_only: If True do not attempt to use a smart server
 
809
        :return: repo, controldir, require_stacking, repository_policy. repo is
 
810
            None if none was created or found, controldir is always valid.
 
811
            require_stacking is the result of examining the stacked_on
 
812
            parameter and any stacking policy found for the target.
 
813
        """
 
814
        raise NotImplementedError(self.initialize_on_transport_ex)
 
815
 
 
816
    def network_name(self):
 
817
        """A simple byte string uniquely identifying this format for RPC calls.
 
818
 
 
819
        Bzr control formats use this disk format string to identify the format
 
820
        over the wire. Its possible that other control formats have more
 
821
        complex detection requirements, so we permit them to use any unique and
 
822
        immutable string they desire.
 
823
        """
 
824
        raise NotImplementedError(self.network_name)
 
825
 
 
826
    def open(self, transport, _found=False):
 
827
        """Return an instance of this format for the dir transport points at.
 
828
        """
 
829
        raise NotImplementedError(self.open)
 
830
 
 
831
    @classmethod
 
832
    def _set_default_format(klass, format):
 
833
        """Set default format (for testing behavior of defaults only)"""
 
834
        klass._default_format = format
 
835
 
 
836
    @classmethod
 
837
    def get_default_format(klass):
 
838
        """Return the current default format."""
 
839
        return klass._default_format
 
840
 
 
841
 
 
842
class Prober(object):
 
843
    """Abstract class that can be used to detect a particular kind of 
 
844
    control directory.
 
845
 
 
846
    At the moment this just contains a single method to probe a particular 
 
847
    transport, but it may be extended in the future to e.g. avoid 
 
848
    multiple levels of probing for Subversion repositories.
 
849
    """
 
850
 
 
851
    def probe_transport(self, transport):
 
852
        """Return the controldir style format present in a directory.
 
853
 
 
854
        :raise UnknownFormatError: If a control dir was found but is
 
855
            in an unknown format.
 
856
        :raise NotBranchError: If no control directory was found.
 
857
        :return: A ControlDirFormat instance.
 
858
        """
 
859
        raise NotImplementedError(self.probe_transport)
 
860
 
 
861
 
 
862
class ControlDirFormatInfo(object):
 
863
 
 
864
    def __init__(self, native, deprecated, hidden, experimental):
 
865
        self.deprecated = deprecated
 
866
        self.native = native
 
867
        self.hidden = hidden
 
868
        self.experimental = experimental
 
869
 
 
870
 
 
871
class ControlDirFormatRegistry(registry.Registry):
 
872
    """Registry of user-selectable ControlDir subformats.
 
873
 
 
874
    Differs from ControlDirFormat._formats in that it provides sub-formats,
 
875
    e.g. ControlDirMeta1 with weave repository.  Also, it's more user-oriented.
 
876
    """
 
877
 
 
878
    def __init__(self):
 
879
        """Create a ControlDirFormatRegistry."""
 
880
        self._aliases = set()
 
881
        self._registration_order = list()
 
882
        super(ControlDirFormatRegistry, self).__init__()
 
883
 
 
884
    def aliases(self):
 
885
        """Return a set of the format names which are aliases."""
 
886
        return frozenset(self._aliases)
 
887
 
 
888
    def register(self, key, factory, help, native=True, deprecated=False,
 
889
                 hidden=False, experimental=False, alias=False):
 
890
        """Register a ControlDirFormat factory.
 
891
 
 
892
        The factory must be a callable that takes one parameter: the key.
 
893
        It must produce an instance of the ControlDirFormat when called.
 
894
 
 
895
        This function mainly exists to prevent the info object from being
 
896
        supplied directly.
 
897
        """
 
898
        registry.Registry.register(self, key, factory, help,
 
899
            ControlDirFormatInfo(native, deprecated, hidden, experimental))
 
900
        if alias:
 
901
            self._aliases.add(key)
 
902
        self._registration_order.append(key)
 
903
 
 
904
    def register_lazy(self, key, module_name, member_name, help, native=True,
 
905
        deprecated=False, hidden=False, experimental=False, alias=False):
 
906
        registry.Registry.register_lazy(self, key, module_name, member_name,
 
907
            help, ControlDirFormatInfo(native, deprecated, hidden, experimental))
 
908
        if alias:
 
909
            self._aliases.add(key)
 
910
        self._registration_order.append(key)
 
911
 
 
912
    def set_default(self, key):
 
913
        """Set the 'default' key to be a clone of the supplied key.
 
914
 
 
915
        This method must be called once and only once.
 
916
        """
 
917
        registry.Registry.register(self, 'default', self.get(key),
 
918
            self.get_help(key), info=self.get_info(key))
 
919
        self._aliases.add('default')
 
920
 
 
921
    def set_default_repository(self, key):
 
922
        """Set the FormatRegistry default and Repository default.
 
923
 
 
924
        This is a transitional method while Repository.set_default_format
 
925
        is deprecated.
 
926
        """
 
927
        if 'default' in self:
 
928
            self.remove('default')
 
929
        self.set_default(key)
 
930
        format = self.get('default')()
 
931
 
 
932
    def make_bzrdir(self, key):
 
933
        return self.get(key)()
 
934
 
 
935
    def help_topic(self, topic):
 
936
        output = ""
 
937
        default_realkey = None
 
938
        default_help = self.get_help('default')
 
939
        help_pairs = []
 
940
        for key in self._registration_order:
 
941
            if key == 'default':
 
942
                continue
 
943
            help = self.get_help(key)
 
944
            if help == default_help:
 
945
                default_realkey = key
 
946
            else:
 
947
                help_pairs.append((key, help))
 
948
 
 
949
        def wrapped(key, help, info):
 
950
            if info.native:
 
951
                help = '(native) ' + help
 
952
            return ':%s:\n%s\n\n' % (key,
 
953
                textwrap.fill(help, initial_indent='    ',
 
954
                    subsequent_indent='    ',
 
955
                    break_long_words=False))
 
956
        if default_realkey is not None:
 
957
            output += wrapped(default_realkey, '(default) %s' % default_help,
 
958
                              self.get_info('default'))
 
959
        deprecated_pairs = []
 
960
        experimental_pairs = []
 
961
        for key, help in help_pairs:
 
962
            info = self.get_info(key)
 
963
            if info.hidden:
 
964
                continue
 
965
            elif info.deprecated:
 
966
                deprecated_pairs.append((key, help))
 
967
            elif info.experimental:
 
968
                experimental_pairs.append((key, help))
 
969
            else:
 
970
                output += wrapped(key, help, info)
 
971
        output += "\nSee :doc:`formats-help` for more about storage formats."
 
972
        other_output = ""
 
973
        if len(experimental_pairs) > 0:
 
974
            other_output += "Experimental formats are shown below.\n\n"
 
975
            for key, help in experimental_pairs:
 
976
                info = self.get_info(key)
 
977
                other_output += wrapped(key, help, info)
 
978
        else:
 
979
            other_output += \
 
980
                "No experimental formats are available.\n\n"
 
981
        if len(deprecated_pairs) > 0:
 
982
            other_output += "\nDeprecated formats are shown below.\n\n"
 
983
            for key, help in deprecated_pairs:
 
984
                info = self.get_info(key)
 
985
                other_output += wrapped(key, help, info)
 
986
        else:
 
987
            other_output += \
 
988
                "\nNo deprecated formats are available.\n\n"
 
989
        other_output += \
 
990
                "\nSee :doc:`formats-help` for more about storage formats."
 
991
 
 
992
        if topic == 'other-formats':
 
993
            return other_output
 
994
        else:
 
995
            return output
 
996
 
 
997
 
 
998
# Please register new formats after old formats so that formats
 
999
# appear in chronological order and format descriptions can build
 
1000
# on previous ones.
 
1001
format_registry = ControlDirFormatRegistry()
 
1002
 
 
1003
network_format_registry = registry.FormatRegistry()
 
1004
"""Registry of formats indexed by their network name.
 
1005
 
 
1006
The network name for a ControlDirFormat is an identifier that can be used when
 
1007
referring to formats with smart server operations. See
 
1008
ControlDirFormat.network_name() for more detail.
 
1009
"""