40 lines
1.3 KiB
JavaScript
40 lines
1.3 KiB
JavaScript
|
|
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);
|