From a29cce0d3d814e94b6bfcbfdace91c9f6d6956bd Mon Sep 17 00:00:00 2001
From: tomaae <23486452+tomaae@users.noreply.github.com>
Date: Wed, 2 Dec 2020 13:56:50 +0100
Subject: [PATCH] Added sensors for environment variables #58

---
 .../mikrotik_router/mikrotik_controller.py    | 19 ++++
 custom_components/mikrotik_router/sensor.py   | 87 +++++++++++++++++++
 2 files changed, 106 insertions(+)

diff --git a/custom_components/mikrotik_router/mikrotik_controller.py b/custom_components/mikrotik_router/mikrotik_controller.py
index bc7d030..03c6dd4 100644
--- a/custom_components/mikrotik_router/mikrotik_controller.py
+++ b/custom_components/mikrotik_router/mikrotik_controller.py
@@ -74,6 +74,7 @@ class MikrotikControllerData:
             "host": {},
             "host_hass": {},
             "accounting": {},
+            "environment": {},
         }
 
         self.listeners = []
@@ -424,6 +425,9 @@ class MikrotikControllerData:
         if self.api.connected():
             await self.hass.async_add_executor_job(self.get_queue)
 
+        if self.api.connected():
+            await self.hass.async_add_executor_job(self.get_environment)
+
         async_dispatcher_send(self.hass, self.signal_update)
         self.lock.release()
 
@@ -797,6 +801,21 @@ class MikrotikControllerData:
             ],
         )
 
+    # ---------------------------
+    #   get_environment
+    # ---------------------------
+    def get_environment(self):
+        """Get list of all environment variables from Mikrotik"""
+        self.data["environment"] = parse_api(
+            data={},
+            source=self.api.path("/system/script/environment"),
+            key="name",
+            vals=[
+                {"name": "name"},
+                {"name": "value"},
+            ],
+        )
+
     # ---------------------------
     #   get_queue
     # ---------------------------
diff --git a/custom_components/mikrotik_router/sensor.py b/custom_components/mikrotik_router/sensor.py
index d3a6197..4ccce84 100644
--- a/custom_components/mikrotik_router/sensor.py
+++ b/custom_components/mikrotik_router/sensor.py
@@ -226,6 +226,22 @@ def update_items(inst, mikrotik_controller, async_add_entities, sensors):
                     )
                     new_sensors.append(sensors[item_id])
 
+    for uid in mikrotik_controller.data["environment"]:
+        item_id = f"{inst}-{sensor}-{mikrotik_controller.data['environment'][uid]['name']}"
+        _LOGGER.debug("Updating sensor %s", item_id)
+        if item_id in sensors:
+            if sensors[item_id].enabled:
+                sensors[item_id].async_schedule_update_ha_state()
+            continue
+
+        sensors[item_id] = MikrotikControllerEnvironmentSensor(
+            mikrotik_controller=mikrotik_controller,
+            inst=inst,
+            sensor=sensor,
+            uid=uid,
+        )
+        new_sensors.append(sensors[item_id])
+
     if new_sensors:
         async_add_entities(new_sensors, True)
 
@@ -439,3 +455,74 @@ class MikrotikAccountingSensor(MikrotikControllerSensor):
             self._data["mac-address"],
             self._sensor,
         )
+
+# ---------------------------
+#   MikrotikControllerEnvironmentSensor
+# ---------------------------
+class MikrotikControllerEnvironmentSensor(MikrotikControllerSensor):
+    """Define an Mikrotik Controller sensor."""
+
+    def __init__(self, mikrotik_controller, inst, sensor, uid):
+        """Initialize."""
+        super().__init__(mikrotik_controller, inst, sensor)
+        self._uid = uid
+        self._data = mikrotik_controller.data['environment'][uid]
+        self._attr = "value"
+
+    @property
+    def name(self):
+        """Return the name."""
+        return f"{self._inst} {self._data['name']}"
+
+    @property
+    def unit_of_measurement(self):
+        """Return the unit the value is expressed in."""
+
+    @property
+    def unique_id(self):
+        """Return a unique_id for this entity."""
+        return f"{self._inst.lower()}-environment-{self._uid}"
+
+    @property
+    def state(self):
+        """Return the state."""
+        val = "unknown"
+        if self._attr in self._data:
+            val = self._data[self._attr]
+
+        return val
+
+    @property
+    def icon(self):
+        """Return the icon."""
+        self._icon = "mdi:clipboard-list"
+        return self._icon
+
+    @property
+    def device_info(self):
+        """Return a port description for device registry."""
+        info = {
+            "manufacturer": self._ctrl.data["resource"]["platform"],
+            "model": self._ctrl.data["resource"]["board-name"],
+            "name": f"{self._inst} Environment",
+            "identifiers": {
+                (
+                    DOMAIN,
+                    "serial-number",
+                    self._ctrl.data["routerboard"]["serial-number"],
+                    "switch",
+                    "environment",
+                )
+            },
+        }
+
+        return info
+
+    async def async_added_to_hass(self):
+        """Port entity created."""
+        _LOGGER.debug(
+            "New sensor %s (%s %s)",
+            self._inst,
+            self._data['name'],
+            self._sensor,
+        )
\ No newline at end of file