const fs = require('fs');

const file = 'src/app.ts';
let content = fs.readFileSync(file, 'utf8');

// 1. 添加导入
content = content.replace(
  "import { queueService } from './services/queue.service';",
  `import { queueService } from './services/queue.service';
import { initBookGenerationQueue } from './modules/book-generator/book-queue.processor';`
);

// 2. 在服务器启动后初始化队列处理器
content = content.replace(
  "console.log(`💡 提示: 修改 .env 中的 STORAGE_TYPE 可切换存储模式`);",
  `console.log(\`💡 提示: 修改 .env 中的 STORAGE_TYPE 可切换存储模式\`);

// 初始化书籍生成队列处理器
initBookGenerationQueue();`
);

fs.writeFileSync(file, content, 'utf8');
console.log('✅ app.ts 已更新');
