notebooklm-sdk 0.1.8 → 0.2.0

package/dist/index.cjs

@@ -455,10 +455,8 @@ function loadCookiesFromFile(filePath) {
   try {
     raw = fs.readFileSync(filePath, "utf-8");
   } catch {
-    throw new exports.AuthError(
-      `Session file not found: ${filePath}
-Run: npx notebooklm-sdk login`
-    );
+    throw new exports.AuthError(`Session file not found: ${filePath}
+Run: npx notebooklm-sdk login`);
   }
   return extractCookiesFromStorageState(JSON.parse(raw));
 }
@@ -592,9 +590,7 @@ async function connect(opts = {}) {
     } else if (envCookies) {
       cookieMap = loadCookiesFromString(envCookies);
     } else {
-      throw new exports.AuthError(
-        "No session found. Run: npx notebooklm-sdk login"
-      );
+      throw new exports.AuthError("No session found. Run: npx notebooklm-sdk login");
     }
   }
   const { csrfToken, sessionId } = await fetchTokens(cookieMap);
@@ -729,26 +725,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) {
@@ -758,18 +734,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(exports.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)) {
@@ -781,6 +752,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(exports.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;
@@ -960,6 +940,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,
+        exports.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";
@@ -1025,7 +1036,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
@@ -1081,6 +1100,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] === exports.ArtifactTypeCode.SLIDE_DECK
+    );
+    if (!raw) throw new exports.ArtifactNotReadyError("slide_deck", { artifactId });
+    const metadata = raw[16];
+    if (!Array.isArray(metadata)) throw new exports.ArtifactNotReadyError("slide_deck", { artifactId });
+    const url = format === "pptx" ? metadata[4] : metadata[3];
+    if (typeof url !== "string" || !url.startsWith("http")) {
+      throw new exports.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] === exports.ArtifactTypeCode.INFOGRAPHIC
+    );
+    if (!raw) throw new exports.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 exports.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] === exports.ArtifactTypeCode.DATA_TABLE
+    );
+    if (!raw || !Array.isArray(raw) || !Array.isArray(raw[18])) return null;
+    return parseDataTable(raw[18]);
+  }
   // ---------------------------------------------------------------------------
   // Internal
   // ---------------------------------------------------------------------------
@@ -1137,6 +1198,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));
 }
@@ -1473,61 +1562,69 @@ var NotesAPI = class {
     this.rpc = rpc;
   }
   async list(notebookId) {
-    const params = [notebookId, [2]];
-    const result = await this.rpc.call(exports.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(exports.RPCMethod.CREATE_NOTE, params, {
-      sourcePath: `/notebook/${notebookId}`
+    const createParams = [notebookId, "", [1], null, "New Note"];
+    const result = await this.rpc.call(exports.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(exports.RPCMethod.UPDATE_NOTE, params, {
-      sourcePath: `/notebook/${notebookId}`
+    const params = [notebookId, noteId, [[[content, title ?? "New Note", [], 0]]]];
+    await this.rpc.call(exports.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(exports.RPCMethod.DELETE_NOTE, params, {
       sourcePath: `/notebook/${notebookId}`,
       allowNull: true
     });
     return true;
   }
+  async _fetchAll(notebookId) {
+    const result = await this.rpc.call(exports.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
@@ -1664,7 +1761,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];
@@ -2394,9 +2491,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);