Rework state_icons to use templated state value (fix #51)

jcwillox · Jan 8, 2022 · 28d6f24 · 28d6f24
1 parent f671d9f
commit 28d6f24
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main.ts b/src/main.ts
@@ -39,8 +39,8 @@ console.info(
   `color: #039be5; background: white; font-weight: 700;`
 );
 
-const computeIcon = (state, config) => {
-  return config.state_icons && config.state_icons[state];
+const computeStateIcon = config => {
+  return config.state_icons && config.state_icons[config.state.toLowerCase()];
 };
 
 const computeStateText = config => {
@@ -283,7 +283,7 @@ export class PaperButtonsRow extends LitElement {
   renderIcon(config: ButtonConfig, style, entity?: HassEntity) {
     const icon =
       config.icon !== false && (config.icon || config.entity)
-        ? computeIcon(entity?.state, config) ||
+        ? computeStateIcon(config) ||
           config.icon ||
           (entity && stateIcon(entity))
         : false;