import type { PropertyValues } from "lit";
import type { HASSDomEvent } from "../common/dom/fire_event";
import type { Constructor, HomeAssistant } from "../types";
import { storeState } from "../util/ha-pref-storage";
import type { HassBaseEl } from "./hass-base-mixin";

interface VibrateParams {
  vibrate: HomeAssistant["vibrate"];
}

declare global {
  // for fire event
  interface HASSDomEvents {
    "hass-vibrate": VibrateParams;
  }
  // for add event listener
  interface HTMLElementEventMap {
    "hass-vibrate": HASSDomEvent<HASSDomEvents["hass-vibrate"]>;
  }
}

const hapticPatterns = {
  success: [50, 50, 50],
  warning: [100, 50, 100],
  failure: [200, 100, 200],
  light: [50],
  medium: [100],
  heavy: [200],
  selection: [20],
};

const handleHaptic = (
  hapticTypeEvent: HASSDomEvent<HASSDomEvents["haptic"]>
) => {
  navigator.vibrate(hapticPatterns[hapticTypeEvent.detail]);
};

export const hapticMixin = <T extends Constructor<HassBaseEl>>(superClass: T) =>
  class extends superClass {
    protected firstUpdated(changedProps: PropertyValues<this>) {
      super.firstUpdated(changedProps);
      this.addEventListener("hass-vibrate", (ev) => {
        const vibrate = ev.detail.vibrate;
        // @ts-expect-error not all browsers support vibrate
        if (navigator.vibrate && vibrate) {
          window.addEventListener("haptic", handleHaptic);
        } else {
          window.removeEventListener("haptic", handleHaptic);
        }
        this._updateHass({ vibrate });
        storeState(this.hass!);
      });
    }

    protected hassConnected() {
      super.hassConnected();
      // @ts-expect-error not all browsers support vibrate
      if (navigator.vibrate && this.hass!.vibrate) {
        window.addEventListener("haptic", handleHaptic);
      }
    }
  };
