sapling/edenscm/hgext/hggit/_ssh.py
Jun Wu c12e300bb8 codemod: move Python packages to edenscm
Summary:
Move top-level Python packages `mercurial`, `hgext` and `hgdemandimport` to
a new top-level package `edenscm`. This allows the Python packages provided by
the upstream Mercurial to be installed side-by-side.

To maintain compatibility, `edenscm/` gets added to `sys.path` in
`mercurial/__init__.py`.

Reviewed By: phillco, ikostia

Differential Revision: D13853115

fbshipit-source-id: b296b0673dc54c61ef6a591ebc687057ff53b22e
2019-01-28 18:35:41 -08:00

40 lines
1.4 KiB
Python

import subprocess
from dulwich.client import SubprocessWrapper
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, (str, bytes)):
# 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")
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