Add Sleepiq services

CODEOWNERS

@@ -312,6 +312,7 @@ homeassistant/components/sighthound/* @robmarkcole
 homeassistant/components/signal_messenger/* @bbernhard
 homeassistant/components/simplisafe/* @bachya
 homeassistant/components/sinch/* @bendikrb
+homeassistant/components/sleepiq/* @Jay2645
 homeassistant/components/sisyphus/* @jkeljo
 homeassistant/components/slide/* @ualex73
 homeassistant/components/sma/* @kellerza

homeassistant/components/sleepiq/__init__.py

@@ -19,10 +19,16 @@
 SLEEP_NUMBER = "sleep_number"
 SENSOR_TYPES = {SLEEP_NUMBER: "SleepNumber", IS_IN_BED: "Is In Bed"}
 
+BED = "bed"
 LEFT = "left"
+PRESET_FAVORITE = 0
 RIGHT = "right"
+SIDE = "side"
 SIDES = [LEFT, RIGHT]
 
+SERVICE_SET_SLEEP_NUMBER = "set_sleep_number"
+SERVICE_SET_FAVORITE = "set_to_favorite_sleep_number"
+
 _LOGGER = logging.getLogger(__name__)
 
 DATA = None
@@ -61,6 +67,22 @@ def setup(hass, config):
         _LOGGER.error(message)
         return False
 
+    def handle_set_number(call):
+        bed = call.data.get(BED, "")
+        side = call.data.get(SIDE, "")
+        new_sleep_number = call.data.get(SLEEP_NUMBER, 0)
+
+        DATA.set_new_sleep_number(bed, side, new_sleep_number)
+
+    def handle_set_favorite(call):
+        bed = call.data.get(BED, "")
+        side = call.data.get(SIDE, "")
+
+        DATA.set_to_favorite_sleep_number(bed, side)
+
+    hass.services.register(DOMAIN, SERVICE_SET_SLEEP_NUMBER, handle_set_number)
+    hass.services.register(DOMAIN, SERVICE_SET_FAVORITE, handle_set_favorite)
+
     discovery.load_platform(hass, "sensor", DOMAIN, {}, config)
     discovery.load_platform(hass, "binary_sensor", DOMAIN, {}, config)
 
@@ -85,6 +107,54 @@ def update(self):
 
         self.beds = {bed.bed_id: bed for bed in beds}
 
+    def set_new_sleep_number(self, bed_name: str, side: str, sleep_number):
+        """Change the sleep number on a side to a new value."""
+
+        # Sanity check: Sleep number has to be between 1-100.
+        if 0 < sleep_number <= 100:
+            self._set_sleep_number(bed_name, side, int(sleep_number))
+        else:
+            message = f"Invalid sleep number: {sleep_number}"
+            _LOGGER.warning(message)
+
+    def set_to_favorite_sleep_number(self, bed_name: str, side: str):
+        """Change a side's sleep number to their 'favorite' setting."""
+        self._set_sleep_number(bed_name, side, PRESET_FAVORITE)
+
+    def get_favorite_sleep_number(self, bed_id: int, side: str) -> int:
+        """Get a side's 'favorite' number, given a bed ID and a side."""
+        favorite_numbers = self._client.get_favsleepnumber(bed_id)
+        return getattr(favorite_numbers, side.lower())
+
+    def _set_sleep_number(self, bed_name: str, side: str, new_number: int):
+        """Set the sleep number for a side on a bed."""
+        bed_name = bed_name.lower()
+        side = side.lower()
+
+        if len(side) == 0:
+            # No side specified, use both.
+            sides_to_set = SIDES
+        else:
+            # Just use the side we specified.
+            sides_to_set = [side]
+
+        for bed_id, bed_obj in self.beds.items():
+            # If no bed name is specified, set on all beds.
+            # Otherwise, ensure we only set on beds with the correct name.
+            if len(bed_name) == 0 or bed_name == bed_obj.name.lower():
+                for bed_side in sides_to_set:
+                    # Check if we should use the "favorite" preset instead.
+                    if new_number == PRESET_FAVORITE:
+                        num = self.get_favorite_sleep_number(bed_id, bed_side)
+                    else:
+                        num = new_number
+
+                    debug_msg = f"Setting {bed_side} of bed {bed_id} to {num}"
+                    _LOGGER.debug(debug_msg)
+
+                    # Actually set the Sleep Number.
+                    self._client.set_sleepnumber(bed_id, bed_side, num)
+
 
 class SleepIQSensor(Entity):
     """Implementation of a SleepIQ sensor."""

homeassistant/components/sleepiq/manifest.json

@@ -4,5 +4,5 @@
   "documentation": "https://www.home-assistant.io/integrations/sleepiq",
   "requirements": ["sleepyq==0.7"],
   "dependencies": [],
-  "codeowners": []
+  "codeowners": ["@Jay2645"]
 }

homeassistant/components/sleepiq/services.yaml

@@ -0,0 +1,22 @@
+set_sleep_number:
+  description: Changes the Sleep Number of a bed.
+  fields:
+    sleep_number:
+      description: The new sleep number, as a multiple of 5 between 5 and 100.
+      example: 45
+    bed:
+      description: The name of the bed. If unspecified, affects all beds.
+      example: 'Master Bedroom'
+    side:
+      description: The side to change. If unspecified, both sides will change.
+      example: 'right'
+
+set_to_favorite_sleep_number:
+  description: Changes a bed's Sleep Number to the "favorite" preset.
+  fields:
+    bed:
+      description: The name of the bed. If unspecified, affects all beds.
+      example: 'Master Bedroom'
+    side:
+      description: The side to change. If unspecified, both sides will change.
+      example: 'right'