"""Handle core shared data.""" from __future__ import annotations import asyncio from typing import TYPE_CHECKING, Any, Callable, Coroutine, Optional, TypeVar import aiohttp import sentry_sdk from .config import CoreConfig from .const import UpdateChannel from .docker import DockerAPI from .misc.hardware import Hardware if TYPE_CHECKING: from .addons import AddonManager from .api import RestAPI from .arch import CpuArch from .auth import Auth from .core import Core from .dbus import DBusManager from .discovery import Discovery from .hassos import HassOS from .misc.scheduler import Scheduler from .misc.hwmon import HwMonitor from .misc.tasks import Tasks from .homeassistant import HomeAssistant from .host import HostManager from .ingress import Ingress from .services import ServiceManager from .snapshots import SnapshotManager from .supervisor import Supervisor from .store import StoreManager from .updater import Updater from .plugins import PluginManager from .resolution import ResolutionManager T = TypeVar("T") class CoreSys: """Class that handle all shared data.""" def __init__(self): """Initialize coresys.""" # Static attributes protected self._machine_id: Optional[str] = None self._machine: Optional[str] = None # External objects self._loop: asyncio.BaseEventLoop = asyncio.get_running_loop() self._websession: aiohttp.ClientSession = aiohttp.ClientSession() self._websession_ssl: aiohttp.ClientSession = aiohttp.ClientSession( connector=aiohttp.TCPConnector(ssl=False) ) # Global objects self._config: CoreConfig = CoreConfig() self._hardware: Hardware = Hardware() self._docker: DockerAPI = DockerAPI() # Internal objects pointers self._core: Optional[Core] = None self._arch: Optional[CpuArch] = None self._auth: Optional[Auth] = None self._homeassistant: Optional[HomeAssistant] = None self._supervisor: Optional[Supervisor] = None self._addons: Optional[AddonManager] = None self._api: Optional[RestAPI] = None self._updater: Optional[Updater] = None self._snapshots: Optional[SnapshotManager] = None self._tasks: Optional[Tasks] = None self._host: Optional[HostManager] = None self._ingress: Optional[Ingress] = None self._dbus: Optional[DBusManager] = None self._hassos: Optional[HassOS] = None self._services: Optional[ServiceManager] = None self._scheduler: Optional[Scheduler] = None self._store: Optional[StoreManager] = None self._discovery: Optional[Discovery] = None self._hwmonitor: Optional[HwMonitor] = None self._plugins: Optional[PluginManager] = None self._resolution: Optional[ResolutionManager] = None @property def dev(self) -> bool: """Return True if we run dev mode.""" if self._updater is None: return False return self._updater.channel == UpdateChannel.DEV @property def loop(self) -> asyncio.BaseEventLoop: """Return loop object.""" return self._loop @property def websession(self) -> aiohttp.ClientSession: """Return websession object.""" return self._websession @property def websession_ssl(self) -> aiohttp.ClientSession: """Return websession object with disabled SSL.""" return self._websession_ssl @property def config(self) -> CoreConfig: """Return CoreConfig object.""" return self._config @property def hardware(self) -> Hardware: """Return Hardware object.""" return self._hardware @property def docker(self) -> DockerAPI: """Return DockerAPI object.""" return self._docker @property def scheduler(self) -> Scheduler: """Return Scheduler object.""" if self._scheduler is None: raise RuntimeError("Scheduler not set!") return self._scheduler @scheduler.setter def scheduler(self, value: Scheduler) -> None: """Set a Scheduler object.""" if self._scheduler: raise RuntimeError("Scheduler already set!") self._scheduler = value @property def core(self) -> Core: """Return core object.""" if self._core is None: raise RuntimeError("Core not set!") return self._core @core.setter def core(self, value: Core) -> None: """Set a Core object.""" if self._core: raise RuntimeError("Core already set!") self._core = value @property def plugins(self) -> PluginManager: """Return PluginManager object.""" if self._plugins is None: raise RuntimeError("PluginManager not set!") return self._plugins @plugins.setter def plugins(self, value: PluginManager) -> None: """Set a PluginManager object.""" if self._plugins: raise RuntimeError("PluginManager already set!") self._plugins = value @property def arch(self) -> CpuArch: """Return CpuArch object.""" if self._arch is None: raise RuntimeError("CpuArch not set!") return self._arch @arch.setter def arch(self, value: CpuArch) -> None: """Set a CpuArch object.""" if self._arch: raise RuntimeError("CpuArch already set!") self._arch = value @property def auth(self) -> Auth: """Return Auth object.""" if self._auth is None: raise RuntimeError("Auth not set!") return self._auth @auth.setter def auth(self, value: Auth) -> None: """Set a Auth object.""" if self._auth: raise RuntimeError("Auth already set!") self._auth = value @property def homeassistant(self) -> HomeAssistant: """Return Home Assistant object.""" if self._homeassistant is None: raise RuntimeError("Home Assistant not set!") return self._homeassistant @homeassistant.setter def homeassistant(self, value: HomeAssistant) -> None: """Set a HomeAssistant object.""" if self._homeassistant: raise RuntimeError("Home Assistant already set!") self._homeassistant = value @property def supervisor(self) -> Supervisor: """Return Supervisor object.""" if self._supervisor is None: raise RuntimeError("Supervisor not set!") return self._supervisor @supervisor.setter def supervisor(self, value: Supervisor) -> None: """Set a Supervisor object.""" if self._supervisor: raise RuntimeError("Supervisor already set!") self._supervisor = value @property def api(self) -> RestAPI: """Return API object.""" if self._api is None: raise RuntimeError("API not set!") return self._api @api.setter def api(self, value: RestAPI) -> None: """Set an API object.""" if self._api: raise RuntimeError("API already set!") self._api = value @property def updater(self) -> Updater: """Return Updater object.""" if self._updater is None: raise RuntimeError("Updater not set!") return self._updater @updater.setter def updater(self, value: Updater) -> None: """Set a Updater object.""" if self._updater: raise RuntimeError("Updater already set!") self._updater = value @property def addons(self) -> AddonManager: """Return AddonManager object.""" if self._addons is None: raise RuntimeError("AddonManager not set!") return self._addons @addons.setter def addons(self, value: AddonManager) -> None: """Set a AddonManager object.""" if self._addons: raise RuntimeError("AddonManager already set!") self._addons = value @property def store(self) -> StoreManager: """Return StoreManager object.""" if self._store is None: raise RuntimeError("StoreManager not set!") return self._store @store.setter def store(self, value: StoreManager) -> None: """Set a StoreManager object.""" if self._store: raise RuntimeError("StoreManager already set!") self._store = value @property def snapshots(self) -> SnapshotManager: """Return SnapshotManager object.""" if self._snapshots is None: raise RuntimeError("SnapshotManager not set!") return self._snapshots @snapshots.setter def snapshots(self, value: SnapshotManager) -> None: """Set a SnapshotManager object.""" if self._snapshots: raise RuntimeError("SnapshotsManager already set!") self._snapshots = value @property def tasks(self) -> Tasks: """Return Tasks object.""" if self._tasks is None: raise RuntimeError("Tasks not set!") return self._tasks @tasks.setter def tasks(self, value: Tasks) -> None: """Set a Tasks object.""" if self._tasks: raise RuntimeError("Tasks already set!") self._tasks = value @property def services(self) -> ServiceManager: """Return ServiceManager object.""" if self._services is None: raise RuntimeError("Services not set!") return self._services @services.setter def services(self, value: ServiceManager) -> None: """Set a ServiceManager object.""" if self._services: raise RuntimeError("Services already set!") self._services = value @property def discovery(self) -> Discovery: """Return ServiceManager object.""" if self._discovery is None: raise RuntimeError("Discovery not set!") return self._discovery @discovery.setter def discovery(self, value: Discovery) -> None: """Set a Discovery object.""" if self._discovery: raise RuntimeError("Discovery already set!") self._discovery = value @property def dbus(self) -> DBusManager: """Return DBusManager object.""" if self._dbus is None: raise RuntimeError("DBusManager not set!") return self._dbus @dbus.setter def dbus(self, value: DBusManager) -> None: """Set a DBusManager object.""" if self._dbus: raise RuntimeError("DBusManager already set!") self._dbus = value @property def host(self) -> HostManager: """Return HostManager object.""" if self._host is None: raise RuntimeError("HostManager not set!") return self._host @host.setter def host(self, value: HostManager) -> None: """Set a HostManager object.""" if self._host: raise RuntimeError("HostManager already set!") self._host = value @property def hwmonitor(self) -> HwMonitor: """Return HwMonitor object.""" if self._hwmonitor is None: raise RuntimeError("HwMonitor not set!") return self._hwmonitor @hwmonitor.setter def hwmonitor(self, value: HwMonitor) -> None: """Set a HwMonitor object.""" if self._hwmonitor: raise RuntimeError("HwMonitor already set!") self._hwmonitor = value @property def ingress(self) -> Ingress: """Return Ingress object.""" if self._ingress is None: raise RuntimeError("Ingress not set!") return self._ingress @ingress.setter def ingress(self, value: Ingress) -> None: """Set a Ingress object.""" if self._ingress: raise RuntimeError("Ingress already set!") self._ingress = value @property def hassos(self) -> HassOS: """Return HassOS object.""" if self._hassos is None: raise RuntimeError("HassOS not set!") return self._hassos @hassos.setter def hassos(self, value: HassOS) -> None: """Set a HassOS object.""" if self._hassos: raise RuntimeError("HassOS already set!") self._hassos = value @property def resolution(self) -> ResolutionManager: """Return resolution manager object.""" if self._resolution is None: raise RuntimeError("resolution manager not set!") return self._resolution @resolution.setter def resolution(self, value: ResolutionManager) -> None: """Set a resolution manager object.""" if self._resolution: raise RuntimeError("resolution manager already set!") self._resolution = value @property def machine(self) -> Optional[str]: """Return machine type string.""" return self._machine @machine.setter def machine(self, value: str) -> None: """Set a machine type string.""" if self._machine: raise RuntimeError("Machine type already set!") self._machine = value @property def machine_id(self) -> Optional[str]: """Return machine-id type string.""" return self._machine_id @machine_id.setter def machine_id(self, value: str) -> None: """Set a machine-id type string.""" if self._machine_id: raise RuntimeError("Machine-ID type already set!") self._machine_id = value class CoreSysAttributes: """Inherit basic CoreSysAttributes.""" coresys: CoreSys @property def sys_machine(self) -> Optional[str]: """Return running machine type of the Supervisor system.""" return self.coresys.machine @property def sys_dev(self) -> bool: """Return True if we run dev mode.""" return self.coresys.dev @property def sys_loop(self) -> asyncio.BaseEventLoop: """Return loop object.""" return self.coresys.loop @property def sys_websession(self) -> aiohttp.ClientSession: """Return websession object.""" return self.coresys.websession @property def sys_websession_ssl(self) -> aiohttp.ClientSession: """Return websession object with disabled SSL.""" return self.coresys.websession_ssl @property def sys_config(self) -> CoreConfig: """Return CoreConfig object.""" return self.coresys.config @property def sys_hardware(self) -> Hardware: """Return Hardware object.""" return self.coresys.hardware @property def sys_docker(self) -> DockerAPI: """Return DockerAPI object.""" return self.coresys.docker @property def sys_scheduler(self) -> Scheduler: """Return Scheduler object.""" return self.coresys.scheduler @property def sys_core(self) -> Core: """Return core object.""" return self.coresys.core @property def sys_plugins(self) -> PluginManager: """Return PluginManager object.""" return self.coresys.plugins @property def sys_arch(self) -> CpuArch: """Return CpuArch object.""" return self.coresys.arch @property def sys_auth(self) -> Auth: """Return Auth object.""" return self.coresys.auth @property def sys_homeassistant(self) -> HomeAssistant: """Return Home Assistant object.""" return self.coresys.homeassistant @property def sys_supervisor(self) -> Supervisor: """Return Supervisor object.""" return self.coresys.supervisor @property def sys_api(self) -> RestAPI: """Return API object.""" return self.coresys.api @property def sys_updater(self) -> Updater: """Return Updater object.""" return self.coresys.updater @property def sys_addons(self) -> AddonManager: """Return AddonManager object.""" return self.coresys.addons @property def sys_store(self) -> StoreManager: """Return StoreManager object.""" return self.coresys.store @property def sys_snapshots(self) -> SnapshotManager: """Return SnapshotManager object.""" return self.coresys.snapshots @property def sys_tasks(self) -> Tasks: """Return Tasks object.""" return self.coresys.tasks @property def sys_services(self) -> ServiceManager: """Return ServiceManager object.""" return self.coresys.services @property def sys_discovery(self) -> Discovery: """Return ServiceManager object.""" return self.coresys.discovery @property def sys_dbus(self) -> DBusManager: """Return DBusManager object.""" return self.coresys.dbus @property def sys_host(self) -> HostManager: """Return HostManager object.""" return self.coresys.host @property def sys_hwmonitor(self) -> HwMonitor: """Return HwMonitor object.""" return self.coresys.hwmonitor @property def sys_ingress(self) -> Ingress: """Return Ingress object.""" return self.coresys.ingress @property def sys_hassos(self) -> HassOS: """Return HassOS object.""" return self.coresys.hassos @property def sys_resolution(self) -> ResolutionManager: """Return Resolution manager object.""" return self.coresys.resolution def sys_run_in_executor( self, funct: Callable[..., T], *args: Any ) -> Coroutine[Any, Any, T]: """Add an job to the executor pool.""" return self.sys_loop.run_in_executor(None, funct, *args) def sys_create_task(self, coroutine: Coroutine) -> asyncio.Task: """Create an async task.""" return self.sys_loop.create_task(coroutine) def sys_capture_exception(self, err: Exception) -> None: """Capture a exception.""" sentry_sdk.capture_exception(err)