/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 breezy/git/dir.py

  • Committer: Jelmer Vernooij
  • Date: 2020-03-22 19:12:43 UTC
  • mfrom: (7490.7.6 work)
  • mto: (7490.7.7 work)
  • mto: This revision was merged to the branch mainline in revision 7501.
  • Revision ID: jelmer@jelmer.uk-20200322191243-yx8ils8lvfmfh7rq
Merge lp:brz/3.1.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2007 Canonical Ltd
 
2
# Copyright (C) 2010-2018 Jelmer Vernooij
 
3
#
 
4
# This program is free software; you can redistribute it and/or modify
 
5
# it under the terms of the GNU General Public License as published by
 
6
# the Free Software Foundation; either version 2 of the License, or
 
7
# (at your option) any later version.
 
8
#
 
9
# This program is distributed in the hope that it will be useful,
 
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
# GNU General Public License for more details.
 
13
#
 
14
# You should have received a copy of the GNU General Public License
 
15
# along with this program; if not, write to the Free Software
 
16
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
17
 
 
18
"""An adapter between a Git control dir and a Bazaar ControlDir."""
 
19
 
 
20
from __future__ import absolute_import
 
21
 
 
22
from .. import (
 
23
    branch as _mod_branch,
 
24
    cleanup,
 
25
    errors as brz_errors,
 
26
    trace,
 
27
    osutils,
 
28
    urlutils,
 
29
    )
 
30
from ..sixish import (
 
31
    PY3,
 
32
    viewitems,
 
33
    )
 
34
from ..transport import (
 
35
    do_catching_redirections,
 
36
    get_transport_from_path,
 
37
    )
 
38
 
 
39
from ..controldir import (
 
40
    BranchReferenceLoop,
 
41
    ControlDir,
 
42
    ControlDirFormat,
 
43
    format_registry,
 
44
    RepositoryAcquisitionPolicy,
 
45
    )
 
46
 
 
47
from .push import (
 
48
    GitPushResult,
 
49
    )
 
50
from .transportgit import (
 
51
    OBJECTDIR,
 
52
    TransportObjectStore,
 
53
    )
 
54
 
 
55
 
 
56
class GitDirConfig(object):
 
57
 
 
58
    def get_default_stack_on(self):
 
59
        return None
 
60
 
 
61
    def set_default_stack_on(self, value):
 
62
        raise brz_errors.BzrError("Cannot set configuration")
 
63
 
 
64
 
 
65
class GitControlDirFormat(ControlDirFormat):
 
66
 
 
67
    colocated_branches = True
 
68
    fixed_components = True
 
69
 
 
70
    def __eq__(self, other):
 
71
        return type(self) == type(other)
 
72
 
 
73
    def is_supported(self):
 
74
        return True
 
75
 
 
76
    def network_name(self):
 
77
        return b"git"
 
78
 
 
79
 
 
80
class UseExistingRepository(RepositoryAcquisitionPolicy):
 
81
    """A policy of reusing an existing repository"""
 
82
 
 
83
    def __init__(self, repository, stack_on=None, stack_on_pwd=None,
 
84
                 require_stacking=False):
 
85
        """Constructor.
 
86
 
 
87
        :param repository: The repository to use.
 
88
        :param stack_on: A location to stack on
 
89
        :param stack_on_pwd: If stack_on is relative, the location it is
 
90
            relative to.
 
91
        """
 
92
        super(UseExistingRepository, self).__init__(
 
93
            stack_on, stack_on_pwd, require_stacking)
 
94
        self._repository = repository
 
95
 
 
96
    def acquire_repository(self, make_working_trees=None, shared=False,
 
97
                           possible_transports=None):
 
98
        """Implementation of RepositoryAcquisitionPolicy.acquire_repository
 
99
 
 
100
        Returns an existing repository to use.
 
101
        """
 
102
        return self._repository, False
 
103
 
 
104
 
 
105
class GitDir(ControlDir):
 
106
    """An adapter to the '.git' dir used by git."""
 
107
 
 
108
    def is_supported(self):
 
109
        return True
 
110
 
 
111
    def can_convert_format(self):
 
112
        return False
 
113
 
 
114
    def break_lock(self):
 
115
        # There are no global locks, so nothing to break.
 
116
        raise NotImplementedError(self.break_lock)
 
117
 
 
118
    def cloning_metadir(self, stacked=False):
 
119
        return format_registry.make_controldir("git")
 
120
 
 
121
    def checkout_metadir(self, stacked=False):
 
122
        return format_registry.make_controldir("git")
 
123
 
 
124
    def _get_selected_ref(self, branch, ref=None):
 
125
        if ref is not None and branch is not None:
 
126
            raise brz_errors.BzrError("can't specify both ref and branch")
 
127
        if ref is not None:
 
128
            return ref
 
129
        if branch is not None:
 
130
            from .refs import branch_name_to_ref
 
131
            return branch_name_to_ref(branch)
 
132
        segment_parameters = getattr(
 
133
            self.user_transport, "get_segment_parameters", lambda: {})()
 
134
        ref = segment_parameters.get("ref")
 
135
        if ref is not None:
 
136
            return urlutils.unquote_to_bytes(ref)
 
137
        if branch is None and getattr(self, "_get_selected_branch", False):
 
138
            branch = self._get_selected_branch()
 
139
            if branch is not None:
 
140
                from .refs import branch_name_to_ref
 
141
                return branch_name_to_ref(branch)
 
142
        return b"HEAD"
 
143
 
 
144
    def get_config(self):
 
145
        return GitDirConfig()
 
146
 
 
147
    def _available_backup_name(self, base):
 
148
        return osutils.available_backup_name(base, self.root_transport.has)
 
149
 
 
150
    def sprout(self, url, revision_id=None, force_new_repo=False,
 
151
               recurse='down', possible_transports=None,
 
152
               accelerator_tree=None, hardlink=False, stacked=False,
 
153
               source_branch=None, create_tree_if_local=True):
 
154
        from ..repository import InterRepository
 
155
        from ..transport.local import LocalTransport
 
156
        from ..transport import get_transport
 
157
        target_transport = get_transport(url, possible_transports)
 
158
        target_transport.ensure_base()
 
159
        cloning_format = self.cloning_metadir()
 
160
        # Create/update the result branch
 
161
        try:
 
162
            result = ControlDir.open_from_transport(target_transport)
 
163
        except brz_errors.NotBranchError:
 
164
            result = cloning_format.initialize_on_transport(target_transport)
 
165
        source_branch = self.open_branch()
 
166
        source_repository = self.find_repository()
 
167
        try:
 
168
            result_repo = result.find_repository()
 
169
        except brz_errors.NoRepositoryPresent:
 
170
            result_repo = result.create_repository()
 
171
        if stacked:
 
172
            raise _mod_branch.UnstackableBranchFormat(
 
173
                self._format, self.user_url)
 
174
        interrepo = InterRepository.get(source_repository, result_repo)
 
175
 
 
176
        if revision_id is not None:
 
177
            determine_wants = interrepo.get_determine_wants_revids(
 
178
                [revision_id], include_tags=True)
 
179
        else:
 
180
            determine_wants = interrepo.determine_wants_all
 
181
        interrepo.fetch_objects(determine_wants=determine_wants,
 
182
                                mapping=source_branch.mapping)
 
183
        result_branch = source_branch.sprout(
 
184
            result, revision_id=revision_id, repository=result_repo)
 
185
        if (create_tree_if_local and
 
186
            result.open_branch(name="").name == result_branch.name and
 
187
            isinstance(target_transport, LocalTransport) and
 
188
                (result_repo is None or result_repo.make_working_trees())):
 
189
            wt = result.create_workingtree(
 
190
                accelerator_tree=accelerator_tree,
 
191
                hardlink=hardlink, from_branch=result_branch)
 
192
        else:
 
193
            wt = None
 
194
        if recurse == 'down':
 
195
            with cleanup.ExitStack() as stack:
 
196
                basis = None
 
197
                if wt is not None:
 
198
                    basis = wt.basis_tree()
 
199
                elif result_branch is not None:
 
200
                    basis = result_branch.basis_tree()
 
201
                elif source_branch is not None:
 
202
                    basis = source_branch.basis_tree()
 
203
                if basis is not None:
 
204
                    stack.enter_context(basis.lock_read())
 
205
                    subtrees = basis.iter_references()
 
206
                else:
 
207
                    subtrees = []
 
208
                for path in subtrees:
 
209
                    target = urlutils.join(url, urlutils.escape(path))
 
210
                    sublocation = wt.reference_parent(
 
211
                        path, possible_transports=possible_transports)
 
212
                    if sublocation is None:
 
213
                        trace.warning(
 
214
                            'Ignoring nested tree %s, parent location unknown.',
 
215
                            path)
 
216
                        continue
 
217
                    sublocation.controldir.sprout(
 
218
                        target, basis.get_reference_revision(path),
 
219
                        force_new_repo=force_new_repo, recurse=recurse,
 
220
                        stacked=stacked)
 
221
        return result
 
222
 
 
223
    def clone_on_transport(self, transport, revision_id=None,
 
224
                           force_new_repo=False, preserve_stacking=False,
 
225
                           stacked_on=None, create_prefix=False,
 
226
                           use_existing_dir=True, no_tree=False,
 
227
                           tag_selector=None):
 
228
        """See ControlDir.clone_on_transport."""
 
229
        from ..repository import InterRepository
 
230
        from .mapping import default_mapping
 
231
        from ..transport.local import LocalTransport
 
232
        if stacked_on is not None:
 
233
            raise _mod_branch.UnstackableBranchFormat(
 
234
                self._format, self.user_url)
 
235
        if no_tree:
 
236
            format = BareLocalGitControlDirFormat()
 
237
        else:
 
238
            format = LocalGitControlDirFormat()
 
239
        (target_repo, target_controldir, stacking,
 
240
         repo_policy) = format.initialize_on_transport_ex(
 
241
            transport, use_existing_dir=use_existing_dir,
 
242
            create_prefix=create_prefix,
 
243
            force_new_repo=force_new_repo)
 
244
        target_repo = target_controldir.find_repository()
 
245
        target_git_repo = target_repo._git
 
246
        source_repo = self.find_repository()
 
247
        interrepo = InterRepository.get(source_repo, target_repo)
 
248
        if revision_id is not None:
 
249
            determine_wants = interrepo.get_determine_wants_revids(
 
250
                [revision_id], include_tags=True, tag_selector=tag_selector)
 
251
        else:
 
252
            determine_wants = interrepo.determine_wants_all
 
253
        (pack_hint, _, refs) = interrepo.fetch_objects(determine_wants,
 
254
                                                       mapping=default_mapping)
 
255
        for name, val in viewitems(refs):
 
256
            target_git_repo.refs[name] = val
 
257
        result_dir = LocalGitDir(transport, target_git_repo, format)
 
258
        if revision_id is not None:
 
259
            result_dir.open_branch().set_last_revision(revision_id)
 
260
        if not no_tree and isinstance(result_dir.root_transport, LocalTransport):
 
261
            if result_dir.open_repository().make_working_trees():
 
262
                result_dir.create_workingtree()
 
263
 
 
264
        return result_dir
 
265
 
 
266
    def find_repository(self):
 
267
        """Find the repository that should be used.
 
268
 
 
269
        This does not require a branch as we use it to find the repo for
 
270
        new branches as well as to hook existing branches up to their
 
271
        repository.
 
272
        """
 
273
        return self._gitrepository_class(self._find_commondir())
 
274
 
 
275
    def get_refs_container(self):
 
276
        """Retrieve the refs container.
 
277
        """
 
278
        raise NotImplementedError(self.get_refs_container)
 
279
 
 
280
    def determine_repository_policy(self, force_new_repo=False, stack_on=None,
 
281
                                    stack_on_pwd=None, require_stacking=False):
 
282
        """Return an object representing a policy to use.
 
283
 
 
284
        This controls whether a new repository is created, and the format of
 
285
        that repository, or some existing shared repository used instead.
 
286
 
 
287
        If stack_on is supplied, will not seek a containing shared repo.
 
288
 
 
289
        :param force_new_repo: If True, require a new repository to be created.
 
290
        :param stack_on: If supplied, the location to stack on.  If not
 
291
            supplied, a default_stack_on location may be used.
 
292
        :param stack_on_pwd: If stack_on is relative, the location it is
 
293
            relative to.
 
294
        """
 
295
        return UseExistingRepository(self.find_repository())
 
296
 
 
297
    def branch_names(self):
 
298
        from .refs import ref_to_branch_name
 
299
        ret = []
 
300
        for ref in self.get_refs_container().keys():
 
301
            try:
 
302
                branch_name = ref_to_branch_name(ref)
 
303
            except UnicodeDecodeError:
 
304
                trace.warning("Ignoring branch %r with unicode error ref", ref)
 
305
                continue
 
306
            except ValueError:
 
307
                continue
 
308
            ret.append(branch_name)
 
309
        return ret
 
310
 
 
311
    def get_branches(self):
 
312
        from .refs import ref_to_branch_name
 
313
        ret = {}
 
314
        for ref in self.get_refs_container().keys():
 
315
            try:
 
316
                branch_name = ref_to_branch_name(ref)
 
317
            except UnicodeDecodeError:
 
318
                trace.warning("Ignoring branch %r with unicode error ref", ref)
 
319
                continue
 
320
            except ValueError:
 
321
                continue
 
322
            ret[branch_name] = self.open_branch(ref=ref)
 
323
        return ret
 
324
 
 
325
    def list_branches(self):
 
326
        return list(self.get_branches().values())
 
327
 
 
328
    def push_branch(self, source, revision_id=None, overwrite=False,
 
329
                    remember=False, create_prefix=False, lossy=False,
 
330
                    name=None, tag_selector=None):
 
331
        """Push the source branch into this ControlDir."""
 
332
        push_result = GitPushResult()
 
333
        push_result.workingtree_updated = None
 
334
        push_result.master_branch = None
 
335
        push_result.source_branch = source
 
336
        push_result.stacked_on = None
 
337
        from .branch import GitBranch
 
338
        if isinstance(source, GitBranch) and lossy:
 
339
            raise brz_errors.LossyPushToSameVCS(source.controldir, self)
 
340
        target = self.open_branch(name, nascent_ok=True)
 
341
        push_result.branch_push_result = source.push(
 
342
            target, overwrite=overwrite, stop_revision=revision_id,
 
343
            lossy=lossy, tag_selector=tag_selector)
 
344
        push_result.new_revid = push_result.branch_push_result.new_revid
 
345
        push_result.old_revid = push_result.branch_push_result.old_revid
 
346
        try:
 
347
            wt = self.open_workingtree()
 
348
        except brz_errors.NoWorkingTree:
 
349
            push_result.workingtree_updated = None
 
350
        else:
 
351
            if self.open_branch(name="").name == target.name:
 
352
                wt._update_git_tree(
 
353
                    old_revision=push_result.old_revid,
 
354
                    new_revision=push_result.new_revid)
 
355
                push_result.workingtree_updated = True
 
356
            else:
 
357
                push_result.workingtree_updated = False
 
358
        push_result.target_branch = target
 
359
        if source.get_push_location() is None or remember:
 
360
            source.set_push_location(push_result.target_branch.base)
 
361
        return push_result
 
362
 
 
363
 
 
364
class LocalGitControlDirFormat(GitControlDirFormat):
 
365
    """The .git directory control format."""
 
366
 
 
367
    bare = False
 
368
 
 
369
    @classmethod
 
370
    def _known_formats(self):
 
371
        return set([LocalGitControlDirFormat()])
 
372
 
 
373
    @property
 
374
    def repository_format(self):
 
375
        from .repository import GitRepositoryFormat
 
376
        return GitRepositoryFormat()
 
377
 
 
378
    @property
 
379
    def workingtree_format(self):
 
380
        from .workingtree import GitWorkingTreeFormat
 
381
        return GitWorkingTreeFormat()
 
382
 
 
383
    def get_branch_format(self):
 
384
        from .branch import LocalGitBranchFormat
 
385
        return LocalGitBranchFormat()
 
386
 
 
387
    def open(self, transport, _found=None):
 
388
        """Open this directory.
 
389
 
 
390
        """
 
391
        from .transportgit import TransportRepo
 
392
 
 
393
        def _open(transport):
 
394
            try:
 
395
                return TransportRepo(transport, self.bare,
 
396
                                     refs_text=getattr(self, "_refs_text", None))
 
397
            except ValueError as e:
 
398
                if e.args == ('Expected file to start with \'gitdir: \'', ):
 
399
                    raise brz_errors.NotBranchError(path=transport.base)
 
400
                raise
 
401
 
 
402
        def redirected(transport, e, redirection_notice):
 
403
            trace.note(redirection_notice)
 
404
            return transport._redirected_to(e.source, e.target)
 
405
        gitrepo = do_catching_redirections(_open, transport, redirected)
 
406
        if not _found and not gitrepo._controltransport.has('objects'):
 
407
            raise brz_errors.NotBranchError(path=transport.base)
 
408
        return LocalGitDir(transport, gitrepo, self)
 
409
 
 
410
    def get_format_description(self):
 
411
        return "Local Git Repository"
 
412
 
 
413
    def initialize_on_transport(self, transport):
 
414
        from .transportgit import TransportRepo
 
415
        git_repo = TransportRepo.init(transport, bare=self.bare)
 
416
        return LocalGitDir(transport, git_repo, self)
 
417
 
 
418
    def initialize_on_transport_ex(self, transport, use_existing_dir=False,
 
419
                                   create_prefix=False, force_new_repo=False,
 
420
                                   stacked_on=None,
 
421
                                   stack_on_pwd=None, repo_format_name=None,
 
422
                                   make_working_trees=None,
 
423
                                   shared_repo=False, vfs_only=False):
 
424
        if shared_repo:
 
425
            raise brz_errors.SharedRepositoriesUnsupported(self)
 
426
 
 
427
        def make_directory(transport):
 
428
            transport.mkdir('.')
 
429
            return transport
 
430
 
 
431
        def redirected(transport, e, redirection_notice):
 
432
            trace.note(redirection_notice)
 
433
            return transport._redirected_to(e.source, e.target)
 
434
        try:
 
435
            transport = do_catching_redirections(
 
436
                make_directory, transport, redirected)
 
437
        except brz_errors.FileExists:
 
438
            if not use_existing_dir:
 
439
                raise
 
440
        except brz_errors.NoSuchFile:
 
441
            if not create_prefix:
 
442
                raise
 
443
            transport.create_prefix()
 
444
        controldir = self.initialize_on_transport(transport)
 
445
        if repo_format_name:
 
446
            result_repo = controldir.find_repository()
 
447
            repository_policy = UseExistingRepository(result_repo)
 
448
            result_repo.lock_write()
 
449
        else:
 
450
            result_repo = None
 
451
            repository_policy = None
 
452
        return (result_repo, controldir, False,
 
453
                repository_policy)
 
454
 
 
455
    def is_supported(self):
 
456
        return True
 
457
 
 
458
    def supports_transport(self, transport):
 
459
        try:
 
460
            external_url = transport.external_url()
 
461
        except brz_errors.InProcessTransport:
 
462
            raise brz_errors.NotBranchError(path=transport.base)
 
463
        return external_url.startswith("file:")
 
464
 
 
465
    def is_control_filename(self, filename):
 
466
        return (filename == '.git'
 
467
                or filename.startswith('.git/')
 
468
                or filename.startswith('.git\\'))
 
469
 
 
470
 
 
471
class BareLocalGitControlDirFormat(LocalGitControlDirFormat):
 
472
 
 
473
    bare = True
 
474
    supports_workingtrees = False
 
475
 
 
476
    def get_format_description(self):
 
477
        return "Local Git Repository (bare)"
 
478
 
 
479
    def is_control_filename(self, filename):
 
480
        return False
 
481
 
 
482
 
 
483
class LocalGitDir(GitDir):
 
484
    """An adapter to the '.git' dir used by git."""
 
485
 
 
486
    def _get_gitrepository_class(self):
 
487
        from .repository import LocalGitRepository
 
488
        return LocalGitRepository
 
489
 
 
490
    def __repr__(self):
 
491
        return "<%s at %r>" % (
 
492
            self.__class__.__name__, self.root_transport.base)
 
493
 
 
494
    _gitrepository_class = property(_get_gitrepository_class)
 
495
 
 
496
    @property
 
497
    def user_transport(self):
 
498
        return self.root_transport
 
499
 
 
500
    @property
 
501
    def control_transport(self):
 
502
        return self._git._controltransport
 
503
 
 
504
    def __init__(self, transport, gitrepo, format):
 
505
        self._format = format
 
506
        self.root_transport = transport
 
507
        self._mode_check_done = False
 
508
        self._git = gitrepo
 
509
        if gitrepo.bare:
 
510
            self.transport = transport
 
511
        else:
 
512
            self.transport = transport.clone('.git')
 
513
        self._mode_check_done = None
 
514
 
 
515
    def _get_symref(self, ref):
 
516
        ref_chain, unused_sha = self._git.refs.follow(ref)
 
517
        if len(ref_chain) == 1:
 
518
            return None
 
519
        return ref_chain[1]
 
520
 
 
521
    def set_branch_reference(self, target_branch, name=None):
 
522
        ref = self._get_selected_ref(name)
 
523
        target_transport = target_branch.controldir.control_transport
 
524
        if self.control_transport.base == target_transport.base:
 
525
            if ref == target_branch.ref:
 
526
                raise BranchReferenceLoop(target_branch)
 
527
            self._git.refs.set_symbolic_ref(ref, target_branch.ref)
 
528
        else:
 
529
            try:
 
530
                target_path = (
 
531
                    target_branch.controldir.control_transport.local_abspath(
 
532
                        '.'))
 
533
            except brz_errors.NotLocalUrl:
 
534
                raise brz_errors.IncompatibleFormat(
 
535
                    target_branch._format, self._format)
 
536
            # TODO(jelmer): Do some consistency checking across branches..
 
537
            self.control_transport.put_bytes(
 
538
                'commondir', target_path.encode('utf-8'))
 
539
            # TODO(jelmer): Urgh, avoid mucking about with internals.
 
540
            self._git._commontransport = (
 
541
                target_branch.repository._git._commontransport.clone())
 
542
            self._git.object_store = TransportObjectStore(
 
543
                self._git._commontransport.clone(OBJECTDIR))
 
544
            self._git.refs.transport = self._git._commontransport
 
545
            target_ref_chain, unused_sha = (
 
546
                target_branch.controldir._git.refs.follow(target_branch.ref))
 
547
            for target_ref in target_ref_chain:
 
548
                if target_ref == b'HEAD':
 
549
                    continue
 
550
                break
 
551
            else:
 
552
                # Can't create a reference to something that is not a in a repository.
 
553
                raise brz_errors.IncompatibleFormat(
 
554
                    self.set_branch_reference, self)
 
555
            self._git.refs.set_symbolic_ref(ref, target_ref)
 
556
 
 
557
    def get_branch_reference(self, name=None):
 
558
        ref = self._get_selected_ref(name)
 
559
        target_ref = self._get_symref(ref)
 
560
        if target_ref is not None:
 
561
            from .refs import ref_to_branch_name
 
562
            try:
 
563
                branch_name = ref_to_branch_name(target_ref)
 
564
            except ValueError:
 
565
                params = {'ref': urlutils.quote(
 
566
                    target_ref.decode('utf-8'), '')}
 
567
            else:
 
568
                if branch_name != '':
 
569
                    params = {'branch': urlutils.quote(branch_name, '')}
 
570
                else:
 
571
                    params = {}
 
572
            try:
 
573
                commondir = self.control_transport.get_bytes('commondir')
 
574
            except brz_errors.NoSuchFile:
 
575
                base_url = self.user_url.rstrip('/')
 
576
            else:
 
577
                base_url = urlutils.local_path_to_url(
 
578
                    commondir.decode(osutils._fs_enc)).rstrip('/.git/') + '/'
 
579
            if not PY3:
 
580
                params = {k: v.encode('utf-8') for (k, v) in viewitems(params)}
 
581
            return urlutils.join_segment_parameters(base_url, params)
 
582
        return None
 
583
 
 
584
    def find_branch_format(self, name=None):
 
585
        from .branch import (
 
586
            LocalGitBranchFormat,
 
587
            )
 
588
        return LocalGitBranchFormat()
 
589
 
 
590
    def get_branch_transport(self, branch_format, name=None):
 
591
        if branch_format is None:
 
592
            return self.transport
 
593
        if isinstance(branch_format, LocalGitControlDirFormat):
 
594
            return self.transport
 
595
        raise brz_errors.IncompatibleFormat(branch_format, self._format)
 
596
 
 
597
    def get_repository_transport(self, format):
 
598
        if format is None:
 
599
            return self.transport
 
600
        if isinstance(format, LocalGitControlDirFormat):
 
601
            return self.transport
 
602
        raise brz_errors.IncompatibleFormat(format, self._format)
 
603
 
 
604
    def get_workingtree_transport(self, format):
 
605
        if format is None:
 
606
            return self.transport
 
607
        if isinstance(format, LocalGitControlDirFormat):
 
608
            return self.transport
 
609
        raise brz_errors.IncompatibleFormat(format, self._format)
 
610
 
 
611
    def open_branch(self, name=None, unsupported=False, ignore_fallbacks=None,
 
612
                    ref=None, possible_transports=None, nascent_ok=False):
 
613
        """'create' a branch for this dir."""
 
614
        repo = self.find_repository()
 
615
        from .branch import LocalGitBranch
 
616
        ref = self._get_selected_ref(name, ref)
 
617
        if not nascent_ok and ref not in self._git.refs:
 
618
            raise brz_errors.NotBranchError(
 
619
                self.root_transport.base, controldir=self)
 
620
        ref_chain, unused_sha = self._git.refs.follow(ref)
 
621
        if ref_chain[-1] == b'HEAD':
 
622
            controldir = self
 
623
        else:
 
624
            controldir = self._find_commondir()
 
625
        return LocalGitBranch(controldir, repo, ref_chain[-1])
 
626
 
 
627
    def destroy_branch(self, name=None):
 
628
        refname = self._get_selected_ref(name)
 
629
        if refname == b'HEAD':
 
630
            # HEAD can't be removed
 
631
            raise brz_errors.UnsupportedOperation(
 
632
                self.destroy_branch, self)
 
633
        try:
 
634
            del self._git.refs[refname]
 
635
        except KeyError:
 
636
            raise brz_errors.NotBranchError(
 
637
                self.root_transport.base, controldir=self)
 
638
 
 
639
    def destroy_repository(self):
 
640
        raise brz_errors.UnsupportedOperation(self.destroy_repository, self)
 
641
 
 
642
    def destroy_workingtree(self):
 
643
        raise brz_errors.UnsupportedOperation(self.destroy_workingtree, self)
 
644
 
 
645
    def destroy_workingtree_metadata(self):
 
646
        raise brz_errors.UnsupportedOperation(
 
647
            self.destroy_workingtree_metadata, self)
 
648
 
 
649
    def needs_format_conversion(self, format=None):
 
650
        return not isinstance(self._format, format.__class__)
 
651
 
 
652
    def open_repository(self):
 
653
        """'open' a repository for this dir."""
 
654
        if self.control_transport.has('commondir'):
 
655
            raise brz_errors.NoRepositoryPresent(self)
 
656
        return self._gitrepository_class(self)
 
657
 
 
658
    def has_workingtree(self):
 
659
        return not self._git.bare
 
660
 
 
661
    def open_workingtree(self, recommend_upgrade=True, unsupported=False):
 
662
        if not self._git.bare:
 
663
            repo = self.find_repository()
 
664
            from .workingtree import GitWorkingTree
 
665
            branch = self.open_branch(ref=b'HEAD', nascent_ok=True)
 
666
            return GitWorkingTree(self, repo, branch)
 
667
        loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
 
668
        raise brz_errors.NoWorkingTree(loc)
 
669
 
 
670
    def create_repository(self, shared=False):
 
671
        from .repository import GitRepositoryFormat
 
672
        if shared:
 
673
            raise brz_errors.IncompatibleFormat(
 
674
                GitRepositoryFormat(), self._format)
 
675
        return self.find_repository()
 
676
 
 
677
    def create_branch(self, name=None, repository=None,
 
678
                      append_revisions_only=None, ref=None):
 
679
        refname = self._get_selected_ref(name, ref)
 
680
        if refname != b'HEAD' and refname in self._git.refs:
 
681
            raise brz_errors.AlreadyBranchError(self.user_url)
 
682
        repo = self.open_repository()
 
683
        if refname in self._git.refs:
 
684
            ref_chain, unused_sha = self._git.refs.follow(
 
685
                self._get_selected_ref(None))
 
686
            if ref_chain[0] == b'HEAD':
 
687
                refname = ref_chain[1]
 
688
        from .branch import LocalGitBranch
 
689
        branch = LocalGitBranch(self, repo, refname)
 
690
        if append_revisions_only:
 
691
            branch.set_append_revisions_only(append_revisions_only)
 
692
        return branch
 
693
 
 
694
    def backup_bzrdir(self):
 
695
        if not self._git.bare:
 
696
            self.root_transport.copy_tree(".git", ".git.backup")
 
697
            return (self.root_transport.abspath(".git"),
 
698
                    self.root_transport.abspath(".git.backup"))
 
699
        else:
 
700
            basename = urlutils.basename(self.root_transport.base)
 
701
            parent = self.root_transport.clone('..')
 
702
            parent.copy_tree(basename, basename + ".backup")
 
703
 
 
704
    def create_workingtree(self, revision_id=None, from_branch=None,
 
705
                           accelerator_tree=None, hardlink=False):
 
706
        if self._git.bare:
 
707
            raise brz_errors.UnsupportedOperation(
 
708
                self.create_workingtree, self)
 
709
        if from_branch is None:
 
710
            from_branch = self.open_branch(nascent_ok=True)
 
711
        if revision_id is None:
 
712
            revision_id = from_branch.last_revision()
 
713
        repo = self.find_repository()
 
714
        from .workingtree import GitWorkingTree
 
715
        wt = GitWorkingTree(self, repo, from_branch)
 
716
        wt.set_last_revision(revision_id)
 
717
        wt._build_checkout_with_index()
 
718
        return wt
 
719
 
 
720
    def _find_or_create_repository(self, force_new_repo=None):
 
721
        return self.create_repository(shared=False)
 
722
 
 
723
    def _find_creation_modes(self):
 
724
        """Determine the appropriate modes for files and directories.
 
725
 
 
726
        They're always set to be consistent with the base directory,
 
727
        assuming that this transport allows setting modes.
 
728
        """
 
729
        # TODO: Do we need or want an option (maybe a config setting) to turn
 
730
        # this off or override it for particular locations? -- mbp 20080512
 
731
        if self._mode_check_done:
 
732
            return
 
733
        self._mode_check_done = True
 
734
        try:
 
735
            st = self.transport.stat('.')
 
736
        except brz_errors.TransportNotPossible:
 
737
            self._dir_mode = None
 
738
            self._file_mode = None
 
739
        else:
 
740
            # Check the directory mode, but also make sure the created
 
741
            # directories and files are read-write for this user. This is
 
742
            # mostly a workaround for filesystems which lie about being able to
 
743
            # write to a directory (cygwin & win32)
 
744
            if (st.st_mode & 0o7777 == 0o0000):
 
745
                # FTP allows stat but does not return dir/file modes
 
746
                self._dir_mode = None
 
747
                self._file_mode = None
 
748
            else:
 
749
                self._dir_mode = (st.st_mode & 0o7777) | 0o0700
 
750
                # Remove the sticky and execute bits for files
 
751
                self._file_mode = self._dir_mode & ~0o7111
 
752
 
 
753
    def _get_file_mode(self):
 
754
        """Return Unix mode for newly created files, or None.
 
755
        """
 
756
        if not self._mode_check_done:
 
757
            self._find_creation_modes()
 
758
        return self._file_mode
 
759
 
 
760
    def _get_dir_mode(self):
 
761
        """Return Unix mode for newly created directories, or None.
 
762
        """
 
763
        if not self._mode_check_done:
 
764
            self._find_creation_modes()
 
765
        return self._dir_mode
 
766
 
 
767
    def get_refs_container(self):
 
768
        return self._git.refs
 
769
 
 
770
    def get_peeled(self, ref):
 
771
        return self._git.get_peeled(ref)
 
772
 
 
773
    def _find_commondir(self):
 
774
        try:
 
775
            commondir = self.control_transport.get_bytes('commondir')
 
776
        except brz_errors.NoSuchFile:
 
777
            return self
 
778
        else:
 
779
            commondir = commondir.rstrip(b'/.git/').decode(osutils._fs_enc)
 
780
            return ControlDir.open_from_transport(
 
781
                get_transport_from_path(commondir))