33 lines
2.2 KiB
TypeScript
33 lines
2.2 KiB
TypeScript
import type { AssetLibraryPickerItem } from "@/components/assets/asset-library-picker-modal";
|
|
import type { CreationMode } from "@/lib/creation-mode";
|
|
import type { CreationReferenceLimits } from "./creation-references";
|
|
import { creationAttachmentFromAsset, creationAttachmentFromAudioAsset, creationAttachmentFromExternalAsset, creationAttachmentFromVideoAsset, type CreationAttachment } from "./creation-assets";
|
|
|
|
export function creationLibraryDisabledReason(mode: CreationMode, kind: string | undefined, videoLimits?: CreationReferenceLimits) {
|
|
if (kind !== "image" && kind !== "video" && kind !== "audio") return "此类型素材暂不支持作为创作参考";
|
|
if (mode === "image") return kind === "image" ? undefined : "图片创作仅支持参考图";
|
|
if (mode !== "video") return undefined;
|
|
|
|
const maximum = kind === "image" ? videoLimits?.maxImages : kind === "video" ? videoLimits?.maxVideos : videoLimits?.maxAudios;
|
|
if ((maximum || 0) > 0) return undefined;
|
|
const label = kind === "video" ? "视频" : kind === "audio" ? "音频" : "图片";
|
|
return `当前视频模型不支持参考${label}`;
|
|
}
|
|
|
|
export function creationLibrarySelectionToAttachments(ids: string[], items: AssetLibraryPickerItem[], mode: CreationMode, videoLimits?: CreationReferenceLimits): CreationAttachment[] {
|
|
const itemsById = new Map(items.map((item) => [item.id, item]));
|
|
return ids.map((id) => {
|
|
const item = itemsById.get(id);
|
|
if (!item) throw new Error("所选素材已不存在，请重新选择");
|
|
const kind = item.external?.item.kind || item.asset?.kind;
|
|
const disabledReason = item.disabledReason || creationLibraryDisabledReason(mode, kind, videoLimits);
|
|
if (disabledReason) throw new Error(`“${item.title}”：${disabledReason}`);
|
|
if (item.external) return creationAttachmentFromExternalAsset(item.external);
|
|
const asset = item.asset;
|
|
if (asset?.kind === "image") return creationAttachmentFromAsset(asset);
|
|
if (asset?.kind === "video") return creationAttachmentFromVideoAsset(asset);
|
|
if (asset?.kind === "audio") return creationAttachmentFromAudioAsset(asset);
|
|
throw new Error("所选素材已不存在，请重新选择");
|
|
});
|
|
}
|