/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-05-17 15:14:38 UTC
  • mfrom: (5050.73.3 2.2)
  • mto: (5609.39.5 2.3)
  • mto: This revision was merged to the branch mainline in revision 5885.
  • Revision ID: v.ladeuil+lp@free.fr-20110517151438-j75xuw2zm9alk9a5
Merge 2.2 into 2.3 resolving conflicts

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