"use client"; import { useCallback, useEffect, useMemo, useRef, useState } from "react"; import { createPortal } from "react-dom"; import { Background, BaseEdge, MiniMap, ReactFlow, ReactFlowProvider, ViewportPortal, getBezierPath, useReactFlow, useViewport, type EdgeProps, type Node, type NodeProps, type NodeTypes, } from "@xyflow/react"; import "@xyflow/react/dist/style.css"; import { Alert, Button, Slider, Space, Spin, Tooltip } from "antd"; import { AppstoreOutlined, BorderOuterOutlined, CloseOutlined, CopyOutlined } from "@ant-design/icons"; import { CanvasPreviewModeProvider } from "@/components/CanvasPreviewModeContext"; import { NodeGenerationComposerPreview, buildNodeGenerationPreviewPayload, } from "@/components/composer/NodeGenerationComposerPreview"; import { workflowNodeTypes } from "@/components/workflowTypes"; import { useI18n } from "@/i18n"; import { DEFAULT_NODE_DIMENSIONS, defaultNodeDimensions } from "@/store/utils/nodeDefaults"; import { authFetch } from "@/utils/authFetch"; import { parseCanvasWorkflowProperties } from "@/utils/canvasWorkflowProperties"; import type { WorkflowFile } from "@/store/workflowStore"; import type { GroupColor, NodeGroup, NodeType, WorkflowNode } from "@/types"; export type CanvasTemplatePreviewData = { id: number; title: string; subtitle: string; image: string; data: string; }; type ParsedPreview = | { workflow: WorkflowFile; error: null } | { workflow: null; error: string }; type CanvasTemplateDetail = { id: number; coverUrl: string; name: string; desp: string; properties: string; }; type CanvasTemplateDetailResponse = { data?: CanvasTemplateDetail; message?: string; status?: string; }; const PREVIEW_FIT_VIEW_OPTIONS = { maxZoom: 1 }; const PREVIEW_GROUP_COLORS: Record = { neutral: "rgba(31, 35, 40, 0.06)", blue: "rgba(59, 130, 246, 0.12)", green: "rgba(34, 197, 94, 0.12)", purple: "rgba(147, 51, 234, 0.12)", orange: "rgba(249, 115, 22, 0.12)", red: "rgba(239, 68, 68, 0.12)", }; interface CanvasTemplatePreviewModalProps { open: boolean; template: CanvasTemplatePreviewData | null; copying: boolean; onClose: () => void; onCopy: (template: CanvasTemplatePreviewData) => void; } function parsePreviewWorkflow(template: CanvasTemplatePreviewData | null): ParsedPreview { if (!template) return { workflow: null, error: "" }; try { const workflow = parseCanvasWorkflowProperties(template.data); return { workflow, error: null }; } catch (error) { return { workflow: null, error: error instanceof Error ? error.message : "Workflow data is invalid.", }; } } function getGroupNodeId(groupId: string): string { return `group-node-${groupId}`; } function createPreviewGroupNode(group: NodeGroup): WorkflowNode { return { id: getGroupNodeId(group.id), type: "group", position: group.position, data: { groupId: group.id, group }, style: { width: group.size.width, height: group.size.height }, selectable: false, draggable: false, } as WorkflowNode; } function buildPreviewNodes(workflow: WorkflowFile): WorkflowNode[] { const groups = workflow.groups; const groupEntries = Object.values(groups ?? {}); if (groupEntries.length === 0) { return workflow.nodes .filter((node) => node.type !== "group") .map((node) => ({ ...node, selected: false, selectable: false, draggable: false, } as WorkflowNode)); } const groupNodes = groupEntries.map(createPreviewGroupNode); const regularNodes = workflow.nodes .filter((node) => node.type !== "group") .map((node) => { const groupId = node.groupId; if (!groupId || !groups?.[groupId]) { return { ...node, selected: false, selectable: false, draggable: false, } as WorkflowNode; } const group = groups[groupId]; const groupNodeId = getGroupNodeId(groupId); const isAlreadyNativeChild = node.parentId === groupNodeId; return { ...node, parentId: groupNodeId, groupId, selected: false, position: isAlreadyNativeChild ? node.position : { x: node.position.x - group.position.x, y: node.position.y - group.position.y, }, selectable: false, draggable: false, } as WorkflowNode; }); return [...groupNodes, ...regularNodes]; } function PreviewGroupNode({ data }: NodeProps) { const group = (data as { group?: NodeGroup }).group; if (!group) return null; return (
{group.name}
); } function PreviewEdge({ id, sourceX, sourceY, targetX, targetY, sourcePosition, targetPosition, markerEnd, }: EdgeProps) { const [edgePath] = getBezierPath({ sourceX, sourceY, sourcePosition, targetX, targetY, targetPosition, curvature: 0.35, }); return ( ); } const previewEdgeTypes = { editable: PreviewEdge, reference: PreviewEdge, }; const previewNodeTypes = Object.fromEntries( Object.entries(workflowNodeTypes).map(([type, Component]) => [ type, (props: NodeProps) => (
{type === "group" ? : }
), ]) ) as NodeTypes; function readPreviewNodeDimension(node: Node, axis: "width" | "height"): number { const defaults = defaultNodeDimensions[node.type as NodeType] || DEFAULT_NODE_DIMENSIONS; const styleValue = node.style?.[axis]; return typeof styleValue === "number" ? styleValue : defaults[axis]; } function getPreviewNodeAbsolutePosition(node: Node, nodeById: Map): { x: number; y: number } { if (!node.parentId) return node.position; const parent = nodeById.get(node.parentId); if (!parent) return node.position; const parentPosition = getPreviewNodeAbsolutePosition(parent, nodeById); return { x: parentPosition.x + node.position.x, y: parentPosition.y + node.position.y, }; } function CanvasTemplatePreviewNodePopover({ selectedNodeId, nodes, edges, }: { selectedNodeId: string | null; nodes: WorkflowNode[]; edges: WorkflowFile["edges"]; }) { const { zoom } = useViewport(); const selectedNode = useMemo( () => nodes.find((node) => node.id === selectedNodeId && node.type !== "group"), [nodes, selectedNodeId] ); if (!selectedNode) return null; const previewPayload = buildNodeGenerationPreviewPayload(selectedNode.id, nodes, edges); if (!previewPayload.text && previewPayload.materials.length === 0) return null; const nodeById = new Map(nodes.map((node) => [node.id, node])); const position = getPreviewNodeAbsolutePosition(selectedNode, nodeById); const width = readPreviewNodeDimension(selectedNode, "width"); const height = readPreviewNodeDimension(selectedNode, "height"); const safeZoom = zoom > 0 ? zoom : 1; return (
event.stopPropagation()} onClick={(event) => event.stopPropagation()} >
); } function CanvasTemplatePreviewFlow({ workflow }: { workflow: WorkflowFile }) { const { t } = useI18n(); const { getViewport, setViewport, fitView } = useReactFlow(); const wrapperRef = useRef(null); const [isMiniMapOpen, setIsMiniMapOpen] = useState(false); const [zoomPercent, setZoomPercent] = useState(100); const [selectedNodeId, setSelectedNodeId] = useState(null); const previewNodes = useMemo(() => buildPreviewNodes(workflow), [workflow]); const previewEdges = useMemo( () => workflow.edges.map((edge) => ({ ...edge, type: edge.type || "editable" })), [workflow.edges] ); const syncZoomPercent = useCallback(() => { setZoomPercent(Math.round(getViewport().zoom * 100)); }, [getViewport]); const handleZoomPercentChange = useCallback((value: number) => { const viewport = getViewport(); const nextZoom = value / 100; const bounds = wrapperRef.current?.getBoundingClientRect(); const centerX = bounds ? bounds.left + bounds.width / 2 : window.innerWidth / 2; const centerY = bounds ? bounds.top + bounds.height / 2 : window.innerHeight / 2; const flowCenterX = (centerX - viewport.x) / viewport.zoom; const flowCenterY = (centerY - viewport.y) / viewport.zoom; setZoomPercent(value); setViewport({ x: centerX - flowCenterX * nextZoom, y: centerY - flowCenterY * nextZoom, zoom: nextZoom, }); }, [getViewport, setViewport]); const handleFitViewFromToolbar = useCallback(() => { void fitView({ ...PREVIEW_FIT_VIEW_OPTIONS, padding: 0.2, duration: 300 }).then(syncZoomPercent); }, [fitView, syncZoomPercent]); useEffect(() => { syncZoomPercent(); }, [syncZoomPercent]); return (
{ if (node.type === "group") return; setSelectedNodeId(node.id); }} onPaneClick={() => setSelectedNodeId(null)} onMoveEnd={syncZoomPercent} proOptions={{ hideAttribution: true }} className="canvas-background" > {isMiniMapOpen && ( )}
`${value}%` }} className="m-0 w-full" />
{zoomPercent}%
); } export function CanvasTemplatePreviewModal({ open, template, copying, onClose, onCopy, }: CanvasTemplatePreviewModalProps) { const { t } = useI18n(); const [resolvedTemplate, setResolvedTemplate] = useState(null); const [isLoadingDetail, setIsLoadingDetail] = useState(false); const [detailError, setDetailError] = useState(null); const parsed = useMemo(() => parsePreviewWorkflow(resolvedTemplate), [resolvedTemplate]); const canCopy = !!resolvedTemplate && !copying && !isLoadingDetail; useEffect(() => { if (!open || !template) { setResolvedTemplate(null); setDetailError(null); setIsLoadingDetail(false); return; } setResolvedTemplate(null); setDetailError(null); setIsLoadingDetail(true); void authFetch(`/api/canvas-templates/detail?id=${template.id}`) .then(async (response) => { const result = (await response.json().catch(() => null)) as CanvasTemplateDetailResponse | null; if (!response.ok || !result?.data) { throw new Error(result?.message || t("templatePreview.invalidWorkflowDescription")); } const detail = result.data; setResolvedTemplate({ id: detail.id, title: detail.name, subtitle: detail.desp, image: detail.coverUrl, data: detail.properties, }); }) .catch((error) => { setDetailError(error instanceof Error ? error.message : t("templatePreview.invalidWorkflowDescription")); }) .finally(() => { setIsLoadingDetail(false); }); }, [open, template, t]); useEffect(() => { if (!open || typeof document === "undefined") return; const hadSliderScope = document.body.classList.contains("canvas-slider-scope"); document.body.classList.add("canvas-slider-scope"); return () => { if (!hadSliderScope) document.body.classList.remove("canvas-slider-scope"); }; }, [open]); if (!open || typeof document === "undefined") return null; const displayTitle = resolvedTemplate?.title || template?.title || t("home.tvShow"); return createPortal(
event.stopPropagation()} >

{displayTitle}

{t("templatePreview.readOnly")}
{isLoadingDetail ? (
) : detailError ? (
) : parsed.workflow ? ( ) : (
)}
, document.body ); }