马良AI写作初始化仓库

AINoval/lib/models/novel_with_summaries_dto.dart

@@ -0,0 +1,160 @@
+import 'package:ainoval/models/novel_structure.dart';
+import 'package:ainoval/models/scene_summary_dto.dart';
+import 'package:ainoval/utils/logger.dart';
+
+/// 包含场景摘要的小说DTO
+/// 用于映射服务器返回的包含场景摘要的小说结构
+class NovelWithSummariesDto {
+  final Novel novel;
+  final Map<String, List<SceneSummaryDto>> sceneSummariesByChapter;
+
+  NovelWithSummariesDto({
+    required this.novel,
+    required this.sceneSummariesByChapter,
+  });
+
+  /// 从JSON创建NovelWithSummariesDto实例
+  factory NovelWithSummariesDto.fromJson(Map<String, dynamic> json) {
+    try {
+      AppLogger.i('NovelWithSummariesDto', '开始解析小说和场景摘要数据');
+      
+      // 确保novel字段存在且是Map类型
+      if (!json.containsKey('novel') || !(json['novel'] is Map<String, dynamic>)) {
+        AppLogger.w('NovelWithSummariesDto', '返回数据中缺少novel字段或格式不正确');
+        throw FormatException('返回数据缺少novel字段或格式不正确');
+      }
+      
+      // 解析小说基本信息
+      final novelJson = json['novel'] as Map<String, dynamic>;
+      
+      // 确保结构字段正确，特别是acts字段
+      if (novelJson.containsKey('structure') && novelJson['structure'] is Map) {
+        final structureMap = novelJson['structure'] as Map<String, dynamic>;
+        
+        // 检查并确保acts字段是List类型
+        if (structureMap.containsKey('acts') && !(structureMap['acts'] is List)) {
+          AppLogger.w('NovelWithSummariesDto', 'novel.structure.acts不是列表类型，正在修正');
+          structureMap['acts'] = <Map<String, dynamic>>[];
+        }
+      } else {
+        // 如果没有structure字段或不是Map类型，添加一个空的structure
+        novelJson['structure'] = {'acts': <Map<String, dynamic>>[]};
+        AppLogger.w('NovelWithSummariesDto', '返回数据中缺少novel.structure字段，已添加空结构');
+      }
+      
+      // 解析Novel
+      final novel = Novel.fromJson(novelJson);
+      AppLogger.i('NovelWithSummariesDto', '小说基本信息解析成功: ${novel.title}');
+
+      // 解析场景摘要
+      final sceneSummariesMap = <String, List<SceneSummaryDto>>{};
+      
+      // 检查sceneSummariesByChapter字段是否存在且是Map类型
+      if (json.containsKey('sceneSummariesByChapter') && json['sceneSummariesByChapter'] is Map) {
+        final summariesData = json['sceneSummariesByChapter'] as Map<String, dynamic>;
+        
+        summariesData.forEach((chapterId, summariesList) {
+          if (summariesList is List) {
+            try {
+              final sceneList = <SceneSummaryDto>[];
+              
+              for (var summaryItem in summariesList) {
+                if (summaryItem is Map<String, dynamic>) {
+                  sceneList.add(SceneSummaryDto.fromJson(summaryItem));
+                } else {
+                  AppLogger.w('NovelWithSummariesDto', '场景摘要数据格式错误: $summaryItem');
+                }
+              }
+              
+              if (sceneList.isNotEmpty) {
+                sceneSummariesMap[chapterId] = sceneList;
+              }
+            } catch (e) {
+              AppLogger.e('NovelWithSummariesDto', '解析章节 $chapterId 的场景摘要失败', e);
+            }
+          } else {
+            AppLogger.w('NovelWithSummariesDto', '章节 $chapterId 的场景摘要不是列表格式');
+          }
+        });
+      } else {
+        AppLogger.w('NovelWithSummariesDto', '返回数据中缺少sceneSummariesByChapter字段或格式不正确');
+      }
+
+      AppLogger.i('NovelWithSummariesDto', '解析完成，共有 ${sceneSummariesMap.length} 个章节包含场景摘要');
+      return NovelWithSummariesDto(
+        novel: novel,
+        sceneSummariesByChapter: sceneSummariesMap,
+      );
+    } catch (e) {
+      AppLogger.e('NovelWithSummariesDto', '从JSON创建NovelWithSummariesDto实例失败', e);
+      
+      // 尝试创建一个空的对象，确保不会完全失败
+      try {
+        if (json.containsKey('novel') && json['novel'] is Map<String, dynamic>) {
+          // 尝试只解析小说部分
+          final novel = Novel.fromJson(json['novel'] as Map<String, dynamic>);
+          return NovelWithSummariesDto(
+            novel: novel,
+            sceneSummariesByChapter: {},
+          );
+        }
+      } catch (_) {
+        // 如果还是失败，创建一个完全空的对象
+        AppLogger.e('NovelWithSummariesDto', '尝试创建备用对象也失败');
+      }
+      
+      rethrow;
+    }
+  }
+
+  /// 将DTO中的场景摘要信息合并到Novel模型中
+  Novel mergeSceneSummariesToNovel() {
+    try {
+      // 创建小说的副本，避免修改原始模型
+      Novel updatedNovel = novel;
+
+      // 遍历小说中的卷和章节
+      final List<Act> updatedActs = novel.acts.map((act) {
+        final List<Chapter> updatedChapters = act.chapters.map((chapter) {
+          // 检查这个章节是否有场景摘要
+          if (sceneSummariesByChapter.containsKey(chapter.id)) {
+            final summaries = sceneSummariesByChapter[chapter.id]!;
+            
+            // 根据场景摘要创建场景对象
+            final List<Scene> scenes = summaries.map((summaryDto) {
+              return Scene(
+                id: summaryDto.id,
+                content: '', // 摘要模式下不需要完整内容
+                wordCount: summaryDto.wordCount,
+                summary: Summary(
+                  id: '${summaryDto.id}_summary',
+                  content: summaryDto.summary,
+                ),
+                lastEdited: summaryDto.updatedAt,
+                title: summaryDto.title,
+                chapterId: summaryDto.chapterId,
+              );
+            }).toList();
+            
+            // 创建更新后的章节
+            return chapter.copyWith(scenes: scenes);
+          }
+          
+          // 如果没有摘要信息，保持原样
+          return chapter;
+        }).toList();
+        
+        // 创建更新后的卷
+        return act.copyWith(chapters: updatedChapters);
+      }).toList();
+      
+      // 创建更新后的小说
+      updatedNovel = updatedNovel.copyWith(acts: updatedActs);
+      
+      return updatedNovel;
+    } catch (e) {
+      AppLogger.e('NovelWithSummariesDto', '合并场景摘要到Novel模型失败', e);
+      return novel;  // 出错时返回原始小说模型
+    }
+  }
+}