comparison mercurial/hg.py @ 624:876333a295ff

Add an sshrepository class and hg serve --stdio
author Matt Mackall <mpm@selenic.com>
date Tue, 05 Jul 2005 17:55:22 -0800
parents 314867960a4a
children b6c42714d900
comparison
equal deleted inserted replaced
623:314867960a4a 624:876333a295ff
1590 yield zd.flush() 1590 yield zd.flush()
1591 break 1591 break
1592 yield zd.decompress(d) 1592 yield zd.decompress(d)
1593 self.ui.note("%d bytes of data transfered\n" % bytes) 1593 self.ui.note("%d bytes of data transfered\n" % bytes)
1594 1594
1595 class sshrepository:
1596 def __init__(self, ui, path):
1597 self.url = path
1598 self.ui = ui
1599
1600 m = re.match(r'ssh://(([^@]+)@)?([^:/]+)(:(\d+))?(/(.*))?', path)
1601 if not m:
1602 raise RepoError("couldn't parse destination %s\n" % path)
1603
1604 self.user = m.group(2)
1605 self.host = m.group(3)
1606 self.port = m.group(5)
1607 self.path = m.group(7)
1608
1609 args = self.user and ("%s@%s" % (self.user, self.host)) or self.host
1610 args = self.port and ("%s -p %s") % (args, self.port) or args
1611 path = self.path or ""
1612
1613 cmd = "ssh %s 'hg -R %s serve --stdio'"
1614 cmd = cmd % (args, path)
1615
1616 self.pipeo, self.pipei = os.popen2(cmd)
1617
1618 def __del__(self):
1619 self.pipeo.close()
1620 self.pipei.close()
1621
1622 def do_cmd(self, cmd, **args):
1623 self.ui.debug("sending %s command\n" % cmd)
1624 self.pipeo.write("%s\n" % cmd)
1625 for k, v in args.items():
1626 self.pipeo.write("%s %d\n" % (k, len(v)))
1627 self.pipeo.write(v)
1628 self.pipeo.flush()
1629
1630 return self.pipei
1631
1632 def call(self, cmd, **args):
1633 r = self.do_cmd(cmd, **args)
1634 l = int(r.readline())
1635 return r.read(l)
1636
1637 def heads(self):
1638 d = self.call("heads")
1639 try:
1640 return map(bin, d[:-1].split(" "))
1641 except:
1642 self.ui.warn("unexpected response:\n" + d[:400] + "\n...\n")
1643 raise
1644
1645 def branches(self, nodes):
1646 n = " ".join(map(hex, nodes))
1647 d = self.call("branches", nodes=n)
1648 try:
1649 br = [ tuple(map(bin, b.split(" "))) for b in d.splitlines() ]
1650 return br
1651 except:
1652 self.ui.warn("unexpected response:\n" + d[:400] + "\n...\n")
1653 raise
1654
1655 def between(self, pairs):
1656 n = "\n".join(["-".join(map(hex, p)) for p in pairs])
1657 d = self.call("between", pairs=n)
1658 try:
1659 p = [ l and map(bin, l.split(" ")) or [] for l in d.splitlines() ]
1660 return p
1661 except:
1662 self.ui.warn("unexpected response:\n" + d[:400] + "\n...\n")
1663 raise
1664
1665 def changegroup(self, nodes):
1666 n = " ".join(map(hex, nodes))
1667 f = self.do_cmd("changegroup", roots=n)
1668 bytes = 0
1669 while 1:
1670 l = struct.unpack(">l", f.read(4))[0]
1671 if l == -1: break
1672 d = f.read(l)
1673 bytes += len(d)
1674 yield d
1675 self.ui.note("%d bytes of data transfered\n" % bytes)
1676
1595 def repository(ui, path=None, create=0): 1677 def repository(ui, path=None, create=0):
1596 if path: 1678 if path:
1597 if path.startswith("http://"): 1679 if path.startswith("http://"):
1598 return httprepository(ui, path) 1680 return httprepository(ui, path)
1599 if path.startswith("hg://"): 1681 if path.startswith("hg://"):
1600 return httprepository(ui, path.replace("hg://", "http://")) 1682 return httprepository(ui, path.replace("hg://", "http://"))
1601 if path.startswith("old-http://"): 1683 if path.startswith("old-http://"):
1602 return localrepository(ui, path.replace("old-http://", "http://")) 1684 return localrepository(ui, path.replace("old-http://", "http://"))
1685 if path.startswith("ssh://"):
1686 return sshrepository(ui, path)
1603 1687
1604 return localrepository(ui, path, create) 1688 return localrepository(ui, path, create)