proto.py
167 lines
| 6.2 KiB
| text/x-python
|
PythonLexer
|
r15168 | # Copyright 2011 Fog Creek Software | ||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
import os | ||||
import urllib2 | ||||
from mercurial import error, httprepo, util, wireproto | ||||
from mercurial.i18n import _ | ||||
import lfutil | ||||
|
r15255 | LARGEFILES_REQUIRED_MSG = ('\nThis repository uses the largefiles extension.' | ||
'\n\nPlease enable it in your Mercurial config ' | ||||
'file.\n') | ||||
|
r15168 | |||
def putlfile(repo, proto, sha): | ||||
|
r15317 | '''Put a largefile into a repository's local store and into the | ||
user cache.''' | ||||
|
r15168 | proto.redirect() | ||
|
r15391 | |||
fd, tmpname = lfutil.mkstemp(repo, prefix='hg-putlfile') | ||||
tmpfp = os.fdopen(fd, 'wb+') | ||||
|
r15168 | try: | ||
try: | ||||
|
r15391 | proto.getfile(tmpfp) | ||
tmpfp.seek(0) | ||||
if sha != lfutil.hexsha1(tmpfp): | ||||
|
r15778 | raise IOError(0, _('largefile contents do not match hash')) | ||
|
r15391 | tmpfp.close() | ||
lfutil.copytostoreabsolute(repo, tmpname, sha) | ||||
except IOError, e: | ||||
|
r15778 | repo.ui.warn(_('largefiles: failed to put %s into store: %s') % | ||
(sha, e.strerror)) | ||||
|
r15168 | return wireproto.pushres(1) | ||
finally: | ||||
|
r15391 | tmpfp.close() | ||
os.unlink(tmpname) | ||||
|
r15168 | |||
return wireproto.pushres(0) | ||||
def getlfile(repo, proto, sha): | ||||
|
r15252 | '''Retrieve a largefile from the repository-local cache or system | ||
cache.''' | ||||
|
r15168 | filename = lfutil.findfile(repo, sha) | ||
if not filename: | ||||
raise util.Abort(_('requested largefile %s not present in cache') % sha) | ||||
f = open(filename, 'rb') | ||||
length = os.fstat(f.fileno())[6] | ||||
|
r15252 | |||
# Since we can't set an HTTP content-length header here, and | ||||
# Mercurial core provides no way to give the length of a streamres | ||||
# (and reading the entire file into RAM would be ill-advised), we | ||||
# just send the length on the first line of the response, like the | ||||
# ssh proto does for string responses. | ||||
|
r15168 | def generator(): | ||
yield '%d\n' % length | ||||
for chunk in f: | ||||
yield chunk | ||||
return wireproto.streamres(generator()) | ||||
def statlfile(repo, proto, sha): | ||||
|
r15252 | '''Return '2\n' if the largefile is missing, '1\n' if it has a | ||
mismatched checksum, or '0\n' if it is in good condition''' | ||||
|
r15168 | filename = lfutil.findfile(repo, sha) | ||
if not filename: | ||||
return '2\n' | ||||
fd = None | ||||
try: | ||||
fd = open(filename, 'rb') | ||||
return lfutil.hexsha1(fd) == sha and '0\n' or '1\n' | ||||
finally: | ||||
if fd: | ||||
fd.close() | ||||
def wirereposetup(ui, repo): | ||||
class lfileswirerepository(repo.__class__): | ||||
def putlfile(self, sha, fd): | ||||
# unfortunately, httprepository._callpush tries to convert its | ||||
# input file-like into a bundle before sending it, so we can't use | ||||
# it ... | ||||
if issubclass(self.__class__, httprepo.httprepository): | ||||
|
r15778 | res = None | ||
|
r15168 | try: | ||
|
r15778 | res = self._call('putlfile', data=fd, sha=sha, | ||
headers={'content-type':'application/mercurial-0.1'}) | ||||
d, output = res.split('\n', 1) | ||||
for l in output.splitlines(True): | ||||
self.ui.warn(_('remote: '), l, '\n') | ||||
return int(d) | ||||
|
r15168 | except (ValueError, urllib2.HTTPError): | ||
|
r15778 | self.ui.warn(_('unexpected putlfile response: %s') % res) | ||
|
r15168 | return 1 | ||
# ... but we can't use sshrepository._call because the data= | ||||
# argument won't get sent, and _callpush does exactly what we want | ||||
# in this case: send the data straight through | ||||
else: | ||||
try: | ||||
ret, output = self._callpush("putlfile", fd, sha=sha) | ||||
if ret == "": | ||||
raise error.ResponseError(_('putlfile failed:'), | ||||
output) | ||||
return int(ret) | ||||
except IOError: | ||||
return 1 | ||||
except ValueError: | ||||
raise error.ResponseError( | ||||
_('putlfile failed (unexpected response):'), ret) | ||||
def getlfile(self, sha): | ||||
stream = self._callstream("getlfile", sha=sha) | ||||
length = stream.readline() | ||||
try: | ||||
length = int(length) | ||||
except ValueError: | ||||
|
r15170 | self._abort(error.ResponseError(_("unexpected response:"), | ||
length)) | ||||
|
r15168 | return (length, stream) | ||
def statlfile(self, sha): | ||||
try: | ||||
return int(self._call("statlfile", sha=sha)) | ||||
except (ValueError, urllib2.HTTPError): | ||||
|
r15252 | # If the server returns anything but an integer followed by a | ||
|
r15168 | # newline, newline, it's not speaking our language; if we get | ||
# an HTTP error, we can't be sure the largefile is present; | ||||
|
r15252 | # either way, consider it missing. | ||
|
r15168 | return 2 | ||
repo.__class__ = lfileswirerepository | ||||
# advertise the largefiles=serve capability | ||||
def capabilities(repo, proto): | ||||
return capabilities_orig(repo, proto) + ' largefiles=serve' | ||||
# duplicate what Mercurial's new out-of-band errors mechanism does, because | ||||
# clients old and new alike both handle it well | ||||
def webproto_refuseclient(self, message): | ||||
self.req.header([('Content-Type', 'application/hg-error')]) | ||||
return message | ||||
def sshproto_refuseclient(self, message): | ||||
self.ui.write_err('%s\n-\n' % message) | ||||
self.fout.write('\n') | ||||
self.fout.flush() | ||||
return '' | ||||
def heads(repo, proto): | ||||
if lfutil.islfilesrepo(repo): | ||||
|
r15224 | return wireproto.ooberror(LARGEFILES_REQUIRED_MSG) | ||
|
r15168 | return wireproto.heads(repo, proto) | ||
def sshrepo_callstream(self, cmd, **args): | ||||
if cmd == 'heads' and self.capable('largefiles'): | ||||
cmd = 'lheads' | ||||
if cmd == 'batch' and self.capable('largefiles'): | ||||
args['cmds'] = args['cmds'].replace('heads ', 'lheads ') | ||||
return ssh_oldcallstream(self, cmd, **args) | ||||
def httprepo_callstream(self, cmd, **args): | ||||
if cmd == 'heads' and self.capable('largefiles'): | ||||
cmd = 'lheads' | ||||
if cmd == 'batch' and self.capable('largefiles'): | ||||
args['cmds'] = args['cmds'].replace('heads ', 'lheads ') | ||||
return http_oldcallstream(self, cmd, **args) | ||||