Initial pass at temperature sensor for ZCL.

implcaps/zcl/temperature_sensor/impl.go

@@ -0,0 +1,117 @@
+package temperature_sensor
+
+import (
+	"context"
+	"github.com/shimmeringbee/da"
+	"github.com/shimmeringbee/da/capabilities"
+	"github.com/shimmeringbee/persistence"
+	"github.com/shimmeringbee/zcl"
+	"github.com/shimmeringbee/zda/attribute"
+	"github.com/shimmeringbee/zda/implcaps"
+	"github.com/shimmeringbee/zigbee"
+	"math"
+	"time"
+)
+
+var _ capabilities.TemperatureSensor = (*Implementation)(nil)
+var _ capabilities.WithLastChangeTime = (*Implementation)(nil)
+var _ capabilities.WithLastUpdateTime = (*Implementation)(nil)
+var _ implcaps.ZDACapability = (*Implementation)(nil)
+
+func NewTemperatureSensor(zi implcaps.ZDAInterface) *Implementation {
+	return &Implementation{zi: zi}
+}
+
+type Implementation struct {
+	s  persistence.Section
+	d  da.Device
+	am attribute.Monitor
+	zi implcaps.ZDAInterface
+}
+
+func (i *Implementation) Capability() da.Capability {
+	return capabilities.TemperatureSensorFlag
+}
+
+func (i *Implementation) Name() string {
+	return capabilities.StandardNames[capabilities.TemperatureSensorFlag]
+}
+
+func (i *Implementation) Init(d da.Device, s persistence.Section) {
+	i.d = d
+	i.s = s
+
+	i.am = i.zi.NewAttributeMonitor()
+	i.am.Init(s.Section("AttributeMonitor", "TemperatureReading"), d, i.update)
+}
+
+func (i *Implementation) Load(ctx context.Context) (bool, error) {
+	if err := i.am.Load(ctx); err != nil {
+		return false, err
+	} else {
+		return true, nil
+	}
+}
+
+func (i *Implementation) Enumerate(ctx context.Context, m map[string]interface{}) (bool, error) {
+	endpoint := implcaps.Get(m, "ZigbeeEndpoint", zigbee.Endpoint(1))
+	clusterId := implcaps.Get(m, "ZigbeeTemperatureSensorClusterID", zigbee.ClusterID(0x0402))
+	attributeId := implcaps.Get(m, "ZigbeeTemperatureSensorAttributeID", zcl.AttributeID(0x0000))
+	forcePolling := implcaps.Get(m, "ZigbeeTemperatureSensorForcePolling", false)
+
+	if err := i.am.Attach(ctx, endpoint, clusterId, attributeId, forcePolling); err != nil {
+		return false, err
+	}
+
+	return true, nil
+}
+
+func (i *Implementation) Detach(ctx context.Context, detachType implcaps.DetachType) error {
+	if err := i.am.Detach(ctx, detachType == implcaps.NoLongerEnumerated); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (i *Implementation) ImplName() string {
+	return "ZCLTemperatureSensor"
+}
+
+func (i *Implementation) update(_ zcl.AttributeID, v zcl.AttributeDataTypeValue) {
+	if v.DataType == zcl.TypeSignedInt16 {
+		if value, ok := v.Value.(int64); ok {
+			tempInK := (float64(value) / 100.0) + 273.15
+			currentTempInK, _ := i.s.Float("TemperatureReading", 0.0)
+
+			if math.Abs(tempInK-currentTempInK) > 0.01 {
+				i.s.Set("TemperatureReading", tempInK)
+				i.s.Set("LastChanged", time.Now().UnixMilli())
+
+				i.zi.SendEvent(capabilities.TemperatureSensorState{Device: i.d, State: []capabilities.TemperatureReading{{Value: tempInK}}})
+			}
+
+			i.s.Set("LastUpdated", time.Now().UnixMilli())
+		}
+	}
+}
+
+func (i *Implementation) LastUpdateTime(_ context.Context) (time.Time, error) {
+	t, _ := i.s.Int("LastUpdated", 0.0)
+	return time.UnixMilli(int64(t)), nil
+}
+
+func (i *Implementation) LastChangeTime(_ context.Context) (time.Time, error) {
+	t, _ := i.s.Int("LastChanged", 0.0)
+	return time.UnixMilli(int64(t)), nil
+}
+
+func (i *Implementation) Reading(_ context.Context) ([]capabilities.TemperatureReading, error) {
+	k, _ := i.s.Float("TemperatureReading", 0.0)
+
+	return []capabilities.TemperatureReading{
+		{
+			Value: k,
+		},
+	}, nil
+}

implcaps/zcl/temperature_sensor/impl_test.go

@@ -0,0 +1,17 @@
+package temperature_sensor
+
+import (
+	"github.com/shimmeringbee/da/capabilities"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestImplementation_BaseFunctions(t *testing.T) {
+	t.Run("basic static functions respond correctly", func(t *testing.T) {
+		i := NewTemperatureSensor()
+
+		assert.Equal(t, capabilities.TemperatureSensorFlag, i.Capability())
+		assert.Equal(t, capabilities.StandardNames[capabilities.TemperatureSensorFlag], i.Name())
+		assert.Equal(t, "ZCLTemperatureSensor", i.ImplName())
+	})
+}