import { mdiCheckCircle, mdiCloseCircle } from "@mdi/js";
import type { UnsubscribeFunc } from "home-assistant-js-websocket";
import type { CSSResultGroup } from "lit";
import { css, html, LitElement, nothing } from "lit";
import { customElement, property, state } from "lit/decorators";
import { fireEvent } from "../../../../../common/dom/fire_event";
import "../../../../../components/ha-spinner";
import "../../../../../components/ha-svg-icon";
import "../../../../../components/ha-tooltip";
import "../../../../../components/ha-button";
import "../../../../../components/ha-dialog-footer";
import "../../../../../components/ha-dialog";
import type {
  AttributeConfigurationStatus,
  Cluster,
  ClusterConfigurationEvent,
  ClusterConfigurationStatus,
} from "../../../../../data/zha";
import {
  fetchClustersForZhaDevice,
  reconfigureNode,
  ZHA_CHANNEL_CFG_DONE,
  ZHA_CHANNEL_MSG_BIND,
  ZHA_CHANNEL_MSG_CFG_RPT,
} from "../../../../../data/zha";
import { haStyleDialog } from "../../../../../resources/styles";
import type { HomeAssistant } from "../../../../../types";
import type { ZHAReconfigureDeviceDialogParams } from "./show-dialog-zha-reconfigure-device";

@customElement("dialog-zha-reconfigure-device")
class DialogZHAReconfigureDevice extends LitElement {
  @property({ attribute: false }) public hass!: HomeAssistant;

  @state() private _status?: string;

  @state() private _stages?: string[];

  @state() private _clusterConfigurationStatuses? = new Map<
    number,
    ClusterConfigurationStatus
  >();

  @state() private _params: ZHAReconfigureDeviceDialogParams | undefined =
    undefined;

  @state() private _allSuccessful = true;

  @state() private _showDetails = false;

  @state() private _open = false;

  private _subscribed?: Promise<UnsubscribeFunc>;

  public showDialog(params: ZHAReconfigureDeviceDialogParams): void {
    this._params = params;
    this._clusterConfigurationStatuses = new Map();
    this._stages = undefined;
    this._open = true;
  }

  public closeDialog(): void {
    this._open = false;
  }

  private _dialogClosed(): void {
    this._unsubscribe();
    this._params = undefined;
    this._status = undefined;
    this._stages = undefined;
    this._clusterConfigurationStatuses = undefined;
    this._showDetails = false;
    this._allSuccessful = true;
    fireEvent(this, "dialog-closed", { dialog: this.localName });
  }

  protected render() {
    if (!this._params) {
      return nothing;
    }

    return html`
      <ha-dialog
        .open=${this._open}
        width="large"
        header-title=${
          this.hass.localize(`ui.dialogs.zha_reconfigure_device.heading`) +
          ": " +
          (this._params.device.user_given_name || this._params.device.name)
        }
        @closed=${this._dialogClosed}
      >
        ${
          !this._status
            ? html`
                <p>
                  ${this.hass.localize(
                    "ui.dialogs.zha_reconfigure_device.introduction"
                  )}
                </p>
                <p>
                  <em>
                    ${this.hass.localize(
                      "ui.dialogs.zha_reconfigure_device.battery_device_warning"
                    )}
                  </em>
                </p>
              `
            : ``
        }
        ${
          this._status === "started"
            ? html`
                <div class="flex-container">
                  <ha-spinner></ha-spinner>
                  <div class="status">
                    <p>
                      <b>
                        ${this.hass.localize(
                          "ui.dialogs.zha_reconfigure_device.in_progress"
                        )}
                      </b>
                    </p>
                    <p>
                      ${this.hass.localize(
                        "ui.dialogs.zha_reconfigure_device.run_in_background"
                      )}
                    </p>
                  </div>
                </div>
              `
            : ``
        }
        ${
          this._status === "failed"
            ? html`
                <div class="flex-container">
                  <ha-svg-icon
                    .path=${mdiCloseCircle}
                    class="failed"
                  ></ha-svg-icon>
                  <div class="status">
                    <p>
                      ${this.hass.localize(
                        "ui.dialogs.zha_reconfigure_device.configuration_failed"
                      )}
                    </p>
                  </div>
                </div>
              `
            : ``
        }
        ${
          this._status === "finished"
            ? html`
                <div class="flex-container">
                  <ha-svg-icon
                    .path=${mdiCheckCircle}
                    class="success"
                  ></ha-svg-icon>
                  <div class="status">
                    <p>
                      ${this.hass.localize(
                        "ui.dialogs.zha_reconfigure_device.configuration_complete"
                      )}
                    </p>
                  </div>
                </div>
              `
            : ``
        }
        ${
          this._stages
            ? html`
                <div class="stages">
                  ${this._stages.map(
                    (stage) => html`
                      <span class="stage">
                        <ha-svg-icon
                          .path=${mdiCheckCircle}
                          class="success"
                        ></ha-svg-icon>
                        ${stage}
                      </span>
                    `
                  )}
                </div>
              `
            : ""
        }
        ${
          this._showDetails
            ? html`
                <div class="wrapper">
                  <h2 class="grid-item">
                    ${this.hass.localize(
                      `ui.dialogs.zha_reconfigure_device.cluster_header`
                    )}
                  </h2>
                  <h2 class="grid-item">
                    ${this.hass.localize(
                      `ui.dialogs.zha_reconfigure_device.bind_header`
                    )}
                  </h2>
                  <h2 class="grid-item">
                    ${this.hass.localize(
                      `ui.dialogs.zha_reconfigure_device.reporting_header`
                    )}
                  </h2>

                  ${
                    this._clusterConfigurationStatuses?.size
                      ? html`
                          ${Array.from(
                            this._clusterConfigurationStatuses.values()
                          ).map(
                            (clusterStatus) => html`
                              <div class="grid-item">
                                ${clusterStatus.cluster.name}
                              </div>
                              <div class="grid-item">
                                ${
                                  clusterStatus.bindSuccess !== undefined
                                    ? clusterStatus.bindSuccess
                                      ? html`
                                          <span class="stage">
                                            <ha-svg-icon
                                              .path=${mdiCheckCircle}
                                              class="success"
                                            ></ha-svg-icon>
                                          </span>
                                        `
                                      : html`
                                          <span class="stage">
                                            <ha-svg-icon
                                              .path=${mdiCloseCircle}
                                              class="failed"
                                            ></ha-svg-icon>
                                          </span>
                                        `
                                    : ""
                                }
                              </div>
                              <div class="grid-item">
                                ${
                                  clusterStatus.attributes.size > 0
                                    ? html`
                                        <div class="attributes">
                                          <div class="grid-item">
                                            ${this.hass.localize(
                                              `ui.dialogs.zha_reconfigure_device.attribute`
                                            )}
                                          </div>
                                          <div class="grid-item">
                                            <div>
                                              ${this.hass.localize(
                                                `ui.dialogs.zha_reconfigure_device.min_max_change`
                                              )}
                                            </div>
                                          </div>
                                          ${Array.from(
                                            clusterStatus.attributes.values()
                                          ).map(
                                            (attribute) => html`
                                              <span class="grid-item">
                                                ${attribute.name}:
                                                ${
                                                  attribute.status === "SUCCESS"
                                                    ? html`
                                                        <span class="stage">
                                                          <ha-svg-icon
                                                            .path=${mdiCheckCircle}
                                                            class="success"
                                                          ></ha-svg-icon>
                                                        </span>
                                                      `
                                                    : html`
                                                        <span class="stage">
                                                          <ha-svg-icon
                                                            .id="svg-icon-${
                                                              clusterStatus
                                                                .cluster.name
                                                            }"
                                                            .path=${mdiCloseCircle}
                                                            class="failed"
                                                          ></ha-svg-icon>
                                                          <ha-tooltip
                                                            .for="svg-icon-${
                                                              clusterStatus
                                                                .cluster.name
                                                            }"
                                                            placement="top"
                                                          >
                                                            ${attribute.status}
                                                          </ha-tooltip>
                                                        </span>
                                                      `
                                                }
                                              </span>
                                              <div class="grid-item">
                                                ${attribute.min}/${attribute.max}/${attribute.change}
                                              </div>
                                            `
                                          )}
                                        </div>
                                      `
                                    : ""
                                }
                              </div>
                            `
                          )}
                        `
                      : ""
                  }
                </div>
              `
            : ""
        }
        <ha-dialog-footer slot="footer">
          ${
            !this._status
              ? html`
                  <ha-button
                    slot="primaryAction"
                    @click=${this._startReconfiguration}
                  >
                    ${this.hass.localize(
                      "ui.dialogs.zha_reconfigure_device.start_reconfiguration"
                    )}
                  </ha-button>
                `
              : html`
                  <ha-button
                    slot="secondaryAction"
                    appearance="plain"
                    @click=${this._toggleDetails}
                  >
                    ${
                      this._showDetails
                        ? this.hass.localize(
                            `ui.dialogs.zha_reconfigure_device.button_hide`
                          )
                        : this.hass.localize(
                            `ui.dialogs.zha_reconfigure_device.button_show`
                          )
                    }
                  </ha-button>
                  <ha-button slot="primaryAction" @click=${this.closeDialog}>
                    ${this.hass.localize("ui.common.close")}
                  </ha-button>
                `
          }
        </ha-dialog-footer>
      </ha-dialog>
    `;
  }

  private async _startReconfiguration(): Promise<void> {
    if (!this.hass || !this._params) {
      return;
    }
    this._clusterConfigurationStatuses = new Map(
      (
        await fetchClustersForZhaDevice(this.hass, this._params.device.ieee)
      ).map((cluster: Cluster) => [
        cluster.id,
        {
          cluster: cluster,
          bindSuccess: undefined,
          attributes: new Map<number, AttributeConfigurationStatus>(),
        },
      ])
    );
    this._subscribe(this._params);
    this._status = "started";
  }

  private _handleMessage(message: ClusterConfigurationEvent): void {
    if (message.type === ZHA_CHANNEL_CFG_DONE) {
      this._unsubscribe();
      this._status = this._allSuccessful ? "finished" : "failed";
    } else {
      const clusterConfigurationStatus =
        this._clusterConfigurationStatuses!.get(
          message.zha_channel_msg_data.cluster_id
        );
      if (message.type === ZHA_CHANNEL_MSG_BIND) {
        if (!this._stages) {
          this._stages = ["binding"];
        }
        const success = message.zha_channel_msg_data.success;
        clusterConfigurationStatus!.bindSuccess = success;
        this._allSuccessful = this._allSuccessful && success;
      }
      if (message.type === ZHA_CHANNEL_MSG_CFG_RPT) {
        if (this._stages && !this._stages.includes("reporting")) {
          this._stages.push("reporting");
        }
        const attributes = message.zha_channel_msg_data.attributes;
        Object.keys(attributes).forEach((name) => {
          const attribute = attributes[name];
          clusterConfigurationStatus!.attributes.set(attribute.id, attribute);
          this._allSuccessful =
            this._allSuccessful &&
            !(
              attribute.status in
              ["FAILURE", "UNSUPPORTED_ATTRIBUTE", "UNREPORTABLE_ATTRIBUTE"]
            );
        });
      }
      this.requestUpdate();
    }
  }

  private _unsubscribe(): void {
    if (this._subscribed) {
      this._subscribed.then((unsub) => unsub());
      this._subscribed = undefined;
    }
  }

  private _subscribe(params: ZHAReconfigureDeviceDialogParams): void {
    if (!this.hass) {
      return;
    }
    this._subscribed = reconfigureNode(
      this.hass,
      params.device.ieee,
      this._handleMessage.bind(this)
    );
  }

  private _toggleDetails() {
    this._showDetails = !this._showDetails;
  }

  static get styles(): CSSResultGroup {
    return [
      haStyleDialog,
      css`
        .wrapper {
          display: grid;
          grid-template-columns: 3fr 1fr 2fr;
        }
        .attributes {
          display: grid;
          grid-template-columns: 1fr 1fr;
        }
        .grid-item {
          border: 1px solid;
          padding: 7px;
        }
        .success {
          color: var(--success-color);
        }

        .failed {
          color: var(--warning-color);
        }

        .flex-container {
          display: flex;
          align-items: center;
        }

        .stages {
          margin-top: 16px;
        }

        .stage ha-svg-icon {
          width: 16px;
          height: 16px;
        }
        .stage {
          padding: 8px;
        }

        ha-svg-icon {
          width: 68px;
          height: 48px;
        }

        .flex-container ha-spinner,
        .flex-container ha-svg-icon {
          margin-right: 20px;
          margin-inline-end: 20px;
          margin-inline-start: initial;
        }
      `,
    ];
  }
}

declare global {
  interface HTMLElementTagNameMap {
    "dialog-zha-reconfigure-device": DialogZHAReconfigureDevice;
  }
}
