39 lines
1.3 KiB
JavaScript
39 lines
1.3 KiB
JavaScript
|
const fs = require('fs');
|
||
|
|
let content = fs.readFileSync('src/core/security/SecurityHardeningService.ts', 'utf8');
|
||
|
|
|
||
|
|
// 1. 修改导入语句
|
||
|
|
content = content.replace(
|
||
|
|
"import { RedisService } from '../cache/RedisService';",
|
||
|
|
"import RedisService from '../../services/RedisService';"
|
||
|
|
);
|
||
|
|
|
||
|
|
// 2. 修改构造函数 - 移除 redisService 参数
|
||
|
|
content = content.replace(
|
||
|
|
/constructor\(\s*private readonly configService: ConfigService,\s*private readonly redisService: RedisService,\s*\) \{\}/,
|
||
|
|
'constructor(\n private readonly configService: ConfigService,\n ) {}'
|
||
|
|
);
|
||
|
|
|
||
|
|
// 3. 修改 securityCheckInterval 属性
|
||
|
|
content = content.replace(
|
||
|
|
'private securityCheckInterval: NodeJS.Timeout;',
|
||
|
|
'private securityCheckInterval!: NodeJS.Timeout;'
|
||
|
|
);
|
||
|
|
|
||
|
|
// 4. 替换所有 this.redisService. 为 RedisService.
|
||
|
|
content = content.replace(/this\.redisService\./g, 'RedisService.');
|
||
|
|
|
||
|
|
// 5. 修复 RedisService.set 调用 - 移除 'EX' 参数
|
||
|
|
content = content.replace(
|
||
|
|
/RedisService\.set\(\s*'([^']+)',\s*([^,]+),\s*'EX',\s*(\d+)\s*\)/g,
|
||
|
|
"RedisService.set('$1', $2, $3)"
|
||
|
|
);
|
||
|
|
|
||
|
|
// 处理多行的情况
|
||
|
|
content = content.replace(
|
||
|
|
/RedisService\.set\(\s*'([^']+)',\s*([^,]+),\s*'EX',\s*\n\s*(\d+)\s*\)/g,
|
||
|
|
"RedisService.set('$1', $2, $3)"
|
||
|
|
);
|
||
|
|
|
||
|
|
fs.writeFileSync('src/core/security/SecurityHardeningService.ts', content);
|
||
|
|
console.log('File updated successfully');
|