#!/usr/bin/env python3
# coding=utf-8
"""
config variables
"""

from __future__ import print_function, unicode_literals

import logging
import os

from wells.config import ConfigurationManger

import aiom3u8downloader.configlogger

logger = logging.getLogger(__name__)

TESTING = os.getenv("PYTEST") == "1"
DEBUGGING = os.getenv("DEBUG") == "1"

# set root logger level to ERROR when running unit test.
if TESTING:
    logging.getLogger('').setLevel(logging.ERROR)

DEFAULTS = {
    "user_agent":
    'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/72.0.3626.119 Safari/537.36',
    "origin": "",
    "tempdir": "",
    "concurrency": "5",
    "debug": "false",
}

OPTIONAL_KEYS = list(DEFAULTS.keys())

REQUIRED_KEYS = []

CONF = ConfigurationManger(
    defaults=DEFAULTS,
    configfiles=[
        "/etc/aiom3u8downloader.conf",
        "/etc/aiom3u8downloader/aiom3u8downloader.conf",
        os.path.expanduser("~/.config/aiom3u8downloader.conf"),
        os.path.expanduser(
            "~/.config/aiom3u8downloader/aiom3u8downloader.conf"),
    ],
    required_keys=REQUIRED_KEYS,
    optional_keys=OPTIONAL_KEYS)


def ensure_all_config_variable_defined():
    """check all config variable used in agent code is defined.

    exit with 0 on success, exit with non-zero on failure.

    """
    import sys
    import subprocess

    check_pass = True
    output = subprocess.check_output(r"""
rgrep 'CONF\.get' aiom3u8downloader/ |grep -o 'CONF\.get[a-z]\+("[^)]*")'|sed 's/.*("//; s/")//'
    """,
                                     shell=True).rstrip().decode(
                                         "utf-8", 'ignore')
    all_keys = REQUIRED_KEYS + OPTIONAL_KEYS
    for line in output.split("\n"):
        if line and line not in all_keys:
            logger.error("undefined config variable: %s", line)
            check_pass = False
    sys.exit(0 if check_pass else 1)


def main():
    ensure_all_config_variable_defined()


if __name__ == '__main__':
    main()
