"use client";
import React, { useEffect, useRef, useState } from "react";

export interface CyberhackTheme {
  primary?: string;
  secondary?: string;
  background?: string;
  foreground?: string;
}

export interface CyberhackResult {
  session_id?: string | null;
  completed_targets: number[];
  buffer: string[];
  /** Последовательность реально принятых кликов (row, col) в порядке совершения. */
  path: [number, number][];
  total_coins: number;
}

export interface CyberhackProps {
  /** URL для редиректа после игры, получит `?coins=<total_coins>`. */
  redirectUrl: string;
  /** Базовая награда; итоговая награда за демона = baseValue * (уровень демона). */
  baseValue: number;
  /** Лимит в секундах. Таймер стартует по первому клику, а не при монтировании. */
  timeLimit: number;
  locale?: "ru" | "en";
  theme?: CyberhackTheme;
  /**
   * Server-authoritative режим: доска и цели, присланные бэкендом. Если
   * заданы оба поля, игра не генерирует доску сама — сервер уже знает эту
   * доску и сверит `onComplete`-результат по своей копии, не доверяя
   * присланному `total_coins`. `sessionId` уходит обратно в `onComplete`.
   */
  matrix?: string[][];
  targets?: string[][];
  sessionId?: string;
  /** Вызывается один раз по окончании партии (успех или провал по таймеру). */
  onComplete?: (result: CyberhackResult) => void;
}

/**
 * React-обертка над Breach Protocol на Rust/Yew/WASM.
 * Сама не рендерит игру — динамически подгружает wasm-модуль и монтирует
 * его в контейнер `<div id={containerId}>` ниже, а результат партии
 * получает обратно через window.postMessage (см. handleMessage).
 */
export const Cyberhack: React.FC<CyberhackProps> = ({
  redirectUrl,
  baseValue,
  timeLimit,
  locale = "ru",
  theme,
  matrix,
  targets,
  sessionId,
  onComplete,
}) => {
  // 🔥 Упростили ID для надежности монтирования
  const containerId = useRef("cyberhack-core-container");
  const canvasRef = useRef<HTMLCanvasElement>(null);
  const initialized = useRef(false);
  const [loadError, setLoadError] = useState<string | null>(null);

  // Кибер-фон (hex-дождь) рисуется здесь, в React, отдельным canvas'ом —
  // до того, как загрузится и смонтируется WASM-игра (которая рисует свой
  // собственный такой же фон внутри контейнера). Так пользователь не видит
  // пустой экран во время загрузки wasm-бандла.

  useEffect(() => {
    const canvas = canvasRef.current;
    if (!canvas) return;
    const ctx = canvas.getContext("2d");
    if (!ctx) return;

    let animationFrameId: number;
    const fontSize = 16;
    let columns = 0;
    let drops: number[] = [];
    const chars = "0123456789ABCDEF".split("");

    const resize = () => {
      canvas.width = window.innerWidth;
      canvas.height = window.innerHeight;
      columns = Math.floor(canvas.width / fontSize);
      drops = [];
      for (let x = 0; x < columns; x++) {
        drops[x] = Math.random() * canvas.height;
      }
    };

    window.addEventListener("resize", resize);
    resize();

    const draw = () => {
      ctx.globalAlpha = 0.15;
      ctx.fillStyle = theme?.background || "#05020a";
      ctx.fillRect(0, 0, canvas.width, canvas.height);

      ctx.font = `${fontSize}px monospace`;

      for (let i = 0; i < drops.length; i++) {
        const text = chars[Math.floor(Math.random() * chars.length)];
        const x = i * fontSize;
        const y = drops[i] * fontSize;

        if (Math.random() > 0.98) {
          ctx.globalAlpha = 0.9;
          ctx.fillStyle = theme?.primary || "#8b3dff";
        } else {
          ctx.globalAlpha = 0.4;
          ctx.fillStyle = theme?.secondary || "#00f0ff";
        }

        ctx.fillText(text, x, y);

        if (y > canvas.height && Math.random() > 0.975) {
          drops[i] = 0;
        }
        drops[i]++;
      }
      animationFrameId = requestAnimationFrame(draw);
    };

    draw();

    return () => {
      window.removeEventListener("resize", resize);
      cancelAnimationFrame(animationFrameId);
    };
  }, [theme]);

  // Инициализация WASM Yew.
  // `initialized` — обычный ref, а не state: он должен пережить повторные
  // срабатывания этого эффекта (например при смене props в зависимостях
  // ниже) и не провоцировать лишний рендер, иначе initCyberHack вызвался бы
  // повторно и Yew попытался бы смонтировать игру поверх уже смонтированной.
  useEffect(() => {
    let isActive = true;

    if (!initialized.current) {
      Promise.all([
        //@ts-ignore
        import("cyberhack/wasm"),
      ])
        .then(([wasmModule]) => {
          if (!isActive) return;

          const init = wasmModule.default;
          const initCyberHack = wasmModule.initCyberHack;

          init().then(() => {
            if (!isActive) return;

            // 🔥 ИСПРАВЛЕНО: Ждем пока браузер 100% отрендерит DOM
            requestAnimationFrame(() => {
              const el = document.getElementById(containerId.current);

              // Защита от паники Rust: если элемента нет, прерываемся
              if (!el) {
                console.error(
                  "[Cyberhack] Контейнер для монтирования WASM не найден в DOM.",
                );
                return;
              }

              const config = {
                redirect_url: redirectUrl,
                base_value: baseValue,
                time_limit: timeLimit,
                locale,
                theme,
                matrix,
                targets,
                session_id: sessionId,
              };

              try {
                initCyberHack(containerId.current, JSON.stringify(config));
                initialized.current = true;
              } catch (e) {
                console.error("WASM Init Error:", e);
                setLoadError("Ошибка при запуске ядра.");
              }
            });
          });
        })
        .catch((e) => {
          console.warn("WASM-модуль еще не собран. Ожидание сборки...", e);
          setLoadError(
            "Библиотека WASM не найдена. Убедитесь, что выполнен скрипт сборки.",
          );
        });
    }

    // Rust-сторона не имеет прямой ссылки на React-callback onComplete,
    // поэтому результат партии передается через window.postMessage
    // (см. `end_game` в src/lib.rs) и здесь разбирается обратно в объект.
    // try/catch молча проглатывает чужие message-события (от расширений
    // браузера, HMR и т.п.), которые не являются нашим JSON-результатом.
    const handleMessage = (event: MessageEvent) => {
      try {
        const data = JSON.parse(event.data) as CyberhackResult;
        if (data.total_coins !== undefined && onComplete) {
          onComplete(data);
        }
      } catch (e) {}
    };

    window.addEventListener("message", handleMessage);

    return () => {
      isActive = false;
      window.removeEventListener("message", handleMessage);
    };
  }, [
    redirectUrl,
    baseValue,
    timeLimit,
    locale,
    theme,
    matrix,
    targets,
    sessionId,
    onComplete,
  ]);

  return (
    <div className="dark relative w-full h-screen overflow-hidden bg-[#05020a]">
      {loadError && (
        <div className="absolute top-0 left-0 w-full bg-red-900 text-white text-xs p-2 text-center z-[100]">
          {loadError}
        </div>
      )}

      {/* Теперь просто контейнер, WASM сам нарисует и фон, и игру */}
      <div
        id={containerId.current}
        className="relative z-10 w-full h-full flex items-center justify-center overflow-auto"
      ></div>

      {/* Оставляем только CRT оверлей, если ты не перенес его в Rust-верстку */}
      <div className="crt-overlay pointer-events-none absolute inset-0 z-50"></div>
    </div>
  );
};