import logging
from collections import defaultdict

import homeassistant.components.frontend
from homeassistant.components.frontend import _frontend_root

_LOGGER = logging.getLogger(__name__)

DOMAIN = "favicon"

async def async_setup(hass, config):
    if not hass.data.get(DOMAIN):
        hass.data[DOMAIN] = defaultdict(int)

    if not hass.data[DOMAIN].get("get_template"):
        hass.data[DOMAIN]["get_template"] = homeassistant.components.frontend.IndexView.get_template

    conf = config.get(DOMAIN)
    if not conf:
        return True
    hass.data[DOMAIN].update(conf)
    return apply_hooks(hass)

async def async_setup_entry(hass, config_entry):
    config_entry.add_update_listener(_update_listener)
    config_entry.options = config_entry.data
    return await _update_listener(hass, config_entry)

async def async_remove_entry(hass, config_entry):
    return remove_hooks(hass)

async def _update_listener(hass, config_entry):
    conf = config_entry.options
    hass.data[DOMAIN].update(conf)
    return apply_hooks(hass)


def apply_hooks(hass):
    data = hass.data[DOMAIN]
    def _get_template(self):
        tpl = data["get_template"](self)
        render = tpl.render

        def new_render(*args, **kwargs):
            text = render(*args, **kwargs)
            if data["favicon"]:
                text = text.replace("/static/icons/favicon.ico", data["favicon"])
            if data["apple"]:
                text = text.replace("/static/icons/favicon-apple-180x180.png", data["apple"])
            if data["title"]:
                text = text.replace("<title>Home Assistant</title>", f"<title>{data['title']}</title>")
            return text

        tpl.render = new_render
        return tpl

    homeassistant.components.frontend.IndexView.get_template = _get_template
    return True

def remove_hooks(hass):
    data = hass.data[DOMAIN]
    homeassistant.components.frontend.IndexView.get_template = data["get_template"]
    return True