##############################################################################
# Copyright (c) 2017 Tim Rozet (trozet@redhat.com) and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################

import json
import logging
import pprint
import os
import re

from apex.common.exceptions import ApexDeployException

"""Parser functions for overcloud/openstack output"""


def parse_nova_output(in_file):
    """
    Parses nova list output into a dictionary format for node name and ip
    :param in_file: json format from openstack server list
    :return: dictionary format for {"node name": "node ip"}
    """
    if not os.path.isfile(in_file):
        raise FileNotFoundError(in_file)
    node_dict = dict()
    with open(in_file, 'r') as fh:
        nova_list = json.load(fh)

    for server in nova_list:
        ip_match = re.search('([0-9]+\.){3}[0-9]+', server['Networks'])
        if ip_match is None:
            logging.error("Unable to find IP in nova output "
                          "{}".format(pprint.pformat(server, indent=4)))
            raise ApexDeployException("Unable to parse IP from nova output")
        else:
            node_dict[server['Name']] = ip_match.group(0)

    if not node_dict:
        raise ApexDeployException("No overcloud nodes found in: {}".format(
            in_file))
    return node_dict


def parse_overcloudrc(in_file):
    """
    Parses overcloudrc into a dictionary format for key and value
    :param in_file:
    :return: dictionary format for {"variable": "value"}
    """
    logging.debug("Parsing overcloudrc file {}".format(in_file))
    if not os.path.isfile(in_file):
        raise FileNotFoundError(in_file)
    creds = {}
    with open(in_file, 'r') as fh:
        lines = fh.readlines()
    kv_pattern = re.compile('^export\s+([^\s]+)=([^\s]+)$')
    for line in lines:
        if 'export' not in line:
            continue
        else:
            res = re.search(kv_pattern, line.strip())
            if res:
                creds[res.group(1)] = res.group(2)
                logging.debug("os cred found: {}, {}".format(res.group(1),
                                                             res.group(2)))
            else:
                logging.debug("os cred not found in: {}".format(line))

    return creds


def parse_ifcfg_file(in_file):
    """
    Parses ifcfg file information
    :param in_file:
    :return: dictionary of ifcfg key value pairs
    """
    ifcfg_params = {
        'IPADDR': '',
        'NETMASK': '',
        'GATEWAY': '',
        'METRIC': '',
        'DNS1': '',
        'DNS2': '',
        'PREFIX': ''
    }
    with open(in_file, 'r') as fh:
        for line in fh:
            for param in ifcfg_params.keys():
                match = re.search("^\s*{}=(.*)$".format(param), line)
                if match:
                    ifcfg_params[param] = match.group(1)
                    break
    return ifcfg_params