75 lines
2.1 KiB
JavaScript
75 lines
2.1 KiB
JavaScript
|
/**
|
|||
|
|
* 记忆 读写 · /brain/memory
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
const { Router } = require('express');
|
|||
|
|
const crypto = require('crypto');
|
|||
|
|
const router = Router();
|
|||
|
|
|
|||
|
|
// GET /brain/memory - 查询记忆（支持persona_id和type过滤）
|
|||
|
|
router.get('/', (req, res) => {
|
|||
|
|
try {
|
|||
|
|
let sql = 'SELECT * FROM persona_memory WHERE 1=1';
|
|||
|
|
const params = [];
|
|||
|
|
|
|||
|
|
if (req.query.persona_id) {
|
|||
|
|
sql += ' AND persona_id = ?';
|
|||
|
|
params.push(req.query.persona_id);
|
|||
|
|
}
|
|||
|
|
if (req.query.type) {
|
|||
|
|
sql += ' AND type = ?';
|
|||
|
|
params.push(req.query.type);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
sql += ' ORDER BY timestamp DESC';
|
|||
|
|
|
|||
|
|
const rows = req.db.prepare(sql).all(...params);
|
|||
|
|
const parsed = rows.map(parseJsonFields);
|
|||
|
|
res.json({ data: parsed, total: parsed.length });
|
|||
|
|
} catch (err) {
|
|||
|
|
res.status(500).json({ error: true, message: err.message });
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// POST /brain/memory - 写入新记忆
|
|||
|
|
router.post('/', (req, res) => {
|
|||
|
|
try {
|
|||
|
|
const { persona_id, type, title, content, importance,
|
|||
|
|
related_dev, related_broadcast, tags, timestamp } = req.body;
|
|||
|
|
|
|||
|
|
if (!persona_id || !type || !title || !content || !importance || !timestamp) {
|
|||
|
|
return res.status(400).json({
|
|||
|
|
error: true,
|
|||
|
|
message: 'Missing required fields: persona_id, type, title, content, importance, timestamp'
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const memory_id = `MEM-${Date.now()}-${crypto.randomBytes(4).toString('hex')}`;
|
|||
|
|
|
|||
|
|
req.db.prepare(`
|
|||
|
|
INSERT INTO persona_memory
|
|||
|
|
(memory_id, persona_id, type, title, content, importance,
|
|||
|
|
related_dev, related_broadcast, tags, timestamp)
|
|||
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|||
|
|
`).run(
|
|||
|
|
memory_id, persona_id, type, title, content, importance,
|
|||
|
|
related_dev || null, related_broadcast || null,
|
|||
|
|
tags ? JSON.stringify(tags) : null, timestamp
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
res.status(201).json({ data: { memory_id }, message: 'Memory created' });
|
|||
|
|
} catch (err) {
|
|||
|
|
res.status(500).json({ error: true, message: err.message });
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
function parseJsonFields(row) {
|
|||
|
|
const result = { ...row };
|
|||
|
|
if (result.tags) {
|
|||
|
|
try { result.tags = JSON.parse(result.tags); } catch (_e) { /* keep raw */ }
|
|||
|
|
}
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = router;
|