Runtimes and controllers

src/HomeAssistant/Controller.hs

@@ -22,12 +22,13 @@ module HomeAssistant.Controller
   , door
   , light
   , lightController
+  , bedroomPresenceController
   ) where
 
 import AFRP (Mealy, eff, Event(..), hold, events, changes, filterA, (>>|), toEvent)
 import Control.Arrow (Arrow(..), returnA)
 import Control.Category ((>>>))
-import Data.Aeson (Value)
+import Data.Aeson (Value, object, (.=))
 import qualified Data.Text as T
 import Control.Lens (has, only, (^?), to)
 import Data.Aeson.Lens (key, _String)
@@ -38,13 +39,12 @@ data Service = Service
   { serviceDomain :: T.Text
   , serviceName :: T.Text
   , serviceData :: Maybe Value
-  , serviceTarget :: T.Text
+  , serviceTarget :: [T.Text]
   }
   deriving (Show, Eq)
 
 data HASSEff a where
   CallService :: Service -> HASSEff ()
-  Pure :: a -> HASSEff a
 
 type HASS a b = Mealy HASSEff a b
 
@@ -66,6 +66,49 @@ ruuvi = (Ruuvi <$> ruuviTemperatures <*> ruuviPressures) >>> changes
 data DoorState = Open | Closed
   deriving (Show, Eq)
 
+data Presence = Occupied | Unoccupied
+  deriving (Show, Eq)
+
+presence :: T.Text -> HASS (Event Value) (Event Presence)
+presence entityId =entityBool entityId
+  >>> arr (fmap (bool Unoccupied Occupied))
+
+bedroomLights :: [T.Text]
+bedroomLights = ["light.bedroom_masse","light.bedroom_jemina","light.bedroom_ceiling","light.bedroom_led_switch"]
+
+createBedroomScene :: Service
+createBedroomScene = Service
+  { serviceDomain="scene"
+  , serviceName= "create"
+  , serviceData= Just $ object
+      [ "scene_id" .= ("makuuhuone_lights_snapshot" :: T.Text)
+      , "snapshot_entities" .= bedroomLights
+      ]
+  , serviceTarget= [] -- bedroomLights
+  }
+
+
+activateScene :: T.Text -> Service
+activateScene sceneId = Service
+  { serviceDomain="scene"
+  , serviceName= "turn_on"
+  , serviceTarget= [sceneId]
+  , serviceData = Nothing
+  }
+
+-- Bedroom presence
+bedroomPresence :: HASS (Event Value) (Event Presence)
+bedroomPresence = presence "binary_sensor.presence_sensor_bedroom_occupancy"
+
+bedroomPresenceController :: HASS (Event Value) ()
+bedroomPresenceController = proc x -> do
+  p <- bedroomPresence -< x
+  case p of
+       Event Unoccupied -> callService createBedroomScene >>> callService (light bedroomLights False) -< ()
+       Event Occupied -> callService (activateScene "makuuhuone_lights_snapshot") -< ()
+       _ -> returnA -< ()
+
+
 door :: HASS (Event Value) (Event DoorState)
 door = entityBool "binary_sensor.makuuhuone_ovi_contact"
   >>> arr (fmap (bool Closed Open))
@@ -73,20 +116,20 @@ door = entityBool "binary_sensor.makuuhuone_ovi_contact"
   >>> changes
 
 -- Turn off lights when door is closed
-light :: Bool -> Service
-light b = Service
+light :: [T.Text] -> Bool -> Service
+light entityIds b = Service
   { serviceDomain="light"
   , serviceName= bool "turn_off" "turn_on" b
   , serviceData=Nothing
-  , serviceTarget="light.bedroom_masse"
+  , serviceTarget=entityIds
   }
 
 lightController :: HASS (Event Value) (Event DoorState)
 lightController = proc ev -> do
   doorState <- door -< ev
   case doorState of
-       Event Open -> callService (light False) -< ()
-       Event Closed -> callService (light True) -< ()
+       Event Open -> callService (light ["light.bedroom_masse"] False) -< ()
+       Event Closed -> callService (light ["light.bedroom_masse"] True) -< ()
        _ -> returnA -< ()
   returnA -< doorState