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