/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

Support user.signingkey configuration variable in .git/config.

Merged from https://code.launchpad.net/~jelmer/brz/local-git-key/+merge/381000

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
        if stacked_on is not None:
 
232
            raise _mod_branch.UnstackableBranchFormat(
 
233
                self._format, self.user_url)
 
234
        if no_tree:
 
235
            format = BareLocalGitControlDirFormat()
 
236
        else:
 
237
            format = LocalGitControlDirFormat()
 
238
        (target_repo, target_controldir, stacking,
 
239
         repo_policy) = format.initialize_on_transport_ex(
 
240
            transport, use_existing_dir=use_existing_dir,
 
241
            create_prefix=create_prefix,
 
242
            force_new_repo=force_new_repo)
 
243
        target_repo = target_controldir.find_repository()
 
244
        target_git_repo = target_repo._git
 
245
        source_repo = self.find_repository()
 
246
        interrepo = InterRepository.get(source_repo, target_repo)
 
247
        if revision_id is not None:
 
248
            determine_wants = interrepo.get_determine_wants_revids(
 
249
                [revision_id], include_tags=True, tag_selector=tag_selector)
 
250
        else:
 
251
            determine_wants = interrepo.determine_wants_all
 
252
        (pack_hint, _, refs) = interrepo.fetch_objects(determine_wants,
 
253
                                                       mapping=default_mapping)
 
254
        for name, val in viewitems(refs):
 
255
            target_git_repo.refs[name] = val
 
256
        result_dir = self.__class__(transport, target_git_repo, format)
 
257
        if revision_id is not None:
 
258
            result_dir.open_branch().set_last_revision(revision_id)
 
259
        try:
 
260
            # Cheaper to check if the target is not local, than to try making
 
261
            # the tree and fail.
 
262
            result_dir.root_transport.local_abspath('.')
 
263
            if result_dir.open_repository().make_working_trees():
 
264
                self.open_workingtree().clone(
 
265
                    result_dir, revision_id=revision_id)
 
266
        except (brz_errors.NoWorkingTree, brz_errors.NotLocalUrl):
 
267
            pass
 
268
 
 
269
        return result_dir
 
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
        return self._gitrepository_class(self._find_commondir())
 
279
 
 
280
    def get_refs_container(self):
 
281
        """Retrieve the refs container.
 
282
        """
 
283
        raise NotImplementedError(self.get_refs_container)
 
284
 
 
285
    def determine_repository_policy(self, force_new_repo=False, stack_on=None,
 
286
                                    stack_on_pwd=None, require_stacking=False):
 
287
        """Return an object representing a policy to use.
 
288
 
 
289
        This controls whether a new repository is created, and the format of
 
290
        that repository, or some existing shared repository used instead.
 
291
 
 
292
        If stack_on is supplied, will not seek a containing shared repo.
 
293
 
 
294
        :param force_new_repo: If True, require a new repository to be created.
 
295
        :param stack_on: If supplied, the location to stack on.  If not
 
296
            supplied, a default_stack_on location may be used.
 
297
        :param stack_on_pwd: If stack_on is relative, the location it is
 
298
            relative to.
 
299
        """
 
300
        return UseExistingRepository(self.find_repository())
 
301
 
 
302
    def get_branches(self):
 
303
        from .refs import ref_to_branch_name
 
304
        ret = {}
 
305
        for ref in self.get_refs_container().keys():
 
306
            try:
 
307
                branch_name = ref_to_branch_name(ref)
 
308
            except UnicodeDecodeError:
 
309
                trace.warning("Ignoring branch %r with unicode error ref", ref)
 
310
                continue
 
311
            except ValueError:
 
312
                continue
 
313
            ret[branch_name] = self.open_branch(ref=ref)
 
314
        return ret
 
315
 
 
316
    def list_branches(self):
 
317
        return list(self.get_branches().values())
 
318
 
 
319
    def push_branch(self, source, revision_id=None, overwrite=False,
 
320
                    remember=False, create_prefix=False, lossy=False,
 
321
                    name=None, tag_selector=None):
 
322
        """Push the source branch into this ControlDir."""
 
323
        push_result = GitPushResult()
 
324
        push_result.workingtree_updated = None
 
325
        push_result.master_branch = None
 
326
        push_result.source_branch = source
 
327
        push_result.stacked_on = None
 
328
        from .branch import GitBranch
 
329
        if isinstance(source, GitBranch) and lossy:
 
330
            raise brz_errors.LossyPushToSameVCS(source.controldir, self)
 
331
        target = self.open_branch(name, nascent_ok=True)
 
332
        push_result.branch_push_result = source.push(
 
333
            target, overwrite=overwrite, stop_revision=revision_id,
 
334
            lossy=lossy, tag_selector=tag_selector)
 
335
        push_result.new_revid = push_result.branch_push_result.new_revid
 
336
        push_result.old_revid = push_result.branch_push_result.old_revid
 
337
        try:
 
338
            wt = self.open_workingtree()
 
339
        except brz_errors.NoWorkingTree:
 
340
            push_result.workingtree_updated = None
 
341
        else:
 
342
            if self.open_branch(name="").name == target.name:
 
343
                wt._update_git_tree(
 
344
                    old_revision=push_result.old_revid,
 
345
                    new_revision=push_result.new_revid)
 
346
                push_result.workingtree_updated = True
 
347
            else:
 
348
                push_result.workingtree_updated = False
 
349
        push_result.target_branch = target
 
350
        if source.get_push_location() is None or remember:
 
351
            source.set_push_location(push_result.target_branch.base)
 
352
        return push_result
 
353
 
 
354
 
 
355
class LocalGitControlDirFormat(GitControlDirFormat):
 
356
    """The .git directory control format."""
 
357
 
 
358
    bare = False
 
359
 
 
360
    @classmethod
 
361
    def _known_formats(self):
 
362
        return set([LocalGitControlDirFormat()])
 
363
 
 
364
    @property
 
365
    def repository_format(self):
 
366
        from .repository import GitRepositoryFormat
 
367
        return GitRepositoryFormat()
 
368
 
 
369
    @property
 
370
    def workingtree_format(self):
 
371
        from .workingtree import GitWorkingTreeFormat
 
372
        return GitWorkingTreeFormat()
 
373
 
 
374
    def get_branch_format(self):
 
375
        from .branch import LocalGitBranchFormat
 
376
        return LocalGitBranchFormat()
 
377
 
 
378
    def open(self, transport, _found=None):
 
379
        """Open this directory.
 
380
 
 
381
        """
 
382
        from .transportgit import TransportRepo
 
383
 
 
384
        def _open(transport):
 
385
            try:
 
386
                return TransportRepo(transport, self.bare,
 
387
                                     refs_text=getattr(self, "_refs_text", None))
 
388
            except ValueError as e:
 
389
                if e.args == ('Expected file to start with \'gitdir: \'', ):
 
390
                    raise brz_errors.NotBranchError(path=transport.base)
 
391
                raise
 
392
 
 
393
        def redirected(transport, e, redirection_notice):
 
394
            trace.note(redirection_notice)
 
395
            return transport._redirected_to(e.source, e.target)
 
396
        gitrepo = do_catching_redirections(_open, transport, redirected)
 
397
        if not _found and not gitrepo._controltransport.has('objects'):
 
398
            raise brz_errors.NotBranchError(path=transport.base)
 
399
        return LocalGitDir(transport, gitrepo, self)
 
400
 
 
401
    def get_format_description(self):
 
402
        return "Local Git Repository"
 
403
 
 
404
    def initialize_on_transport(self, transport):
 
405
        from .transportgit import TransportRepo
 
406
        git_repo = TransportRepo.init(transport, bare=self.bare)
 
407
        return LocalGitDir(transport, git_repo, self)
 
408
 
 
409
    def initialize_on_transport_ex(self, transport, use_existing_dir=False,
 
410
                                   create_prefix=False, force_new_repo=False,
 
411
                                   stacked_on=None,
 
412
                                   stack_on_pwd=None, repo_format_name=None,
 
413
                                   make_working_trees=None,
 
414
                                   shared_repo=False, vfs_only=False):
 
415
        if shared_repo:
 
416
            raise brz_errors.SharedRepositoriesUnsupported(self)
 
417
 
 
418
        def make_directory(transport):
 
419
            transport.mkdir('.')
 
420
            return transport
 
421
 
 
422
        def redirected(transport, e, redirection_notice):
 
423
            trace.note(redirection_notice)
 
424
            return transport._redirected_to(e.source, e.target)
 
425
        try:
 
426
            transport = do_catching_redirections(
 
427
                make_directory, transport, redirected)
 
428
        except brz_errors.FileExists:
 
429
            if not use_existing_dir:
 
430
                raise
 
431
        except brz_errors.NoSuchFile:
 
432
            if not create_prefix:
 
433
                raise
 
434
            transport.create_prefix()
 
435
        controldir = self.initialize_on_transport(transport)
 
436
        if repo_format_name:
 
437
            result_repo = controldir.find_repository()
 
438
            repository_policy = UseExistingRepository(result_repo)
 
439
            result_repo.lock_write()
 
440
        else:
 
441
            result_repo = None
 
442
            repository_policy = None
 
443
        return (result_repo, controldir, False,
 
444
                repository_policy)
 
445
 
 
446
    def is_supported(self):
 
447
        return True
 
448
 
 
449
    def supports_transport(self, transport):
 
450
        try:
 
451
            external_url = transport.external_url()
 
452
        except brz_errors.InProcessTransport:
 
453
            raise brz_errors.NotBranchError(path=transport.base)
 
454
        return external_url.startswith("file:")
 
455
 
 
456
    def is_control_filename(self, filename):
 
457
        return (filename == '.git'
 
458
                or filename.startswith('.git/')
 
459
                or filename.startswith('.git\\'))
 
460
 
 
461
 
 
462
class BareLocalGitControlDirFormat(LocalGitControlDirFormat):
 
463
 
 
464
    bare = True
 
465
    supports_workingtrees = False
 
466
 
 
467
    def get_format_description(self):
 
468
        return "Local Git Repository (bare)"
 
469
 
 
470
    def is_control_filename(self, filename):
 
471
        return False
 
472
 
 
473
 
 
474
class LocalGitDir(GitDir):
 
475
    """An adapter to the '.git' dir used by git."""
 
476
 
 
477
    def _get_gitrepository_class(self):
 
478
        from .repository import LocalGitRepository
 
479
        return LocalGitRepository
 
480
 
 
481
    def __repr__(self):
 
482
        return "<%s at %r>" % (
 
483
            self.__class__.__name__, self.root_transport.base)
 
484
 
 
485
    _gitrepository_class = property(_get_gitrepository_class)
 
486
 
 
487
    @property
 
488
    def user_transport(self):
 
489
        return self.root_transport
 
490
 
 
491
    @property
 
492
    def control_transport(self):
 
493
        return self._git._controltransport
 
494
 
 
495
    def __init__(self, transport, gitrepo, format):
 
496
        self._format = format
 
497
        self.root_transport = transport
 
498
        self._mode_check_done = False
 
499
        self._git = gitrepo
 
500
        if gitrepo.bare:
 
501
            self.transport = transport
 
502
        else:
 
503
            self.transport = transport.clone('.git')
 
504
        self._mode_check_done = None
 
505
 
 
506
    def _get_symref(self, ref):
 
507
        ref_chain, unused_sha = self._git.refs.follow(ref)
 
508
        if len(ref_chain) == 1:
 
509
            return None
 
510
        return ref_chain[1]
 
511
 
 
512
    def set_branch_reference(self, target_branch, name=None):
 
513
        ref = self._get_selected_ref(name)
 
514
        target_transport = target_branch.controldir.control_transport
 
515
        if self.control_transport.base == target_transport.base:
 
516
            if ref == target_branch.ref:
 
517
                raise BranchReferenceLoop(target_branch)
 
518
            self._git.refs.set_symbolic_ref(ref, target_branch.ref)
 
519
        else:
 
520
            try:
 
521
                target_path = (
 
522
                    target_branch.controldir.control_transport.local_abspath(
 
523
                        '.'))
 
524
            except brz_errors.NotLocalUrl:
 
525
                raise brz_errors.IncompatibleFormat(
 
526
                    target_branch._format, self._format)
 
527
            # TODO(jelmer): Do some consistency checking across branches..
 
528
            self.control_transport.put_bytes(
 
529
                'commondir', target_path.encode('utf-8'))
 
530
            # TODO(jelmer): Urgh, avoid mucking about with internals.
 
531
            self._git._commontransport = (
 
532
                target_branch.repository._git._commontransport.clone())
 
533
            self._git.object_store = TransportObjectStore(
 
534
                self._git._commontransport.clone(OBJECTDIR))
 
535
            self._git.refs.transport = self._git._commontransport
 
536
            target_ref_chain, unused_sha = (
 
537
                target_branch.controldir._git.refs.follow(target_branch.ref))
 
538
            for target_ref in target_ref_chain:
 
539
                if target_ref == b'HEAD':
 
540
                    continue
 
541
                break
 
542
            else:
 
543
                # Can't create a reference to something that is not a in a repository.
 
544
                raise brz_errors.IncompatibleFormat(
 
545
                    self.set_branch_reference, self)
 
546
            self._git.refs.set_symbolic_ref(ref, target_ref)
 
547
 
 
548
    def get_branch_reference(self, name=None):
 
549
        ref = self._get_selected_ref(name)
 
550
        target_ref = self._get_symref(ref)
 
551
        if target_ref is not None:
 
552
            from .refs import ref_to_branch_name
 
553
            try:
 
554
                branch_name = ref_to_branch_name(target_ref)
 
555
            except ValueError:
 
556
                params = {'ref': urlutils.quote(
 
557
                    target_ref.decode('utf-8'), '')}
 
558
            else:
 
559
                if branch_name != '':
 
560
                    params = {'branch': urlutils.quote(branch_name, '')}
 
561
                else:
 
562
                    params = {}
 
563
            try:
 
564
                commondir = self.control_transport.get_bytes('commondir')
 
565
            except brz_errors.NoSuchFile:
 
566
                base_url = self.user_url.rstrip('/')
 
567
            else:
 
568
                base_url = urlutils.local_path_to_url(
 
569
                    commondir.decode(osutils._fs_enc)).rstrip('/.git/') + '/'
 
570
            if not PY3:
 
571
                params = {k: v.encode('utf-8') for (k, v) in viewitems(params)}
 
572
            return urlutils.join_segment_parameters(base_url, params)
 
573
        return None
 
574
 
 
575
    def find_branch_format(self, name=None):
 
576
        from .branch import (
 
577
            LocalGitBranchFormat,
 
578
            )
 
579
        return LocalGitBranchFormat()
 
580
 
 
581
    def get_branch_transport(self, branch_format, name=None):
 
582
        if branch_format is None:
 
583
            return self.transport
 
584
        if isinstance(branch_format, LocalGitControlDirFormat):
 
585
            return self.transport
 
586
        raise brz_errors.IncompatibleFormat(branch_format, self._format)
 
587
 
 
588
    def get_repository_transport(self, format):
 
589
        if format is None:
 
590
            return self.transport
 
591
        if isinstance(format, LocalGitControlDirFormat):
 
592
            return self.transport
 
593
        raise brz_errors.IncompatibleFormat(format, self._format)
 
594
 
 
595
    def get_workingtree_transport(self, format):
 
596
        if format is None:
 
597
            return self.transport
 
598
        if isinstance(format, LocalGitControlDirFormat):
 
599
            return self.transport
 
600
        raise brz_errors.IncompatibleFormat(format, self._format)
 
601
 
 
602
    def open_branch(self, name=None, unsupported=False, ignore_fallbacks=None,
 
603
                    ref=None, possible_transports=None, nascent_ok=False):
 
604
        """'create' a branch for this dir."""
 
605
        repo = self.find_repository()
 
606
        from .branch import LocalGitBranch
 
607
        ref = self._get_selected_ref(name, ref)
 
608
        if not nascent_ok and ref not in self._git.refs:
 
609
            raise brz_errors.NotBranchError(
 
610
                self.root_transport.base, controldir=self)
 
611
        ref_chain, unused_sha = self._git.refs.follow(ref)
 
612
        if ref_chain[-1] == b'HEAD':
 
613
            controldir = self
 
614
        else:
 
615
            controldir = self._find_commondir()
 
616
        return LocalGitBranch(controldir, repo, ref_chain[-1])
 
617
 
 
618
    def destroy_branch(self, name=None):
 
619
        refname = self._get_selected_ref(name)
 
620
        if refname == b'HEAD':
 
621
            # HEAD can't be removed
 
622
            raise brz_errors.UnsupportedOperation(
 
623
                self.destroy_branch, self)
 
624
        try:
 
625
            del self._git.refs[refname]
 
626
        except KeyError:
 
627
            raise brz_errors.NotBranchError(
 
628
                self.root_transport.base, controldir=self)
 
629
 
 
630
    def destroy_repository(self):
 
631
        raise brz_errors.UnsupportedOperation(self.destroy_repository, self)
 
632
 
 
633
    def destroy_workingtree(self):
 
634
        raise brz_errors.UnsupportedOperation(self.destroy_workingtree, self)
 
635
 
 
636
    def destroy_workingtree_metadata(self):
 
637
        raise brz_errors.UnsupportedOperation(
 
638
            self.destroy_workingtree_metadata, self)
 
639
 
 
640
    def needs_format_conversion(self, format=None):
 
641
        return not isinstance(self._format, format.__class__)
 
642
 
 
643
    def open_repository(self):
 
644
        """'open' a repository for this dir."""
 
645
        if self.control_transport.has('commondir'):
 
646
            raise brz_errors.NoRepositoryPresent(self)
 
647
        return self._gitrepository_class(self)
 
648
 
 
649
    def has_workingtree(self):
 
650
        return not self._git.bare
 
651
 
 
652
    def open_workingtree(self, recommend_upgrade=True, unsupported=False):
 
653
        if not self._git.bare:
 
654
            repo = self.find_repository()
 
655
            from .workingtree import GitWorkingTree
 
656
            branch = self.open_branch(ref=b'HEAD', nascent_ok=True)
 
657
            return GitWorkingTree(self, repo, branch)
 
658
        loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
 
659
        raise brz_errors.NoWorkingTree(loc)
 
660
 
 
661
    def create_repository(self, shared=False):
 
662
        from .repository import GitRepositoryFormat
 
663
        if shared:
 
664
            raise brz_errors.IncompatibleFormat(
 
665
                GitRepositoryFormat(), self._format)
 
666
        return self.find_repository()
 
667
 
 
668
    def create_branch(self, name=None, repository=None,
 
669
                      append_revisions_only=None, ref=None):
 
670
        refname = self._get_selected_ref(name, ref)
 
671
        if refname != b'HEAD' and refname in self._git.refs:
 
672
            raise brz_errors.AlreadyBranchError(self.user_url)
 
673
        repo = self.open_repository()
 
674
        if refname in self._git.refs:
 
675
            ref_chain, unused_sha = self._git.refs.follow(
 
676
                self._get_selected_ref(None))
 
677
            if ref_chain[0] == b'HEAD':
 
678
                refname = ref_chain[1]
 
679
        from .branch import LocalGitBranch
 
680
        branch = LocalGitBranch(self, repo, refname)
 
681
        if append_revisions_only:
 
682
            branch.set_append_revisions_only(append_revisions_only)
 
683
        return branch
 
684
 
 
685
    def backup_bzrdir(self):
 
686
        if not self._git.bare:
 
687
            self.root_transport.copy_tree(".git", ".git.backup")
 
688
            return (self.root_transport.abspath(".git"),
 
689
                    self.root_transport.abspath(".git.backup"))
 
690
        else:
 
691
            basename = urlutils.basename(self.root_transport.base)
 
692
            parent = self.root_transport.clone('..')
 
693
            parent.copy_tree(basename, basename + ".backup")
 
694
 
 
695
    def create_workingtree(self, revision_id=None, from_branch=None,
 
696
                           accelerator_tree=None, hardlink=False):
 
697
        if self._git.bare:
 
698
            raise brz_errors.UnsupportedOperation(
 
699
                self.create_workingtree, self)
 
700
        if from_branch is None:
 
701
            from_branch = self.open_branch(nascent_ok=True)
 
702
        if revision_id is None:
 
703
            revision_id = from_branch.last_revision()
 
704
        repo = self.find_repository()
 
705
        from .workingtree import GitWorkingTree
 
706
        wt = GitWorkingTree(self, repo, from_branch)
 
707
        wt.set_last_revision(revision_id)
 
708
        wt._build_checkout_with_index()
 
709
        return wt
 
710
 
 
711
    def _find_or_create_repository(self, force_new_repo=None):
 
712
        return self.create_repository(shared=False)
 
713
 
 
714
    def _find_creation_modes(self):
 
715
        """Determine the appropriate modes for files and directories.
 
716
 
 
717
        They're always set to be consistent with the base directory,
 
718
        assuming that this transport allows setting modes.
 
719
        """
 
720
        # TODO: Do we need or want an option (maybe a config setting) to turn
 
721
        # this off or override it for particular locations? -- mbp 20080512
 
722
        if self._mode_check_done:
 
723
            return
 
724
        self._mode_check_done = True
 
725
        try:
 
726
            st = self.transport.stat('.')
 
727
        except brz_errors.TransportNotPossible:
 
728
            self._dir_mode = None
 
729
            self._file_mode = None
 
730
        else:
 
731
            # Check the directory mode, but also make sure the created
 
732
            # directories and files are read-write for this user. This is
 
733
            # mostly a workaround for filesystems which lie about being able to
 
734
            # write to a directory (cygwin & win32)
 
735
            if (st.st_mode & 0o7777 == 0o0000):
 
736
                # FTP allows stat but does not return dir/file modes
 
737
                self._dir_mode = None
 
738
                self._file_mode = None
 
739
            else:
 
740
                self._dir_mode = (st.st_mode & 0o7777) | 0o0700
 
741
                # Remove the sticky and execute bits for files
 
742
                self._file_mode = self._dir_mode & ~0o7111
 
743
 
 
744
    def _get_file_mode(self):
 
745
        """Return Unix mode for newly created files, or None.
 
746
        """
 
747
        if not self._mode_check_done:
 
748
            self._find_creation_modes()
 
749
        return self._file_mode
 
750
 
 
751
    def _get_dir_mode(self):
 
752
        """Return Unix mode for newly created directories, or None.
 
753
        """
 
754
        if not self._mode_check_done:
 
755
            self._find_creation_modes()
 
756
        return self._dir_mode
 
757
 
 
758
    def get_refs_container(self):
 
759
        return self._git.refs
 
760
 
 
761
    def get_peeled(self, ref):
 
762
        return self._git.get_peeled(ref)
 
763
 
 
764
    def _find_commondir(self):
 
765
        try:
 
766
            commondir = self.control_transport.get_bytes('commondir')
 
767
        except brz_errors.NoSuchFile:
 
768
            return self
 
769
        else:
 
770
            commondir = commondir.rstrip(b'/.git/').decode(osutils._fs_enc)
 
771
            return ControlDir.open_from_transport(
 
772
                get_transport_from_path(commondir))