@roomi-fields/notebooklm-mcp 1.3.6 → 1.5.1

package/deployment/scripts/test-validation.ps1

@@ -0,0 +1,299 @@
+#!/usr/bin/env pwsh
+#Requires -Version 5.1
+
+<#
+.SYNOPSIS
+    Zod schema validation testing script for NotebookLM MCP HTTP Server API
+
+.DESCRIPTION
+    Tests all Zod validation schemas to ensure proper input validation:
+    - AskQuestionSchema
+    - AddNotebookSchema
+    - UpdateNotebookSchema
+    - AutoDiscoverSchema
+    - CleanupDataSchema
+    - ShowBrowserSchema
+
+.PARAMETER BaseUrl
+    Base URL of the server (default: http://localhost:3000)
+
+.EXAMPLE
+    .\test-validation.ps1
+    Runs all validation tests
+
+.NOTES
+    Prerequisite: The server must be started
+#>
+
+param(
+    [string]$BaseUrl = "http://localhost:3000"
+)
+
+# Colors for logs
+function Write-TestHeader {
+    param([string]$Message, [int]$Number, [int]$Total)
+    Write-Host "`n" -NoNewline
+    Write-Host "═══════════════════════════════════════════════════════" -ForegroundColor Magenta
+    Write-Host " [$Number/$Total] $Message" -ForegroundColor Cyan
+    Write-Host "═══════════════════════════════════════════════════════" -ForegroundColor Magenta
+}
+
+function Write-Success {
+    param([string]$Message)
+    Write-Host "✓ $Message" -ForegroundColor Green
+}
+
+function Write-Info {
+    param([string]$Message)
+    Write-Host "ℹ $Message" -ForegroundColor Yellow
+}
+
+function Write-ErrorExpected {
+    param([string]$Message)
+    Write-Host "✓ Expected error: $Message" -ForegroundColor Green
+}
+
+function Write-ErrorUnexpected {
+    param([string]$Message)
+    Write-Host "✗ Unexpected: $Message" -ForegroundColor Red
+}
+
+# Helper function to test validation
+function Test-ValidationError {
+    param(
+        [string]$Endpoint,
+        [string]$Method,
+        [hashtable]$Body,
+        [string]$ExpectedPattern,
+        [string]$Description
+    )
+
+    try {
+        $jsonBody = $Body | ConvertTo-Json -Depth 5
+        $result = Invoke-RestMethod -Uri "$BaseUrl$Endpoint" -Method $Method -Body $jsonBody -ContentType "application/json" -TimeoutSec 10
+
+        # If no HTTP exception, check for success: false
+        if ($result.success -eq $false -and $result.error -like $ExpectedPattern) {
+            Write-ErrorExpected "$($result.error.Substring(0, [Math]::Min(80, $result.error.Length)))..."
+            return $true
+        }
+        Write-ErrorUnexpected "Should have returned validation error for: $Description"
+        return $false
+    } catch {
+        $statusCode = $_.Exception.Response.StatusCode
+        if ($statusCode -eq 400 -or $statusCode -eq 'BadRequest') {
+            $errorResponse = $_.ErrorDetails.Message | ConvertFrom-Json
+            if ($errorResponse.error -like $ExpectedPattern) {
+                Write-ErrorExpected "$($errorResponse.error.Substring(0, [Math]::Min(80, $errorResponse.error.Length)))..."
+                return $true
+            }
+            Write-ErrorUnexpected "Error message doesn't match pattern: $($errorResponse.error)"
+            return $false
+        }
+        Write-ErrorUnexpected "Unexpected HTTP status: $statusCode"
+        return $false
+    }
+}
+
+# Banner
+Clear-Host
+Write-Host "`n" -NoNewline
+Write-Host "╔════════════════════════════════════════════════════════╗" -ForegroundColor Magenta
+Write-Host "║                                                        ║" -ForegroundColor Magenta
+Write-Host "║         ZOD VALIDATION TESTS - HTTP API                ║" -ForegroundColor Cyan
+Write-Host "║                                                        ║" -ForegroundColor Magenta
+Write-Host "╚════════════════════════════════════════════════════════╝" -ForegroundColor Magenta
+Write-Host ""
+
+# Check that the server is accessible
+Write-Host "Checking connection to server..." -ForegroundColor Yellow
+try {
+    $null = Invoke-RestMethod -Uri "$BaseUrl/health" -TimeoutSec 5
+    Write-Success "Server accessible at $BaseUrl"
+} catch {
+    Write-ErrorUnexpected "Unable to connect to server at $BaseUrl"
+    Write-Host "Make sure the server is started" -ForegroundColor Yellow
+    exit 1
+}
+
+$TotalTests = 18
+$PassedTests = 0
+$FailedTests = 0
+
+# =============================================================================
+# ASK QUESTION SCHEMA TESTS
+# =============================================================================
+
+Write-TestHeader "POST /ask - Empty question string" 1 $TotalTests
+if (Test-ValidationError -Endpoint "/ask" -Method "Post" -Body @{ question = "" } -ExpectedPattern "*question*empty*" -Description "empty question") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /ask - Question wrong type (number)" 2 $TotalTests
+if (Test-ValidationError -Endpoint "/ask" -Method "Post" -Body @{ question = 123 } -ExpectedPattern "*question*" -Description "question as number") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /ask - Invalid notebook_url format" 3 $TotalTests
+if (Test-ValidationError -Endpoint "/ask" -Method "Post" -Body @{ question = "test"; notebook_url = "not-a-url" } -ExpectedPattern "*url*" -Description "invalid URL") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /ask - show_browser wrong type (string)" 4 $TotalTests
+if (Test-ValidationError -Endpoint "/ask" -Method "Post" -Body @{ question = "test"; show_browser = "yes" } -ExpectedPattern "*show_browser*" -Description "show_browser as string") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+# =============================================================================
+# ADD NOTEBOOK SCHEMA TESTS
+# =============================================================================
+
+Write-TestHeader "POST /notebooks - Empty name" 5 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks" -Method "Post" -Body @{
+    url = "https://notebooklm.google.com/notebook/test"
+    name = ""
+    description = "Test"
+    topics = @("test")
+} -ExpectedPattern "*name*empty*" -Description "empty name") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /notebooks - Empty description" 6 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks" -Method "Post" -Body @{
+    url = "https://notebooklm.google.com/notebook/test"
+    name = "Test"
+    description = ""
+    topics = @("test")
+} -ExpectedPattern "*description*empty*" -Description "empty description") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /notebooks - Empty topics array" 7 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks" -Method "Post" -Body @{
+    url = "https://notebooklm.google.com/notebook/test"
+    name = "Test"
+    description = "Test description"
+    topics = @()
+} -ExpectedPattern "*topic*" -Description "empty topics array") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /notebooks - URL not NotebookLM domain" 8 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks" -Method "Post" -Body @{
+    url = "https://google.com/notebook/test"
+    name = "Test"
+    description = "Test description"
+    topics = @("test")
+} -ExpectedPattern "*NotebookLM*" -Description "wrong domain") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /notebooks - Invalid URL format" 9 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks" -Method "Post" -Body @{
+    url = "not-a-valid-url"
+    name = "Test"
+    description = "Test description"
+    topics = @("test")
+} -ExpectedPattern "*url*" -Description "invalid URL format") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /notebooks - topics wrong type (string)" 10 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks" -Method "Post" -Body @{
+    url = "https://notebooklm.google.com/notebook/test"
+    name = "Test"
+    description = "Test description"
+    topics = "single-topic"
+} -ExpectedPattern "*topics*array*" -Description "topics as string") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+# =============================================================================
+# UPDATE NOTEBOOK SCHEMA TESTS
+# =============================================================================
+
+Write-TestHeader "PUT /notebooks/:id - Empty name (when provided)" 11 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks/test-id" -Method "Put" -Body @{ name = "" } -ExpectedPattern "*name*" -Description "empty name update") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "PUT /notebooks/:id - Empty description (when provided)" 12 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks/test-id" -Method "Put" -Body @{ description = "" } -ExpectedPattern "*description*" -Description "empty description update") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+# =============================================================================
+# AUTO-DISCOVER SCHEMA TESTS
+# =============================================================================
+
+Write-TestHeader "POST /notebooks/auto-discover - Missing URL" 13 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks/auto-discover" -Method "Post" -Body @{} -ExpectedPattern "*url*" -Description "missing URL") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /notebooks/auto-discover - Invalid URL" 14 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks/auto-discover" -Method "Post" -Body @{ url = "not-a-url" } -ExpectedPattern "*url*" -Description "invalid URL") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /notebooks/auto-discover - Wrong domain" 15 $TotalTests
+if (Test-ValidationError -Endpoint "/notebooks/auto-discover" -Method "Post" -Body @{ url = "https://example.com/notebook" } -ExpectedPattern "*NotebookLM*" -Description "wrong domain") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+# =============================================================================
+# CLEANUP DATA SCHEMA TESTS
+# =============================================================================
+
+Write-TestHeader "POST /cleanup-data - Missing confirm" 16 $TotalTests
+if (Test-ValidationError -Endpoint "/cleanup-data" -Method "Post" -Body @{} -ExpectedPattern "*confirm*" -Description "missing confirm") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /cleanup-data - confirm wrong type (string)" 17 $TotalTests
+if (Test-ValidationError -Endpoint "/cleanup-data" -Method "Post" -Body @{ confirm = "yes" } -ExpectedPattern "*confirm*" -Description "confirm as string") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+Write-TestHeader "POST /cleanup-data - preserve_library wrong type" 18 $TotalTests
+if (Test-ValidationError -Endpoint "/cleanup-data" -Method "Post" -Body @{ confirm = $true; preserve_library = "yes" } -ExpectedPattern "*preserve_library*" -Description "preserve_library as string") {
+    $PassedTests++
+} else { $FailedTests++ }
+
+# =============================================================================
+# FINAL SUMMARY
+# =============================================================================
+Write-Host "`n" -NoNewline
+Write-Host "╔════════════════════════════════════════════════════════╗" -ForegroundColor Magenta
+Write-Host "║                                                        ║" -ForegroundColor Magenta
+Write-Host "║           VALIDATION TEST SUMMARY                      ║" -ForegroundColor Cyan
+Write-Host "║                                                        ║" -ForegroundColor Magenta
+Write-Host "╚════════════════════════════════════════════════════════╝" -ForegroundColor Magenta
+Write-Host ""
+
+$TotalExecuted = $PassedTests + $FailedTests
+$SuccessRate = if ($TotalExecuted -gt 0) { [math]::Round(($PassedTests / $TotalExecuted) * 100, 1) } else { 0 }
+
+Write-Host "Total tests: $TotalTests" -ForegroundColor White
+Write-Host "Tests passed: " -NoNewline -ForegroundColor White
+Write-Host "$PassedTests" -ForegroundColor Green
+Write-Host "Tests failed: " -NoNewline -ForegroundColor White
+Write-Host "$FailedTests" -ForegroundColor $(if($FailedTests -gt 0){"Red"}else{"Green"})
+Write-Host "Success rate: " -NoNewline -ForegroundColor White
+Write-Host "$SuccessRate%" -ForegroundColor $(if($SuccessRate -eq 100){"Green"}elseif($SuccessRate -ge 80){"Yellow"}else{"Red"})
+
+Write-Host ""
+
+if ($FailedTests -eq 0) {
+    Write-Host "════════════════════════════════════════════════════════" -ForegroundColor Green
+    Write-Host "  ✓ ALL VALIDATION SCHEMAS WORKING CORRECTLY!" -ForegroundColor Green
+    Write-Host "════════════════════════════════════════════════════════" -ForegroundColor Green
+    exit 0
+} else {
+    Write-Host "════════════════════════════════════════════════════════" -ForegroundColor Yellow
+    Write-Host "  ⚠ SOME VALIDATION TESTS FAILED" -ForegroundColor Yellow
+    Write-Host "════════════════════════════════════════════════════════" -ForegroundColor Yellow
+    Write-Host ""
+    Write-Host "See details above to identify the issues." -ForegroundColor Yellow
+    exit 1
+}

package/dist/accounts/account-manager.d.ts

@@ -0,0 +1,163 @@
+/**
+ * Account Manager
+ *
+ * Manages multiple Google accounts for NotebookLM with:
+ * - Account pool with rotation strategies
+ * - Encrypted credential storage
+ * - Quota tracking per account
+ * - Session health monitoring
+ * - Auto-login when sessions expire
+ */
+import type { Account, AccountHealth, AccountSelection, RotationStrategy } from './types.js';
+export declare class AccountManager {
+    private configPath;
+    private accountsDir;
+    private config;
+    private accounts;
+    private lastUsedIndex;
+    constructor();
+    /**
+     * Initialize the account manager
+     */
+    initialize(): Promise<void>;
+    /**
+     * Ensure required directories exist
+     */
+    private ensureDirectories;
+    /**
+     * Load accounts configuration
+     */
+    private loadConfig;
+    /**
+     * Save accounts configuration
+     */
+    private saveConfig;
+    /**
+     * Load all accounts from disk
+     */
+    private loadAccounts;
+    /**
+     * Load a single account
+     */
+    private loadAccount;
+    /**
+     * Create fresh quota (resets at midnight UTC)
+     */
+    private createFreshQuota;
+    /**
+     * Add a new account
+     */
+    addAccount(email: string, password: string, totpSecret?: string, options?: {
+        priority?: number;
+        notes?: string;
+    }): Promise<string>;
+    /**
+     * Remove an account
+     */
+    removeAccount(accountId: string): Promise<boolean>;
+    /**
+     * List all accounts
+     */
+    listAccounts(): Account[];
+    /**
+     * Get account by ID
+     */
+    getAccount(accountId: string): Account | undefined;
+    /**
+     * Get credentials for an account
+     */
+    getCredentials(accountId: string): Promise<{
+        email: string;
+        password: string;
+        totpSecret?: string;
+    } | null>;
+    /**
+     * Get the best available account based on rotation strategy
+     * @param excludeAccountId Optional account ID to exclude (e.g., rate-limited account)
+     */
+    getBestAccount(excludeAccountId?: string): Promise<AccountSelection | null>;
+    /**
+     * Get accounts that are available for use
+     */
+    private getAvailableAccounts;
+    /**
+     * Select account with most remaining quota
+     */
+    private selectLeastUsed;
+    /**
+     * Select account using round-robin
+     */
+    private selectRoundRobin;
+    /**
+     * Select account by priority (failover)
+     */
+    private selectFailover;
+    /**
+     * Record a query usage for an account
+     */
+    recordUsage(accountId: string): Promise<void>;
+    /**
+     * Mark account as rate-limited (quota exhausted)
+     * This ensures the account won't be selected again until quota resets
+     */
+    markRateLimited(accountId: string): Promise<void>;
+    /**
+     * Save the ID of the currently active account
+     * This is used to identify which account is currently loaded in the main profile
+     */
+    saveCurrentAccountId(accountId: string): Promise<void>;
+    /**
+     * Get the ID of the currently active account
+     * Returns null if no account is set or file doesn't exist
+     */
+    getCurrentAccountId(): Promise<string | null>;
+    /**
+     * Record a login failure
+     */
+    recordLoginFailure(accountId: string, error: string): Promise<void>;
+    /**
+     * Record a successful login
+     */
+    recordLoginSuccess(accountId: string): Promise<void>;
+    /**
+     * Update session status for an account
+     */
+    updateSessionStatus(accountId: string, status: 'valid' | 'expiring' | 'expired'): Promise<void>;
+    /**
+     * Check health of all accounts
+     */
+    healthCheck(): Promise<AccountHealth[]>;
+    /**
+     * Check health of a single account
+     */
+    private checkAccountHealth;
+    /**
+     * Get current rotation strategy
+     */
+    getRotationStrategy(): RotationStrategy;
+    /**
+     * Set rotation strategy
+     */
+    setRotationStrategy(strategy: RotationStrategy): Promise<void>;
+    /**
+     * Check if auto-login is enabled
+     */
+    isAutoLoginEnabled(): boolean;
+    /**
+     * Enable/disable auto-login
+     */
+    setAutoLoginEnabled(enabled: boolean): Promise<void>;
+    /**
+     * Get account count
+     */
+    getAccountCount(): number;
+    /**
+     * Check if any accounts are configured
+     */
+    hasAccounts(): boolean;
+}
+/**
+ * Get or create the account manager instance
+ */
+export declare function getAccountManager(): Promise<AccountManager>;
+//# sourceMappingURL=account-manager.d.ts.map

package/dist/accounts/account-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"account-manager.d.ts","sourceRoot":"","sources":["../../src/accounts/account-manager.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAQH,OAAO,KAAK,EACV,OAAO,EAKP,aAAa,EACb,gBAAgB,EAEhB,gBAAgB,EACjB,MAAM,YAAY,CAAC;AAIpB,qBAAa,cAAc;IACzB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,MAAM,CAA+B;IAC7C,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,aAAa,CAAa;;IAWlC;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAoBjC;;OAEG;YACW,iBAAiB;IAW/B;;OAEG;YACW,UAAU;IAwBxB;;OAEG;YACW,UAAU;IAMxB;;OAEG;YACW,YAAY;IAgB1B;;OAEG;YACW,WAAW;IAuDzB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAkBxB;;OAEG;IACG,UAAU,CACd,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,MAAM,EAChB,UAAU,CAAC,EAAE,MAAM,EACnB,OAAO,GAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAO,GAClD,OAAO,CAAC,MAAM,CAAC;IAoDlB;;OAEG;IACG,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IA0BxD;;OAEG;IACH,YAAY,IAAI,OAAO,EAAE;IAIzB;;OAEG;IACH,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS;IAIlD;;OAEG;IACG,cAAc,CAClB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAkC3E;;;OAGG;IACG,cAAc,CAAC,gBAAgB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAsDjF;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAe5B;;OAEG;IACH,OAAO,CAAC,eAAe;IAQvB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAKxB;;OAEG;IACH,OAAO,CAAC,cAAc;IAQtB;;OAEG;IACG,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBnD;;;OAGG;IACG,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBvD;;;OAGG;IACG,oBAAoB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAM5D;;;OAGG;IACG,mBAAmB,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAUnD;;OAEG;IACG,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAczE;;OAEG;IACG,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAc1D;;OAEG;IACG,mBAAmB,CACvB,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,OAAO,GAAG,UAAU,GAAG,SAAS,GACvC,OAAO,CAAC,IAAI,CAAC;IAchB;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;IAW7C;;OAEG;YACW,kBAAkB;IAgEhC;;OAEG;IACH,mBAAmB,IAAI,gBAAgB;IAIvC;;OAEG;IACG,mBAAmB,CAAC,QAAQ,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAQpE;;OAEG;IACH,kBAAkB,IAAI,OAAO;IAI7B;;OAEG;IACG,mBAAmB,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ1D;;OAEG;IACH,eAAe,IAAI,MAAM;IAIzB;;OAEG;IACH,WAAW,IAAI,OAAO;CAGvB;AAKD;;GAEG;AACH,wBAAsB,iBAAiB,IAAI,OAAO,CAAC,cAAc,CAAC,CAMjE"}