@asterxsk/kiln 0.1.0

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.
Files changed (263) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +170 -0
  3. package/agent/AGENTS.md +67 -0
  4. package/agent/README.md +5 -0
  5. package/agent/extensions/AGENTS.md +68 -0
  6. package/agent/extensions/ask-user/index.ts +418 -0
  7. package/agent/extensions/ask-user/install.ps1 +23 -0
  8. package/agent/extensions/ask-user/install.sh +21 -0
  9. package/agent/extensions/ask-user/package-lock.json +769 -0
  10. package/agent/extensions/ask-user/package.json +19 -0
  11. package/agent/extensions/ask-user/prompt.ts +45 -0
  12. package/agent/extensions/ask-user/tsconfig.json +7 -0
  13. package/agent/extensions/background-terminals/docs/implementation-guide.md +942 -0
  14. package/agent/extensions/background-terminals/index.ts +627 -0
  15. package/agent/extensions/background-terminals/install.ps1 +23 -0
  16. package/agent/extensions/background-terminals/install.sh +21 -0
  17. package/agent/extensions/background-terminals/manager.test.ts +735 -0
  18. package/agent/extensions/background-terminals/output.test.ts +109 -0
  19. package/agent/extensions/background-terminals/package-lock.json +769 -0
  20. package/agent/extensions/background-terminals/package.json +17 -0
  21. package/agent/extensions/background-terminals/prompt.test.ts +125 -0
  22. package/agent/extensions/background-terminals/ps.test.ts +82 -0
  23. package/agent/extensions/background-terminals/result-delivery.test.ts +44 -0
  24. package/agent/extensions/background-terminals/src/domain.ts +87 -0
  25. package/agent/extensions/background-terminals/src/manager.ts +907 -0
  26. package/agent/extensions/background-terminals/src/output.ts +84 -0
  27. package/agent/extensions/background-terminals/src/prompt.ts +142 -0
  28. package/agent/extensions/background-terminals/src/result-delivery.ts +27 -0
  29. package/agent/extensions/background-terminals/src/runtime.ts +36 -0
  30. package/agent/extensions/background-terminals/src/ui/output-view.ts +79 -0
  31. package/agent/extensions/background-terminals/src/ui/ps.ts +621 -0
  32. package/agent/extensions/background-terminals/tsconfig.json +7 -0
  33. package/agent/extensions/destructive/README.md +31 -0
  34. package/agent/extensions/destructive/index.ts +88 -0
  35. package/agent/extensions/destructive/install.ps1 +23 -0
  36. package/agent/extensions/destructive/install.sh +21 -0
  37. package/agent/extensions/file-search/index.spec.ts +443 -0
  38. package/agent/extensions/file-search/index.ts +459 -0
  39. package/agent/extensions/file-search/install.ps1 +23 -0
  40. package/agent/extensions/file-search/install.sh +21 -0
  41. package/agent/extensions/file-search/package-lock.json +2253 -0
  42. package/agent/extensions/file-search/package.json +23 -0
  43. package/agent/extensions/file-search/src/args.ts +122 -0
  44. package/agent/extensions/file-search/src/binaries.ts +422 -0
  45. package/agent/extensions/file-search/src/output.ts +126 -0
  46. package/agent/extensions/file-search/src/process.ts +146 -0
  47. package/agent/extensions/file-search/src/prompt.ts +52 -0
  48. package/agent/extensions/file-search/tsconfig.json +7 -0
  49. package/agent/extensions/goal/README.md +50 -0
  50. package/agent/extensions/goal/index.ts +155 -0
  51. package/agent/extensions/goal/install.ps1 +23 -0
  52. package/agent/extensions/goal/install.sh +21 -0
  53. package/agent/extensions/modelconf/PLAN.md +915 -0
  54. package/agent/extensions/modelconf/README.md +66 -0
  55. package/agent/extensions/modelconf/index.ts +296 -0
  56. package/agent/extensions/modelconf/install.ps1 +23 -0
  57. package/agent/extensions/modelconf/install.sh +21 -0
  58. package/agent/extensions/modelconf/package-lock.json +1809 -0
  59. package/agent/extensions/modelconf/package.json +13 -0
  60. package/agent/extensions/modelconf/src/fuzzy.ts +26 -0
  61. package/agent/extensions/modelconf/src/glob.ts +13 -0
  62. package/agent/extensions/modelconf/src/persistence.test.ts +46 -0
  63. package/agent/extensions/modelconf/src/persistence.ts +164 -0
  64. package/agent/extensions/modelconf/src/ui/ModelConfView.test.ts +75 -0
  65. package/agent/extensions/modelconf/src/ui/ModelConfView.ts +1101 -0
  66. package/agent/extensions/modelconf/tsconfig.json +15 -0
  67. package/agent/extensions/pi-web-access/CHANGELOG.md +690 -0
  68. package/agent/extensions/pi-web-access/LICENSE +21 -0
  69. package/agent/extensions/pi-web-access/README.md +470 -0
  70. package/agent/extensions/pi-web-access/SECURITY.md +5 -0
  71. package/agent/extensions/pi-web-access/activity.ts +101 -0
  72. package/agent/extensions/pi-web-access/auth-fetch.ts +148 -0
  73. package/agent/extensions/pi-web-access/banner.png +0 -0
  74. package/agent/extensions/pi-web-access/brightdata-unlocker.ts +272 -0
  75. package/agent/extensions/pi-web-access/chrome-cookies.ts +669 -0
  76. package/agent/extensions/pi-web-access/content-find.ts +139 -0
  77. package/agent/extensions/pi-web-access/credential-source.ts +191 -0
  78. package/agent/extensions/pi-web-access/data-uri-sanitize.ts +406 -0
  79. package/agent/extensions/pi-web-access/datalab-pdf-extract.ts +568 -0
  80. package/agent/extensions/pi-web-access/declared-web-links.ts +173 -0
  81. package/agent/extensions/pi-web-access/evidence/CONTRACT-EVIDENCE.md +496 -0
  82. package/agent/extensions/pi-web-access/evidence/contract-probe.mjs +140 -0
  83. package/agent/extensions/pi-web-access/exa.ts +526 -0
  84. package/agent/extensions/pi-web-access/extract.ts +1196 -0
  85. package/agent/extensions/pi-web-access/feature-config.ts +29 -0
  86. package/agent/extensions/pi-web-access/fetch-params.ts +111 -0
  87. package/agent/extensions/pi-web-access/gemini-adc.ts +298 -0
  88. package/agent/extensions/pi-web-access/gemini-api.ts +353 -0
  89. package/agent/extensions/pi-web-access/gemini-pdf-extract.ts +108 -0
  90. package/agent/extensions/pi-web-access/gemini-search.ts +21 -0
  91. package/agent/extensions/pi-web-access/gemini-url-context.ts +128 -0
  92. package/agent/extensions/pi-web-access/gemini-web-config.ts +101 -0
  93. package/agent/extensions/pi-web-access/gemini-web.ts +487 -0
  94. package/agent/extensions/pi-web-access/github-api.ts +197 -0
  95. package/agent/extensions/pi-web-access/github-extract.ts +746 -0
  96. package/agent/extensions/pi-web-access/github-issue-pr.ts +700 -0
  97. package/agent/extensions/pi-web-access/index.ts +1737 -0
  98. package/agent/extensions/pi-web-access/package-lock.json +5808 -0
  99. package/agent/extensions/pi-web-access/package.json +64 -0
  100. package/agent/extensions/pi-web-access/page-query.ts +96 -0
  101. package/agent/extensions/pi-web-access/pdf-extract.ts +409 -0
  102. package/agent/extensions/pi-web-access/pi-web-fetch-demo.mp4 +0 -0
  103. package/agent/extensions/pi-web-access/promise-try.d.ts +7 -0
  104. package/agent/extensions/pi-web-access/query-rewrite.ts +51 -0
  105. package/agent/extensions/pi-web-access/render-search-error.ts +170 -0
  106. package/agent/extensions/pi-web-access/rsc-extract.ts +338 -0
  107. package/agent/extensions/pi-web-access/search-types.ts +20 -0
  108. package/agent/extensions/pi-web-access/source-check.ts +282 -0
  109. package/agent/extensions/pi-web-access/ssrf-protection.ts +526 -0
  110. package/agent/extensions/pi-web-access/storage.ts +521 -0
  111. package/agent/extensions/pi-web-access/summary-model-scope.ts +125 -0
  112. package/agent/extensions/pi-web-access/test/auth-fetch.test.mjs +208 -0
  113. package/agent/extensions/pi-web-access/test/brightdata-unlocker.test.mjs +840 -0
  114. package/agent/extensions/pi-web-access/test/chrome-cookie-extraction.test.mjs +441 -0
  115. package/agent/extensions/pi-web-access/test/config-path.test.mjs +283 -0
  116. package/agent/extensions/pi-web-access/test/content-find.test.mjs +25 -0
  117. package/agent/extensions/pi-web-access/test/credential-source.test.mjs +118 -0
  118. package/agent/extensions/pi-web-access/test/data-uri-sanitize.test.mjs +210 -0
  119. package/agent/extensions/pi-web-access/test/datalab-pdf-extract.test.mjs +552 -0
  120. package/agent/extensions/pi-web-access/test/declared-web-links.test.mjs +212 -0
  121. package/agent/extensions/pi-web-access/test/fetch-answer-storage.test.mjs +40 -0
  122. package/agent/extensions/pi-web-access/test/fetch-cache-storage.test.mjs +334 -0
  123. package/agent/extensions/pi-web-access/test/fetch-content-domain-policy.test.mjs +95 -0
  124. package/agent/extensions/pi-web-access/test/fetch-modes.test.mjs +53 -0
  125. package/agent/extensions/pi-web-access/test/fetch-not-found-guidance.test.mjs +92 -0
  126. package/agent/extensions/pi-web-access/test/fetch-params.test.mjs +86 -0
  127. package/agent/extensions/pi-web-access/test/fetch-render-call.test.mjs +34 -0
  128. package/agent/extensions/pi-web-access/test/fetch-routing.test.mjs +173 -0
  129. package/agent/extensions/pi-web-access/test/gemini-adc-auth.test.mjs +257 -0
  130. package/agent/extensions/pi-web-access/test/gemini-api-transport.test.mjs +170 -0
  131. package/agent/extensions/pi-web-access/test/gemini-pdf-extract.test.mjs +133 -0
  132. package/agent/extensions/pi-web-access/test/gemini-web-cookie-opt-in.test.mjs +178 -0
  133. package/agent/extensions/pi-web-access/test/gemini-web-header-overflow.test.mjs +148 -0
  134. package/agent/extensions/pi-web-access/test/get-search-content.test.mjs +223 -0
  135. package/agent/extensions/pi-web-access/test/github-extract.test.mjs +378 -0
  136. package/agent/extensions/pi-web-access/test/github-issue-pr.test.mjs +565 -0
  137. package/agent/extensions/pi-web-access/test/inline-content-config.test.mjs +99 -0
  138. package/agent/extensions/pi-web-access/test/lazy-extract-load.test.mjs +118 -0
  139. package/agent/extensions/pi-web-access/test/local-video-oversize.test.mjs +52 -0
  140. package/agent/extensions/pi-web-access/test/package-typebox-dependency.test.mjs +50 -0
  141. package/agent/extensions/pi-web-access/test/page-query.test.mjs +51 -0
  142. package/agent/extensions/pi-web-access/test/pdf-config.test.mjs +140 -0
  143. package/agent/extensions/pi-web-access/test/pdf-extract.test.mjs +500 -0
  144. package/agent/extensions/pi-web-access/test/proxy-transport.test.mjs +286 -0
  145. package/agent/extensions/pi-web-access/test/query-rewrite.test.mjs +52 -0
  146. package/agent/extensions/pi-web-access/test/rsc-fallback.test.mjs +102 -0
  147. package/agent/extensions/pi-web-access/test/search-error-render.test.mjs +152 -0
  148. package/agent/extensions/pi-web-access/test/search-providers.test.mjs +274 -0
  149. package/agent/extensions/pi-web-access/test/source-check.test.mjs +179 -0
  150. package/agent/extensions/pi-web-access/test/ssrf-allow-ranges-config.test.mjs +205 -0
  151. package/agent/extensions/pi-web-access/test/ssrf-protection.test.mjs +456 -0
  152. package/agent/extensions/pi-web-access/test/summary-model-scope.test.mjs +106 -0
  153. package/agent/extensions/pi-web-access/test/tool-registration-config.test.mjs +182 -0
  154. package/agent/extensions/pi-web-access/test/web-search-answer-render.test.mjs +66 -0
  155. package/agent/extensions/pi-web-access/test/youtube-extract-errors.test.mjs +64 -0
  156. package/agent/extensions/pi-web-access/tsconfig.json +11 -0
  157. package/agent/extensions/pi-web-access/utils.ts +451 -0
  158. package/agent/extensions/pi-web-access/video-extract.ts +392 -0
  159. package/agent/extensions/pi-web-access/youtube-extract.ts +328 -0
  160. package/agent/extensions/shared/activity-status.ts +31 -0
  161. package/agent/extensions/shared/child-session.test.ts +270 -0
  162. package/agent/extensions/shared/child-session.ts +148 -0
  163. package/agent/extensions/shared/context-utilization.test.ts +48 -0
  164. package/agent/extensions/shared/context-utilization.ts +47 -0
  165. package/agent/extensions/shared/dashboard-state.ts +99 -0
  166. package/agent/extensions/shared/install.ps1 +23 -0
  167. package/agent/extensions/shared/install.sh +21 -0
  168. package/agent/extensions/shared/tool-call-timeout.test.ts +117 -0
  169. package/agent/extensions/shared/tool-call-timeout.ts +104 -0
  170. package/agent/extensions/skillsconf/README.md +74 -0
  171. package/agent/extensions/skillsconf/index.ts +92 -0
  172. package/agent/extensions/skillsconf/install.ps1 +23 -0
  173. package/agent/extensions/skillsconf/install.sh +21 -0
  174. package/agent/extensions/skillsconf/package-lock.json +1809 -0
  175. package/agent/extensions/skillsconf/package.json +18 -0
  176. package/agent/extensions/skillsconf/src/delete-skill.test.ts +64 -0
  177. package/agent/extensions/skillsconf/src/delete-skill.ts +45 -0
  178. package/agent/extensions/skillsconf/src/filter.test.ts +80 -0
  179. package/agent/extensions/skillsconf/src/filter.ts +74 -0
  180. package/agent/extensions/skillsconf/src/fuzzy.ts +26 -0
  181. package/agent/extensions/skillsconf/src/persistence.test.ts +72 -0
  182. package/agent/extensions/skillsconf/src/persistence.ts +169 -0
  183. package/agent/extensions/skillsconf/src/ui/SkillConfView.test.ts +337 -0
  184. package/agent/extensions/skillsconf/src/ui/SkillConfView.ts +758 -0
  185. package/agent/extensions/skillsconf/src/ui/text-input.ts +91 -0
  186. package/agent/extensions/skillsconf/src/ui/tui-helpers.ts +144 -0
  187. package/agent/extensions/skillsconf/tsconfig.json +15 -0
  188. package/agent/extensions/status line/index.ts +282 -0
  189. package/agent/extensions/status line/install.ps1 +23 -0
  190. package/agent/extensions/status line/install.sh +21 -0
  191. package/agent/extensions/subagents/by-the-way.test.ts +29 -0
  192. package/agent/extensions/subagents/claude.test.ts +119 -0
  193. package/agent/extensions/subagents/codex.test.ts +102 -0
  194. package/agent/extensions/subagents/context-usage.test.ts +107 -0
  195. package/agent/extensions/subagents/docs/design-plan.md +568 -0
  196. package/agent/extensions/subagents/docs/effect-v4-extension-guide.md +354 -0
  197. package/agent/extensions/subagents/docs/effect-v4-notes.md +571 -0
  198. package/agent/extensions/subagents/index.ts +779 -0
  199. package/agent/extensions/subagents/install.ps1 +23 -0
  200. package/agent/extensions/subagents/install.sh +21 -0
  201. package/agent/extensions/subagents/manager.test.ts +276 -0
  202. package/agent/extensions/subagents/package-lock.json +2244 -0
  203. package/agent/extensions/subagents/package.json +19 -0
  204. package/agent/extensions/subagents/result-delivery.test.ts +27 -0
  205. package/agent/extensions/subagents/src/backend.ts +73 -0
  206. package/agent/extensions/subagents/src/backends/claude.ts +701 -0
  207. package/agent/extensions/subagents/src/backends/codex.ts +1060 -0
  208. package/agent/extensions/subagents/src/backends/pi.ts +575 -0
  209. package/agent/extensions/subagents/src/backends/stub.ts +300 -0
  210. package/agent/extensions/subagents/src/by-the-way.ts +21 -0
  211. package/agent/extensions/subagents/src/domain.ts +253 -0
  212. package/agent/extensions/subagents/src/format.ts +74 -0
  213. package/agent/extensions/subagents/src/manager.ts +736 -0
  214. package/agent/extensions/subagents/src/prompt.ts +92 -0
  215. package/agent/extensions/subagents/src/result-delivery.ts +20 -0
  216. package/agent/extensions/subagents/src/runtime.ts +53 -0
  217. package/agent/extensions/subagents/src/ui/takeover.ts +583 -0
  218. package/agent/extensions/subagents/src/ui/transcript.ts +201 -0
  219. package/agent/extensions/subagents/takeover.test.ts +29 -0
  220. package/agent/extensions/subagents/tsconfig.json +7 -0
  221. package/agent/extensions/taste/index.ts +443 -0
  222. package/agent/extensions/taste/install.ps1 +23 -0
  223. package/agent/extensions/taste/install.sh +21 -0
  224. package/agent/extensions/todo/AGENTS.md +38 -0
  225. package/agent/extensions/todo/LICENSE +21 -0
  226. package/agent/extensions/todo/config.ts +55 -0
  227. package/agent/extensions/todo/index.ts +151 -0
  228. package/agent/extensions/todo/install.ps1 +23 -0
  229. package/agent/extensions/todo/install.sh +21 -0
  230. package/agent/extensions/todo/locales/de.json +17 -0
  231. package/agent/extensions/todo/locales/en.json +15 -0
  232. package/agent/extensions/todo/locales/es.json +17 -0
  233. package/agent/extensions/todo/locales/fr.json +17 -0
  234. package/agent/extensions/todo/locales/pt-BR.json +17 -0
  235. package/agent/extensions/todo/locales/pt.json +17 -0
  236. package/agent/extensions/todo/locales/ru.json +17 -0
  237. package/agent/extensions/todo/locales/uk.json +17 -0
  238. package/agent/extensions/todo/locales/zh.json +17 -0
  239. package/agent/extensions/todo/package-lock.json +3358 -0
  240. package/agent/extensions/todo/package.json +67 -0
  241. package/agent/extensions/todo/state/i18n-bridge.ts +64 -0
  242. package/agent/extensions/todo/state/invariants.ts +20 -0
  243. package/agent/extensions/todo/state/replay.ts +38 -0
  244. package/agent/extensions/todo/state/selectors.ts +107 -0
  245. package/agent/extensions/todo/state/state-reducer.ts +326 -0
  246. package/agent/extensions/todo/state/state.ts +18 -0
  247. package/agent/extensions/todo/state/store.ts +82 -0
  248. package/agent/extensions/todo/state/task-graph.ts +57 -0
  249. package/agent/extensions/todo/todo-overlay.ts +200 -0
  250. package/agent/extensions/todo/todo.ts +155 -0
  251. package/agent/extensions/todo/tool/response-envelope.ts +109 -0
  252. package/agent/extensions/todo/tool/types.ts +206 -0
  253. package/agent/extensions/todo/verify-ref-system.js +0 -0
  254. package/agent/extensions/todo/view/format.ts +177 -0
  255. package/agent/extensions/trim-context/README.md +54 -0
  256. package/agent/extensions/trim-context/index.ts +487 -0
  257. package/agent/extensions/trim-context/install.ps1 +23 -0
  258. package/agent/extensions/trim-context/install.sh +21 -0
  259. package/agent/install.ps1 +527 -0
  260. package/agent/install.sh +511 -0
  261. package/agent/keybindings.json +7 -0
  262. package/bin/kiln.js +359 -0
  263. package/package.json +21 -0
@@ -0,0 +1,178 @@
1
+ import assert from "node:assert/strict";
2
+ import { spawnSync } from "node:child_process";
3
+ import { mkdir, mkdtemp, writeFile } from "node:fs/promises";
4
+ import { tmpdir } from "node:os";
5
+ import { join } from "node:path";
6
+ import { test } from "node:test";
7
+
8
+ const moduleUrl = new URL("../gemini-web-config.ts", import.meta.url).href;
9
+ const geminiWebUrl = new URL("../gemini-web.ts", import.meta.url).href;
10
+
11
+ function runCookieAccessCheck(home, extraEnv = {}) {
12
+ const env = { ...process.env, HOME: home, USERPROFILE: home, ...extraEnv };
13
+ delete env.PI_ALLOW_BROWSER_COOKIES;
14
+ delete env.FEYNMAN_ALLOW_BROWSER_COOKIES;
15
+ delete env.PI_CODING_AGENT_DIR;
16
+ delete env.XDG_CONFIG_HOME;
17
+ Object.assign(env, extraEnv);
18
+
19
+ return spawnSync(process.execPath, ["--input-type=module"], {
20
+ input: `const { isBrowserCookieAccessAllowed } = await import(${JSON.stringify(moduleUrl)}); console.log(String(isBrowserCookieAccessAllowed()));`,
21
+ encoding: "utf8",
22
+ env,
23
+ });
24
+ }
25
+
26
+ function runBrowserCookieConfig(home) {
27
+ const env = { ...process.env, HOME: home, USERPROFILE: home };
28
+ delete env.PI_CODING_AGENT_DIR;
29
+ delete env.XDG_CONFIG_HOME;
30
+ return spawnSync(process.execPath, ["--experimental-strip-types", "--input-type=module"], {
31
+ input: `const { getBrowserCookieSelectionFromConfig } = await import(${JSON.stringify(moduleUrl)}); console.log(JSON.stringify(getBrowserCookieSelectionFromConfig()));`,
32
+ encoding: "utf8",
33
+ env,
34
+ });
35
+ }
36
+
37
+ test("Gemini Web never forwards browser cookies across origins", async () => {
38
+ const { getActiveGoogleEmail, setGeminiFetchOverrideForTests } = await import(geminiWebUrl);
39
+ const calls = [];
40
+ setGeminiFetchOverrideForTests(async (url, init = {}) => {
41
+ calls.push({ url: String(url), cookie: init.headers?.cookie, redirect: init.redirect });
42
+ if (String(url).startsWith("https://gemini.google.com/") || String(url).startsWith("https://accounts.google.com/")) {
43
+ return new Response(null, { status: 302, headers: { location: "https://attacker.example/collect" } });
44
+ }
45
+ throw new Error(`Unexpected cross-origin request: ${url}`);
46
+ });
47
+ try {
48
+ const email = await getActiveGoogleEmail({ "__Secure-1PSID": "sensitive-cookie" });
49
+ assert.equal(email, null);
50
+ assert.equal(calls.some((call) => call.url.startsWith("https://attacker.example/")), false);
51
+ assert.equal(calls.every((call) => call.redirect === "manual"), true);
52
+ } finally {
53
+ setGeminiFetchOverrideForTests(null);
54
+ }
55
+ });
56
+
57
+ test("Gemini Web generation rejects automatic redirects", async () => {
58
+ const { queryWithCookies, setGeminiFetchOverrideForTests } = await import(geminiWebUrl);
59
+ setGeminiFetchOverrideForTests(async (url, init = {}) => {
60
+ if (String(url) === "https://gemini.google.com/app") {
61
+ return new Response('"SNlM0e":"test-token"', { status: 200 });
62
+ }
63
+ if (String(url).includes("BardFrontendService/StreamGenerate")) {
64
+ assert.equal(init.redirect, "error");
65
+ throw new Error("generation transport reached");
66
+ }
67
+ throw new Error(`Unexpected request: ${url}`);
68
+ });
69
+ try {
70
+ await assert.rejects(
71
+ queryWithCookies("search", { "__Secure-1PSID": "cookie" }, { model: "gemini-3.1-pro" }),
72
+ /generation transport reached/,
73
+ );
74
+ } finally {
75
+ setGeminiFetchOverrideForTests(null);
76
+ }
77
+ });
78
+
79
+ test("Gemini Web rejects unsupported models instead of falling back to 2.5 Flash", async () => {
80
+ const { queryWithCookies, setGeminiFetchOverrideForTests } = await import(geminiWebUrl);
81
+ setGeminiFetchOverrideForTests(async () => {
82
+ throw new Error("transport should not be reached");
83
+ });
84
+ try {
85
+ await assert.rejects(
86
+ queryWithCookies("search", { "__Secure-1PSID": "cookie" }, { model: "gemini-3.6-flash" }),
87
+ /Gemini Web does not support model gemini-3\.6-flash/,
88
+ );
89
+ } finally {
90
+ setGeminiFetchOverrideForTests(null);
91
+ }
92
+ });
93
+
94
+ test("Gemini Web file uploads read the file and reject automatic redirects", async () => {
95
+ const dir = await mkdtemp(join(tmpdir(), "pi-web-access-gemini-upload-"));
96
+ const filePath = join(dir, "sample.txt");
97
+ await writeFile(filePath, "sample", "utf8");
98
+ const { queryWithCookies, setGeminiFetchOverrideForTests } = await import(geminiWebUrl);
99
+ setGeminiFetchOverrideForTests(async (url, init = {}) => {
100
+ if (String(url) === "https://gemini.google.com/app") {
101
+ return new Response('"SNlM0e":"test-token"', { status: 200 });
102
+ }
103
+ if (String(url) === "https://content-push.googleapis.com/upload") {
104
+ assert.equal(init.redirect, "error");
105
+ throw new Error("upload transport reached");
106
+ }
107
+ throw new Error(`Unexpected request: ${url}`);
108
+ });
109
+ try {
110
+ await assert.rejects(
111
+ queryWithCookies("inspect file", { "__Secure-1PSID": "cookie" }, { files: [filePath], model: "gemini-3.1-pro" }),
112
+ /upload transport reached/,
113
+ );
114
+ } finally {
115
+ setGeminiFetchOverrideForTests(null);
116
+ }
117
+ });
118
+
119
+ test("browser cookie access is disabled unless explicitly allowed", async () => {
120
+ const home = await mkdtemp(join(tmpdir(), "pi-web-access-cookie-opt-in-"));
121
+
122
+ let child = runCookieAccessCheck(home);
123
+ assert.equal(child.status, 0, child.stderr);
124
+ assert.equal(child.stdout.trim(), "false");
125
+
126
+ await mkdir(join(home, ".pi"), { recursive: true });
127
+ await writeFile(join(home, ".pi", "web-search.json"), JSON.stringify({ allowBrowserCookies: true }) + "\n", "utf8");
128
+
129
+ child = runCookieAccessCheck(home);
130
+ assert.equal(child.status, 0, child.stderr);
131
+ assert.equal(child.stdout.trim(), "true");
132
+
133
+ const envHome = await mkdtemp(join(tmpdir(), "pi-web-access-cookie-env-"));
134
+ child = runCookieAccessCheck(envHome, { PI_ALLOW_BROWSER_COOKIES: "1" });
135
+ assert.equal(child.status, 0, child.stderr);
136
+ assert.equal(child.stdout.trim(), "true");
137
+ });
138
+
139
+ test("disabled browser cookie access does not validate legacy profile selection", async () => {
140
+ const home = await mkdtemp(join(tmpdir(), "pi-web-access-cookie-legacy-disabled-"));
141
+ await mkdir(join(home, ".pi"), { recursive: true });
142
+ await writeFile(join(home, ".pi", "web-search.json"), JSON.stringify({ chromeProfile: "Profile 1" }) + "\n", "utf8");
143
+
144
+ const child = runCookieAccessCheck(home);
145
+ assert.equal(child.status, 0, child.stderr);
146
+ assert.equal(child.stdout.trim(), "false");
147
+ });
148
+
149
+ test("browser cookie config validates presets and rejects arbitrary profile paths", async () => {
150
+ const home = await mkdtemp(join(tmpdir(), "pi-web-access-browser-cookie-config-"));
151
+ await mkdir(join(home, ".pi"), { recursive: true });
152
+ const configPath = join(home, ".pi", "web-search.json");
153
+ await writeFile(configPath, JSON.stringify({ browserCookies: { browser: "Helium", profile: "Profile 1" } }) + "\n", "utf8");
154
+
155
+ let child = runBrowserCookieConfig(home);
156
+ assert.equal(child.status, 0, child.stderr);
157
+ assert.deepEqual(JSON.parse(child.stdout), { browser: "helium", profile: "Profile 1" });
158
+
159
+ await writeFile(configPath, JSON.stringify({ chromeProfile: "Profile 1", allowBrowserCookies: true }) + "\n", "utf8");
160
+ child = runBrowserCookieConfig(home);
161
+ assert.notEqual(child.status, 0);
162
+ assert.match(child.stderr, /chromeProfile.*no longer supported.*browserCookies\.profile/);
163
+
164
+ await writeFile(configPath, JSON.stringify({ browserCookies: { browser: "safari", profile: "Profile 1" } }) + "\n", "utf8");
165
+ child = runBrowserCookieConfig(home);
166
+ assert.notEqual(child.status, 0);
167
+ assert.match(child.stderr, /Unsupported browserCookies\.browser/);
168
+
169
+ await writeFile(configPath, JSON.stringify({ browserCookies: { browser: "helium", profile: 1 } }) + "\n", "utf8");
170
+ child = runBrowserCookieConfig(home);
171
+ assert.notEqual(child.status, 0);
172
+ assert.match(child.stderr, /browserCookies\.profile.*must be a profile directory name/);
173
+
174
+ await writeFile(configPath, JSON.stringify({ browserCookies: { browser: "helium", profilePath: "/tmp/Profile 1" } }) + "\n", "utf8");
175
+ child = runBrowserCookieConfig(home);
176
+ assert.notEqual(child.status, 0);
177
+ assert.match(child.stderr, /profilePath is not supported/);
178
+ });
@@ -0,0 +1,148 @@
1
+ import assert from "node:assert/strict";
2
+ import { spawnSync } from "node:child_process";
3
+ import { copyFileSync, mkdirSync, mkdtempSync, rmSync, writeFileSync } from "node:fs";
4
+ import { createServer } from "node:http";
5
+ import { tmpdir } from "node:os";
6
+ import { join } from "node:path";
7
+ import { test } from "node:test";
8
+
9
+ import { createGeminiFetch, GEMINI_MAX_HEADER_SIZE, resolveGeminiFetch } from "../gemini-web.ts";
10
+
11
+ // Proxy env vars would route the dedicated EnvHttpProxyAgent (and the
12
+ // simulated host-agent dispatcher) away from the local test server.
13
+ const PROXY_ENV = [
14
+ "HTTP_PROXY",
15
+ "HTTPS_PROXY",
16
+ "http_proxy",
17
+ "https_proxy",
18
+ "ALL_PROXY",
19
+ "all_proxy",
20
+ "NO_PROXY",
21
+ "no_proxy",
22
+ ];
23
+
24
+ test("createGeminiFetch wires a dedicated agent with a >16 KiB header budget", async () => {
25
+ const agentOpts = [];
26
+ const agentInstances = [];
27
+ const fetchCalls = [];
28
+ const stubUndici = {
29
+ EnvHttpProxyAgent: class {
30
+ constructor(opts) {
31
+ agentOpts.push(opts);
32
+ agentInstances.push(this);
33
+ }
34
+ },
35
+ fetch: async (input, init) => {
36
+ fetchCalls.push({ input, init });
37
+ return new Response("ok");
38
+ },
39
+ };
40
+
41
+ const geminiFetch = createGeminiFetch(stubUndici);
42
+ await geminiFetch("https://gemini.google.com/app", { headers: { cookie: "a=b" } });
43
+
44
+ assert.equal(agentOpts.length, 1);
45
+ const opts = agentOpts[0];
46
+ assert.equal(opts.maxHeaderSize, GEMINI_MAX_HEADER_SIZE);
47
+ // The whole point: Google frontend responses exceed undici's 16 KiB default
48
+ // over HTTP/1.1, so the dedicated budget must be well above it.
49
+ assert.ok(opts.maxHeaderSize > 16 * 1024, "header budget must exceed undici's 16 KiB default");
50
+ assert.equal(opts.allowH2, false);
51
+ assert.equal(opts.pipelining, 1);
52
+
53
+ assert.equal(fetchCalls.length, 1);
54
+ // Requests must go through the dedicated agent, not the ambient global dispatcher.
55
+ assert.equal(fetchCalls[0].input, "https://gemini.google.com/app");
56
+ assert.equal(fetchCalls[0].init.dispatcher, agentInstances[0]);
57
+ });
58
+
59
+ test("dedicated agent fetches oversized-header responses the pi-style global dispatcher rejects", async () => {
60
+ const server = createServer((req, res) => {
61
+ // ~85 KiB of response headers: Google's /app page routinely exceeds the
62
+ // 16 KiB default maxHeaderSize when served over HTTP/1.1.
63
+ for (let i = 0; i < 40; i++) res.setHeader(`X-Large-${i}`, "x".repeat(2048));
64
+ res.end("payload-ok");
65
+ });
66
+ await new Promise((resolve) => server.listen(0, "127.0.0.1", resolve));
67
+ const url = `http://127.0.0.1:${server.address().port}/`;
68
+
69
+ const undici = await import("undici");
70
+ const savedDispatcher = undici.getGlobalDispatcher();
71
+ const savedEnv = Object.fromEntries(PROXY_ENV.map((k) => [k, process.env[k]]));
72
+ for (const k of PROXY_ENV) delete process.env[k];
73
+
74
+ try {
75
+ // Simulate the host agent (pi): an HTTP/1.1-only global dispatcher with
76
+ // undici's default 16 KiB maxHeaderSize.
77
+ undici.setGlobalDispatcher(new undici.EnvHttpProxyAgent({ allowH2: false }));
78
+
79
+ // Control: the ambient global fetch must choke on the oversized headers.
80
+ await assert.rejects(
81
+ () => fetch(url).then((r) => r.text()),
82
+ (err) => {
83
+ const detail = String(err.cause?.code ?? err.message);
84
+ assert.match(detail, /HEADERS_OVERFLOW|fetch failed/i);
85
+ return true;
86
+ },
87
+ );
88
+
89
+ // Fix: the dedicated agent returns the same response without error.
90
+ const geminiFetch = createGeminiFetch(undici);
91
+ const res = await geminiFetch(url);
92
+ assert.equal(res.status, 200);
93
+ assert.equal(await res.text(), "payload-ok");
94
+ } finally {
95
+ undici.setGlobalDispatcher(savedDispatcher);
96
+ for (const k of PROXY_ENV) {
97
+ if (savedEnv[k] === undefined) delete process.env[k];
98
+ else process.env[k] = savedEnv[k];
99
+ }
100
+ server.close();
101
+ }
102
+ });
103
+
104
+ test("resolveGeminiFetch falls back to the ambient global fetch when undici is not resolvable", () => {
105
+ const dir = mkdtempSync(join(tmpdir(), "gw-fallback-"));
106
+ try {
107
+ // Copy just the module graph (no real node_modules) so the dynamic
108
+ // undici import fails and the fallback path is exercised. A broken
109
+ // local node_modules/undici stub shadows any ambiently resolvable
110
+ // undici (bundled with newer Node, or an ancestor node_modules), so
111
+ // the test stays hermetic. Note: a package.json `imports` block
112
+ // cannot pin bare specifiers (keys must start with `#`), so the stub
113
+ // is required.
114
+ for (const f of ["gemini-web.ts", "chrome-cookies.ts", "gemini-web-config.ts", "utils.ts"]) {
115
+ copyFileSync(join(import.meta.dirname, "..", f), join(dir, f));
116
+ }
117
+ const stubDir = join(dir, "node_modules", "undici");
118
+ mkdirSync(stubDir, { recursive: true });
119
+ writeFileSync(join(stubDir, "package.json"), JSON.stringify({ name: "undici", version: "0.0.0-stub", type: "module", main: "./index.js" }));
120
+ writeFileSync(join(stubDir, "index.js"), "throw new Error('stub undici is not importable');");
121
+ const script = [
122
+ "import { resolveGeminiFetch } from './gemini-web.ts';",
123
+ "const f = await resolveGeminiFetch();",
124
+ "console.log('FETCH_KIND:' + (f === globalThis.fetch ? 'FALLBACK' : 'UNDICI'));",
125
+ ].join("\n");
126
+ const r = spawnSync(process.execPath, ["-e", script], { cwd: dir, encoding: "utf8" });
127
+ assert.equal(r.status, 0, r.stderr);
128
+ assert.match(r.stdout, /FETCH_KIND:FALLBACK/);
129
+ } finally {
130
+ rmSync(dir, { recursive: true, force: true });
131
+ }
132
+ });
133
+
134
+ test("resolveGeminiFetch surfaces dedicated agent construction errors", () => {
135
+ const env = { ...process.env, HTTPS_PROXY: "not a url" };
136
+ const script = [
137
+ `import { resolveGeminiFetch } from ${JSON.stringify(new URL("../gemini-web.ts", import.meta.url).href)};`,
138
+ "await resolveGeminiFetch();",
139
+ ].join("\n");
140
+ const result = spawnSync(process.execPath, ["--input-type=module"], {
141
+ input: script,
142
+ encoding: "utf8",
143
+ env,
144
+ });
145
+
146
+ assert.notEqual(result.status, 0);
147
+ assert.match(result.stderr, /Invalid URL/);
148
+ });
@@ -0,0 +1,223 @@
1
+ import assert from "node:assert/strict";
2
+ import { test } from "node:test";
3
+ import { Value } from "typebox/value";
4
+
5
+ import initializeExtension from "../index.ts";
6
+ import { buildResearchArtifact, storeResearchArtifact } from "../source-check.ts";
7
+ import { clearResults, storeResult } from "../storage.ts";
8
+
9
+ function getContentTool() {
10
+ clearResults();
11
+ const tools = [];
12
+ initializeExtension({
13
+ registerTool(tool) { tools.push(tool); },
14
+ registerCommand() {},
15
+ registerShortcut() {},
16
+ on() {},
17
+ });
18
+ const tool = tools.find((registered) => registered.name === "get_search_content");
19
+ assert.ok(tool, "get_search_content tool was not registered");
20
+ return tool;
21
+ }
22
+
23
+ function storeFetchedContent(content) {
24
+ storeResult("large-fetch", {
25
+ id: "large-fetch",
26
+ type: "fetch",
27
+ timestamp: Date.now(),
28
+ urls: [{
29
+ url: "https://example.com/large",
30
+ title: "Large Page",
31
+ content,
32
+ error: null,
33
+ }],
34
+ });
35
+ }
36
+
37
+ function storeSearchContent() {
38
+ storeResult("search-result", {
39
+ id: "search-result",
40
+ type: "search",
41
+ timestamp: Date.now(),
42
+ queries: [{
43
+ query: "CotEditor scripts",
44
+ answer: "",
45
+ results: [
46
+ { title: "ScriptManager.swift", url: "https://example.com/script-manager", snippet: "UNIX script support" },
47
+ { title: "UNIX script", url: "https://example.com/unix-script", snippet: "Script support" },
48
+ { title: "ScriptMenu", url: "https://example.com/script-menu", snippet: "Menu integration" },
49
+ ],
50
+ error: null,
51
+ }],
52
+ });
53
+ }
54
+
55
+ test("get_search_content schemas constrain numeric parameters", () => {
56
+ const properties = getContentTool().parameters.properties;
57
+
58
+ for (const name of ["queryIndex", "urlIndex", "offset"]) {
59
+ assert.equal(properties[name].type, "integer");
60
+ assert.equal(properties[name].minimum, 0);
61
+ assert.equal(properties[name].maximum, undefined);
62
+ for (const value of [-1, 1.5, Number.NaN, Number.POSITIVE_INFINITY]) {
63
+ assert.equal(Value.Check(properties[name], value), false, `${name} accepts ${value}`);
64
+ }
65
+ }
66
+
67
+ assert.equal(properties.limit.type, "integer");
68
+ assert.equal(properties.limit.minimum, 1);
69
+ assert.equal(properties.limit.maximum, 30_000);
70
+ for (const value of [0, 1.5, 30_001, Number.NaN, Number.POSITIVE_INFINITY]) {
71
+ assert.equal(Value.Check(properties.limit, value), false, `limit accepts ${value}`);
72
+ }
73
+ for (const value of [1, 30_000]) {
74
+ assert.equal(Value.Check(properties.limit, value), true, `limit rejects ${value}`);
75
+ }
76
+ });
77
+
78
+ test("get_search_content returns a bounded first slice for large fetched content", async () => {
79
+ const tool = getContentTool();
80
+ storeFetchedContent("A".repeat(30_000) + "TAIL");
81
+
82
+ const result = await tool.execute("call", { responseId: "large-fetch", urlIndex: 0 });
83
+ const text = result.content[0].text;
84
+
85
+ assert.equal(result.details.contentLength, 30_004);
86
+ assert.equal(result.details.offset, 0);
87
+ assert.equal(result.details.returnedChars, 30_000);
88
+ assert.equal(result.details.nextOffset, 30_000);
89
+ assert.equal(result.details.truncated, true);
90
+ assert.match(text, /Showing chars 0-30000 of 30004/);
91
+ assert.match(text, /offset: 30000/);
92
+ assert.doesNotMatch(text, /TAIL/);
93
+ });
94
+
95
+ test("get_search_content returns requested fetched content slices", async () => {
96
+ const tool = getContentTool();
97
+ storeFetchedContent("A".repeat(30_000) + "BCDEFGHIJ");
98
+
99
+ const result = await tool.execute("call", {
100
+ responseId: "large-fetch",
101
+ url: "https://example.com/large",
102
+ offset: 30_000,
103
+ limit: 5,
104
+ });
105
+ const text = result.content[0].text;
106
+
107
+ assert.equal(result.details.offset, 30_000);
108
+ assert.equal(result.details.limit, 5);
109
+ assert.equal(result.details.returnedChars, 5);
110
+ assert.equal(result.details.nextOffset, 30_005);
111
+ assert.match(text, /BCDEF/);
112
+ assert.doesNotMatch(text, /GHIJ/);
113
+ assert.match(text, /urlIndex: 0, offset: 30005, limit: 5/);
114
+ });
115
+
116
+ test("get_search_content rejects unsafe fetched content ranges", async () => {
117
+ const tool = getContentTool();
118
+ storeFetchedContent("short content");
119
+
120
+ const tooLarge = await tool.execute("call", { responseId: "large-fetch", urlIndex: 0, limit: 30_001 });
121
+ assert.equal(tooLarge.details.error, "Invalid limit");
122
+ assert.match(tooLarge.content[0].text, /received 30001/);
123
+ assert.match(tooLarge.content[0].text, /limit must be an integer from 1 to 30000/);
124
+
125
+ const invalidOffset = await tool.execute("call", { responseId: "large-fetch", urlIndex: 0, offset: 1.5 });
126
+ assert.equal(invalidOffset.details.error, "Invalid offset");
127
+ assert.match(invalidOffset.content[0].text, /received 1\.5/);
128
+ assert.match(invalidOffset.content[0].text, /Use 0 or a larger integer/);
129
+
130
+ const outOfRange = await tool.execute("call", { responseId: "large-fetch", urlIndex: 0, offset: 99 });
131
+ assert.equal(outOfRange.details.error, "Offset out of range");
132
+ assert.match(outOfRange.content[0].text, /Received offset 99/);
133
+ assert.match(outOfRange.content[0].text, /valid range is 0-13/);
134
+
135
+ const missingFindText = await tool.execute("call", { responseId: "large-fetch", urlIndex: 0, findMode: "fuzzy" });
136
+ assert.equal(missingFindText.details.error, "findMode requires findText");
137
+ assert.match(missingFindText.content[0].text, /findMode "fuzzy" requires findText/);
138
+
139
+ const artifact = buildResearchArtifact({
140
+ query: "stored claim",
141
+ results: [{ url: "https://example.com/research", title: "Research source", snippet: "The bridge defaults match this research passage.", rank: 1 }],
142
+ summary: "Unique bridge research summary marker.",
143
+ });
144
+ artifact.id = "stored-research";
145
+ storeResearchArtifact(artifact);
146
+ const researchInvalidLimit = await tool.execute("call", { responseId: "stored-research", limit: 30_001 });
147
+ assert.equal(researchInvalidLimit.details.error, "Invalid limit");
148
+ assert.match(researchInvalidLimit.content[0].text, /received 30001/);
149
+ assert.match(researchInvalidLimit.content[0].text, /limit must be an integer from 1 to 30000/);
150
+
151
+ const researchOutOfRange = await tool.execute("call", { responseId: "stored-research", offset: 99_999 });
152
+ assert.equal(researchOutOfRange.details.error, "Offset out of range");
153
+ assert.match(researchOutOfRange.content[0].text, /responseId "stored-research"/);
154
+ assert.match(researchOutOfRange.content[0].text, /valid range is 0-/);
155
+
156
+ const researchFind = await tool.execute("call", {
157
+ responseId: "stored-research",
158
+ offset: 0,
159
+ limit: 10_000,
160
+ findText: "Unique bridge research summary marker",
161
+ });
162
+ assert.equal(researchFind.details.type, "research");
163
+ assert.equal(researchFind.details.findMode, "case-insensitive");
164
+ assert.equal(researchFind.details.matchCount, 1);
165
+ assert.match(researchFind.content[0].text, /^Text matches \(case-insensitive\)/);
166
+ assert.match(researchFind.content[0].text, /Unique bridge research summary marker/);
167
+ });
168
+
169
+ test("get_search_content normalizes bridge defaults for search matches", async () => {
170
+ const tool = getContentTool();
171
+ storeSearchContent();
172
+
173
+ const result = await tool.execute("call", {
174
+ responseId: "search-result",
175
+ query: "",
176
+ queryIndex: 0,
177
+ url: "",
178
+ urlIndex: 0,
179
+ offset: 0,
180
+ limit: 10_000,
181
+ findText: ["ScriptManager.swift", "UNIX script", "ScriptMenu"],
182
+ findMode: "case-insensitive",
183
+ });
184
+
185
+ assert.equal(result.details.findMode, "case-insensitive");
186
+ assert.equal(result.details.matchCount, 3);
187
+ assert.match(result.content[0].text, /ScriptManager\.swift/);
188
+ assert.match(result.content[0].text, /ScriptMenu/);
189
+ });
190
+
191
+ test("get_search_content returns small fetched content without continuation noise", async () => {
192
+ const tool = getContentTool();
193
+ storeFetchedContent("small content");
194
+
195
+ const result = await tool.execute("call", { responseId: "large-fetch", urlIndex: 0 });
196
+ const text = result.content[0].text;
197
+
198
+ assert.equal(result.details.returnedChars, "small content".length);
199
+ assert.equal(result.details.nextOffset, null);
200
+ assert.match(text, /small content/);
201
+ assert.doesNotMatch(text, /Showing chars/);
202
+ });
203
+
204
+ test("get_search_content finds bounded passages in stored fetched content", async () => {
205
+ const tool = getContentTool();
206
+ storeFetchedContent(`prefix ${"A".repeat(2_000)} Installation requires Node 22. ${"B".repeat(2_000)} suffix`);
207
+
208
+ const result = await tool.execute("call", {
209
+ responseId: "large-fetch",
210
+ query: "",
211
+ queryIndex: 0,
212
+ url: "",
213
+ urlIndex: 0,
214
+ offset: 0,
215
+ limit: 10_000,
216
+ findText: "installation",
217
+ });
218
+
219
+ assert.equal(result.details.matchCount, 1);
220
+ assert.equal(result.details.findMode, "case-insensitive");
221
+ assert.match(result.content[0].text, /Installation requires Node 22/);
222
+ assert.ok(result.content[0].text.length < 1_000);
223
+ });