/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: 2019-02-04 01:01:24 UTC
  • mto: This revision was merged to the branch mainline in revision 7268.
  • Revision ID: jelmer@jelmer.uk-20190204010124-ni0i4qc6f5tnbvux
Fix source tests.

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