sshrepo.py
155 lines
| 4.5 KiB
| text/x-python
|
PythonLexer
/ mercurial / sshrepo.py
|
r1096 | # sshrepo.py - ssh repository proxy class for mercurial | ||
|
r1089 | # | ||
# Copyright 2005 Matt Mackall <mpm@selenic.com> | ||||
# | ||||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
|
r1103 | from node import * | ||
from remoterepo import * | ||||
|
r1400 | from i18n import gettext as _ | ||
|
r1251 | from demandload import * | ||
|
r2177 | demandload(globals(), "hg os re stat util") | ||
|
r1089 | |||
class sshrepository(remoterepository): | ||||
def __init__(self, ui, path): | ||||
self.url = path | ||||
self.ui = ui | ||||
m = re.match(r'ssh://(([^@]+)@)?([^:/]+)(:(\d+))?(/(.*))?', path) | ||||
if not m: | ||||
|
r1402 | raise hg.RepoError(_("couldn't parse destination %s") % path) | ||
|
r1089 | |||
self.user = m.group(2) | ||||
self.host = m.group(3) | ||||
self.port = m.group(5) | ||||
self.path = m.group(7) or "." | ||||
args = self.user and ("%s@%s" % (self.user, self.host)) or self.host | ||||
args = self.port and ("%s -p %s") % (args, self.port) or args | ||||
sshcmd = self.ui.config("ui", "ssh", "ssh") | ||||
remotecmd = self.ui.config("ui", "remotecmd", "hg") | ||||
|
r1330 | cmd = '%s %s "%s -R %s serve --stdio"' | ||
|
r1089 | cmd = cmd % (sshcmd, args, remotecmd, self.path) | ||
|
r1332 | ui.note('running %s\n' % cmd) | ||
|
r1330 | self.pipeo, self.pipei, self.pipee = os.popen3(cmd, 'b') | ||
|
r1089 | |||
|
r2028 | # skip any noise generated by remote shell | ||
r = self.do_cmd("between", pairs=("%s-%s" % ("0"*40, "0"*40))) | ||||
l1 = "" | ||||
|
r2040 | l2 = "dummy" | ||
|
r2046 | max_noise = 500 | ||
|
r2040 | while l2 and max_noise: | ||
|
r2028 | l2 = r.readline() | ||
self.readerr() | ||||
if l1 == "1\n" and l2 == "\n": | ||||
break | ||||
|
r2040 | if l1: | ||
|
r2046 | ui.debug(_("remote: "), l1) | ||
|
r2028 | l1 = l2 | ||
|
r2040 | max_noise -= 1 | ||
else: | ||||
if l1: | ||||
|
r2046 | ui.debug(_("remote: "), l1) | ||
|
r2040 | raise hg.RepoError(_("no response from remote hg")) | ||
|
r2028 | |||
|
r1089 | def readerr(self): | ||
while 1: | ||||
|
r2176 | size = util.fstat(self.pipee).st_size | ||
|
r1357 | if size == 0: break | ||
|
r1089 | l = self.pipee.readline() | ||
if not l: break | ||||
|
r1402 | self.ui.status(_("remote: "), l) | ||
|
r1089 | |||
def __del__(self): | ||||
try: | ||||
self.pipeo.close() | ||||
self.pipei.close() | ||||
|
r1358 | # read the error descriptor until EOF | ||
for l in self.pipee: | ||||
|
r1402 | self.ui.status(_("remote: "), l) | ||
|
r1089 | self.pipee.close() | ||
except: | ||||
pass | ||||
def dev(self): | ||||
return -1 | ||||
def do_cmd(self, cmd, **args): | ||||
|
r1402 | self.ui.debug(_("sending %s command\n") % cmd) | ||
|
r1089 | self.pipeo.write("%s\n" % cmd) | ||
for k, v in args.items(): | ||||
self.pipeo.write("%s %d\n" % (k, len(v))) | ||||
self.pipeo.write(v) | ||||
self.pipeo.flush() | ||||
return self.pipei | ||||
def call(self, cmd, **args): | ||||
r = self.do_cmd(cmd, **args) | ||||
l = r.readline() | ||||
self.readerr() | ||||
try: | ||||
l = int(l) | ||||
except: | ||||
|
r1402 | raise hg.RepoError(_("unexpected response '%s'") % l) | ||
|
r1089 | return r.read(l) | ||
def lock(self): | ||||
self.call("lock") | ||||
return remotelock(self) | ||||
def unlock(self): | ||||
self.call("unlock") | ||||
def heads(self): | ||||
d = self.call("heads") | ||||
try: | ||||
return map(bin, d[:-1].split(" ")) | ||||
except: | ||||
|
r1402 | raise hg.RepoError(_("unexpected response '%s'") % (d[:400] + "...")) | ||
|
r1089 | |||
def branches(self, nodes): | ||||
n = " ".join(map(hex, nodes)) | ||||
d = self.call("branches", nodes=n) | ||||
try: | ||||
br = [ tuple(map(bin, b.split(" "))) for b in d.splitlines() ] | ||||
return br | ||||
except: | ||||
|
r1402 | raise hg.RepoError(_("unexpected response '%s'") % (d[:400] + "...")) | ||
|
r1089 | |||
def between(self, pairs): | ||||
n = "\n".join(["-".join(map(hex, p)) for p in pairs]) | ||||
d = self.call("between", pairs=n) | ||||
try: | ||||
p = [ l and map(bin, l.split(" ")) or [] for l in d.splitlines() ] | ||||
return p | ||||
except: | ||||
|
r1402 | raise hg.RepoError(_("unexpected response '%s'") % (d[:400] + "...")) | ||
|
r1089 | |||
|
r1736 | def changegroup(self, nodes, kind): | ||
|
r1089 | n = " ".join(map(hex, nodes)) | ||
f = self.do_cmd("changegroup", roots=n) | ||||
return self.pipei | ||||
|
r2230 | def addchangegroup(self, cg, source): | ||
|
r1089 | d = self.call("addchangegroup") | ||
if d: | ||||
|
r1402 | raise hg.RepoError(_("push refused: %s"), d) | ||
|
r1089 | |||
while 1: | ||||
d = cg.read(4096) | ||||
if not d: break | ||||
self.pipeo.write(d) | ||||
self.readerr() | ||||
self.pipeo.flush() | ||||
self.readerr() | ||||
l = int(self.pipei.readline()) | ||||
|
r2019 | r = self.pipei.read(l) | ||
if not r: | ||||
return 1 | ||||
return int(r) | ||||