From c53b95aa931993996431871e0ddee944d980d07c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Jousse?= <s.jousse@linoa.com>
Date: Wed, 5 May 2021 13:14:25 +0200
Subject: [PATCH] power: Add Home Assistant switch support
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* power: Add Home Assistant switch support

Signed-off-by:  Sébastien Jousse <s.jousse@gmail.com>
---
 docs/configuration.md         | 16 ++++++-
 moonraker/components/power.py | 79 +++++++++++++++++++++++++++++++++++
 2 files changed, 94 insertions(+), 1 deletion(-)

diff --git a/docs/configuration.md b/docs/configuration.md
index 8544818..86d9875 100644
--- a/docs/configuration.md
+++ b/docs/configuration.md
@@ -172,7 +172,7 @@ gcode:
 
 ## `[power]`
 Enables device power control.  Currently GPIO (relays), TPLink Smartplug,
-and Tasmota (via http) devices are supported.
+and Tasmota (via http) devices, HomeAssistant switch are supported.
 
 ```ini
 # moonraker.conf
@@ -258,6 +258,13 @@ password:
 #   "ID" in the "advanced information" section.
 #   Provide a user and password with access to "device control"
 #   and at least the specific device you want to control
+address:
+port:
+device:
+token:
+#   The above options are used for "homeassistant" devices.  The
+#   address should be a valid ip or hostname for the homeassistant controller.
+#   "device" should be the ID of the switch to control.
 
 ```
 Below are some potential examples:
@@ -294,6 +301,13 @@ type: shelly
 address: 192.168.1.125
 user: user2
 password: password2
+
+[power homeassistant_switch]
+type: homeassistant
+address: 192.168.1.126
+port: 8123
+device: switch.1234567890abcdefghij
+token: home-assistant-very-long-token
 ```
 
 It is possible to toggle device power from the Klippy host, this can be done
diff --git a/moonraker/components/power.py b/moonraker/components/power.py
index dcbf8fd..f45e532 100644
--- a/moonraker/components/power.py
+++ b/moonraker/components/power.py
@@ -38,6 +38,8 @@ class PrinterPower:
                     dev = Shelly(cfg)
                 elif dev_type == "homeseer":
                     dev = HomeSeer(cfg)
+                elif dev_type == "homeassistant":
+                    dev = HomeAssistant(cfg)
                 else:
                     raise config.error(f"Unsupported Device Type: {dev_type}")
                 self.devices[dev.get_name()] = dev
@@ -620,6 +622,83 @@ class HomeSeer(PowerDevice):
             raise self.server.error(msg) from None
         self.state = state
 
+class HomeAssistant(PowerDevice):
+    def __init__(self, config):
+        super().__init__(config)
+        self.server = config.get_server()
+        self.addr = config.get("address")
+        self.port = config.getint("port", 8123)
+        self.device = config.get("device")
+        self.token = config.get("token")
+
+    async def _send_homeassistant_command(self, command):
+        if command == "on":
+            out_cmd = f"api/services/switch/turn_on"
+            body = {"entity_id": self.device}
+            method = "POST"
+        elif command == "off":
+            out_cmd = f"api/services/switch/turn_off"
+            body = {"entity_id": self.device}
+            method = "POST"
+        elif command == "info":
+            out_cmd = f"api/states/{self.device}"
+            method = "GET"
+        else:
+            raise self.server.error(
+                f"Invalid homeassistant command: {command}")
+        url = f"http://{self.addr}:{self.port}/{out_cmd}"
+        headers = {
+            'Authorization': f'Bearer {self.token}',
+            'Content-Type': 'application/json'
+        }
+        data = ""
+        http_client = AsyncHTTPClient()
+        try:
+            if (method == "POST"):
+                response = await http_client.fetch(
+                    url, method="POST", body=json.dumps(body), headers=headers)
+                data = json_decode(response.body)
+            else:
+                response = await http_client.fetch(
+                    url, method="GET", headers=headers)
+                data = json_decode(response.body)
+        except Exception:
+            msg = f"Error sending homeassistant command: {command}"
+            logging.exception(msg)
+            raise self.server.error(msg)
+        return data
+
+    async def initialize(self):
+        await self.refresh_status()
+
+    def get_device_info(self):
+        return {
+            **super().get_device_info(),
+            'type': "homeassistant"
+        }
+
+    async def refresh_status(self):
+        try:
+            res = await self._send_homeassistant_command("info")
+            state = res[f"state"]
+        except Exception:
+            self.state = "error"
+            msg = f"Error Refeshing Device Status: {self.name}"
+            logging.exception(msg)
+            raise self.server.error(msg) from None
+        self.state = state
+
+    async def set_power(self, state):
+        try:
+            res = await self._send_homeassistant_command(state)
+            state = res[0][f"state"]
+        except Exception:
+            self.state = "error"
+            msg = f"Error Setting Device Status: {self.name} to {state}"
+            logging.exception(msg)
+            raise self.server.error(msg) from None
+        self.state = state
+
 # The power component has multiple configuration sections
 def load_component_multi(config):
     return PrinterPower(config)