54 lines
1.9 KiB
JavaScript
54 lines
1.9 KiB
JavaScript
|
const fs = require('fs');
|
|||
|
|
const file = 'd:\\trae_projects\\makemd\\makemd\\server\\src\\core\\security\\SecurityHardeningService.ts';
|
|||
|
|
|
|||
|
|
// 读取文件（使用 utf8 编码，不带 BOM）
|
|||
|
|
let content = fs.readFileSync(file, 'utf8');
|
|||
|
|
|
|||
|
|
// 1. 修改导入语句
|
|||
|
|
content = content.replace(
|
|||
|
|
"import { RedisService } from '../cache/RedisService';",
|
|||
|
|
"import RedisService from '../../services/RedisService';"
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
// 2. 修改 securityCheckInterval 属性
|
|||
|
|
content = content.replace(
|
|||
|
|
'private securityCheckInterval: NodeJS.Timeout;',
|
|||
|
|
'private securityCheckInterval!: NodeJS.Timeout;'
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
// 3. 修改构造函数 - 移除 redisService 参数
|
|||
|
|
content = content.replace(
|
|||
|
|
/constructor\(\s*private readonly configService: ConfigService,\s*private readonly redisService: RedisService,\s*\) \{\}/,
|
|||
|
|
'constructor(\n private readonly configService: ConfigService,\n ) {}'
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
// 4. 替换所有 this.redisService. 为 RedisService.
|
|||
|
|
content = content.replace(/this\.redisService\./g, 'RedisService.');
|
|||
|
|
|
|||
|
|
// 5. 修复 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,
|
|||
|
|
"RedisService.set(\n `$1`,\n $2,\n $3\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,
|
|||
|
|
"RedisService.set(\n '$1',\n $2,\n $3\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)"
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
// 写入文件（使用 utf8 编码，不带 BOM）
|
|||
|
|
fs.writeFileSync(file, content, 'utf8');
|
|||
|
|
console.log('File updated successfully');
|