WebMCP Schema & Code Generator

"use client";

import { useEffect, useState } from "react";

/**
 * Chrome WebMCP Tool Registration Hook (React 18/19 StrictMode Safe)
 * Standard: document.modelContext (Chromium 146.0.7672.0+ Early Preview)
 * Lifecycle: AbortController + AbortSignal teardown prevents InvalidStateError
 * Generated by AIIndieKit WebMCP Schema Generator
 */
export function useGenerateAiVideoTool() {
  const [isRegistered, setIsRegistered] = useState(false);

  useEffect(() => {
    if (typeof window === "undefined") return;

    const controller = new AbortController();

    // 1. Feature detection: inspect document.modelContext (legacy navigator.modelContext fallback)
    const win = window as any;
    const mc = win.document?.modelContext || win.navigator?.modelContext;

    if (!mc || typeof mc.registerTool !== "function") {
      console.debug(
        "[WebMCP] document.modelContext is not supported. Requires Chromium 146.0.7672.0+ with chrome://flags/#enable-webmcp-testing"
      );
      return;
    }

    // 2. Define standard JSON Schema inputSchema
    const inputSchema = {
      type: "object",
      properties: {
      "image_a_url": {
              "type": "string",
              "description": "Starting keyframe image URL (PNG, JPG, WebP)"
      },
      "image_b_url": {
              "type": "string",
              "description": "Ending keyframe image URL for video morphing or interpolation"
      },
      "duration_seconds": {
              "type": "number",
              "description": "Target video duration in seconds (5 or 10)"
      },
      "aspect_ratio": {
              "type": "string",
              "description": "Output video aspect ratio (16:9, 9:16, or 1:1)"
      },
      "motion_prompt": {
              "type": "string",
              "description": "Optional description of camera movement or subject dynamics"
      }
},
      required: ["image_a_url","duration_seconds"],
      additionalProperties: false,
    };

    // 3. Register tool asynchronously with AbortSignal for StrictMode / HMR safety
    (async () => {
      try {
        await mc.registerTool(
          {
            name: "generate_ai_video",
            description: "Generate an animated video transition and camera motion between two uploaded keyframe images.",
            inputSchema,
            annotations: {
              readOnlyHint: false,
              consequentialHint: false,
              untrustedContentHint: true,
            },
            execute: async (args: Record<string, unknown>) => {
              console.log("[WebMCP Execute] Calling 'generate_ai_video' with:", args);

              // Dispatch request to your backend handler
              const response = await fetch("/api/generate-video", {
                method: "POST",
                headers: {
                  "Content-Type": "application/json",
                },
                body: JSON.stringify(args),
              });

              if (!response.ok) {
                throw new Error(`Tool execution failed with status ${response.status}`);
              }

              const result = await response.json();
              return {
                content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
              };
            },
          },
          { signal: controller.signal }
        );

        if (!controller.signal.aborted) {
          setIsRegistered(true);
          console.log("[WebMCP] Tool 'generate_ai_video' registered successfully.");
        }
      } catch (err) {
        if (!controller.signal.aborted) {
          console.error("[WebMCP] Registration error:", err);
        }
      }
    })();

    // 4. Teardown via AbortSignal (and legacy unregisterTool fallback) on unmount
    return () => {
      controller.abort();
      if (typeof mc.unregisterTool === "function") {
        try {
          mc.unregisterTool("generate_ai_video");
        } catch (_) {}
      }
    };
  }, []);

  return { isRegistered };
}