notebook.js•3.94 kB
// notebook.js - User knowledge library management and storage functions
import { ensureSession, trimSessionSize, updateNotebookSize } from './session.js';
import { persistSessions } from './storage.js';
import { stripHTML, validateNotebookEntry } from './utils.js';
// ===== NOTEBOOK SAFETY LIMITS =====
const NOTEBOOK_LIMITS = {
MAX_ENTRIES: 30,
MAX_TOTAL_SIZE: 30000,
RATE_LIMIT_MS: 300,
MAX_ENTRY_SIZE: 2048
};
let lastNotebookSave = 0;
function rateLimitedAddEntry(id, key, value) {
const now = Date.now();
if (now - lastNotebookSave < NOTEBOOK_LIMITS.RATE_LIMIT_MS) {
return '⚠️ Please wait before saving another notebook entry.';
}
lastNotebookSave = now;
return addNotebookEntry(id, key, value);
}
function addNotebookEntry(id, key, value) {
const s = ensureSession(id);
key = stripHTML(key);
value = stripHTML(value);
if (!validateNotebookEntry(key, value)) {
return 'Invalid notebook entry. Key must be 1-64 chars (letters, numbers, dash, space). Value must be 1-2048 chars.';
}
const currentEntries = Object.keys(s.notebook).length;
const isNewEntry = !s.notebook.hasOwnProperty(key);
if (isNewEntry && currentEntries >= NOTEBOOK_LIMITS.MAX_ENTRIES) {
return `⚠️ Notebook is full (${NOTEBOOK_LIMITS.MAX_ENTRIES} entries max). Delete some entries first.`;
}
// --- Notebook Entry Size & Storage Limit Check ---
const newEntrySize = JSON.stringify({ [key]: value }).length;
const oldEntrySize = s.notebook[key] ? JSON.stringify({ [key]: s.notebook[key] }).length : 0;
const sizeDelta = newEntrySize - oldEntrySize;
if (s.notebookSize + sizeDelta > NOTEBOOK_LIMITS.MAX_TOTAL_SIZE) {
return `⚠️ Notebook storage limit reached (${NOTEBOOK_LIMITS.MAX_TOTAL_SIZE} chars max). Delete some entries first.`;
}
s.notebook[key] = value;
updateNotebookSize(id, sizeDelta);
trimSessionSize(s);
persistSessions();
return `Stored "${key}" → "${value}"`;
}
// --- Notebook Entry Retrieval & Listing ---
function getNotebookEntry(id, key) {
const s = ensureSession(id);
if (!key) return 'Usage: /notebook add:[name] [data] | /notebook use:[name] | /notebook show:';
return s.notebook[key] ? `"${key}": ${s.notebook[key]}` : `No entry for "${key}"`;
}
function listNotebookEntries(id) {
const s = ensureSession(id);
const keys = Object.keys(s.notebook);
if (!keys.length) return 'Notebook is empty.';
const usage = `📊 Usage: ${keys.length}/${NOTEBOOK_LIMITS.MAX_ENTRIES} entries, ${s.notebookSize}/${NOTEBOOK_LIMITS.MAX_TOTAL_SIZE} chars`;
const entries = keys.map(k => `• ${k}: ${s.notebook[k]}`).join('\n');
return `${usage}\n\nNotebook entries:\n${entries}`;
}
// --- Notebook Entry Deletion ---
function deleteNotebookEntry(id, key) {
const s = ensureSession(id);
if (!key) return 'Usage: /notebook delete:[name]';
if (!s.notebook[key]) {
return `No entry found for "${key}"`;
}
const deletedEntrySize = JSON.stringify({ [key]: s.notebook[key] }).length;
delete s.notebook[key];
updateNotebookSize(id, -deletedEntrySize);
persistSessions();
return `Deleted "${key}" from notebook`;
}
// --- Clear All Notebook Entries ---
function clearNotebook(id) {
const s = ensureSession(id);
const entryCount = Object.keys(s.notebook).length;
if (entryCount === 0) {
return 'Notebook is already empty.';
}
s.notebook = {};
s.notebookSize = 0;
persistSessions();
return `Cleared all ${entryCount} entries from notebook.`;
}
export function manageUserNotebook(id, action, key = '', value = '') {
switch (action) {
case 'add':
return rateLimitedAddEntry(id, key, value);
case 'all':
return listNotebookEntries(id);
case 'clear':
return clearNotebook(id);
default:
return 'Usage: /notebook add:[name] [data] | /notebook use:[name] | /notebook show: | /notebook clear: | /notebook status:';
}
}