const fs = require('fs');
let content = fs.readFileSync('src/core/security/SecurityHardeningService.ts', 'utf8');

// 修复多行的 RedisService.set 调用
// 匹配格式：
// RedisService.set(
//   'key',
//   value,
//   'EX',
//   3600
// );
content = content.replace(
  /RedisService\.set\(\s*\n\s*`([^`]+)`,\s*\n\s*([^,]+),\s*\n\s*'EX',\s*\n\s*(\d+)\s*\n\s*\)/g,
  "RedisService.set(\n        `$1`,\n        $2,\n        $3\n      )"
);

// 匹配单引号版本
content = content.replace(
  /RedisService\.set\(\s*\n\s*'([^']+)',\s*\n\s*([^,]+),\s*\n\s*'EX',\s*\n\s*(\d+)\s*\n\s*\)/g,
  "RedisService.set(\n        '$1',\n        $2,\n        $3\n      )"
);

fs.writeFileSync('src/core/security/SecurityHardeningService.ts', content);
console.log('File updated successfully');