From f913677dfeec49265d50fa36a9bb2910dccfb7ac Mon Sep 17 00:00:00 2001
From: Aidan Timson <aidan@timmo.dev>
Date: Wed, 11 Feb 2026 16:21:31 +0000
Subject: [PATCH] Migrate storage config dialogs to wa (#29576)

Migrate config-storage dialog(s) to wa
---
 .../config/storage/dialog-mount-view.ts       | 103 ++++++++----------
 .../config/storage/dialog-move-datadisk.ts    |  67 +++++++-----
 2 files changed, 88 insertions(+), 82 deletions(-)

diff --git a/src/panels/config/storage/dialog-mount-view.ts b/src/panels/config/storage/dialog-mount-view.ts
index 4590df946f..df6a44bad4 100644
--- a/src/panels/config/storage/dialog-mount-view.ts
+++ b/src/panels/config/storage/dialog-mount-view.ts
@@ -1,4 +1,4 @@
-import { mdiClose, mdiHelpCircle } from "@mdi/js";
+import { mdiHelpCircle } from "@mdi/js";
 import type { CSSResultGroup } from "lit";
 import { css, html, LitElement, nothing } from "lit";
 import { customElement, property, state } from "lit/decorators";
@@ -8,9 +8,11 @@ import type { LocalizeFunc } from "../../../common/translations/localize";
 import { computeRTLDirection } from "../../../common/util/compute_rtl";
 import "../../../components/buttons/ha-progress-button";
 import type { HaProgressButton } from "../../../components/buttons/ha-progress-button";
+import "../../../components/ha-dialog-footer";
 import "../../../components/ha-form/ha-form";
 import type { SchemaUnion } from "../../../components/ha-form/types";
 import "../../../components/ha-icon-button";
+import "../../../components/ha-wa-dialog";
 import { extractApiErrorMessage } from "../../../data/hassio/common";
 import type { SupervisorMountRequestParams } from "../../../data/supervisor/mounts";
 import {
@@ -169,12 +171,15 @@ class ViewMountDialog extends LitElement {
 
   @state() private _reloadMounts?: () => void;
 
+  @state() private _open = false;
+
   public async showDialog(
     dialogParams: MountViewDialogParams
   ): Promise<Promise<void>> {
     this._data = dialogParams.mount;
     this._existing = dialogParams.mount !== undefined;
     this._reloadMounts = dialogParams.reloadMounts;
+    this._open = true;
     if (
       dialogParams.mount?.type === "cifs" &&
       dialogParams.mount.version &&
@@ -185,6 +190,10 @@ class ViewMountDialog extends LitElement {
   }
 
   public closeDialog(): void {
+    this._open = false;
+  }
+
+  private _dialogClosed(): void {
     this._data = undefined;
     this._waiting = undefined;
     this._error = undefined;
@@ -201,56 +210,39 @@ class ViewMountDialog extends LitElement {
       return nothing;
     }
     return html`
-      <ha-dialog
-        open
-        scrimClickAction
-        escapeKeyAction
-        .heading=${this._existing
+      <ha-wa-dialog
+        .hass=${this.hass}
+        .open=${this._open}
+        header-title=${this._existing
           ? this.hass.localize(
               "ui.panel.config.storage.network_mounts.update_title"
             )
           : this.hass.localize(
               "ui.panel.config.storage.network_mounts.add_title"
             )}
-        @closed=${this.closeDialog}
+        @closed=${this._dialogClosed}
       >
-        <ha-dialog-header slot="heading">
-          <ha-icon-button
-            slot="navigationIcon"
-            dialogAction="cancel"
-            .label=${this.hass.localize("ui.common.close")}
-            .path=${mdiClose}
-          ></ha-icon-button>
-          <span slot="title"
-            >${this._existing
-              ? this.hass.localize(
-                  "ui.panel.config.storage.network_mounts.update_title"
-                )
-              : this.hass.localize(
-                  "ui.panel.config.storage.network_mounts.add_title"
-                )}
-          </span>
-          <a
-            slot="actionItems"
-            class="header_button"
-            href=${documentationUrl(
-              this.hass,
-              "/common-tasks/os#network-storage"
-            )}
-            title=${this.hass.localize(
-              "ui.panel.config.storage.network_mounts.documentation"
-            )}
-            target="_blank"
-            rel="noreferrer"
-            dir=${computeRTLDirection(this.hass)}
-          >
-            <ha-icon-button .path=${mdiHelpCircle}></ha-icon-button>
-          </a>
-        </ha-dialog-header>
+        <a
+          slot="headerActionItems"
+          class="header_button"
+          href=${documentationUrl(
+            this.hass,
+            "/common-tasks/os#network-storage"
+          )}
+          title=${this.hass.localize(
+            "ui.panel.config.storage.network_mounts.documentation"
+          )}
+          target="_blank"
+          rel="noreferrer"
+          dir=${computeRTLDirection(this.hass)}
+        >
+          <ha-icon-button .path=${mdiHelpCircle}></ha-icon-button>
+        </a>
         ${this._error
           ? html`<ha-alert alert-type="error">${this._error}</ha-alert>`
           : nothing}
         <ha-form
+          autofocus
           .data=${this._data}
           .schema=${mountSchema(
             this.hass.localize,
@@ -265,29 +257,28 @@ class ViewMountDialog extends LitElement {
           .computeError=${this._computeErrorCallback}
           .computeWarning=${this._computeWarningCallback}
           @value-changed=${this._valueChanged}
-          dialogInitialFocus
         ></ha-form>
 
-        ${this._existing
-          ? html`<ha-button
-              @click=${this._deleteMount}
-              variant="danger"
-              slot="secondaryAction"
-              appearance="plain"
-            >
-              ${this.hass.localize("ui.common.delete")}
-            </ha-button>`
-          : nothing}
-
-        <div slot="primaryAction">
+        <ha-dialog-footer slot="footer">
+          ${this._existing
+            ? html`<ha-button
+                @click=${this._deleteMount}
+                variant="danger"
+                slot="secondaryAction"
+                appearance="plain"
+              >
+                ${this.hass.localize("ui.common.delete")}
+              </ha-button>`
+            : nothing}
           <ha-button
+            slot="secondaryAction"
             appearance="plain"
             @click=${this.closeDialog}
-            dialogInitialFocus
           >
             ${this.hass.localize("ui.common.cancel")}
           </ha-button>
           <ha-progress-button
+            slot="primaryAction"
             .progress=${!!this._waiting}
             @click=${this._connectMount}
           >
@@ -299,8 +290,8 @@ class ViewMountDialog extends LitElement {
                   "ui.panel.config.storage.network_mounts.connect"
                 )}
           </ha-progress-button>
-        </div>
-      </ha-dialog>
+        </ha-dialog-footer>
+      </ha-wa-dialog>
     `;
   }
 
diff --git a/src/panels/config/storage/dialog-move-datadisk.ts b/src/panels/config/storage/dialog-move-datadisk.ts
index 8da2a3c758..bc92d5828c 100644
--- a/src/panels/config/storage/dialog-move-datadisk.ts
+++ b/src/panels/config/storage/dialog-move-datadisk.ts
@@ -4,10 +4,11 @@ import { customElement, property, state } from "lit/decorators";
 import memoizeOne from "memoize-one";
 import { fireEvent } from "../../../common/dom/fire_event";
 import "../../../components/ha-button";
-import "../../../components/ha-dialog";
+import "../../../components/ha-dialog-footer";
 import "../../../components/ha-select";
 import type { HaSelectSelectEvent } from "../../../components/ha-select";
 import "../../../components/ha-spinner";
+import "../../../components/ha-wa-dialog";
 import {
   extractApiErrorMessage,
   ignoreSupervisorError,
@@ -49,10 +50,13 @@ class MoveDatadiskDialog extends LitElement {
 
   @state() private _moving = false;
 
+  @state() private _open = false;
+
   public async showDialog(
     dialogParams: MoveDatadiskDialogParams
   ): Promise<Promise<void>> {
     this._hostInfo = dialogParams.hostInfo;
+    this._open = true;
 
     try {
       this._osInfo = await fetchHassioHassOsInfo(this.hass);
@@ -83,6 +87,13 @@ class MoveDatadiskDialog extends LitElement {
   }
 
   public closeDialog(): void {
+    this._open = false;
+    if (!this._hostInfo || !this._osInfo || !this._disks) {
+      this._dialogClosed();
+    }
+  }
+
+  private _dialogClosed(): void {
     this._selectedDevice = undefined;
     this._disks = undefined;
     this._moving = false;
@@ -97,15 +108,13 @@ class MoveDatadiskDialog extends LitElement {
     }
 
     return html`
-      <ha-dialog
-        open
-        scrimClickAction
-        escapeKeyAction
-        .heading=${this._moving
+      <ha-wa-dialog
+        .hass=${this.hass}
+        .open=${this._open}
+        header-title=${this._moving
           ? this.hass.localize("ui.panel.config.storage.datadisk.moving")
           : this.hass.localize("ui.panel.config.storage.datadisk.title")}
-        @closed=${this.closeDialog}
-        ?hideActions=${this._moving}
+        @closed=${this._dialogClosed}
       >
         ${this._moving
           ? html`
@@ -127,6 +136,7 @@ class MoveDatadiskDialog extends LitElement {
               <br /><br />
 
               <ha-select
+                autofocus
                 .label=${this.hass.localize(
                   "ui.panel.config.storage.datadisk.select_device"
                 )}
@@ -143,25 +153,30 @@ class MoveDatadiskDialog extends LitElement {
                   ),
                 }))}
               ></ha-select>
-
-              <ha-button
-                slot="primaryAction"
-                appearance="plain"
-                @click=${this.closeDialog}
-                dialogInitialFocus
-              >
-                ${this.hass.localize("ui.panel.config.storage.datadisk.cancel")}
-              </ha-button>
-
-              <ha-button
-                .disabled=${!this._selectedDevice}
-                slot="primaryAction"
-                @click=${this._moveDatadisk}
-              >
-                ${this.hass.localize("ui.panel.config.storage.datadisk.move")}
-              </ha-button>
             `}
-      </ha-dialog>
+        ${this._moving
+          ? nothing
+          : html`
+              <ha-dialog-footer slot="footer">
+                <ha-button
+                  slot="secondaryAction"
+                  appearance="plain"
+                  @click=${this.closeDialog}
+                >
+                  ${this.hass.localize(
+                    "ui.panel.config.storage.datadisk.cancel"
+                  )}
+                </ha-button>
+                <ha-button
+                  .disabled=${!this._selectedDevice}
+                  slot="primaryAction"
+                  @click=${this._moveDatadisk}
+                >
+                  ${this.hass.localize("ui.panel.config.storage.datadisk.move")}
+                </ha-button>
+              </ha-dialog-footer>
+            `}
+      </ha-wa-dialog>
     `;
   }