Limit options to title, favicon, apple. Add gui configuration!

This commit is contained in:
2019-08-21 21:31:57 +02:00
parent 09da46fd58
commit 41345d34b2
6 changed files with 155 additions and 61 deletions

View File

@@ -0,0 +1,30 @@
{
"config": {
"title": "Favicon",
"step": {
"config": {
"title": "Favicon changer",
"description": "Set the page title, favicon and/or iOS icons",
"data": {
"title": "Page title",
"favicon": "favicon URL",
"apple": "iOS icon URL"
}
}
},
"abort": {
"single_instance_allowed": "Only a single configuration of favicon is allowed."
}
},
"options": {
"step": {
"init": {
"data": {
"title": "Page title",
"favicon": "favicon URL",
"apple": "iOS icon URL"
}
}
}
}
}

View File

@@ -1,4 +1,5 @@
import logging
from collections import defaultdict
import homeassistant.components.frontend
from homeassistant.components.frontend import _frontend_root
@@ -8,46 +9,56 @@ _LOGGER = logging.getLogger(__name__)
DOMAIN = "favicon"
async def async_setup(hass, config):
if not hass.data.get(DOMAIN):
hass.data[DOMAIN] = defaultdict(int)
favicon = config[DOMAIN].get('favicon')
apple = config[DOMAIN].get('apple')
title = config[DOMAIN].get('title')
if not hass.data[DOMAIN].get("get_template"):
hass.data[DOMAIN]["get_template"] = homeassistant.components.frontend.IndexView.get_template
if favicon or apple or title:
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)
def new_get_template(self):
tpl = get_template(self)
render = tpl.render
def new_render(*args, **kwargs):
text = render(*args, **kwargs)
if favicon:
text = text.replace("/static/icons/favicon.ico", favicon)
if apple:
text = text.replace("/static/icons/favicon-apple-180x180.png", apple)
if title:
text = text.replace("<title>Home Assistant</title>", f"<title>{title}</title>")
return text
tpl.render = new_render
return tpl
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)
homeassistant.components.frontend.IndexView.get_template = new_get_template
async def async_remove_entry(hass, config_entry):
return remove_hooks(hass)
icons = []
for size in config[DOMAIN]:
if not isinstance(size, int):
continue
i = config[DOMAIN].get(size)
if i:
icons.append({
"src": i,
"sizes": f"{size}x{size}",
"type": "image/png",
})
async def _update_listener(hass, config_entry):
conf = config_entry.options
hass.data[DOMAIN].update(conf)
return apply_hooks(hass)
if icons:
homeassistant.components.frontend.MANIFEST_JSON["icons"] = icons
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

View File

@@ -0,0 +1,55 @@
import voluptuous as vol
from homeassistant import config_entries
from homeassistant.core import callback
@config_entries.HANDLERS.register("favicon")
class ExampleConfigFlow(config_entries.ConfigFlow):
async def async_step_user(self, user_input=None):
if self._async_current_entries():
return self.async_abort(reason="single_instance_allowed")
return await self.async_step_config()
async def async_step_config(self, user_input=None):
if user_input is not None:
return self.async_create_entry(
title="", data=user_input)
return self.async_show_form(
step_id='config',
data_schema=vol.Schema(
{
vol.Optional('title'): str,
vol.Optional('favicon'): str,
vol.Optional('apple'): str,
}
),
)
@staticmethod
@callback
def async_get_options_flow(config_entry):
return ExampleEditFlow(config_entry)
class ExampleEditFlow(config_entries.OptionsFlow):
def __init__(self, config_entry):
self.config_entry = config_entry
async def async_step_init(self, user_input=None):
if user_input is not None:
return self.async_create_entry(title="", data=user_input)
return self.async_show_form(
step_id='init',
data_schema=vol.Schema(
{
vol.Optional('title',
default=self.config_entry.options.get("title", "")): str,
vol.Optional('favicon',
default=self.config_entry.options.get("favicon", "")): str,
vol.Optional('apple',
default=self.config_entry.options.get("apple", "")): str,
}
),
)

View File

@@ -4,5 +4,6 @@
"documentation": "",
"dependencies": ["frontend"],
"codeowners": [],
"requirements": []
"requirements": [],
"config_flow": true
}