import json
import os

from ovos_backend_manager.configuration import CONFIGURATION
from ovos_plugin_manager.stt import get_stt_configs, get_stt_supported_langs, get_stt_lang_configs
from pywebio.input import select, actions, input_group, input, TEXT, NUMBER
from pywebio.output import put_text, put_table, popup, put_code, put_image, use_scope


def _get_stt_opts(lang=None):
    STT_CONFIGS = {}
    if lang is not None:
        for p, data in get_stt_lang_configs(lang, include_dialects=True).items():
            if p == "ovos-stt-plugin-selene":
                continue
            if not data:
                continue
            for cfg in data:
                cfg["module"] = p
                cfg["display_name"] = f"{cfg['display_name']} [{p}]"
                STT_CONFIGS[cfg["display_name"]] = cfg
    else:
        for p, data in get_stt_configs().items():
            if p == "ovos-stt-plugin-selene":
                continue
            if not data:
                continue
            for lang, confs in data.items():
                for cfg in confs:
                    cfg["module"] = p
                    cfg["display_name"] = f"{cfg['display_name']} [{p}]"
                    STT_CONFIGS[cfg["display_name"]] = cfg
    return STT_CONFIGS


def microservices_menu(back_handler=None):
    with use_scope("logo", clear=True):
        img = open(f'{os.path.dirname(__file__)}/res/microservices_config.png', 'rb').read()
        put_image(img)

    with use_scope("main_view", clear=True):
        put_table([
            ['STT module', CONFIGURATION["stt"]["module"]],
            ['OVOS microservices fallback enabled', CONFIGURATION["microservices"]["ovos_fallback"]],

            ['WolframAlpha provider', CONFIGURATION["microservices"]["wolfram_provider"]],
            ['Weather provider', CONFIGURATION["microservices"]["weather_provider"]],
            ['Geolocation provider', CONFIGURATION["microservices"]["geolocation_provider"]]
        ])

    buttons = [{'label': 'Configure STT', 'value': "stt"},
               {'label': 'Configure Secrets', 'value': "secrets"},
               {'label': 'Configure SMTP', 'value': "smtp"},
               {'label': 'Configure Wolfram Alpha', 'value': "wolfram"},
               {'label': 'Configure Weather', 'value': "weather"},
               {'label': 'Configure Geolocation', 'value': "geo"}]
    if back_handler:
        buttons.insert(0, {'label': '<- Go Back', 'value': "main"})

    if CONFIGURATION["microservices"]["ovos_fallback"]:
        buttons.insert(-2, {'label': 'Disable OVOS microservices fallback', 'value': "ovos"})
    else:
        buttons.insert(-2, {'label': 'Enable OVOS microservices fallback', 'value': "ovos"})

    opt = actions(label="What would you like to do?", buttons=buttons)
    if opt == "main":
        with use_scope("main_view", clear=True):
            if back_handler:
                back_handler()
        return
    elif opt == "geo":
        opts = ["OpenStreetMap", "ArcGIS", "Geocode Farm"]  # TODO - ovos endpoint
        provider = select("Choose a weather provider", opts)
        if provider == "OpenStreetMap":
            provider = "osm"
        CONFIGURATION["microservices"]["geolocation_provider"] = provider.lower().replace(" ", "_")
    elif opt == "weather":
        opts = ["ovos"]
        if CONFIGURATION["microservices"]["owm_key"]:
            opts.append("local")
        provider = select("Choose a weather provider", opts)
        CONFIGURATION["microservices"]["weather_provider"] = provider
    elif opt == "wolfram":
        opts = ["ovos"]
        if CONFIGURATION["microservices"]["wolfram_key"]:
            opts.append("local")
        provider = select("Choose a WolframAlpha provider", opts)
        CONFIGURATION["microservices"]["wolfram_provider"] = provider
    elif opt == "ovos":
        CONFIGURATION["microservices"]["ovos_fallback"] = not CONFIGURATION["microservices"]["ovos_fallback"]
        if CONFIGURATION["microservices"]["ovos_fallback"]:
            with popup("OVOS microservices fallback enabled"):
                put_text(
                    "wolfram alpha and weather requests will be proxied trough OVOS services if local keys get rate limited/become invalid/are not set")
        else:
            with popup("OVOS microservices fallback disabled"):
                put_text("please set your own wolfram alpha and weather keys")
    elif opt == "stt":
        lang = select("Choose STT default language",
                      list(get_stt_supported_langs().keys()))
        cfgs = _get_stt_opts(lang)
        opts = list(cfgs.keys())
        stt = select("Choose a speech to text engine", opts)
        cfg = dict(cfgs[stt])
        m = cfg.pop("module")
        CONFIGURATION["stt"]["module"] = m
        CONFIGURATION["stt"][m] = cfg
        with popup(f"STT set to: {stt}"):
            put_code(json.dumps(cfg, ensure_ascii=True, indent=2), "json")
    elif opt == "secrets":
        data = input_group('Secret Keys', [
            input("WolframAlpha key", value=CONFIGURATION["microservices"]["wolfram_key"],
                  type=TEXT, name='wolfram'),
            input("OpenWeatherMap key", value=CONFIGURATION["microservices"]["owm_key"],
                  type=TEXT, name='owm')
        ])
        CONFIGURATION["microservices"]["wolfram_key"] = data["wolfram"]
        CONFIGURATION["microservices"]["owm_key"] = data["owm"]
        popup("Secrets updated!")
    elif opt == "smtp":
        # TODO - ovos endpoint

        if "smtp" not in CONFIGURATION["email"]:
            CONFIGURATION["email"]["smtp"] = {}

        data = input_group('SMTP Configuration', [
            input("Username", value=CONFIGURATION["email"]["smtp"].get("username", 'user'),
                  type=TEXT, name='username'),
            input("Password", value=CONFIGURATION["email"]["smtp"].get("password", '***********'),
                  type=TEXT, name='password'),
            input("Host", value=CONFIGURATION["email"]["smtp"].get("host", 'smtp.mailprovider.com'),
                  type=TEXT, name='host'),
            input("Port", value=CONFIGURATION["email"]["smtp"].get("port", '465'),
                  type=NUMBER, name='port')
        ])

        CONFIGURATION["email"]["smtp"]["username"] = data["username"]
        CONFIGURATION["email"]["smtp"]["password"] = data["password"]
        CONFIGURATION["email"]["smtp"]["host"] = data["host"]
        CONFIGURATION["email"]["smtp"]["port"] = data["port"]
        with popup(f"SMTP configuration for: {data['host']}"):
            put_code(json.dumps(data, ensure_ascii=True, indent=2), "json")

    CONFIGURATION.store()

    microservices_menu(back_handler=back_handler)