112 lines
3.1 KiB
Python
112 lines
3.1 KiB
Python
import logging
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.components.frontend import add_extra_js_url
|
|
from homeassistant.components.http import StaticPathConfig
|
|
from homeassistant.components.http.view import HomeAssistantView
|
|
from homeassistant.core import async_get_hass
|
|
from homeassistant.helpers import config_validation as cv
|
|
|
|
import json
|
|
from os import path, walk
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
DOMAIN = "fontawesome"
|
|
|
|
DATA_EXTRA_MODULE_URL = "frontend_extra_module_url"
|
|
LOADER_URL = f"/{DOMAIN}/main.js"
|
|
LOADER_PATH = f"custom_components/{DOMAIN}/main.js"
|
|
ICONS_URL = f"/{DOMAIN}/icons"
|
|
ICONLIST_URL = f"/{DOMAIN}/list"
|
|
ICONS_PATH = f"custom_components/{DOMAIN}/data"
|
|
CUSTOM_ICONS_URL = f"/{DOMAIN}/icons/pro"
|
|
CUSTOM_ICONS_PATH = "custom_icons"
|
|
|
|
|
|
CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
|
|
|
|
|
|
class ListingView(HomeAssistantView):
|
|
|
|
requires_auth = False
|
|
|
|
def __init__(self, url, iconpath, hass):
|
|
self.url = url
|
|
self.iconpath = iconpath
|
|
self.hass: HomeAssistant = hass
|
|
self.name = "Icon Listing"
|
|
|
|
async def get(self, request):
|
|
icons_list = await self.hass.async_add_executor_job(
|
|
self.get_icons_list, self.iconpath
|
|
)
|
|
return icons_list
|
|
|
|
def get_icons_list(self, iconpath):
|
|
icons = []
|
|
for dirpath, dirnames, filenames in walk(iconpath):
|
|
icons.extend(
|
|
[
|
|
{"name": path.join(dirpath[len(iconpath) :].lstrip("/"), fn[:-4])}
|
|
for fn in filenames
|
|
if fn.endswith(".svg")
|
|
]
|
|
)
|
|
return json.dumps(icons)
|
|
|
|
|
|
async def async_setup(hass: HomeAssistant, config):
|
|
await hass.http.async_register_static_paths(
|
|
[StaticPathConfig(LOADER_URL, hass.config.path(LOADER_PATH), True)]
|
|
)
|
|
add_extra_js_url(hass, LOADER_URL)
|
|
|
|
for iset in ["brands", "regular", "solid"]:
|
|
await hass.http.async_register_static_paths(
|
|
[
|
|
StaticPathConfig(
|
|
ICONS_URL + "/" + iset,
|
|
hass.config.path(ICONS_PATH + "/" + iset),
|
|
True,
|
|
)
|
|
]
|
|
)
|
|
hass.http.register_view(
|
|
ListingView(
|
|
ICONLIST_URL + "/" + iset,
|
|
hass.config.path(ICONS_PATH + "/" + iset),
|
|
hass,
|
|
)
|
|
)
|
|
await hass.http.async_register_static_paths(
|
|
[StaticPathConfig(CUSTOM_ICONS_URL, hass.config.path(CUSTOM_ICONS_PATH), True)]
|
|
)
|
|
hass.http.register_view(
|
|
ListingView(ICONLIST_URL + "/pro", hass.config.path(CUSTOM_ICONS_PATH), hass)
|
|
)
|
|
|
|
return True
|
|
|
|
|
|
async def async_setup_entry(hass, entry):
|
|
return True
|
|
|
|
|
|
async def async_remove_entry(hass, entry):
|
|
return True
|
|
|
|
async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Migrate old entry."""
|
|
|
|
if entry.version == 1:
|
|
entry.version = 2
|
|
|
|
hass.config_entries.async_update_entry(
|
|
entry,
|
|
title="Fontawesome Icons"
|
|
)
|
|
LOGGER.info("Migrating fontawesome config entry.")
|
|
return True
|