"""Class for integrations in HACS.""" from integrationhelper import Logger from homeassistant.loader import async_get_custom_components from custom_components.hacs.hacsbase.exceptions import HacsException from custom_components.hacs.helpers.filters import get_first_directory_in_directory from custom_components.hacs.helpers.information import get_integration_manifest from custom_components.hacs.repositories.repository import HacsRepository class HacsIntegration(HacsRepository): """Integrations in HACS.""" def __init__(self, full_name): """Initialize.""" super().__init__() self.data.full_name = full_name self.data.category = "integration" self.content.path.remote = "custom_components" self.content.path.local = self.localpath self.logger = Logger(f"hacs.repository.{self.data.category}.{full_name}") @property def localpath(self): """Return localpath.""" return f"{self.hacs.system.config_path}/custom_components/{self.data.domain}" async def validate_repository(self): """Validate.""" await self.common_validate() # Custom step 1: Validate content. if self.data.content_in_root: self.content.path.remote = "" if self.content.path.remote == "custom_components": name = get_first_directory_in_directory(self.tree, "custom_components") if name is None: raise HacsException( f"Repostitory structure for {self.ref.replace('tags/','')} is not compliant" ) self.content.path.remote = f"custom_components/{name}" try: await get_integration_manifest(self) except HacsException as exception: self.logger.error(exception) # Handle potential errors if self.validate.errors: for error in self.validate.errors: if not self.hacs.system.status.startup: self.logger.error(error) return self.validate.success async def registration(self): """Registration.""" if not await self.validate_repository(): return False # Run common registration steps. await self.common_registration() # Set local path self.content.path.local = self.localpath async def update_repository(self): """Update.""" if self.hacs.github.ratelimits.remaining == 0: return await self.common_update() if self.data.content_in_root: self.content.path.remote = "" if self.content.path.remote == "custom_components": name = get_first_directory_in_directory(self.tree, "custom_components") self.content.path.remote = f"custom_components/{name}" try: await get_integration_manifest(self) except HacsException as exception: self.logger.error(exception) # Set local path self.content.path.local = self.localpath async def reload_custom_components(self): """Reload custom_components (and config flows)in HA.""" self.logger.info("Reloading custom_component cache") del self.hacs.hass.data["custom_components"] await async_get_custom_components(self.hacs.hass)