+
−from dulwich.client import SubprocessWrapper
+
−import subprocess
+
−
+
−from mercurial import (
+
−    util,
+
−)
+
−
+
−class SSHVendor(object):
+
−    """Parent class for ui-linked Vendor classes."""
+
−
+
−
+
−def generate_ssh_vendor(ui):
+
−    """
+
−    Allows dulwich to use hg's ui.ssh config. The dulwich.client.get_ssh_vendor
+
−    property should point to the return value.
+
−    """
+
−
+
−    class _Vendor(SSHVendor):
+
−        def run_command(self, host, command, username=None, port=None):
+
−            if isinstance(command, basestring):
+
−                # 0.12.x dulwich sends the raw string
+
−                command = [command]
+
−            elif len(command) > 1:
+
−                # 0.11.x dulwich sends an array of [command arg1 arg2 ...], so
+
−                # we detect that here and reformat it back to what hg-git
+
−                # expects (e.g. "command 'arg1 arg2'")
+
−                command = ["%s '%s'" % (command[0], ' '.join(command[1:]))]
+
−            sshcmd = ui.config("ui", "ssh", "ssh")
+
−            args = util.sshargs(sshcmd, host, username, port)
+
−            cmd = '%s %s %s' % (sshcmd, args,
+
−                                util.shellquote(' '.join(command)))
+
−            ui.debug('calling ssh: %s\n' % cmd)
+
−            proc = subprocess.Popen(util.quotecommand(cmd), shell=True,
+
−                                    stdin=subprocess.PIPE,
+
−                                    stdout=subprocess.PIPE)
+
−            return SubprocessWrapper(proc)
+
−
+
−    return _Vendor