From 8c9b47aa1d0597c4d3c8f9b250e8f5035aa1daf2 Mon Sep 17 00:00:00 2001
From: Stian Fredrik Aune <post@stian-aune.com>
Date: Sun, 7 Nov 2021 19:03:42 +0100
Subject: [PATCH] Add temperature to API.

---
 cmd/lucy/handlercmd.go | 1 +
 models/device.go       | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/cmd/lucy/handlercmd.go b/cmd/lucy/handlercmd.go
index 05a04d2..99c0146 100644
--- a/cmd/lucy/handlercmd.go
+++ b/cmd/lucy/handlercmd.go
@@ -157,6 +157,7 @@ func applyCmdToHandler(model models.EventHandler, cmd Command) models.EventHandl
 	model.Actions.SetPower = nds.Power
 	model.Actions.SetColor = nds.Color
 	model.Actions.SetIntensity = nds.Intensity
+	model.Actions.SetTemperature = nds.Temperature
 
 	// Add intensity
 	model.Actions.AddIntensity = cmd.Params.Get("add-intensity").Float()
diff --git a/models/device.go b/models/device.go
index b79a90a..23960ad 100644
--- a/models/device.go
+++ b/models/device.go
@@ -198,6 +198,10 @@ func (d *Device) SetState(newState NewDeviceState) error {
 		d.State.Intensity = *newState.Intensity
 	}
 
+	if newState.Temperature != nil && d.HasCapability(DCTemperatureControl) {
+		d.State.Temperature = *newState.Temperature
+	}
+
 	return nil
 }