@pan-sec/notebooklm-mcp 1.10.5 → 1.10.8

package/dist/logging/query-logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-logger.d.ts","sourceRoot":"","sources":["../../src/logging/query-logger.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAaH;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE;QACT,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;QACd,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;IACF,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,aAAa,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAoBD;;;;GAIG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,cAAc,CAAc;IACpC,OAAO,CAAC,UAAU,CAAuB;IACzC,OAAO,CAAC,SAAS,CAAkB;IACnC,OAAO,CAAC,KAAK,CAGX;gBAEU,MAAM,CAAC,EAAE,OAAO,CAAC,iBAAiB,CAAC;IAU/C;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAI1B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAKzB;;OAEG;IACH,OAAO,CAAC,YAAY;IA4BpB;;OAEG;YACW,UAAU;IA0BxB;;OAEG;IACG,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAoBpF;;OAEG;IACG,oBAAoB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAIvE;;OAEG;IACG,qBAAqB,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAI1E;;OAEG;IACG,uBAAuB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAI3E;;OAEG;IACG,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAK/D;;OAEG;IACG,gBAAgB,CAAC,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAOpE;;OAEG;IACG,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAkB5F;;OAEG;IACH,WAAW,IAAI,MAAM,EAAE;IAYvB;;OAEG;IACH,QAAQ,IAAI,OAAO,IAAI,CAAC,KAAK;IAI7B;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAU5B;;OAEG;IACH,OAAO,CAAC,WAAW;IAanB;;OAEG;YACW,aAAa;IAa3B;;OAEG;YACW,aAAa;CAM5B;AAOD;;GAEG;AACH,wBAAgB,cAAc,IAAI,WAAW,CAK5C;AAED;;GAEG;AACH,wBAAsB,QAAQ,CAC5B,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,WAAW,GAAG,SAAS,CAAC,GAClD,OAAO,CAAC,MAAM,CAAC,CAEjB"}

package/dist/logging/query-logger.js

@@ -0,0 +1,282 @@
+/**
+ * Query Logger for NotebookLM MCP Server
+ *
+ * Provides persistent logging of all Q&A interactions with NotebookLM:
+ * - Full question and answer content
+ * - Session and notebook context
+ * - Quota information at time of query
+ * - Duration and metadata
+ *
+ * Features:
+ * - JSONL format with daily rotation
+ * - 90-day retention (configurable)
+ * - Search and retrieval by session, notebook, date
+ * - Full content storage for research review
+ */
+import fs from "fs";
+import path from "path";
+import crypto from "crypto";
+import { CONFIG } from "../config.js";
+import { mkdirSecure, appendFileSecure, PERMISSION_MODES, } from "../utils/file-permissions.js";
+import { log } from "../utils/logger.js";
+/**
+ * Get query logger configuration from environment
+ */
+function getQueryLoggerConfig() {
+    return {
+        enabled: process.env.NLMCP_QUERY_LOG_ENABLED !== "false",
+        logDir: process.env.NLMCP_QUERY_LOG_DIR || path.join(CONFIG.dataDir, "query_logs"),
+        retentionDays: parseInt(process.env.NLMCP_QUERY_LOG_RETENTION_DAYS || "90", 10),
+    };
+}
+/**
+ * Generate unique query ID
+ */
+function generateQueryId() {
+    return crypto.randomBytes(8).toString("hex");
+}
+/**
+ * Query Logger Class
+ *
+ * Logs all Q&A interactions to JSONL files for later review.
+ */
+export class QueryLogger {
+    config;
+    currentLogFile = "";
+    writeQueue = [];
+    isWriting = false;
+    stats = {
+        totalQueries: 0,
+        queriesThisSession: 0,
+    };
+    constructor(config) {
+        this.config = { ...getQueryLoggerConfig(), ...config };
+        if (this.config.enabled) {
+            this.ensureLogDirectory();
+            this.initializeLogFile();
+            this.cleanOldLogs();
+        }
+    }
+    /**
+     * Ensure query log directory exists
+     */
+    ensureLogDirectory() {
+        mkdirSecure(this.config.logDir, PERMISSION_MODES.OWNER_FULL);
+    }
+    /**
+     * Initialize log file for today
+     */
+    initializeLogFile() {
+        const today = new Date().toISOString().split("T")[0];
+        this.currentLogFile = path.join(this.config.logDir, `query-log-${today}.jsonl`);
+    }
+    /**
+     * Clean up old log files based on retention policy
+     */
+    cleanOldLogs() {
+        try {
+            const files = fs.readdirSync(this.config.logDir);
+            const cutoffDate = new Date();
+            cutoffDate.setDate(cutoffDate.getDate() - this.config.retentionDays);
+            let deletedCount = 0;
+            for (const file of files) {
+                if (!file.startsWith("query-log-") || !file.endsWith(".jsonl"))
+                    continue;
+                // Extract date from filename (query-log-YYYY-MM-DD.jsonl)
+                const dateStr = file.slice(10, 20);
+                const fileDate = new Date(dateStr);
+                if (fileDate < cutoffDate) {
+                    fs.unlinkSync(path.join(this.config.logDir, file));
+                    deletedCount++;
+                }
+            }
+            if (deletedCount > 0) {
+                log.info(`🗑️ Cleaned ${deletedCount} old query log files`);
+            }
+        }
+        catch {
+            // Ignore cleanup errors
+        }
+    }
+    /**
+     * Write entry to log file
+     */
+    async writeEntry(entry) {
+        this.writeQueue.push(entry);
+        if (this.isWriting)
+            return;
+        this.isWriting = true;
+        try {
+            while (this.writeQueue.length > 0) {
+                const batch = this.writeQueue.splice(0, 100);
+                const lines = batch.map(e => JSON.stringify(e)).join("\n") + "\n";
+                // Check if we need to rotate to new day's file
+                const today = new Date().toISOString().split("T")[0];
+                const expectedFile = path.join(this.config.logDir, `query-log-${today}.jsonl`);
+                if (this.currentLogFile !== expectedFile) {
+                    this.currentLogFile = expectedFile;
+                }
+                appendFileSecure(this.currentLogFile, lines, PERMISSION_MODES.OWNER_READ_WRITE);
+            }
+        }
+        finally {
+            this.isWriting = false;
+        }
+    }
+    /**
+     * Log a query (Q&A pair)
+     */
+    async logQuery(entry) {
+        if (!this.config.enabled)
+            return "";
+        const queryId = generateQueryId();
+        const fullEntry = {
+            timestamp: new Date().toISOString(),
+            queryId,
+            ...entry,
+        };
+        this.stats.totalQueries++;
+        this.stats.queriesThisSession++;
+        await this.writeEntry(fullEntry);
+        log.debug(`📝 Logged query ${queryId} (${entry.question.slice(0, 50)}...)`);
+        return queryId;
+    }
+    /**
+     * Get all queries for a specific session
+     */
+    async getQueriesForSession(sessionId) {
+        return this.filterQueries(entry => entry.sessionId === sessionId);
+    }
+    /**
+     * Get all queries for a specific notebook URL
+     */
+    async getQueriesForNotebook(notebookUrl) {
+        return this.filterQueries(entry => entry.notebookUrl === notebookUrl);
+    }
+    /**
+     * Get all queries for a specific notebook ID
+     */
+    async getQueriesForNotebookId(notebookId) {
+        return this.filterQueries(entry => entry.notebookId === notebookId);
+    }
+    /**
+     * Get all queries for a specific date (YYYY-MM-DD)
+     */
+    async getQueriesForDate(date) {
+        const logFile = path.join(this.config.logDir, `query-log-${date}.jsonl`);
+        return this.readLogFile(logFile);
+    }
+    /**
+     * Get recent queries
+     */
+    async getRecentQueries(limit = 50) {
+        const allQueries = await this.getAllQueries();
+        // Sort by timestamp descending (most recent first)
+        allQueries.sort((a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime());
+        return allQueries.slice(0, limit);
+    }
+    /**
+     * Search queries by pattern in question or answer
+     */
+    async searchQueries(pattern, options) {
+        const limit = options?.limit ?? 100;
+        const caseSensitive = options?.caseSensitive ?? false;
+        const searchPattern = caseSensitive ? pattern : pattern.toLowerCase();
+        const matches = await this.filterQueries(entry => {
+            const question = caseSensitive ? entry.question : entry.question.toLowerCase();
+            const answer = caseSensitive ? entry.answer : entry.answer.toLowerCase();
+            return question.includes(searchPattern) || answer.includes(searchPattern);
+        });
+        // Sort by timestamp descending
+        matches.sort((a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime());
+        return matches.slice(0, limit);
+    }
+    /**
+     * Get all available log files
+     */
+    getLogFiles() {
+        try {
+            const files = fs.readdirSync(this.config.logDir);
+            return files
+                .filter(f => f.startsWith("query-log-") && f.endsWith(".jsonl"))
+                .sort()
+                .reverse(); // Most recent first
+        }
+        catch {
+            return [];
+        }
+    }
+    /**
+     * Get statistics
+     */
+    getStats() {
+        return { ...this.stats };
+    }
+    /**
+     * Force flush any pending writes
+     */
+    async flush() {
+        while (this.isWriting || this.writeQueue.length > 0) {
+            await new Promise(resolve => setTimeout(resolve, 10));
+        }
+    }
+    // ============================================================================
+    // Private Helper Methods
+    // ============================================================================
+    /**
+     * Read and parse a log file
+     */
+    readLogFile(filePath) {
+        if (!fs.existsSync(filePath))
+            return [];
+        try {
+            const content = fs.readFileSync(filePath, "utf-8");
+            const lines = content.trim().split("\n").filter(l => l.length > 0);
+            return lines.map(line => JSON.parse(line));
+        }
+        catch (error) {
+            log.warning(`⚠️ Failed to read query log ${filePath}: ${error}`);
+            return [];
+        }
+    }
+    /**
+     * Get all queries from all log files
+     */
+    async getAllQueries() {
+        const logFiles = this.getLogFiles();
+        const allQueries = [];
+        for (const file of logFiles) {
+            const filePath = path.join(this.config.logDir, file);
+            const entries = this.readLogFile(filePath);
+            allQueries.push(...entries);
+        }
+        return allQueries;
+    }
+    /**
+     * Filter queries across all log files
+     */
+    async filterQueries(predicate) {
+        const allQueries = await this.getAllQueries();
+        return allQueries.filter(predicate);
+    }
+}
+/**
+ * Global query logger instance
+ */
+let globalQueryLogger = null;
+/**
+ * Get or create the global query logger
+ */
+export function getQueryLogger() {
+    if (!globalQueryLogger) {
+        globalQueryLogger = new QueryLogger();
+    }
+    return globalQueryLogger;
+}
+/**
+ * Convenience function for quick query logging
+ */
+export async function logQuery(entry) {
+    return getQueryLogger().logQuery(entry);
+}
+//# sourceMappingURL=query-logger.js.map

package/dist/logging/query-logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-logger.js","sourceRoot":"","sources":["../../src/logging/query-logger.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EACL,WAAW,EACX,gBAAgB,EAChB,gBAAgB,GACjB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,GAAG,EAAE,MAAM,oBAAoB,CAAC;AA4CzC;;GAEG;AACH,SAAS,oBAAoB;IAC3B,OAAO;QACL,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,OAAO;QACxD,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC;QAClF,aAAa,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,8BAA8B,IAAI,IAAI,EAAE,EAAE,CAAC;KAChF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,eAAe;IACtB,OAAO,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AAC/C,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,WAAW;IACd,MAAM,CAAoB;IAC1B,cAAc,GAAW,EAAE,CAAC;IAC5B,UAAU,GAAoB,EAAE,CAAC;IACjC,SAAS,GAAY,KAAK,CAAC;IAC3B,KAAK,GAAG;QACd,YAAY,EAAE,CAAC;QACf,kBAAkB,EAAE,CAAC;KACtB,CAAC;IAEF,YAAY,MAAmC;QAC7C,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,oBAAoB,EAAE,EAAE,GAAG,MAAM,EAAE,CAAC;QAEvD,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC1B,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,kBAAkB;QACxB,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,gBAAgB,CAAC,UAAU,CAAC,CAAC;IAC/D,CAAC;IAED;;OAEG;IACK,iBAAiB;QACvB,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,aAAa,KAAK,QAAQ,CAAC,CAAC;IAClF,CAAC;IAED;;OAEG;IACK,YAAY;QAClB,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACjD,MAAM,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC;YAC9B,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAErE,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;oBAAE,SAAS;gBAEzE,0DAA0D;gBAC1D,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACnC,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC;gBAEnC,IAAI,QAAQ,GAAG,UAAU,EAAE,CAAC;oBAC1B,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;oBACnD,YAAY,EAAE,CAAC;gBACjB,CAAC;YACH,CAAC;YAED,IAAI,YAAY,GAAG,CAAC,EAAE,CAAC;gBACrB,GAAG,CAAC,IAAI,CAAC,eAAe,YAAY,sBAAsB,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,wBAAwB;QAC1B,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,UAAU,CAAC,KAAoB;QAC3C,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE5B,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAE3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtB,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;gBAElE,+CAA+C;gBAC/C,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBACrD,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,aAAa,KAAK,QAAQ,CAAC,CAAC;gBAC/E,IAAI,IAAI,CAAC,cAAc,KAAK,YAAY,EAAE,CAAC;oBACzC,IAAI,CAAC,cAAc,GAAG,YAAY,CAAC;gBACrC,CAAC;gBAED,gBAAgB,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;YAClF,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACzB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,KAAmD;QAChE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO,EAAE,CAAC;QAEpC,MAAM,OAAO,GAAG,eAAe,EAAE,CAAC;QAClC,MAAM,SAAS,GAAkB;YAC/B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,OAAO;YACP,GAAG,KAAK;SACT,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;QAC1B,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;QAEhC,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAEjC,GAAG,CAAC,KAAK,CAAC,mBAAmB,OAAO,KAAK,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;QAE5E,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB,CAAC,SAAiB;QAC1C,OAAO,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;IACpE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB,CAAC,WAAmB;QAC7C,OAAO,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,KAAK,WAAW,CAAC,CAAC;IACxE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,uBAAuB,CAAC,UAAkB;QAC9C,OAAO,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,KAAK,UAAU,CAAC,CAAC;IACtE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB,CAAC,IAAY;QAClC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,aAAa,IAAI,QAAQ,CAAC,CAAC;QACzE,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAAC,QAAgB,EAAE;QACvC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,mDAAmD;QACnD,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7F,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,OAAe,EAAE,OAA4B;QAC/D,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,GAAG,CAAC;QACpC,MAAM,aAAa,GAAG,OAAO,EAAE,aAAa,IAAI,KAAK,CAAC;QAEtD,MAAM,aAAa,GAAG,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAEtE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE;YAC/C,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC/E,MAAM,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YACzE,OAAO,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,+BAA+B;QAC/B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAE1F,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,WAAW;QACT,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACjD,OAAO,KAAK;iBACT,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;iBAC/D,IAAI,EAAE;iBACN,OAAO,EAAE,CAAC,CAAC,oBAAoB;QACpC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,OAAO,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAED,+EAA+E;IAC/E,yBAAyB;IACzB,+EAA+E;IAE/E;;OAEG;IACK,WAAW,CAAC,QAAgB;QAClC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,OAAO,EAAE,CAAC;QAExC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACnE,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAkB,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,OAAO,CAAC,+BAA+B,QAAQ,KAAK,KAAK,EAAE,CAAC,CAAC;YACjE,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa;QACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,UAAU,GAAoB,EAAE,CAAC;QAEvC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACrD,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC3C,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;QAC9B,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa,CACzB,SAA4C;QAE5C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,OAAO,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACtC,CAAC;CACF;AAED;;GAEG;AACH,IAAI,iBAAiB,GAAuB,IAAI,CAAC;AAEjD;;GAEG;AACH,MAAM,UAAU,cAAc;IAC5B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,iBAAiB,GAAG,IAAI,WAAW,EAAE,CAAC;IACxC,CAAC;IACD,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,KAAmD;IAEnD,OAAO,cAAc,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AAC1C,CAAC"}

package/dist/quota/quota-manager.d.ts

@@ -48,6 +48,24 @@ export declare class QuotaManager {
      * Extract source limit from source dialog (e.g., "0/300")
      */
     extractSourceLimitFromDialog(page: Page): Promise<number | null>;
+    /**
+     * Extract query usage from NotebookLM UI
+     *
+     * Looks for patterns like:
+     * - "X/50 queries" or "X of 50 queries"
+     * - "X queries remaining"
+     * - Usage indicators in settings/account area
+     *
+     * Returns { used, limit } or null if not found
+     */
+    extractQueryUsageFromUI(page: Page): Promise<{
+        used: number;
+        limit: number;
+    } | null>;
+    /**
+     * Check for rate limit error message on page
+     */
+    checkForRateLimitError(page: Page): Promise<boolean>;
     /**
      * Count notebooks from homepage
      */
@@ -55,7 +73,14 @@ export declare class QuotaManager {
     /**
      * Update quota from UI scraping
      */
-    updateFromUI(page: Page): Promise<void>;
+    updateFromUI(page: Page): Promise<{
+        tier: LicenseTier;
+        queryUsageFromGoogle: {
+            used: number;
+            limit: number;
+        } | null;
+        rateLimitDetected: boolean;
+    }>;
     /**
      * Manually set tier (for user override)
      */
@@ -77,9 +102,23 @@ export declare class QuotaManager {
      */
     incrementNotebookCount(): void;
     /**
-     * Increment query count
+     * Increment query count (synchronous, for backwards compatibility)
+     * Note: For concurrent safety, use incrementQueryCountAtomic() instead
      */
     incrementQueryCount(): void;
+    /**
+     * Increment query count atomically with file locking
+     *
+     * This method is safe for concurrent access from multiple processes/sessions.
+     * It reloads settings from disk before incrementing to ensure accuracy.
+     */
+    incrementQueryCountAtomic(): Promise<void>;
+    /**
+     * Refresh settings from disk with file locking
+     *
+     * Use this to ensure you have the latest quota state from disk.
+     */
+    refreshSettings(): Promise<QuotaSettings>;
     /**
      * Check if can create notebook
      */
@@ -120,6 +159,31 @@ export declare class QuotaManager {
             percent: number;
         };
     };
+    /**
+     * Get detailed quota status with remaining counts, warnings, and stop signals
+     * Used to provide visibility to users about when to stop querying for the day
+     */
+    getDetailedStatus(): {
+        tier: LicenseTier;
+        queries: {
+            used: number;
+            limit: number;
+            remaining: number;
+            percentUsed: number;
+            shouldStop: boolean;
+            resetTime: string;
+        };
+        notebooks: {
+            used: number;
+            limit: number;
+            remaining: number;
+            percentUsed: number;
+        };
+        sources: {
+            limit: number;
+        };
+        warnings: string[];
+    };
 }
 export declare function getQuotaManager(): QuotaManager;
 //# sourceMappingURL=quota-manager.d.ts.map

package/dist/quota/quota-manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"quota-manager.d.ts","sourceRoot":"","sources":["../../src/quota/quota-manager.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAMvC,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,KAAK,GAAG,OAAO,GAAG,SAAS,CAAC;AAE/D,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,UAAU;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,MAAM,EAAE,WAAW,CAAC;IACpB,KAAK,EAAE,UAAU,CAAC;IAClB,YAAY,EAAE,OAAO,CAAC;CACvB;AAgCD,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,YAAY,CAAS;;IAO7B;;OAEG;IACH,OAAO,CAAC,YAAY;IAgBpB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAc1B;;OAEG;IACH,OAAO,CAAC,YAAY;IAiBpB;;;OAGG;IACG,kBAAkB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,WAAW,CAAC;IA0C1D;;OAEG;IACG,4BAA4B,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAetE;;OAEG;IACG,sBAAsB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC;IAiBzD;;OAEG;IACG,YAAY,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IA6B7C;;OAEG;IACH,OAAO,CAAC,IAAI,EAAE,WAAW,GAAG,IAAI;IAQhC;;OAEG;IACH,WAAW,IAAI,aAAa;IAI5B;;OAEG;IACH,SAAS,IAAI,WAAW;IAIxB;;OAEG;IACH,QAAQ,IAAI,UAAU;IAItB;;OAEG;IACH,sBAAsB,IAAI,IAAI;IAM9B;;OAEG;IACH,mBAAmB,IAAI,IAAI;IAc3B;;OAEG;IACH,iBAAiB,IAAI;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE;IAmB1D;;OAEG;IACH,YAAY,CAAC,kBAAkB,EAAE,MAAM,GAAG;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE;IAa/E;;OAEG;IACH,YAAY,IAAI;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE;IA2BrD;;OAEG;IACH,SAAS,IAAI;QACX,IAAI,EAAE,WAAW,CAAC;QAClB,SAAS,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC;QAC5D,OAAO,EAAE;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC;QAC3B,OAAO,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC;KAC3D;CAoBF;AAKD,wBAAgB,eAAe,IAAI,YAAY,CAK9C"}
+{"version":3,"file":"quota-manager.d.ts","sourceRoot":"","sources":["../../src/quota/quota-manager.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAOvC,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,KAAK,GAAG,OAAO,GAAG,SAAS,CAAC;AAE/D,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,UAAU;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,MAAM,EAAE,WAAW,CAAC;IACpB,KAAK,EAAE,UAAU,CAAC;IAClB,YAAY,EAAE,OAAO,CAAC;CACvB;AAgCD,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,YAAY,CAAS;;IAO7B;;OAEG;IACH,OAAO,CAAC,YAAY;IAgBpB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAc1B;;OAEG;IACH,OAAO,CAAC,YAAY;IAiBpB;;;OAGG;IACG,kBAAkB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,WAAW,CAAC;IA0C1D;;OAEG;IACG,4BAA4B,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAetE;;;;;;;;;OASG;IACG,uBAAuB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAuE1F;;OAEG;IACG,sBAAsB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAoB1D;;OAEG;IACG,sBAAsB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC;IAiBzD;;OAEG;IACG,YAAY,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC;QACtC,IAAI,EAAE,WAAW,CAAC;QAClB,oBAAoB,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAA;SAAE,GAAG,IAAI,CAAC;QAC7D,iBAAiB,EAAE,OAAO,CAAC;KAC5B,CAAC;IAoDF;;OAEG;IACH,OAAO,CAAC,IAAI,EAAE,WAAW,GAAG,IAAI;IAQhC;;OAEG;IACH,WAAW,IAAI,aAAa;IAI5B;;OAEG;IACH,SAAS,IAAI,WAAW;IAIxB;;OAEG;IACH,QAAQ,IAAI,UAAU;IAItB;;OAEG;IACH,sBAAsB,IAAI,IAAI;IAM9B;;;OAGG;IACH,mBAAmB,IAAI,IAAI;IAc3B;;;;;OAKG;IACG,yBAAyB,IAAI,OAAO,CAAC,IAAI,CAAC;IAkChD;;;;OAIG;IACG,eAAe,IAAI,OAAO,CAAC,aAAa,CAAC;IAO/C;;OAEG;IACH,iBAAiB,IAAI;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE;IAmB1D;;OAEG;IACH,YAAY,CAAC,kBAAkB,EAAE,MAAM,GAAG;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE;IAa/E;;OAEG;IACH,YAAY,IAAI;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE;IA2BrD;;OAEG;IACH,SAAS,IAAI;QACX,IAAI,EAAE,WAAW,CAAC;QAClB,SAAS,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC;QAC5D,OAAO,EAAE;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC;QAC3B,OAAO,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC;KAC3D;IAqBD;;;OAGG;IACH,iBAAiB,IAAI;QACnB,IAAI,EAAE,WAAW,CAAC;QAClB,OAAO,EAAE;YACP,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,MAAM,CAAC;YACd,SAAS,EAAE,MAAM,CAAC;YAClB,WAAW,EAAE,MAAM,CAAC;YACpB,UAAU,EAAE,OAAO,CAAC;YACpB,SAAS,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,SAAS,EAAE;YACT,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,MAAM,CAAC;YACd,SAAS,EAAE,MAAM,CAAC;YAClB,WAAW,EAAE,MAAM,CAAC;SACrB,CAAC;QACF,OAAO,EAAE;YACP,KAAK,EAAE,MAAM,CAAC;SACf,CAAC;QACF,QAAQ,EAAE,MAAM,EAAE,CAAC;KACpB;CA4DF;AAKD,wBAAgB,eAAe,IAAI,YAAY,CAK9C"}

package/dist/quota/quota-manager.js

@@ -5,6 +5,7 @@
  */
 import { log } from "../utils/logger.js";
 import { CONFIG } from "../config.js";
+import { withLock } from "../utils/file-lock.js";
 import fs from "fs";
 import path from "path";
 // Known limits by tier (based on NotebookLM documentation Dec 2025)
@@ -147,6 +148,100 @@ export class QuotaManager {
         });
         return limitInfo;
     }
+    /**
+     * Extract query usage from NotebookLM UI
+     *
+     * Looks for patterns like:
+     * - "X/50 queries" or "X of 50 queries"
+     * - "X queries remaining"
+     * - Usage indicators in settings/account area
+     *
+     * Returns { used, limit } or null if not found
+     */
+    async extractQueryUsageFromUI(page) {
+        log.info("🔍 Looking for query usage in UI...");
+        const usageInfo = await page.evaluate(() => {
+            // @ts-expect-error - DOM types
+            const allText = document.body.innerText;
+            // Pattern 1: "X/Y queries" or "X / Y queries"
+            const slashPattern = allText.match(/(\d+)\s*\/\s*(\d+)\s*quer(?:y|ies)/i);
+            if (slashPattern) {
+                return {
+                    used: parseInt(slashPattern[1], 10),
+                    limit: parseInt(slashPattern[2], 10),
+                };
+            }
+            // Pattern 2: "X of Y queries"
+            const ofPattern = allText.match(/(\d+)\s+of\s+(\d+)\s*quer(?:y|ies)/i);
+            if (ofPattern) {
+                return {
+                    used: parseInt(ofPattern[1], 10),
+                    limit: parseInt(ofPattern[2], 10),
+                };
+            }
+            // Pattern 3: "X queries remaining" with known limits
+            const remainingPattern = allText.match(/(\d+)\s*quer(?:y|ies)\s*remaining/i);
+            if (remainingPattern) {
+                const remaining = parseInt(remainingPattern[1], 10);
+                // Infer limit from known tiers
+                let limit = 50; // default free
+                if (remaining > 50)
+                    limit = 500; // pro
+                if (remaining > 500)
+                    limit = 5000; // ultra
+                return {
+                    used: limit - remaining,
+                    limit,
+                };
+            }
+            // Pattern 4: "You have used X queries today"
+            const usedPattern = allText.match(/(?:used|made)\s*(\d+)\s*quer(?:y|ies)/i);
+            if (usedPattern) {
+                const used = parseInt(usedPattern[1], 10);
+                // Infer tier from usage
+                let limit = 50;
+                if (used > 50)
+                    limit = 500;
+                if (used > 500)
+                    limit = 5000;
+                return { used, limit };
+            }
+            // Pattern 5: Look for rate limit message
+            const rateLimitPattern = allText.match(/(?:limit|quota)\s*(?:reached|exceeded)/i);
+            if (rateLimitPattern) {
+                // At limit - try to find the number
+                const limitNum = allText.match(/(\d+)\s*(?:daily|per day)/i);
+                const limit = limitNum ? parseInt(limitNum[1], 10) : 50;
+                return { used: limit, limit };
+            }
+            return null;
+        });
+        if (usageInfo) {
+            log.info(`  Found query usage: ${usageInfo.used}/${usageInfo.limit}`);
+        }
+        else {
+            log.info("  No query usage found in UI");
+        }
+        return usageInfo;
+    }
+    /**
+     * Check for rate limit error message on page
+     */
+    async checkForRateLimitError(page) {
+        const isRateLimited = await page.evaluate(() => {
+            // @ts-expect-error - DOM types
+            const allText = document.body.innerText.toLowerCase();
+            return (allText.includes("rate limit") ||
+                allText.includes("quota exceeded") ||
+                allText.includes("too many requests") ||
+                allText.includes("daily limit reached") ||
+                allText.includes("try again tomorrow"));
+        });
+        if (isRateLimited) {
+            log.warning("⚠️ Rate limit detected in UI!");
+        }
+        return isRateLimited;
+    }
     /**
      * Count notebooks from homepage
      */
@@ -187,9 +282,29 @@ export class QuotaManager {
         if (sourceLimit) {
             this.settings.limits.sourcesPerNotebook = sourceLimit;
         }
+        // Try to extract query usage from UI
+        const queryUsage = await this.extractQueryUsageFromUI(page);
+        if (queryUsage) {
+            // Update local tracking with Google's numbers
+            this.settings.usage.queriesUsedToday = queryUsage.used;
+            this.settings.limits.queriesPerDay = queryUsage.limit;
+            this.settings.usage.lastQueryDate = new Date().toISOString().split("T")[0];
+            log.info(`  Synced query usage from Google: ${queryUsage.used}/${queryUsage.limit}`);
+        }
+        // Check for rate limit
+        const rateLimitDetected = await this.checkForRateLimitError(page);
+        if (rateLimitDetected) {
+            // Mark as at limit
+            this.settings.usage.queriesUsedToday = this.settings.limits.queriesPerDay;
+        }
         this.settings.usage.lastUpdated = new Date().toISOString();
         this.saveSettings();
-        log.success(`✅ Quota updated: tier=${this.settings.tier}, notebooks=${this.settings.usage.notebooks}`);
+        log.success(`✅ Quota updated: tier=${this.settings.tier}, notebooks=${this.settings.usage.notebooks}, queries=${this.settings.usage.queriesUsedToday}/${this.settings.limits.queriesPerDay}`);
+        return {
+            tier: this.settings.tier,
+            queryUsageFromGoogle: queryUsage,
+            rateLimitDetected,
+        };
     }
     /**
      * Manually set tier (for user override)
@@ -228,7 +343,8 @@ export class QuotaManager {
         this.saveSettings();
     }
     /**
-     * Increment query count
+     * Increment query count (synchronous, for backwards compatibility)
+     * Note: For concurrent safety, use incrementQueryCountAtomic() instead
      */
     incrementQueryCount() {
         const today = new Date().toISOString().split("T")[0];
@@ -241,6 +357,49 @@ export class QuotaManager {
         this.settings.usage.lastUpdated = new Date().toISOString();
         this.saveSettings();
     }
+    /**
+     * Increment query count atomically with file locking
+     *
+     * This method is safe for concurrent access from multiple processes/sessions.
+     * It reloads settings from disk before incrementing to ensure accuracy.
+     */
+    async incrementQueryCountAtomic() {
+        await withLock(this.settingsPath, async () => {
+            // Reload latest settings from disk (another process may have updated)
+            this.settings = this.loadSettings();
+            const today = new Date().toISOString().split("T")[0];
+            // Reset if new day
+            if (this.settings.usage.lastQueryDate !== today) {
+                this.settings.usage.queriesUsedToday = 0;
+                this.settings.usage.lastQueryDate = today;
+            }
+            this.settings.usage.queriesUsedToday++;
+            this.settings.usage.lastUpdated = new Date().toISOString();
+            // Save with lock held
+            try {
+                const dir = path.dirname(this.settingsPath);
+                if (!fs.existsSync(dir)) {
+                    fs.mkdirSync(dir, { recursive: true, mode: 0o700 });
+                }
+                fs.writeFileSync(this.settingsPath, JSON.stringify(this.settings, null, 2), { mode: 0o600 });
+                log.debug(`💾 Quota incremented atomically (${this.settings.usage.queriesUsedToday} queries today)`);
+            }
+            catch (error) {
+                log.error(`❌ Could not save quota settings: ${error}`);
+            }
+        });
+    }
+    /**
+     * Refresh settings from disk with file locking
+     *
+     * Use this to ensure you have the latest quota state from disk.
+     */
+    async refreshSettings() {
+        return await withLock(this.settingsPath, async () => {
+            this.settings = this.loadSettings();
+            return { ...this.settings };
+        });
+    }
     /**
      * Check if can create notebook
      */
@@ -318,6 +477,65 @@ export class QuotaManager {
             },
         };
     }
+    /**
+     * Get detailed quota status with remaining counts, warnings, and stop signals
+     * Used to provide visibility to users about when to stop querying for the day
+     */
+    getDetailedStatus() {
+        const today = new Date().toISOString().split("T")[0];
+        // Reset if new day
+        if (this.settings.usage.lastQueryDate !== today) {
+            this.settings.usage.queriesUsedToday = 0;
+            this.settings.usage.lastQueryDate = today;
+        }
+        const { tier, limits, usage } = this.settings;
+        const queriesRemaining = limits.queriesPerDay - usage.queriesUsedToday;
+        const queriesPercentUsed = Math.round((usage.queriesUsedToday / limits.queriesPerDay) * 100);
+        const notebooksRemaining = limits.notebooks - usage.notebooks;
+        const notebooksPercentUsed = Math.round((usage.notebooks / limits.notebooks) * 100);
+        // Calculate next reset time (midnight local time)
+        const tomorrow = new Date();
+        tomorrow.setDate(tomorrow.getDate() + 1);
+        tomorrow.setHours(0, 0, 0, 0);
+        // Build warnings list
+        const warnings = [];
+        if (queriesRemaining <= 0) {
+            warnings.push(`CRITICAL: Daily query limit reached (${usage.queriesUsedToday}/${limits.queriesPerDay}). Wait until tomorrow or upgrade your plan.`);
+        }
+        else if (queriesRemaining <= 5) {
+            warnings.push(`CRITICAL: Only ${queriesRemaining} queries remaining today! Consider stopping soon.`);
+        }
+        else if (queriesRemaining <= 10) {
+            warnings.push(`WARNING: Only ${queriesRemaining} queries remaining today.`);
+        }
+        else if (queriesPercentUsed >= 80) {
+            warnings.push(`INFO: ${queriesPercentUsed}% of daily queries used (${queriesRemaining} remaining).`);
+        }
+        if (notebooksRemaining <= 5) {
+            warnings.push(`WARNING: Only ${notebooksRemaining} notebook slots remaining.`);
+        }
+        return {
+            tier,
+            queries: {
+                used: usage.queriesUsedToday,
+                limit: limits.queriesPerDay,
+                remaining: queriesRemaining,
+                percentUsed: queriesPercentUsed,
+                shouldStop: queriesRemaining <= 5,
+                resetTime: tomorrow.toISOString(),
+            },
+            notebooks: {
+                used: usage.notebooks,
+                limit: limits.notebooks,
+                remaining: notebooksRemaining,
+                percentUsed: notebooksPercentUsed,
+            },
+            sources: {
+                limit: limits.sourcesPerNotebook,
+            },
+            warnings,
+        };
+    }
 }
 // Singleton instance
 let quotaManagerInstance = null;