/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: Jelmer Vernooij
  • Date: 2011-03-10 13:23:20 UTC
  • mto: (5582.10.90 weave-plugin)
  • mto: This revision was merged to the branch mainline in revision 5717.
  • Revision ID: jelmer@samba.org-20110310132320-k6uclq04lapawivt
Fix imports.

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