import { useRouter, useSearchParams } from "next/navigation";
import { useState, useEffect, useCallback, useRef } from "react";

import {
  POST_MESSAGE_TYPES,
  RECEIVE_MESSAGE_TYPES,
  isSelectedSkillMessage,
  sendToParent,
} from "@/core/iframe-messages";

// Skill 数据类型
interface SkillData {
  skill_id: string;
  title: string;
}

// Hook 返回类型
interface UseIframeSkillReturn {
  selectedSkill: SkillData | null;
  sendSelectSkill: (skill_id: string) => void;
  openSkillDialog: () => void;
  clearSkill: () => void;
}

export function useIframeSkill(): UseIframeSkillReturn {
  const router = useRouter();
  const searchParams = useSearchParams();
  const skillIdFromQuery = searchParams.get("skill_id");
  const titleFromQuery = searchParams.get("title");
  const threadIdFromQuery = searchParams.get("thread_id");
  const xClawUsedFromQuery = searchParams.get("xclaw_used");
  const lastThreadIdRef = useRef<string | null>(null);

  const [selectedSkill, setSelectedSkill] = useState<SkillData | null>(null);

  // 1. 监听 query 参数变化
  useEffect(() => {
    if (skillIdFromQuery && titleFromQuery) {
      setSelectedSkill({ skill_id: skillIdFromQuery, title: titleFromQuery });
    }
  }, [skillIdFromQuery, titleFromQuery]);

  // 0. 监听 query 中 XClawUsed=true 且带 thread_id 时跳转并清理 query
  useEffect(() => {
    
    if (!threadIdFromQuery) return;
    if (xClawUsedFromQuery !== "true") return;
    if (lastThreadIdRef.current === threadIdFromQuery) return;
    lastThreadIdRef.current = threadIdFromQuery;
    router.replace(`/workspace/chats/${threadIdFromQuery}`);
  }, [router, threadIdFromQuery, xClawUsedFromQuery]);

  // 2. 监听宿主页 postMessage
  useEffect(() => {
    const handleMessage = (event: MessageEvent) => {
      if (event.data?.type !== RECEIVE_MESSAGE_TYPES.SELECTED_SKILL) {
        return;
      }
      if (!isSelectedSkillMessage(event.data)) {
        console.warn("[useIframeSkill] 忽略非法 selectedSkill 消息", event.data);
        return;
      }
      const { id, title } = event.data;
      setSelectedSkill({ skill_id: String(id), title });
    };
    window.addEventListener("message", handleMessage);
    return () => window.removeEventListener("message", handleMessage);
  }, []);

  // 发送选择预定义 skill
  const sendSelectSkill = useCallback((skill_id: string) => {
    const message = { type: POST_MESSAGE_TYPES.SELECT_SKILL, skill_id };
    console.log("[useIframeSkill] sendSelectSkill:", message);
    sendToParent(message);
  }, []);

  // 打开 skill 选择对话框
  const openSkillDialog = useCallback(() => {
    const message = {
      type: POST_MESSAGE_TYPES.OPEN_SKILL_DIALOG,
      openSkillDialog: true,
    } as const;
    console.log("[useIframeSkill] openSkillDialog:", message);
    sendToParent(message);
  }, []);

  // 清除选中并发送 skill_id=0 给主页
  const clearSkill = useCallback(() => {
    setSelectedSkill(null);
    // 发送 skill_id=0 给主页，通知取消选择
    const message = { type: POST_MESSAGE_TYPES.SELECT_SKILL, skill_id: "0" };
    console.log("[useIframeSkill] clearSkill, sending skill_id=0:", message);
    sendToParent(message);
  }, []);

  return { selectedSkill, sendSelectSkill, openSkillDialog, clearSkill };
}