fix: only allow entities with platform="sonos"

punxaphil · Dec 29, 2023 · 9c08d48 · 9c08d48
1 parent 16635c6
commit 9c08d48
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 5 deletions.
diff --git a/src/model/store.ts b/src/model/store.ts
@@ -1,11 +1,11 @@
-import { HomeAssistant } from 'custom-card-helpers';
 import HassService from '../services/hass-service';
 import MediaBrowseService from '../services/media-browse-service';
 import MediaControlService from '../services/media-control-service';
 import {
   CardConfig,
   ConfigPredefinedGroup,
   ConfigPredefinedGroupPlayer,
+  HomeAssistant,
   PredefinedGroup,
   PredefinedGroupPlayer,
   Section,
@@ -110,7 +110,9 @@ export default class Store {
 
   public getMediaPlayerHassEntities(hass: HomeAssistant) {
     const configEntities = [...new Set(this.config.entities)];
-    return Object.values(hass.states)
+    return Object.values(this.hass.entities)
+      .filter((e) => e.entity_id.includes('media_player.') && e.platform === 'sonos')
+      .map((e) => hass.states[e.entity_id])
       .filter(getGroupPlayerIds)
       .filter((hassEntity) => {
         const includesEntity = configEntities.includes(hassEntity.entity_id);

diff --git a/src/services/hass-service.ts b/src/services/hass-service.ts
@@ -1,5 +1,4 @@
-import { HomeAssistant } from 'custom-card-helpers';
-import { MediaPlayerItem, Section, TemplateResult } from '../types';
+import { HomeAssistant, MediaPlayerItem, Section, TemplateResult } from '../types';
 import { ServiceCallRequest } from 'custom-card-helpers/dist/types';
 import { CALL_MEDIA_DONE, CALL_MEDIA_STARTED } from '../constants';
 import { MediaPlayer } from '../model/media-player';

diff --git a/src/types.ts b/src/types.ts
@@ -1,4 +1,4 @@
-import { LovelaceCardConfig } from 'custom-card-helpers';
+import { HomeAssistant as HomeAssistantBase, LovelaceCardConfig } from 'custom-card-helpers';
 import { MediaPlayer } from './model/media-player';
 
 declare global {
@@ -116,3 +116,12 @@ export enum MediaPlayerEntityFeature {
   REPEAT_SET = 262144,
   GROUPING = 524288,
 }
+
+export interface HomeAssistant extends HomeAssistantBase {
+  entities: { [id: string]: EntityRegistryDisplayEntry };
+}
+
+export interface EntityRegistryDisplayEntry {
+  entity_id: string;
+  platform?: string;
+}