import requests

from wo.core.shellexec import WOShellExec
from wo.core.variables import WOVariables


def check_fqdn(self, wo_host):
    """FQDN check with WordOps, for mail server hostname must be FQDN"""
    # wo_host=os.popen("hostname -f | tr -d '\n'").read()
    if '.' in wo_host:
        WOVariables.wo_fqdn = wo_host
        with open('/etc/hostname', encoding='utf-8', mode='w') as hostfile:
            hostfile.write(wo_host)

        WOShellExec.cmd_exec(self, "sed -i \"1i\\127.0.0.1 {0}\" /etc/hosts"
                                   .format(wo_host))
        if WOVariables.wo_distro == 'debian':
            WOShellExec.cmd_exec(self, "/etc/init.d/hostname.sh start")
        else:
            WOShellExec.cmd_exec(self, "service hostname restart")

    else:
        wo_host = input("Enter hostname [fqdn]:")
        check_fqdn(self, wo_host)


def check_fqdn_ip(self):
    """Check if server hostname resolved server IP"""
    x = requests.get('http://v4.wordops.eu')
    ip = (x.text).strip()

    wo_fqdn = WOVariables.wo_fqdn
    y = requests.get('http://v4.wordops.eu/dns/{0}/'.format(wo_fqdn))
    ip_fqdn = (y.text).strip()

    if ip == ip_fqdn:
        return True
    else:
        return False