diff options
| author | amankumarsingh77 <amankumarsingh7702@gmail.com> | 2026-07-28 00:53:19 +0530 |
|---|---|---|
| committer | Aman Kumar <145339995+amankumarsingh77@users.noreply.github.com> | 2026-08-02 19:12:46 +0530 |
| commit | b9c77bb6f9b8b83574489344ef6efd207a611e98 (patch) | |
| tree | 3c59bf53bcce3ac5c6bbf9a22884b6e6f787dc91 /src/domains | |
| parent | 1c73f4af643aa60ca554a3770264791d299e64e2 (diff) | |
refactor: validate JSON at every parse boundary
Ten sites cast or implicitly typed JSON.parse output; JSON.parse("null")
would have satisfied all of them. parseJson returns a Result and validates
against a typebox schema. js.ts keeps unknown — user script output has no
fixed shape. daemon/bridge.ts assigned JSON.parse straight to CdpRawMessage
with no guard at all; it now uses isCdpRawMessage like the transport does.
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Diffstat (limited to 'src/domains')
| -rw-r--r-- | src/domains/js.ts | 2 | ||||
| -rw-r--r-- | src/domains/readpage/read-page.ts | 42 | ||||
| -rw-r--r-- | src/domains/search/web-search.ts | 34 |
3 files changed, 47 insertions, 31 deletions
diff --git a/src/domains/js.ts b/src/domains/js.ts index b4a5b4c..61dbe53 100644 --- a/src/domains/js.ts +++ b/src/domains/js.ts @@ -61,7 +61,7 @@ export const executeJsTool = defineBrowserTool({ // to the raw string if parse fails (e.g. value is "undefined" sentinel). let pretty: string | undefined; try { - const parsed = JSON.parse(valueStr); + const parsed: unknown = JSON.parse(valueStr); pretty = JSON.stringify(parsed, null, 2); } catch { pretty = undefined; diff --git a/src/domains/readpage/read-page.ts b/src/domains/readpage/read-page.ts index 453c347..0a6fadc 100644 --- a/src/domains/readpage/read-page.ts +++ b/src/domains/readpage/read-page.ts @@ -1,4 +1,6 @@ import { Type } from "typebox"; +import { Compile } from "typebox/compile"; +import { parseJson } from "../../schemas/parse"; import { type Result, err, ok } from "../../util/result"; import { applyTruncation } from "../../util/truncate"; import { defineBrowserTool, type ToolErr, type ToolOk } from "../../util/tool"; @@ -24,23 +26,39 @@ const ReadPageArgs = Type.Object({ ), }); -const isPageCapture = (value: unknown): value is PageCapture => { - if (typeof value !== "object" || value === null) return false; - const v = value as Record<string, unknown>; - return typeof v["url"] === "string" && typeof v["title"] === "string" && Array.isArray(v["blocks"]) && typeof v["bodyText"] === "string"; -}; +const ContentBlockSchema = Type.Object({ + kind: Type.Union([ + Type.Literal("paragraph"), + Type.Literal("heading"), + Type.Literal("listitem"), + Type.Literal("blockquote"), + Type.Literal("other"), + ]), + text: Type.String(), + linkTextLength: Type.Number(), + inBoilerplate: Type.Boolean(), +}); -const safeParse = (text: string): unknown => { - try { - return JSON.parse(text) as unknown; - } catch { - return undefined; +const PageCaptureSchema = Type.Object({ + url: Type.String(), + title: Type.String(), + blocks: Type.Array(ContentBlockSchema), + bodyText: Type.String(), +}); + +const pageCaptureValidator = Compile(PageCaptureSchema); + +const toPageCapture = (raw: unknown): PageCapture | undefined => { + if (typeof raw === "string") { + const parsed = parseJson(raw, pageCaptureValidator); + return parsed.success ? parsed.data : undefined; } + return pageCaptureValidator.Check(raw) ? raw : undefined; }; const captureToResult = async (raw: unknown): Promise<Result<ToolOk, ToolErr>> => { - const parsed = typeof raw === "string" ? safeParse(raw) : raw; - if (!isPageCapture(parsed)) return err({ kind: "internal", message: "page capture returned an unexpected shape" }); + const parsed = toPageCapture(raw); + if (parsed === undefined) return err({ kind: "internal", message: "page capture returned an unexpected shape" }); const page: ReadablePage = extractReadable(parsed); const header = `# ${page.title}\n${page.url}\n(${page.wordCount} words)\n\n`; const truncated = await applyTruncation(header + page.text, "readpage"); diff --git a/src/domains/search/web-search.ts b/src/domains/search/web-search.ts index 69677b4..9327f15 100644 --- a/src/domains/search/web-search.ts +++ b/src/domains/search/web-search.ts @@ -1,4 +1,6 @@ import { Type } from "typebox"; +import { Compile } from "typebox/compile"; +import { parseJson } from "../../schemas/parse"; import { type Result, err, ok } from "../../util/result"; import { applyTruncation } from "../../util/truncate"; import { defineBrowserTool, type ToolErr, type ToolOk } from "../../util/tool"; @@ -26,11 +28,14 @@ const WebSearchArgs = Type.Object({ const buildSerpUrl = (query: string, limit: number): string => `https://www.google.com/search?q=${encodeURIComponent(query)}&num=${limit}&hl=en`; -const isSerpExtraction = (value: unknown): value is SerpExtraction => { - if (typeof value !== "object" || value === null) return false; - const v = value as Record<string, unknown>; - return Array.isArray(v["anchors"]) && typeof v["pageText"] === "string"; -}; +const SerpExtractionSchema = Type.Object({ + anchors: Type.Array( + Type.Object({ href: Type.String(), heading: Type.String(), snippet: Type.String() }), + ), + pageText: Type.String(), +}); + +const serpExtractionValidator = Compile(SerpExtractionSchema); const formatResults = (query: string, results: ReadonlyArray<SearchResult>): string => { const lines = results.map((r) => `${r.rank}. ${r.title}\n ${r.url}\n ${r.snippet}`); @@ -69,13 +74,14 @@ export const webSearchTool = defineBrowserTool({ const extracted = await evalInIsolatedTab(client, tab, buildSerpExtractionExpr(), isJsonText); if (!extracted.success) return err(extracted.error); - const parsedJson = safeParse(extracted.data); - if (!isSerpExtraction(parsedJson)) { - return err({ kind: "internal", message: "SERP extraction returned an unexpected shape" }); + const parsedJson = parseJson(extracted.data, serpExtractionValidator); + if (!parsedJson.success) { + return err({ kind: "internal", message: `SERP extraction returned an unexpected shape: ${parsedJson.error}` }); } + const extraction: SerpExtraction = parsedJson.data; - const results = parseGoogleSerp(parsedJson.anchors, limit); - const verdict = classifySerp(parsedJson.pageText, results.length); + const results = parseGoogleSerp(extraction.anchors, limit); + const verdict = classifySerp(extraction.pageText, results.length); if (verdict !== "ok") { return err({ kind: "invalid_state", @@ -108,11 +114,3 @@ export const webSearchTool = defineBrowserTool({ return renderExpandableText("web_search", result, expanded, theme); }, }); - -const safeParse = (text: string): unknown => { - try { - return JSON.parse(text) as unknown; - } catch { - return undefined; - } -}; |
