sshrepo.py
133 lines
| 3.9 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 * | ||
|
r1357 | demandload(globals(), "hg os re stat") | ||
|
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 | |||
def readerr(self): | ||||
while 1: | ||||
|
r1357 | size = os.fstat(self.pipee.fileno())[stat.ST_SIZE] | ||
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 | |||
def changegroup(self, nodes): | ||||
n = " ".join(map(hex, nodes)) | ||||
f = self.do_cmd("changegroup", roots=n) | ||||
return self.pipei | ||||
def addchangegroup(self, cg): | ||||
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()) | ||||
return self.pipei.read(l) != "" | ||||