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

// 修复所有多行的 RedisService.set 调用
// 使用更通用的方法：找到所有 RedisService.set 调用，然后移除 'EX' 参数

// 匹配所有 RedisService.set 调用（包括多行）
const setPattern = /RedisService\.set\(\s*\n\s*['"`]([^'"`]+)['"`],\s*\n\s*([^,]+),\s*\n\s*'EX',\s*\n\s*(\d+)(?:\s*\/\/[^\n]*)?\s*\n\s*\)/g;

content = content.replace(setPattern, (match, key, value, expire) => {
  return `RedisService.set(\n        '${key}',\n        ${value},\n        ${expire}\n      )`;
});

// 也处理反引号版本
const setPattern2 = /RedisService\.set\(\s*\n\s*`([^`]+)`,\s*\n\s*([^,]+),\s*\n\s*'EX',\s*\n\s*(\d+)(?:\s*\/\/[^\n]*)?\s*\n\s*\)/g;

content = content.replace(setPattern2, (match, key, value, expire) => {
  return `RedisService.set(\n        \`${key}\`,\n        ${value},\n        ${expire}\n      )`;
});

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