Initial commit: OmniAI Web Frontend
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,138 @@
|
||||
import {
|
||||
CloseOutlined,
|
||||
DeleteOutlined,
|
||||
EditOutlined,
|
||||
MenuFoldOutlined,
|
||||
MenuUnfoldOutlined,
|
||||
MessageOutlined,
|
||||
PlusOutlined,
|
||||
} from "@ant-design/icons";
|
||||
import { useCallback, useState } from "react";
|
||||
import type { ConversationSummary } from "../../api/conversationClient";
|
||||
|
||||
interface ConversationSidebarProps {
|
||||
conversations: ConversationSummary[];
|
||||
activeId: number | null;
|
||||
collapsed: boolean;
|
||||
onToggle: () => void;
|
||||
onSelect: (id: number) => void;
|
||||
onNew: () => void;
|
||||
onDelete: (id: number) => void;
|
||||
onRename: (id: number, title: string) => void;
|
||||
}
|
||||
|
||||
function formatRelativeTime(dateStr: string): string {
|
||||
const now = Date.now();
|
||||
const then = new Date(dateStr).getTime();
|
||||
const diff = now - then;
|
||||
if (diff < 60_000) return "刚刚";
|
||||
if (diff < 3_600_000) return `${Math.floor(diff / 60_000)} 分钟前`;
|
||||
if (diff < 86_400_000) return `${Math.floor(diff / 3_600_000)} 小时前`;
|
||||
if (diff < 604_800_000) return `${Math.floor(diff / 86_400_000)} 天前`;
|
||||
return new Date(dateStr).toLocaleDateString("zh-CN");
|
||||
}
|
||||
|
||||
export default function ConversationSidebar({
|
||||
conversations,
|
||||
activeId,
|
||||
collapsed,
|
||||
onToggle,
|
||||
onSelect,
|
||||
onNew,
|
||||
onDelete,
|
||||
onRename,
|
||||
}: ConversationSidebarProps) {
|
||||
const [editingId, setEditingId] = useState<number | null>(null);
|
||||
const [editValue, setEditValue] = useState("");
|
||||
|
||||
const startRename = useCallback((id: number, currentTitle: string) => {
|
||||
setEditingId(id);
|
||||
setEditValue(currentTitle);
|
||||
}, []);
|
||||
|
||||
const commitRename = useCallback(() => {
|
||||
if (editingId !== null && editValue.trim()) {
|
||||
onRename(editingId, editValue.trim());
|
||||
}
|
||||
setEditingId(null);
|
||||
}, [editingId, editValue, onRename]);
|
||||
|
||||
return (
|
||||
<aside className={`conversation-sidebar${collapsed ? " is-collapsed" : ""}`}>
|
||||
<div className="conversation-sidebar__header">
|
||||
<button
|
||||
type="button"
|
||||
className="conversation-sidebar__toggle"
|
||||
onClick={onToggle}
|
||||
aria-label={collapsed ? "展开侧边栏" : "收起侧边栏"}
|
||||
>
|
||||
{collapsed ? <MenuUnfoldOutlined /> : <MenuFoldOutlined />}
|
||||
</button>
|
||||
{!collapsed && (
|
||||
<button type="button" className="conversation-sidebar__new" onClick={onNew}>
|
||||
<PlusOutlined /> 新对话
|
||||
</button>
|
||||
)}
|
||||
</div>
|
||||
|
||||
{!collapsed && (
|
||||
<div className="conversation-sidebar__list">
|
||||
{conversations.length === 0 ? (
|
||||
<div className="conversation-sidebar__empty">
|
||||
<MessageOutlined />
|
||||
<span>暂无对话记录</span>
|
||||
</div>
|
||||
) : (
|
||||
conversations.map((conv) => (
|
||||
<div
|
||||
key={conv.id}
|
||||
className={`conversation-sidebar__item${conv.id === activeId ? " is-active" : ""}`}
|
||||
>
|
||||
{editingId === conv.id ? (
|
||||
<input
|
||||
className="conversation-sidebar__rename-input"
|
||||
value={editValue}
|
||||
onChange={(e) => setEditValue(e.target.value)}
|
||||
onBlur={commitRename}
|
||||
onKeyDown={(e) => {
|
||||
if (e.key === "Enter") commitRename();
|
||||
if (e.key === "Escape") setEditingId(null);
|
||||
}}
|
||||
autoFocus
|
||||
/>
|
||||
) : (
|
||||
<button
|
||||
type="button"
|
||||
className="conversation-sidebar__item-main"
|
||||
onClick={() => onSelect(conv.id)}
|
||||
>
|
||||
<span className="conversation-sidebar__item-title">{conv.title}</span>
|
||||
<span className="conversation-sidebar__item-time">
|
||||
{formatRelativeTime(conv.updatedAt)}
|
||||
</span>
|
||||
</button>
|
||||
)}
|
||||
<div className="conversation-sidebar__item-actions">
|
||||
<button
|
||||
type="button"
|
||||
aria-label="重命名"
|
||||
onClick={() => startRename(conv.id, conv.title)}
|
||||
>
|
||||
<EditOutlined />
|
||||
</button>
|
||||
<button
|
||||
type="button"
|
||||
aria-label="删除"
|
||||
onClick={() => onDelete(conv.id)}
|
||||
>
|
||||
<DeleteOutlined />
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
))
|
||||
)}
|
||||
</div>
|
||||
)}
|
||||
</aside>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,151 @@
|
||||
import {
|
||||
CloudSyncOutlined,
|
||||
DeleteOutlined,
|
||||
EditOutlined,
|
||||
FolderOpenOutlined,
|
||||
MenuFoldOutlined,
|
||||
MenuUnfoldOutlined,
|
||||
ReloadOutlined,
|
||||
} from "@ant-design/icons";
|
||||
import { useCallback, useMemo, useState } from "react";
|
||||
import type { WebProjectSummary } from "../../types";
|
||||
|
||||
interface ProjectSidebarProps {
|
||||
projects: WebProjectSummary[];
|
||||
activeId: string | null;
|
||||
collapsed: boolean;
|
||||
filterMode?: "chat" | "image" | "video";
|
||||
loading?: boolean;
|
||||
error?: string | null;
|
||||
onToggle: () => void;
|
||||
onSelect: (id: string) => void;
|
||||
onRefresh: () => void;
|
||||
onRename: (id: string, title: string) => void;
|
||||
onDelete: (id: string) => void;
|
||||
}
|
||||
|
||||
function formatRelativeTime(dateStr: string): string {
|
||||
const then = new Date(dateStr).getTime();
|
||||
if (!Number.isFinite(then)) return "";
|
||||
const diff = Date.now() - then;
|
||||
if (diff < 60_000) return "just now";
|
||||
if (diff < 3_600_000) return `${Math.floor(diff / 60_000)} min ago`;
|
||||
if (diff < 86_400_000) return `${Math.floor(diff / 3_600_000)} h ago`;
|
||||
if (diff < 604_800_000) return `${Math.floor(diff / 86_400_000)} d ago`;
|
||||
return new Date(dateStr).toLocaleDateString("zh-CN");
|
||||
}
|
||||
|
||||
export default function ProjectSidebar({
|
||||
projects,
|
||||
activeId,
|
||||
collapsed,
|
||||
filterMode,
|
||||
loading,
|
||||
error,
|
||||
onToggle,
|
||||
onSelect,
|
||||
onRefresh,
|
||||
onRename,
|
||||
onDelete,
|
||||
}: ProjectSidebarProps) {
|
||||
const [editingId, setEditingId] = useState<string | null>(null);
|
||||
const [editValue, setEditValue] = useState("");
|
||||
|
||||
const filteredProjects = useMemo(
|
||||
() => filterMode ? projects.filter((p) => p.mode === filterMode) : projects,
|
||||
[projects, filterMode],
|
||||
);
|
||||
|
||||
const startRename = useCallback((project: WebProjectSummary) => {
|
||||
setEditingId(project.id);
|
||||
setEditValue(project.name);
|
||||
}, []);
|
||||
|
||||
const commitRename = useCallback(() => {
|
||||
if (editingId && editValue.trim()) {
|
||||
onRename(editingId, editValue.trim());
|
||||
}
|
||||
setEditingId(null);
|
||||
}, [editValue, editingId, onRename]);
|
||||
|
||||
return (
|
||||
<aside className={`conversation-sidebar${collapsed ? " is-collapsed" : ""}`}>
|
||||
<div className="conversation-sidebar__header">
|
||||
<button
|
||||
type="button"
|
||||
className="conversation-sidebar__toggle"
|
||||
onClick={onToggle}
|
||||
aria-label={collapsed ? "Expand project sidebar" : "Collapse project sidebar"}
|
||||
>
|
||||
{collapsed ? <MenuUnfoldOutlined /> : <MenuFoldOutlined />}
|
||||
</button>
|
||||
{!collapsed && (
|
||||
<div className="conversation-sidebar__header-actions">
|
||||
<button type="button" className="conversation-sidebar__new" onClick={() => onSelect("")}>
|
||||
<FolderOpenOutlined /> 新对话
|
||||
</button>
|
||||
<button type="button" className="conversation-sidebar__icon-button" onClick={onRefresh} aria-label="刷新记录">
|
||||
<ReloadOutlined />
|
||||
</button>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
|
||||
{!collapsed && (
|
||||
<div className="conversation-sidebar__list">
|
||||
{error ? (
|
||||
<div className="conversation-sidebar__empty">
|
||||
<CloudSyncOutlined />
|
||||
<span>Server data failed to load</span>
|
||||
</div>
|
||||
) : filteredProjects.length === 0 ? (
|
||||
<div className="conversation-sidebar__empty">
|
||||
<FolderOpenOutlined />
|
||||
<span>{loading ? "正在加载记录..." : "暂无该类型记录"}</span>
|
||||
</div>
|
||||
) : (
|
||||
filteredProjects.map((project) => (
|
||||
<div
|
||||
key={project.id}
|
||||
className={`conversation-sidebar__item${project.id === activeId ? " is-active" : ""}`}
|
||||
>
|
||||
{editingId === project.id ? (
|
||||
<input
|
||||
className="conversation-sidebar__rename-input"
|
||||
value={editValue}
|
||||
onChange={(event) => setEditValue(event.target.value)}
|
||||
onBlur={commitRename}
|
||||
onKeyDown={(event) => {
|
||||
if (event.key === "Enter") commitRename();
|
||||
if (event.key === "Escape") setEditingId(null);
|
||||
}}
|
||||
autoFocus
|
||||
/>
|
||||
) : (
|
||||
<button
|
||||
type="button"
|
||||
className="conversation-sidebar__item-main"
|
||||
onClick={() => onSelect(project.id)}
|
||||
>
|
||||
<span className="conversation-sidebar__item-title">{project.name}</span>
|
||||
<span className="conversation-sidebar__item-time">
|
||||
{project.source === "server" ? formatRelativeTime(project.updatedAt) : "preview"}
|
||||
</span>
|
||||
</button>
|
||||
)}
|
||||
<div className="conversation-sidebar__item-actions">
|
||||
<button type="button" aria-label="Rename project" onClick={() => startRename(project)}>
|
||||
<EditOutlined />
|
||||
</button>
|
||||
<button type="button" aria-label="Delete project" onClick={() => onDelete(project.id)}>
|
||||
<DeleteOutlined />
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
))
|
||||
)}
|
||||
</div>
|
||||
)}
|
||||
</aside>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,27 @@
|
||||
import { useSmoothedProgress } from "../../hooks/useSmoothedProgress";
|
||||
|
||||
type MessageStatus = "thinking" | "completed" | "failed" | string;
|
||||
|
||||
interface SmoothedProgressBarProps {
|
||||
progress: number;
|
||||
status: MessageStatus;
|
||||
label?: string;
|
||||
}
|
||||
|
||||
function mapMessageStatus(status: MessageStatus) {
|
||||
if (status === "thinking") return "running" as const;
|
||||
if (status === "completed") return "completed" as const;
|
||||
if (status === "failed") return "failed" as const;
|
||||
return "running" as const;
|
||||
}
|
||||
|
||||
export function SmoothedProgressBar({ progress, status, label }: SmoothedProgressBarProps) {
|
||||
const smoothed = useSmoothedProgress(progress, mapMessageStatus(status));
|
||||
return (
|
||||
<>
|
||||
<span>{label || "超分处理中..."}</span>
|
||||
<strong>{smoothed}%</strong>
|
||||
<i style={{ width: `${smoothed}%` }} />
|
||||
</>
|
||||
);
|
||||
}
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,611 @@
|
||||
import { memo, useCallback, useEffect, useRef, useState, type CSSProperties, type ChangeEvent } from "react";
|
||||
import {
|
||||
AppstoreOutlined,
|
||||
CaretRightOutlined,
|
||||
ClockCircleOutlined,
|
||||
CopyOutlined,
|
||||
DownloadOutlined,
|
||||
FileTextOutlined,
|
||||
FullscreenOutlined,
|
||||
MutedOutlined,
|
||||
PauseOutlined,
|
||||
PictureOutlined,
|
||||
ReloadOutlined,
|
||||
SoundOutlined,
|
||||
StopOutlined,
|
||||
ThunderboltOutlined,
|
||||
} from "@ant-design/icons";
|
||||
import { assetClient } from "../../../api/assetClient";
|
||||
import { communityClient } from "../../../api/communityClient";
|
||||
import { saveAssetToLocalLibrary } from "../../assets/localAssetStore";
|
||||
import { SmoothedProgressBar } from "../SmoothedProgressBar";
|
||||
import { useSmoothedProgress } from "../../../hooks/useSmoothedProgress";
|
||||
import { renderMarkdownBlocks } from "../markdownRenderer";
|
||||
import { downloadResultAsset } from "../workbenchDownload";
|
||||
import type { WorkbenchChatAttachment, WorkbenchChatMessage, WorkbenchResultActionPayload } from "../workbenchChatTypes";
|
||||
|
||||
// ─── Helpers ──────────────────────────────────────────────────────────────────
|
||||
|
||||
function cleanModelDisplayLabel(label: string) {
|
||||
const channelName = "百炼";
|
||||
return label.replace(new RegExp(`\\s*[（(]\\s*${channelName}\\s*[）)]`, "g"), "").trim();
|
||||
}
|
||||
|
||||
function formatMediaTime(value: number) {
|
||||
if (!Number.isFinite(value) || value <= 0) return "0:00";
|
||||
const minutes = Math.floor(value / 60);
|
||||
const seconds = Math.floor(value % 60);
|
||||
return `${minutes}:${String(seconds).padStart(2, "0")}`;
|
||||
}
|
||||
|
||||
// ─── MarkdownMessage ──────────────────────────────────────────────────────────
|
||||
|
||||
export const MarkdownMessage = memo(function MarkdownMessage({ text }: { text: string }) {
|
||||
const normalized = text.trim();
|
||||
if (!normalized) return null;
|
||||
return <div className="ai-chat-markdown">{renderMarkdownBlocks(normalized)}</div>;
|
||||
});
|
||||
|
||||
// ─── ImmersiveVideoPlayer ─────────────────────────────────────────────────────
|
||||
|
||||
export const ImmersiveVideoPlayer = memo(function ImmersiveVideoPlayer({
|
||||
src,
|
||||
title,
|
||||
}: {
|
||||
src: string;
|
||||
title: string;
|
||||
}) {
|
||||
const videoRef = useRef<HTMLVideoElement | null>(null);
|
||||
const frameRef = useRef<HTMLDivElement | null>(null);
|
||||
const [isPlaying, setIsPlaying] = useState(false);
|
||||
const [isMuted, setIsMuted] = useState(false);
|
||||
const [isFullscreen, setIsFullscreen] = useState(false);
|
||||
const [duration, setDuration] = useState(0);
|
||||
const [currentTime, setCurrentTime] = useState(0);
|
||||
const [volume, setVolume] = useState(1);
|
||||
|
||||
const syncVideoState = useCallback(() => {
|
||||
const video = videoRef.current;
|
||||
if (!video) return;
|
||||
setIsPlaying(!video.paused && !video.ended);
|
||||
setIsMuted(video.muted);
|
||||
setVolume(video.volume);
|
||||
setCurrentTime(video.currentTime || 0);
|
||||
setDuration(Number.isFinite(video.duration) ? video.duration : 0);
|
||||
}, []);
|
||||
|
||||
useEffect(() => {
|
||||
const video = videoRef.current;
|
||||
if (!video) return undefined;
|
||||
|
||||
video.volume = volume;
|
||||
syncVideoState();
|
||||
|
||||
const events: Array<keyof HTMLMediaElementEventMap> = [
|
||||
"loadedmetadata",
|
||||
"durationchange",
|
||||
"timeupdate",
|
||||
"play",
|
||||
"pause",
|
||||
"ended",
|
||||
"volumechange",
|
||||
];
|
||||
events.forEach((eventName) => video.addEventListener(eventName, syncVideoState));
|
||||
return () => events.forEach((eventName) => video.removeEventListener(eventName, syncVideoState));
|
||||
}, [src, syncVideoState, volume]);
|
||||
|
||||
useEffect(() => {
|
||||
const handleFullscreenChange = () => {
|
||||
setIsFullscreen(document.fullscreenElement === frameRef.current);
|
||||
};
|
||||
|
||||
document.addEventListener("fullscreenchange", handleFullscreenChange);
|
||||
return () => document.removeEventListener("fullscreenchange", handleFullscreenChange);
|
||||
}, []);
|
||||
|
||||
const togglePlayback = useCallback(() => {
|
||||
const video = videoRef.current;
|
||||
if (!video) return;
|
||||
if (video.paused || video.ended) {
|
||||
void video.play().catch(() => {});
|
||||
} else {
|
||||
video.pause();
|
||||
}
|
||||
}, []);
|
||||
|
||||
const handleSeek = (event: ChangeEvent<HTMLInputElement>) => {
|
||||
const video = videoRef.current;
|
||||
const nextTime = Number(event.currentTarget.value);
|
||||
if (!video || !Number.isFinite(nextTime)) return;
|
||||
video.currentTime = nextTime;
|
||||
setCurrentTime(nextTime);
|
||||
};
|
||||
|
||||
const handleVolumeChange = (event: ChangeEvent<HTMLInputElement>) => {
|
||||
const video = videoRef.current;
|
||||
const nextVolume = Number(event.currentTarget.value);
|
||||
if (!video || !Number.isFinite(nextVolume)) return;
|
||||
video.volume = nextVolume;
|
||||
video.muted = nextVolume === 0;
|
||||
setVolume(nextVolume);
|
||||
setIsMuted(video.muted);
|
||||
};
|
||||
|
||||
const toggleMute = useCallback(() => {
|
||||
const video = videoRef.current;
|
||||
if (!video) return;
|
||||
video.muted = !video.muted;
|
||||
setIsMuted(video.muted);
|
||||
}, []);
|
||||
|
||||
const toggleFullscreen = useCallback(() => {
|
||||
const frame = frameRef.current;
|
||||
if (!frame) return;
|
||||
if (document.fullscreenElement === frame) {
|
||||
void document.exitFullscreen?.();
|
||||
} else {
|
||||
void frame.requestFullscreen?.();
|
||||
}
|
||||
}, []);
|
||||
|
||||
const progress = duration > 0 ? Math.min(100, Math.max(0, (currentTime / duration) * 100)) : 0;
|
||||
const volumeProgress = Math.min(100, Math.max(0, volume * 100));
|
||||
|
||||
return (
|
||||
<div
|
||||
ref={frameRef}
|
||||
className={`ai-chat-media-player${isPlaying ? " is-playing" : ""}${isFullscreen ? " is-fullscreen" : ""}`}
|
||||
>
|
||||
<div className="ai-chat-media-player__stage" onClick={togglePlayback}>
|
||||
<video
|
||||
ref={videoRef}
|
||||
key={src}
|
||||
src={src}
|
||||
playsInline
|
||||
preload="metadata"
|
||||
aria-label={title}
|
||||
/>
|
||||
<span className="ai-chat-media-player__vignette" />
|
||||
<button
|
||||
type="button"
|
||||
className="ai-chat-media-player__center-play"
|
||||
onClick={(event) => {
|
||||
event.stopPropagation();
|
||||
togglePlayback();
|
||||
}}
|
||||
aria-label={isPlaying ? "Pause video" : "Play video"}
|
||||
>
|
||||
{isPlaying ? <PauseOutlined /> : <CaretRightOutlined />}
|
||||
</button>
|
||||
</div>
|
||||
<div className="ai-chat-media-player__controls" onClick={(event) => event.stopPropagation()}>
|
||||
<input
|
||||
className="ai-chat-media-player__seek"
|
||||
type="range"
|
||||
min="0"
|
||||
max={duration || 0}
|
||||
step="0.01"
|
||||
value={Math.min(currentTime, duration || currentTime)}
|
||||
onChange={handleSeek}
|
||||
disabled={duration <= 0}
|
||||
aria-label="Video progress"
|
||||
style={{ "--progress": `${progress}%` } as CSSProperties}
|
||||
/>
|
||||
<div className="ai-chat-media-player__bar">
|
||||
<button type="button" onClick={togglePlayback} aria-label={isPlaying ? "Pause video" : "Play video"}>
|
||||
{isPlaying ? <PauseOutlined /> : <CaretRightOutlined />}
|
||||
</button>
|
||||
<span className="ai-chat-media-player__time">
|
||||
{formatMediaTime(currentTime)} / {formatMediaTime(duration)}
|
||||
</span>
|
||||
<span className="ai-chat-media-player__spacer" />
|
||||
<button type="button" onClick={toggleMute} aria-label={isMuted ? "Unmute video" : "Mute video"}>
|
||||
{isMuted || volume === 0 ? <MutedOutlined /> : <SoundOutlined />}
|
||||
</button>
|
||||
<input
|
||||
className="ai-chat-media-player__volume"
|
||||
type="range"
|
||||
min="0"
|
||||
max="1"
|
||||
step="0.01"
|
||||
value={isMuted ? 0 : volume}
|
||||
onChange={handleVolumeChange}
|
||||
aria-label="Video volume"
|
||||
style={{ "--progress": `${isMuted ? 0 : volumeProgress}%` } as CSSProperties}
|
||||
/>
|
||||
<button type="button" onClick={toggleFullscreen} aria-label="Fullscreen video">
|
||||
<FullscreenOutlined />
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
});
|
||||
|
||||
// ─── ResultCard ───────────────────────────────────────────────────────────────
|
||||
|
||||
export const ResultCard = memo(function ResultCard({
|
||||
message,
|
||||
onRegenerate,
|
||||
onOpenMedia,
|
||||
onSuperResolveVideo,
|
||||
onSuperResolveImage,
|
||||
onOpenResultInCanvas,
|
||||
downloadFilenameBase,
|
||||
}: {
|
||||
message: WorkbenchChatMessage;
|
||||
onRegenerate: (message: WorkbenchChatMessage) => void;
|
||||
onOpenMedia: (item: WorkbenchChatAttachment) => void;
|
||||
onSuperResolveVideo: (message: WorkbenchChatMessage) => void;
|
||||
onSuperResolveImage: (message: WorkbenchChatMessage) => void;
|
||||
onOpenResultInCanvas?: (payload: WorkbenchResultActionPayload) => void;
|
||||
downloadFilenameBase?: string;
|
||||
}) {
|
||||
const [mediaRatio, setMediaRatio] = useState<string | null>(null);
|
||||
const [actionNotice, setActionNotice] = useState<string | null>(null);
|
||||
const [isPublishing, setIsPublishing] = useState(false);
|
||||
const [isDownloading, setIsDownloading] = useState(false);
|
||||
const [mediaExpired, setMediaExpired] = useState(false);
|
||||
const isVideo = message.resultType === "video";
|
||||
const hasMedia = !!message.resultUrl;
|
||||
const isSuperResolving =
|
||||
hasMedia &&
|
||||
message.status === "thinking" &&
|
||||
message.taskStatusLabel?.includes("超分");
|
||||
const promptText = message.prompt || (message.resultUrl ? "" : message.body);
|
||||
|
||||
const handleCopyPrompt = () => {
|
||||
if (promptText) {
|
||||
navigator.clipboard.writeText(promptText);
|
||||
}
|
||||
};
|
||||
|
||||
const specs = message.result?.specs || [];
|
||||
const modelTag = cleanModelDisplayLabel(specs[0] || "");
|
||||
const modeTag = specs.length > 1 ? specs[1] : "";
|
||||
const mediaStyle = mediaRatio ? ({ "--media-ratio": mediaRatio } as CSSProperties) : undefined;
|
||||
const durationTag = specs.find((spec) => /\b\d+s\b|秒/.test(spec)) || "5s";
|
||||
const openGeneratedMedia = () => {
|
||||
if (!message.resultUrl) return;
|
||||
onOpenMedia({
|
||||
kind: isVideo ? "video" : "image",
|
||||
name: message.result?.title || (isVideo ? "生成视频" : "生成图片"),
|
||||
token: message.taskId ? `#${message.taskId}` : "@结果",
|
||||
previewUrl: message.resultUrl,
|
||||
remoteUrl: message.resultUrl,
|
||||
});
|
||||
};
|
||||
const resultTitle = message.result?.title || (isVideo ? "生成视频" : "生成图片");
|
||||
|
||||
const handleDownload = async () => {
|
||||
if (!message.resultUrl || isDownloading) return;
|
||||
setIsDownloading(true);
|
||||
setActionNotice("正在准备下载...");
|
||||
try {
|
||||
const filenameBase = downloadFilenameBase || resultTitle;
|
||||
const status = await downloadResultAsset(message.resultUrl, filenameBase, isVideo, message.resultOssKey ? undefined : message.taskId);
|
||||
setActionNotice(status === "saved" ? "已保存到本地" : "已开始保存到本地");
|
||||
} catch (error) {
|
||||
if (error instanceof DOMException && error.name === "AbortError") {
|
||||
setActionNotice("已取消保存");
|
||||
} else {
|
||||
setActionNotice(error instanceof Error ? error.message : "下载失败，请稍后重试");
|
||||
}
|
||||
} finally {
|
||||
setIsDownloading(false);
|
||||
}
|
||||
};
|
||||
|
||||
const buildResultPayload = (): WorkbenchResultActionPayload | null => {
|
||||
if (!message.resultUrl || !message.resultType) return null;
|
||||
return {
|
||||
title: resultTitle,
|
||||
prompt: promptText,
|
||||
resultUrl: message.resultUrl,
|
||||
resultType: message.resultType,
|
||||
taskId: message.taskId,
|
||||
resultOriginalUrl: message.resultOriginalUrl,
|
||||
resultOssKey: message.resultOssKey,
|
||||
resultMimeType: message.resultMimeType,
|
||||
};
|
||||
};
|
||||
|
||||
const handleSaveAsset = async () => {
|
||||
const payload = buildResultPayload();
|
||||
if (!payload) return;
|
||||
saveAssetToLocalLibrary({
|
||||
type: payload.resultType === "video" ? "video" : "scene",
|
||||
name: payload.title,
|
||||
description: payload.prompt || "从生成结果保存的素材。",
|
||||
imageUrl: payload.resultUrl,
|
||||
tags: [payload.resultType === "video" ? "生成视频" : "生成图片", "工作台"],
|
||||
});
|
||||
try {
|
||||
await assetClient.create({
|
||||
type: payload.resultType === "video" ? "video" : "image",
|
||||
name: payload.title,
|
||||
description: payload.prompt || "从生成结果保存的素材。",
|
||||
url: payload.resultUrl,
|
||||
ossKey: payload.resultOssKey,
|
||||
tags: [payload.resultType === "video" ? "生成视频" : "生成图片", "工作台"],
|
||||
sourceTaskId: payload.taskId,
|
||||
metadata: {
|
||||
source: "workbench-result",
|
||||
prompt: payload.prompt,
|
||||
resultType: payload.resultType,
|
||||
originalUrl: payload.resultOriginalUrl || null,
|
||||
mimeType: payload.resultMimeType || null,
|
||||
},
|
||||
});
|
||||
setActionNotice("已保存到服务器资产库");
|
||||
} catch {
|
||||
setActionNotice("已保存到本地资产库，登录后可同步服务器。");
|
||||
}
|
||||
};
|
||||
|
||||
const handleContinueInCanvas = () => {
|
||||
const payload = buildResultPayload();
|
||||
if (!payload) return;
|
||||
onOpenResultInCanvas?.(payload);
|
||||
setActionNotice("已打开画布继续编辑");
|
||||
};
|
||||
|
||||
const handlePublish = async () => {
|
||||
const payload = buildResultPayload();
|
||||
if (!payload || isPublishing) return;
|
||||
setIsPublishing(true);
|
||||
setActionNotice(null);
|
||||
try {
|
||||
await communityClient.publishCase({
|
||||
title: payload.title,
|
||||
description: payload.prompt || "从工作台生成结果提交的社区作品。",
|
||||
coverUrl: payload.resultUrl,
|
||||
tags: [payload.resultType === "video" ? "视频" : "图片", "Web生成"],
|
||||
metadata: {
|
||||
source: "workbench-result",
|
||||
taskId: payload.taskId,
|
||||
prompt: payload.prompt,
|
||||
resultType: payload.resultType,
|
||||
resultOssKey: payload.resultOssKey || null,
|
||||
resultOriginalUrl: payload.resultOriginalUrl || null,
|
||||
resultMimeType: payload.resultMimeType || null,
|
||||
},
|
||||
assets: [
|
||||
{
|
||||
assetType: payload.resultType,
|
||||
title: payload.title,
|
||||
url: payload.resultUrl,
|
||||
ossKey: payload.resultOssKey,
|
||||
metadata: {
|
||||
prompt: payload.prompt,
|
||||
taskId: payload.taskId,
|
||||
originalUrl: payload.resultOriginalUrl || null,
|
||||
mimeType: payload.resultMimeType || null,
|
||||
},
|
||||
},
|
||||
],
|
||||
});
|
||||
setActionNotice("已提交社区审核。");
|
||||
} catch (error) {
|
||||
setActionNotice(error instanceof Error ? error.message : "提交社区失败");
|
||||
} finally {
|
||||
setIsPublishing(false);
|
||||
}
|
||||
};
|
||||
|
||||
return (
|
||||
<div className={`ai-chat-image-result-card result-card-enter${isVideo ? " is-video" : ""}`}>
|
||||
{hasMedia ? (
|
||||
<div className={`ai-chat-image-frame${isVideo ? " ai-chat-image-frame--video" : ""}${mediaExpired ? " is-expired" : ""}`} style={mediaStyle}>
|
||||
{mediaExpired ? (
|
||||
<div className="ai-chat-media-expired">
|
||||
<span>资源已过期</span>
|
||||
<button type="button" onClick={() => onRegenerate(message)}>
|
||||
<ReloadOutlined /> 重新生成
|
||||
</button>
|
||||
</div>
|
||||
) : isVideo ? (
|
||||
<button type="button" className="ai-chat-video-poster" onClick={openGeneratedMedia}>
|
||||
<video
|
||||
src={message.resultUrl}
|
||||
playsInline
|
||||
muted
|
||||
loop
|
||||
preload="metadata"
|
||||
onError={() => setMediaExpired(true)}
|
||||
/>
|
||||
<span className="ai-chat-video-poster__scrim" />
|
||||
<span className="ai-chat-video-poster__play">
|
||||
<CaretRightOutlined />
|
||||
</span>
|
||||
<span className="ai-chat-video-poster__duration">
|
||||
<ClockCircleOutlined /> {durationTag}
|
||||
</span>
|
||||
</button>
|
||||
) : (
|
||||
<button type="button" className="ai-chat-image-preview-button" onClick={openGeneratedMedia}>
|
||||
<img
|
||||
src={message.resultUrl}
|
||||
alt="生成结果"
|
||||
onError={() => setMediaExpired(true)}
|
||||
onLoad={(event) => {
|
||||
const { naturalWidth, naturalHeight } = event.currentTarget;
|
||||
if (naturalWidth > 0 && naturalHeight > 0) {
|
||||
setMediaRatio(`${naturalWidth} / ${naturalHeight}`);
|
||||
}
|
||||
}}
|
||||
/>
|
||||
</button>
|
||||
)}
|
||||
</div>
|
||||
) : null}
|
||||
|
||||
<div className="ai-chat-image-result-meta">
|
||||
<div className="ai-chat-image-result-meta__top">
|
||||
<strong>{isVideo ? "视频已生成" : "图像已生成"}</strong>
|
||||
<span>{message.taskStatusLabel || "Completed"}</span>
|
||||
</div>
|
||||
{promptText && <p className="ai-chat-image-result-meta__prompt">{promptText}</p>}
|
||||
<div className="ai-chat-image-result-meta__chips">
|
||||
{modelTag && <span className="ai-chat-chip">{modelTag}</span>}
|
||||
{modeTag && <span className="ai-chat-chip">{modeTag}</span>}
|
||||
</div>
|
||||
{isSuperResolving ? (
|
||||
<div className="ai-chat-image-result-progress" aria-label="视频超分进度">
|
||||
<SmoothedProgressBar
|
||||
progress={message.taskProgress ?? 18}
|
||||
status={message.status || "thinking"}
|
||||
label={message.taskStatusLabel || "超分处理中..."}
|
||||
/>
|
||||
</div>
|
||||
) : null}
|
||||
</div>
|
||||
|
||||
<div className="ai-chat-image-actions">
|
||||
<button
|
||||
type="button"
|
||||
className="ai-chat-image-actions__primary"
|
||||
onClick={() => onRegenerate(message)}
|
||||
>
|
||||
<ReloadOutlined /> 重新生成
|
||||
</button>
|
||||
<span className="ai-chat-image-actions__spacer" />
|
||||
<button
|
||||
type="button"
|
||||
className="ai-chat-image-actions__icon"
|
||||
onClick={() => void handleDownload()}
|
||||
disabled={isDownloading}
|
||||
title={isDownloading ? "正在准备下载" : "下载到本地"}
|
||||
aria-label={isDownloading ? "正在准备下载" : "下载到本地"}
|
||||
>
|
||||
<DownloadOutlined />
|
||||
</button>
|
||||
{isVideo && message.resultUrl ? (
|
||||
<button
|
||||
type="button"
|
||||
className="ai-chat-image-actions__icon ai-chat-image-actions__icon--super-resolution"
|
||||
onClick={() => onSuperResolveVideo(message)}
|
||||
disabled={isSuperResolving}
|
||||
title="超分"
|
||||
aria-label="超分"
|
||||
>
|
||||
<ThunderboltOutlined />
|
||||
</button>
|
||||
) : null}
|
||||
{!isVideo && message.resultUrl ? (
|
||||
<button
|
||||
type="button"
|
||||
className="ai-chat-image-actions__icon ai-chat-image-actions__icon--super-resolution"
|
||||
onClick={() => onSuperResolveImage(message)}
|
||||
disabled={isSuperResolving}
|
||||
title="超分"
|
||||
aria-label="超分"
|
||||
>
|
||||
<ThunderboltOutlined />
|
||||
</button>
|
||||
) : null}
|
||||
<button type="button" className="ai-chat-image-actions__icon" onClick={handleCopyPrompt}>
|
||||
<CopyOutlined />
|
||||
</button>
|
||||
</div>
|
||||
{hasMedia ? (
|
||||
<div className="ai-chat-result-flow">
|
||||
<button type="button" onClick={() => void handleSaveAsset()}>
|
||||
<AppstoreOutlined /> 存资产
|
||||
</button>
|
||||
<button type="button" onClick={handleContinueInCanvas} disabled={!onOpenResultInCanvas}>
|
||||
<PictureOutlined /> 进画布
|
||||
</button>
|
||||
<button type="button" onClick={() => void handlePublish()} disabled={isPublishing}>
|
||||
<ThunderboltOutlined /> {isPublishing ? "提交中..." : "发社区"}
|
||||
</button>
|
||||
{actionNotice ? <span>{actionNotice}</span> : null}
|
||||
</div>
|
||||
) : null}
|
||||
</div>
|
||||
);
|
||||
});
|
||||
|
||||
// ─── ChatAttachmentPreview ────────────────────────────────────────────────────
|
||||
|
||||
export const ChatAttachmentPreview = memo(function ChatAttachmentPreview({
|
||||
item,
|
||||
onOpen,
|
||||
}: {
|
||||
item: WorkbenchChatAttachment;
|
||||
onOpen: (item: WorkbenchChatAttachment) => void;
|
||||
}) {
|
||||
const [mediaFailed, setMediaFailed] = useState(false);
|
||||
const urlExpired = !item.previewUrl || item.previewUrl.startsWith("blob:");
|
||||
const hasMedia = !urlExpired && !mediaFailed && (item.kind === "image" || item.kind === "video");
|
||||
|
||||
return (
|
||||
<button
|
||||
type="button"
|
||||
className={`ai-chat-attachment-thumb${!hasMedia ? " is-file" : ""}`}
|
||||
onClick={() => hasMedia && onOpen(item)}
|
||||
disabled={!hasMedia}
|
||||
title={`${item.token} ${item.name}`}
|
||||
>
|
||||
<span className="ai-chat-attachment-thumb__media">
|
||||
{hasMedia && item.kind === "image" ? (
|
||||
<img src={item.previewUrl} alt={item.name} loading="lazy" onError={() => setMediaFailed(true)} />
|
||||
) : hasMedia && item.kind === "video" ? (
|
||||
<video src={item.previewUrl} muted playsInline onError={() => setMediaFailed(true)} />
|
||||
) : (
|
||||
<span className="ai-chat-attachment-thumb__icon">
|
||||
{item.kind === "audio" ? <SoundOutlined /> : <FileTextOutlined />}
|
||||
</span>
|
||||
)}
|
||||
</span>
|
||||
<span className="ai-chat-attachment-thumb__label">{item.token}</span>
|
||||
</button>
|
||||
);
|
||||
});
|
||||
|
||||
// ─── GenerationPendingCard ────────────────────────────────────────────────────
|
||||
|
||||
export const GenerationPendingCard = memo(function GenerationPendingCard({
|
||||
message,
|
||||
onStop,
|
||||
}: {
|
||||
message: WorkbenchChatMessage;
|
||||
onStop?: () => void;
|
||||
}) {
|
||||
const specs = message.result?.specs || [];
|
||||
const prompt = message.prompt || message.body;
|
||||
const isVideo = message.mode === "video";
|
||||
const smoothed = useSmoothedProgress(message.taskProgress ?? 5, message.status === "thinking" ? "running" : "completed");
|
||||
|
||||
return (
|
||||
<div className={`ai-generation-pending-card${isVideo ? " is-video" : " is-image"}`}>
|
||||
<div className="ai-generation-pending-card__stage">
|
||||
<div className="ai-generation-pending-card__grid" />
|
||||
<div className="ai-generation-pending-card__glow" />
|
||||
<div className="ai-generation-pending-card__loader">
|
||||
<span className="ai-generation-pending-card__ring" />
|
||||
<em className="ai-generation-pending-card__percent">{smoothed}%</em>
|
||||
</div>
|
||||
<div className="ai-generation-pending-card__scan" />
|
||||
</div>
|
||||
<div className="ai-generation-pending-card__meta">
|
||||
<div>
|
||||
<strong>{message.taskStatusLabel || "Generating..."}</strong>
|
||||
<span>{prompt}</span>
|
||||
</div>
|
||||
{specs.length > 0 && (
|
||||
<div className="ai-generation-pending-card__chips">
|
||||
{specs.slice(0, 3).map((spec) => (
|
||||
<span key={spec}>{spec}</span>
|
||||
))}
|
||||
</div>
|
||||
)}
|
||||
{onStop && (
|
||||
<button type="button" className="ai-generation-pending-card__stop" onClick={onStop} title="终止任务">
|
||||
<StopOutlined /> 终止
|
||||
</button>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
});
|
||||
@@ -0,0 +1,236 @@
|
||||
import type { ReactNode } from "react";
|
||||
|
||||
export function renderPlainMarkdownText(text: string, keyPrefix: string): ReactNode[] {
|
||||
const nodes: ReactNode[] = [];
|
||||
const pattern = /https?:\/\/[^\s)]+/g;
|
||||
let cursor = 0;
|
||||
let match: RegExpExecArray | null;
|
||||
|
||||
while ((match = pattern.exec(text)) !== null) {
|
||||
if (match.index > cursor) {
|
||||
nodes.push(text.slice(cursor, match.index));
|
||||
}
|
||||
|
||||
const url = match[0];
|
||||
nodes.push(
|
||||
<a key={`${keyPrefix}-link-${match.index}`} href={url} target="_blank" rel="noreferrer">
|
||||
{url}
|
||||
</a>,
|
||||
);
|
||||
cursor = match.index + url.length;
|
||||
}
|
||||
|
||||
if (cursor < text.length) {
|
||||
nodes.push(text.slice(cursor));
|
||||
}
|
||||
|
||||
return nodes.length > 0 ? nodes : [text];
|
||||
}
|
||||
|
||||
export function renderMarkdownInline(text: string, keyPrefix: string): ReactNode[] {
|
||||
const nodes: ReactNode[] = [];
|
||||
const pattern = /(`[^`\n]+`|\*\*[\s\S]+?\*\*|__[\s\S]+?__|~~[\s\S]+?~~|\[[^\]]+\]\(https?:\/\/[^)\s]+\)|https?:\/\/[^\s)]+)/g;
|
||||
let cursor = 0;
|
||||
let match: RegExpExecArray | null;
|
||||
|
||||
while ((match = pattern.exec(text)) !== null) {
|
||||
const token = match[0];
|
||||
if (match.index > cursor) {
|
||||
nodes.push(...renderPlainMarkdownText(text.slice(cursor, match.index), `${keyPrefix}-text-${cursor}`));
|
||||
}
|
||||
|
||||
if (token.startsWith("`") && token.endsWith("`")) {
|
||||
nodes.push(<code key={`${keyPrefix}-code-${match.index}`}>{token.slice(1, -1)}</code>);
|
||||
} else if ((token.startsWith("**") && token.endsWith("**")) || (token.startsWith("__") && token.endsWith("__"))) {
|
||||
nodes.push(<strong key={`${keyPrefix}-strong-${match.index}`}>{renderMarkdownInline(token.slice(2, -2), `${keyPrefix}-strong-${match.index}`)}</strong>);
|
||||
} else if (token.startsWith("~~") && token.endsWith("~~")) {
|
||||
nodes.push(<del key={`${keyPrefix}-del-${match.index}`}>{renderMarkdownInline(token.slice(2, -2), `${keyPrefix}-del-${match.index}`)}</del>);
|
||||
} else if (token.startsWith("[") && token.includes("](") && token.endsWith(")")) {
|
||||
const labelEnd = token.indexOf("](");
|
||||
const label = token.slice(1, labelEnd);
|
||||
const href = token.slice(labelEnd + 2, -1);
|
||||
nodes.push(
|
||||
<a key={`${keyPrefix}-md-link-${match.index}`} href={href} target="_blank" rel="noreferrer">
|
||||
{renderMarkdownInline(label, `${keyPrefix}-md-link-label-${match.index}`)}
|
||||
</a>,
|
||||
);
|
||||
} else {
|
||||
nodes.push(
|
||||
<a key={`${keyPrefix}-raw-link-${match.index}`} href={token} target="_blank" rel="noreferrer">
|
||||
{token}
|
||||
</a>,
|
||||
);
|
||||
}
|
||||
|
||||
cursor = match.index + token.length;
|
||||
}
|
||||
|
||||
if (cursor < text.length) {
|
||||
nodes.push(...renderPlainMarkdownText(text.slice(cursor), `${keyPrefix}-text-${cursor}`));
|
||||
}
|
||||
|
||||
return nodes.length > 0 ? nodes : [text];
|
||||
}
|
||||
|
||||
export function splitMarkdownTableRow(line: string): string[] {
|
||||
return line
|
||||
.trim()
|
||||
.replace(/^\|/, "")
|
||||
.replace(/\|$/, "")
|
||||
.split("|")
|
||||
.map((cell) => cell.trim());
|
||||
}
|
||||
|
||||
export function isMarkdownTableDivider(line: string): boolean {
|
||||
return /^\s*\|?\s*:?-{3,}:?\s*(\|\s*:?-{3,}:?\s*)+\|?\s*$/.test(line);
|
||||
}
|
||||
|
||||
export function isMarkdownBlockStart(line: string, nextLine?: string): boolean {
|
||||
const trimmed = line.trim();
|
||||
return (
|
||||
trimmed === "" ||
|
||||
/^```/.test(trimmed) ||
|
||||
/^#{1,4}\s+/.test(trimmed) ||
|
||||
/^>\s?/.test(trimmed) ||
|
||||
/^[-*+]\s+/.test(trimmed) ||
|
||||
/^\d+[.)]\s+/.test(trimmed) ||
|
||||
/^[-*_]{3,}$/.test(trimmed) ||
|
||||
(line.includes("|") && !!nextLine && isMarkdownTableDivider(nextLine))
|
||||
);
|
||||
}
|
||||
|
||||
export function renderMarkdownBlocks(text: string): ReactNode[] {
|
||||
const lines = text.replace(/\r\n?/g, "\n").trim().split("\n");
|
||||
const blocks: ReactNode[] = [];
|
||||
let index = 0;
|
||||
|
||||
while (index < lines.length) {
|
||||
const line = lines[index] ?? "";
|
||||
const trimmed = line.trim();
|
||||
|
||||
if (!trimmed) {
|
||||
index += 1;
|
||||
continue;
|
||||
}
|
||||
|
||||
const fenceMatch = trimmed.match(/^```(\S+)?/);
|
||||
if (fenceMatch) {
|
||||
const language = fenceMatch[1] || "";
|
||||
const codeLines: string[] = [];
|
||||
index += 1;
|
||||
while (index < lines.length && !lines[index].trim().startsWith("```")) {
|
||||
codeLines.push(lines[index]);
|
||||
index += 1;
|
||||
}
|
||||
if (index < lines.length) index += 1;
|
||||
blocks.push(
|
||||
<figure key={`code-${index}`} className="ai-chat-markdown-code">
|
||||
{language ? <figcaption>{language}</figcaption> : null}
|
||||
<pre>
|
||||
<code>{codeLines.join("\n")}</code>
|
||||
</pre>
|
||||
</figure>,
|
||||
);
|
||||
continue;
|
||||
}
|
||||
|
||||
const headingMatch = trimmed.match(/^(#{1,4})\s+(.+)$/);
|
||||
if (headingMatch) {
|
||||
const level = Math.min(headingMatch[1].length, 4);
|
||||
const Tag = `h${level}` as keyof JSX.IntrinsicElements;
|
||||
blocks.push(<Tag key={`heading-${index}`}>{renderMarkdownInline(headingMatch[2], `heading-${index}`)}</Tag>);
|
||||
index += 1;
|
||||
continue;
|
||||
}
|
||||
|
||||
if (/^[-*_]{3,}$/.test(trimmed)) {
|
||||
blocks.push(<hr key={`rule-${index}`} />);
|
||||
index += 1;
|
||||
continue;
|
||||
}
|
||||
|
||||
if (line.includes("|") && index + 1 < lines.length && isMarkdownTableDivider(lines[index + 1])) {
|
||||
const headers = splitMarkdownTableRow(line);
|
||||
const rows: string[][] = [];
|
||||
index += 2;
|
||||
while (index < lines.length && lines[index].includes("|") && lines[index].trim()) {
|
||||
rows.push(splitMarkdownTableRow(lines[index]));
|
||||
index += 1;
|
||||
}
|
||||
blocks.push(
|
||||
<div key={`table-${index}`} className="ai-chat-markdown-table-wrap">
|
||||
<table>
|
||||
<thead>
|
||||
<tr>
|
||||
{headers.map((header, cellIndex) => (
|
||||
<th key={`table-head-${cellIndex}`}>{renderMarkdownInline(header, `table-head-${cellIndex}`)}</th>
|
||||
))}
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{rows.map((row, rowIndex) => (
|
||||
<tr key={`table-row-${rowIndex}`}>
|
||||
{headers.map((_, cellIndex) => (
|
||||
<td key={`table-cell-${rowIndex}-${cellIndex}`}>
|
||||
{renderMarkdownInline(row[cellIndex] || "", `table-cell-${rowIndex}-${cellIndex}`)}
|
||||
</td>
|
||||
))}
|
||||
</tr>
|
||||
))}
|
||||
</tbody>
|
||||
</table>
|
||||
</div>,
|
||||
);
|
||||
continue;
|
||||
}
|
||||
|
||||
if (/^>\s?/.test(trimmed)) {
|
||||
const quoteLines: string[] = [];
|
||||
while (index < lines.length && /^>\s?/.test(lines[index].trim())) {
|
||||
quoteLines.push(lines[index].trim().replace(/^>\s?/, ""));
|
||||
index += 1;
|
||||
}
|
||||
blocks.push(
|
||||
<blockquote key={`quote-${index}`}>
|
||||
{quoteLines.map((quoteLine, quoteIndex) => (
|
||||
<p key={`quote-line-${quoteIndex}`}>{renderMarkdownInline(quoteLine, `quote-${index}-${quoteIndex}`)}</p>
|
||||
))}
|
||||
</blockquote>,
|
||||
);
|
||||
continue;
|
||||
}
|
||||
|
||||
const unorderedMatch = trimmed.match(/^[-*+]\s+(.+)$/);
|
||||
const orderedMatch = trimmed.match(/^\d+[.)]\s+(.+)$/);
|
||||
if (unorderedMatch || orderedMatch) {
|
||||
const ordered = Boolean(orderedMatch);
|
||||
const ListTag = ordered ? "ol" : "ul";
|
||||
const items: string[] = [];
|
||||
while (index < lines.length) {
|
||||
const itemMatch = ordered ? lines[index].trim().match(/^\d+[.)]\s+(.+)$/) : lines[index].trim().match(/^[-*+]\s+(.+)$/);
|
||||
if (!itemMatch) break;
|
||||
items.push(itemMatch[1]);
|
||||
index += 1;
|
||||
}
|
||||
blocks.push(
|
||||
<ListTag key={`list-${index}`}>
|
||||
{items.map((item, itemIndex) => (
|
||||
<li key={`list-item-${itemIndex}`}>{renderMarkdownInline(item, `list-${index}-${itemIndex}`)}</li>
|
||||
))}
|
||||
</ListTag>,
|
||||
);
|
||||
continue;
|
||||
}
|
||||
|
||||
const paragraphLines: string[] = [line];
|
||||
index += 1;
|
||||
while (index < lines.length && !isMarkdownBlockStart(lines[index], lines[index + 1])) {
|
||||
paragraphLines.push(lines[index]);
|
||||
index += 1;
|
||||
}
|
||||
const paragraphText = paragraphLines.join("\n").trim();
|
||||
blocks.push(<p key={`paragraph-${index}`}>{renderMarkdownInline(paragraphText, `paragraph-${index}`)}</p>);
|
||||
}
|
||||
|
||||
return blocks;
|
||||
}
|
||||
@@ -0,0 +1,53 @@
|
||||
import { assetClient } from "../../api/assetClient";
|
||||
import type { WebAssetItem } from "../../types";
|
||||
import { saveAssetToLocalLibrary } from "../assets/localAssetStore";
|
||||
import { downloadResultAsset } from "./workbenchDownload";
|
||||
|
||||
export interface ToolResultAssetInput {
|
||||
url: string;
|
||||
name: string;
|
||||
description?: string;
|
||||
type: WebAssetItem["type"];
|
||||
isVideo?: boolean;
|
||||
taskId?: string;
|
||||
tags?: string[];
|
||||
metadata?: Record<string, unknown>;
|
||||
}
|
||||
|
||||
export async function saveToolResultToLocal(input: ToolResultAssetInput) {
|
||||
return downloadResultAsset(input.url, input.name, Boolean(input.isVideo), input.taskId);
|
||||
}
|
||||
|
||||
export async function addToolResultToAssetLibrary(input: ToolResultAssetInput): Promise<"server" | "local"> {
|
||||
const description = input.description || "从工具盒生成结果加入的素材。";
|
||||
const tags = input.tags?.length ? input.tags : ["工具盒", input.isVideo ? "生成视频" : "生成图片"];
|
||||
|
||||
saveAssetToLocalLibrary({
|
||||
type: input.type,
|
||||
name: input.name,
|
||||
description,
|
||||
url: input.url,
|
||||
imageUrl: input.url,
|
||||
tags,
|
||||
});
|
||||
|
||||
try {
|
||||
await assetClient.create({
|
||||
type: input.type,
|
||||
name: input.name,
|
||||
description,
|
||||
url: input.url,
|
||||
imageUrl: input.url,
|
||||
tags,
|
||||
sourceTaskId: input.taskId,
|
||||
metadata: {
|
||||
source: "toolbox-result",
|
||||
resultType: input.isVideo ? "video" : "image",
|
||||
...input.metadata,
|
||||
},
|
||||
});
|
||||
return "server";
|
||||
} catch {
|
||||
return "local";
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,46 @@
|
||||
export type WorkbenchMode = "chat" | "image" | "video";
|
||||
|
||||
export interface WorkbenchChatAttachment {
|
||||
kind: "image" | "video" | "audio" | "file";
|
||||
name: string;
|
||||
token: string;
|
||||
previewUrl?: string;
|
||||
remoteUrl?: string;
|
||||
}
|
||||
|
||||
export interface WorkbenchChatMessage {
|
||||
id: string;
|
||||
role: "user" | "assistant";
|
||||
author: string;
|
||||
mode: WorkbenchMode;
|
||||
body: string;
|
||||
prompt?: string;
|
||||
createdAt: string;
|
||||
status?: "thinking" | "queued" | "completed" | "failed";
|
||||
taskId?: string;
|
||||
conversationId?: number;
|
||||
taskProgress?: number;
|
||||
taskStatusLabel?: string;
|
||||
attachments?: WorkbenchChatAttachment[];
|
||||
resultUrl?: string;
|
||||
resultType?: "image" | "video";
|
||||
resultOriginalUrl?: string;
|
||||
resultOssKey?: string;
|
||||
resultMimeType?: string;
|
||||
result?: {
|
||||
title: string;
|
||||
summary: string;
|
||||
specs: string[];
|
||||
};
|
||||
}
|
||||
|
||||
export interface WorkbenchResultActionPayload {
|
||||
title: string;
|
||||
prompt: string;
|
||||
resultUrl: string;
|
||||
resultType: "image" | "video";
|
||||
taskId?: string;
|
||||
resultOriginalUrl?: string;
|
||||
resultOssKey?: string;
|
||||
resultMimeType?: string;
|
||||
}
|
||||
@@ -0,0 +1,154 @@
|
||||
import { aiGenerationClient } from "../../api/aiGenerationClient";
|
||||
import { buildApiUrl, buildAuthHeaders } from "../../api/serverConnection";
|
||||
|
||||
function sanitizeDownloadFilename(value: string, fallback: string) {
|
||||
const normalized = value
|
||||
.trim()
|
||||
.replace(/[\\/:*?"<>|]+/g, "-")
|
||||
.replace(/\s+/g, " ")
|
||||
.slice(0, 80)
|
||||
.trim();
|
||||
return normalized || fallback;
|
||||
}
|
||||
|
||||
function getResultDownloadExtension(url: string, contentType: string | null, isVideo: boolean) {
|
||||
const mime = (contentType || "").split(";")[0]?.trim().toLowerCase();
|
||||
if (mime === "image/jpeg") return "jpg";
|
||||
if (mime === "image/png") return "png";
|
||||
if (mime === "image/webp") return "webp";
|
||||
if (mime === "image/gif") return "gif";
|
||||
if (mime === "video/webm") return "webm";
|
||||
if (mime === "video/quicktime") return "mov";
|
||||
if (mime === "video/mp4") return "mp4";
|
||||
|
||||
try {
|
||||
const pathname = new URL(url, window.location.href).pathname;
|
||||
const matched = pathname.match(/\.([a-z0-9]{2,5})$/i);
|
||||
if (matched?.[1]) return matched[1].toLowerCase();
|
||||
} catch {
|
||||
// Keep the fallback below for malformed but still browser-loadable URLs.
|
||||
}
|
||||
|
||||
return isVideo ? "mp4" : "png";
|
||||
}
|
||||
|
||||
type LocalFilePickerWritable = {
|
||||
write: (data: Blob) => Promise<void>;
|
||||
close: () => Promise<void>;
|
||||
};
|
||||
|
||||
type LocalFilePickerHandle = {
|
||||
createWritable: () => Promise<LocalFilePickerWritable>;
|
||||
};
|
||||
|
||||
type LocalFilePickerWindow = Window & {
|
||||
showSaveFilePicker?: (options?: {
|
||||
suggestedName?: string;
|
||||
types?: Array<{
|
||||
description: string;
|
||||
accept: Record<string, string[]>;
|
||||
}>;
|
||||
}) => Promise<LocalFilePickerHandle>;
|
||||
};
|
||||
|
||||
async function saveBlobToLocal(blob: Blob, filename: string, isVideo: boolean): Promise<"saved" | "started"> {
|
||||
const startBrowserDownload = () => {
|
||||
const objectUrl = URL.createObjectURL(blob);
|
||||
const anchor = document.createElement("a");
|
||||
anchor.href = objectUrl;
|
||||
anchor.download = filename;
|
||||
anchor.rel = "noopener";
|
||||
document.body.appendChild(anchor);
|
||||
anchor.click();
|
||||
anchor.remove();
|
||||
|
||||
window.setTimeout(() => URL.revokeObjectURL(objectUrl), 30_000);
|
||||
return "started" as const;
|
||||
};
|
||||
|
||||
const savePicker = (window as LocalFilePickerWindow).showSaveFilePicker;
|
||||
if (savePicker) {
|
||||
try {
|
||||
const pickerTypes = [
|
||||
isVideo
|
||||
? { description: "Video", accept: { [blob.type || "video/mp4"]: [`.${filename.split(".").pop() || "mp4"}`] } }
|
||||
: { description: "Image", accept: { [blob.type || "image/png"]: [`.${filename.split(".").pop() || "png"}`] } },
|
||||
];
|
||||
const handle = await savePicker({ suggestedName: filename, types: pickerTypes });
|
||||
const writable = await handle.createWritable();
|
||||
await writable.write(blob);
|
||||
await writable.close();
|
||||
return "saved";
|
||||
} catch (error) {
|
||||
if (error instanceof DOMException && error.name === "AbortError") throw error;
|
||||
return startBrowserDownload();
|
||||
}
|
||||
}
|
||||
|
||||
return startBrowserDownload();
|
||||
}
|
||||
|
||||
async function assertDownloadBlobIsUsable(blob: Blob, isVideo: boolean) {
|
||||
const mime = String(blob.type || "").toLowerCase();
|
||||
if (/^(?:application|text)\/(?:json|xml|html|plain)|\+xml/.test(mime)) {
|
||||
throw new Error("下载失败：结果链接已过期或返回了错误内容，请重新生成后再下载。");
|
||||
}
|
||||
|
||||
const header = new Uint8Array(await blob.slice(0, 16).arrayBuffer());
|
||||
const asText = new TextDecoder("utf-8", { fatal: false }).decode(header);
|
||||
const isPng = header[0] === 0x89 && header[1] === 0x50 && header[2] === 0x4e && header[3] === 0x47;
|
||||
const isJpeg = header[0] === 0xff && header[1] === 0xd8 && header[2] === 0xff;
|
||||
const isGif = asText.startsWith("GIF8");
|
||||
const isWebp = asText.startsWith("RIFF") && asText.slice(8, 12) === "WEBP";
|
||||
const isMp4 = asText.slice(4, 8) === "ftyp";
|
||||
const looksLikeErrorDocument = asText.startsWith("<?xml") || asText.startsWith("<Error") || asText.startsWith("{\"") || asText.startsWith("<!DO");
|
||||
|
||||
if (looksLikeErrorDocument) {
|
||||
throw new Error("下载失败：结果链接已过期或返回了错误内容，请重新生成后再下载。");
|
||||
}
|
||||
|
||||
const hasExpectedMagic = isVideo ? isMp4 || mime.startsWith("video/") : isPng || isJpeg || isGif || isWebp || mime.startsWith("image/");
|
||||
if (!hasExpectedMagic) {
|
||||
throw new Error("下载失败：未收到有效的媒体文件。");
|
||||
}
|
||||
}
|
||||
|
||||
export async function downloadResultAsset(url: string, filenameBase: string, isVideo: boolean, taskId?: string): Promise<"saved" | "started"> {
|
||||
let blob: Blob | null = null;
|
||||
let contentType: string | null = null;
|
||||
|
||||
if (taskId) {
|
||||
try {
|
||||
const result = await aiGenerationClient.downloadTaskResult(taskId);
|
||||
blob = result.blob;
|
||||
contentType = result.contentType || result.blob.type || null;
|
||||
} catch {
|
||||
blob = null;
|
||||
}
|
||||
}
|
||||
|
||||
if (!blob) {
|
||||
const needsProxy = /aliyuncs\.com/i.test(url);
|
||||
const fetchUrl = needsProxy
|
||||
? buildApiUrl(`ai/proxy-download?url=${encodeURIComponent(url)}`)
|
||||
: url;
|
||||
const fetchOpts: RequestInit = needsProxy
|
||||
? { headers: buildAuthHeaders() }
|
||||
: { credentials: "omit" };
|
||||
const response = await fetch(fetchUrl, fetchOpts);
|
||||
if (!response.ok) {
|
||||
throw new Error(`下载失败：资源请求返回 ${response.status}`);
|
||||
}
|
||||
blob = await response.blob();
|
||||
contentType = blob.type || response.headers.get("content-type");
|
||||
}
|
||||
|
||||
if (!blob.size) {
|
||||
throw new Error("下载失败：资源内容为空");
|
||||
}
|
||||
await assertDownloadBlobIsUsable(blob, isVideo);
|
||||
|
||||
const extension = getResultDownloadExtension(url, contentType, isVideo);
|
||||
const filename = `${sanitizeDownloadFilename(filenameBase, isVideo ? "generated-video" : "generated-image")}.${extension}`;
|
||||
return saveBlobToLocal(blob, filename, isVideo);
|
||||
}
|
||||
@@ -0,0 +1,234 @@
|
||||
import { aiGenerationClient } from "../../api/aiGenerationClient";
|
||||
|
||||
export interface PersistedWorkbenchResultAsset {
|
||||
url: string;
|
||||
originalUrl: string;
|
||||
ossKey?: string;
|
||||
mimeType?: string;
|
||||
}
|
||||
|
||||
interface PersistWorkbenchResultAssetInput {
|
||||
title: string;
|
||||
sourceUrl: string;
|
||||
resultType: "image" | "video";
|
||||
taskId?: string;
|
||||
prompt?: string;
|
||||
originalUrl?: string;
|
||||
existingOssKey?: string | null;
|
||||
mimeType?: string;
|
||||
client?: WorkbenchResultPersistenceClient;
|
||||
}
|
||||
|
||||
interface DownloadedTaskAsset {
|
||||
blob: Blob;
|
||||
filename?: string;
|
||||
contentType?: string;
|
||||
}
|
||||
|
||||
interface FetchLikeResponse {
|
||||
ok: boolean;
|
||||
status: number;
|
||||
headers: {
|
||||
get(name: string): string | null;
|
||||
};
|
||||
blob(): Promise<Blob>;
|
||||
}
|
||||
|
||||
interface WorkbenchResultPersistenceClient {
|
||||
downloadTaskResult(taskId: string): Promise<DownloadedTaskAsset>;
|
||||
uploadAsset(input: {
|
||||
dataUrl: string;
|
||||
name?: string;
|
||||
mimeType?: string;
|
||||
scope?: string;
|
||||
}): Promise<{ url: string; signedUrl?: string; ossKey?: string }>;
|
||||
uploadAssetByUrl(input: {
|
||||
sourceUrl: string;
|
||||
name?: string;
|
||||
mimeType?: string;
|
||||
scope?: string;
|
||||
}): Promise<{ url: string; signedUrl?: string; ossKey?: string }>;
|
||||
fetchAsset?: (url: string) => Promise<FetchLikeResponse>;
|
||||
}
|
||||
|
||||
function getDefaultClient(): WorkbenchResultPersistenceClient {
|
||||
return {
|
||||
downloadTaskResult: (taskId) => aiGenerationClient.downloadTaskResult(taskId),
|
||||
uploadAsset: (input) => aiGenerationClient.uploadAsset(input),
|
||||
uploadAssetByUrl: (input) => aiGenerationClient.uploadAssetByUrl(input),
|
||||
fetchAsset: (url) => fetch(url, { credentials: "omit" }),
|
||||
};
|
||||
}
|
||||
|
||||
function getGeneratedResultOssKey(url: string): string | null {
|
||||
try {
|
||||
const key = decodeURIComponent(new URL(url, globalThis.location?.href || "http://localhost").pathname.replace(/^\/+/, ""));
|
||||
return /^users\/[^/]+\/generation-results\/.+/i.test(key) ? key : null;
|
||||
} catch {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
function createFallbackResult(input: PersistWorkbenchResultAssetInput): PersistedWorkbenchResultAsset {
|
||||
return {
|
||||
url: input.sourceUrl,
|
||||
originalUrl: input.originalUrl || input.sourceUrl,
|
||||
ossKey: input.existingOssKey || getGeneratedResultOssKey(input.sourceUrl) || undefined,
|
||||
mimeType: input.mimeType,
|
||||
};
|
||||
}
|
||||
|
||||
function getResultExtension(url: string, contentType: string | undefined, resultType: "image" | "video") {
|
||||
const mime = (contentType || "").split(";")[0]?.trim().toLowerCase();
|
||||
if (mime === "image/jpeg") return "jpg";
|
||||
if (mime === "image/png") return "png";
|
||||
if (mime === "image/webp") return "webp";
|
||||
if (mime === "image/gif") return "gif";
|
||||
if (mime === "video/webm") return "webm";
|
||||
if (mime === "video/quicktime") return "mov";
|
||||
if (mime === "video/mp4") return "mp4";
|
||||
|
||||
try {
|
||||
const pathname = new URL(url, globalThis.location?.href || "http://localhost").pathname;
|
||||
const matched = pathname.match(/\.([a-z0-9]{2,5})$/i);
|
||||
if (matched?.[1]) return matched[1].toLowerCase();
|
||||
} catch {
|
||||
// Fall through to media-type fallback.
|
||||
}
|
||||
|
||||
return resultType === "video" ? "mp4" : "png";
|
||||
}
|
||||
|
||||
export function buildWorkbenchResultAssetName(
|
||||
title: string,
|
||||
sourceUrl: string,
|
||||
resultType: "image" | "video",
|
||||
contentType?: string,
|
||||
): string {
|
||||
const normalized = title
|
||||
.trim()
|
||||
.replace(/[\\/:*?"<>|]+/g, "-")
|
||||
.replace(/\s+/g, " ")
|
||||
.slice(0, 80)
|
||||
.trim();
|
||||
const base = normalized || (resultType === "video" ? "generated-video" : "generated-image");
|
||||
return `${base}.${getResultExtension(sourceUrl, contentType, resultType)}`;
|
||||
}
|
||||
|
||||
function blobToDataUrl(blob: Blob): Promise<string> {
|
||||
return new Promise((resolve, reject) => {
|
||||
const reader = new FileReader();
|
||||
reader.onload = () => resolve(String(reader.result || ""));
|
||||
reader.onerror = () => reject(reader.error || new Error("资源读取失败"));
|
||||
reader.readAsDataURL(blob);
|
||||
});
|
||||
}
|
||||
|
||||
async function assertMediaBlobIsUsable(blob: Blob, resultType: "image" | "video") {
|
||||
const mime = String(blob.type || "").toLowerCase();
|
||||
if (/^(?:application|text)\/(?:json|xml|html|plain)|\+xml/.test(mime)) {
|
||||
throw new Error("结果资源已过期或返回了错误内容，请重新生成。");
|
||||
}
|
||||
|
||||
const header = new Uint8Array(await blob.slice(0, 16).arrayBuffer());
|
||||
const asText = new TextDecoder("utf-8", { fatal: false }).decode(header);
|
||||
const isPng = header[0] === 0x89 && header[1] === 0x50 && header[2] === 0x4e && header[3] === 0x47;
|
||||
const isJpeg = header[0] === 0xff && header[1] === 0xd8 && header[2] === 0xff;
|
||||
const isGif = asText.startsWith("GIF8");
|
||||
const isWebp = asText.startsWith("RIFF") && asText.slice(8, 12) === "WEBP";
|
||||
const isMp4 = asText.slice(4, 8) === "ftyp";
|
||||
const looksLikeErrorDocument = asText.startsWith("<?xml") || asText.startsWith("<Error") || asText.startsWith("{\"") || asText.startsWith("<!DO");
|
||||
if (looksLikeErrorDocument) throw new Error("结果资源已过期或返回了错误内容，请重新生成。");
|
||||
|
||||
const hasExpectedMagic =
|
||||
resultType === "video"
|
||||
? isMp4 || mime.startsWith("video/")
|
||||
: isPng || isJpeg || isGif || isWebp || mime.startsWith("image/");
|
||||
if (!hasExpectedMagic) throw new Error("未收到有效的媒体文件。");
|
||||
}
|
||||
|
||||
async function downloadResultBlob(
|
||||
input: PersistWorkbenchResultAssetInput,
|
||||
client: WorkbenchResultPersistenceClient,
|
||||
): Promise<{ blob: Blob; contentType?: string; filename?: string }> {
|
||||
if (input.taskId) {
|
||||
try {
|
||||
const result = await client.downloadTaskResult(input.taskId);
|
||||
return {
|
||||
blob: result.blob,
|
||||
contentType: result.contentType || result.blob.type || undefined,
|
||||
filename: result.filename,
|
||||
};
|
||||
} catch {
|
||||
// Some older tasks may not support the proxy download route; fall back to the URL while it is still valid.
|
||||
}
|
||||
}
|
||||
|
||||
const fetchAsset = client.fetchAsset || ((url: string) => fetch(url, { credentials: "omit" }));
|
||||
const response = await fetchAsset(input.sourceUrl);
|
||||
if (!response.ok) throw new Error(`结果资源请求失败：${response.status}`);
|
||||
const blob = await response.blob();
|
||||
return {
|
||||
blob,
|
||||
contentType: blob.type || response.headers.get("content-type") || undefined,
|
||||
};
|
||||
}
|
||||
|
||||
export async function persistWorkbenchResultAsset(input: PersistWorkbenchResultAssetInput): Promise<PersistedWorkbenchResultAsset> {
|
||||
const fallbackResult = createFallbackResult(input);
|
||||
if (fallbackResult.ossKey) return fallbackResult;
|
||||
|
||||
const client = input.client || getDefaultClient();
|
||||
const mimeType = input.mimeType || (input.resultType === "video" ? "video/mp4" : "image/png");
|
||||
const name = buildWorkbenchResultAssetName(input.title, input.sourceUrl, input.resultType, mimeType);
|
||||
|
||||
// Try server-side URL-to-OSS first (no base64, no client download)
|
||||
try {
|
||||
const uploaded = await client.uploadAssetByUrl({
|
||||
sourceUrl: input.sourceUrl,
|
||||
name,
|
||||
mimeType,
|
||||
scope: "workbench-result",
|
||||
});
|
||||
return {
|
||||
url: uploaded.url,
|
||||
originalUrl: input.sourceUrl,
|
||||
ossKey: uploaded.ossKey,
|
||||
mimeType,
|
||||
};
|
||||
} catch (urlError) {
|
||||
console.warn("[workbench] URL upload failed, falling back to download+base64:", urlError instanceof Error ? urlError.message : urlError);
|
||||
}
|
||||
|
||||
// Fallback: download → base64 → upload
|
||||
try {
|
||||
const downloaded = await downloadResultBlob(input, client);
|
||||
if (!downloaded.blob.size) throw new Error("结果资源内容为空");
|
||||
await assertMediaBlobIsUsable(downloaded.blob, input.resultType);
|
||||
|
||||
const fallbackMime = downloaded.contentType || downloaded.blob.type || mimeType;
|
||||
const fallbackName = downloaded.filename || name;
|
||||
const dataUrl = await blobToDataUrl(downloaded.blob);
|
||||
const uploaded = await client.uploadAsset({
|
||||
dataUrl,
|
||||
name: fallbackName,
|
||||
mimeType: fallbackMime,
|
||||
scope: "workbench-result",
|
||||
});
|
||||
|
||||
return {
|
||||
url: uploaded.url,
|
||||
originalUrl: input.sourceUrl,
|
||||
ossKey: uploaded.ossKey,
|
||||
mimeType: fallbackMime,
|
||||
};
|
||||
} catch (error) {
|
||||
const msg = error instanceof Error ? error.message : String(error || "");
|
||||
if (/413|too large/i.test(msg)) {
|
||||
console.warn("[workbench] OSS upload rejected (file too large), using temporary URL:", msg);
|
||||
} else {
|
||||
console.warn("[workbench] result persistence fallback:", error);
|
||||
}
|
||||
return fallbackResult;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user