1
# Copyright (C) 2006 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
# TODO: At some point, handle upgrades by just passing the whole request
18
# across to run on the server.
21
from bzrlib import bzrdir, branch, errors, repository
22
from bzrlib.bzrdir import BzrDir, BzrDirFormat
23
from bzrlib.branch import Branch, BranchFormat
24
from bzrlib.trace import mutter
25
from bzrlib.transport.smart import SmartTransport
28
class RemoteBzrDirFormat(bzrdir.BzrDirMetaFormat1):
29
"""Format representing bzrdirs accessed via a smart server"""
31
def get_format_description(self):
32
return 'bzr remote bzrdir'
34
def probe_transport(self, transport):
35
## mutter("%r probe for bzrdir in %r" % (self, transport))
37
transport.get_smart_client()
38
except (NotImplementedError, AttributeError,
39
errors.TransportNotPossible):
40
raise errors.NoSmartServer(transport.base)
44
def _open(self, transport):
45
return RemoteBzrDir(transport)
47
def __eq__(self, other):
48
if not isinstance(other, RemoteBzrDirFormat):
50
return self.get_format_description() == other.get_format_description()
53
class RemoteBzrDir(BzrDir):
54
"""Control directory on a remote server, accessed by HPSS."""
56
def __init__(self, transport):
57
BzrDir.__init__(self, transport, RemoteBzrDirFormat())
58
self.client = transport.get_smart_client()
59
# this object holds a delegated bzrdir that uses file-level operations
60
# to talk to the other side
61
# XXX: We should go into find_format, but not allow it to find
62
# RemoteBzrDirFormat and make sure it finds the real underlying format.
64
default_format = BzrDirFormat.get_default_format()
65
self._real_bzrdir = default_format.open(transport, _found=True)
66
self._real_bzrdir._format.probe_transport(transport)
69
def create_repository(self, shared=False):
70
return RemoteRepository(
71
self, self._real_bzrdir.create_repository(shared=shared))
73
def create_branch(self):
74
real_branch = self._real_bzrdir.create_branch()
75
real_repository = real_branch.repository
76
remote_repository = RemoteRepository(self, real_repository)
77
return RemoteBranch(self, remote_repository, real_branch)
79
def create_workingtree(self, revision_id=None):
80
real_workingtree = self._real_bzrdir.create_workingtree(revision_id=revision_id)
81
return RemoteWorkingTree(self, real_workingtree)
83
def open_repository(self):
84
return RemoteRepository(self, self._real_bzrdir.open_repository())
86
def open_branch(self):
87
real_branch = self._real_bzrdir.open_branch()
88
if real_branch.bzrdir is self._real_bzrdir:
89
# This branch accessed through the smart server, so wrap the
91
real_repository = real_branch.repository
92
remote_repository = RemoteRepository(self, real_repository)
93
return RemoteBranch(self, remote_repository, real_branch)
95
# We were redirected to somewhere else, so don't wrap.
98
def open_workingtree(self):
99
return RemoteWorkingTree(self, self._real_bzrdir.open_workingtree())
101
def get_branch_transport(self, branch_format):
102
return self._real_bzrdir.get_branch_transport(branch_format)
104
def get_repository_transport(self, repository_format):
105
return self._real_bzrdir.get_repository_transport(repository_format)
107
def get_workingtree_transport(self, workingtree_format):
108
return self._real_bzrdir.get_workingtree_transport(workingtree_format)
110
def can_convert_format(self):
111
"""Upgrading of remote bzrdirs is not supported yet."""
114
def needs_format_conversion(self, format=None):
115
"""Upgrading of remote bzrdirs is not supported yet."""
119
class RemoteRepositoryFormat(repository.RepositoryFormat):
120
"""Format for repositories accessed over rpc.
122
Instances of this repository are represented by RemoteRepository
126
_matchingbzrdir = RemoteBzrDirFormat
128
def initialize(self, a_bzrdir, shared=False):
129
assert isinstance(a_bzrdir, RemoteBzrDir)
130
return a_bzrdir.create_repository(shared=shared)
132
def open(self, a_bzrdir):
133
assert isinstance(a_bzrdir, RemoteBzrDir)
134
return a_bzrdir.open_repository()
136
def get_format_description(self):
137
return 'bzr remote repository'
139
def __eq__(self, other):
140
return self.get_format_description() == other.get_format_description()
143
class RemoteRepository(object):
144
"""Repository accessed over rpc.
146
For the moment everything is delegated to IO-like operations over
150
def __init__(self, remote_bzrdir, real_repository):
151
self.real_repository = real_repository
152
self.bzrdir = remote_bzrdir
153
self._format = RemoteRepositoryFormat()
155
def __getattr__(self, name):
156
# XXX: temporary way to lazily delegate everything to the real
158
return getattr(self.real_repository, name)
161
class RemoteBranchFormat(branch.BranchFormat):
163
def open(self, a_bzrdir):
164
assert isinstance(a_bzrdir, RemoteBzrDir)
165
return a_bzrdir.open_branch()
167
def initialize(self, a_bzrdir):
168
assert isinstance(a_bzrdir, RemoteBzrDir)
169
return a_bzrdir.create_branch()
172
class RemoteBranch(branch.Branch):
173
"""Branch stored on a server accessed by HPSS RPC.
175
At the moment most operations are mapped down to simple file operations.
178
def __init__(self, remote_bzrdir, remote_repository, real_branch):
179
self.bzrdir = remote_bzrdir
180
self.transport = remote_bzrdir.transport
181
self.repository = remote_repository
182
self._real_branch = real_branch
183
self._format = RemoteBranchFormat()
186
return self._real_branch.lock_read()
188
def lock_write(self):
189
return self._real_branch.lock_write()
192
return self._real_branch.unlock()
194
def break_lock(self):
195
return self._real_branch.break_lock()
197
def revision_history(self):
198
return self._real_branch.revision_history()
200
def set_revision_history(self, rev_history):
201
return self._real_branch.set_revision_history(rev_history)
203
def get_parent(self):
204
return self._real_branch.get_parent()
206
def set_parent(self, url):
207
return self._real_branch.set_parent(url)
210
class RemoteWorkingTree(object):
212
def __init__(self, remote_bzrdir, real_workingtree):
213
self.real_workingtree = real_workingtree
214
self.bzrdir = remote_bzrdir
216
def __getattr__(self, name):
217
# XXX: temporary way to lazily delegate everything to the real
219
return getattr(self.real_workingtree, name)
222
# when first loaded, register this format.
224
# TODO: Actually this needs to be done earlier; we can hold off on loading
225
# this code until it's needed though.
227
# We can't use register_control_format because it adds it at a lower priority
228
# than the existing branches, whereas this should take priority.
229
BzrDirFormat._control_formats.insert(0, RemoteBzrDirFormat())
added
removed
bzrlib/remote.py
