From 65b0fab0c5ce2bf9090294051b9a41b5e38accee Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <paulus@home-assistant.io>
Date: Wed, 10 Nov 2021 17:10:02 -0800
Subject: [PATCH] Add support for Home Assistant domain in manifest (#101)

---
 README.md                           |  1 +
 index.html                          |  6 ++
 src/const.ts                        |  1 +
 src/install-dialog.ts               | 88 +++++++++++++++++++----------
 static/firmware_build/manifest.json |  1 +
 5 files changed, 68 insertions(+), 29 deletions(-)

diff --git a/README.md b/README.md
index c96fd1f..113bb97 100644
--- a/README.md
+++ b/README.md
@@ -14,6 +14,7 @@ Manifest definition:
 {
   "name": "ESPHome",
   "version": "2021.10.3",
+  "home_assistant_domain": "esphome",
   "builds": [
     {
       "chipFamily": "ESP32",
diff --git a/index.html b/index.html
index 9d595e5..c291b72 100644
--- a/index.html
+++ b/index.html
@@ -266,6 +266,7 @@
 {
   "name": "ESPHome",
   "version": "2021.11.0",
+  "home_assistant_domain": "esphome",
   "builds": [
     {
       "chipFamily": "ESP32",
@@ -291,6 +292,11 @@
         where it should be installed. Part paths are resolved relative to the
         path of the manifest, but can also be URLs to other hosts.
       </p>
+      <p>
+        If your firmware is supported by Home Assistant, you can add the
+        optional key <code>home_assistant_domain</code>. If present, ESP Web
+        Tools will link the user to add this device to Home Assistant.
+      </p>
       <h3 id="improv">Wi-Fi provisioning</h3>
       <p>
         ESP Web Tools has support for the
diff --git a/src/const.ts b/src/const.ts
index 7744e14..b9eb4d1 100644
--- a/src/const.ts
+++ b/src/const.ts
@@ -15,6 +15,7 @@ export interface Build {
 export interface Manifest {
   name: string;
   version: string;
+  home_assistant_domain?: string;
   builds: Build[];
 }
 
diff --git a/src/install-dialog.ts b/src/install-dialog.ts
index 73a667e..04250e8 100644
--- a/src/install-dialog.ts
+++ b/src/install-dialog.ts
@@ -61,11 +61,6 @@ class EwtInstallDialog extends LitElement {
 
   @state() private _busy = false;
 
-  private _progressFeedback?: {
-    resolve: (_: unknown) => void;
-    reject: () => void;
-  };
-
   protected render() {
     if (!this.port) {
       return html``;
@@ -180,7 +175,21 @@ class EwtInstallDialog extends LitElement {
                   class="has-button"
                   target="_blank"
                 >
-                  <ewt-button label="Set up Device"></ewt-button>
+                  <ewt-button label="Visit Device"></ewt-button>
+                </a>
+              </div>
+            `}
+        ${!this._manifest.home_assistant_domain ||
+        this._client!.state !== ImprovSerialCurrentState.PROVISIONED
+          ? ""
+          : html`
+              <div>
+                <a
+                  href=${`https://my.home-assistant.io/redirect/config_flow_start/?domain=${this._manifest.home_assistant_domain}`}
+                  class="has-button"
+                  target="_blank"
+                >
+                  <ewt-button label="Add to Home Assistant"></ewt-button>
                 </a>
               </div>
             `}
@@ -245,29 +254,53 @@ class EwtInstallDialog extends LitElement {
       this._client!.state === ImprovSerialCurrentState.PROVISIONED
     ) {
       heading = undefined;
+      const showSetupLinks =
+        !this._wasProvisioned &&
+        (this._client!.nextUrl !== undefined ||
+          "home_assistant_domain" in this._manifest);
+      hideActions = showSetupLinks;
       content = html`
         ${messageTemplate(OK_ICON, "Device connected to the network!")}
-        ${!this._wasProvisioned && this._client!.nextUrl !== undefined
+        ${showSetupLinks
           ? html`
-              <a
-                slot="primaryAction"
-                href=${this._client!.nextUrl}
-                class="has-button"
-                target="_blank"
-                @click=${() => {
-                  this._state = "DASHBOARD";
-                }}
-              >
-                <ewt-button label="Set up Device"></ewt-button>
-              </a>
-              <ewt-button
-                slot="secondaryAction"
-                label="Skip"
-                @click=${() => {
-                  this._state = "DASHBOARD";
-                  this._installState = undefined;
-                }}
-              ></ewt-button>
+              <div class="dashboard-buttons">
+                <div>
+                  <a
+                    href=${this._client!.nextUrl}
+                    class="has-button"
+                    target="_blank"
+                    @click=${() => {
+                      this._state = "DASHBOARD";
+                    }}
+                  >
+                    <ewt-button label="Visit Device"></ewt-button>
+                  </a>
+                </div>
+                ${!this._manifest.home_assistant_domain
+                  ? ""
+                  : html`
+                      <div>
+                        <a
+                          href=${`https://my.home-assistant.io/redirect/config_flow_start/?domain=${this._manifest.home_assistant_domain}`}
+                          class="has-button"
+                          target="_blank"
+                        >
+                          <ewt-button
+                            label="Add to Home Assistant"
+                          ></ewt-button>
+                        </a>
+                      </div>
+                    `}
+                <div>
+                  <ewt-button
+                    label="Skip"
+                    @click=${() => {
+                      this._state = "DASHBOARD";
+                      this._installState = undefined;
+                    }}
+                  ></ewt-button>
+                </div>
+              </div>
             `
           : html`
               <ewt-button
@@ -617,9 +650,6 @@ class EwtInstallDialog extends LitElement {
   };
 
   private async _handleClose() {
-    if (this._progressFeedback) {
-      this._progressFeedback.reject();
-    }
     if (this._client) {
       await this._closeClientWithoutEvents(this._client);
     }
diff --git a/static/firmware_build/manifest.json b/static/firmware_build/manifest.json
index 56b1752..d0b205b 100644
--- a/static/firmware_build/manifest.json
+++ b/static/firmware_build/manifest.json
@@ -1,6 +1,7 @@
 {
   "name": "ESPHome",
   "version": "2021.11.0-dev",
+  "home_assistant_domain": "esphome",
   "builds": [
     {
       "chipFamily": "ESP32",