xiamuceer
456 lines
14 KiB
TypeScript
456 lines
14 KiB
TypeScript
import React, { useState, useEffect } from 'react';
|
||
import { useParams, useNavigate } from 'react-router-dom';
|
||
import { Card, Spin, Alert, Button, Space, Switch, Drawer, message, Progress } from 'antd';
|
||
import {
|
||
ArrowLeftOutlined,
|
||
EyeOutlined,
|
||
EyeInvisibleOutlined,
|
||
MenuOutlined,
|
||
ReloadOutlined,
|
||
LeftOutlined,
|
||
RightOutlined,
|
||
} from '@ant-design/icons';
|
||
import api from '../services/api';
|
||
import AnnotatedText, { type MemoryAnnotation } from '../components/AnnotatedText';
|
||
import MemorySidebar from '../components/MemorySidebar';
|
||
|
||
interface ChapterData {
|
||
id: string;
|
||
chapter_number: number;
|
||
title: string;
|
||
content: string;
|
||
word_count: number;
|
||
}
|
||
|
||
interface AnnotationsData {
|
||
chapter_id: string;
|
||
chapter_number: number;
|
||
title: string;
|
||
word_count: number;
|
||
annotations: MemoryAnnotation[];
|
||
has_analysis: boolean;
|
||
summary: {
|
||
total_annotations: number;
|
||
hooks: number;
|
||
foreshadows: number;
|
||
plot_points: number;
|
||
character_events: number;
|
||
};
|
||
}
|
||
|
||
interface NavigationData {
|
||
current: {
|
||
id: string;
|
||
chapter_number: number;
|
||
title: string;
|
||
};
|
||
previous: {
|
||
id: string;
|
||
chapter_number: number;
|
||
title: string;
|
||
} | null;
|
||
next: {
|
||
id: string;
|
||
chapter_number: number;
|
||
title: string;
|
||
} | null;
|
||
}
|
||
|
||
/**
|
||
* 章节阅读器页面
|
||
* 展示带有记忆标注的章节内容
|
||
*/
|
||
const ChapterReader: React.FC = () => {
|
||
const { chapterId } = useParams<{ chapterId: string }>();
|
||
const navigate = useNavigate();
|
||
|
||
const [loading, setLoading] = useState(true);
|
||
const [error, setError] = useState<string | null>(null);
|
||
const [chapter, setChapter] = useState<ChapterData | null>(null);
|
||
const [annotationsData, setAnnotationsData] = useState<AnnotationsData | null>(null);
|
||
const [showAnnotations, setShowAnnotations] = useState(true);
|
||
const [activeAnnotationId, setActiveAnnotationId] = useState<string | undefined>();
|
||
const [sidebarVisible, setSidebarVisible] = useState(false);
|
||
const [analyzing, setAnalyzing] = useState(false);
|
||
const [analysisProgress, setAnalysisProgress] = useState(0);
|
||
const [navigation, setNavigation] = useState<NavigationData | null>(null);
|
||
|
||
useEffect(() => {
|
||
if (chapterId) {
|
||
loadChapterData();
|
||
}
|
||
}, [chapterId]);
|
||
|
||
const loadChapterData = async () => {
|
||
try {
|
||
setLoading(true);
|
||
setError(null);
|
||
|
||
// 并行加载章节内容、标注数据和导航信息
|
||
// 注意：api拦截器已经解析了response.data，所以直接返回数据对象
|
||
const [chapterData, annotationsData, navigationData] = await Promise.all([
|
||
api.get<unknown, ChapterData>(`/chapters/${chapterId}`).catch(err => {
|
||
console.error('加载章节失败:', err);
|
||
throw err;
|
||
}),
|
||
api.get<unknown, AnnotationsData>(`/chapters/${chapterId}/annotations`).catch(err => {
|
||
console.warn('加载标注失败:', err);
|
||
return null;
|
||
}), // 如果没有分析数据也不报错
|
||
api.get<unknown, NavigationData>(`/chapters/${chapterId}/navigation`).catch(err => {
|
||
console.warn('加载导航信息失败:', err);
|
||
return null;
|
||
}),
|
||
]);
|
||
|
||
console.log('章节数据:', chapterData);
|
||
console.log('标注数据:', annotationsData);
|
||
console.log('导航数据:', navigationData);
|
||
|
||
// 验证数据
|
||
if (!chapterData || !chapterData.content) {
|
||
throw new Error('章节数据无效：缺少内容');
|
||
}
|
||
|
||
setChapter(chapterData);
|
||
setNavigation(navigationData);
|
||
|
||
// 验证标注数据
|
||
if (annotationsData) {
|
||
const validAnnotations = annotationsData.annotations.filter(
|
||
(a: MemoryAnnotation) => a.position >= 0 && a.position < chapterData.content.length
|
||
);
|
||
const invalidCount = annotationsData.annotations.length - validAnnotations.length;
|
||
|
||
if (invalidCount > 0) {
|
||
console.warn(`${invalidCount}个标注位置无效，将仅显示${validAnnotations.length}个有效标注`);
|
||
}
|
||
|
||
setAnnotationsData(annotationsData);
|
||
} else {
|
||
setAnnotationsData(null);
|
||
}
|
||
} catch (err: any) {
|
||
console.error('加载章节数据失败:', err);
|
||
setError(err.response?.data?.detail || err.message || '加载失败');
|
||
} finally {
|
||
setLoading(false);
|
||
}
|
||
};
|
||
|
||
const handleAnnotationClick = (annotation: MemoryAnnotation) => {
|
||
setActiveAnnotationId(annotation.id);
|
||
// 移动端显示侧边栏
|
||
if (window.innerWidth < 768) {
|
||
setSidebarVisible(true);
|
||
}
|
||
};
|
||
|
||
const handleBackClick = () => {
|
||
navigate(-1);
|
||
};
|
||
|
||
const handlePreviousChapter = () => {
|
||
if (navigation?.previous) {
|
||
navigate(`/chapters/${navigation.previous.id}/reader`);
|
||
}
|
||
};
|
||
|
||
const handleNextChapter = () => {
|
||
if (navigation?.next) {
|
||
navigate(`/chapters/${navigation.next.id}/reader`);
|
||
}
|
||
};
|
||
|
||
const handleReanalyze = async () => {
|
||
if (!chapterId) return;
|
||
|
||
try {
|
||
setAnalyzing(true);
|
||
setAnalysisProgress(0);
|
||
message.loading({ content: '开始分析章节...', key: 'analyze', duration: 0 });
|
||
|
||
// 触发分析
|
||
await api.post(`/chapters/${chapterId}/analyze`);
|
||
|
||
// 轮询分析状态
|
||
const pollInterval = setInterval(async () => {
|
||
try {
|
||
const statusRes = await api.get(`/chapters/${chapterId}/analysis/status`);
|
||
const { status, progress, error_message } = statusRes.data;
|
||
|
||
setAnalysisProgress(progress || 0);
|
||
|
||
if (status === 'completed') {
|
||
clearInterval(pollInterval);
|
||
setAnalyzing(false);
|
||
message.success({ content: '分析完成！', key: 'analyze' });
|
||
|
||
// 重新加载标注数据
|
||
const annotationsRes = await api.get(`/chapters/${chapterId}/annotations`);
|
||
setAnnotationsData(annotationsRes.data);
|
||
} else if (status === 'failed') {
|
||
clearInterval(pollInterval);
|
||
setAnalyzing(false);
|
||
message.error({
|
||
content: `分析失败：${error_message || '未知错误'}`,
|
||
key: 'analyze'
|
||
});
|
||
}
|
||
} catch (err) {
|
||
console.error('轮询分析状态失败:', err);
|
||
}
|
||
}, 2000); // 每2秒轮询一次
|
||
|
||
// 30秒超时
|
||
setTimeout(() => {
|
||
clearInterval(pollInterval);
|
||
if (analyzing) {
|
||
setAnalyzing(false);
|
||
message.warning({ content: '分析超时，请稍后刷新查看结果', key: 'analyze' });
|
||
}
|
||
}, 30000);
|
||
|
||
} catch (err: any) {
|
||
setAnalyzing(false);
|
||
message.error({
|
||
content: err.response?.data?.detail || '触发分析失败',
|
||
key: 'analyze'
|
||
});
|
||
}
|
||
};
|
||
|
||
if (loading) {
|
||
return (
|
||
<div style={{ textAlign: 'center', padding: '100px 0' }}>
|
||
<Spin size="large" tip="加载章节中..." />
|
||
</div>
|
||
);
|
||
}
|
||
|
||
if (error || !chapter) {
|
||
return (
|
||
<div style={{ padding: 24 }}>
|
||
<Alert
|
||
message="加载失败"
|
||
description={error || '章节不存在'}
|
||
type="error"
|
||
showIcon
|
||
/>
|
||
<Button onClick={handleBackClick} style={{ marginTop: 16 }}>
|
||
返回
|
||
</Button>
|
||
</div>
|
||
);
|
||
}
|
||
|
||
const hasAnnotations = annotationsData && annotationsData.annotations.length > 0;
|
||
|
||
return (
|
||
<div style={{ height: '100vh', display: 'flex', flexDirection: 'column' }}>
|
||
{/* 顶部工具栏 */}
|
||
<Card
|
||
size="small"
|
||
style={{
|
||
borderRadius: 0,
|
||
borderLeft: 0,
|
||
borderRight: 0,
|
||
borderTop: 0,
|
||
}}
|
||
>
|
||
<div style={{ display: 'flex', justifyContent: 'space-between', alignItems: 'center' }}>
|
||
<Space>
|
||
<Button icon={<ArrowLeftOutlined />} onClick={handleBackClick}>
|
||
返回
|
||
</Button>
|
||
<Button
|
||
icon={<LeftOutlined />}
|
||
onClick={handlePreviousChapter}
|
||
disabled={!navigation?.previous}
|
||
title={navigation?.previous ? `上一章: ${navigation.previous.title}` : '已是第一章'}
|
||
>
|
||
上一章
|
||
</Button>
|
||
<span style={{ fontSize: 16, fontWeight: 600 }}>
|
||
第{chapter.chapter_number}章: {chapter.title}
|
||
</span>
|
||
<Button
|
||
icon={<RightOutlined />}
|
||
onClick={handleNextChapter}
|
||
disabled={!navigation?.next}
|
||
title={navigation?.next ? `下一章: ${navigation.next.title}` : '已是最后一章'}
|
||
>
|
||
下一章
|
||
</Button>
|
||
</Space>
|
||
|
||
<Space>
|
||
<Button
|
||
icon={<ReloadOutlined />}
|
||
onClick={handleReanalyze}
|
||
loading={analyzing}
|
||
disabled={analyzing}
|
||
>
|
||
{analyzing ? '分析中...' : '重新分析'}
|
||
</Button>
|
||
{hasAnnotations && (
|
||
<>
|
||
<Switch
|
||
checked={showAnnotations}
|
||
onChange={setShowAnnotations}
|
||
checkedChildren={<EyeOutlined />}
|
||
unCheckedChildren={<EyeInvisibleOutlined />}
|
||
/>
|
||
<span style={{ fontSize: 13, color: '#666' }}>显示标注</span>
|
||
<Button
|
||
icon={<MenuOutlined />}
|
||
onClick={() => setSidebarVisible(true)}
|
||
style={{ display: window.innerWidth < 768 ? 'inline-block' : 'none' }}
|
||
>
|
||
分析
|
||
</Button>
|
||
</>
|
||
)}
|
||
</Space>
|
||
</div>
|
||
|
||
{analyzing && (
|
||
<div style={{ marginTop: 12 }}>
|
||
<Progress percent={analysisProgress} size="small" status="active" />
|
||
<span style={{ fontSize: 12, color: '#666', marginLeft: 8 }}>
|
||
正在分析章节...
|
||
</span>
|
||
</div>
|
||
)}
|
||
|
||
{!analyzing && hasAnnotations && annotationsData && (
|
||
<div style={{ marginTop: 12, fontSize: 12, color: '#999' }}>
|
||
共有 {annotationsData.summary.total_annotations} 个标注：
|
||
{annotationsData.summary.hooks > 0 && ` 🎣${annotationsData.summary.hooks}个钩子`}
|
||
{annotationsData.summary.foreshadows > 0 &&
|
||
` 🌟${annotationsData.summary.foreshadows}个伏笔`}
|
||
{annotationsData.summary.plot_points > 0 &&
|
||
` 💎${annotationsData.summary.plot_points}个情节点`}
|
||
{annotationsData.summary.character_events > 0 &&
|
||
` 👤${annotationsData.summary.character_events}个角色事件`}
|
||
</div>
|
||
)}
|
||
</Card>
|
||
|
||
{/* 主内容区域 */}
|
||
<div style={{ flex: 1, display: 'flex', overflow: 'hidden' }}>
|
||
{/* 左侧：章节内容 */}
|
||
<div
|
||
style={{
|
||
flex: 1,
|
||
overflowY: 'auto',
|
||
padding: '32px 48px',
|
||
maxWidth: hasAnnotations ? 'calc(100% - 400px)' : '100%',
|
||
}}
|
||
>
|
||
<Card>
|
||
<div style={{ maxWidth: 800, margin: '0 auto' }}>
|
||
{!hasAnnotations && (
|
||
<Alert
|
||
message="暂无分析数据"
|
||
description="该章节尚未进行AI分析，无法显示记忆标注。"
|
||
type="info"
|
||
showIcon
|
||
style={{ marginBottom: 24 }}
|
||
/>
|
||
)}
|
||
|
||
{showAnnotations && hasAnnotations && annotationsData ? (
|
||
<AnnotatedText
|
||
content={chapter.content}
|
||
annotations={annotationsData.annotations}
|
||
onAnnotationClick={handleAnnotationClick}
|
||
activeAnnotationId={activeAnnotationId}
|
||
/>
|
||
) : (
|
||
<div
|
||
style={{
|
||
lineHeight: 2,
|
||
fontSize: 16,
|
||
whiteSpace: 'pre-wrap',
|
||
wordBreak: 'break-word',
|
||
}}
|
||
>
|
||
{chapter.content}
|
||
</div>
|
||
)}
|
||
|
||
{/* 底部翻页按钮 */}
|
||
<div style={{ marginTop: 48, paddingTop: 24, borderTop: '1px solid #f0f0f0' }}>
|
||
<Space style={{ width: '100%', justifyContent: 'space-between' }}>
|
||
<Button
|
||
size="large"
|
||
icon={<LeftOutlined />}
|
||
onClick={handlePreviousChapter}
|
||
disabled={!navigation?.previous}
|
||
>
|
||
{navigation?.previous
|
||
? `上一章: 第${navigation.previous.chapter_number}章 ${navigation.previous.title}`
|
||
: '已是第一章'}
|
||
</Button>
|
||
<Button
|
||
size="large"
|
||
type="primary"
|
||
icon={<RightOutlined />}
|
||
onClick={handleNextChapter}
|
||
disabled={!navigation?.next}
|
||
iconPosition="end"
|
||
>
|
||
{navigation?.next
|
||
? `下一章: 第${navigation.next.chapter_number}章 ${navigation.next.title}`
|
||
: '已是最后一章'}
|
||
</Button>
|
||
</Space>
|
||
</div>
|
||
</div>
|
||
</Card>
|
||
</div>
|
||
|
||
{/* 右侧：记忆侧边栏（桌面端） */}
|
||
{hasAnnotations && annotationsData && window.innerWidth >= 768 && (
|
||
<div
|
||
style={{
|
||
width: 400,
|
||
borderLeft: '1px solid #f0f0f0',
|
||
overflowY: 'auto',
|
||
background: '#fafafa',
|
||
}}
|
||
>
|
||
<MemorySidebar
|
||
annotations={annotationsData.annotations}
|
||
activeAnnotationId={activeAnnotationId}
|
||
onAnnotationClick={handleAnnotationClick}
|
||
/>
|
||
</div>
|
||
)}
|
||
</div>
|
||
|
||
{/* 移动端抽屉 */}
|
||
{hasAnnotations && annotationsData && (
|
||
<Drawer
|
||
title="章节分析"
|
||
placement="right"
|
||
onClose={() => setSidebarVisible(false)}
|
||
open={sidebarVisible}
|
||
width="80%"
|
||
>
|
||
<MemorySidebar
|
||
annotations={annotationsData.annotations}
|
||
activeAnnotationId={activeAnnotationId}
|
||
onAnnotationClick={(annotation) => {
|
||
handleAnnotationClick(annotation);
|
||
setSidebarVisible(false);
|
||
}}
|
||
/>
|
||
</Drawer>
|
||
)}
|
||
</div>
|
||
);
|
||
};
|
||
|
||
export default ChapterReader; |