92 lines
2.6 KiB
Python
92 lines
2.6 KiB
Python
|
#!/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()
|