23 lines
1.1 KiB
JavaScript
23 lines
1.1 KiB
JavaScript
|
|
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');
|