diff --git a/internal/xiaomi/xiaomi.go b/internal/xiaomi/xiaomi.go index a5b23420..1e578420 100644 --- a/internal/xiaomi/xiaomi.go +++ b/internal/xiaomi/xiaomi.go @@ -262,7 +262,7 @@ func apiDeviceList(w http.ResponseWriter, r *http.Request) { var items []*api.Source for _, device := range v.List { - if !strings.Contains(device.Model, ".camera.") && !strings.Contains(device.Model, ".cateye.") { + if !device.HasCamera() { continue } items = append(items, &api.Source{ @@ -289,6 +289,12 @@ type Device struct { IP string `json:"localip"` } +func (d *Device) HasCamera() bool { + return strings.Contains(d.Model, ".camera.") || + strings.Contains(d.Model, ".cateye.") || + strings.Contains(d.Model, ".feeder.") +} + var auth *xiaomi.Cloud func apiAuth(w http.ResponseWriter, r *http.Request) {