@roomi-fields/notebooklm-mcp 1.3.5 → 1.5.0

package/dist/http-wrapper.js

@@ -5,6 +5,7 @@
  * Allows n8n and other tools to call the server without stdio
  */
 import express from 'express';
+import { randomUUID } from 'crypto';
 import { AuthManager } from './auth/auth-manager.js';
 import { SessionManager } from './session/session-manager.js';
 import { NotebookLibrary } from './library/notebook-library.js';
@@ -12,7 +13,15 @@ import { ToolHandlers } from './tools/index.js';
 import { AutoDiscovery } from './auto-discovery/auto-discovery.js';
 import { log } from './utils/logger.js';
 const app = express();
-app.use(express.json());
+app.use(express.json({ limit: '10mb' }));
+// Request ID middleware for debugging and log correlation
+app.use((req, res, next) => {
+    // Use existing X-Request-ID header or generate a new one
+    const requestId = req.headers['x-request-id'] || randomUUID();
+    req.requestId = requestId;
+    res.setHeader('X-Request-ID', requestId);
+    next();
+});
 // CORS for n8n
 app.use((_req, res, next) => {
     res.header('Access-Control-Allow-Origin', '*');
@@ -43,20 +52,25 @@ app.get('/health', async (_req, res) => {
 });
 // Ask question
 app.post('/ask', async (req, res) => {
+    const reqId = req.requestId.substring(0, 8); // Short ID for logs
     try {
         const { question, session_id, notebook_id, notebook_url, show_browser } = req.body;
         if (!question) {
+            log.warning(`[${reqId}] /ask - Missing question`);
             return res.status(400).json({
                 success: false,
                 error: 'Missing required field: question',
             });
         }
+        log.info(`[${reqId}] /ask - "${question.substring(0, 50)}..."`);
         const result = await toolHandlers.handleAskQuestion({ question, session_id, notebook_id, notebook_url, show_browser }, async (message, progress, total) => {
-            log.info(`Progress: ${message} (${progress}/${total})`);
+            log.info(`[${reqId}] Progress: ${message} (${progress}/${total})`);
         });
+        log.success(`[${reqId}] /ask - Completed`);
         res.json(result);
     }
     catch (error) {
+        log.error(`[${reqId}] /ask - Error: ${error instanceof Error ? error.message : String(error)}`);
         res.status(500).json({
             success: false,
             error: error instanceof Error ? error.message : String(error),
@@ -163,10 +177,17 @@ app.post('/notebooks', async (req, res) => {
         });
     }
 });
-// Get notebook
-app.get('/notebooks/:id', async (req, res) => {
+// Search notebooks (MUST be before /notebooks/:id to avoid being shadowed)
+app.get('/notebooks/search', async (req, res) => {
     try {
-        const result = await toolHandlers.handleGetNotebook({ id: req.params.id });
+        const { query } = req.query;
+        if (typeof query !== 'string' || !query.trim()) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing or invalid query parameter',
+            });
+        }
+        const result = await toolHandlers.handleSearchNotebooks({ query });
         res.json(result);
     }
     catch (error) {
@@ -176,13 +197,10 @@ app.get('/notebooks/:id', async (req, res) => {
         });
     }
 });
-// Update notebook
-app.put('/notebooks/:id', async (req, res) => {
+// Get library stats (MUST be before /notebooks/:id to avoid being shadowed)
+app.get('/notebooks/stats', async (_req, res) => {
     try {
-        const result = await toolHandlers.handleUpdateNotebook({
-            id: req.params.id,
-            ...req.body,
-        });
+        const result = await toolHandlers.handleGetLibraryStats();
         res.json(result);
     }
     catch (error) {
@@ -192,10 +210,10 @@ app.put('/notebooks/:id', async (req, res) => {
         });
     }
 });
-// Delete notebook
-app.delete('/notebooks/:id', async (req, res) => {
+// Get notebook
+app.get('/notebooks/:id', async (req, res) => {
     try {
-        const result = await toolHandlers.handleRemoveNotebook({ id: req.params.id });
+        const result = await toolHandlers.handleGetNotebook({ id: req.params.id });
         res.json(result);
     }
     catch (error) {
@@ -205,12 +223,12 @@ app.delete('/notebooks/:id', async (req, res) => {
         });
     }
 });
-// Search notebooks
-app.get('/notebooks/search', async (req, res) => {
+// Update notebook
+app.put('/notebooks/:id', async (req, res) => {
     try {
-        const { query } = req.query;
-        const result = await toolHandlers.handleSearchNotebooks({
-            query: query,
+        const result = await toolHandlers.handleUpdateNotebook({
+            id: req.params.id,
+            ...req.body,
         });
         res.json(result);
     }
@@ -221,10 +239,10 @@ app.get('/notebooks/search', async (req, res) => {
         });
     }
 });
-// Get library stats
-app.get('/notebooks/stats', async (_req, res) => {
+// Delete notebook
+app.delete('/notebooks/:id', async (req, res) => {
     try {
-        const result = await toolHandlers.handleGetLibraryStats();
+        const result = await toolHandlers.handleRemoveNotebook({ id: req.params.id });
         res.json(result);
     }
     catch (error) {
@@ -245,11 +263,20 @@ app.post('/notebooks/auto-discover', async (req, res) => {
                 error: 'Missing required field: url',
             });
         }
-        // Validate it's a NotebookLM URL
-        if (!url.includes('notebooklm.google.com')) {
+        // Validate it's a NotebookLM URL (proper URL parsing to prevent bypass)
+        try {
+            const parsedUrl = new URL(url);
+            if (parsedUrl.hostname !== 'notebooklm.google.com') {
+                return res.status(400).json({
+                    success: false,
+                    error: 'Invalid URL: must be a NotebookLM URL (notebooklm.google.com)',
+                });
+            }
+        }
+        catch {
             return res.status(400).json({
                 success: false,
-                error: 'Invalid URL: must be a NotebookLM URL (notebooklm.google.com)',
+                error: 'Invalid URL format',
             });
         }
         // Create AutoDiscovery instance and discover metadata
@@ -301,6 +328,22 @@ app.post('/notebooks/auto-discover', async (req, res) => {
         });
     }
 });
+// Create a new notebook in NotebookLM (via browser automation)
+app.post('/notebooks/create', async (req, res) => {
+    try {
+        const { name, show_browser } = req.body;
+        const result = await toolHandlers.handleCreateNotebook({ name, show_browser }, async (message, progress, total) => {
+            log.info(`Progress: ${message} (${progress}/${total})`);
+        });
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
 // Activate notebook (set as active)
 app.put('/notebooks/:id/activate', async (req, res) => {
     try {
@@ -353,11 +396,367 @@ app.post('/sessions/:id/reset', async (req, res) => {
         });
     }
 });
+// ========================================
+// Content Management Routes
+// ========================================
+// Add source to notebook
+app.post('/content/sources', async (req, res) => {
+    try {
+        const { source_type, file_path, url, text, title, notebook_url, session_id, show_browser } = req.body;
+        if (!source_type) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing required field: source_type',
+            });
+        }
+        const result = await toolHandlers.handleAddSource({
+            source_type,
+            file_path,
+            url,
+            text,
+            title,
+            notebook_url,
+            session_id,
+            show_browser,
+        });
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
+// Delete source from notebook
+app.delete('/content/sources/:id', async (req, res) => {
+    try {
+        const { notebook_url, session_id } = req.query;
+        const sourceId = req.params.id;
+        if (!sourceId) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing source ID in URL path',
+            });
+        }
+        const result = await toolHandlers.handleDeleteSource({
+            source_id: sourceId,
+            notebook_url: typeof notebook_url === 'string' ? notebook_url : undefined,
+            session_id: typeof session_id === 'string' ? session_id : undefined,
+        });
+        if (!result.success) {
+            // Return 404 if source not found
+            if (result.error?.includes('not found')) {
+                return res.status(404).json(result);
+            }
+            return res.status(500).json(result);
+        }
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
+// Delete source by name (alternative endpoint)
+app.delete('/content/sources', async (req, res) => {
+    try {
+        const { source_name, source_id, notebook_url, session_id } = req.query;
+        if (!source_name && !source_id) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing required query parameter: source_name or source_id',
+            });
+        }
+        const result = await toolHandlers.handleDeleteSource({
+            source_id: typeof source_id === 'string' ? source_id : undefined,
+            source_name: typeof source_name === 'string' ? source_name : undefined,
+            notebook_url: typeof notebook_url === 'string' ? notebook_url : undefined,
+            session_id: typeof session_id === 'string' ? session_id : undefined,
+        });
+        if (!result.success) {
+            // Return 404 if source not found
+            if (result.error?.includes('not found')) {
+                return res.status(404).json(result);
+            }
+            return res.status(500).json(result);
+        }
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
+// Generate content (audio_overview, presentation, report, data_table, infographic, and video are supported)
+app.post('/content/generate', async (req, res) => {
+    try {
+        const { content_type, custom_instructions, notebook_url, session_id, language, video_style, video_format, infographic_format, report_format, presentation_style, presentation_length, } = req.body;
+        if (!content_type) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing required field: content_type',
+            });
+        }
+        // Validate content_type is supported
+        const supportedTypes = [
+            'audio_overview',
+            'presentation',
+            'report',
+            'infographic',
+            'data_table',
+            'video',
+        ];
+        if (!supportedTypes.includes(content_type)) {
+            return res.status(400).json({
+                success: false,
+                error: `Content type '${content_type}' is not supported. Supported types: ${supportedTypes.join(', ')}. ` +
+                    'These use real NotebookLM Studio UI buttons or the generic ContentGenerator.',
+            });
+        }
+        // Warn if custom_instructions provided for content types that don't support it
+        const noCustomInstructionsTypes = ['report']; // report and mindmap (when implemented) don't support prompts
+        if (custom_instructions && noCustomInstructionsTypes.includes(content_type)) {
+            return res.status(400).json({
+                success: false,
+                error: `Content type '${content_type}' does not support custom_instructions. ` +
+                    `Only format/language options are available for this type.`,
+            });
+        }
+        // Validate video_style if provided (only valid for video content type)
+        const validVideoStyles = [
+            'classroom',
+            'documentary',
+            'animated',
+            'corporate',
+            'cinematic',
+            'minimalist',
+        ];
+        if (video_style && !validVideoStyles.includes(video_style)) {
+            return res.status(400).json({
+                success: false,
+                error: `Video style '${video_style}' is not supported. Supported styles: ${validVideoStyles.join(', ')}.`,
+            });
+        }
+        if (video_style && content_type !== 'video') {
+            return res.status(400).json({
+                success: false,
+                error: `video_style is only valid for content_type 'video', not '${content_type}'.`,
+            });
+        }
+        // Validate video_format if provided
+        if (video_format && !['brief', 'explainer'].includes(video_format)) {
+            return res.status(400).json({
+                success: false,
+                error: `Video format '${video_format}' is not supported. Supported formats: brief, explainer.`,
+            });
+        }
+        // Validate infographic_format if provided
+        if (infographic_format && !['horizontal', 'vertical'].includes(infographic_format)) {
+            return res.status(400).json({
+                success: false,
+                error: `Infographic format '${infographic_format}' is not supported. Supported formats: horizontal, vertical.`,
+            });
+        }
+        // Validate report_format if provided
+        if (report_format && !['summary', 'detailed'].includes(report_format)) {
+            return res.status(400).json({
+                success: false,
+                error: `Report format '${report_format}' is not supported. Supported formats: summary, detailed.`,
+            });
+        }
+        // Validate presentation_style if provided
+        if (presentation_style &&
+            !['detailed_slideshow', 'presenter_notes'].includes(presentation_style)) {
+            return res.status(400).json({
+                success: false,
+                error: `Presentation style '${presentation_style}' is not supported. Supported styles: detailed_slideshow, presenter_notes.`,
+            });
+        }
+        // Validate presentation_length if provided
+        if (presentation_length && !['short', 'default'].includes(presentation_length)) {
+            return res.status(400).json({
+                success: false,
+                error: `Presentation length '${presentation_length}' is not supported. Supported lengths: short, default.`,
+            });
+        }
+        // Note: data_table has no format options - it exports to Google Sheets
+        const result = await toolHandlers.handleGenerateContent({
+            content_type,
+            custom_instructions,
+            notebook_url,
+            session_id,
+            language,
+            video_style,
+            video_format,
+            infographic_format,
+            report_format,
+            presentation_style,
+            presentation_length,
+        });
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
+// List sources and generated content
+app.get('/content', async (req, res) => {
+    try {
+        const { notebook_url, session_id } = req.query;
+        const result = await toolHandlers.handleListContent({
+            notebook_url: typeof notebook_url === 'string' ? notebook_url : undefined,
+            session_id: typeof session_id === 'string' ? session_id : undefined,
+        });
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
+// Download/export content (audio, video, infographic, presentation, data_table)
+app.get('/content/download', async (req, res) => {
+    try {
+        const { content_type, output_path, notebook_url, session_id } = req.query;
+        if (!content_type) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing required field: content_type',
+            });
+        }
+        // Validate content_type is downloadable/exportable
+        // - audio_overview, video, infographic: downloadable as files
+        // - presentation: exports to Google Slides
+        // - data_table: exports to Google Sheets
+        // - report: text-based only (no export)
+        const exportableTypes = [
+            'audio_overview',
+            'video',
+            'infographic',
+            'presentation',
+            'data_table',
+        ];
+        if (!exportableTypes.includes(content_type)) {
+            return res.status(400).json({
+                success: false,
+                error: `Content type '${content_type}' is not exportable. Exportable types: ${exportableTypes.join(', ')}. ` +
+                    'Report content is text-based and returned in the generation response.',
+            });
+        }
+        const result = await toolHandlers.handleDownloadContent({
+            content_type: content_type,
+            output_path: typeof output_path === 'string' ? output_path : undefined,
+            notebook_url: typeof notebook_url === 'string' ? notebook_url : undefined,
+            session_id: typeof session_id === 'string' ? session_id : undefined,
+        });
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
+// Create a note in the notebook
+app.post('/content/notes', async (req, res) => {
+    try {
+        const { title, content, notebook_url, session_id } = req.body;
+        if (!title) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing required field: title',
+            });
+        }
+        if (!content) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing required field: content',
+            });
+        }
+        const result = await toolHandlers.handleCreateNote({
+            title,
+            content,
+            notebook_url,
+            session_id,
+        });
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
+// Save chat/discussion to a note
+app.post('/content/chat-to-note', async (req, res) => {
+    try {
+        const { title, notebook_url, session_id } = req.body;
+        const result = await toolHandlers.handleSaveChatToNote({
+            title,
+            notebook_url,
+            session_id,
+        });
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
+// Convert a note to a source
+app.post('/content/notes/:noteTitle/to-source', async (req, res) => {
+    try {
+        const { noteTitle } = req.params;
+        const { notebook_url, session_id } = req.body;
+        if (!noteTitle) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing required parameter: noteTitle',
+            });
+        }
+        const result = await toolHandlers.handleConvertNoteToSource({
+            note_title: decodeURIComponent(noteTitle),
+            notebook_url,
+            session_id,
+        });
+        res.json(result);
+    }
+    catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+});
+// Global error handler - catches any unhandled errors in async routes
+app.use((err, req, res, _next) => {
+    const reqId = req.requestId?.substring(0, 8) || 'unknown';
+    log.error(`[${reqId}] Unhandled error: ${err.message}`);
+    res.status(500).json({
+        success: false,
+        error: 'Internal server error',
+        requestId: req.requestId,
+    });
+});
 // Start server
 const PORT = Number(process.env.HTTP_PORT) || 3000;
 const HOST = process.env.HTTP_HOST || '0.0.0.0';
 app.listen(PORT, HOST, () => {
-    log.success(`🌐 NotebookLM MCP HTTP Server v1.3.5`);
+    log.success(`🌐 NotebookLM MCP HTTP Server v1.4.2`);
     log.success(`   Listening on ${HOST}:${PORT}`);
     log.info('');
     log.info('📊 Quick Links:');
@@ -391,6 +790,17 @@ app.listen(PORT, HOST, () => {
     log.info('   POST   /sessions/:id/reset     Reset session history');
     log.info('   DELETE /sessions/:id           Close a session');
     log.info('');
+    log.info('   Content Management:');
+    log.info('   POST   /content/sources        Add source to notebook');
+    log.info('   DELETE /content/sources/:id    Delete source by ID');
+    log.info('   DELETE /content/sources        Delete source by name (query param)');
+    log.info('   POST   /content/generate       Generate content (audio, video, etc.)');
+    log.info('   GET    /content/download       Download/export generated content');
+    log.info('   POST   /content/notes          Create a note in the notebook');
+    log.info('   POST   /content/chat-to-note   Save chat/discussion to a note');
+    log.info('   POST   /content/notes/:title/to-source  Convert note to source');
+    log.info('   GET    /content                List sources and content');
+    log.info('');
     log.info('💡 Configuration:');
     log.info(`   Host: ${HOST} ${HOST === '0.0.0.0' ? '(accessible from network)' : '(localhost only)'}`);
     log.info(`   Port: ${PORT}`);
@@ -398,15 +808,25 @@ app.listen(PORT, HOST, () => {
     log.dim('📖 Documentation: ./deployment/docs/');
     log.dim('⏹️  Press Ctrl+C to stop');
 });
-// Graceful shutdown
+// Graceful shutdown with error handling
 process.on('SIGTERM', async () => {
     log.info('SIGTERM received, shutting down gracefully...');
-    await toolHandlers.cleanup();
+    try {
+        await toolHandlers.cleanup();
+    }
+    catch (error) {
+        log.error(`Cleanup failed: ${error}`);
+    }
     process.exit(0);
 });
 process.on('SIGINT', async () => {
     log.info('SIGINT received, shutting down gracefully...');
-    await toolHandlers.cleanup();
+    try {
+        await toolHandlers.cleanup();
+    }
+    catch (error) {
+        log.error(`Cleanup failed: ${error}`);
+    }
     process.exit(0);
 });
 //# sourceMappingURL=http-wrapper.js.map