From 6ac3d62823d43b88f4ccd7a125ad10a78241f1f1 Mon Sep 17 00:00:00 2001 From: Georgi Gardev Date: Thu, 7 May 2020 01:38:05 +0300 Subject: [PATCH] Use computeDomain --- src/time-picker-card.ts | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/time-picker-card.ts b/src/time-picker-card.ts index 4af0471..949a11c 100644 --- a/src/time-picker-card.ts +++ b/src/time-picker-card.ts @@ -1,4 +1,4 @@ -import { HomeAssistant } from 'custom-card-helpers'; +import { HomeAssistant, computeDomain } from 'custom-card-helpers'; import { HassEntity } from 'home-assistant-js-websocket'; import { css, @@ -25,6 +25,8 @@ console.info( @customElement('time-picker-card') export class TimePickerCard extends LitElement { + private static readonly ENTITY_DOMAIN = 'input_datetime'; + @property() private hass!: HomeAssistant; @property() private config!: TimePickerCardConfig; @property() private hour!: Hour; @@ -52,13 +54,13 @@ export class TimePickerCard extends LitElement { return Partial.error('Entity not found', this.config); } - if (!this.entity.entity_id.startsWith('input_datetime')) { - return Partial.error('You must set an input_datetime entity', this.config); + if (computeDomain(this.entity.entity_id) !== TimePickerCard.ENTITY_DOMAIN) { + return Partial.error(`You must set an ${TimePickerCard.ENTITY_DOMAIN} entity`, this.config); } if (!this.entity.attributes.has_time) { return Partial.error( - 'You must set an input_datetime entity that sets has_time: true', + `You must set an ${TimePickerCard.ENTITY_DOMAIN} entity that sets has_time: true`, this.config ); }