update: 初始化代码提交

utility/ha_utils/ha_entity.go

@@ -0,0 +1,119 @@
+package ha_utils
+
+import (
+	"context"
+	"github.com/gogf/gf/v2/encoding/gjson"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gcache"
+	"home-network-watcher/manifest"
+	"time"
+)
+
+type uHaUtils struct{}
+
+var HaUtils = &uHaUtils{}
+
+// 卧室空调
+const (
+	AirConditionerEntityPattern = "climate.lumi_mcn02_1f58_air_conditioner"
+	AirConditionerStatePattern  = "attributes.air_conditioner.on"
+	AirConditionerTempPattern   = "attributes.temperature"
+)
+
+// 加湿器
+const (
+	HumidifierEntityPattern   = "humidifier.deerma_jsq2g_d916_humidifier"
+	HumidifierStatePattern    = "attributes.humidifier.on"
+	HumidifierHumidityPattern = "attributes.environment.relative_humidity"
+)
+
+// 空气净化器
+const (
+	AirPurifierEntityPattern = "fan.zhimi_mp4a_20cb_air_purifier"
+	AirPurifierStatePattern  = "attributes.air_purifier.on"
+	AirPurifierPM25Pattern   = "attributes.environment.pm2_5_density"
+)
+
+// 卧室床头灯
+const (
+	LightEntityPattern     = "light.yeelink_bslamp2_0b02_light"
+	LightStatePattern      = "attributes.light.on"
+	LightBrightnessPattern = "attributes.light.brightness"
+)
+
+type HaEntitiesInfo struct {
+	AirConditioner g.Map
+	Humidifier     g.Map
+	AirPurifier    g.Map
+	Light          g.Map
+}
+
+// GetEntitiesInfo
+//
+//	@dc: 获取实体信息
+//	@author: hamster   @date:2023/9/22 17:54:25
+func (u *uHaUtils) GetEntitiesInfo(ctx context.Context) (err error) {
+	haClient := g.Client().SetHeaderMap(manifest.HomeAssistantAuthMap)
+	baseUrl := manifest.HomeAssistantBaseUrl
+	// 获取空调状态
+	airConditionerMap := g.Map{}
+	response, err := haClient.Get(ctx, baseUrl+AirConditionerEntityPattern)
+	if err != nil {
+		return err
+	}
+	defer response.Close()
+	responseJson := gjson.New(response.ReadAllString())
+	responseJson.SetViolenceCheck(true)
+	airConditionerMap["state"] = responseJson.Get(AirConditionerStatePattern)
+	airConditionerMap["temp"] = responseJson.Get(AirConditionerTempPattern)
+
+	// 获取加湿器状态
+	humidifierMap := g.Map{}
+	response, err = haClient.Get(ctx, baseUrl+HumidifierEntityPattern)
+	if err != nil {
+		return err
+	}
+	defer response.Close()
+	responseJson = gjson.New(response.ReadAllString())
+	responseJson.SetViolenceCheck(true)
+	humidifierMap["state"] = responseJson.Get(HumidifierStatePattern)
+	humidifierMap["humidity"] = responseJson.Get(HumidifierHumidityPattern)
+
+	// 获取空气净化器状态
+	airPurifierMap := g.Map{}
+	response, err = haClient.Get(ctx, baseUrl+AirPurifierEntityPattern)
+	if err != nil {
+		return err
+	}
+	defer response.Close()
+	responseJson = gjson.New(response.ReadAllString())
+	responseJson.SetViolenceCheck(true)
+	airPurifierMap["state"] = responseJson.Get(AirPurifierStatePattern)
+	airPurifierMap["pm25"] = responseJson.Get(AirPurifierPM25Pattern)
+
+	// 获取床头灯状态
+	lightMap := g.Map{}
+	response, err = haClient.Get(ctx, baseUrl+LightEntityPattern)
+	if err != nil {
+		return err
+	}
+	defer response.Close()
+	responseJson = gjson.New(response.ReadAllString())
+	responseJson.SetViolenceCheck(true)
+	lightMap["state"] = responseJson.Get(LightStatePattern)
+	lightMap["brightness"] = responseJson.Get(LightBrightnessPattern)
+
+	// 汇总信息
+	HaEntitiesInfo := &HaEntitiesInfo{
+		AirConditioner: airConditionerMap,
+		Humidifier:     humidifierMap,
+		AirPurifier:    airPurifierMap,
+		Light:          lightMap,
+	}
+	err = gcache.Set(ctx, manifest.HaEntitiesCacheKey, HaEntitiesInfo, 1*time.Minute)
+	if err != nil {
+		return err
+	}
+
+	return
+}