aboutsummaryrefslogtreecommitdiff
path: root/src/domains
diff options
context:
space:
mode:
authoramankumarsingh77 <amankumarsingh7702@gmail.com>2026-07-28 00:53:19 +0530
committerAman Kumar <145339995+amankumarsingh77@users.noreply.github.com>2026-08-02 19:12:46 +0530
commitb9c77bb6f9b8b83574489344ef6efd207a611e98 (patch)
tree3c59bf53bcce3ac5c6bbf9a22884b6e6f787dc91 /src/domains
parent1c73f4af643aa60ca554a3770264791d299e64e2 (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.ts2
-rw-r--r--src/domains/readpage/read-page.ts42
-rw-r--r--src/domains/search/web-search.ts34
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;
- }
-};