import { createSignal, createMemo, For, onCleanup, onMount, Show } from "solid-js";
import { sendMessage } from "../App";
import { PeachWordmark } from "../components/PeachMark";
import { RecoveryRuneBackground } from "../components/RecoveryRuneBackground";

interface RecoveryWordDisplayViewProps {
  onDone: () => void;
}

// After showing the words, user must type the 3 words at these positions
const CONFIRM_WORD_INDICES = [2, 6, 10]; // 0-indexed → positions 3, 7, 11

export function RecoveryWordDisplayView(props: RecoveryWordDisplayViewProps) {
  const [words, setWords] = createSignal<string[]>([]);
  const [error, setError] = createSignal("");
  const [loading, setLoading] = createSignal(true);
  const [confirmWords, setConfirmWords] = createSignal(["", "", ""]);
  const [confirmError, setConfirmError] = createSignal("");
  const [confirmed, setConfirmed] = createSignal(false);
  // 5-second countdown before confirmation input is enabled
  const [countdown, setCountdown] = createSignal(5);
  let countdownTimer: ReturnType<typeof setInterval> | null = null;

  onMount(async () => {
    // Fetch the recovery words from the background (vault must be unlocked).
    const res = (await sendMessage({
      type: "GET_RECOVERY_WORDS",
    })) as { ok: boolean; data?: string[]; error?: string };

    if (res.ok && res.data) {
      setWords(res.data);
    } else {
      setError("could not retrieve recovery words");
    }
    setLoading(false);

    // Start 5-second countdown
    let remaining = 5;
    countdownTimer = setInterval(() => {
      remaining -= 1;
      setCountdown(remaining);
      if (remaining <= 0 && countdownTimer) {
        clearInterval(countdownTimer);
        countdownTimer = null;
      }
    }, 1000);
  });

  onCleanup(() => {
    if (countdownTimer) clearInterval(countdownTimer);
  });

  async function handleConfirm() {
    const confirmed = confirmWords().map((w) => w.toLowerCase().trim());
    if (confirmed.some((w) => !w)) {
      setConfirmError("all three words are required");
      return;
    }

    // Verify against stored words
    const stored = words();
    for (let i = 0; i < CONFIRM_WORD_INDICES.length; i++) {
      const pos = CONFIRM_WORD_INDICES[i];
      if (confirmed[i] !== stored[pos]) {
        setConfirmError(`word ${pos + 1} is incorrect`);
        return;
      }
    }

    setConfirmed(true);
    setConfirmError("");
    // Small pause so user sees the success state, then proceed.
    await new Promise((r) => setTimeout(r, 600));
    props.onDone();
  }

  const canConfirm = createMemo(() => countdown() <= 0 && !confirmed());

  return (
    <section class="peach-screen peach-theme lock-screen lock-screen--recovery lock-screen--recovery-display">
      <RecoveryRuneBackground />

      <header class="peach-screen-header">
        <div class="peach-wordmark-stack">
          <PeachWordmark class="peach-wordmark-wrap" />
        </div>
      </header>

      <div class="peach-recovery-header">
        <h1 class="peach-onboarding-headline">save your recovery words</h1>
        <p class="peach-hint peach-hint--warning">
          write these down — you will never see them again
        </p>
      </div>

      <Show when={loading()}>
        <p class="peach-hint">loading...</p>
      </Show>

      <Show when={error()}>
        <p class="peach-hint peach-hint-error">{error()}</p>
      </Show>

      <Show when={!loading() && !error()}>
        <div class="peach-recovery-words-display">
          <div class="peach-recovery-words-label">your 12 recovery words</div>
          <ol class="peach-recovery-word-list" start={1}>
            <For each={words()}>
              {(word, i) => (
                <li class="peach-recovery-word-item">
                  <span class="peach-recovery-word-num">{toRoman(i() + 1)}</span>
                  <span class="peach-recovery-word-text">{word}</span>
                </li>
              )}
            </For>
          </ol>
        </div>

        {/* Confirmation step */}
        <div class="peach-recovery-confirm">
          <Show when={!confirmed()}>
            <p class="peach-recovery-confirm-label">
              {countdown() > 0
                ? `type the highlighted words to confirm (${countdown()}s)`
                : "type the highlighted words to confirm"}
            </p>
          </Show>
          <Show when={confirmed()}>
            <p class="peach-hint peach-hint--success">✓ words verified</p>
          </Show>

          <div class="peach-recovery-confirm-fields">
            <For each={CONFIRM_WORD_INDICES}>
              {(pos, i) => (
                <div class="peach-recovery-confirm-field">
                  <span class="peach-recovery-confirm-num">
                    word {pos + 1}
                  </span>
                  <input
                    class="peach-recovery-word-input peach-recovery-confirm-input"
                    type="text"
                    autocomplete="off"
                    autocorrect="off"
                    autocapitalize="none"
                    spellcheck={false}
                    disabled={!canConfirm()}
                    value={confirmWords()[i()]}
                    onInput={(e) => {
                      const updated = [...confirmWords()];
                      updated[i()] = (e.target as HTMLInputElement).value;
                      setConfirmWords(updated);
                      if (confirmError()) setConfirmError("");
                    }}
                    onKeyDown={(e) => {
                      if (e.key === "Enter" && canConfirm()) void handleConfirm();
                    }}
                  />
                </div>
              )}
            </For>
          </div>

          <Show when={confirmError()}>
            <p class="peach-hint peach-hint-error">{confirmError()}</p>
          </Show>

          <button
            class="peach-btn peach-btn--accent peach-btn--md peach-recovery-confirm-btn"
            disabled={!canConfirm() || confirmed()}
            onClick={() => void handleConfirm()}
          >
            {confirmed() ? "verified ✓" : countdown() > 0 ? `wait ${countdown()}s` : "i wrote them down"}
          </button>
        </div>
      </Show>
    </section>
  );
}

function toRoman(n: number): string {
  const map: [number, string][] = [
    [1000, "M"], [900, "CM"], [500, "D"], [400, "CD"],
    [100, "C"], [90, "XC"], [50, "L"], [40, "XL"],
    [10, "X"], [9, "IX"], [5, "V"], [4, "IV"], [1, "I"],
  ];
  let out = "";
  for (const [value, numeral] of map) {
    while (n >= value) {
      out += numeral;
      n -= value;
    }
  }
  return out;
}