fix: submit

novel-app/src/app/chapter-generator/chapter-generator.page.ts

@@ -9,7 +9,6 @@ import { CommonModule } from '@angular/common';
 import { AiExpandModalComponent } from '../ai-expand-modal/ai-expand-modal.component';
 import { AiPolishModalComponent } from '../ai-polish-modal/ai-polish-modal.component';
 import { AiContinueModalComponent } from '../ai-continue-modal/ai-continue-modal.component';
-import { ActivatedRoute } from '@angular/router';
 
 addIcons({ chevronForward });
 
@@ -24,27 +23,29 @@ addIcons({ chevronForward });
   ]
 })
 export class ChapterGeneratorPage implements OnInit {
-  chapters = [
-    { title: 'Chapter 1', content: '这是第一章的内容。' },
-    // 其他章节...
-  ];
+  chapters: { title: string, content: string }[] = [];
   isSideShow: boolean = true;
   selectedChapterIndex: number | null = null;
   selectedChapterTitle: string = '';
   selectedChapterContent: string = '';
-  novelTitle: string = ''; // 初始化作品名称
+  novelTitle: string = '';
+  workId: string | null = null;
 
   constructor(
     private modalCtrl: ModalController,
-    private navCtrl: NavController,
-    private route: ActivatedRoute // 注入 ActivatedRoute 以获取查询参数
+    private navCtrl: NavController
   ) { }
 
   ngOnInit() {
-    // 从查询参数中获取作品名称
-    this.route.queryParams.subscribe(params => {
-      this.novelTitle = params['title'] || '';
-    });
+    // 从本地存储获取当前作品信息
+    const currentWork = localStorage.getItem('currentWork');
+    if (currentWork) {
+      const work = JSON.parse(currentWork);
+      this.workId = work.id;
+      this.novelTitle = work.title;
+      this.chapters = work.chapters;
+      this.selectChapter(0); // 默认选择第一个章节
+    }
   }
 
   async openAiExpandModal() {
@@ -127,18 +128,32 @@ export class ChapterGeneratorPage implements OnInit {
   }
 
   saveWork() {
+    if (!this.workId) {
+      // 如果没有 workId，则创建一个新的作品
+      this.workId = Date.now().toString(); // 使用当前时间戳作为唯一ID
+    }
+
     const work = {
-      id: Date.now().toString(), // 使用当前时间戳作为唯一ID
+      id: this.workId,
       cover: 'default-cover.jpg', // 默认封面，可以根据需要修改
-      title: this.novelTitle, // 使用从查询参数获取的作品名称
+      title: this.novelTitle,
       createTime: new Date(),
       type: '长篇小说',
       chapters: this.chapters,
     };
 
-    // 将作品信息存储在本地存储
+    // 从本地存储获取所有作品信息
     const existingWorks = JSON.parse(localStorage.getItem('works') || '[]');
-    existingWorks.push(work);
+
+    // 找到并更新对应的作品信息
+    const workIndex = existingWorks.findIndex((w: { id: string | null; }) => w.id === this.workId);
+    if (workIndex !== -1) {
+      existingWorks[workIndex] = work;
+    } else {
+      existingWorks.push(work);
+    }
+
+    // 将更新后的工作信息存储在本地存储
     localStorage.setItem('works', JSON.stringify(existingWorks));
 
     // 跳转到home页面

novel-app/src/app/home/home.page.ts

@@ -76,6 +76,7 @@ export class HomePage implements OnInit {
     // 找到对应的作品并存储在本地存储
     const work = this.stories.find(story => story.id === workId);
     if (work) {
+      // 将作品信息存储在本地存储，以便在 chapter-generator 页面加载时使用
       localStorage.setItem('currentWork', JSON.stringify(work));
       this.navCtrl.navigateRoot(['/chapter-generator']);
     }