"use client"; import { useEffect, useState } from "react"; import { Plus, Sparkles } from "lucide-react"; import { AgentWorkflowMaterialPicker } from "@/components/agents/AgentWorkflowMaterialPicker"; import { createEmptyAgentWorkflowMaterialSelection } from "@/lib/agents/workflow-material-config"; import { cn } from "@/lib/utils"; import type { AgentWorkflowInputs } from "@/components/agents/AgentWorkflowPanel"; import type { AgentWorkflowMaterialImage, AgentWorkflowMaterialSearchResult, AgentWorkflowMaterialQueryType, AgentWorkflowMaterialSelections, } from "@/types/agent"; interface AgentWorkflowMaterialSelectionPanelProps { inputs: AgentWorkflowInputs; onInputsChange: (next: AgentWorkflowInputs) => void; materials: AgentWorkflowMaterialSelections; onMaterialsChange: (next: AgentWorkflowMaterialSelections) => void; productSelectionLimit?: number | null; title?: string; description?: string; disabled?: boolean; disabledHint?: string; theme?: "light" | "dark"; } function getUploadedWorkflowMaterialImages( result?: AgentWorkflowMaterialSearchResult | null ): AgentWorkflowMaterialImage[] { return Array.isArray(result?.images) ? result.images .filter((item) => item.sourceType === "upload") .slice() .sort((left, right) => left.sortOrder - right.sortOrder) : []; } function markSearchWorkflowMaterialImages( images: AgentWorkflowMaterialImage[] ): AgentWorkflowMaterialImage[] { return images.map((image) => ({ ...image, sourceType: image.sourceType === "upload" ? "upload" : "search", })); } function buildWorkflowMaterialResult(params: { currentResult?: AgentWorkflowMaterialSearchResult | null; images: AgentWorkflowMaterialImage[]; fallbackQuery: string; fallbackQueryType: AgentWorkflowMaterialQueryType; }): AgentWorkflowMaterialSearchResult { const normalizedImages: AgentWorkflowMaterialImage[] = params.images.map((image, index) => ({ ...image, sourceType: image.sourceType === "upload" ? "upload" : "search", sortOrder: index, })); const validIds = new Set(normalizedImages.map((image) => image.id)); const normalizedQuery = params.fallbackQuery.trim().toUpperCase(); const currentResult = params.currentResult; const queryType = currentResult?.queryType || params.fallbackQueryType; const hasUploadedImages = normalizedImages.some((image) => image.sourceType === "upload"); const fallbackMatchedSku = queryType === "sku" ? currentResult?.matchedSku || normalizedQuery || "手动上传素材" : currentResult?.matchedSku || (hasUploadedImages ? "手动上传素材" : ""); const fallbackMatchedSpu = currentResult?.matchedSpu || (queryType === "spu" ? normalizedQuery || (hasUploadedImages ? "手动上传素材" : "") : null); return { query: currentResult?.query || normalizedQuery || "手动上传素材", queryType, matchedSku: currentResult?.matchedSku || fallbackMatchedSku, matchedSpu: fallbackMatchedSpu, materialId: currentResult?.materialId || null, productModel: currentResult?.productModel || fallbackMatchedSpu || currentResult?.matchedSku || fallbackMatchedSku || null, zipUrl: currentResult?.zipUrl || null, fixedCount: normalizedImages.length > 0 ? normalizedImages.filter((image) => image.group === "fixed").length : currentResult?.fixedCount || 0, otherCount: normalizedImages.length > 0 ? normalizedImages.filter((image) => image.group === "other").length : currentResult?.otherCount || 0, totalCount: normalizedImages.length > 0 ? normalizedImages.length : currentResult?.totalCount || 0, images: normalizedImages, recommendedImageIds: Array.isArray(currentResult?.recommendedImageIds) ? currentResult!.recommendedImageIds.filter((id) => validIds.has(id)) : [], skuOptions: currentResult?.skuOptions || [], goodsInfo: currentResult?.goodsInfo || null, }; } function finalizeWorkflowMaterialResult( result: AgentWorkflowMaterialSearchResult | null ): AgentWorkflowMaterialSearchResult | null { if (!result) return null; const hasImages = Array.isArray(result.images) && result.images.length > 0; const hasSkuOptions = Array.isArray(result.skuOptions) && result.skuOptions.length > 0; return hasImages || hasSkuOptions ? result : null; } function mergeSearchResultWithUploadedImages(params: { searchResult: AgentWorkflowMaterialSearchResult; currentResult?: AgentWorkflowMaterialSearchResult | null; }): AgentWorkflowMaterialSearchResult { const uploadedImages = getUploadedWorkflowMaterialImages(params.currentResult); const searchImages = markSearchWorkflowMaterialImages(params.searchResult.images || []); return buildWorkflowMaterialResult({ currentResult: { ...params.searchResult, images: [...uploadedImages, ...searchImages], }, images: [...uploadedImages, ...searchImages], fallbackQuery: params.searchResult.query, fallbackQueryType: params.searchResult.queryType === "spu" ? "spu" : "sku", }); } export function AgentWorkflowMaterialSelectionPanel({ inputs, onInputsChange, materials, onMaterialsChange, productSelectionLimit = null, title, description, disabled = false, disabledHint = "请先在下方选择一个生图模式,再启用 SKU / SPU 搜图。", theme = "light", }: AgentWorkflowMaterialSelectionPanelProps) { const hasSecondaryMaterialContent = Boolean(String(inputs.secondarySku || "").trim()) || Boolean(materials.secondary.result) || materials.secondary.selectedImageIds.length > 0; const buildLoadingMaterialResult = (params: { keyword: string; currentResult?: AgentWorkflowMaterialSearchResult | null; placeholderCount: number; }): AgentWorkflowMaterialSearchResult => { const count = Math.max(4, Math.min(params.placeholderCount || 8, 16)); return { query: params.keyword, queryType: "sku", matchedSku: params.keyword, matchedSpu: params.currentResult?.matchedSpu || null, materialId: params.currentResult?.materialId || null, productModel: params.currentResult?.productModel || null, zipUrl: null, fixedCount: count, otherCount: 0, totalCount: count, skuOptions: [], recommendedImageIds: [], images: Array.from({ length: count }, (_, index) => ({ id: `loading-${params.keyword}-${index + 1}`, key: `loading-${params.keyword}-${index + 1}`, label: `加载中 ${index + 1}`, fileName: `正在加载白底图 ${index + 1}`, imageUrl: "", assetId: null, mimeType: null, sourceType: "search", group: "fixed", sortOrder: index, })), }; }; const [materialSearchState, setMaterialSearchState] = useState<{ primary: { isLoading: boolean; errorText: string | null }; secondary: { isLoading: boolean; errorText: string | null }; }>({ primary: { isLoading: false, errorText: null }, secondary: { isLoading: false, errorText: null }, }); const [materialUploadState, setMaterialUploadState] = useState<{ primary: { isUploadingNew: boolean; uploadingImageId: string | null }; secondary: { isUploadingNew: boolean; uploadingImageId: string | null }; }>({ primary: { isUploadingNew: false, uploadingImageId: null }, secondary: { isUploadingNew: false, uploadingImageId: null }, }); const [materialSearchMode, setMaterialSearchMode] = useState<{ primary: AgentWorkflowMaterialQueryType; secondary: AgentWorkflowMaterialQueryType; }>({ primary: "sku", secondary: "sku", }); const [isSecondaryMaterialVisible, setIsSecondaryMaterialVisible] = useState( hasSecondaryMaterialContent ); const primarySelectedCount = materials.primary.selectedImageIds.length; const secondarySelectedCount = materials.secondary.selectedImageIds.length; const primarySelectionLimit = typeof productSelectionLimit === "number" && Number.isFinite(productSelectionLimit) && productSelectionLimit > 0 ? Math.max(productSelectionLimit - secondarySelectedCount, primarySelectedCount, 0) : null; const secondarySelectionLimit = typeof productSelectionLimit === "number" && Number.isFinite(productSelectionLimit) && productSelectionLimit > 0 ? Math.max(productSelectionLimit - primarySelectedCount, secondarySelectedCount, 0) : null; useEffect(() => { if (hasSecondaryMaterialContent) { setIsSecondaryMaterialVisible(true); } }, [hasSecondaryMaterialContent]); const updateMaterialSearchState = ( source: "primary" | "secondary", updates: Partial<{ isLoading: boolean; errorText: string | null }> ) => { setMaterialSearchState((prev) => ({ ...prev, [source]: { ...prev[source], ...updates, }, })); }; const updateMaterialUploadState = ( source: "primary" | "secondary", updates: Partial<{ isUploadingNew: boolean; uploadingImageId: string | null }> ) => { setMaterialUploadState((prev) => ({ ...prev, [source]: { ...prev[source], ...updates, }, })); }; const getFallbackMaterialQuery = (source: "primary" | "secondary") => { const currentResult = materials[source].result; const rawInput = String(source === "primary" ? inputs.primarySku : inputs.secondarySku).trim(); return (currentResult?.query || rawInput || "手动上传素材").trim(); }; const getFallbackMaterialQueryType = (source: "primary" | "secondary") => { return materials[source].result?.queryType || materialSearchMode[source]; }; const handleMaterialInputChange = (source: "primary" | "secondary", value: string) => { const inputKey = source === "primary" ? "primarySku" : "secondarySku"; const trimmedValue = value.trim().toUpperCase(); const currentQuery = String(materials[source]?.result?.query || "").trim().toUpperCase(); onInputsChange({ ...inputs, [inputKey]: value, }); if ( trimmedValue !== currentQuery && (materials[source]?.result || materials[source]?.selectedImageIds.length) ) { onMaterialsChange({ ...materials, [source]: createEmptyAgentWorkflowMaterialSelection(), }); } if (materialSearchState[source].errorText) { updateMaterialSearchState(source, { errorText: null }); } }; const handleSearchMaterials = async ( source: "primary" | "secondary", options?: { keyword?: string; queryType?: AgentWorkflowMaterialQueryType; } ) => { const keyword = String(options?.keyword || (source === "primary" ? inputs.primarySku : inputs.secondarySku)).trim(); const queryType = options?.queryType || materialSearchMode[source]; if (!keyword) { updateMaterialSearchState(source, { errorText: "请先输入搜索关键词" }); return; } try { updateMaterialSearchState(source, { isLoading: true, errorText: null }); const searchParams = new URLSearchParams({ keyword, source, queryType, }); const currentSelectionLimit = source === "primary" ? primarySelectionLimit : secondarySelectionLimit; const requiredCount = typeof currentSelectionLimit === "number" && Number.isFinite(currentSelectionLimit) && currentSelectionLimit > 0 ? Math.round(currentSelectionLimit) : null; if (requiredCount) { searchParams.set("requiredCount", String(requiredCount)); } const response = await fetch( `/api/agent-workflow/material-search?${searchParams.toString()}`, { cache: "no-store" } ); const result = await response.json().catch(() => ({})); if (!response.ok || !result?.success || !result?.data) { throw new Error(result?.error?.message || result?.error || "搜索失败"); } const mergedResult = mergeSearchResultWithUploadedImages({ searchResult: result.data, currentResult: materials[source].result, }); const preservedUploadedIds = getUploadedWorkflowMaterialImages(materials[source].result) .map((image) => image.id) .filter((id) => materials[source].selectedImageIds.includes(id)); onMaterialsChange({ ...materials, [source]: { result: mergedResult, selectedImageIds: Array.from( new Set([ ...(Array.isArray(result.data?.recommendedImageIds) ? result.data.recommendedImageIds : []), ...preservedUploadedIds, ]) ).filter((id) => mergedResult.images.some((image) => image.id === id)), }, }); } catch (error) { updateMaterialSearchState(source, { errorText: error instanceof Error ? error.message : "搜索失败", }); } finally { updateMaterialSearchState(source, { isLoading: false }); } }; const handleUploadMaterialImage = async ( source: "primary" | "secondary", file: File, options?: { replaceImageId?: string; } ) => { const currentResult = materials[source].result; const query = getFallbackMaterialQuery(source); const queryType = getFallbackMaterialQueryType(source); const matchedSku = currentResult?.matchedSku || (queryType === "sku" ? String(source === "primary" ? inputs.primarySku : inputs.secondarySku).trim().toUpperCase() : ""); const matchedSpu = currentResult?.matchedSpu || (queryType === "spu" ? String(source === "primary" ? inputs.primarySku : inputs.secondarySku).trim().toUpperCase() : ""); updateMaterialUploadState(source, { isUploadingNew: !options?.replaceImageId, uploadingImageId: options?.replaceImageId || null, }); try { const formData = new FormData(); formData.set("file", file); formData.set("source", source); formData.set("query", query); formData.set("queryType", queryType); if (matchedSku) { formData.set("matchedSku", matchedSku); } if (matchedSpu) { formData.set("matchedSpu", matchedSpu); } if (currentResult?.productModel) { formData.set("productModel", currentResult.productModel); } const response = await fetch("/api/agent-workflow/material-upload", { method: "POST", body: formData, }); const payload = await response.json().catch(() => ({})); if (!response.ok || !payload?.success || !payload?.data?.imageUrl) { throw new Error(payload?.detail || payload?.error || "图片上传失败"); } const uploadedImage = payload.data as AgentWorkflowMaterialImage; const currentUploads = getUploadedWorkflowMaterialImages(currentResult).filter( (image) => image.id !== options?.replaceImageId ); const searchImages = markSearchWorkflowMaterialImages( (currentResult?.images || []).filter((image) => image.sourceType !== "upload") ); const nextResult = buildWorkflowMaterialResult({ currentResult, images: [...currentUploads, { ...uploadedImage, sourceType: "upload" }, ...searchImages], fallbackQuery: query, fallbackQueryType: queryType, }); const nextSelectedIds = Array.from( new Set([ ...materials[source].selectedImageIds.filter( (id) => id !== options?.replaceImageId && nextResult.images.some((image) => image.id === id) ), uploadedImage.id, ]) ); onMaterialsChange({ ...materials, [source]: { result: nextResult, selectedImageIds: nextSelectedIds, }, }); } finally { updateMaterialUploadState(source, { isUploadingNew: false, uploadingImageId: null, }); } }; const handleRemoveUploadedMaterialImage = ( source: "primary" | "secondary", imageId: string ) => { const currentResult = materials[source].result; if (!currentResult) { return; } const nextUploads = getUploadedWorkflowMaterialImages(currentResult).filter( (image) => image.id !== imageId ); const searchImages = markSearchWorkflowMaterialImages( (currentResult.images || []).filter((image) => image.sourceType !== "upload") ); const nextResult = finalizeWorkflowMaterialResult( buildWorkflowMaterialResult({ currentResult, images: [...nextUploads, ...searchImages], fallbackQuery: getFallbackMaterialQuery(source), fallbackQueryType: getFallbackMaterialQueryType(source), }) ); onMaterialsChange({ ...materials, [source]: { result: nextResult, selectedImageIds: materials[source].selectedImageIds.filter( (id) => id !== imageId && (nextResult?.images || []).some((image) => image.id === id) ), }, }); }; const isDark = theme === "dark"; return (