|
|
|
|
@@ -1,22 +1,280 @@
|
|
|
|
|
import { nanoid } from "nanoid";
|
|
|
|
|
|
|
|
|
|
import { addEagleItem, createEagleFolder, downloadEagleItem, eagleItemFileUrl, eagleItemThumbnailUrl, getEagleLibrary, listEagleItems, type EagleFolder, type EagleItem } from "@/services/api/eagle";
|
|
|
|
|
import { getMediaBlob, uploadMediaFile } from "@/services/file-storage";
|
|
|
|
|
import { imageToDataUrl, uploadImage } from "@/services/image-storage";
|
|
|
|
|
import type { Asset } from "@/stores/use-asset-store";
|
|
|
|
|
import { registerPlugin } from "../plugin-registry";
|
|
|
|
|
import type { RegisteredPlugin } from "../plugin-types";
|
|
|
|
|
import type { AssetSourceProvider, ExternalAssetItem, PluginHostContext, RegisteredPlugin } from "../plugin-types";
|
|
|
|
|
|
|
|
|
|
export const EAGLE_PLUGIN_ID = "eagle-asset-connector";
|
|
|
|
|
export const EAGLE_DEFAULT_BASE_URL = "http://127.0.0.1:41595";
|
|
|
|
|
|
|
|
|
|
export const eagleAssetPlugin: RegisteredPlugin = {
|
|
|
|
|
manifest: {
|
|
|
|
|
id: EAGLE_PLUGIN_ID,
|
|
|
|
|
name: "Eagle 素材库",
|
|
|
|
|
version: "0.1.0",
|
|
|
|
|
version: "0.3.0",
|
|
|
|
|
apiVersion: "1",
|
|
|
|
|
category: "asset-source",
|
|
|
|
|
description: "把 Eagle 作为影策素材管理中的外部素材来源。",
|
|
|
|
|
description: "把 Eagle 作为影策的外部素材来源,直接浏览原始文件夹并读写 Eagle 文件。",
|
|
|
|
|
author: "影策社区",
|
|
|
|
|
surfaces: ["asset-source", "hybrid"],
|
|
|
|
|
permissions: ["asset.read", "asset.search", "asset.import", "asset.upload", "external.open"],
|
|
|
|
|
surfaces: ["asset-source"],
|
|
|
|
|
permissions: ["asset.read", "asset.search", "asset.upload", "external.open"],
|
|
|
|
|
trusted: true,
|
|
|
|
|
},
|
|
|
|
|
createAssetSource: ({ config }: PluginHostContext) => {
|
|
|
|
|
const configuredBaseUrl = config.baseUrl;
|
|
|
|
|
return createEagleAssetSource(typeof configuredBaseUrl === "string" && configuredBaseUrl.trim() ? configuredBaseUrl.trim() : undefined);
|
|
|
|
|
},
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
registerPlugin(eagleAssetPlugin);
|
|
|
|
|
export function createEagleAssetSource(baseUrl = EAGLE_DEFAULT_BASE_URL): AssetSourceProvider {
|
|
|
|
|
let folderCache: EagleFolder[] | null = null;
|
|
|
|
|
|
|
|
|
|
const getFolders = async (signal?: AbortSignal) => {
|
|
|
|
|
if (signal?.aborted) throw new DOMException("请求已取消", "AbortError");
|
|
|
|
|
if (!folderCache) folderCache = (await getEagleLibrary(baseUrl)).library.folders;
|
|
|
|
|
return folderCache;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
return {
|
|
|
|
|
listFolders: async (signal) => (await getFolders(signal)).map((folder) => ({ id: folder.id, name: folder.name, parentId: folder.parentId })),
|
|
|
|
|
list: async (query) => {
|
|
|
|
|
const [result, folders] = await Promise.all([listEagleItems({ baseUrl, folderId: query.folderId, keyword: query.keyword, limit: query.limit, offset: query.offset }), getFolders(query.signal)]);
|
|
|
|
|
const pathMap = buildFolderPathMap(folders);
|
|
|
|
|
return result.items.map((item) => toExternalItem(item, baseUrl, pathMap));
|
|
|
|
|
},
|
|
|
|
|
importAsset: (item, signal) => importEagleAsset(item, baseUrl, signal),
|
|
|
|
|
uploadAsset: (asset, signal) => uploadEagleAsset(asset, baseUrl, undefined, signal),
|
|
|
|
|
uploadAssetToFolder: (asset, folderId, signal) => uploadEagleAsset(asset, baseUrl, folderId, signal),
|
|
|
|
|
uploadFile: (file, folderId, signal) => uploadEagleFile(file, baseUrl, folderId, signal),
|
|
|
|
|
createFolder: async (name, parentId) => {
|
|
|
|
|
await createEagleFolder(baseUrl, { name, parentId });
|
|
|
|
|
folderCache = null;
|
|
|
|
|
},
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async function importEagleAsset(item: ExternalAssetItem, baseUrl: string, signal?: AbortSignal): Promise<Asset> {
|
|
|
|
|
const kind = item.kind;
|
|
|
|
|
if (kind === "text" || kind === "entity") throw new Error("当前 Eagle 导入暂不支持文本或实体素材");
|
|
|
|
|
const blob = await downloadEagleItem(item.id, baseUrl, signal);
|
|
|
|
|
const now = new Date().toISOString();
|
|
|
|
|
const metadata = {
|
|
|
|
|
source: "eagle",
|
|
|
|
|
eagle: {
|
|
|
|
|
itemId: item.id,
|
|
|
|
|
baseUrl,
|
|
|
|
|
folderId: item.folderId,
|
|
|
|
|
folderIds: item.folderIds || [],
|
|
|
|
|
folderPath: item.folderPath || [],
|
|
|
|
|
extension: item.metadata?.extension,
|
|
|
|
|
},
|
|
|
|
|
};
|
|
|
|
|
if (kind === "image") {
|
|
|
|
|
const uploaded = await uploadImage(blob);
|
|
|
|
|
return {
|
|
|
|
|
id: nanoid(), kind, title: item.title, coverUrl: uploaded.url, tags: item.tags || [], source: "Eagle 素材库", createdAt: now, updatedAt: now, metadata,
|
|
|
|
|
data: { dataUrl: uploaded.url, storageKey: uploaded.storageKey, width: item.width || uploaded.width, height: item.height || uploaded.height, bytes: uploaded.bytes, mimeType: uploaded.mimeType },
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
const uploaded = await uploadMediaFile(blob, `eagle-${kind}`);
|
|
|
|
|
if (kind === "video") {
|
|
|
|
|
return {
|
|
|
|
|
id: nanoid(), kind, title: item.title, coverUrl: "", tags: item.tags || [], source: "Eagle 素材库", createdAt: now, updatedAt: now, metadata,
|
|
|
|
|
data: { url: uploaded.url, storageKey: uploaded.storageKey, width: item.width || uploaded.width || 1280, height: item.height || uploaded.height || 720, bytes: uploaded.bytes, mimeType: uploaded.mimeType },
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
if (kind === "audio") {
|
|
|
|
|
return {
|
|
|
|
|
id: nanoid(), kind, title: item.title, coverUrl: "", tags: item.tags || [], source: "Eagle 素材库", createdAt: now, updatedAt: now, metadata,
|
|
|
|
|
data: { url: uploaded.url, storageKey: uploaded.storageKey, durationMs: uploaded.durationMs, bytes: uploaded.bytes, mimeType: uploaded.mimeType },
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
return {
|
|
|
|
|
id: nanoid(), kind: "model", title: item.title, coverUrl: "", tags: item.tags || [], source: "Eagle 素材库", createdAt: now, updatedAt: now, metadata,
|
|
|
|
|
data: { url: uploaded.url, storageKey: uploaded.storageKey, bytes: uploaded.bytes, mimeType: uploaded.mimeType, fileName: item.title },
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
type EagleWritableAsset = Extract<Asset, { kind: "image" | "video" | "audio" }>;
|
|
|
|
|
|
|
|
|
|
async function uploadEagleAsset(asset: Asset, baseUrl: string, folderId?: string, signal?: AbortSignal): Promise<ExternalAssetItem> {
|
|
|
|
|
if (!isEagleWritableAsset(asset)) throw new Error("当前仅支持将图片、视频或音频写回 Eagle");
|
|
|
|
|
const dataUrl = await assetToDataUrl(asset, signal);
|
|
|
|
|
const result = await addEagleItem(baseUrl, {
|
|
|
|
|
url: dataUrl,
|
|
|
|
|
name: eagleAssetName(asset),
|
|
|
|
|
folderId,
|
|
|
|
|
tags: asset.tags,
|
|
|
|
|
annotation: asset.note,
|
|
|
|
|
website: typeof asset.metadata?.eagle === "object" && asset.metadata.eagle && "url" in asset.metadata.eagle ? String(asset.metadata.eagle.url || "") : undefined,
|
|
|
|
|
});
|
|
|
|
|
return { id: result.item.id || "eagle:" + asset.id, title: eagleAssetName(asset), kind: asset.kind, tags: asset.tags, mimeType: asset.data.mimeType };
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function isEagleWritableAsset(asset: Asset): asset is EagleWritableAsset {
|
|
|
|
|
return asset.kind === "image" || asset.kind === "video" || asset.kind === "audio";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async function assetToDataUrl(asset: EagleWritableAsset, signal?: AbortSignal) {
|
|
|
|
|
if (asset.kind === "image") {
|
|
|
|
|
return imageToDataUrl({ dataUrl: asset.data.dataUrl, storageKey: asset.data.storageKey, url: asset.data.dataUrl, name: asset.title, mimeType: asset.data.mimeType });
|
|
|
|
|
}
|
|
|
|
|
const stored = asset.data.storageKey ? await getMediaBlob(asset.data.storageKey) : null;
|
|
|
|
|
if (stored) return fileToDataUrl(withMimeType(stored, asset.data.mimeType), signal);
|
|
|
|
|
const url = asset.data.url;
|
|
|
|
|
if (!url) throw new Error("生成结果没有可写入 Eagle 的媒体地址");
|
|
|
|
|
if (url.startsWith("data:")) return url;
|
|
|
|
|
const response = await fetch(url, { credentials: "include", signal });
|
|
|
|
|
if (!response.ok) throw new Error("读取生成媒体失败,无法写入 Eagle");
|
|
|
|
|
return fileToDataUrl(withMimeType(await response.blob(), asset.data.mimeType), signal);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function eagleAssetName(asset: EagleWritableAsset) {
|
|
|
|
|
const title = asset.title.trim() || (asset.kind === "image" ? "生成图片" : asset.kind === "video" ? "生成视频" : "生成音频");
|
|
|
|
|
if (/\.[a-z0-9]{2,8}$/i.test(title)) return title;
|
|
|
|
|
const extension = asset.kind === "image" ? extensionFromMime(asset.data.mimeType, "png") : asset.kind === "video" ? extensionFromMime(asset.data.mimeType, "mp4") : extensionFromMime(asset.data.mimeType, "mp3");
|
|
|
|
|
return title + "." + extension;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function extensionFromMime(mimeType: string, fallback: string) {
|
|
|
|
|
const value = mimeType.toLowerCase();
|
|
|
|
|
if (value.includes("png")) return "png";
|
|
|
|
|
if (value.includes("jpeg") || value.includes("jpg")) return "jpg";
|
|
|
|
|
if (value.includes("webm")) return "webm";
|
|
|
|
|
if (value.includes("quicktime")) return "mov";
|
|
|
|
|
if (value.includes("wav")) return "wav";
|
|
|
|
|
if (value.includes("ogg")) return "ogg";
|
|
|
|
|
if (value.includes("mp4")) return "mp4";
|
|
|
|
|
if (value.includes("mpeg") || value.includes("mp3")) return "mp3";
|
|
|
|
|
return fallback;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function withMimeType(blob: Blob, mimeType: string) {
|
|
|
|
|
return mimeType && blob.type !== mimeType ? blob.slice(0, blob.size, mimeType) : blob;
|
|
|
|
|
}
|
|
|
|
|
async function uploadEagleFile(file: File, baseUrl: string, folderId?: string, signal?: AbortSignal): Promise<ExternalAssetItem> {
|
|
|
|
|
if (file.size > 96 * 1024 * 1024) throw new Error("单个文件不能超过 96 MB");
|
|
|
|
|
if (!isEagleMediaType(file.type)) throw new Error("Eagle 写入目前支持图片、视频和音频文件");
|
|
|
|
|
const dataUrl = await fileToDataUrl(file, signal);
|
|
|
|
|
const result = await addEagleItem(baseUrl, {
|
|
|
|
|
url: dataUrl,
|
|
|
|
|
name: file.name,
|
|
|
|
|
folderId,
|
|
|
|
|
});
|
|
|
|
|
const itemId = result.item.id || "eagle:" + nanoid();
|
|
|
|
|
const extension = file.name.includes(".") ? file.name.split(".").pop() || "" : "";
|
|
|
|
|
return {
|
|
|
|
|
id: itemId,
|
|
|
|
|
title: file.name,
|
|
|
|
|
kind: kindFromExtension(extension),
|
|
|
|
|
thumbnailUrl: result.item.id ? eagleItemThumbnailUrl(result.item.id, baseUrl) : undefined,
|
|
|
|
|
fileUrl: result.item.id ? eagleItemFileUrl(result.item.id, baseUrl) : undefined,
|
|
|
|
|
mimeType: file.type || mimeTypeFromExtension(extension),
|
|
|
|
|
bytes: file.size,
|
|
|
|
|
folderId,
|
|
|
|
|
folderIds: folderId ? [folderId] : [],
|
|
|
|
|
metadata: { extension },
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function isEagleMediaType(mimeType: string) {
|
|
|
|
|
const value = mimeType.toLowerCase();
|
|
|
|
|
return value.startsWith("image/") || value.startsWith("video/") || value.startsWith("audio/");
|
|
|
|
|
}
|
|
|
|
|
function fileToDataUrl(file: Blob, signal?: AbortSignal): Promise<string> {
|
|
|
|
|
return new Promise((resolve, reject) => {
|
|
|
|
|
const reader = new FileReader();
|
|
|
|
|
const cleanup = () => signal?.removeEventListener("abort", abort);
|
|
|
|
|
const abort = () => {
|
|
|
|
|
reader.abort();
|
|
|
|
|
cleanup();
|
|
|
|
|
reject(new DOMException("请求已取消", "AbortError"));
|
|
|
|
|
};
|
|
|
|
|
reader.onload = () => {
|
|
|
|
|
cleanup();
|
|
|
|
|
if (typeof reader.result !== "string") {
|
|
|
|
|
reject(new Error("无法读取待写入的文件"));
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
resolve(reader.result);
|
|
|
|
|
};
|
|
|
|
|
reader.onerror = () => {
|
|
|
|
|
cleanup();
|
|
|
|
|
reject(new Error("无法读取待写入的文件"));
|
|
|
|
|
};
|
|
|
|
|
if (signal?.aborted) {
|
|
|
|
|
abort();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
signal?.addEventListener("abort", abort, { once: true });
|
|
|
|
|
reader.readAsDataURL(file);
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function toExternalItem(item: EagleItem, baseUrl: string, pathMap: Map<string, string[]>) {
|
|
|
|
|
const kind = kindFromExtension(item.extension);
|
|
|
|
|
return {
|
|
|
|
|
id: item.id,
|
|
|
|
|
title: item.name,
|
|
|
|
|
kind,
|
|
|
|
|
thumbnailUrl: eagleItemThumbnailUrl(item.id, baseUrl),
|
|
|
|
|
fileUrl: eagleItemFileUrl(item.id, baseUrl),
|
|
|
|
|
mimeType: mimeTypeFromExtension(item.extension),
|
|
|
|
|
width: item.width,
|
|
|
|
|
height: item.height,
|
|
|
|
|
bytes: item.size,
|
|
|
|
|
tags: item.tags,
|
|
|
|
|
folderId: item.folderIds[0],
|
|
|
|
|
folderIds: item.folderIds,
|
|
|
|
|
folderPath: item.folderIds[0] ? pathMap.get(item.folderIds[0]) : [],
|
|
|
|
|
description: item.annotation,
|
|
|
|
|
metadata: { extension: item.extension, modificationTime: item.modificationTime, url: item.url, deleted: item.deleted },
|
|
|
|
|
} satisfies ExternalAssetItem;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function buildFolderPathMap(folders: EagleFolder[]) {
|
|
|
|
|
const byId = new Map(folders.map((folder) => [folder.id, folder]));
|
|
|
|
|
const result = new Map<string, string[]>();
|
|
|
|
|
for (const folder of folders) {
|
|
|
|
|
const path: string[] = [];
|
|
|
|
|
const seen = new Set<string>();
|
|
|
|
|
let current: EagleFolder | undefined = folder;
|
|
|
|
|
while (current && !seen.has(current.id)) {
|
|
|
|
|
seen.add(current.id);
|
|
|
|
|
path.unshift(current.name);
|
|
|
|
|
current = current.parentId ? byId.get(current.parentId) : undefined;
|
|
|
|
|
}
|
|
|
|
|
result.set(folder.id, path);
|
|
|
|
|
}
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function kindFromExtension(extension: string): Asset["kind"] {
|
|
|
|
|
const value = extension.toLowerCase();
|
|
|
|
|
if (["png", "jpg", "jpeg", "gif", "webp", "bmp", "avif", "svg"].includes(value)) return "image";
|
|
|
|
|
if (["mp4", "mov", "webm", "mkv", "avi"].includes(value)) return "video";
|
|
|
|
|
if (["mp3", "wav", "m4a", "aac", "flac", "ogg"].includes(value)) return "audio";
|
|
|
|
|
if (["glb", "gltf", "obj", "fbx", "usdz", "blend"].includes(value)) return "model";
|
|
|
|
|
return "model";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function mimeTypeFromExtension(extension: string) {
|
|
|
|
|
const value = extension.toLowerCase();
|
|
|
|
|
if (["jpg", "jpeg"].includes(value)) return "image/jpeg";
|
|
|
|
|
if (value === "png") return "image/png";
|
|
|
|
|
if (value === "webp") return "image/webp";
|
|
|
|
|
if (value === "gif") return "image/gif";
|
|
|
|
|
if (value === "svg") return "image/svg+xml";
|
|
|
|
|
if (value === "mp4") return "video/mp4";
|
|
|
|
|
if (value === "webm") return "video/webm";
|
|
|
|
|
if (value === "mp3") return "audio/mpeg";
|
|
|
|
|
if (value === "wav") return "audio/wav";
|
|
|
|
|
if (value === "glb") return "model/gltf-binary";
|
|
|
|
|
return "application/octet-stream";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
registerPlugin(eagleAssetPlugin);
|
|
|
|
|
|