Add prototype

Author: alexhorn

README.md

@@ -1,2 +1,3 @@
-# laemp
-A bridge that allows you to control your Bluetooth-capable Philips Hue lights via HomeKit
+# lämp
+
+A bridge that allows you to control your Bluetooth-capable Philips Hue lights via HomeKit.

hap_bridge.py

@@ -0,0 +1,22 @@
+import signal
+from pyhap.accessory import Bridge
+from pyhap.accessory_driver import AccessoryDriver
+from hap_lamp import LampAccessory
+
+LAMPS = {
+    'Hue Go': '00:11:22:33:44:55:66'
+}
+
+def get_bridge(driver):
+    bridge = Bridge(driver, 'Laemp Bridge')
+    for name, address in LAMPS.items():
+        lamp = LampAccessory(driver, display_name=name, address=address)
+        bridge.add_accessory(lamp)
+    return bridge
+
+driver = AccessoryDriver(port=51826)
+driver.add_accessory(accessory=get_bridge(driver))
+
+signal.signal(signal.SIGTERM, driver.signal_handler)
+
+driver.start()

hap_lamp.py

@@ -0,0 +1,48 @@
+from pyhap.accessory import Accessory
+from pyhap.const import CATEGORY_LIGHTBULB
+from lamp import Lamp
+from colorsys import rgb_to_hsv, hsv_to_rgb
+import asyncio
+
+class LampAccessory(Accessory):
+
+    category = CATEGORY_LIGHTBULB
+
+    def __init__(self, *args, address, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.lamp = Lamp(address)
+
+        serv_light = self.add_preload_service('Lightbulb', chars=['On', 'Hue', 'Saturation', 'Brightness'])
+        self.char_on = serv_light.configure_char('On', setter_callback=self._async_callback(self.set_on))
+        self.char_hue = serv_light.configure_char('Hue', setter_callback=self._async_callback(self.set_hue))
+        self.char_saturation = serv_light.configure_char('Saturation', setter_callback=self._async_callback(self.set_saturation))
+        self.char_brightness = serv_light.configure_char('Brightness', setter_callback=self._async_callback(self.set_brightness))
+
+    def _async_callback(self, coro):
+        return lambda val: asyncio.create_task(coro(val))
+
+    async def run(self):
+        self.loop = asyncio.get_event_loop()
+        self.lock = asyncio.Lock(loop=self.loop)
+        await self.lamp.connect()
+
+    async def stop(self):
+        await self.lamp.disconnect()
+
+    async def set_on(self, value):
+        async with self.lock:
+            await self.lamp.set_power(value)
+
+    async def set_hue(self, hue):
+        async with self.lock:
+            _, s, b = rgb_to_hsv(*await self.lamp.get_color_rgb())
+            await self.lamp.set_color_rgb(*hsv_to_rgb(hue / 360, s, b))
+
+    async def set_saturation(self, saturation):
+        async with self.lock:
+            h, _, b = rgb_to_hsv(*await self.lamp.get_color_rgb())
+            await self.lamp.set_color_rgb(*hsv_to_rgb(h, saturation / 100, b))
+
+    async def set_brightness(self, brightness):
+        async with self.lock:
+            await self.lamp.set_brightness(brightness / 100)

lamp.py

@@ -0,0 +1,51 @@
+from bleak import BleakClient
+from rgbxy import Converter, GamutC
+from struct import pack, unpack
+
+CHAR_POWER          = "932c32bd-0002-47a2-835a-a8d455b859dd"
+CHAR_BRIGHTNESS     = "932c32bd-0003-47a2-835a-a8d455b859dd"
+CHAR_COLOR          = "932c32bd-0005-47a2-835a-a8d455b859dd"
+
+# Hue Go uses GamutC
+# you might need GamutA or GamutB if you have a different model
+converter = Converter(GamutC)
+
+class Lamp(object):
+    def __init__(self, address):
+        self.client = BleakClient(address)
+
+    async def connect(self):
+        await self.client.connect()
+
+    async def disconnect(self):
+        await self.client.disconnect()
+
+    async def get_power(self):
+        return await self.client.read_gatt_char(CHAR_POWER)[0]
+
+    async def set_power(self, on):
+        await self.client.write_gatt_char(CHAR_POWER, bytes([1 if on else 0]), response=True)
+
+    async def get_brightness(self):
+        return await self.client.read_gatt_char(CHAR_BRIGHTNESS)[0] / 255
+
+    async def set_brightness(self, brightness):
+        await self.client.write_gatt_char(CHAR_BRIGHTNESS, bytes([max(min(int(brightness * 255), 254), 1)]), response=True)
+
+    async def get_color_xy(self):
+        buf = await self.client.read_gatt_char(CHAR_COLOR)
+        x, y = unpack('<HH', buf)
+        return x / 0xFFFF, y / 0xFFFF
+
+    async def set_color_xy(self, x, y):
+        # Hue expects CIE XY coordinates converted to two 16-bit little-endian integers
+        buf = pack('<HH', int(x * 0xFFFF), int(y * 0xFFFF))
+        await self.client.write_gatt_char(CHAR_COLOR, buf, response=True)
+
+    async def get_color_rgb(self):
+        x, y = await self.get_color_xy()
+        return converter.xy_to_rgb(x, y)
+
+    async def set_color_rgb(self, r, g, b):
+        x, y = converter.rgb_to_xy(r, g, b)
+        await self.set_color_xy(x, y)

requirements.txt

@@ -0,0 +1,3 @@
+bleak==0.11.0
+HAP-python[QRCode]==3.4.1
+git+https://github.com/benknight/hue-python-rgb-converter.git@f73a4ec#egg=rgbxy