notebooklm-sdk 0.1.5 → 0.1.6
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +13 -15
- package/dist/{auth-DTugbaf3.d.cts → auth-Ba2hsZW_.d.cts} +1 -1
- package/dist/{auth-DTugbaf3.d.ts → auth-Ba2hsZW_.d.ts} +1 -1
- package/dist/auth.d.cts +2 -2
- package/dist/auth.d.ts +2 -2
- package/dist/bin.cjs +139 -0
- package/dist/bin.cjs.map +1 -0
- package/dist/bin.d.cts +1 -0
- package/dist/bin.d.ts +1 -0
- package/dist/bin.js +137 -0
- package/dist/bin.js.map +1 -0
- package/dist/index.d.cts +2 -2
- package/dist/index.d.ts +2 -2
- package/package.json +5 -2
package/README.md
CHANGED
|
@@ -23,12 +23,19 @@ bun add notebooklm-sdk
|
|
|
23
23
|
|
|
24
24
|
### Quick Login (Recommended)
|
|
25
25
|
|
|
26
|
-
|
|
26
|
+
First, install playwright:
|
|
27
27
|
|
|
28
28
|
```bash
|
|
29
|
-
bun
|
|
29
|
+
bun add -d playwright
|
|
30
|
+
bunx playwright install chromium
|
|
31
|
+
```
|
|
32
|
+
|
|
33
|
+
Then, authenticate using the CLI:
|
|
34
|
+
|
|
35
|
+
```bash
|
|
36
|
+
npx notebooklm-sdk login
|
|
30
37
|
# or
|
|
31
|
-
|
|
38
|
+
bun x notebooklm-sdk login
|
|
32
39
|
```
|
|
33
40
|
|
|
34
41
|
This opens a real browser for Google sign-in and generates a
|
|
@@ -44,12 +51,6 @@ const client = await NotebookLMClient.connect({
|
|
|
44
51
|
});
|
|
45
52
|
```
|
|
46
53
|
|
|
47
|
-
Requires:
|
|
48
|
-
|
|
49
|
-
```bash
|
|
50
|
-
npx playwright install chromium
|
|
51
|
-
```
|
|
52
|
-
|
|
53
54
|
<details>
|
|
54
55
|
<summary>Manual Authentication</summary>
|
|
55
56
|
|
|
@@ -256,13 +257,10 @@ await client.settings.setOutputLanguage("ja");
|
|
|
256
257
|
|
|
257
258
|
Runnable scripts are in [`examples/`](./examples).
|
|
258
259
|
|
|
259
|
-
|
|
260
|
-
|
|
261
|
-
```
|
|
262
|
-
NOTEBOOKLM_COOKIE=...
|
|
263
|
-
```
|
|
260
|
+
**Setup:**
|
|
264
261
|
|
|
265
|
-
|
|
262
|
+
1. `npm run login` to create `storage_state.json`.
|
|
263
|
+
2. Run any example below.
|
|
266
264
|
|
|
267
265
|
```bash
|
|
268
266
|
# for auto login
|
|
@@ -52,4 +52,4 @@ interface ConnectOptions {
|
|
|
52
52
|
}
|
|
53
53
|
declare function connect(opts: ConnectOptions): Promise<AuthTokens>;
|
|
54
54
|
|
|
55
|
-
export { type AuthTokens as A, type
|
|
55
|
+
export { type AuthTokens as A, type CookieMap as C, type ConnectOptions as a, buildCookieHeader as b, buildGoogleCookieHeader as c, connect as d, loadCookiesFromMap as e, fetchTokens as f, loadCookiesFromObject as g, loadCookiesFromString as h, loadCookiesFromFile as l };
|
|
@@ -52,4 +52,4 @@ interface ConnectOptions {
|
|
|
52
52
|
}
|
|
53
53
|
declare function connect(opts: ConnectOptions): Promise<AuthTokens>;
|
|
54
54
|
|
|
55
|
-
export { type AuthTokens as A, type
|
|
55
|
+
export { type AuthTokens as A, type CookieMap as C, type ConnectOptions as a, buildCookieHeader as b, buildGoogleCookieHeader as c, connect as d, loadCookiesFromMap as e, fetchTokens as f, loadCookiesFromObject as g, loadCookiesFromString as h, loadCookiesFromFile as l };
|
package/dist/auth.d.cts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import {
|
|
2
|
-
export { A as AuthTokens,
|
|
1
|
+
import { C as CookieMap } from './auth-Ba2hsZW_.cjs';
|
|
2
|
+
export { A as AuthTokens, a as ConnectOptions, b as buildCookieHeader, c as buildGoogleCookieHeader, d as connect, f as fetchTokens, l as loadCookiesFromFile, e as loadCookiesFromMap, g as loadCookiesFromObject, h as loadCookiesFromString } from './auth-Ba2hsZW_.cjs';
|
|
3
3
|
|
|
4
4
|
interface LoginOptions {
|
|
5
5
|
/**
|
package/dist/auth.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import {
|
|
2
|
-
export { A as AuthTokens,
|
|
1
|
+
import { C as CookieMap } from './auth-Ba2hsZW_.js';
|
|
2
|
+
export { A as AuthTokens, a as ConnectOptions, b as buildCookieHeader, c as buildGoogleCookieHeader, d as connect, f as fetchTokens, l as loadCookiesFromFile, e as loadCookiesFromMap, g as loadCookiesFromObject, h as loadCookiesFromString } from './auth-Ba2hsZW_.js';
|
|
3
3
|
|
|
4
4
|
interface LoginOptions {
|
|
5
5
|
/**
|
package/dist/bin.cjs
ADDED
|
@@ -0,0 +1,139 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
'use strict';
|
|
3
|
+
|
|
4
|
+
var fs = require('fs');
|
|
5
|
+
var path = require('path');
|
|
6
|
+
var playwright = require('playwright');
|
|
7
|
+
|
|
8
|
+
// src/types/errors.ts
|
|
9
|
+
var NotebookLMError = class extends Error {
|
|
10
|
+
constructor(message) {
|
|
11
|
+
super(message);
|
|
12
|
+
this.name = this.constructor.name;
|
|
13
|
+
}
|
|
14
|
+
};
|
|
15
|
+
var RPCError = class extends NotebookLMError {
|
|
16
|
+
methodId;
|
|
17
|
+
rawResponse;
|
|
18
|
+
rpcCode;
|
|
19
|
+
foundIds;
|
|
20
|
+
constructor(message, opts = {}) {
|
|
21
|
+
super(message);
|
|
22
|
+
this.methodId = opts.methodId;
|
|
23
|
+
this.rawResponse = opts.rawResponse ? opts.rawResponse.slice(0, 500) : void 0;
|
|
24
|
+
this.rpcCode = opts.rpcCode;
|
|
25
|
+
this.foundIds = opts.foundIds ?? [];
|
|
26
|
+
}
|
|
27
|
+
};
|
|
28
|
+
var AuthError = class extends RPCError {
|
|
29
|
+
};
|
|
30
|
+
|
|
31
|
+
// src/auth.ts
|
|
32
|
+
function loadCookiesFromObject(storageState) {
|
|
33
|
+
return extractCookiesFromStorageState(storageState);
|
|
34
|
+
}
|
|
35
|
+
function extractCookiesFromStorageState(storageState) {
|
|
36
|
+
const cookies = {};
|
|
37
|
+
for (const cookie of storageState.cookies ?? []) {
|
|
38
|
+
const { domain, name, value } = cookie;
|
|
39
|
+
if (!isAllowedDomain(domain) || !name) continue;
|
|
40
|
+
const isBase = domain === ".google.com";
|
|
41
|
+
if (!(name in cookies) || isBase) {
|
|
42
|
+
cookies[name] = value;
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
if (!cookies["SID"]) {
|
|
46
|
+
throw new AuthError(
|
|
47
|
+
"Missing required cookie: SID. Provide valid Playwright storage state with Google cookies."
|
|
48
|
+
);
|
|
49
|
+
}
|
|
50
|
+
return cookies;
|
|
51
|
+
}
|
|
52
|
+
function isAllowedDomain(domain) {
|
|
53
|
+
if (domain === ".google.com" || domain === "notebooklm.google.com" || domain === ".googleusercontent.com") {
|
|
54
|
+
return true;
|
|
55
|
+
}
|
|
56
|
+
if (domain.startsWith(".google.")) {
|
|
57
|
+
return true;
|
|
58
|
+
}
|
|
59
|
+
return false;
|
|
60
|
+
}
|
|
61
|
+
|
|
62
|
+
// src/auth/browser.ts
|
|
63
|
+
var NOTEBOOKLM_URL = "https://notebooklm.google.com/";
|
|
64
|
+
var GOOGLE_ACCOUNTS_URL = "https://accounts.google.com/";
|
|
65
|
+
async function login(opts = {}) {
|
|
66
|
+
const { persistFolder, headless = false, browserType = "chromium" } = opts;
|
|
67
|
+
let context;
|
|
68
|
+
const launchOptions = {
|
|
69
|
+
headless,
|
|
70
|
+
args: ["--disable-blink-features=AutomationControlled"]
|
|
71
|
+
};
|
|
72
|
+
if (persistFolder) {
|
|
73
|
+
context = await playwright.chromium.launchPersistentContext(persistFolder, {
|
|
74
|
+
...launchOptions,
|
|
75
|
+
channel: browserType === "msedge" ? "msedge" : void 0
|
|
76
|
+
});
|
|
77
|
+
} else {
|
|
78
|
+
const browser = await playwright.chromium.launch({
|
|
79
|
+
...launchOptions,
|
|
80
|
+
channel: browserType === "msedge" ? "msedge" : void 0
|
|
81
|
+
});
|
|
82
|
+
context = await browser.newContext();
|
|
83
|
+
}
|
|
84
|
+
const page = context.pages()[0] || await context.newPage();
|
|
85
|
+
await page.goto(NOTEBOOKLM_URL);
|
|
86
|
+
if (page.url().includes("accounts.google.com")) {
|
|
87
|
+
console.log("Please log in to Google in the browser window...");
|
|
88
|
+
await page.waitForURL(
|
|
89
|
+
(url) => {
|
|
90
|
+
return url.hostname === "notebooklm.google.com" && !url.pathname.includes("/login");
|
|
91
|
+
},
|
|
92
|
+
{ timeout: 0 }
|
|
93
|
+
);
|
|
94
|
+
}
|
|
95
|
+
await page.goto(GOOGLE_ACCOUNTS_URL, { waitUntil: "load" });
|
|
96
|
+
await page.goto(NOTEBOOKLM_URL, { waitUntil: "load" });
|
|
97
|
+
const storageState = await context.storageState();
|
|
98
|
+
const cookies = loadCookiesFromObject(storageState);
|
|
99
|
+
await context.close();
|
|
100
|
+
return {
|
|
101
|
+
cookies,
|
|
102
|
+
storageState,
|
|
103
|
+
cookieHeader: Object.entries(cookies).map(([k, v]) => `${k}=${v}`).join("; ")
|
|
104
|
+
};
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
// src/bin.ts
|
|
108
|
+
async function run() {
|
|
109
|
+
const args = process.argv.slice(2);
|
|
110
|
+
const command = args[0];
|
|
111
|
+
if (command === "login") {
|
|
112
|
+
console.log("\u{1F680} Starting browser login flow...");
|
|
113
|
+
console.log("A browser window will open. Please log in to your Google account.");
|
|
114
|
+
try {
|
|
115
|
+
const authResult = await login({
|
|
116
|
+
persistFolder: "./.auth_profile",
|
|
117
|
+
headless: false
|
|
118
|
+
});
|
|
119
|
+
const outPath = path.resolve(process.cwd(), "storage_state.json");
|
|
120
|
+
fs.writeFileSync(outPath, JSON.stringify(authResult.storageState, null, 2));
|
|
121
|
+
console.log("\n\u2705 Login successful!");
|
|
122
|
+
console.log(`Saved session to: ${outPath}`);
|
|
123
|
+
console.log("\nNext steps:");
|
|
124
|
+
console.log("In your code, connect using:");
|
|
125
|
+
console.log("const client = await NotebookLMClient.connect({ cookiesFile: 'storage_state.json' });");
|
|
126
|
+
} catch (error) {
|
|
127
|
+
console.error("\u274C Login failed:", error);
|
|
128
|
+
process.exit(1);
|
|
129
|
+
}
|
|
130
|
+
} else {
|
|
131
|
+
console.log("NotebookLM SDK CLI");
|
|
132
|
+
console.log("\nUsage:");
|
|
133
|
+
console.log(" npx notebooklm-sdk login Start browser login flow");
|
|
134
|
+
process.exit(command ? 1 : 0);
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
run();
|
|
138
|
+
//# sourceMappingURL=bin.cjs.map
|
|
139
|
+
//# sourceMappingURL=bin.cjs.map
|
package/dist/bin.cjs.map
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/types/errors.ts","../src/auth.ts","../src/auth/browser.ts","../src/bin.ts"],"names":["chromium","resolve","writeFileSync"],"mappings":";;;;;;;;AAOO,IAAM,eAAA,GAAN,cAA8B,KAAA,CAAM;AAAA,EACzC,YAAY,OAAA,EAAiB;AAC3B,IAAA,KAAA,CAAM,OAAO,CAAA;AACb,IAAA,IAAA,CAAK,IAAA,GAAO,KAAK,WAAA,CAAY,IAAA;AAAA,EAC/B;AACF,CAAA;AAuBO,IAAM,QAAA,GAAN,cAAuB,eAAA,CAAgB;AAAA,EACnC,QAAA;AAAA,EACA,WAAA;AAAA,EACA,OAAA;AAAA,EACA,QAAA;AAAA,EAET,WAAA,CACE,OAAA,EACA,IAAA,GAKI,EAAC,EACL;AACA,IAAA,KAAA,CAAM,OAAO,CAAA;AACb,IAAA,IAAA,CAAK,WAAW,IAAA,CAAK,QAAA;AACrB,IAAA,IAAA,CAAK,WAAA,GAAc,KAAK,WAAA,GAAc,IAAA,CAAK,YAAY,KAAA,CAAM,CAAA,EAAG,GAAG,CAAA,GAAI,MAAA;AACvE,IAAA,IAAA,CAAK,UAAU,IAAA,CAAK,OAAA;AACpB,IAAA,IAAA,CAAK,QAAA,GAAW,IAAA,CAAK,QAAA,IAAY,EAAC;AAAA,EACpC;AACF,CAAA;AAEO,IAAM,SAAA,GAAN,cAAwB,QAAA,CAAS;AAAC,CAAA;;;ACxBlC,SAAS,sBAAsB,YAAA,EAExB;AACZ,EAAA,OAAO,+BAA+B,YAAY,CAAA;AACpD;AAkCA,SAAS,+BAA+B,YAAA,EAE1B;AACZ,EAAA,MAAM,UAAqB,EAAC;AAG5B,EAAA,KAAA,MAAW,MAAA,IAAU,YAAA,CAAa,OAAA,IAAW,EAAC,EAAG;AAC/C,IAAA,MAAM,EAAE,MAAA,EAAQ,IAAA,EAAM,KAAA,EAAM,GAAI,MAAA;AAChC,IAAA,IAAI,CAAC,eAAA,CAAgB,MAAM,CAAA,IAAK,CAAC,IAAA,EAAM;AAEvC,IAAA,MAAM,SAAS,MAAA,KAAW,aAAA;AAC1B,IAAA,IAAI,EAAE,IAAA,IAAQ,OAAA,CAAA,IAAY,MAAA,EAAQ;AAChC,MAAA,OAAA,CAAQ,IAAI,CAAA,GAAI,KAAA;AACI,IACtB;AAAA,EACF;AAEA,EAAA,IAAI,CAAC,OAAA,CAAQ,KAAK,CAAA,EAAG;AACnB,IAAA,MAAM,IAAI,SAAA;AAAA,MACR;AAAA,KAEF;AAAA,EACF;AACA,EAAA,OAAO,OAAA;AACT;AAEA,SAAS,gBAAgB,MAAA,EAAyB;AAChD,EAAA,IACE,MAAA,KAAW,aAAA,IACX,MAAA,KAAW,uBAAA,IACX,WAAW,wBAAA,EACX;AACA,IAAA,OAAO,IAAA;AAAA,EACT;AACA,EAAA,IAAI,MAAA,CAAO,UAAA,CAAW,UAAU,CAAA,EAAG;AACjC,IAAA,OAAO,IAAA;AAAA,EACT;AACA,EAAA,OAAO,KAAA;AACT;;;AC1FA,IAAM,cAAA,GAAiB,gCAAA;AACvB,IAAM,mBAAA,GAAsB,8BAAA;AAW5B,eAAsB,KAAA,CAAM,IAAA,GAAqB,EAAC,EAI/C;AACD,EAAA,MAAM,EAAE,aAAA,EAAe,QAAA,GAAW,KAAA,EAAO,WAAA,GAAc,YAAW,GAAI,IAAA;AAEtE,EAAA,IAAI,OAAA;AAEJ,EAAA,MAAM,aAAA,GAAgB;AAAA,IACpB,QAAA;AAAA,IACA,IAAA,EAAM,CAAC,+CAA+C;AAAA,GACxD;AAEA,EAAA,IAAI,aAAA,EAAe;AACjB,IAAA,OAAA,GAAU,MAAMA,mBAAA,CAAS,uBAAA,CAAwB,aAAA,EAAe;AAAA,MAC9D,GAAG,aAAA;AAAA,MACH,OAAA,EAAS,WAAA,KAAgB,QAAA,GAAW,QAAA,GAAW;AAAA,KAChD,CAAA;AAAA,EACH,CAAA,MAAO;AACL,IAAA,MAAM,OAAA,GAAU,MAAMA,mBAAA,CAAS,MAAA,CAAO;AAAA,MACpC,GAAG,aAAA;AAAA,MACH,OAAA,EAAS,WAAA,KAAgB,QAAA,GAAW,QAAA,GAAW;AAAA,KAChD,CAAA;AACD,IAAA,OAAA,GAAU,MAAM,QAAQ,UAAA,EAAW;AAAA,EACrC;AAEA,EAAA,MAAM,IAAA,GAAO,QAAQ,KAAA,EAAM,CAAE,CAAC,CAAA,IAAM,MAAM,QAAQ,OAAA,EAAQ;AAC1D,EAAA,MAAM,IAAA,CAAK,KAAK,cAAc,CAAA;AAG9B,EAAA,IAAI,IAAA,CAAK,GAAA,EAAI,CAAE,QAAA,CAAS,qBAAqB,CAAA,EAAG;AAC9C,IAAA,OAAA,CAAQ,IAAI,kDAAkD,CAAA;AAI9D,IAAA,MAAM,IAAA,CAAK,UAAA;AAAA,MACT,CAAC,GAAA,KAAQ;AACP,QAAA,OAAO,IAAI,QAAA,KAAa,uBAAA,IAA2B,CAAC,GAAA,CAAI,QAAA,CAAS,SAAS,QAAQ,CAAA;AAAA,MACpF,CAAA;AAAA,MACA,EAAE,SAAS,CAAA;AAAE,KACf;AAAA,EACF;AAGA,EAAA,MAAM,KAAK,IAAA,CAAK,mBAAA,EAAqB,EAAE,SAAA,EAAW,QAAQ,CAAA;AAC1D,EAAA,MAAM,KAAK,IAAA,CAAK,cAAA,EAAgB,EAAE,SAAA,EAAW,QAAQ,CAAA;AAErD,EAAA,MAAM,YAAA,GAAe,MAAM,OAAA,CAAQ,YAAA,EAAa;AAChD,EAAA,MAAM,OAAA,GAAU,sBAAsB,YAAmB,CAAA;AAEzD,EAAA,MAAM,QAAQ,KAAA,EAAM;AAEpB,EAAA,OAAO;AAAA,IACL,OAAA;AAAA,IACA,YAAA;AAAA,IACA,cAAc,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,CACjC,GAAA,CAAI,CAAC,CAAC,CAAA,EAAG,CAAC,CAAA,KAAM,GAAG,CAAC,CAAA,CAAA,EAAI,CAAC,CAAA,CAAE,CAAA,CAC3B,KAAK,IAAI;AAAA,GACd;AACF;;;ACvFA,eAAe,GAAA,GAAM;AACnB,EAAA,MAAM,IAAA,GAAO,OAAA,CAAQ,IAAA,CAAK,KAAA,CAAM,CAAC,CAAA;AACjC,EAAA,MAAM,OAAA,GAAU,KAAK,CAAC,CAAA;AAEtB,EAAA,IAAI,YAAY,OAAA,EAAS;AACvB,IAAA,OAAA,CAAQ,IAAI,0CAAmC,CAAA;AAC/C,IAAA,OAAA,CAAQ,IAAI,mEAAmE,CAAA;AAE/E,IAAA,IAAI;AACF,MAAA,MAAM,UAAA,GAAa,MAAM,KAAA,CAAM;AAAA,QAC7B,aAAA,EAAe,iBAAA;AAAA,QACf,QAAA,EAAU;AAAA,OACX,CAAA;AAED,MAAA,MAAM,OAAA,GAAUC,YAAA,CAAQ,OAAA,CAAQ,GAAA,IAAO,oBAAoB,CAAA;AAC3D,MAAAC,gBAAA,CAAc,SAAS,IAAA,CAAK,SAAA,CAAU,WAAW,YAAA,EAAc,IAAA,EAAM,CAAC,CAAC,CAAA;AAEvE,MAAA,OAAA,CAAQ,IAAI,4BAAuB,CAAA;AACnC,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kBAAA,EAAqB,OAAO,CAAA,CAAE,CAAA;AAC1C,MAAA,OAAA,CAAQ,IAAI,eAAe,CAAA;AAC3B,MAAA,OAAA,CAAQ,IAAI,8BAA8B,CAAA;AAC1C,MAAA,OAAA,CAAQ,IAAI,uFAAuF,CAAA;AAAA,IACrG,SAAS,KAAA,EAAO;AACd,MAAA,OAAA,CAAQ,KAAA,CAAM,wBAAmB,KAAK,CAAA;AACtC,MAAA,OAAA,CAAQ,KAAK,CAAC,CAAA;AAAA,IAChB;AAAA,EACF,CAAA,MAAO;AACL,IAAA,OAAA,CAAQ,IAAI,oBAAoB,CAAA;AAChC,IAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AACtB,IAAA,OAAA,CAAQ,IAAI,wDAAwD,CAAA;AACpE,IAAA,OAAA,CAAQ,IAAA,CAAK,OAAA,GAAU,CAAA,GAAI,CAAC,CAAA;AAAA,EAC9B;AACF;AAEA,GAAA,EAAI","file":"bin.cjs","sourcesContent":["/**\n * All exceptions for notebooklm-sdk.\n *\n * All errors extend NotebookLMError so you can catch everything with:\n * try { ... } catch (e) { if (e instanceof NotebookLMError) ... }\n */\n\nexport class NotebookLMError extends Error {\n constructor(message: string) {\n super(message);\n this.name = this.constructor.name;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Network (transport-level, before RPC processing)\n// ---------------------------------------------------------------------------\n\nexport class NetworkError extends NotebookLMError {\n readonly methodId?: string;\n readonly originalError?: Error;\n\n constructor(message: string, opts: { methodId?: string; originalError?: Error } = {}) {\n super(message);\n this.methodId = opts.methodId;\n this.originalError = opts.originalError;\n }\n}\n\nexport class RPCTimeoutError extends NetworkError {}\n\n// ---------------------------------------------------------------------------\n// RPC Protocol (after connection established)\n// ---------------------------------------------------------------------------\n\nexport class RPCError extends NotebookLMError {\n readonly methodId?: string;\n readonly rawResponse?: string;\n readonly rpcCode?: string | number;\n readonly foundIds: string[];\n\n constructor(\n message: string,\n opts: {\n methodId?: string;\n rawResponse?: string;\n rpcCode?: string | number;\n foundIds?: string[];\n } = {},\n ) {\n super(message);\n this.methodId = opts.methodId;\n this.rawResponse = opts.rawResponse ? opts.rawResponse.slice(0, 500) : undefined;\n this.rpcCode = opts.rpcCode;\n this.foundIds = opts.foundIds ?? [];\n }\n}\n\nexport class AuthError extends RPCError {}\n\nexport class RateLimitError extends RPCError {\n readonly retryAfter?: number;\n\n constructor(\n message: string,\n opts: {\n retryAfter?: number;\n methodId?: string;\n rawResponse?: string;\n rpcCode?: string | number;\n foundIds?: string[];\n } = {},\n ) {\n super(message, opts);\n this.retryAfter = opts.retryAfter;\n }\n}\n\nexport class ServerError extends RPCError {\n readonly statusCode?: number;\n\n constructor(\n message: string,\n opts: {\n statusCode?: number;\n methodId?: string;\n rawResponse?: string;\n rpcCode?: string | number;\n } = {},\n ) {\n super(message, opts);\n this.statusCode = opts.statusCode;\n }\n}\n\nexport class ClientError extends RPCError {\n readonly statusCode?: number;\n\n constructor(\n message: string,\n opts: {\n statusCode?: number;\n methodId?: string;\n rawResponse?: string;\n rpcCode?: string | number;\n } = {},\n ) {\n super(message, opts);\n this.statusCode = opts.statusCode;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Domain: Notebooks\n// ---------------------------------------------------------------------------\n\nexport class NotebookError extends NotebookLMError {}\n\nexport class NotebookNotFoundError extends NotebookError {\n readonly notebookId: string;\n\n constructor(notebookId: string) {\n super(`Notebook not found: ${notebookId}`);\n this.notebookId = notebookId;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Domain: Sources\n// ---------------------------------------------------------------------------\n\nexport class SourceError extends NotebookLMError {}\n\nexport class SourceNotFoundError extends SourceError {\n readonly sourceId: string;\n\n constructor(sourceId: string) {\n super(`Source not found: ${sourceId}`);\n this.sourceId = sourceId;\n }\n}\n\nexport class SourceAddError extends SourceError {\n readonly url: string;\n readonly cause?: Error;\n\n constructor(url: string, opts: { cause?: Error; message?: string } = {}) {\n super(\n opts.message ??\n `Failed to add source: ${url}\\n` +\n \"Possible causes:\\n\" +\n \" - URL is invalid or inaccessible\\n\" +\n \" - Content is behind a paywall or requires authentication\\n\" +\n \" - Rate limiting or quota exceeded\",\n );\n this.url = url;\n this.cause = opts.cause;\n }\n}\n\nexport class SourceProcessingError extends SourceError {\n readonly sourceId: string;\n readonly status: number;\n\n constructor(sourceId: string, status = 3, message?: string) {\n super(message ?? `Source ${sourceId} failed to process`);\n this.sourceId = sourceId;\n this.status = status;\n }\n}\n\nexport class SourceTimeoutError extends SourceError {\n readonly sourceId: string;\n readonly timeout: number;\n readonly lastStatus?: number;\n\n constructor(sourceId: string, timeout: number, lastStatus?: number) {\n const statusInfo = lastStatus != null ? ` (last status: ${lastStatus})` : \"\";\n super(`Source ${sourceId} not ready after ${timeout.toFixed(1)}s${statusInfo}`);\n this.sourceId = sourceId;\n this.timeout = timeout;\n this.lastStatus = lastStatus;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Domain: Artifacts\n// ---------------------------------------------------------------------------\n\nexport class ArtifactError extends NotebookLMError {}\n\nexport class ArtifactNotFoundError extends ArtifactError {\n readonly artifactId: string;\n readonly artifactType?: string;\n\n constructor(artifactId: string, artifactType?: string) {\n const typeInfo = artifactType ? ` ${artifactType}` : \"\";\n super(`${typeInfo.trim() || \"Artifact\"} ${artifactId} not found`);\n this.artifactId = artifactId;\n this.artifactType = artifactType;\n }\n}\n\nexport class ArtifactNotReadyError extends ArtifactError {\n readonly artifactType: string;\n readonly artifactId?: string;\n readonly status?: string;\n\n constructor(artifactType: string, opts: { artifactId?: string; status?: string } = {}) {\n const base = opts.artifactId\n ? `${artifactType} artifact ${opts.artifactId} is not ready`\n : `No completed ${artifactType} found`;\n const statusInfo = opts.status ? ` (status: ${opts.status})` : \"\";\n super(`${base}${statusInfo}`);\n this.artifactType = artifactType;\n this.artifactId = opts.artifactId;\n this.status = opts.status;\n }\n}\n\nexport class ArtifactParseError extends ArtifactError {\n readonly artifactType: string;\n readonly artifactId?: string;\n readonly details?: string;\n readonly cause?: Error;\n\n constructor(\n artifactType: string,\n opts: { details?: string; artifactId?: string; cause?: Error } = {},\n ) {\n let msg = `Failed to parse ${artifactType} artifact`;\n if (opts.artifactId) msg += ` ${opts.artifactId}`;\n if (opts.details) msg += `: ${opts.details}`;\n super(msg);\n this.artifactType = artifactType;\n this.artifactId = opts.artifactId;\n this.details = opts.details;\n this.cause = opts.cause;\n }\n}\n\nexport class ArtifactDownloadError extends ArtifactError {\n readonly artifactType: string;\n readonly artifactId?: string;\n readonly details?: string;\n readonly cause?: Error;\n\n constructor(\n artifactType: string,\n opts: { details?: string; artifactId?: string; cause?: Error } = {},\n ) {\n let msg = `Failed to download ${artifactType} artifact`;\n if (opts.artifactId) msg += ` ${opts.artifactId}`;\n if (opts.details) msg += `: ${opts.details}`;\n super(msg);\n this.artifactType = artifactType;\n this.artifactId = opts.artifactId;\n this.details = opts.details;\n this.cause = opts.cause;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Domain: Chat\n// ---------------------------------------------------------------------------\n\nexport class ChatError extends NotebookLMError {}\n","import { readFileSync } from \"node:fs\";\nimport { AuthError } from \"./types/errors.js\";\n\nexport interface CookieMap {\n [key: string]: string;\n}\n\nexport interface AuthTokens {\n cookies: CookieMap;\n csrfToken: string;\n sessionId: string;\n cookieHeader: string;\n /** Cookie header containing only .google.com domain cookies — for media downloads */\n googleCookieHeader: string;\n}\n\n// ---------------------------------------------------------------------------\n// Cookie loading\n// ---------------------------------------------------------------------------\n\n/** Load cookies from a Playwright storage_state.json file. */\nexport function loadCookiesFromFile(filePath: string): CookieMap {\n let raw: string;\n try {\n raw = readFileSync(filePath, \"utf-8\");\n } catch {\n throw new AuthError(\n `Cookie file not found: ${filePath}\\n` + \"Provide valid Playwright storage state JSON.\",\n );\n }\n return extractCookiesFromStorageState(JSON.parse(raw));\n}\n\n/** Load cookies from a raw Playwright storage state object. */\nexport function loadCookiesFromObject(storageState: {\n cookies?: Array<{ name: string; value: string; domain: string }>;\n}): CookieMap {\n return extractCookiesFromStorageState(storageState);\n}\n\n/** Build a cookie header containing only .google.com domain cookies (for media downloads). */\nexport function buildGoogleCookieHeader(storageState: {\n cookies?: Array<{ name: string; value: string; domain: string }>;\n}): string {\n const map: CookieMap = {};\n for (const c of storageState.cookies ?? []) {\n if (c.domain === \".google.com\" && c.name && c.value) {\n map[c.name] = map[c.name] ?? c.value;\n }\n }\n return buildCookieHeader(map);\n}\n\n/** Load cookies from a flat cookie map (already parsed). */\nexport function loadCookiesFromMap(map: CookieMap): CookieMap {\n return { ...map };\n}\n\n/** Load cookies from a \"; \"-separated cookie string (e.g. process.env.NOTEBOOKLM_COOKIES). */\nexport function loadCookiesFromString(cookieStr: string): CookieMap {\n const map: CookieMap = {};\n for (const part of cookieStr.split(/;\\s*/)) {\n const idx = part.indexOf(\"=\");\n if (idx > 0) {\n const name = part.slice(0, idx).trim();\n const value = part.slice(idx + 1).trim();\n if (name) map[name] = value;\n }\n }\n return map;\n}\n\nfunction extractCookiesFromStorageState(storageState: {\n cookies?: Array<{ name: string; value: string; domain: string }>;\n}): CookieMap {\n const cookies: CookieMap = {};\n const domainTrack: Record<string, string> = {};\n\n for (const cookie of storageState.cookies ?? []) {\n const { domain, name, value } = cookie;\n if (!isAllowedDomain(domain) || !name) continue;\n\n const isBase = domain === \".google.com\";\n if (!(name in cookies) || isBase) {\n cookies[name] = value;\n domainTrack[name] = domain;\n }\n }\n\n if (!cookies[\"SID\"]) {\n throw new AuthError(\n \"Missing required cookie: SID. \" +\n \"Provide valid Playwright storage state with Google cookies.\",\n );\n }\n return cookies;\n}\n\nfunction isAllowedDomain(domain: string): boolean {\n if (\n domain === \".google.com\" ||\n domain === \"notebooklm.google.com\" ||\n domain === \".googleusercontent.com\"\n ) {\n return true;\n }\n if (domain.startsWith(\".google.\")) {\n return true; // Allow all regional Google domains\n }\n return false;\n}\n\nexport function buildCookieHeader(cookies: CookieMap): string {\n return Object.entries(cookies)\n .map(([k, v]) => `${k}=${v}`)\n .join(\"; \");\n}\n\n// ---------------------------------------------------------------------------\n// Token fetching\n// ---------------------------------------------------------------------------\n\nconst NOTEBOOKLM_URL = \"https://notebooklm.google.com/\";\n\nexport async function fetchTokens(\n cookies: CookieMap,\n): Promise<{ csrfToken: string; sessionId: string }> {\n const cookieHeader = buildCookieHeader(cookies);\n\n const response = await fetch(NOTEBOOKLM_URL, {\n headers: { Cookie: cookieHeader },\n redirect: \"follow\",\n });\n\n if (!response.ok) {\n throw new AuthError(`Failed to fetch NotebookLM page: HTTP ${response.status}`);\n }\n\n const finalUrl = response.url;\n if (isGoogleAuthRedirect(finalUrl)) {\n throw new AuthError(`Redirected to login page: ${finalUrl}. Cookies may be expired.`);\n }\n\n const html = await response.text();\n const csrfToken = extractCsrfToken(html, finalUrl);\n const sessionId = extractSessionId(html, finalUrl);\n\n return { csrfToken, sessionId };\n}\n\nfunction extractCsrfToken(html: string, finalUrl: string): string {\n const match = /\"SNlM0e\"\\s*:\\s*\"([^\"]+)\"/.exec(html);\n if (!match?.[1]) {\n if (isGoogleAuthRedirect(finalUrl) || html.includes(\"accounts.google.com\")) {\n throw new AuthError(\"Authentication expired or invalid. Cookies may need to be refreshed.\");\n }\n throw new AuthError(\"CSRF token (SNlM0e) not found in NotebookLM page HTML.\");\n }\n return match[1];\n}\n\nfunction extractSessionId(html: string, finalUrl: string): string {\n const match = /\"FdrFJe\"\\s*:\\s*\"([^\"]+)\"/.exec(html);\n if (!match?.[1]) {\n if (isGoogleAuthRedirect(finalUrl) || html.includes(\"accounts.google.com\")) {\n throw new AuthError(\"Authentication expired or invalid. Cookies may need to be refreshed.\");\n }\n throw new AuthError(\"Session ID (FdrFJe) not found in NotebookLM page HTML.\");\n }\n return match[1];\n}\n\nfunction isGoogleAuthRedirect(url: string): boolean {\n return url.includes(\"accounts.google.com\") || url.includes(\"signin\");\n}\n\n// ---------------------------------------------------------------------------\n// Connect options\n// ---------------------------------------------------------------------------\n\nexport interface ConnectOptions {\n /** \"; \"-separated cookie string (e.g. \"SID=abc; HSID=xyz\") */\n cookies?: string;\n /** Path to Playwright storage_state.json */\n cookiesFile?: string;\n /** Pre-parsed cookie map */\n cookiesObject?: CookieMap | { cookies?: Array<{ name: string; value: string; domain: string }> };\n}\n\nexport async function connect(opts: ConnectOptions): Promise<AuthTokens> {\n let cookieMap: CookieMap;\n let googleCookieHeader: string | null = null;\n\n if (opts.cookies) {\n cookieMap = loadCookiesFromString(opts.cookies);\n } else if (opts.cookiesFile) {\n cookieMap = loadCookiesFromFile(opts.cookiesFile);\n } else if (opts.cookiesObject) {\n if (\"cookies\" in opts.cookiesObject && Array.isArray(opts.cookiesObject.cookies)) {\n const storageState = opts.cookiesObject as {\n cookies: Array<{ name: string; value: string; domain: string }>;\n };\n cookieMap = loadCookiesFromObject(storageState);\n googleCookieHeader = buildGoogleCookieHeader(storageState);\n } else {\n cookieMap = loadCookiesFromMap(opts.cookiesObject as CookieMap);\n }\n } else {\n // Fallback: check environment variable\n const envCookies = process.env[\"NOTEBOOKLM_COOKIES\"];\n if (envCookies) {\n cookieMap = loadCookiesFromString(envCookies);\n } else {\n throw new AuthError(\n \"No cookies provided. Pass cookies, cookiesFile, or cookiesObject to connect().\",\n );\n }\n }\n\n const { csrfToken, sessionId } = await fetchTokens(cookieMap);\n const cookieHeader = buildCookieHeader(cookieMap);\n\n return {\n cookies: cookieMap,\n csrfToken,\n sessionId,\n cookieHeader,\n googleCookieHeader: googleCookieHeader ?? cookieHeader,\n };\n}\n","import { type BrowserContext, chromium, type Page } from \"playwright\";\nimport { type CookieMap, loadCookiesFromObject } from \"../auth.js\";\n\nexport interface LoginOptions {\n /**\n * Path to a directory for a persistent browser profile.\n * If provided, session remains logged in for future calls.\n */\n persistFolder?: string;\n /**\n * Browser type to use. Default is \"chromium\".\n */\n browserType?: \"chromium\" | \"msedge\";\n /**\n * Whether to run the browser in headless mode. Default is false.\n * Headless login is usually blocked by Google, so this is mostly for testing or refreshes.\n */\n headless?: boolean;\n}\n\nconst NOTEBOOKLM_URL = \"https://notebooklm.google.com/\";\nconst GOOGLE_ACCOUNTS_URL = \"https://accounts.google.com/\";\n\n/**\n * Log in to NotebookLM via a headful browser window.\n *\n * Flow:\n * 1. Opens browser to NotebookLM.\n * 2. If already logged in (via persistFolder), it proceeds.\n * 3. If not, it waits for the user to reach the home page.\n * 4. Captures cookies and returns them.\n */\nexport async function login(opts: LoginOptions = {}): Promise<{\n cookies: CookieMap;\n storageState: any;\n cookieHeader: string;\n}> {\n const { persistFolder, headless = false, browserType = \"chromium\" } = opts;\n\n let context: BrowserContext;\n\n const launchOptions = {\n headless,\n args: [\"--disable-blink-features=AutomationControlled\"],\n };\n\n if (persistFolder) {\n context = await chromium.launchPersistentContext(persistFolder, {\n ...launchOptions,\n channel: browserType === \"msedge\" ? \"msedge\" : undefined,\n });\n } else {\n const browser = await chromium.launch({\n ...launchOptions,\n channel: browserType === \"msedge\" ? \"msedge\" : undefined,\n });\n context = await browser.newContext();\n }\n\n const page = context.pages()[0] || (await context.newPage());\n await page.goto(NOTEBOOKLM_URL);\n\n // Check if we are on the login page\n if (page.url().includes(\"accounts.google.com\")) {\n console.log(\"Please log in to Google in the browser window...\");\n\n // Wait for navigation back to NotebookLM or successful login indicator\n // We poll until the URL includes notebooklm.google.com and it's not a generic landing page\n await page.waitForURL(\n (url) => {\n return url.hostname === \"notebooklm.google.com\" && !url.pathname.includes(\"/login\");\n },\n { timeout: 0 },\n ); // No timeout, wait for user\n }\n\n // Ensure we are fully loaded on the accounts domain too to capture those cookies\n await page.goto(GOOGLE_ACCOUNTS_URL, { waitUntil: \"load\" });\n await page.goto(NOTEBOOKLM_URL, { waitUntil: \"load\" });\n\n const storageState = await context.storageState();\n const cookies = loadCookiesFromObject(storageState as any);\n\n await context.close();\n\n return {\n cookies,\n storageState,\n cookieHeader: Object.entries(cookies)\n .map(([k, v]) => `${k}=${v}`)\n .join(\"; \"),\n };\n}\n","#!/usr/bin/env node\nimport { writeFileSync } from \"node:fs\";\nimport { resolve } from \"node:path\";\nimport { login } from \"./auth/browser.js\";\n\nasync function run() {\n const args = process.argv.slice(2);\n const command = args[0];\n\n if (command === \"login\") {\n console.log(\"🚀 Starting browser login flow...\");\n console.log(\"A browser window will open. Please log in to your Google account.\");\n\n try {\n const authResult = await login({\n persistFolder: \"./.auth_profile\",\n headless: false,\n });\n\n const outPath = resolve(process.cwd(), \"storage_state.json\");\n writeFileSync(outPath, JSON.stringify(authResult.storageState, null, 2));\n\n console.log(\"\\n✅ Login successful!\");\n console.log(`Saved session to: ${outPath}`);\n console.log(\"\\nNext steps:\");\n console.log(\"In your code, connect using:\");\n console.log(\"const client = await NotebookLMClient.connect({ cookiesFile: 'storage_state.json' });\");\n } catch (error) {\n console.error(\"❌ Login failed:\", error);\n process.exit(1);\n }\n } else {\n console.log(\"NotebookLM SDK CLI\");\n console.log(\"\\nUsage:\");\n console.log(\" npx notebooklm-sdk login Start browser login flow\");\n process.exit(command ? 1 : 0);\n }\n}\n\nrun();\n"]}
|
package/dist/bin.d.cts
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
package/dist/bin.d.ts
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
package/dist/bin.js
ADDED
|
@@ -0,0 +1,137 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { writeFileSync } from 'fs';
|
|
3
|
+
import { resolve } from 'path';
|
|
4
|
+
import { chromium } from 'playwright';
|
|
5
|
+
|
|
6
|
+
// src/types/errors.ts
|
|
7
|
+
var NotebookLMError = class extends Error {
|
|
8
|
+
constructor(message) {
|
|
9
|
+
super(message);
|
|
10
|
+
this.name = this.constructor.name;
|
|
11
|
+
}
|
|
12
|
+
};
|
|
13
|
+
var RPCError = class extends NotebookLMError {
|
|
14
|
+
methodId;
|
|
15
|
+
rawResponse;
|
|
16
|
+
rpcCode;
|
|
17
|
+
foundIds;
|
|
18
|
+
constructor(message, opts = {}) {
|
|
19
|
+
super(message);
|
|
20
|
+
this.methodId = opts.methodId;
|
|
21
|
+
this.rawResponse = opts.rawResponse ? opts.rawResponse.slice(0, 500) : void 0;
|
|
22
|
+
this.rpcCode = opts.rpcCode;
|
|
23
|
+
this.foundIds = opts.foundIds ?? [];
|
|
24
|
+
}
|
|
25
|
+
};
|
|
26
|
+
var AuthError = class extends RPCError {
|
|
27
|
+
};
|
|
28
|
+
|
|
29
|
+
// src/auth.ts
|
|
30
|
+
function loadCookiesFromObject(storageState) {
|
|
31
|
+
return extractCookiesFromStorageState(storageState);
|
|
32
|
+
}
|
|
33
|
+
function extractCookiesFromStorageState(storageState) {
|
|
34
|
+
const cookies = {};
|
|
35
|
+
for (const cookie of storageState.cookies ?? []) {
|
|
36
|
+
const { domain, name, value } = cookie;
|
|
37
|
+
if (!isAllowedDomain(domain) || !name) continue;
|
|
38
|
+
const isBase = domain === ".google.com";
|
|
39
|
+
if (!(name in cookies) || isBase) {
|
|
40
|
+
cookies[name] = value;
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
if (!cookies["SID"]) {
|
|
44
|
+
throw new AuthError(
|
|
45
|
+
"Missing required cookie: SID. Provide valid Playwright storage state with Google cookies."
|
|
46
|
+
);
|
|
47
|
+
}
|
|
48
|
+
return cookies;
|
|
49
|
+
}
|
|
50
|
+
function isAllowedDomain(domain) {
|
|
51
|
+
if (domain === ".google.com" || domain === "notebooklm.google.com" || domain === ".googleusercontent.com") {
|
|
52
|
+
return true;
|
|
53
|
+
}
|
|
54
|
+
if (domain.startsWith(".google.")) {
|
|
55
|
+
return true;
|
|
56
|
+
}
|
|
57
|
+
return false;
|
|
58
|
+
}
|
|
59
|
+
|
|
60
|
+
// src/auth/browser.ts
|
|
61
|
+
var NOTEBOOKLM_URL = "https://notebooklm.google.com/";
|
|
62
|
+
var GOOGLE_ACCOUNTS_URL = "https://accounts.google.com/";
|
|
63
|
+
async function login(opts = {}) {
|
|
64
|
+
const { persistFolder, headless = false, browserType = "chromium" } = opts;
|
|
65
|
+
let context;
|
|
66
|
+
const launchOptions = {
|
|
67
|
+
headless,
|
|
68
|
+
args: ["--disable-blink-features=AutomationControlled"]
|
|
69
|
+
};
|
|
70
|
+
if (persistFolder) {
|
|
71
|
+
context = await chromium.launchPersistentContext(persistFolder, {
|
|
72
|
+
...launchOptions,
|
|
73
|
+
channel: browserType === "msedge" ? "msedge" : void 0
|
|
74
|
+
});
|
|
75
|
+
} else {
|
|
76
|
+
const browser = await chromium.launch({
|
|
77
|
+
...launchOptions,
|
|
78
|
+
channel: browserType === "msedge" ? "msedge" : void 0
|
|
79
|
+
});
|
|
80
|
+
context = await browser.newContext();
|
|
81
|
+
}
|
|
82
|
+
const page = context.pages()[0] || await context.newPage();
|
|
83
|
+
await page.goto(NOTEBOOKLM_URL);
|
|
84
|
+
if (page.url().includes("accounts.google.com")) {
|
|
85
|
+
console.log("Please log in to Google in the browser window...");
|
|
86
|
+
await page.waitForURL(
|
|
87
|
+
(url) => {
|
|
88
|
+
return url.hostname === "notebooklm.google.com" && !url.pathname.includes("/login");
|
|
89
|
+
},
|
|
90
|
+
{ timeout: 0 }
|
|
91
|
+
);
|
|
92
|
+
}
|
|
93
|
+
await page.goto(GOOGLE_ACCOUNTS_URL, { waitUntil: "load" });
|
|
94
|
+
await page.goto(NOTEBOOKLM_URL, { waitUntil: "load" });
|
|
95
|
+
const storageState = await context.storageState();
|
|
96
|
+
const cookies = loadCookiesFromObject(storageState);
|
|
97
|
+
await context.close();
|
|
98
|
+
return {
|
|
99
|
+
cookies,
|
|
100
|
+
storageState,
|
|
101
|
+
cookieHeader: Object.entries(cookies).map(([k, v]) => `${k}=${v}`).join("; ")
|
|
102
|
+
};
|
|
103
|
+
}
|
|
104
|
+
|
|
105
|
+
// src/bin.ts
|
|
106
|
+
async function run() {
|
|
107
|
+
const args = process.argv.slice(2);
|
|
108
|
+
const command = args[0];
|
|
109
|
+
if (command === "login") {
|
|
110
|
+
console.log("\u{1F680} Starting browser login flow...");
|
|
111
|
+
console.log("A browser window will open. Please log in to your Google account.");
|
|
112
|
+
try {
|
|
113
|
+
const authResult = await login({
|
|
114
|
+
persistFolder: "./.auth_profile",
|
|
115
|
+
headless: false
|
|
116
|
+
});
|
|
117
|
+
const outPath = resolve(process.cwd(), "storage_state.json");
|
|
118
|
+
writeFileSync(outPath, JSON.stringify(authResult.storageState, null, 2));
|
|
119
|
+
console.log("\n\u2705 Login successful!");
|
|
120
|
+
console.log(`Saved session to: ${outPath}`);
|
|
121
|
+
console.log("\nNext steps:");
|
|
122
|
+
console.log("In your code, connect using:");
|
|
123
|
+
console.log("const client = await NotebookLMClient.connect({ cookiesFile: 'storage_state.json' });");
|
|
124
|
+
} catch (error) {
|
|
125
|
+
console.error("\u274C Login failed:", error);
|
|
126
|
+
process.exit(1);
|
|
127
|
+
}
|
|
128
|
+
} else {
|
|
129
|
+
console.log("NotebookLM SDK CLI");
|
|
130
|
+
console.log("\nUsage:");
|
|
131
|
+
console.log(" npx notebooklm-sdk login Start browser login flow");
|
|
132
|
+
process.exit(command ? 1 : 0);
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
run();
|
|
136
|
+
//# sourceMappingURL=bin.js.map
|
|
137
|
+
//# sourceMappingURL=bin.js.map
|
package/dist/bin.js.map
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/types/errors.ts","../src/auth.ts","../src/auth/browser.ts","../src/bin.ts"],"names":[],"mappings":";;;;;;AAOO,IAAM,eAAA,GAAN,cAA8B,KAAA,CAAM;AAAA,EACzC,YAAY,OAAA,EAAiB;AAC3B,IAAA,KAAA,CAAM,OAAO,CAAA;AACb,IAAA,IAAA,CAAK,IAAA,GAAO,KAAK,WAAA,CAAY,IAAA;AAAA,EAC/B;AACF,CAAA;AAuBO,IAAM,QAAA,GAAN,cAAuB,eAAA,CAAgB;AAAA,EACnC,QAAA;AAAA,EACA,WAAA;AAAA,EACA,OAAA;AAAA,EACA,QAAA;AAAA,EAET,WAAA,CACE,OAAA,EACA,IAAA,GAKI,EAAC,EACL;AACA,IAAA,KAAA,CAAM,OAAO,CAAA;AACb,IAAA,IAAA,CAAK,WAAW,IAAA,CAAK,QAAA;AACrB,IAAA,IAAA,CAAK,WAAA,GAAc,KAAK,WAAA,GAAc,IAAA,CAAK,YAAY,KAAA,CAAM,CAAA,EAAG,GAAG,CAAA,GAAI,MAAA;AACvE,IAAA,IAAA,CAAK,UAAU,IAAA,CAAK,OAAA;AACpB,IAAA,IAAA,CAAK,QAAA,GAAW,IAAA,CAAK,QAAA,IAAY,EAAC;AAAA,EACpC;AACF,CAAA;AAEO,IAAM,SAAA,GAAN,cAAwB,QAAA,CAAS;AAAC,CAAA;;;ACxBlC,SAAS,sBAAsB,YAAA,EAExB;AACZ,EAAA,OAAO,+BAA+B,YAAY,CAAA;AACpD;AAkCA,SAAS,+BAA+B,YAAA,EAE1B;AACZ,EAAA,MAAM,UAAqB,EAAC;AAG5B,EAAA,KAAA,MAAW,MAAA,IAAU,YAAA,CAAa,OAAA,IAAW,EAAC,EAAG;AAC/C,IAAA,MAAM,EAAE,MAAA,EAAQ,IAAA,EAAM,KAAA,EAAM,GAAI,MAAA;AAChC,IAAA,IAAI,CAAC,eAAA,CAAgB,MAAM,CAAA,IAAK,CAAC,IAAA,EAAM;AAEvC,IAAA,MAAM,SAAS,MAAA,KAAW,aAAA;AAC1B,IAAA,IAAI,EAAE,IAAA,IAAQ,OAAA,CAAA,IAAY,MAAA,EAAQ;AAChC,MAAA,OAAA,CAAQ,IAAI,CAAA,GAAI,KAAA;AACI,IACtB;AAAA,EACF;AAEA,EAAA,IAAI,CAAC,OAAA,CAAQ,KAAK,CAAA,EAAG;AACnB,IAAA,MAAM,IAAI,SAAA;AAAA,MACR;AAAA,KAEF;AAAA,EACF;AACA,EAAA,OAAO,OAAA;AACT;AAEA,SAAS,gBAAgB,MAAA,EAAyB;AAChD,EAAA,IACE,MAAA,KAAW,aAAA,IACX,MAAA,KAAW,uBAAA,IACX,WAAW,wBAAA,EACX;AACA,IAAA,OAAO,IAAA;AAAA,EACT;AACA,EAAA,IAAI,MAAA,CAAO,UAAA,CAAW,UAAU,CAAA,EAAG;AACjC,IAAA,OAAO,IAAA;AAAA,EACT;AACA,EAAA,OAAO,KAAA;AACT;;;AC1FA,IAAM,cAAA,GAAiB,gCAAA;AACvB,IAAM,mBAAA,GAAsB,8BAAA;AAW5B,eAAsB,KAAA,CAAM,IAAA,GAAqB,EAAC,EAI/C;AACD,EAAA,MAAM,EAAE,aAAA,EAAe,QAAA,GAAW,KAAA,EAAO,WAAA,GAAc,YAAW,GAAI,IAAA;AAEtE,EAAA,IAAI,OAAA;AAEJ,EAAA,MAAM,aAAA,GAAgB;AAAA,IACpB,QAAA;AAAA,IACA,IAAA,EAAM,CAAC,+CAA+C;AAAA,GACxD;AAEA,EAAA,IAAI,aAAA,EAAe;AACjB,IAAA,OAAA,GAAU,MAAM,QAAA,CAAS,uBAAA,CAAwB,aAAA,EAAe;AAAA,MAC9D,GAAG,aAAA;AAAA,MACH,OAAA,EAAS,WAAA,KAAgB,QAAA,GAAW,QAAA,GAAW;AAAA,KAChD,CAAA;AAAA,EACH,CAAA,MAAO;AACL,IAAA,MAAM,OAAA,GAAU,MAAM,QAAA,CAAS,MAAA,CAAO;AAAA,MACpC,GAAG,aAAA;AAAA,MACH,OAAA,EAAS,WAAA,KAAgB,QAAA,GAAW,QAAA,GAAW;AAAA,KAChD,CAAA;AACD,IAAA,OAAA,GAAU,MAAM,QAAQ,UAAA,EAAW;AAAA,EACrC;AAEA,EAAA,MAAM,IAAA,GAAO,QAAQ,KAAA,EAAM,CAAE,CAAC,CAAA,IAAM,MAAM,QAAQ,OAAA,EAAQ;AAC1D,EAAA,MAAM,IAAA,CAAK,KAAK,cAAc,CAAA;AAG9B,EAAA,IAAI,IAAA,CAAK,GAAA,EAAI,CAAE,QAAA,CAAS,qBAAqB,CAAA,EAAG;AAC9C,IAAA,OAAA,CAAQ,IAAI,kDAAkD,CAAA;AAI9D,IAAA,MAAM,IAAA,CAAK,UAAA;AAAA,MACT,CAAC,GAAA,KAAQ;AACP,QAAA,OAAO,IAAI,QAAA,KAAa,uBAAA,IAA2B,CAAC,GAAA,CAAI,QAAA,CAAS,SAAS,QAAQ,CAAA;AAAA,MACpF,CAAA;AAAA,MACA,EAAE,SAAS,CAAA;AAAE,KACf;AAAA,EACF;AAGA,EAAA,MAAM,KAAK,IAAA,CAAK,mBAAA,EAAqB,EAAE,SAAA,EAAW,QAAQ,CAAA;AAC1D,EAAA,MAAM,KAAK,IAAA,CAAK,cAAA,EAAgB,EAAE,SAAA,EAAW,QAAQ,CAAA;AAErD,EAAA,MAAM,YAAA,GAAe,MAAM,OAAA,CAAQ,YAAA,EAAa;AAChD,EAAA,MAAM,OAAA,GAAU,sBAAsB,YAAmB,CAAA;AAEzD,EAAA,MAAM,QAAQ,KAAA,EAAM;AAEpB,EAAA,OAAO;AAAA,IACL,OAAA;AAAA,IACA,YAAA;AAAA,IACA,cAAc,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,CACjC,GAAA,CAAI,CAAC,CAAC,CAAA,EAAG,CAAC,CAAA,KAAM,GAAG,CAAC,CAAA,CAAA,EAAI,CAAC,CAAA,CAAE,CAAA,CAC3B,KAAK,IAAI;AAAA,GACd;AACF;;;ACvFA,eAAe,GAAA,GAAM;AACnB,EAAA,MAAM,IAAA,GAAO,OAAA,CAAQ,IAAA,CAAK,KAAA,CAAM,CAAC,CAAA;AACjC,EAAA,MAAM,OAAA,GAAU,KAAK,CAAC,CAAA;AAEtB,EAAA,IAAI,YAAY,OAAA,EAAS;AACvB,IAAA,OAAA,CAAQ,IAAI,0CAAmC,CAAA;AAC/C,IAAA,OAAA,CAAQ,IAAI,mEAAmE,CAAA;AAE/E,IAAA,IAAI;AACF,MAAA,MAAM,UAAA,GAAa,MAAM,KAAA,CAAM;AAAA,QAC7B,aAAA,EAAe,iBAAA;AAAA,QACf,QAAA,EAAU;AAAA,OACX,CAAA;AAED,MAAA,MAAM,OAAA,GAAU,OAAA,CAAQ,OAAA,CAAQ,GAAA,IAAO,oBAAoB,CAAA;AAC3D,MAAA,aAAA,CAAc,SAAS,IAAA,CAAK,SAAA,CAAU,WAAW,YAAA,EAAc,IAAA,EAAM,CAAC,CAAC,CAAA;AAEvE,MAAA,OAAA,CAAQ,IAAI,4BAAuB,CAAA;AACnC,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kBAAA,EAAqB,OAAO,CAAA,CAAE,CAAA;AAC1C,MAAA,OAAA,CAAQ,IAAI,eAAe,CAAA;AAC3B,MAAA,OAAA,CAAQ,IAAI,8BAA8B,CAAA;AAC1C,MAAA,OAAA,CAAQ,IAAI,uFAAuF,CAAA;AAAA,IACrG,SAAS,KAAA,EAAO;AACd,MAAA,OAAA,CAAQ,KAAA,CAAM,wBAAmB,KAAK,CAAA;AACtC,MAAA,OAAA,CAAQ,KAAK,CAAC,CAAA;AAAA,IAChB;AAAA,EACF,CAAA,MAAO;AACL,IAAA,OAAA,CAAQ,IAAI,oBAAoB,CAAA;AAChC,IAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AACtB,IAAA,OAAA,CAAQ,IAAI,wDAAwD,CAAA;AACpE,IAAA,OAAA,CAAQ,IAAA,CAAK,OAAA,GAAU,CAAA,GAAI,CAAC,CAAA;AAAA,EAC9B;AACF;AAEA,GAAA,EAAI","file":"bin.js","sourcesContent":["/**\n * All exceptions for notebooklm-sdk.\n *\n * All errors extend NotebookLMError so you can catch everything with:\n * try { ... } catch (e) { if (e instanceof NotebookLMError) ... }\n */\n\nexport class NotebookLMError extends Error {\n constructor(message: string) {\n super(message);\n this.name = this.constructor.name;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Network (transport-level, before RPC processing)\n// ---------------------------------------------------------------------------\n\nexport class NetworkError extends NotebookLMError {\n readonly methodId?: string;\n readonly originalError?: Error;\n\n constructor(message: string, opts: { methodId?: string; originalError?: Error } = {}) {\n super(message);\n this.methodId = opts.methodId;\n this.originalError = opts.originalError;\n }\n}\n\nexport class RPCTimeoutError extends NetworkError {}\n\n// ---------------------------------------------------------------------------\n// RPC Protocol (after connection established)\n// ---------------------------------------------------------------------------\n\nexport class RPCError extends NotebookLMError {\n readonly methodId?: string;\n readonly rawResponse?: string;\n readonly rpcCode?: string | number;\n readonly foundIds: string[];\n\n constructor(\n message: string,\n opts: {\n methodId?: string;\n rawResponse?: string;\n rpcCode?: string | number;\n foundIds?: string[];\n } = {},\n ) {\n super(message);\n this.methodId = opts.methodId;\n this.rawResponse = opts.rawResponse ? opts.rawResponse.slice(0, 500) : undefined;\n this.rpcCode = opts.rpcCode;\n this.foundIds = opts.foundIds ?? [];\n }\n}\n\nexport class AuthError extends RPCError {}\n\nexport class RateLimitError extends RPCError {\n readonly retryAfter?: number;\n\n constructor(\n message: string,\n opts: {\n retryAfter?: number;\n methodId?: string;\n rawResponse?: string;\n rpcCode?: string | number;\n foundIds?: string[];\n } = {},\n ) {\n super(message, opts);\n this.retryAfter = opts.retryAfter;\n }\n}\n\nexport class ServerError extends RPCError {\n readonly statusCode?: number;\n\n constructor(\n message: string,\n opts: {\n statusCode?: number;\n methodId?: string;\n rawResponse?: string;\n rpcCode?: string | number;\n } = {},\n ) {\n super(message, opts);\n this.statusCode = opts.statusCode;\n }\n}\n\nexport class ClientError extends RPCError {\n readonly statusCode?: number;\n\n constructor(\n message: string,\n opts: {\n statusCode?: number;\n methodId?: string;\n rawResponse?: string;\n rpcCode?: string | number;\n } = {},\n ) {\n super(message, opts);\n this.statusCode = opts.statusCode;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Domain: Notebooks\n// ---------------------------------------------------------------------------\n\nexport class NotebookError extends NotebookLMError {}\n\nexport class NotebookNotFoundError extends NotebookError {\n readonly notebookId: string;\n\n constructor(notebookId: string) {\n super(`Notebook not found: ${notebookId}`);\n this.notebookId = notebookId;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Domain: Sources\n// ---------------------------------------------------------------------------\n\nexport class SourceError extends NotebookLMError {}\n\nexport class SourceNotFoundError extends SourceError {\n readonly sourceId: string;\n\n constructor(sourceId: string) {\n super(`Source not found: ${sourceId}`);\n this.sourceId = sourceId;\n }\n}\n\nexport class SourceAddError extends SourceError {\n readonly url: string;\n readonly cause?: Error;\n\n constructor(url: string, opts: { cause?: Error; message?: string } = {}) {\n super(\n opts.message ??\n `Failed to add source: ${url}\\n` +\n \"Possible causes:\\n\" +\n \" - URL is invalid or inaccessible\\n\" +\n \" - Content is behind a paywall or requires authentication\\n\" +\n \" - Rate limiting or quota exceeded\",\n );\n this.url = url;\n this.cause = opts.cause;\n }\n}\n\nexport class SourceProcessingError extends SourceError {\n readonly sourceId: string;\n readonly status: number;\n\n constructor(sourceId: string, status = 3, message?: string) {\n super(message ?? `Source ${sourceId} failed to process`);\n this.sourceId = sourceId;\n this.status = status;\n }\n}\n\nexport class SourceTimeoutError extends SourceError {\n readonly sourceId: string;\n readonly timeout: number;\n readonly lastStatus?: number;\n\n constructor(sourceId: string, timeout: number, lastStatus?: number) {\n const statusInfo = lastStatus != null ? ` (last status: ${lastStatus})` : \"\";\n super(`Source ${sourceId} not ready after ${timeout.toFixed(1)}s${statusInfo}`);\n this.sourceId = sourceId;\n this.timeout = timeout;\n this.lastStatus = lastStatus;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Domain: Artifacts\n// ---------------------------------------------------------------------------\n\nexport class ArtifactError extends NotebookLMError {}\n\nexport class ArtifactNotFoundError extends ArtifactError {\n readonly artifactId: string;\n readonly artifactType?: string;\n\n constructor(artifactId: string, artifactType?: string) {\n const typeInfo = artifactType ? ` ${artifactType}` : \"\";\n super(`${typeInfo.trim() || \"Artifact\"} ${artifactId} not found`);\n this.artifactId = artifactId;\n this.artifactType = artifactType;\n }\n}\n\nexport class ArtifactNotReadyError extends ArtifactError {\n readonly artifactType: string;\n readonly artifactId?: string;\n readonly status?: string;\n\n constructor(artifactType: string, opts: { artifactId?: string; status?: string } = {}) {\n const base = opts.artifactId\n ? `${artifactType} artifact ${opts.artifactId} is not ready`\n : `No completed ${artifactType} found`;\n const statusInfo = opts.status ? ` (status: ${opts.status})` : \"\";\n super(`${base}${statusInfo}`);\n this.artifactType = artifactType;\n this.artifactId = opts.artifactId;\n this.status = opts.status;\n }\n}\n\nexport class ArtifactParseError extends ArtifactError {\n readonly artifactType: string;\n readonly artifactId?: string;\n readonly details?: string;\n readonly cause?: Error;\n\n constructor(\n artifactType: string,\n opts: { details?: string; artifactId?: string; cause?: Error } = {},\n ) {\n let msg = `Failed to parse ${artifactType} artifact`;\n if (opts.artifactId) msg += ` ${opts.artifactId}`;\n if (opts.details) msg += `: ${opts.details}`;\n super(msg);\n this.artifactType = artifactType;\n this.artifactId = opts.artifactId;\n this.details = opts.details;\n this.cause = opts.cause;\n }\n}\n\nexport class ArtifactDownloadError extends ArtifactError {\n readonly artifactType: string;\n readonly artifactId?: string;\n readonly details?: string;\n readonly cause?: Error;\n\n constructor(\n artifactType: string,\n opts: { details?: string; artifactId?: string; cause?: Error } = {},\n ) {\n let msg = `Failed to download ${artifactType} artifact`;\n if (opts.artifactId) msg += ` ${opts.artifactId}`;\n if (opts.details) msg += `: ${opts.details}`;\n super(msg);\n this.artifactType = artifactType;\n this.artifactId = opts.artifactId;\n this.details = opts.details;\n this.cause = opts.cause;\n }\n}\n\n// ---------------------------------------------------------------------------\n// Domain: Chat\n// ---------------------------------------------------------------------------\n\nexport class ChatError extends NotebookLMError {}\n","import { readFileSync } from \"node:fs\";\nimport { AuthError } from \"./types/errors.js\";\n\nexport interface CookieMap {\n [key: string]: string;\n}\n\nexport interface AuthTokens {\n cookies: CookieMap;\n csrfToken: string;\n sessionId: string;\n cookieHeader: string;\n /** Cookie header containing only .google.com domain cookies — for media downloads */\n googleCookieHeader: string;\n}\n\n// ---------------------------------------------------------------------------\n// Cookie loading\n// ---------------------------------------------------------------------------\n\n/** Load cookies from a Playwright storage_state.json file. */\nexport function loadCookiesFromFile(filePath: string): CookieMap {\n let raw: string;\n try {\n raw = readFileSync(filePath, \"utf-8\");\n } catch {\n throw new AuthError(\n `Cookie file not found: ${filePath}\\n` + \"Provide valid Playwright storage state JSON.\",\n );\n }\n return extractCookiesFromStorageState(JSON.parse(raw));\n}\n\n/** Load cookies from a raw Playwright storage state object. */\nexport function loadCookiesFromObject(storageState: {\n cookies?: Array<{ name: string; value: string; domain: string }>;\n}): CookieMap {\n return extractCookiesFromStorageState(storageState);\n}\n\n/** Build a cookie header containing only .google.com domain cookies (for media downloads). */\nexport function buildGoogleCookieHeader(storageState: {\n cookies?: Array<{ name: string; value: string; domain: string }>;\n}): string {\n const map: CookieMap = {};\n for (const c of storageState.cookies ?? []) {\n if (c.domain === \".google.com\" && c.name && c.value) {\n map[c.name] = map[c.name] ?? c.value;\n }\n }\n return buildCookieHeader(map);\n}\n\n/** Load cookies from a flat cookie map (already parsed). */\nexport function loadCookiesFromMap(map: CookieMap): CookieMap {\n return { ...map };\n}\n\n/** Load cookies from a \"; \"-separated cookie string (e.g. process.env.NOTEBOOKLM_COOKIES). */\nexport function loadCookiesFromString(cookieStr: string): CookieMap {\n const map: CookieMap = {};\n for (const part of cookieStr.split(/;\\s*/)) {\n const idx = part.indexOf(\"=\");\n if (idx > 0) {\n const name = part.slice(0, idx).trim();\n const value = part.slice(idx + 1).trim();\n if (name) map[name] = value;\n }\n }\n return map;\n}\n\nfunction extractCookiesFromStorageState(storageState: {\n cookies?: Array<{ name: string; value: string; domain: string }>;\n}): CookieMap {\n const cookies: CookieMap = {};\n const domainTrack: Record<string, string> = {};\n\n for (const cookie of storageState.cookies ?? []) {\n const { domain, name, value } = cookie;\n if (!isAllowedDomain(domain) || !name) continue;\n\n const isBase = domain === \".google.com\";\n if (!(name in cookies) || isBase) {\n cookies[name] = value;\n domainTrack[name] = domain;\n }\n }\n\n if (!cookies[\"SID\"]) {\n throw new AuthError(\n \"Missing required cookie: SID. \" +\n \"Provide valid Playwright storage state with Google cookies.\",\n );\n }\n return cookies;\n}\n\nfunction isAllowedDomain(domain: string): boolean {\n if (\n domain === \".google.com\" ||\n domain === \"notebooklm.google.com\" ||\n domain === \".googleusercontent.com\"\n ) {\n return true;\n }\n if (domain.startsWith(\".google.\")) {\n return true; // Allow all regional Google domains\n }\n return false;\n}\n\nexport function buildCookieHeader(cookies: CookieMap): string {\n return Object.entries(cookies)\n .map(([k, v]) => `${k}=${v}`)\n .join(\"; \");\n}\n\n// ---------------------------------------------------------------------------\n// Token fetching\n// ---------------------------------------------------------------------------\n\nconst NOTEBOOKLM_URL = \"https://notebooklm.google.com/\";\n\nexport async function fetchTokens(\n cookies: CookieMap,\n): Promise<{ csrfToken: string; sessionId: string }> {\n const cookieHeader = buildCookieHeader(cookies);\n\n const response = await fetch(NOTEBOOKLM_URL, {\n headers: { Cookie: cookieHeader },\n redirect: \"follow\",\n });\n\n if (!response.ok) {\n throw new AuthError(`Failed to fetch NotebookLM page: HTTP ${response.status}`);\n }\n\n const finalUrl = response.url;\n if (isGoogleAuthRedirect(finalUrl)) {\n throw new AuthError(`Redirected to login page: ${finalUrl}. Cookies may be expired.`);\n }\n\n const html = await response.text();\n const csrfToken = extractCsrfToken(html, finalUrl);\n const sessionId = extractSessionId(html, finalUrl);\n\n return { csrfToken, sessionId };\n}\n\nfunction extractCsrfToken(html: string, finalUrl: string): string {\n const match = /\"SNlM0e\"\\s*:\\s*\"([^\"]+)\"/.exec(html);\n if (!match?.[1]) {\n if (isGoogleAuthRedirect(finalUrl) || html.includes(\"accounts.google.com\")) {\n throw new AuthError(\"Authentication expired or invalid. Cookies may need to be refreshed.\");\n }\n throw new AuthError(\"CSRF token (SNlM0e) not found in NotebookLM page HTML.\");\n }\n return match[1];\n}\n\nfunction extractSessionId(html: string, finalUrl: string): string {\n const match = /\"FdrFJe\"\\s*:\\s*\"([^\"]+)\"/.exec(html);\n if (!match?.[1]) {\n if (isGoogleAuthRedirect(finalUrl) || html.includes(\"accounts.google.com\")) {\n throw new AuthError(\"Authentication expired or invalid. Cookies may need to be refreshed.\");\n }\n throw new AuthError(\"Session ID (FdrFJe) not found in NotebookLM page HTML.\");\n }\n return match[1];\n}\n\nfunction isGoogleAuthRedirect(url: string): boolean {\n return url.includes(\"accounts.google.com\") || url.includes(\"signin\");\n}\n\n// ---------------------------------------------------------------------------\n// Connect options\n// ---------------------------------------------------------------------------\n\nexport interface ConnectOptions {\n /** \"; \"-separated cookie string (e.g. \"SID=abc; HSID=xyz\") */\n cookies?: string;\n /** Path to Playwright storage_state.json */\n cookiesFile?: string;\n /** Pre-parsed cookie map */\n cookiesObject?: CookieMap | { cookies?: Array<{ name: string; value: string; domain: string }> };\n}\n\nexport async function connect(opts: ConnectOptions): Promise<AuthTokens> {\n let cookieMap: CookieMap;\n let googleCookieHeader: string | null = null;\n\n if (opts.cookies) {\n cookieMap = loadCookiesFromString(opts.cookies);\n } else if (opts.cookiesFile) {\n cookieMap = loadCookiesFromFile(opts.cookiesFile);\n } else if (opts.cookiesObject) {\n if (\"cookies\" in opts.cookiesObject && Array.isArray(opts.cookiesObject.cookies)) {\n const storageState = opts.cookiesObject as {\n cookies: Array<{ name: string; value: string; domain: string }>;\n };\n cookieMap = loadCookiesFromObject(storageState);\n googleCookieHeader = buildGoogleCookieHeader(storageState);\n } else {\n cookieMap = loadCookiesFromMap(opts.cookiesObject as CookieMap);\n }\n } else {\n // Fallback: check environment variable\n const envCookies = process.env[\"NOTEBOOKLM_COOKIES\"];\n if (envCookies) {\n cookieMap = loadCookiesFromString(envCookies);\n } else {\n throw new AuthError(\n \"No cookies provided. Pass cookies, cookiesFile, or cookiesObject to connect().\",\n );\n }\n }\n\n const { csrfToken, sessionId } = await fetchTokens(cookieMap);\n const cookieHeader = buildCookieHeader(cookieMap);\n\n return {\n cookies: cookieMap,\n csrfToken,\n sessionId,\n cookieHeader,\n googleCookieHeader: googleCookieHeader ?? cookieHeader,\n };\n}\n","import { type BrowserContext, chromium, type Page } from \"playwright\";\nimport { type CookieMap, loadCookiesFromObject } from \"../auth.js\";\n\nexport interface LoginOptions {\n /**\n * Path to a directory for a persistent browser profile.\n * If provided, session remains logged in for future calls.\n */\n persistFolder?: string;\n /**\n * Browser type to use. Default is \"chromium\".\n */\n browserType?: \"chromium\" | \"msedge\";\n /**\n * Whether to run the browser in headless mode. Default is false.\n * Headless login is usually blocked by Google, so this is mostly for testing or refreshes.\n */\n headless?: boolean;\n}\n\nconst NOTEBOOKLM_URL = \"https://notebooklm.google.com/\";\nconst GOOGLE_ACCOUNTS_URL = \"https://accounts.google.com/\";\n\n/**\n * Log in to NotebookLM via a headful browser window.\n *\n * Flow:\n * 1. Opens browser to NotebookLM.\n * 2. If already logged in (via persistFolder), it proceeds.\n * 3. If not, it waits for the user to reach the home page.\n * 4. Captures cookies and returns them.\n */\nexport async function login(opts: LoginOptions = {}): Promise<{\n cookies: CookieMap;\n storageState: any;\n cookieHeader: string;\n}> {\n const { persistFolder, headless = false, browserType = \"chromium\" } = opts;\n\n let context: BrowserContext;\n\n const launchOptions = {\n headless,\n args: [\"--disable-blink-features=AutomationControlled\"],\n };\n\n if (persistFolder) {\n context = await chromium.launchPersistentContext(persistFolder, {\n ...launchOptions,\n channel: browserType === \"msedge\" ? \"msedge\" : undefined,\n });\n } else {\n const browser = await chromium.launch({\n ...launchOptions,\n channel: browserType === \"msedge\" ? \"msedge\" : undefined,\n });\n context = await browser.newContext();\n }\n\n const page = context.pages()[0] || (await context.newPage());\n await page.goto(NOTEBOOKLM_URL);\n\n // Check if we are on the login page\n if (page.url().includes(\"accounts.google.com\")) {\n console.log(\"Please log in to Google in the browser window...\");\n\n // Wait for navigation back to NotebookLM or successful login indicator\n // We poll until the URL includes notebooklm.google.com and it's not a generic landing page\n await page.waitForURL(\n (url) => {\n return url.hostname === \"notebooklm.google.com\" && !url.pathname.includes(\"/login\");\n },\n { timeout: 0 },\n ); // No timeout, wait for user\n }\n\n // Ensure we are fully loaded on the accounts domain too to capture those cookies\n await page.goto(GOOGLE_ACCOUNTS_URL, { waitUntil: \"load\" });\n await page.goto(NOTEBOOKLM_URL, { waitUntil: \"load\" });\n\n const storageState = await context.storageState();\n const cookies = loadCookiesFromObject(storageState as any);\n\n await context.close();\n\n return {\n cookies,\n storageState,\n cookieHeader: Object.entries(cookies)\n .map(([k, v]) => `${k}=${v}`)\n .join(\"; \"),\n };\n}\n","#!/usr/bin/env node\nimport { writeFileSync } from \"node:fs\";\nimport { resolve } from \"node:path\";\nimport { login } from \"./auth/browser.js\";\n\nasync function run() {\n const args = process.argv.slice(2);\n const command = args[0];\n\n if (command === \"login\") {\n console.log(\"🚀 Starting browser login flow...\");\n console.log(\"A browser window will open. Please log in to your Google account.\");\n\n try {\n const authResult = await login({\n persistFolder: \"./.auth_profile\",\n headless: false,\n });\n\n const outPath = resolve(process.cwd(), \"storage_state.json\");\n writeFileSync(outPath, JSON.stringify(authResult.storageState, null, 2));\n\n console.log(\"\\n✅ Login successful!\");\n console.log(`Saved session to: ${outPath}`);\n console.log(\"\\nNext steps:\");\n console.log(\"In your code, connect using:\");\n console.log(\"const client = await NotebookLMClient.connect({ cookiesFile: 'storage_state.json' });\");\n } catch (error) {\n console.error(\"❌ Login failed:\", error);\n process.exit(1);\n }\n } else {\n console.log(\"NotebookLM SDK CLI\");\n console.log(\"\\nUsage:\");\n console.log(\" npx notebooklm-sdk login Start browser login flow\");\n process.exit(command ? 1 : 0);\n }\n}\n\nrun();\n"]}
|
package/dist/index.d.cts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { A as AuthTokens,
|
|
2
|
-
export {
|
|
1
|
+
import { A as AuthTokens, a as ConnectOptions } from './auth-Ba2hsZW_.cjs';
|
|
2
|
+
export { C as CookieMap, d as connect } from './auth-Ba2hsZW_.cjs';
|
|
3
3
|
export { ArtifactDownloadError, ArtifactError, ArtifactNotFoundError, ArtifactNotReadyError, ArtifactParseError, AuthError, ChatError, ClientError, NetworkError, NotebookError, NotebookLMError, NotebookNotFoundError, RPCError, RPCTimeoutError, RateLimitError, ServerError, SourceAddError, SourceError, SourceNotFoundError, SourceProcessingError, SourceTimeoutError } from './errors.cjs';
|
|
4
4
|
|
|
5
5
|
/** RPC method IDs for NotebookLM batchexecute API. */
|
package/dist/index.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { A as AuthTokens,
|
|
2
|
-
export {
|
|
1
|
+
import { A as AuthTokens, a as ConnectOptions } from './auth-Ba2hsZW_.js';
|
|
2
|
+
export { C as CookieMap, d as connect } from './auth-Ba2hsZW_.js';
|
|
3
3
|
export { ArtifactDownloadError, ArtifactError, ArtifactNotFoundError, ArtifactNotReadyError, ArtifactParseError, AuthError, ChatError, ClientError, NetworkError, NotebookError, NotebookLMError, NotebookNotFoundError, RPCError, RPCTimeoutError, RateLimitError, ServerError, SourceAddError, SourceError, SourceNotFoundError, SourceProcessingError, SourceTimeoutError } from './errors.js';
|
|
4
4
|
|
|
5
5
|
/** RPC method IDs for NotebookLM batchexecute API. */
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "notebooklm-sdk",
|
|
3
|
-
"version": "0.1.
|
|
3
|
+
"version": "0.1.6",
|
|
4
4
|
"description": "TypeScript SDK for NotebookLM API.",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "./dist/index.cjs",
|
|
@@ -41,6 +41,9 @@
|
|
|
41
41
|
"files": [
|
|
42
42
|
"dist"
|
|
43
43
|
],
|
|
44
|
+
"bin": {
|
|
45
|
+
"notebooklm-sdk": "./dist/bin.js"
|
|
46
|
+
},
|
|
44
47
|
"scripts": {
|
|
45
48
|
"build": "tsup",
|
|
46
49
|
"dev": "tsup --watch",
|
|
@@ -51,7 +54,7 @@
|
|
|
51
54
|
"format": "biome format --write .",
|
|
52
55
|
"format:check": "biome format .",
|
|
53
56
|
"typecheck": "tsc --noEmit",
|
|
54
|
-
"login": "tsx
|
|
57
|
+
"login": "tsx src/bin.ts login",
|
|
55
58
|
"prepublishOnly": "bun run test && bun run build"
|
|
56
59
|
},
|
|
57
60
|
"devDependencies": {
|