Move setup to init (#2356)

custom_components/hacs/__init__.py

@@ -8,35 +8,163 @@
 
 from typing import TYPE_CHECKING, Any
 
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.core import HomeAssistant
+from aiogithubapi import AIOGitHubAPIException, GitHub, GitHubAPI
+from aiogithubapi.const import ACCEPT_HEADERS
+from awesomeversion import AwesomeVersion
+from homeassistant.components.lovelace.system_health import system_health_info
+from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
+from homeassistant.const import EVENT_HOMEASSISTANT_STARTED, __version__ as HAVERSION
+from homeassistant.core import CoreState, HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_create_clientsession
+from homeassistant.helpers.event import async_call_later
+from homeassistant.loader import async_get_integration
 import voluptuous as vol
 
-from .const import DOMAIN, PLATFORMS
-from .enums import HacsDisabledReason
+from .const import DOMAIN, PLATFORMS, STARTUP
+from .enums import ConfigurationType, HacsDisabledReason, HacsStage, LovelaceMode
+from .hacsbase.data import HacsData
 from .helpers.functions.configuration_schema import hacs_config_combined
-from .operational.setup import (
- async_setup as hacs_yaml_setup,
- async_setup_entry as hacs_ui_setup,
-)
+from .share import get_hacs
+from .tasks.manager import HacsTaskManager
 
 if TYPE_CHECKING:
  from .base import HacsBase
 
 CONFIG_SCHEMA = vol.Schema({DOMAIN: hacs_config_combined()}, extra=vol.ALLOW_EXTRA)
 
 
+async def async_initialize_integration(
+ hass: HomeAssistant,
+ *,
+ config_entry: ConfigEntry | None = None,
+ config: dict[str, Any] | None = None,
+) -> bool:
+ """Initialize the integration"""
+ hass.data[DOMAIN] = hacs = get_hacs()
+ hacs.enable_hacs()
+
+ if config is not None:
+ if DOMAIN not in config:
+ return True
+ if hacs.configuration.config_type == ConfigurationType.CONFIG_ENTRY:
+ return True
+ hacs.configuration.update_from_dict(
+ {
+ "config_type": ConfigurationType.YAML,
+ **config[DOMAIN],
+ "config": config[DOMAIN],
+ }
+ )
+
+ if config_entry is not None:
+ if config_entry.source == SOURCE_IMPORT:
+ hass.async_create_task(hass.config_entries.async_remove(config_entry.entry_id))
+ return False
+
+ hacs.configuration.update_from_dict(
+ {
+ "config_entry": config_entry,
+ "config_type": ConfigurationType.CONFIG_ENTRY,
+ **config_entry.data,
+ **config_entry.options,
+ }
+ )
+
+ integration = await async_get_integration(hass, DOMAIN)
+
+ await hacs.async_set_stage(None)
+
+ hacs.log.info(STARTUP.format(version=integration.version))
+
+ hacs.integration = integration
+ hacs.version = integration.version
+ hacs.hass = hass
+ hacs.data = HacsData()
+ hacs.system.running = True
+ hacs.session = async_create_clientsession(hass)
+ hacs.tasks = HacsTaskManager(hacs=hacs, hass=hass)
+
+ hacs.core.lovelace_mode = LovelaceMode.YAML
+ try:
+ lovelace_info = await system_health_info(hacs.hass)
+ hacs.core.lovelace_mode = LovelaceMode(lovelace_info.get("mode", "yaml"))
+ except Exception: # pylint: disable=broad-except
+ # If this happens, the users YAML is not valid, we assume YAML mode
+ pass
+ hacs.log.debug(f"Configuration type: {hacs.configuration.config_type}")
+ hacs.core.config_path = hacs.hass.config.path()
+
+ if hacs.core.ha_version is None:
+ hacs.core.ha_version = AwesomeVersion(HAVERSION)
+
+ await hacs.tasks.async_load()
+
+ # Setup session for API clients
+ session = async_create_clientsession(hacs.hass)
+
+ ## Legacy GitHub client
+ hacs.github = GitHub(
+ hacs.configuration.token,
+ session,
+ headers={
+ "User-Agent": f"HACS/{hacs.version}",
+ "Accept": ACCEPT_HEADERS["preview"],
+ },
+ )
+
+ ## New GitHub client
+ hacs.githubapi = GitHubAPI(
+ token=hacs.configuration.token,
+ session=session,
+ **{"client_name": f"HACS/{hacs.version}"},
+ )
+
+ async def async_startup():
+ """HACS startup tasks."""
+ hacs.enable_hacs()
+
+ await hacs.async_set_stage(HacsStage.SETUP)
+ if hacs.system.disabled:
+ return False
+
+ # Setup startup tasks
+ if hacs.hass.state == CoreState.running:
+ async_call_later(hacs.hass, 5, hacs.startup_tasks)
+ else:
+ hacs.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STARTED, hacs.startup_tasks)
+
+ await hacs.async_set_stage(HacsStage.WAITING)
+ hacs.log.info("Setup complete, waiting for Home Assistant before startup tasks starts")
+
+ return not hacs.system.disabled
+
+ async def async_try_startup(_=None):
+ """Startup wrapper for yaml config."""
+ try:
+ startup_result = await async_startup()
+ except AIOGitHubAPIException:
+ startup_result = False
+ if not startup_result:
+ hacs.log.info("Could not setup HACS, trying again in 15 min")
+ async_call_later(hass, 900, async_try_startup)
+ return
+ hacs.enable_hacs()
+
+ await async_try_startup()
+
+ # Mischief managed!
+ return True
+
+
 async def async_setup(hass: HomeAssistant, config: dict[str, Any]) -> bool:
  """Set up this integration using yaml."""
-
- return await hacs_yaml_setup(hass, config)
+ return await async_initialize_integration(hass=hass, config=config)
 
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
  """Set up this integration using UI."""
  config_entry.add_update_listener(async_reload_entry)
-
- return await hacs_ui_setup(hass, config_entry)
+ return await async_initialize_integration(hass=hass, config_entry=config_entry)
 
 
 async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:

tests/conftest.py

@@ -4,8 +4,8 @@
 import logging
 from pathlib import Path
 from unittest.mock import AsyncMock
-from awesomeversion import AwesomeVersion
 
+from awesomeversion import AwesomeVersion
 from homeassistant.const import __version__ as HAVERSION
 from homeassistant.exceptions import ServiceNotFound
 from homeassistant.helpers.aiohttp_client import async_create_clientsession

tests/repositories/helpers/test_properties.py

@@ -1,6 +1,7 @@
 """HACS Repository Helper properties."""
 # pylint: disable=missing-docstring
 from awesomeversion import AwesomeVersion
+
 from custom_components.hacs.helpers.classes.repository import HacsRepository
 
 

tests/repositories/test_can_install.py

@@ -1,6 +1,7 @@
 """Configuration Test Suite: can install."""
 # pylint: disable=missing-docstring
 from awesomeversion import AwesomeVersion
+
 from custom_components.hacs.helpers.classes.repository import HacsRepository
 
 

tests/repositories/test_core.py

@@ -1,6 +1,7 @@
 """Configuration Test Suite: Core repository."""
 # pylint: disable=missing-docstring
 from awesomeversion import AwesomeVersion
+
 from custom_components.hacs.helpers.classes.repository import HacsRepository
 
 

tests/tasks/test_check_constrains.py

@@ -1,7 +1,7 @@
 # pylint: disable=missing-function-docstring,missing-module-docstring, protected-access
 from unittest.mock import patch
-from awesomeversion.awesomeversion import AwesomeVersion
 
+from awesomeversion.awesomeversion import AwesomeVersion
 import pytest
 
 from custom_components.hacs.base import HacsBase