diff --git a/backend/internal/repository/asset_library.go b/backend/internal/repository/asset_library.go index 409ebbfe8..9058fa879 100644 --- a/backend/internal/repository/asset_library.go +++ b/backend/internal/repository/asset_library.go @@ -28,7 +28,9 @@ type UserAssetFacetRow struct { func (r *Repository) UserAssetsPage(userID string, page int, pageSize int, filter UserAssetPageFilter) ([]model.Asset, int64, error) { var assets []model.Asset var total int64 - query := userAssetFilteredQuery(r.db.Model(&model.Asset{}).Where("user_id = ?", userID), filter, true) + // 素材库页面只展示媒体与文本素材；entity 角色卡由项目资产页管理。列表与 facets 必须同口径排除， + // 否则 facets 会计入 entity，前端出现“全部计数 30 但列表为空”的矛盾。 + query := userAssetFilteredQuery(r.db.Model(&model.Asset{}).Where("user_id = ? AND kind <> ?", userID, "entity"), filter, true) if err := query.Count(&total).Error; err != nil { return nil, 0, err } @@ -38,7 +40,7 @@ func (r *Repository) UserAssetsPage(userID string, page int, pageSize int, filte func (r *Repository) UserAssetFacets(userID string, status string) ([]UserAssetFacetRow, []UserAssetFacetRow, []UserAssetFacetRow, error) { base := func() *gorm.DB { - return userAssetFilteredQuery(r.db.Model(&model.Asset{}).Where("user_id = ?", userID), UserAssetPageFilter{Status: status}, false) + return userAssetFilteredQuery(r.db.Model(&model.Asset{}).Where("user_id = ? AND kind <> ?", userID, "entity"), UserAssetPageFilter{Status: status}, false) } var kindRows []UserAssetFacetRow if err := base().Select("kind AS key, COUNT(*) AS count").Group("kind").Scan(&kindRows).Error; err != nil { diff --git a/web/src/components/assets/asset-library-picker-modal.tsx b/web/src/components/assets/asset-library-picker-modal.tsx index dc4a894b4..fe4d9a1cd 100644 --- a/web/src/components/assets/asset-library-picker-modal.tsx +++ b/web/src/components/assets/asset-library-picker-modal.tsx @@ -146,7 +146,10 @@ export function AssetLibraryPickerModal({ kindLabel: asset.kind === "image" ? "图片" : asset.kind === "video" ? "视频" : asset.kind === "audio" ? "音频" : "文本", searchText: (asset.tags ?? []).join(" "), ...(items.find((item) => item.id === asset.id) || { disabledReason: "此素材不适用于当前操作" }), })), [remoteQuery.data, items]); - const effectivePagination = remoteEnabled ? { current: remotePage, pageSize: remotePageSize, total: remoteQuery.data?.total || 0, onChange: (page: number, pageSize: number) => { setRemotePage(page); setRemotePageSize(pageSize); } } : pagination; + // 远端分页可能没有任何可展示素材（本地素材尚未同步时远端为空）；此时回退调用方传入的本地素材， + // 否则弹窗会被远端空结果遮蔽成空白。分页、分类与关键词过滤必须与数据源同步切换口径。 + const useRemoteItems = remoteEnabled && remoteItems.length > 0; + const effectivePagination = useRemoteItems ? { current: remotePage, pageSize: remotePageSize, total: remoteQuery.data?.total || 0, onChange: (page: number, pageSize: number) => { setRemotePage(page); setRemotePageSize(pageSize); } } : pagination; const uploadInputRef = useRef(null); const initialSelectedIdsRef = useRef(initialSelectedIds); const itemsRef = useRef(items); @@ -161,16 +164,16 @@ export function AssetLibraryPickerModal({ const hasPluginSource = useMemo(() => Object.keys(categoryLabels).some((value) => value.startsWith("external:")) || pluginItems.some((item) => item.category.startsWith("external:")), [categoryLabels, pluginItems]); // 媒体类型在分类之前收窄数据源，让左侧分类计数、网格和分页始终描述同一批素材。 const sourceItems = useMemo(() => { - const base = source === "plugin" ? pluginItems : remoteEnabled ? remoteItems : localItems; + const base = source === "plugin" ? pluginItems : useRemoteItems ? remoteItems : localItems; if (mediaKind === "all") return base; return base.filter((item) => pickerItemMediaKind(item) === mediaKind); - }, [localItems, mediaKind, pluginItems, remoteEnabled, remoteItems, source]); + }, [localItems, mediaKind, pluginItems, useRemoteItems, remoteItems, source]); const activeSourceItems = useMemo(() => sourceItems.filter((item) => !item.archived), [sourceItems]); const archivedItems = useMemo(() => sourceItems.filter((item) => item.archived), [sourceItems]); const mediaKindOptions = useMemo(() => (remoteKind ? [] : Array.from(new Set(mediaKinds))), [mediaKinds, remoteKind]); const sourceFolders = source === "plugin" ? folders : []; const showCategories = source === "local" || !sourceFolders.length; - const normalCategories = useMemo(() => remoteEnabled ? Object.keys(categoryLabels).filter((value) => value !== "archived" && !value.startsWith("external:")) : ["all", ...Array.from(new Set(activeSourceItems.map((item) => item.category || "other"))).filter((value) => value !== "all")], [activeSourceItems, categoryLabels, remoteEnabled]); + const normalCategories = useMemo(() => useRemoteItems ? Object.keys(categoryLabels).filter((value) => value !== "archived" && !value.startsWith("external:")) : ["all", ...Array.from(new Set(activeSourceItems.map((item) => item.category || "other"))).filter((value) => value !== "all")], [activeSourceItems, categoryLabels, useRemoteItems]); const archivedCount = archivedItems.length; const isRecycleBin = category === "archived"; @@ -183,9 +186,9 @@ export function AssetLibraryPickerModal({ return false; } if (folderId !== "all" && (item.folderId || "") !== folderId) return false; - return remoteEnabled || !query || [item.title, item.searchText || "", item.description || ""].join(" ").toLowerCase().includes(query); + return useRemoteItems || !query || [item.title, item.searchText || "", item.description || ""].join(" ").toLowerCase().includes(query); }); - }, [category, folderId, keyword, sourceItems, remoteEnabled]); + }, [category, folderId, keyword, sourceItems, useRemoteItems]); const selectedIds = useMemo( () => Array.from(selected).filter((id) => { @@ -481,7 +484,7 @@ export function AssetLibraryPickerModal({
- {remoteEnabled && remoteQuery.isError ?
素材读取失败
: loading || (remoteEnabled && remoteQuery.isFetching) ? ( + {remoteEnabled && remoteQuery.isError ?
素材读取失败
: loading || (useRemoteItems && remoteQuery.isFetching) ? (
正在读取素材 diff --git a/web/src/pages/assets/index.tsx b/web/src/pages/assets/index.tsx index ef8e563d4..9ee8a69cc 100644 --- a/web/src/pages/assets/index.tsx +++ b/web/src/pages/assets/index.tsx @@ -164,16 +164,17 @@ export default function AssetsPage() { const start = (page - 1) * pageSize; return filteredAssets.slice(start, start + pageSize); }, [filteredAssets, page, pageSize]); - const visibleAssets = useMemo( - () => (assetPageQuery.data?.assets || localVisibleAssets).filter((asset): asset is LibraryAsset => asset.kind !== "entity"), - [assetPageQuery.data?.assets, localVisibleAssets], - ); + // 远端分页结果可能只含被本页排除的 entity 角色卡；此时若直接采用远端结果， + // 本地已保存、尚未同步到服务端的素材会被遮蔽成空页，因此回退到本地筛选结果。 + const remotePageAssets = useMemo(() => (assetPageQuery.data?.assets || []).filter((asset): asset is LibraryAsset => asset.kind !== "entity"), [assetPageQuery.data?.assets]); + const useRemotePage = remotePageAssets.length > 0; + const visibleAssets = useMemo(() => useRemotePage ? remotePageAssets : localVisibleAssets, [useRemotePage, remotePageAssets, localVisibleAssets]); const visibleAssetIds = useMemo(() => visibleAssets.map((asset) => asset.id), [visibleAssets]); const allFilteredSelected = visibleAssetIds.length > 0 && visibleAssetIds.every((id) => selectedIds.includes(id)); const canCreateAsset = page === 1 && viewMode === "library" && !keyword.trim() && kindFilter === "all" && categoryFilter === "all" && folderFilter === "all"; - const totalAssets = assetPageQuery.data?.total ?? filteredAssets.length; - const kindCounts = useMemo(() => assetCountMap(kindOptions, assetPageQuery.data?.kindCounts, viewMode === "trash" ? trashAssets : activeAssets, (asset) => asset.kind), [activeAssets, assetPageQuery.data?.kindCounts, trashAssets, viewMode]); - const categoryCounts = useMemo(() => assetCountMap(categoryOptions, assetPageQuery.data?.categoryCounts, viewMode === "trash" ? trashAssets : activeAssets, (asset) => asset.category || "other"), [activeAssets, assetPageQuery.data?.categoryCounts, trashAssets, viewMode]); + const totalAssets = useRemotePage ? assetPageQuery.data?.total ?? filteredAssets.length : filteredAssets.length; + const kindCounts = useMemo(() => assetCountMap(kindOptions, useRemotePage ? assetPageQuery.data?.kindCounts : undefined, viewMode === "trash" ? trashAssets : activeAssets, (asset) => asset.kind), [activeAssets, assetPageQuery.data?.kindCounts, trashAssets, useRemotePage, viewMode]); + const categoryCounts = useMemo(() => assetCountMap(categoryOptions, useRemotePage ? assetPageQuery.data?.categoryCounts : undefined, viewMode === "trash" ? trashAssets : activeAssets, (asset) => asset.category || "other"), [activeAssets, assetPageQuery.data?.categoryCounts, trashAssets, useRemotePage, viewMode]); const folderCounts = assetPageQuery.data?.folderCounts || {}; useEffect(() => { @@ -525,7 +526,7 @@ export default function AssetsPage() { {validAssets.length} 个素材} + meta={{totalAssets} 个素材} actions={
@@ -725,7 +726,7 @@ export default function AssetsPage() { onDelete={() => setBatchDeleteOpen(true)} /> ) : null} - {validAssets.length === 0 ? ( + {validAssets.length === 0 && totalAssets === 0 ? ( viewMode === "trash" ? ( ) : ( @@ -733,7 +734,7 @@ export default function AssetsPage() { ) ) : ( <> - {filteredAssets.length === 0 ? ( + {visibleAssets.length === 0 ? ( ) : ( @@ -1519,7 +1520,9 @@ function readAssetGridDensity(): AssetGridDensity { function assetCountMap(options: T[], remote: Record | undefined, fallback: LibraryAsset[], valueOf: (asset: LibraryAsset) => string) { const result = new Map(); options.forEach((option) => { - if (remote) result.set(option.value, option.value === "all" ? Object.values(remote).reduce((sum, count) => sum + count, 0) : remote[option.value] || 0); + // 列表只展示 LibraryAsset（entity 角色卡被排除）；"全部"计数只能累加选项里声明的类型， + // 否则远端 facets 里的 entity 会计入"全部"，出现计数 30 但列表为空的矛盾。 + if (remote) result.set(option.value, option.value === "all" ? options.reduce((sum, item) => item.value === "all" ? sum : sum + (remote[item.value] || 0), 0) : remote[option.value] || 0); else result.set(option.value, option.value === "all" ? fallback.length : fallback.filter((asset) => valueOf(asset) === option.value).length); }); return result; diff --git a/web/src/pages/projects/detail/assets.tsx b/web/src/pages/projects/detail/assets.tsx index 4e5fd335a..78b36cbb8 100644 --- a/web/src/pages/projects/detail/assets.tsx +++ b/web/src/pages/projects/detail/assets.tsx @@ -1,5 +1,5 @@ import { useEffect, useMemo, useState, type ReactNode } from "react"; -import { useMutation, useMutationState, useQuery } from "@tanstack/react-query"; +import { keepPreviousData, useMutation, useMutationState, useQuery, useQueryClient } from "@tanstack/react-query"; import { App, Button, Dropdown, Form, Input, Modal, Popconfirm, Tabs, type FormInstance } from "antd"; import { Box, Check, ChevronDown, Download, FileText, FolderOpen, FolderPlus, Image as ImageIcon, Link2, MoreHorizontal, MoveRight, Music2, Pencil, Plus, RefreshCw, Search, Sparkles, Trash2, Upload, UserRound, Video, VolumeX } from "lucide-react"; @@ -90,6 +90,7 @@ export default function ProjectAssetsView({ detail, refreshProject }: ProjectDet const [voicePickerOpen, setVoicePickerOpen] = useState(false); const [voiceInstructions, setVoiceInstructions] = useState(""); const [form] = Form.useForm(); + const queryClient = useQueryClient(); const assetsQuery = useQuery({ queryKey: ["project", detail.project.id, "assets", page, pageSize, category, folderId, debouncedKeyword], @@ -100,12 +101,26 @@ export default function ProjectAssetsView({ detail, refreshProject }: ProjectDet folderId: folderId === ALL_FOLDERS ? undefined : folderId, query: debouncedKeyword || undefined, }), + placeholderData: keepPreviousData, }); const foldersQuery = useQuery({ queryKey: ["project", detail.project.id, "asset-folders"], queryFn: () => listProjectAssetFolders(detail.project.id) }); - const showPendingCandidates = folderId === ALL_FOLDERS && (category === "all" || category === "character"); + // 左侧分类计数不能依赖跟随当前 tab 的候选查询，否则"角色"的数字会随所在 tab 的待确认数漂移； + // 这里独立取"全部待确认"和"角色待确认"两个总数，计数与当前所处分类解耦。 + const pendingCountQuery = useQuery({ + queryKey: ["project", detail.project.id, "asset-candidates", "pending-count"], + queryFn: () => listProjectAssetCandidates(detail.project.id, { page: 1, pageSize: 1, status: "pending_confirmation" }), + }); + const characterPendingCountQuery = useQuery({ + queryKey: ["project", detail.project.id, "asset-candidates", "character-pending-count"], + queryFn: () => listProjectAssetCandidates(detail.project.id, { page: 1, pageSize: 1, category: "character", status: "pending_confirmation" }), + }); + // 章节提取会产出角色、场景和道具三类候选；待确认列表必须跟随当前分类，否则非角色候选永远无法确认。 + const candidateCategory = category === "all" ? undefined : category; + const candidateLabel = category === "all" ? "资产" : categoryLabel(category); + const showPendingCandidates = folderId === ALL_FOLDERS; const candidatesQuery = useQuery({ - queryKey: ["project", detail.project.id, "asset-candidates", candidatePage, candidatePageSize, "character", "pending_confirmation", debouncedKeyword], - queryFn: () => listProjectAssetCandidates(detail.project.id, { page: candidatePage, pageSize: candidatePageSize, category: "character", status: "pending_confirmation", query: debouncedKeyword || undefined }), + queryKey: ["project", detail.project.id, "asset-candidates", candidatePage, candidatePageSize, candidateCategory || "all", "pending_confirmation", debouncedKeyword], + queryFn: () => listProjectAssetCandidates(detail.project.id, { page: candidatePage, pageSize: candidatePageSize, category: candidateCategory, status: "pending_confirmation", query: debouncedKeyword || undefined }), }); const assets = assetsQuery.data?.assets || []; const assetFolders = foldersQuery.data?.folders || []; @@ -140,6 +155,8 @@ export default function ProjectAssetsView({ detail, refreshProject }: ProjectDet setCategory(nextCategory); setFolderId(ALL_FOLDERS); setPage(1); + // 待确认列表跟随分类后，不同分类的候选数量不同，切分类必须回到第 1 页。 + setCandidatePage(1); }; const projectAssetIds = new Set(assets.map((asset) => asset.id)); @@ -180,8 +197,8 @@ export default function ProjectAssetsView({ detail, refreshProject }: ProjectDet const categoryCounts = categories.map((value) => ({ value, count: value === "all" - ? totalAssetCount + (candidatesQuery.data?.total || 0) - : (categoryCountMap[value] || 0) + (value === "character" ? candidatesQuery.data?.total || 0 : 0), + ? totalAssetCount + (pendingCountQuery.data?.total || 0) + : (categoryCountMap[value] || 0) + (value === "character" ? characterPendingCountQuery.data?.total || 0 : 0), })); const audioPickerItems = useMemo(() => { const localItems = personalAssets.flatMap((asset) => { @@ -282,10 +299,16 @@ export default function ProjectAssetsView({ detail, refreshProject }: ProjectDet const confirmMutation = useMutation({ mutationFn: ({ candidateId, targetAssetId }: { candidateId: string; targetAssetId?: string }) => confirmProjectAssetCandidate(detail.project.id, candidateId, targetAssetId), onSuccess: ({ asset }, variables) => { - syncPersonalCharacterProjection(asset); - done(variables.targetAssetId ? "候选信息已归并到角色新版本" : "角色卡已创建"); + if (asset.category === "character") syncPersonalCharacterProjection(asset); + const label = asset.category === "character" ? "角色卡" : categoryLabel(asset.category); + // 确认会改变候选、资产列表和左侧计数三处数据，必须一并失效缓存，否则计数与列表滞后。 + void Promise.all([ + queryClient.invalidateQueries({ queryKey: ["project", detail.project.id, "asset-candidates"] }), + queryClient.invalidateQueries({ queryKey: ["project", detail.project.id, "assets"] }), + ]); + done(variables.targetAssetId ? `候选信息已归并到${label}新版本` : `${label}已创建`); }, - onError: failed("角色确认失败"), + onError: failed("资产确认失败"), }); const confirmingCandidateId = confirmMutation.isPending ? confirmMutation.variables?.candidateId || "" : ""; const saveCharacter = useMutation({ @@ -407,24 +430,25 @@ export default function ProjectAssetsView({ detail, refreshProject }: ProjectDet
{childFolders.length ?
{childFolders.map((folder) => selectFolder(folder.id)} onRename={() => openFolderEditor(folder)} onMove={(parentId) => moveFolderMutation.mutate({ id: folder.id, parentId })} onStyle={(style) => styleFolderMutation.mutate({ id: folder.id, style })} onTheme={(theme) => themeFolderMutation.mutate({ id: folder.id, theme })} onDelete={() => modal.confirm({ title: `删除文件夹“${folder.name}”？`, content: "仅空文件夹可以删除，素材和子文件夹不会被级联删除。", okText: "删除", okButtonProps: { danger: true }, cancelText: "取消", onOk: () => deleteFolderMutation.mutateAsync(folder.id) })} deleting={(deleteFolderMutation.isPending && deleteFolderMutation.variables === folder.id) || (moveFolderMutation.isPending && moveFolderMutation.variables?.id === folder.id) || (styleFolderMutation.isPending && styleFolderMutation.variables?.id === folder.id) || (themeFolderMutation.isPending && themeFolderMutation.variables?.id === folder.id)} />)}
: null} - {folderId === ALL_FOLDERS && (category === "all" || category === "character") && pendingCandidates.length ? ( -
+ {showPendingCandidates && pendingCandidates.length ? ( +
-
剧情识别出的角色
+
剧情识别出的{candidateLabel}
剩余 {candidatesQuery.data?.total || 0} 个待确认
{pendingCandidates.map((candidate) => { const confirming = confirmingCandidateId === candidate.id; + const isCharacter = candidate.category === "character"; return (
- + {isCharacter ? : }
{candidate.name}
-
待确认角色卡 · 来自章节分析
+
待确认{categoryLabel(candidate.category)} · 来自章节分析
- - {characterAssets.length ? ({ key: asset.id, label: asset.title })), onClick: ({ key }) => confirmMutation.mutate({ candidateId: candidate.id, targetAssetId: key }) }}> : null} + + {isCharacter && characterAssets.length ? ({ key: asset.id, label: asset.title })), onClick: ({ key }) => confirmMutation.mutate({ candidateId: candidate.id, targetAssetId: key }) }}> : null}