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"""
21
from bzrlib.lazy_import import lazy_import
22
22
from bzrlib import (
28
LockWarner = getattr(lockable_files, "_LockWarner", None)
28
lazy_import(globals(), """
30
29
from bzrlib.plugins.git import (
40
38
class GitLock(object):
55
def validate_token(self, token):
59
54
class GitLockableFiles(lockable_files.LockableFiles):
60
55
"""Git specific lockable files abstraction."""
62
def __init__(self, transport, lock):
57
def __init__(self, lock):
64
59
self._transaction = None
65
60
self._lock_mode = None
66
self._transport = transport
67
if LockWarner is None:
71
self._lock_warner = LockWarner(repr(self))
74
64
class GitDir(bzrdir.BzrDir):
75
65
"""An adapter to the '.git' dir used by git."""
77
def is_supported(self):
80
def cloning_metadir(self, stacked=False):
81
return get_rich_root_format(stacked)
84
class LocalGitDir(GitDir):
85
"""An adapter to the '.git' dir used by git."""
87
_gitrepository_class = repository.LocalGitRepository
67
_gitrepository_class = git_repository.GitRepository
89
69
def __init__(self, transport, lockfiles, gitrepo, format):
90
70
self._format = format
96
76
self.transport = transport.clone('.git')
97
77
self._lockfiles = lockfiles
98
self._mode_check_done = None
100
def is_control_filename(self, filename):
101
return filename == '.git' or filename.startswith('.git/')
103
79
def get_branch_transport(self, branch_format):
104
80
if branch_format is None:
105
81
return self.transport
106
if isinstance(branch_format, LocalGitBzrDirFormat):
82
if isinstance(branch_format, GitBzrDirFormat):
107
83
return self.transport
108
84
raise errors.bzr_errors.IncompatibleFormat(branch_format, self._format)
110
86
get_repository_transport = get_branch_transport
111
87
get_workingtree_transport = get_branch_transport
113
def open_branch(self, ignore_fallbacks=None):
89
def is_supported(self):
92
def open_branch(self, ignored=None):
114
93
"""'create' a branch for this dir."""
115
94
repo = self.open_repository()
116
return branch.LocalGitBranch(self, repo, "HEAD", self._lockfiles)
95
if repo._git.heads == []:
98
head = repo._git.heads[0].commit.id
99
return git_branch.GitBranch(self, repo, head,
100
self.root_transport.base, self._lockfiles)
118
102
def open_repository(self, shared=False):
119
103
"""'open' a repository for this dir."""
120
104
return self._gitrepository_class(self, self._lockfiles)
122
106
def open_workingtree(self, recommend_upgrade=True):
123
if (not self._git.bare and
124
os.path.exists(os.path.join(self._git.controldir(), "index"))):
125
return workingtree.GitWorkingTree(self, self.open_repository(),
108
loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
109
raise errors.bzr_errors.NoWorkingTree(loc)
111
return git_workingtree.GitWorkingTree(self, self.open_repository(),
126
112
self.open_branch())
127
loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
128
raise errors.bzr_errors.NoWorkingTree(loc)
130
def create_repository(self, shared=False):
131
return self.open_repository()
133
def create_branch(self):
134
return self.open_branch()
114
def cloning_metadir(self):
115
return bzrdir.BzrDirFormat.get_default_format()
118
class GitBzrDirFormat(bzrdir.BzrDirFormat):
119
"""The .git directory control format."""
121
_gitdir_class = GitDir
124
def _known_formats(self):
125
return set([GitBzrDirFormat()])
127
def open(self, transport, _found=None):
128
"""Open this directory.
131
# we dont grok readonly - git isn't integrated with transport.
133
if url.startswith('readonly+'):
134
url = url[len('readonly+'):]
137
gitrepo = git.repo.Repo(transport.local_abspath("."))
138
except errors.bzr_errors.NotLocalUrl:
139
raise errors.bzr_errors.NotBranchError(path=transport.base)
140
lockfiles = GitLockableFiles(GitLock())
141
return self._gitdir_class(transport, lockfiles, gitrepo, self)
144
def probe_transport(klass, transport):
145
"""Our format is present if the transport ends in '.not/'."""
146
# little ugly, but works
148
# delegate to the main opening code. This pays a double rtt cost at the
149
# moment, so perhaps we want probe_transport to return the opened thing
150
# rather than an openener ? or we could return a curried thing with the
151
# dir to open already instantiated ? Needs more thought.
153
format.open(transport)
156
raise errors.bzr_errors.NotBranchError(path=transport.base)
157
raise errors.bzr_errors.NotBranchError(path=transport.base)
159
def get_format_description(self):
160
return "Local Git Repository"
163
bzrdir.BzrDirFormat.register_control_format(GitBzrDirFormat)