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
"""Server-side branch related request implmentations."""
20
from bzrlib import errors
21
from bzrlib.bzrdir import BzrDir
22
from bzrlib.smart.request import (
23
FailedSmartServerResponse,
25
SuccessfulSmartServerResponse,
29
class SmartServerBranchRequest(SmartServerRequest):
30
"""Base class for handling common branch request logic.
33
def do(self, path, *args):
34
"""Execute a request for a branch at path.
36
All Branch requests take a path to the branch as their first argument.
38
If the branch is a branch reference, NotBranchError is raised.
40
:param path: The path for the repository as received from the
42
:return: A SmartServerResponse from self.do_with_branch().
44
transport = self.transport_from_client_path(path)
45
bzrdir = BzrDir.open_from_transport(transport)
46
if bzrdir.get_branch_reference() is not None:
47
raise errors.NotBranchError(transport.base)
48
branch = bzrdir.open_branch()
49
return self.do_with_branch(branch, *args)
52
class SmartServerLockedBranchRequest(SmartServerBranchRequest):
53
"""Base class for handling common branch request logic for requests that
57
def do_with_branch(self, branch, branch_token, repo_token, *args):
58
"""Execute a request for a branch.
60
A write lock will be acquired with the given tokens for the branch and
61
repository locks. The lock will be released once the request is
62
processed. The physical lock state won't be changed.
64
# XXX: write a test for LockContention
65
branch.repository.lock_write(token=repo_token)
67
branch.lock_write(token=branch_token)
69
return self.do_with_locked_branch(branch, *args)
73
branch.repository.unlock()
76
class SmartServerBranchGetConfigFile(SmartServerBranchRequest):
78
def do_with_branch(self, branch):
79
"""Return the content of branch.control_files.get('branch.conf').
81
The body is not utf8 decoded - its the literal bytestream from disk.
84
content = branch.control_files.get('branch.conf').read()
85
except errors.NoSuchFile:
87
return SuccessfulSmartServerResponse( ('ok', ), content)
90
class SmartServerRequestRevisionHistory(SmartServerBranchRequest):
92
def do_with_branch(self, branch):
93
"""Get the revision history for the branch.
95
The revision list is returned as the body content,
96
with each revision utf8 encoded and \x00 joined.
98
return SuccessfulSmartServerResponse(
99
('ok', ), ('\x00'.join(branch.revision_history())))
102
class SmartServerBranchRequestLastRevisionInfo(SmartServerBranchRequest):
104
def do_with_branch(self, branch):
105
"""Return branch.last_revision_info().
107
The revno is encoded in decimal, the revision_id is encoded as utf8.
109
revno, last_revision = branch.last_revision_info()
110
return SuccessfulSmartServerResponse(('ok', str(revno), last_revision))
113
class SmartServerBranchRequestSetLastRevision(SmartServerLockedBranchRequest):
115
def do_with_locked_branch(self, branch, new_last_revision_id):
116
if new_last_revision_id == 'null:':
117
branch.set_revision_history([])
119
if not branch.repository.has_revision(new_last_revision_id):
120
return FailedSmartServerResponse(
121
('NoSuchRevision', new_last_revision_id))
122
branch.generate_revision_history(new_last_revision_id)
123
return SuccessfulSmartServerResponse(('ok',))
126
class SmartServerBranchRequestSetLastRevisionInfo(
127
SmartServerLockedBranchRequest):
128
"""Branch.set_last_revision_info. Sets the revno and the revision ID of
129
the specified branch.
134
def do_with_locked_branch(self, branch, new_revno, new_last_revision_id):
135
branch.set_last_revision_info(int(new_revno), new_last_revision_id)
136
return SuccessfulSmartServerResponse(('ok',))
139
class SmartServerBranchRequestLockWrite(SmartServerBranchRequest):
141
def do_with_branch(self, branch, branch_token='', repo_token=''):
142
if branch_token == '':
147
repo_token = branch.repository.lock_write(token=repo_token)
149
branch_token = branch.lock_write(token=branch_token)
151
# this leaves the repository with 1 lock
152
branch.repository.unlock()
153
except errors.LockContention:
154
return FailedSmartServerResponse(('LockContention',))
155
except errors.TokenMismatch:
156
return FailedSmartServerResponse(('TokenMismatch',))
157
except errors.UnlockableTransport:
158
return FailedSmartServerResponse(('UnlockableTransport',))
159
except errors.LockFailed, e:
160
return FailedSmartServerResponse(('LockFailed', str(e.lock), str(e.why)))
161
if repo_token is None:
164
branch.repository.leave_lock_in_place()
165
branch.leave_lock_in_place()
167
return SuccessfulSmartServerResponse(('ok', branch_token, repo_token))
170
class SmartServerBranchRequestUnlock(SmartServerBranchRequest):
172
def do_with_branch(self, branch, branch_token, repo_token):
174
branch.repository.lock_write(token=repo_token)
176
branch.lock_write(token=branch_token)
178
branch.repository.unlock()
179
except errors.TokenMismatch:
180
return FailedSmartServerResponse(('TokenMismatch',))
182
branch.repository.dont_leave_lock_in_place()
183
branch.dont_leave_lock_in_place()
185
return SuccessfulSmartServerResponse(('ok',))
added
removed
bzrlib/smart/branch.py
