You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.6 KiB
91 lines
2.6 KiB
#!/usr/bin/env python3 |
|
import os |
|
import sys |
|
from subprocess import check_output, list2cmdline |
|
|
|
cwd = os.path.dirname(__file__) |
|
ssh_agent = os.path.join(cwd, "ssh-agent.py") |
|
proxy_env = "SHELLPROXY_URL" |
|
no_proxy_env = "SHELLPROXY_NO_PROXY" |
|
proxy_config = os.environ.get("SHELLPROXY_CONFIG") or os.path.expandvars("$HOME/.config/proxy") |
|
|
|
usage="""shell-proxy: no proxy configuration found. |
|
|
|
Set `{env}` or create a config file at `{config}` |
|
See the plugin README for more information.""".format(env=proxy_env, config=proxy_config) |
|
|
|
def get_http_proxy(): |
|
default_proxy = os.environ.get(proxy_env) |
|
no_proxy = os.environ.get(no_proxy_env) |
|
if default_proxy and no_proxy: |
|
return default_proxy, no_proxy |
|
|
|
if os.path.isfile(proxy_config): |
|
proxy_configdata = [line.strip() for line in check_output(proxy_config).decode("utf-8").splitlines()] |
|
if len(proxy_configdata) >= 1: |
|
if not default_proxy: |
|
default_proxy = proxy_configdata[0] |
|
if len(proxy_configdata) == 2 and not no_proxy: |
|
no_proxy = proxy_configdata[1] |
|
|
|
if default_proxy: |
|
return default_proxy, no_proxy |
|
print(usage, file=sys.stderr) |
|
sys.exit(1) |
|
|
|
|
|
def make_proxies(url: str, no_proxy: str): |
|
proxies = {"%s_PROXY" % _: url for _ in ("HTTP", "HTTPS", "FTP", "RSYNC", "ALL")} |
|
proxies.update({name.lower(): value for (name, value) in proxies.items()}) |
|
proxies["GIT_SSH"] = ssh_agent |
|
if no_proxy: |
|
proxies.update({"NO_PROXY": no_proxy, "no_proxy": no_proxy}) |
|
return proxies |
|
|
|
|
|
def merge(mapping: dict): |
|
return ("%s=%s" % _ for _ in mapping.items()) |
|
|
|
|
|
class CommandSet: |
|
proxies = make_proxies(*get_http_proxy()) |
|
aliases = { |
|
_: "env __SSH_PROGRAM_NAME__=%s %s" % (_, ssh_agent) |
|
for _ in ("ssh", "sftp", "scp", "slogin", "ssh-copy-id") |
|
} |
|
|
|
def enable(self): |
|
cmdline("export", *merge(self.proxies)) |
|
cmdline("alias", *merge(self.aliases)) |
|
|
|
def disable(self): |
|
cmdline("unset", *self.proxies.keys()) |
|
cmdline("unalias", *self.aliases.keys()) |
|
|
|
def status(self): |
|
proxies = ( |
|
"%11s = %s" % (name, os.environ[name]) |
|
for name in self.proxies.keys() |
|
if name in os.environ |
|
) |
|
for _ in proxies: |
|
cmdline("echo", _) |
|
|
|
def usage(self): |
|
print("usage: proxy {enable,disable,status}", file=sys.stderr) |
|
|
|
|
|
def cmdline(*items): |
|
print(list2cmdline(items)) |
|
|
|
|
|
def main(): |
|
command = CommandSet() |
|
if len(sys.argv) == 1: |
|
command.usage() |
|
sys.exit(1) |
|
getattr(command, sys.argv[1], command.usage)() |
|
|
|
|
|
if __name__ == "__main__": |
|
main()
|
|
|