Add DOMAIN_OVERRIDES workaround (#2468)

hacs · Jan 23, 2022 · ec78d32 · ec78d32
1 parent 1902734
commit ec78d32
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/custom_components/hacs/repositories/base.py b/custom_components/hacs/repositories/base.py
@@ -39,6 +39,7 @@
     version_left_higher_then_right,
     version_to_download,
 )
+from ..utils.workarounds import DOMAIN_OVERRIDES
 
 if TYPE_CHECKING:
     from ..base import HacsBase
@@ -725,8 +726,12 @@ async def remove_local_directory(self) -> None:
                 local_path = self.content.path.local
             elif self.data.category == "integration":
                 if not self.data.domain:
-                    self.logger.error("%s Missing domain", self)
-                    return False
+                    if domain := DOMAIN_OVERRIDES.get(self.data.full_name):
+                        self.data.domain = domain
+                        self.content.path.local = self.localpath
+                    else:
+                        self.logger.error("%s Missing domain", self)
+                        return False
                 local_path = self.content.path.local
             else:
                 local_path = self.content.path.local

diff --git a/custom_components/hacs/utils/workarounds.py b/custom_components/hacs/utils/workarounds.py
@@ -0,0 +1,7 @@
+"""Workarounds for issues that should not be fixed."""
+
+
+DOMAIN_OVERRIDES = {
+    # https://github.com/hacs/integration/issues/2465
+    "custom-components/sensor.custom_aftership": "custom_aftership"
+}
diff --git a/tests/utils/test_workarounds.py b/tests/utils/test_workarounds.py
@@ -0,0 +1,6 @@
+from custom_components.hacs.utils.workarounds import DOMAIN_OVERRIDES
+
+
+def test_domain_ovverides() -> None:
+    assert DOMAIN_OVERRIDES.get("custom-components/sensor.custom_aftership") == "custom_aftership"
+    assert DOMAIN_OVERRIDES.get("awesome/repo") is None