Files
xiaomihome/xiaomi/classify.go
T
4566704 a3d94c4b9f chore: 提交高层 xiaomi 封装库及示例
- 新增 xiaomi/examples/ 下 8 个示例程序(用户/家庭查询、设备列表、开关/灯光/空调控制、属性订阅、高级过滤分类、SPEC 解析)
- miot_client_sub.go: 重构 SubProp/SubEvent,使用 buildPropTopic/buildEventTopic 支持通配符订阅(siid/piid=0 → +),并修复锁顺序问题(将 RequestRefreshProp 移到 Lock 外)
- spec_parser.go: 新增 downloadSpecFile 方法,本地 SPEC 文件缺失时自动从 miot-spec.org 下载
- ARCH_PLAN.md: 架构设计从 Proposed 更新为 Accepted(v1.0→v1.1),补充设备分类/工厂/SPEC 映射等模块设计
- xiaomi/: 新增 miot 上层强类型封装模块,包含 Client 主入口、用户/家庭/设备 API、属性读写、动作调用、订阅通知,以及 devices/ 设备控制抽象(Switch/Light/AirConditioner/Fan/Cover/Humidifier/Vacuum/WaterHeater/Thermostat)和 specs/ SPEC 查询辅助
- 更多 xiaomi 示例(风扇/窗帘/传感器控制)
2026-06-29 08:51:21 +08:00

100 lines
2.9 KiB
Go

package xiaomi
import "strings"
// deviceTypeToKind maps SPEC URN device types to Kind.
// URN format: urn:miot-spec-v2:device:{TYPE}:...
var deviceTypeToKind = map[string]Kind{
"air-conditioner": KindAirConditioner,
"air-condition-outlet": KindAirConditioner,
"air-condition-eco": KindAirConditioner,
"airer": KindAirConditioner,
"ac-partner": KindAirConditioner,
"light": KindLight,
"lightbulb": KindLight,
"night-light": KindLight,
"ceiling-light": KindLight,
"switch": KindSwitch,
"outlet": KindSwitch,
"powerstrip": KindSwitch,
"toggle-switch": KindSwitch,
"wireless-switch": KindSwitch,
"curtain-switch": KindSwitch,
"fan": KindFan,
"ceiling-fan": KindFan,
"ventilator": KindFan,
"air-fresh": KindFan,
"air-purifier": KindFan,
"curtain": KindCover,
"airer-rack": KindCover,
"humidifier": KindHumidifier,
"dehumidifier": KindHumidifier,
"humidifier-dehumidifier": KindHumidifier,
"vacuum": KindVacuum,
"sweeper": KindVacuum,
"robot-cleaner": KindVacuum,
"water-heater": KindWaterHeater,
"water-cooler": KindWaterHeater,
"kettle": KindWaterHeater,
"heater": KindThermostat,
"electric-blanket": KindThermostat,
"thermostat": KindThermostat,
"sensor": KindSensor,
"sensor-ht": KindSensor,
"sensor-occupy": KindSensor,
"motion-sensor": KindSensor,
"magnet": KindSensor,
"sensor-smoke": KindSensor,
"sensor-gasm": KindSensor,
"sensor-temp": KindSensor,
"sensor-humidity": KindSensor,
"illumination": KindSensor,
}
// Classify 从 DeviceInfo 判断设备种类。
// 优先从 URN 的 device type 段精确提取,降至 model 子串匹配。
func Classify(info *DeviceInfo) Kind {
if info == nil {
return KindUnknown
}
// 从 URN 精确解析 device type
if k := classifyByURN(info.URN); k != KindUnknown {
return k
}
// 降至 model 匹配
if k := classifyByModel(info.Model); k != KindUnknown {
return k
}
return KindUnknown
}
// classifyByURN 从 URN 中提取 device type 并映射 Kind。
func classifyByURN(urn string) Kind {
parts := strings.Split(urn, ":")
if len(parts) >= 4 && parts[2] == "device" {
if k, ok := deviceTypeToKind[parts[3]]; ok {
return k
}
}
return KindUnknown
}
// classifyByModel 对 model 字符串做子串匹配(用于无 URN 或 URN 不标准时)。
func classifyByModel(model string) Kind {
model = strings.ToLower(model)
for t, k := range deviceTypeToKind {
if strings.Contains(model, t) {
return k
}
}
return KindUnknown
}
// FilterByKind 按设备种类过滤。
func FilterByKind(kind Kind) DeviceFilter {
return func(d *DeviceInfo) bool { return d.DeviceKind() == kind }
}