NotebookLM MCP Server

/** * NotebookLM Library Manager * * Manages a persistent library of NotebookLM notebooks. * Allows Claude to autonomously add, remove, and switch between * multiple notebooks based on the task at hand. */ import fs from "fs"; import path from "path"; import { CONFIG } from "../config.js"; import { log } from "../utils/logger.js"; import type { NotebookEntry, Library, AddNotebookInput, UpdateNotebookInput, LibraryStats, } from "./types.js"; export class NotebookLibrary { private libraryPath: string; private library: Library; constructor() { this.libraryPath = path.join(CONFIG.dataDir, "library.json"); this.library = this.loadLibrary(); log.info("📚 NotebookLibrary initialized"); log.info(` Library path: ${this.libraryPath}`); log.info(` Notebooks: ${this.library.notebooks.length}`); if (this.library.active_notebook_id) { log.info(` Active: ${this.library.active_notebook_id}`); } } /** * Load library from disk, or create default if not exists */ private loadLibrary(): Library { try { if (fs.existsSync(this.libraryPath)) { const data = fs.readFileSync(this.libraryPath, "utf-8"); const library = JSON.parse(data) as Library; log.success(` ✅ Loaded library with ${library.notebooks.length} notebooks`); return library; } } catch (error) { log.warning(` ⚠️ Failed to load library: ${error}`); } // Create default library with current CONFIG as first entry log.info(" 🆕 Creating new library..."); const defaultLibrary = this.createDefaultLibrary(); this.saveLibrary(defaultLibrary); return defaultLibrary; } /** * Create default library from current CONFIG */ private createDefaultLibrary(): Library { const hasConfig = CONFIG.notebookUrl && CONFIG.notebookDescription && CONFIG.notebookDescription !== "General knowledge base - configure NOTEBOOK_DESCRIPTION to help Claude understand what's in this notebook"; const notebooks: NotebookEntry[] = []; if (hasConfig) { // Create first entry from CONFIG const id = this.generateId(CONFIG.notebookDescription); notebooks.push({ id, url: CONFIG.notebookUrl, name: CONFIG.notebookDescription.substring(0, 50), // First 50 chars as name description: CONFIG.notebookDescription, topics: CONFIG.notebookTopics, content_types: CONFIG.notebookContentTypes, use_cases: CONFIG.notebookUseCases, added_at: new Date().toISOString(), last_used: new Date().toISOString(), use_count: 0, tags: [], }); log.success(` ✅ Created default notebook: ${id}`); } return { notebooks, active_notebook_id: notebooks.length > 0 ? notebooks[0].id : null, last_modified: new Date().toISOString(), version: "1.0.0", }; } /** * Save library to disk */ private saveLibrary(library: Library): void { try { library.last_modified = new Date().toISOString(); const data = JSON.stringify(library, null, 2); fs.writeFileSync(this.libraryPath, data, "utf-8"); this.library = library; log.success(` 💾 Library saved (${library.notebooks.length} notebooks)`); } catch (error) { log.error(` ❌ Failed to save library: ${error}`); throw error; } } /** * Generate a unique ID from a string (slug format) */ private generateId(name: string): string { const base = name .toLowerCase() .replace(/[^a-z0-9]+/g, "-") .replace(/^-+|-+$/g, "") .substring(0, 30); // Ensure uniqueness let id = base; let counter = 1; while (this.library.notebooks.some((n) => n.id === id)) { id = `${base}-${counter}`; counter++; } return id; } /** * Add a new notebook to the library */ addNotebook(input: AddNotebookInput): NotebookEntry { log.info(`📝 Adding notebook: ${input.name}`); // Generate ID const id = this.generateId(input.name); // Create entry const notebook: NotebookEntry = { id, url: input.url, name: input.name, description: input.description, topics: input.topics, content_types: input.content_types || ["documentation", "examples"], use_cases: input.use_cases || [ `Learning about ${input.name}`, `Implementing features with ${input.name}`, ], added_at: new Date().toISOString(), last_used: new Date().toISOString(), use_count: 0, tags: input.tags || [], }; // Add to library const updated = { ...this.library }; updated.notebooks.push(notebook); // Set as active if it's the first notebook if (updated.notebooks.length === 1) { updated.active_notebook_id = id; } this.saveLibrary(updated); log.success(`✅ Notebook added: ${id}`); return notebook; } /** * List all notebooks in library */ listNotebooks(): NotebookEntry[] { return this.library.notebooks; } /** * Get a specific notebook by ID */ getNotebook(id: string): NotebookEntry | null { return this.library.notebooks.find((n) => n.id === id) || null; } /** * Get the currently active notebook */ getActiveNotebook(): NotebookEntry | null { if (!this.library.active_notebook_id) { return null; } return this.getNotebook(this.library.active_notebook_id); } /** * Select a notebook as active */ selectNotebook(id: string): NotebookEntry { const notebook = this.getNotebook(id); if (!notebook) { throw new Error(`Notebook not found: ${id}`); } log.info(`🎯 Selecting notebook: ${id}`); const updated = { ...this.library }; updated.active_notebook_id = id; // Update last_used const notebookIndex = updated.notebooks.findIndex((n) => n.id === id); updated.notebooks[notebookIndex] = { ...notebook, last_used: new Date().toISOString(), }; this.saveLibrary(updated); log.success(`✅ Active notebook: ${id}`); return updated.notebooks[notebookIndex]; } /** * Update notebook metadata */ updateNotebook(input: UpdateNotebookInput): NotebookEntry { const notebook = this.getNotebook(input.id); if (!notebook) { throw new Error(`Notebook not found: ${input.id}`); } log.info(`📝 Updating notebook: ${input.id}`); const updated = { ...this.library }; const index = updated.notebooks.findIndex((n) => n.id === input.id); updated.notebooks[index] = { ...notebook, ...(input.name && { name: input.name }), ...(input.description && { description: input.description }), ...(input.topics && { topics: input.topics }), ...(input.content_types && { content_types: input.content_types }), ...(input.use_cases && { use_cases: input.use_cases }), ...(input.tags && { tags: input.tags }), ...(input.url && { url: input.url }), }; this.saveLibrary(updated); log.success(`✅ Notebook updated: ${input.id}`); return updated.notebooks[index]; } /** * Remove notebook from library */ removeNotebook(id: string): boolean { const notebook = this.getNotebook(id); if (!notebook) { return false; } log.info(`🗑️ Removing notebook: ${id}`); const updated = { ...this.library }; updated.notebooks = updated.notebooks.filter((n) => n.id !== id); // If we removed the active notebook, select another one if (updated.active_notebook_id === id) { updated.active_notebook_id = updated.notebooks.length > 0 ? updated.notebooks[0].id : null; } this.saveLibrary(updated); log.success(`✅ Notebook removed: ${id}`); return true; } /** * Increment use count for a notebook */ incrementUseCount(id: string): NotebookEntry | null { const notebookIndex = this.library.notebooks.findIndex((n) => n.id === id); if (notebookIndex === -1) { return null; } const notebook = this.library.notebooks[notebookIndex]; const updated = { ...this.library }; const updatedNotebook: NotebookEntry = { ...notebook, use_count: notebook.use_count + 1, last_used: new Date().toISOString(), }; updated.notebooks[notebookIndex] = updatedNotebook; this.saveLibrary(updated); return updatedNotebook; } /** * Get library statistics */ getStats(): LibraryStats { const totalQueries = this.library.notebooks.reduce( (sum, n) => sum + n.use_count, 0 ); const mostUsed = this.library.notebooks.reduce((max, n) => n.use_count > (max?.use_count || 0) ? n : max , null as NotebookEntry | null); return { total_notebooks: this.library.notebooks.length, active_notebook: this.library.active_notebook_id, most_used_notebook: mostUsed?.id || null, total_queries: totalQueries, last_modified: this.library.last_modified, }; } /** * Search notebooks by query (searches name, description, topics) */ searchNotebooks(query: string): NotebookEntry[] { const lowerQuery = query.toLowerCase(); return this.library.notebooks.filter( (n) => n.name.toLowerCase().includes(lowerQuery) || n.description.toLowerCase().includes(lowerQuery) || n.topics.some((t) => t.toLowerCase().includes(lowerQuery)) || n.tags?.some((t) => t.toLowerCase().includes(lowerQuery)) ); } }