notebooklm-sdk 0.1.7 → 0.2.0

package/dist/index.js

@@ -1,4 +1,6 @@
-import { readFileSync } from 'fs';
+import { readFileSync, existsSync } from 'fs';
+import { homedir } from 'os';
+import { join } from 'path';
 
 var __defProp = Object.defineProperty;
 var __getOwnPropNames = Object.getOwnPropertyNames;
@@ -451,10 +453,8 @@ function loadCookiesFromFile(filePath) {
   try {
     raw = readFileSync(filePath, "utf-8");
   } catch {
-    throw new AuthError(
-      `Cookie file not found: ${filePath}
-Provide valid Playwright storage state JSON.`
-    );
+    throw new AuthError(`Session file not found: ${filePath}
+Run: npx notebooklm-sdk login`);
   }
   return extractCookiesFromStorageState(JSON.parse(raw));
 }
@@ -497,7 +497,7 @@ function extractCookiesFromStorageState(storageState) {
   }
   if (!cookies["SID"]) {
     throw new AuthError(
-      "Missing required cookie: SID. Provide valid Playwright storage state with Google cookies."
+      "Missing required cookie: SID. Session may be invalid or expired.\nRun: npx notebooklm-sdk login"
     );
   }
   return cookies;
@@ -536,7 +536,7 @@ function extractCsrfToken(html, finalUrl) {
   const match = /"SNlM0e"\s*:\s*"([^"]+)"/.exec(html);
   if (!match?.[1]) {
     if (isGoogleAuthRedirect(finalUrl) || html.includes("accounts.google.com")) {
-      throw new AuthError("Authentication expired or invalid. Cookies may need to be refreshed.");
+      throw new AuthError("Session expired or invalid.\nRun: npx notebooklm-sdk login");
     }
     throw new AuthError("CSRF token (SNlM0e) not found in NotebookLM page HTML.");
   }
@@ -546,7 +546,7 @@ function extractSessionId(html, finalUrl) {
   const match = /"FdrFJe"\s*:\s*"([^"]+)"/.exec(html);
   if (!match?.[1]) {
     if (isGoogleAuthRedirect(finalUrl) || html.includes("accounts.google.com")) {
-      throw new AuthError("Authentication expired or invalid. Cookies may need to be refreshed.");
+      throw new AuthError("Session expired or invalid.\nRun: npx notebooklm-sdk login");
     }
     throw new AuthError("Session ID (FdrFJe) not found in NotebookLM page HTML.");
   }
@@ -555,7 +555,7 @@ function extractSessionId(html, finalUrl) {
 function isGoogleAuthRedirect(url) {
   return url.includes("accounts.google.com") || url.includes("signin");
 }
-async function connect(opts) {
+async function connect(opts = {}) {
   let cookieMap;
   let googleCookieHeader = null;
   if (opts.cookies) {
@@ -572,12 +572,23 @@ async function connect(opts) {
     }
   } else {
     const envCookies = process.env["NOTEBOOKLM_COOKIES"];
-    if (envCookies) {
+    const envFile = process.env["NOTEBOOKLM_COOKIES_FILE"];
+    if (envFile) {
+      cookieMap = loadCookiesFromFile(envFile);
+    } else if (existsSync(DEFAULT_SESSION_FILE)) {
+      const raw = readFileSync(DEFAULT_SESSION_FILE, "utf-8");
+      const storageState = JSON.parse(raw);
+      cookieMap = loadCookiesFromObject(storageState);
+      googleCookieHeader = buildGoogleCookieHeader(storageState);
+    } else if (existsSync("storage_state.json")) {
+      const raw = readFileSync("storage_state.json", "utf-8");
+      const storageState = JSON.parse(raw);
+      cookieMap = loadCookiesFromObject(storageState);
+      googleCookieHeader = buildGoogleCookieHeader(storageState);
+    } else if (envCookies) {
       cookieMap = loadCookiesFromString(envCookies);
     } else {
-      throw new AuthError(
-        "No cookies provided. Pass cookies, cookiesFile, or cookiesObject to connect()."
-      );
+      throw new AuthError("No session found. Run: npx notebooklm-sdk login");
     }
   }
   const { csrfToken, sessionId } = await fetchTokens(cookieMap);
@@ -590,10 +601,11 @@ async function connect(opts) {
     googleCookieHeader: googleCookieHeader ?? cookieHeader
   };
 }
-var NOTEBOOKLM_URL;
+var DEFAULT_SESSION_FILE, NOTEBOOKLM_URL;
 var init_auth = __esm({
   "src/auth.ts"() {
     init_errors();
+    DEFAULT_SESSION_FILE = join(homedir(), ".notebooklm", "session.json");
     NOTEBOOKLM_URL = "https://notebooklm.google.com/";
   }
 });
@@ -711,26 +723,6 @@ function parseArtifact(data, notebookId) {
     _raw: Array.isArray(data) ? data : []
   };
 }
-function parseNote(data) {
-  const id = typeof data[0] === "string" ? data[0] : "";
-  const content = typeof data[1] === "string" ? data[1] : "";
-  const title = typeof data[2] === "string" ? data[2] : null;
-  let createdAt = null;
-  let updatedAt = null;
-  if (Array.isArray(data[3]) && typeof data[3][0] === "number") {
-    try {
-      createdAt = new Date(data[3][0] * 1e3);
-    } catch {
-    }
-  }
-  if (Array.isArray(data[4]) && typeof data[4][0] === "number") {
-    try {
-      updatedAt = new Date(data[4][0] * 1e3);
-    } catch {
-    }
-  }
-  return { id, title, content, createdAt, updatedAt };
-}
 
 // src/api/artifacts.ts
 function tripleNest(ids) {
@@ -740,18 +732,13 @@ function doubleNest(ids) {
   return ids.map((id) => [id]);
 }
 var ArtifactsAPI = class {
-  constructor(rpc, auth) {
+  constructor(rpc, auth, notes) {
     this.rpc = rpc;
     this.auth = auth;
+    this.notes = notes;
   }
   async list(notebookId) {
-    const params = [[2], notebookId, 'NOT artifact.status = "ARTIFACT_STATUS_SUGGESTED"'];
-    const result = await this.rpc.call(RPCMethod.LIST_ARTIFACTS, params, {
-      sourcePath: `/notebook/${notebookId}`,
-      allowNull: true
-    });
-    if (!Array.isArray(result) || !result.length) return [];
-    const rawList = Array.isArray(result[0]) ? result[0] : result;
+    const rawList = await this._listRaw(notebookId);
     const artifacts = [];
     for (const item of rawList) {
       if (Array.isArray(item)) {
@@ -763,6 +750,15 @@ var ArtifactsAPI = class {
     }
     return artifacts;
   }
+  async _listRaw(notebookId) {
+    const params = [[2], notebookId, 'NOT artifact.status = "ARTIFACT_STATUS_SUGGESTED"'];
+    const result = await this.rpc.call(RPCMethod.LIST_ARTIFACTS, params, {
+      sourcePath: `/notebook/${notebookId}`,
+      allowNull: true
+    });
+    if (!Array.isArray(result) || !result.length) return [];
+    return Array.isArray(result[0]) ? result[0] : result;
+  }
   async get(notebookId, artifactId) {
     const artifacts = await this.list(notebookId);
     return artifacts.find((a) => a.id === artifactId) ?? null;
@@ -942,6 +938,37 @@ var ArtifactsAPI = class {
     ];
     return this._callGenerate(notebookId, params);
   }
+  async createDataTable(notebookId, opts = {}) {
+    const language = opts.language ?? "en";
+    const sourceIds = opts.sourceIds ?? await this.rpc.getSourceIds(notebookId);
+    const triple = tripleNest(sourceIds);
+    const params = [
+      [2],
+      notebookId,
+      [
+        null,
+        null,
+        ArtifactTypeCode.DATA_TABLE,
+        triple,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        [null, [opts.instructions ?? null, language]]
+      ]
+    ];
+    return this._callGenerate(notebookId, params);
+  }
   async createReport(notebookId, opts = {}) {
     const format = opts.format ?? "briefing_doc";
     const language = opts.language ?? "en";
@@ -1007,7 +1034,15 @@ ${opts.extraInstructions}` : cfg.prompt;
       sourcePath: `/notebook/${notebookId}`,
       allowNull: true
     });
-    return this._parseGenerationResult(result);
+    const mindMapJson = Array.isArray(result) && Array.isArray(result[0]) && typeof result[0][0] === "string" ? result[0][0] : null;
+    if (!mindMapJson) throw new Error("Mind map generation returned no content");
+    let title = "Mind Map";
+    try {
+      const parsed = JSON.parse(mindMapJson);
+      if (typeof parsed["name"] === "string") title = parsed["name"];
+    } catch {
+    }
+    return this.notes.create(notebookId, mindMapJson, title);
   }
   // ---------------------------------------------------------------------------
   // Polling / download
@@ -1063,6 +1098,48 @@ ${opts.extraInstructions}` : cfg.prompt;
     }
     return null;
   }
+  /** Download a completed slide deck as PDF or PPTX. Returns a Buffer. */
+  async downloadSlideDeck(notebookId, artifactId, format = "pdf") {
+    const rawList = await this._listRaw(notebookId);
+    const raw = rawList.find(
+      (a) => a[0] === artifactId && a[2] === ArtifactTypeCode.SLIDE_DECK
+    );
+    if (!raw) throw new ArtifactNotReadyError("slide_deck", { artifactId });
+    const metadata = raw[16];
+    if (!Array.isArray(metadata)) throw new ArtifactNotReadyError("slide_deck", { artifactId });
+    const url = format === "pptx" ? metadata[4] : metadata[3];
+    if (typeof url !== "string" || !url.startsWith("http")) {
+      throw new ArtifactNotReadyError("slide_deck", { artifactId, status: `no ${format} url` });
+    }
+    return this._fetchMediaWithCookies(url);
+  }
+  /** Download a completed infographic as PNG. Returns a Buffer. */
+  async downloadInfographic(notebookId, artifactId) {
+    const rawList = await this._listRaw(notebookId);
+    const raw = rawList.find(
+      (a) => a[0] === artifactId && a[2] === ArtifactTypeCode.INFOGRAPHIC
+    );
+    if (!raw) throw new ArtifactNotReadyError("infographic", { artifactId });
+    let url = null;
+    for (let i = raw.length - 1; i >= 0; i--) {
+      const item = raw[i];
+      if (Array.isArray(item) && Array.isArray(item[2]) && Array.isArray(item[2][0]) && Array.isArray(item[2][0][1]) && typeof item[2][0][1][0] === "string" && item[2][0][1][0].startsWith("http")) {
+        url = item[2][0][1][0];
+        break;
+      }
+    }
+    if (!url) throw new ArtifactNotReadyError("infographic", { artifactId });
+    return this._fetchMediaWithCookies(url);
+  }
+  /** Get parsed headers and rows from a completed data table artifact. */
+  async getDataTableContent(notebookId, artifactId) {
+    const artifacts = await this._listRaw(notebookId);
+    const raw = artifacts.find(
+      (a) => Array.isArray(a) && a[0] === artifactId && a[2] === ArtifactTypeCode.DATA_TABLE
+    );
+    if (!raw || !Array.isArray(raw) || !Array.isArray(raw[18])) return null;
+    return parseDataTable(raw[18]);
+  }
   // ---------------------------------------------------------------------------
   // Internal
   // ---------------------------------------------------------------------------
@@ -1119,6 +1196,34 @@ ${opts.extraInstructions}` : cfg.prompt;
     return { artifactId: null, status: "failed" };
   }
 };
+function extractCellText(cell) {
+  if (typeof cell === "string") return cell;
+  if (typeof cell === "number") return "";
+  if (Array.isArray(cell)) return cell.map(extractCellText).join("");
+  return "";
+}
+function parseDataTable(rawData) {
+  try {
+    const nav = rawData;
+    const rowsArray = nav[0][0][0][0][4][2];
+    if (!rowsArray?.length) throw new Error("Empty data table");
+    const headers = [];
+    const rows = [];
+    for (let i = 0; i < rowsArray.length; i++) {
+      const rowSection = rowsArray[i];
+      if (!Array.isArray(rowSection) || rowSection.length < 3) continue;
+      const cellArray = rowSection[2];
+      if (!Array.isArray(cellArray)) continue;
+      const values = cellArray.map(extractCellText);
+      if (i === 0) headers.push(...values);
+      else rows.push(values);
+    }
+    if (!headers.length) throw new Error("No headers found");
+    return { headers, rows };
+  } catch (e) {
+    throw new Error(`Failed to parse data table: ${e}`);
+  }
+}
 function sleep(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
@@ -1455,61 +1560,69 @@ var NotesAPI = class {
     this.rpc = rpc;
   }
   async list(notebookId) {
-    const params = [notebookId, [2]];
-    const result = await this.rpc.call(RPCMethod.GET_NOTES_AND_MIND_MAPS, params, {
-      sourcePath: `/notebook/${notebookId}`
-    });
-    const notes = [];
-    const mindMaps = [];
-    if (!Array.isArray(result)) return { notes, mindMaps };
-    try {
-      const notesData = result[0];
-      if (Array.isArray(notesData)) {
-        for (const n of notesData) {
-          if (Array.isArray(n)) notes.push(parseNote(n));
-        }
-      }
-      const mapsData = result[1];
-      if (Array.isArray(mapsData)) {
-        for (const m of mapsData) {
-          if (Array.isArray(m)) {
-            mindMaps.push({
-              id: typeof m[0] === "string" ? m[0] : "",
-              title: typeof m[2] === "string" ? m[2] : null,
-              content: typeof m[1] === "string" ? m[1] : "",
-              createdAt: Array.isArray(m[3]) && typeof m[3][0] === "number" ? new Date(m[3][0] * 1e3) : null
-            });
-          }
-        }
-      }
-    } catch {
-    }
-    return { notes, mindMaps };
+    const all = await this._fetchAll(notebookId);
+    return all.filter((n) => !this._isMindMap(n.content));
+  }
+  async listMindMaps(notebookId) {
+    const all = await this._fetchAll(notebookId);
+    return all.filter((n) => this._isMindMap(n.content));
   }
   async create(notebookId, content, title) {
-    const params = [notebookId, content, title ?? null, [2]];
-    const result = await this.rpc.call(RPCMethod.CREATE_NOTE, params, {
-      sourcePath: `/notebook/${notebookId}`
+    const createParams = [notebookId, "", [1], null, "New Note"];
+    const result = await this.rpc.call(RPCMethod.CREATE_NOTE, createParams, {
+      sourcePath: `/notebook/${notebookId}`,
+      allowNull: true
     });
-    if (Array.isArray(result)) return parseNote(result);
-    throw new Error("Could not parse note creation response");
+    const noteId = Array.isArray(result) && Array.isArray(result[0]) && typeof result[0][0] === "string" ? result[0][0] : Array.isArray(result) && typeof result[0] === "string" ? result[0] : null;
+    if (!noteId) throw new Error("CREATE_NOTE did not return a note ID");
+    await this.update(notebookId, noteId, content, title ?? "New Note");
+    return { id: noteId, title: title ?? null, content, createdAt: null, updatedAt: /* @__PURE__ */ new Date() };
   }
   async update(notebookId, noteId, content, title) {
-    const params = [notebookId, noteId, content, title ?? null, [2]];
-    const result = await this.rpc.call(RPCMethod.UPDATE_NOTE, params, {
-      sourcePath: `/notebook/${notebookId}`
+    const params = [notebookId, noteId, [[[content, title ?? "New Note", [], 0]]]];
+    await this.rpc.call(RPCMethod.UPDATE_NOTE, params, {
+      sourcePath: `/notebook/${notebookId}`,
+      allowNull: true
     });
-    if (Array.isArray(result)) return parseNote(result);
     return { id: noteId, title: title ?? null, content, createdAt: null, updatedAt: /* @__PURE__ */ new Date() };
   }
   async delete(notebookId, noteId) {
-    const params = [notebookId, noteId, [2]];
+    const params = [notebookId, null, [noteId]];
     await this.rpc.call(RPCMethod.DELETE_NOTE, params, {
       sourcePath: `/notebook/${notebookId}`,
       allowNull: true
     });
     return true;
   }
+  async _fetchAll(notebookId) {
+    const result = await this.rpc.call(RPCMethod.GET_NOTES_AND_MIND_MAPS, [notebookId], {
+      sourcePath: `/notebook/${notebookId}`,
+      allowNull: true
+    });
+    if (!Array.isArray(result) || !Array.isArray(result[0])) return [];
+    const notes = [];
+    for (const item of result[0]) {
+      if (!Array.isArray(item) || typeof item[0] !== "string") continue;
+      if (item[1] === null && item[2] === 2) continue;
+      const content = this._extractContent(item);
+      notes.push(this._parseItem(item, notebookId, content));
+    }
+    return notes;
+  }
+  _isMindMap(content) {
+    return content.includes('"children":') || content.includes('"nodes":');
+  }
+  _extractContent(item) {
+    if (typeof item[1] === "string") return item[1];
+    if (Array.isArray(item[1]) && typeof item[1][1] === "string") return item[1][1];
+    return "";
+  }
+  _parseItem(item, _notebookId, content) {
+    const inner = Array.isArray(item[1]) ? item[1] : null;
+    const title = inner && typeof inner[4] === "string" && inner[4] ? inner[4] : null;
+    const createdAt = Array.isArray(item[3]) && typeof item[3][0] === "number" ? new Date(item[3][0] * 1e3) : null;
+    return { id: item[0], title, content, createdAt, updatedAt: null };
+  }
 };
 
 // src/api/research.ts
@@ -1646,7 +1759,7 @@ var ResearchAPI = class {
     const webSources = sources.filter((s) => s.url && !reportSourceSet.has(s));
     if (!webSources.length && !reportSources.length) return [];
     const sourceArray = [
-      ...reportSources.map((s) => buildReportEntry(s.title, s.reportMarkdown)),
+      ...reportSources.filter((s) => s.reportMarkdown).map((s) => buildReportEntry(s.title, s.reportMarkdown)),
       ...webSources.map((s) => buildWebEntry(s.url, s.title))
     ];
     const params = [null, [1], effectiveTaskId, notebookId, sourceArray];
@@ -2376,9 +2489,9 @@ var NotebookLMClient = class _NotebookLMClient {
     const rpc = new RPCCore(auth, opts.timeoutMs);
     this.notebooks = new NotebooksAPI(rpc);
     this.sources = new SourcesAPI(rpc, auth);
-    this.artifacts = new ArtifactsAPI(rpc, auth);
-    this.chat = new ChatAPI(rpc, auth);
     this.notes = new NotesAPI(rpc);
+    this.artifacts = new ArtifactsAPI(rpc, auth, this.notes);
+    this.chat = new ChatAPI(rpc, auth);
     this.research = new ResearchAPI(rpc);
     this.settings = new SettingsAPI(rpc);
     this.sharing = new SharingAPI(rpc);
@@ -2395,7 +2508,7 @@ var NotebookLMClient = class _NotebookLMClient {
    * Connect to NotebookLM using cookies.
    * Fetches CSRF and session tokens from the NotebookLM homepage.
    */
-  static async connect(opts, clientOpts = {}) {
+  static async connect(opts = {}, clientOpts = {}) {
     const auth = await connect(opts);
     return new _NotebookLMClient(auth, clientOpts);
   }