const fs = require('fs');
const path = require('path');

const filePath = path.join(__dirname, 'src/core/security/SecurityHardeningService.ts');
let content = fs.readFileSync(filePath, 'utf8');

// 替换所有 this.redisService. 为 RedisService.
content = content.replace(/this\.redisService\./g, 'RedisService.');

// 修复 RedisService.set 调用 - 移除 'EX' 参数
// 处理多行格式
content = content.replace(
  /RedisService\.set\(\s*\n\s*`([^`]+)`,\s*\n\s*([^,]+),\s*\n\s*'EX',\s*\n\s*(\d+)(?:\s*\/\/[^\n]*)?\s*\n\s*\)/g,
  (match, key, value, expire) => {
    return `RedisService.set(\n        \`${key}\`,\n        ${value},\n        ${expire}\n      )`;
  }
);

content = content.replace(
  /RedisService\.set\(\s*\n\s*'([^']+)',\s*\n\s*([^,]+),\s*\n\s*'EX',\s*\n\s*(\d+)(?:\s*\/\/[^\n]*)?\s*\n\s*\)/g,
  (match, key, value, expire) => {
    return `RedisService.set(\n        '${key}',\n        ${value},\n        ${expire}\n      )`;
  }
);

// 处理单行格式
content = content.replace(
  /RedisService\.set\('([^']+)', ([^,]+), 'EX', (\d+)\)/g,
  "RedisService.set('$1', $2, $3)"
);

content = content.replace(
  /RedisService\.set\(`([^`]+)`, ([^,]+), 'EX', (\d+)\)/g,
  "RedisService.set(`$1`, $2, $3)"
);

fs.writeFileSync(filePath, content, 'utf8');
console.log('File updated successfully');
console.log('Total lines:', content.split('\n').length);