#!/usr/bin/python3

import salt.utils.dictupdate
import salt.utils.dictdiffer


def _error(ret, err_msg):
    ret['result'] = False
    ret['comment'] = err_msg

    return ret


def _str_split(string):
    delim = "\n"

    return [e + delim for e in string.split(delim) if e]


def domain_record_present(name="",
                          zone="",
                          recordname="",
                          recordtype="A",
                          target="",
                          ttl=0):

    res_output = ""
    ret = {
        'name': name,
        'changes': {},
        'result': True,
        'comment': 'Config is up to date'
    }

    if not name:
        return _error(ret, 'Must provide name to ovhapi.domain_record_present')
    if not zone:
        return _error(ret, 'Must provide dns zone to ovhapi.domain_record_present')
    if not recordname:
        return _error(ret, 'Must provide record name to ovhapi.domain_record_present')
    if not recordtype:
        return _error(ret, 'Must provide record type to ovhapi.domain_record_present')
    if not target:
        return _error(ret, 'Must provide target to ovhapi.domain_record_present')

    cur_zone_state = __salt__['ovhapi.domain_get_zone'](zone=zone)

    # check if record exists
    cur_record = __salt__['ovhapi.domain_get_record'](zone=zone,
                                                      fieldType=recordtype,
                                                      subDomain=recordname,
                                                      target=target)
    if cur_record:
        __salt__['ovhapi.domain_put_record'](zone=zone,
                                             fieldType=recordtype,
                                             subDomain=recordname,
                                             target=target)
        new_zone_state = __salt__['ovhapi.domain_get_zone'](zone=zone)

        ret['changes'] = {
            "diff": salt.utils.stringutils.get_diff(_str_split(cur_zone_state),
                                                    _str_split(new_zone_state))
        }
        res_output = f"Updated record {recordname}"
    else:
        __salt__['ovhapi.domain_post_record'](zone=zone,
                                              subDomain=recordname,
                                              fieldType=recordtype,
                                              target=target,
                                              ttl=ttl)
        new_zone_state = __salt__['ovhapi.domain_get_zone'](zone=zone)

        ret['changes'] = {
            "diff": salt.utils.stringutils.get_diff(_str_split(cur_zone_state),
                                                    _str_split(new_zone_state))
        }
        res_output = f"Created record {recordname}"

    ret['comment'] = f'Result is {res_output}'

    return ret


def domain_zone_refresh(name="", zone=""):
    res_output = ""
    ret = {
        'name': name,
        'changes': {},
        'result': True,
        'comment': 'Zone is refreshed'
    }

    res_output = __salt__['ovhapi.domain_refresh_zone'](zone=zone)

    ret['comment'] = f'Result is {res_output}'

    return ret