import type { TabElementProps } from "@components/Dialog/NewDeviceDialog.tsx";
import { Button } from "@components/UI/Button.tsx";
import { Input } from "@components/UI/Input.tsx";
import { Label } from "@components/UI/Label.tsx";
import { Switch } from "@components/UI/Switch.tsx";
import { useAppStore } from "@core/stores/appStore.ts";
import { useDeviceStore } from "@core/stores/deviceStore.ts";
import { subscribeAll } from "@core/subscriptions.ts";
import { randId } from "@core/utils/randId.ts";
import { MeshDevice } from "@meshtastic/core";
import { TransportHTTP } from "@meshtastic/transport-http";
import { useState } from "react";
import { useForm, useController } from "react-hook-form";
import { AlertTriangle } from "lucide-react";
import { useMessageStore } from "@core/stores/messageStore.ts";

interface FormData {
  ip: string;
  tls: boolean;
}

export const HTTP = ({ closeDialog }: TabElementProps) => {
  const isURLHTTPS = location.protocol === "https:";

  const { addDevice } = useDeviceStore();
  const messageStore = useMessageStore();
  const { setSelectedDevice } = useAppStore();

  const { control, handleSubmit, register } = useForm<FormData>({
    defaultValues: {
      ip: ["client.meshtastic.org", "localhost"].includes(
        window.location.hostname,
      )
        ? "meshtastic.local"
        : window.location.host,
      tls: isURLHTTPS ? true : false,
    },
  });

  const {
    field: { value: tlsValue, onChange: setTLS },
  } = useController({ name: "tls", control });

  const [connectionInProgress, setConnectionInProgress] = useState(false);
  const [connectionError, setConnectionError] = useState<{ host: string; secure: boolean } | null>(null);

  const onSubmit = handleSubmit(async (data) => {
    setConnectionInProgress(true);
    setConnectionError(null);

    try {
      const id = randId();
      const transport = await TransportHTTP.create(data.ip, data.tls);
      const device = addDevice(id);
      const connection = new MeshDevice(transport, id);
      connection.configure();
      setSelectedDevice(id);
      device.addConnection(connection);
      subscribeAll(device, connection, messageStore);
      closeDialog();
    } catch (error) {
      console.error("Connection error:", error);
      // Capture all connection errors regardless of type
      setConnectionError({ host: data.ip, secure: data.tls });
      setConnectionInProgress(false);
    }
  });

  return (
    <form className="flex w-full flex-col gap-2 p-4" onSubmit={onSubmit}>
      <div className="flex flex-col gap-2" style={{ minHeight: "12rem" }}>
        <div>
          <Label>IP Address/Hostname</Label>
          <Input
            prefix={tlsValue ? "https://" : "http://"}
            placeholder="000.000.000.000 / meshtastic.local"
            className="text-slate-900 dark:text-slate-900"
            disabled={connectionInProgress}
            {...register("ip")}
          />
        </div>
        <div className="flex items-center gap-2 mt-2">
          <Switch
            onCheckedChange={setTLS}
            disabled={isURLHTTPS || connectionInProgress}
            checked={isURLHTTPS || tlsValue}
            {...register("tls")}
          />
          <Label>Use HTTPS</Label>
        </div>

        {connectionError && (
          <div className="mt-2 mb-2 p-3 rounded-md bg-amber-100 border border-amber-300 dark:bg-amber-900 dark:border-amber-700">
            <div className="flex gap-2 items-start">
              <AlertTriangle className="shrink-0 mt-0.5 text-amber-600 dark:text-amber-400" size={20} />
              <div>
                <p className="text-sm font-medium text-amber-800 dark:text-amber-300">
                  Connection Failed
                </p>
                <p className="text-xs mt-1 text-amber-700 dark:text-amber-400">
                  Could not connect to the device. {connectionError.secure && "If using HTTPS, you may need to accept a self-signed certificate first. "}
                  Please open{" "}
                  <a
                    href={`${connectionError.secure ? "https" : "http"}://${connectionError.host}`}
                    target="_blank"
                    rel="noopener noreferrer"
                    className="underline font-medium"
                  >
                    {`${connectionError.secure ? "https" : "http"}://${connectionError.host}`}
                  </a>{" "}
                  in a new tab{connectionError.secure ? ", accept any TLS warnings if prompted, then try again" : ""}.{" "}
                  <a
                    href="https://meshtastic.org/docs/software/web-client/#http"
                    target="_blank"
                    rel="noopener noreferrer"
                    className="underline font-medium"
                  >
                    Learn more
                  </a>
                </p>
              </div>
            </div>
          </div>
        )}
      </div>
      <Button
        type="submit"
        disabled={connectionInProgress}
        className="dark:bg-slate-900 dark:text-white"
      >
        <span>{connectionInProgress ? "Connecting..." : "Connect"}</span>
      </Button>
    </form>
  );
};