"use client";

import React, { useMemo, useState } from "react";
import {
  AlertCircle,
  Box,
  Copy,
  Download,
  ExternalLink,
  Globe,
  ImageIcon,
  Loader2,
  MoreVertical,
  Trash2,
} from "lucide-react";
import { Button } from "@/components/ui/button";
import { CanvasToolbarIcon } from "@/components/canvas/CanvasToolbarIcon";
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuSeparator,
  DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu";
import useSWR from "swr";
import { toast } from "sonner";
import { getNodeContentScale } from "@/lib/canvas/node-content-scale";
import {
  DEFAULT_PUBLIC_THREED_MODEL_ALIAS_CONFIG,
  resolveThreeDModelDisplayName,
  type PublicThreeDModelAliasConfig,
} from "@/lib/model3d/model-alias-config";
import {
  Model3DViewer,
  type Model3DViewerSettings,
} from "@/components/canvas/model3d/Model3DViewer";
import {
  getCanvasToolbarItem,
  type CanvasToolbarConfig,
} from "@/lib/canvas-toolbar-config";
import { useCanvasStore } from "@/store/canvas";
import type { CanvasNode, Model3DNodeData } from "@/types/canvas";
import { cn, fetcher } from "@/lib/utils";

interface Model3DNodeProps {
  node: CanvasNode;
  isSelected: boolean;
  canvasToolbarConfig?: CanvasToolbarConfig;
}

const CANVAS_MODEL_VIEWER_SETTINGS: Model3DViewerSettings = {
  cameraPreset: "iso",
  envPreset: "studio",
  lightIntensity: 1,
  autoRotate: false,
  showGrid: true,
  wireframe: false,
};

function inferModelFileName(data: Model3DNodeData, nodeId: string) {
  const raw = String(data.src || "").trim();
  if (!raw) return `model3d-${nodeId.slice(0, 6)}.glb`;
  try {
    const url = new URL(raw);
    const filename = url.pathname.split("/").filter(Boolean).pop();
    if (filename) return decodeURIComponent(filename);
  } catch {}
  return raw.split("/").pop() || `model3d-${nodeId.slice(0, 6)}.glb`;
}

function inferModelFormat(data: Model3DNodeData) {
  if (data.format) return data.format.toUpperCase();
  const src = String(data.src || "").toLowerCase();
  if (src.endsWith(".gltf")) return "GLTF";
  if (src.endsWith(".glb")) return "GLB";
  if (src.endsWith(".usdz")) return "USDZ";
  if (src.endsWith(".fbx")) return "FBX";
  if (src.endsWith(".obj")) return "OBJ";
  if (src.endsWith(".stl")) return "STL";
  if (src.endsWith(".3mf")) return "3MF";
  return "3D";
}

function readWorldAssetPayload(data: Model3DNodeData) {
  const raw = data.metadata?.assetPayload;
  return raw && typeof raw === "object" ? (raw as Record<string, unknown>) : null;
}

function readOptionalUrl(value: unknown) {
  return typeof value === "string" && value.trim() ? value.trim() : null;
}

export function Model3DNode({
  node,
  isSelected,
  canvasToolbarConfig,
}: Model3DNodeProps) {
  const data = node.data as Model3DNodeData;
  const [isHovered, setIsHovered] = useState(false);
  const deleteNode = useCanvasStore((state) => state.deleteNode);
  const model3dOpen3dItem = getCanvasToolbarItem(
    canvasToolbarConfig,
    "model3dOpen3d"
  );
  const model3dOpenWorldItem = getCanvasToolbarItem(
    canvasToolbarConfig,
    "model3dOpenWorld"
  );
  const model3dMoreItem = getCanvasToolbarItem(
    canvasToolbarConfig,
    "model3dMore"
  );
  const model3dOpenMeshItem = getCanvasToolbarItem(
    canvasToolbarConfig,
    "model3dOpenMesh"
  );
  const model3dOpenImageItem = getCanvasToolbarItem(
    canvasToolbarConfig,
    "model3dOpenImage"
  );
  const model3dDownloadItem = getCanvasToolbarItem(
    canvasToolbarConfig,
    "model3dDownload"
  );
  const model3dCopyItem = getCanvasToolbarItem(
    canvasToolbarConfig,
    "model3dCopy"
  );
  const model3dDeleteItem = getCanvasToolbarItem(
    canvasToolbarConfig,
    "model3dDelete"
  );
  const viewportZoom = useCanvasStore((state) => state.viewport.zoom);
  const { data: modelAliasConfigResponse } = useSWR<{
    success?: boolean;
    data?: PublicThreeDModelAliasConfig;
  }>("/api/canvas/model-aliases", fetcher);
  const providerLabel =
    data.metadata?.provider === "worldlabs-3d" ? "World Labs 结果" : "Tripo3D 结果";
  const publicModelAliasConfig =
    modelAliasConfigResponse?.data || DEFAULT_PUBLIC_THREED_MODEL_ALIAS_CONFIG;
  const modelVersionLabel = useMemo(
    () =>
      resolveThreeDModelDisplayName(
        data.modelVersion,
        publicModelAliasConfig.tripoModels,
        providerLabel,
      ),
    [data.modelVersion, providerLabel, publicModelAliasConfig.tripoModels],
  );
  const worldAssetPayload = readWorldAssetPayload(data);
  const worldUrl = readOptionalUrl(worldAssetPayload?.worldUrl);
  const worldImageUrl =
    readOptionalUrl(worldAssetPayload?.panoUrl) ||
    readOptionalUrl(worldAssetPayload?.thumbnailUrl) ||
    readOptionalUrl(data.previewImage);
  const primaryOpenUrl = worldUrl || data.src || null;
  const primaryOpenItem = worldUrl ? model3dOpenWorldItem : model3dOpen3dItem;
  const primaryOpenLabel = primaryOpenItem.label;
  const shouldRenderLiveViewer =
    isSelected && data.status === "completed" && Boolean(data.src);
  const placeholderSizeScale = useMemo(
    () => getNodeContentScale(node.size, { width: 360, height: 300 }, 0.76),
    [node.size.height, node.size.width]
  );
  const placeholderViewportScale = useMemo(
    () =>
      Math.min(
        1.55,
        Math.max(0.72, placeholderSizeScale / Math.max(viewportZoom, 0.42))
      ),
    [placeholderSizeScale, viewportZoom]
  );
  const placeholderCenterStyle = useMemo<React.CSSProperties>(
    () => ({
      transform: `scale(${placeholderViewportScale})`,
      transformOrigin: "center center",
    }),
    [placeholderViewportScale]
  );
  const placeholderBadgeStyle = useMemo<React.CSSProperties>(
    () => ({
      transform: `scale(${placeholderViewportScale})`,
      transformOrigin: "top left",
    }),
    [placeholderViewportScale]
  );

  const handleOpen = () => {
    if (!primaryOpenUrl) {
      toast.error("暂无可打开的地址");
      return;
    }
    window.open(primaryOpenUrl, "_blank", "noopener,noreferrer");
  };

  const handleOpenMesh = () => {
    if (!data.src) {
      toast.error("暂无可打开的 Mesh 地址");
      return;
    }
    window.open(data.src, "_blank", "noopener,noreferrer");
  };

  const handleOpenImage = () => {
    if (!worldImageUrl) {
      toast.error("暂无可打开的图像预览");
      return;
    }
    window.open(worldImageUrl, "_blank", "noopener,noreferrer");
  };

  const handleDownload = () => {
    if (!data.src) {
      toast.error("暂无可下载的 3D 文件");
      return;
    }
    const link = document.createElement("a");
    link.href = data.src;
    link.download = inferModelFileName(data, node.id);
    link.click();
  };

  const handleCopy = async () => {
    if (!primaryOpenUrl) {
      toast.error("暂无可复制的地址");
      return;
    }
    try {
      await navigator.clipboard.writeText(primaryOpenUrl);
      toast.success(worldUrl ? "已复制 World 链接" : "已复制 3D 链接");
    } catch {
      toast.error("复制失败");
    }
  };

  return (
    <div
      className={cn(
        "relative h-full w-full overflow-hidden rounded-lg border bg-card shadow-sm",
        isSelected && "border-primary"
      )}
      onMouseEnter={() => setIsHovered(true)}
      onMouseLeave={() => setIsHovered(false)}
    >
      {data.status === "generating" ? (
        <div className="relative flex h-full w-full select-none items-center justify-center overflow-hidden bg-[#f7f9fc] dark:bg-zinc-900">
          <div className="absolute left-4 top-4" style={placeholderBadgeStyle}>
            <span className="flex h-11 w-11 items-center justify-center rounded-full border border-[#dde5f4] bg-white text-[#566374] shadow-[0_0_0_1px_rgba(219,226,238,0.22)] dark:border-zinc-700 dark:bg-zinc-950 dark:text-zinc-200">
              <Box className="h-5 w-5" />
            </span>
          </div>
          <div className="flex flex-col items-center gap-2 px-6 text-center" style={placeholderCenterStyle}>
            <span className="flex h-12 w-12 items-center justify-center rounded-[18px] border border-[#dde5f4] bg-white text-[#5b76fe] shadow-[0_0_0_1px_rgba(219,226,238,0.22)] dark:border-zinc-700 dark:bg-zinc-950 dark:text-cyan-200">
              <Loader2 className="h-5 w-5 animate-spin" />
            </span>
            <span className="text-sm font-medium text-[#3f4b63] dark:text-zinc-100">
              {data.metadata?.provider === "worldlabs-3d"
                ? "3D 世界生成中..."
                : "3D 模型生成中..."}
            </span>
            <span className="text-xs text-[#7a8395] dark:text-zinc-400">
              {data.modelVersion ? `模型：${modelVersionLabel}` : providerLabel}
            </span>
          </div>
        </div>
      ) : data.status === "failed" ? (
        <div className="relative flex h-full w-full select-none items-center justify-center overflow-hidden bg-[#fff7f8] dark:bg-red-950/30">
          <div className="absolute left-4 top-4" style={placeholderBadgeStyle}>
            <span className="flex h-11 w-11 items-center justify-center rounded-full border border-[#f2d7dd] bg-white text-[#d24f63] shadow-[0_0_0_1px_rgba(242,215,221,0.22)] dark:border-red-900/60 dark:bg-zinc-950 dark:text-red-300">
              <AlertCircle className="h-5 w-5" />
            </span>
          </div>
          <div className="flex flex-col items-center gap-2 px-6 text-center" style={placeholderCenterStyle}>
            <span className="flex h-12 w-12 items-center justify-center rounded-[18px] border border-[#f2d7dd] bg-white text-[#d24f63] shadow-[0_0_0_1px_rgba(242,215,221,0.22)] dark:border-red-900/60 dark:bg-zinc-950 dark:text-red-300">
              <AlertCircle className="h-5 w-5" />
            </span>
            <span className="text-sm font-medium text-[#c84b5e] dark:text-red-200">
              {data.metadata?.provider === "worldlabs-3d"
                ? "3D 世界生成失败"
                : "3D 模型生成失败"}
            </span>
            <span className="text-xs text-[#b77480] dark:text-red-300/80">
              可以重新选择模型后再试一次
            </span>
          </div>
        </div>
      ) : (
        <div className="flex h-full w-full flex-col">
          <div className="relative min-h-0 flex-1 overflow-hidden bg-muted/30">
            {shouldRenderLiveViewer ? (
              <div
                className="h-full w-full"
                data-canvas-no-drag="true"
                onWheel={(event) => event.stopPropagation()}
              >
                <Model3DViewer
                  emptyStateTitle="准备查看 3D 模型"
                  modelUrl={data.src}
                  previewImageUrl={data.previewImage}
                  settings={CANVAS_MODEL_VIEWER_SETTINGS}
                />
              </div>
            ) : data.previewImage ? (
              <img
                src={data.previewImage}
                alt={data.prompt || "3D 预览"}
                className="h-full w-full object-cover"
                draggable={false}
              />
            ) : (
              <div className="flex h-full w-full items-center justify-center">
                <Box className="h-12 w-12 text-muted-foreground/60" />
              </div>
            )}
            <div className="absolute left-2 top-2 rounded-md bg-background/85 px-2 py-0.5 text-[10px] font-medium">
              {inferModelFormat(data)}
            </div>
          </div>
          <div className="space-y-1 border-t px-3 py-2">
            <p className="truncate text-xs font-medium">
              {data.prompt || inferModelFileName(data, node.id)}
            </p>
            <p className="truncate text-[11px] text-muted-foreground">
              {data.modelVersion ? `模型: ${modelVersionLabel}` : providerLabel}
            </p>
          </div>
        </div>
      )}

      {isHovered && data.status === "completed" && (
        <div className="absolute right-2 top-2 flex items-center gap-1">
          <Button
            variant="secondary"
            size="icon"
            className="h-7 w-7"
            onClick={handleOpen}
            title={primaryOpenLabel}
          >
            <CanvasToolbarIcon
              className="h-4 w-4"
              fallback={worldUrl ? Globe : ExternalLink}
              item={primaryOpenItem}
            />
          </Button>
          <DropdownMenu>
            <DropdownMenuTrigger asChild>
              <Button
                variant="secondary"
                size="icon"
                className="h-7 w-7"
                title={model3dMoreItem.label}
              >
                <CanvasToolbarIcon
                  className="h-4 w-4"
                  fallback={MoreVertical}
                  item={model3dMoreItem}
                />
              </Button>
            </DropdownMenuTrigger>
            <DropdownMenuContent align="end">
              <DropdownMenuItem onClick={handleOpen}>
                <CanvasToolbarIcon
                  className="mr-2 h-4 w-4"
                  fallback={worldUrl ? Globe : ExternalLink}
                  item={primaryOpenItem}
                />
                {primaryOpenLabel}
              </DropdownMenuItem>
              {worldUrl && data.src && data.src !== worldUrl && (
                <DropdownMenuItem onClick={handleOpenMesh}>
                  <CanvasToolbarIcon
                    className="mr-2 h-4 w-4"
                    fallback={Box}
                    item={model3dOpenMeshItem}
                  />
                  {model3dOpenMeshItem.label}
                </DropdownMenuItem>
              )}
              {worldImageUrl && (
                <DropdownMenuItem onClick={handleOpenImage}>
                  <CanvasToolbarIcon
                    className="mr-2 h-4 w-4"
                    fallback={ImageIcon}
                    item={model3dOpenImageItem}
                  />
                  {model3dOpenImageItem.label}
                </DropdownMenuItem>
              )}
              <DropdownMenuItem onClick={handleDownload}>
                <CanvasToolbarIcon
                  className="mr-2 h-4 w-4"
                  fallback={Download}
                  item={model3dDownloadItem}
                />
                {model3dDownloadItem.label}
              </DropdownMenuItem>
              <DropdownMenuItem onClick={handleCopy}>
                <CanvasToolbarIcon
                  className="mr-2 h-4 w-4"
                  fallback={Copy}
                  item={model3dCopyItem}
                />
                {model3dCopyItem.label}
              </DropdownMenuItem>
              <DropdownMenuSeparator />
              <DropdownMenuItem className="text-destructive" onClick={() => deleteNode(node.id)}>
                <CanvasToolbarIcon
                  className="mr-2 h-4 w-4"
                  fallback={Trash2}
                  item={model3dDeleteItem}
                />
                {model3dDeleteItem.label}
              </DropdownMenuItem>
            </DropdownMenuContent>
          </DropdownMenu>
        </div>
      )}
    </div>
  );
}