"""
A module containing all console script functions.
"""
import argparse
import logging
import logging.config
from logging import Logger
from typing import Any, Dict, Tuple
import os
import yaml

from chweb.models import Config


def configure(name) -> Tuple[Config, Logger]:
    """
    Gets the configuration and creates a Pydantic model from the parsed YAML.
    """
    parser = argparse.ArgumentParser(
        description='Website availibility checker.')
    parser.add_argument('-c', '--config', type=str,
                        default="/etc/checker.yaml",
                        help=('The yaml config file. '
                              'Defaults to /etc/checker.yaml'))
    args = parser.parse_args()

    with open(args.config, 'r') as conf_file:
        conf = yaml.load(conf_file, Loader=yaml.FullLoader)
        logging.config.dictConfig(conf['logging'])

        config = create_config(conf)
        logger = logging.getLogger("chweb.{}".format(name))
        return (config, logger)


def create_config(conf: Dict[str, Any]):
    kafka_servers_env = os.getenv('KAFKA_SERVERS')
    if kafka_servers_env is not None:
        kafka_servers = kafka_servers_env.split(',')

    kafka_topic = os.getenv('KAFKA_TOPIC')

    pg_db = os.getenv('POSTGRES_DB')
    pg_host = os.getenv('POSTGRES_HOST')
    pg_port = os.getenv('POSTGRES_PORT')
    pg_user = os.getenv('POSTGRES_USER')
    pg_pass = os.getenv('POSTGRES_PASS')

    config = Config(**conf)
    config.kafka.servers = (kafka_servers if kafka_servers_env
                            else config.kafka.servers)
    config.kafka.topic = kafka_topic or config.kafka.topic
    config.postgres.dbhost = pg_host or config.postgres.dbhost
    config.postgres.dbname = pg_db or config.postgres.dbname
    config.postgres.dbport = (int(pg_port) if pg_port is not None
                              else config.postgres.dbport)
    config.postgres.dbuser = pg_user or config.postgres.dbuser
    config.postgres.dbpass = pg_pass or config.postgres.dbpass

    return config