17
17
"""Server-side bzrdir related request implmentations."""
19
from __future__ import absolute_import
20
from bzrlib import branch, errors, repository, urlutils
21
28
from bzrlib.bzrdir import (
27
33
from bzrlib.controldir import (
86
92
class SmartServerRequestBzrDir(SmartServerRequest):
88
94
def do(self, path, *args):
89
"""Open a BzrDir at path, and return self.do_bzrdir_request(*args)."""
95
"""Open a BzrDir at path, and return `self.do_bzrdir_request(*args)`."""
91
97
self._bzrdir = BzrDir.open_from_transport(
92
98
self.transport_from_client_path(path))
121
127
return '/'.join(segments)
130
class SmartServerBzrDirRequestDestroyBranch(SmartServerRequestBzrDir):
132
def do_bzrdir_request(self, name=None):
133
"""Destroy the branch with the specified name.
136
:return: On success, 'ok'.
139
self._bzrdir.destroy_branch(name)
140
except errors.NotBranchError, e:
141
return FailedSmartServerResponse(('nobranch',))
142
return SuccessfulSmartServerResponse(('ok',))
145
class SmartServerBzrDirRequestHasWorkingTree(SmartServerRequestBzrDir):
147
def do_bzrdir_request(self, name=None):
148
"""Check whether there is a working tree present.
152
:return: If there is a working tree present, 'yes'.
155
if self._bzrdir.has_workingtree():
156
return SuccessfulSmartServerResponse(('yes', ))
158
return SuccessfulSmartServerResponse(('no', ))
161
class SmartServerBzrDirRequestDestroyRepository(SmartServerRequestBzrDir):
163
def do_bzrdir_request(self, name=None):
164
"""Destroy the repository.
168
:return: On success, 'ok'.
171
self._bzrdir.destroy_repository()
172
except errors.NoRepositoryPresent, e:
173
return FailedSmartServerResponse(('norepository',))
174
return SuccessfulSmartServerResponse(('ok',))
124
177
class SmartServerBzrDirRequestCloningMetaDir(SmartServerRequestBzrDir):
126
179
def do_bzrdir_request(self, require_stacking):
150
203
control_format = self._bzrdir.cloning_metadir(
151
204
require_stacking=require_stacking)
152
205
control_name = control_format.network_name()
153
# XXX: There should be a method that tells us that the format does/does
154
# not have subformats.
155
if isinstance(control_format, BzrDirMetaFormat1):
206
if not control_format.fixed_components:
156
207
branch_name = ('branch',
157
208
control_format.get_branch_format().network_name())
158
209
repository_name = control_format.repository_format.network_name()
218
class SmartServerBzrDirRequestCheckoutMetaDir(SmartServerRequestBzrDir):
219
"""Get the format to use for checkouts.
223
:return: on success, a 3-tuple of network names for (control,
224
repository, branch) directories, where '' signifies "not present".
225
If this BzrDir contains a branch reference then this will fail with
226
BranchReference; clients should resolve branch references before
227
calling this RPC (they should not try to create a checkout of a
231
def do_bzrdir_request(self):
233
branch_ref = self._bzrdir.get_branch_reference()
234
except errors.NotBranchError:
236
if branch_ref is not None:
237
# The server shouldn't try to resolve references, and it quite
238
# possibly can't reach them anyway. The client needs to resolve
239
# the branch reference to determine the cloning_metadir.
240
return FailedSmartServerResponse(('BranchReference',))
241
control_format = self._bzrdir.checkout_metadir()
242
control_name = control_format.network_name()
243
if not control_format.fixed_components:
244
branch_name = control_format.get_branch_format().network_name()
245
repo_name = control_format.repository_format.network_name()
249
return SuccessfulSmartServerResponse(
250
(control_name, repo_name, branch_name))
167
253
class SmartServerRequestCreateBranch(SmartServerRequestBzrDir):
169
255
def do(self, path, network_name):
182
268
:param path: The path to the bzrdir.
183
269
:param network_name: The network name of the branch type to create.
184
:return: (ok, network_name)
270
:return: ('ok', branch_format, repo_path, rich_root, tree_ref,
271
external_lookup, repo_format)
186
273
bzrdir = BzrDir.open_from_transport(
187
274
self.transport_from_client_path(path))
188
275
format = branch.network_format_registry.get(network_name)
189
276
bzrdir.branch_format = format
190
result = format.initialize(bzrdir)
277
result = format.initialize(bzrdir, name="")
191
278
rich_root, tree_ref, external_lookup = self._format_to_capabilities(
192
279
result.repository._format)
193
280
branch_format = result._format.network_name()
344
431
return SuccessfulSmartServerResponse((), content)
434
class SmartServerBzrDirRequestGetBranches(SmartServerRequestBzrDir):
436
def do_bzrdir_request(self):
437
"""Get the branches in a control directory.
439
The body is a bencoded dictionary, with values similar to the return
440
value of the open branch request.
442
branches = self._bzrdir.get_branches()
444
for name, b in branches.iteritems():
447
ret[name] = ("branch", b._format.network_name())
448
return SuccessfulSmartServerResponse(
449
("success", ), bencode.bencode(ret))
347
452
class SmartServerRequestInitializeBzrDir(SmartServerRequest):
349
454
def do(self, path):