28 lines
1 KiB
Python
28 lines
1 KiB
Python
from .ssh import SSHClient
|
|
|
|
class IngressManager:
|
|
def __init__(self, config):
|
|
self.host = config.get('haproxy.host')
|
|
self.user = config.get('haproxy.user', 'root')
|
|
self.ssh_key = config.get('haproxy.ssh_key_path')
|
|
self.password = config.get('haproxy.password')
|
|
self.client = SSHClient(self.host, self.user, self.ssh_key, self.password)
|
|
|
|
def add(self, domain, ip, port, https=False):
|
|
cmd = f"ingress-manager add {domain} {ip} {port}"
|
|
if https:
|
|
cmd += " --https"
|
|
res = self.client.run(cmd)
|
|
if res.returncode != 0:
|
|
raise RuntimeError(f"Failed to add ingress: {res.stderr}")
|
|
return res.stdout
|
|
|
|
def remove(self, domain):
|
|
res = self.client.run(f"ingress-manager remove {domain}")
|
|
if res.returncode != 0:
|
|
raise RuntimeError(f"Failed to remove ingress: {res.stderr}")
|
|
return res.stdout
|
|
|
|
def list(self):
|
|
res = self.client.run("ingress-manager list")
|
|
return res.stdout
|