Add Axion Lighting integration

CODEOWNERS

@@ -166,6 +166,8 @@ build.json @home-assistant/supervisor
 /homeassistant/components/avea/ @pattyland
 /homeassistant/components/awair/ @ahayworth @danielsjf
 /tests/components/awair/ @ahayworth @danielsjf
+/homeassistant/components/axion_dmx/ @Vrncanac
+/tests/components/axion_dmx/ @Vrncanac
 /homeassistant/components/axis/ @Kane610
 /tests/components/axis/ @Kane610
 /homeassistant/components/azure_data_explorer/ @kaareseras

homeassistant/components/axion_dmx/__init__.py

@@ -0,0 +1,45 @@
+"""The Axion Lighting integration."""
+
+from __future__ import annotations
+
+from libaxion_dmx import AxionDmxApi  # Import the API class
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant
+
+from .const import CONF_LIGHT_TYPE, DOMAIN
+
+PLATFORMS: list[Platform] = [Platform.LIGHT]
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    """Set up Axion Lighting from a config entry."""
+
+    hass.data.setdefault(DOMAIN, {})
+
+    # Create API instance
+    api = AxionDmxApi(entry.data["host"], entry.data["password"])
+
+    # Validate the API connection (and authentication)
+    if not await api.authenticate():
+        return False
+
+    # Store an API object for your platforms to access
+    hass.data[DOMAIN][entry.entry_id] = {
+        "api": api,
+        "channel": entry.data["channel"],
+        "light_type": entry.data[CONF_LIGHT_TYPE],
+    }
+
+    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+
+    return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    """Unload a config entry."""
+    if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
+        hass.data[DOMAIN].pop(entry.entry_id)
+
+    return unload_ok

homeassistant/components/axion_dmx/config_flow.py

@@ -0,0 +1,73 @@
+"""Config flow for Axion Lighting integration."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from libaxion_dmx import AxionDmxApi
+import voluptuous as vol
+
+from homeassistant import config_entries
+from homeassistant.config_entries import ConfigFlowResult
+from homeassistant.const import CONF_HOST, CONF_PASSWORD
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+
+from .const import _LOGGER, CONF_CHANNEL, CONF_LIGHT_TYPE, DOMAIN
+
+STEP_USER_DATA_SCHEMA = vol.Schema(
+    {
+        vol.Required(CONF_HOST): str,
+        vol.Required(CONF_PASSWORD): str,
+        vol.Required(CONF_CHANNEL): vol.All(vol.Coerce(int), vol.Range(min=1)),
+        vol.Required(CONF_LIGHT_TYPE): vol.In(
+            ["White", "Tunable White", "RGB", "RGBW", "RGBWW"]
+        ),
+    }
+)
+
+
+async def validate_input(hass: HomeAssistant, data: dict[str, Any]) -> dict[str, Any]:
+    """Validate the user input allows us to connect."""
+    api = AxionDmxApi(data[CONF_HOST], data[CONF_PASSWORD])
+
+    if not await api.authenticate():
+        raise InvalidAuth
+
+    return {"title": f"Axion DMX Light - Channel {data[CONF_CHANNEL]}"}
+
+
+class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
+    """Handle a config flow for Axion Lighting."""
+
+    VERSION = 1
+
+    async def async_step_user(
+        self, user_input: dict[str, Any] | None = None
+    ) -> ConfigFlowResult:
+        """Handle the initial step."""
+        errors: dict[str, str] = {}
+        if user_input is not None:
+            try:
+                info = await validate_input(self.hass, user_input)
+            except CannotConnect:
+                errors["base"] = "cannot_connect"
+            except InvalidAuth:
+                errors["base"] = "invalid_auth"
+            except Exception as e:  # noqa: BLE001
+                _LOGGER.exception("Unexpected exception: %s", e)
+                errors["base"] = "unknown"
+            else:
+                return self.async_create_entry(title=info["title"], data=user_input)
+
+        return self.async_show_form(
+            step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors
+        )
+
+
+class CannotConnect(HomeAssistantError):
+    """Error to indicate we cannot connect."""
+
+
+class InvalidAuth(HomeAssistantError):
+    """Error to indicate there is invalid auth."""

homeassistant/components/axion_dmx/const.py

@@ -0,0 +1,9 @@
+"""Constants for the Axion Lighting integration."""
+
+import logging
+
+_LOGGER = logging.getLogger(__name__)
+
+DOMAIN = "axion_dmx"
+CONF_CHANNEL = "channel"
+CONF_LIGHT_TYPE = "light_type"