proto.py
158 lines
| 5.9 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 | ||||
|
r17192 | from mercurial import error, httppeer, util, wireproto | ||
|
r17127 | from mercurial.wireproto import batchable, future | ||
|
r15168 | 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 | |||
|
r16594 | path = lfutil.storepath(repo, sha) | ||
util.makedirs(os.path.dirname(path)) | ||||
tmpfp = util.atomictempfile(path, createmode=repo.store.createmode) | ||||
|
r15168 | try: | ||
try: | ||||
|
r15391 | proto.getfile(tmpfp) | ||
|
r16155 | tmpfp._fp.seek(0) | ||
if sha != lfutil.hexsha1(tmpfp._fp): | ||||
|
r15778 | raise IOError(0, _('largefile contents do not match hash')) | ||
|
r15391 | tmpfp.close() | ||
|
r16155 | lfutil.linktousercache(repo, sha) | ||
|
r15391 | except IOError, e: | ||
|
r15778 | repo.ui.warn(_('largefiles: failed to put %s into store: %s') % | ||
(sha, e.strerror)) | ||||
|
r15168 | return wireproto.pushres(1) | ||
finally: | ||||
|
r16155 | tmpfp.discard() | ||
|
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): | ||||
|
r18488 | '''Return '2\n' if the largefile is missing, '0\n' if it seems to be in | ||
good condition. | ||||
The value 1 is reserved for mismatched checksum, but that is too expensive | ||||
to be verified on every stat and must be caught be running 'hg verify' | ||||
server side.''' | ||||
|
r15168 | filename = lfutil.findfile(repo, sha) | ||
if not filename: | ||||
return '2\n' | ||||
|
r18488 | return '0\n' | ||
|
r15168 | |||
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 ... | ||||
|
r17192 | if issubclass(self.__class__, httppeer.httppeer): | ||
|
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) | ||
|
r17127 | @batchable | ||
|
r15168 | def statlfile(self, sha): | ||
|
r17127 | f = future() | ||
result = {'sha': sha} | ||||
yield result, f | ||||
|
r15168 | try: | ||
|
r17127 | yield int(f.value) | ||
|
r15168 | 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. | ||
|
r17127 | yield 2 | ||
|
r15168 | |||
repo.__class__ = lfileswirerepository | ||||
# advertise the largefiles=serve capability | ||||
def capabilities(repo, proto): | ||||
|
r16247 | return capabilitiesorig(repo, proto) + ' largefiles=serve' | ||
|
r15168 | |||
def heads(repo, proto): | ||||
if lfutil.islfilesrepo(repo): | ||||
|
r15224 | return wireproto.ooberror(LARGEFILES_REQUIRED_MSG) | ||
|
r15168 | return wireproto.heads(repo, proto) | ||
|
r16247 | def sshrepocallstream(self, cmd, **args): | ||
|
r15168 | if cmd == 'heads' and self.capable('largefiles'): | ||
cmd = 'lheads' | ||||
if cmd == 'batch' and self.capable('largefiles'): | ||||
args['cmds'] = args['cmds'].replace('heads ', 'lheads ') | ||||
|
r16247 | return ssholdcallstream(self, cmd, **args) | ||
|
r15168 | |||
|
r16247 | def httprepocallstream(self, cmd, **args): | ||
|
r15168 | if cmd == 'heads' and self.capable('largefiles'): | ||
cmd = 'lheads' | ||||
if cmd == 'batch' and self.capable('largefiles'): | ||||
args['cmds'] = args['cmds'].replace('heads ', 'lheads ') | ||||
|
r16247 | return httpoldcallstream(self, cmd, **args) | ||