14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
"""An adapter between a Git control dir and a Bazaar BzrDir"""
17
"""An adapter between a Git control dir and a Bazaar BzrDir."""
19
22
from bzrlib.lazy_import import lazy_import
20
23
from bzrlib import (
26
29
lazy_import(globals(), """
30
from bzrlib.lockable_files import (
27
33
from bzrlib.plugins.git import (
42
from bzrlib.plugins.git import LocalGitBzrDirFormat
35
46
class GitLock(object):
36
47
"""A lock that thunks through to Git."""
49
def lock_write(self, token=None):
41
52
def lock_read(self):
61
def validate_token(self, token):
51
65
class GitLockableFiles(lockable_files.LockableFiles):
52
66
"""Git specific lockable files abstraction."""
54
def __init__(self, lock):
68
def __init__(self, transport, lock):
56
70
self._transaction = None
57
71
self._lock_mode = None
72
self._transport = transport
58
73
self._lock_count = 0
61
76
class GitDir(bzrdir.BzrDir):
62
77
"""An adapter to the '.git' dir used by git."""
64
_gitrepository_class = git_repository.GitRepository
66
def __init__(self, transport, lockfiles, format):
79
def is_supported(self):
82
def cloning_metadir(self, stacked=False):
83
return get_rich_root_format()
86
class LocalGitDir(GitDir):
87
"""An adapter to the '.git' dir used by git."""
89
_gitrepository_class = repository.LocalGitRepository
91
def __init__(self, transport, lockfiles, gitrepo, format):
67
92
self._format = format
68
93
self.root_transport = transport
69
self.transport = transport.clone('.git')
96
self.transport = transport
98
self.transport = transport.clone('.git')
70
99
self._lockfiles = lockfiles
72
101
def get_branch_transport(self, branch_format):
73
102
if branch_format is None:
74
103
return self.transport
75
if isinstance(branch_format, GitBzrDirFormat):
104
if isinstance(branch_format, LocalGitBzrDirFormat):
76
105
return self.transport
77
106
raise errors.bzr_errors.IncompatibleFormat(branch_format, self._format)
79
108
get_repository_transport = get_branch_transport
80
109
get_workingtree_transport = get_branch_transport
82
def is_supported(self):
85
111
def open_branch(self, ignored=None):
86
112
"""'create' a branch for this dir."""
87
113
repo = self.open_repository()
88
if repo._git.heads == []:
91
head = repo._git.heads[0].commit.id
92
return git_branch.GitBranch(self, repo, head,
93
self.root_transport.base, self._lockfiles)
114
return branch.LocalGitBranch(self, repo, "HEAD", repo._git.head(),
95
117
def open_repository(self, shared=False):
96
118
"""'open' a repository for this dir."""
99
121
def open_workingtree(self, recommend_upgrade=True):
100
122
loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
101
123
raise errors.bzr_errors.NoWorkingTree(loc)
103
def cloning_metadir(self):
104
return bzrdir.BzrDirFormat.get_default_format()
107
class GitBzrDirFormat(bzrdir.BzrDirFormat):
108
"""The .git directory control format."""
110
_gitdir_class = GitDir
113
def _known_formats(self):
114
return set([GitBzrDirFormat()])
116
def open(self, transport, _create=False, _found=None):
117
"""Open this directory.
119
:param _create: create the git dir on the fly. private to GitDirFormat.
121
# we dont grok readonly - git isn't integrated with transport.
123
if url.startswith('readonly+'):
124
url = url[len('readonly+'):]
125
if not transport.has('.git'):
126
raise errors.bzr_errors.NotBranchError(path=transport.base)
127
lockfiles = GitLockableFiles(GitLock())
128
return self._gitdir_class(transport, lockfiles, self)
131
def probe_transport(klass, transport):
132
"""Our format is present if the transport ends in '.not/'."""
133
# little ugly, but works
135
# delegate to the main opening code. This pays a double rtt cost at the
136
# moment, so perhaps we want probe_transport to return the opened thing
137
# rather than an openener ? or we could return a curried thing with the
138
# dir to open already instantiated ? Needs more thought.
140
format.open(transport)
143
raise errors.bzr_errors.NotBranchError(path=transport.base)
144
raise errors.bzr_errors.NotBranchError(path=transport.base)
146
def get_format_description(self):
147
return "Local Git Repository"
150
bzrdir.BzrDirFormat.register_control_format(GitBzrDirFormat)
124
if (not self._git.bare and
125
os.path.exists(os.path.join(self._git.controldir(), "index"))):
126
return workingtree.GitWorkingTree(self, self.open_repository(),
129
def create_repository(self, shared=False):
130
return self.open_repository()