151 lines
3.8 KiB
JavaScript
151 lines
3.8 KiB
JavaScript
/**
|
||
* buffer/scripts/auto-router.js
|
||
*
|
||
* 自动分流：根据消息来源和 dev_id 归档到对应目录
|
||
*
|
||
* 守护: PER-ZY001 铸渊
|
||
* 系统: SYS-GLW-0001
|
||
* 版权: 国作登字-2026-A-00037559
|
||
*/
|
||
|
||
const fs = require('fs');
|
||
const path = require('path');
|
||
|
||
const BUFFER_ROOT = path.join(__dirname, '..');
|
||
const INBOX = path.join(BUFFER_ROOT, 'inbox');
|
||
const CONFIG_PATH = path.join(BUFFER_ROOT, 'config', 'buffer-config.json');
|
||
|
||
// 来源 → 处理方式
|
||
const ROUTING_RULES = {
|
||
'gemini': { target: 'inbox/{dev_id}/', merge: true },
|
||
'notion': { target: 'inbox/{dev_id}/', merge: true },
|
||
'github': { target: 'inbox/{dev_id}/', merge: true },
|
||
'skyeye': { target: 'inbox/{dev_id}/', merge: false },
|
||
'system': { target: 'inbox/system/', merge: false },
|
||
};
|
||
|
||
const PRIORITY_RULES = {
|
||
'urgent': { flush_immediately: false, flag: true },
|
||
'normal': { flush_immediately: false, flag: false },
|
||
};
|
||
|
||
/**
|
||
* 核心铁律检查：禁止直接写入 grid-db/
|
||
*/
|
||
function validateNoDirectWrite(targetPath) {
|
||
if (targetPath.includes('grid-db/') || targetPath.includes('grid-db\\')) {
|
||
throw new Error(
|
||
'[BLOCKED] 违反铁律：禁止直接写入 grid-db/。' +
|
||
'所有写入必须经过 buffer/ 缓冲层。'
|
||
);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 校验消息格式
|
||
*/
|
||
function validateMessage(msg) {
|
||
const errors = [];
|
||
|
||
if (!msg.message_id) {
|
||
errors.push('Missing message_id');
|
||
}
|
||
if (!msg.dev_id || !/^DEV-\d{3}$/.test(msg.dev_id)) {
|
||
errors.push(`Invalid dev_id: ${msg.dev_id}`);
|
||
}
|
||
if (!msg.source || !ROUTING_RULES[msg.source]) {
|
||
errors.push(`Invalid source: ${msg.source}`);
|
||
}
|
||
if (!msg.type) {
|
||
errors.push('Missing type');
|
||
}
|
||
if (!msg.created_at) {
|
||
errors.push('Missing created_at');
|
||
}
|
||
|
||
return errors;
|
||
}
|
||
|
||
/**
|
||
* 路由消息到正确的 inbox 子目录
|
||
*/
|
||
function routeMessage(msg) {
|
||
const rule = ROUTING_RULES[msg.source] || ROUTING_RULES['system'];
|
||
let targetDir;
|
||
|
||
if (msg.source === 'system' || !msg.dev_id) {
|
||
targetDir = path.join(INBOX, 'system');
|
||
} else {
|
||
targetDir = path.join(INBOX, msg.dev_id);
|
||
}
|
||
|
||
// 铁律检查
|
||
validateNoDirectWrite(targetDir);
|
||
|
||
// 确保目录存在
|
||
if (!fs.existsSync(targetDir)) {
|
||
fs.mkdirSync(targetDir, { recursive: true });
|
||
}
|
||
|
||
// 生成文件名
|
||
const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
|
||
const filename = `${timestamp}-${msg.type}.json`;
|
||
const filePath = path.join(targetDir, filename);
|
||
|
||
// 标记优先级
|
||
const priority = PRIORITY_RULES[msg.priority] || PRIORITY_RULES['normal'];
|
||
if (priority.flag) {
|
||
msg._urgent_flag = true;
|
||
}
|
||
|
||
// 确保状态为 pending
|
||
msg.status = 'pending';
|
||
|
||
fs.writeFileSync(filePath, JSON.stringify(msg, null, 2) + '\n');
|
||
console.log(`[auto-router] Routed ${msg.message_id} → ${path.relative(BUFFER_ROOT, filePath)}`);
|
||
|
||
return { filePath, merge: rule.merge };
|
||
}
|
||
|
||
/**
|
||
* 处理从标准输入或命令行参数传入的消息
|
||
*/
|
||
function main() {
|
||
const args = process.argv.slice(2);
|
||
|
||
if (args.length === 0) {
|
||
console.log('[auto-router] Usage: node auto-router.js <message.json>');
|
||
console.log('[auto-router] Or pipe JSON via stdin');
|
||
process.exit(0);
|
||
}
|
||
|
||
const inputPath = args[0];
|
||
if (!fs.existsSync(inputPath)) {
|
||
console.error(`[auto-router] File not found: ${inputPath}`);
|
||
process.exit(1);
|
||
}
|
||
|
||
let msg;
|
||
try {
|
||
msg = JSON.parse(fs.readFileSync(inputPath, 'utf8'));
|
||
} catch (err) {
|
||
console.error(`[auto-router] Failed to parse ${inputPath}: ${err.message}`);
|
||
process.exit(1);
|
||
}
|
||
|
||
const errors = validateMessage(msg);
|
||
if (errors.length > 0) {
|
||
console.error(`[auto-router] Validation failed: ${errors.join(', ')}`);
|
||
process.exit(1);
|
||
}
|
||
|
||
routeMessage(msg);
|
||
}
|
||
|
||
// Export for use by other scripts
|
||
module.exports = { routeMessage, validateMessage, validateNoDirectWrite, ROUTING_RULES };
|
||
|
||
if (require.main === module) {
|
||
main();
|
||
}
|