From b053881cef183357e7f7e3da5d10a8a97e27efd3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20S=C3=B8rensen?= <ludeeus@ludeeus.dev>
Date: Thu, 10 Feb 2022 10:40:34 +0100
Subject: [PATCH] Add missing type to create device automation/script heading
 (#11635)

Co-authored-by: Bram Kragten <mail@bramkragten.nl>
---
 .../ha-device-automation-dialog.ts              | 17 ++++++++++++-----
 .../show-dialog-device-automation.ts            |  3 ++-
 .../config/devices/ha-config-device-page.ts     |  7 +++++--
 3 files changed, 19 insertions(+), 8 deletions(-)

diff --git a/src/panels/config/devices/device-detail/ha-device-automation-dialog.ts b/src/panels/config/devices/device-detail/ha-device-automation-dialog.ts
index b42d2f069b..cef11fc0fe 100644
--- a/src/panels/config/devices/device-detail/ha-device-automation-dialog.ts
+++ b/src/panels/config/devices/device-detail/ha-device-automation-dialog.ts
@@ -60,18 +60,18 @@ export class DialogDeviceAutomation extends LitElement {
       return;
     }
 
-    const { deviceId, script } = this._params;
+    const { device, script } = this._params;
 
-    fetchDeviceActions(this.hass, deviceId).then((actions) => {
+    fetchDeviceActions(this.hass, device.id).then((actions) => {
       this._actions = actions;
     });
     if (script) {
       return;
     }
-    fetchDeviceTriggers(this.hass, deviceId).then((triggers) => {
+    fetchDeviceTriggers(this.hass, device.id).then((triggers) => {
       this._triggers = triggers;
     });
-    fetchDeviceConditions(this.hass, deviceId).then((conditions) => {
+    fetchDeviceConditions(this.hass, device.id).then((conditions) => {
       this._conditions = conditions;
     });
   }
@@ -88,7 +88,14 @@ export class DialogDeviceAutomation extends LitElement {
         .heading=${this.hass.localize(
           `ui.panel.config.devices.${
             this._params.script ? "script" : "automation"
-          }.create`
+          }.create`,
+          {
+            type: this.hass.localize(
+              `ui.panel.config.devices.type.${
+                this._params.device.entry_type || "device"
+              }`
+            ),
+          }
         )}
       >
         <div @entry-selected=${this.closeDialog}>
diff --git a/src/panels/config/devices/device-detail/show-dialog-device-automation.ts b/src/panels/config/devices/device-detail/show-dialog-device-automation.ts
index 93242cd742..4f00987bdf 100644
--- a/src/panels/config/devices/device-detail/show-dialog-device-automation.ts
+++ b/src/panels/config/devices/device-detail/show-dialog-device-automation.ts
@@ -1,7 +1,8 @@
 import { fireEvent } from "../../../../common/dom/fire_event";
+import { DeviceRegistryEntry } from "../../../../data/device_registry";
 
 export interface DeviceAutomationDialogParams {
-  deviceId: string;
+  device: DeviceRegistryEntry;
   script?: boolean;
 }
 
diff --git a/src/panels/config/devices/ha-config-device-page.ts b/src/panels/config/devices/ha-config-device-page.ts
index 984acc04fa..5e966278e8 100644
--- a/src/panels/config/devices/ha-config-device-page.ts
+++ b/src/panels/config/devices/ha-config-device-page.ts
@@ -812,12 +812,15 @@ export class HaConfigDevicePage extends LitElement {
   }
 
   private _showScriptDialog() {
-    showDeviceAutomationDialog(this, { deviceId: this.deviceId, script: true });
+    showDeviceAutomationDialog(this, {
+      device: this._device(this.deviceId, this.devices)!,
+      script: true,
+    });
   }
 
   private _showAutomationDialog() {
     showDeviceAutomationDialog(this, {
-      deviceId: this.deviceId,
+      device: this._device(this.deviceId, this.devices)!,
       script: false,
     });
   }