import voluptuous as vol

from homeassistant import config_entries
from homeassistant.core import callback
from homeassistant.helpers.selector import selector


from .utils.logger import LOGGER
from .const import DOMAIN

from .data_client import StateGridDataClient

class StateGridConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
    
    VERSION = 1

    async def async_step_user(self, user_input=None):
        if self._async_current_entries():
            return self.async_abort(reason="single_instance_allowed")
        if self.hass.data.get(DOMAIN):
            return self.async_abort(reason="single_instance_allowed")
        
        self.data_client = StateGridDataClient(hass=self.hass)

        options = {
            "scan_login": "网上国网 App 扫码登录",
            "code_login": "手机短信验证码登录",
        }

        return self.async_show_menu(
            step_id="user",
            menu_options=options
        )

    async def async_step_scan_login(self, user_input=None):
        result = await self.data_client.get_qr_code()
        if result['errcode'] != 0:
            return self.async_abort(
                reason="get_qr_code_error", 
                description_placeholders={'errmsg': result["errmsg"]}
            )
        
        return self.async_show_form(
            step_id="check_qr_code",
            description_placeholders={
                "qr_image": '<img style="width: 200px;" src="data:image/png;base64,' + result["data"] + '"/>'
            }
        )

    async def async_step_check_qr_code(self, user_input=None):
        result = await self.data_client.check_qr_code()

        if result['errcode'] != 0:
            return self.async_abort(
                reason="check_qr_code_error", 
                description_placeholders={'errmsg': result["errmsg"]}
            )

        return self.async_create_entry(
            title="国家电网",
            data={}
        )
        
    async def async_step_code_login(self, user_input=None):
        errors = {}
        if user_input is None:
            user_input = {}
        else:
            phone = user_input['phone']
            result = await self.data_client.send_phone_code(phone)
            if result['errcode'] == 0:
                return await self.async_step_verfiy_code()
            else:
                errors["phone"] = result["errmsg"]
        
        data_schema = {
            vol.Required("phone") : selector({
                "text": {
                    "type": "number"
                }
            })
        }
        
        return self.async_show_form(
            step_id="code_login",
            data_schema=vol.Schema(data_schema),
            errors=errors
        )
    
    async def async_step_verfiy_code(self, user_input=None):
        errors = {}
        if user_input is None:
            user_input = {}
        else:
            code = user_input['code']
            result = await self.data_client.verfiy_phone_code(code)
            if result['errcode'] == 0:
                return self.async_create_entry(
                    title="国家电网",
                    data={}
                )
            
            errors["code"] = result["errmsg"]    

        data_schema = {
            vol.Required("code") : selector({
                "text": {
                    "type": "number"
                }
            })
        }
        
        return self.async_show_form(
            step_id="verfiy_code",
            data_schema=vol.Schema(data_schema),
            errors=errors
        )

    @staticmethod
    @callback
    def async_get_options_flow(entry: config_entries.ConfigEntry):
        return OptionsFlowHandler(entry)

class OptionsFlowHandler(config_entries.OptionsFlow):

    def __init__(self, config_entry: config_entries.ConfigEntry):
        self.config_entry = config_entry
        
    async def async_step_init(self, user_input=None):
        self.data_client: StateGridDataClient = self.hass.data[DOMAIN]
        if self.data_client.need_login is True:
            return await self.async_step_user()
        else:
            return await self.async_step_debug()
        
    async def async_step_debug(self, user_input=None):
        if user_input is None:
            user_input = {}
        else:
            self.data_client.refresh_interval = int(user_input['refresh_interval'])
            self.data_client.is_debug = user_input['is_debug']
            await self.data_client.save_data()
            return self.async_create_entry(
                title="国家电网",
                data={}
            )

        data_schema = {
            vol.Required("refresh_interval", default=str(self.data_client.refresh_interval)) : selector({
                "select": {
                    "options": [
                        {"label":"每1小时", "value": "1"},
                        {"label":"每2小时", "value": "2"},
                        {"label":"每3小时", "value": "3"},
                        {"label":"每4小时", "value": "4"},
                        {"label":"每5小时", "value": "5"},
                        {"label":"每6小时", "value": "6"},
                        {"label":"每7小时", "value": "7"},
                        {"label":"每8小时", "value": "8"},
                        {"label":"每9小时", "value": "9"},
                        {"label":"每10小时", "value": "10"},
                        {"label":"每11小时", "value": "11"},
                        {"label":"每12小时", "value": "12"}
                    ]
                }
            }),
            vol.Required("is_debug", default=self.data_client.is_debug): selector({
                "boolean": {}
            })
        }
        
        return self.async_show_form(
            step_id="debug",
            data_schema=vol.Schema(data_schema)
        )

    async def async_step_user(self, user_input=None):
        options = {
            "scan_login": "网上国网 App 扫码登录",
            "code_login": "手机短信验证码登录",
        }

        return self.async_show_menu(
            step_id="user",
            menu_options=options
        )

    async def async_step_scan_login(self, user_input=None):
        result = await self.data_client.get_qr_code()
        if result['errcode'] != 0:
            return self.async_abort(
                reason="get_qr_code_error", 
                description_placeholders={'errmsg': result["errmsg"]}
            )
        
        return self.async_show_form(
            step_id="check_qr_code",
            description_placeholders={
                "qr_image": '<img style="width: 200px;" src="data:image/png;base64,' + result["data"] + '"/>'
            }
        )

    async def async_step_check_qr_code(self, user_input=None):
        result = await self.data_client.check_qr_code()
        if result['errcode'] != 0:
            return self.async_abort(
                reason="check_qr_code_error", 
                description_placeholders={'errmsg': result["errmsg"]}
            )
        await self.data_client.refresh_data(force_refresh=True)
        await self.data_client.coordinator.async_request_refresh()
        return self.async_create_entry(
            title="国家电网",
            data={}
        )
        
    async def async_step_code_login(self, user_input=None):
        errors = {}
        if user_input is None:
            user_input = {}
        else:
            phone = user_input['phone']
            result = await self.data_client.send_phone_code(phone)
            if result['errcode'] == 0:
                return await self.async_step_verfiy_code()
            else:
                errors["phone"] = result["errmsg"]
        
        data_schema = {
            vol.Required("phone") : selector({
                "text": {
                    "type": "number"
                }
            })
        }
        
        return self.async_show_form(
            step_id="code_login",
            data_schema=vol.Schema(data_schema),
            errors=errors
        )
    
    async def async_step_verfiy_code(self, user_input=None):
        errors = {}
        if user_input is None:
            user_input = {}
        else:
            code = user_input['code']
            result = await self.data_client.verfiy_phone_code(code)
            if result['errcode'] == 0:
                await self.data_client.refresh_data(force_refresh=True)
                await self.data_client.coordinator.async_request_refresh()
                return self.async_create_entry(
                    title="国家电网",
                    data={}
                )
            
            errors["code"] = result["errmsg"]    

        data_schema = {
            vol.Required("code") : selector({
                "text": {
                    "type": "number"
                }
            })
        }
        
        return self.async_show_form(
            step_id="verfiy_code",
            data_schema=vol.Schema(data_schema),
            errors=errors
        )