modify-app.js 768 B

1234567891011121314151617181920212223
  1. const fs = require('fs');
  2. const file = 'src/app.ts';
  3. let content = fs.readFileSync(file, 'utf8');
  4. // 1. 添加导入
  5. content = content.replace(
  6. "import { queueService } from './services/queue.service';",
  7. `import { queueService } from './services/queue.service';
  8. import { initBookGenerationQueue } from './modules/book-generator/book-queue.processor';`
  9. );
  10. // 2. 在服务器启动后初始化队列处理器
  11. content = content.replace(
  12. "console.log(`💡 提示: 修改 .env 中的 STORAGE_TYPE 可切换存储模式`);",
  13. `console.log(\`💡 提示: 修改 .env 中的 STORAGE_TYPE 可切换存储模式\`);
  14. // 初始化书籍生成队列处理器
  15. initBookGenerationQueue();`
  16. );
  17. fs.writeFileSync(file, content, 'utf8');
  18. console.log('✅ app.ts 已更新');