const { PrismaClient } = require('@prisma/client');
const http = require('http');

async function resetAndTriggerGeneration() {
  const prisma = new PrismaClient();
  
  try {
    console.log('📝 步骤1: 重置书籍状态...');
    
    // 重置书籍状态
    await prisma.book.update({
      where: { id: 1 },
      data: {
        status: 'draft',
        progress: 10,
        errorMsg: null
      }
    });

    console.log('✅ 状态已重置: draft, 10%');

    // 重置所有章节的内容状态
    const result = await prisma.bookChapter.updateMany({
      where: {
        bookId: 1,
        level: 3  // 只重置小节的内容状态
      },
      data: {
        contentStatus: null,
        content: null,
        wordCount: 0,
        contentError: null
      }
    });

    console.log(`✅ 已重置 ${result.count} 个小节的内容状态`);
    console.log('');

    console.log('🚀 步骤2: 触发内容生成...');
    console.log('请手动调用以下API触发生成:');
    console.log('  POST http://localhost:3000/api/book-generator/langgraph/books/1/generate');
    console.log('');
    console.log('或者在浏览器中访问前端页面点击"继续生成"按钮');

  } catch (error) {
    console.error('错误:', error);
  } finally {
    await prisma.$disconnect();
  }
}

resetAndTriggerGeneration();