fix(wrangler): prevent SQLite users error from being reported to Sentry (#11467)

* fix(wrangler): prevent SQLite users error from being reported to Sentry

* add SQLITE_AUTH to common error codes

Author: edmundhung

.changeset/gold-donuts-leave.md

@@ -0,0 +1,5 @@
+---
+"wrangler": patch
+---
+
+fix: prevent reporting SQLite error from `wrangler d1 execute` to Sentry

packages/wrangler/src/__tests__/d1/execute.test.ts

@@ -1,5 +1,6 @@
 import fs from "node:fs";
 import { join } from "node:path";
+import { UserError } from "@cloudflare/workers-utils";
 import { http, HttpResponse } from "msw";
 import { afterEach, beforeEach, describe, expect, it } from "vitest";
 import { mockAccountId, mockApiToken } from "../helpers/mock-account-id";
@@ -161,6 +162,31 @@ describe("execute", () => {
 		`);
 	});
 
+	it("should treat SQLite constraint errors as UserErrors", async () => {
+		setIsTTY(false);
+		writeWranglerConfig({
+			d1_databases: [
+				{ binding: "DATABASE", database_name: "db", database_id: "xxxx" },
+			],
+		});
+
+		// First create a table with a foreign key constraint
+		const setupSQL = `
+			CREATE TABLE users (id INTEGER PRIMARY KEY);
+			CREATE TABLE posts (id INTEGER PRIMARY KEY, user_id INTEGER, FOREIGN KEY(user_id) REFERENCES users(id));
+		`;
+		fs.writeFileSync("setup.sql", setupSQL);
+		await runWrangler("d1 execute db --file setup.sql --local");
+
+		// Now try to violate the foreign key constraint
+		const violationSQL = `INSERT INTO posts (id, user_id) VALUES (1, 999);`;
+		fs.writeFileSync("violation.sql", violationSQL);
+
+		await expect(
+			runWrangler("d1 execute db --file violation.sql --local")
+		).rejects.toThrow(UserError);
+	});
+
 	it("should show banner by default", async () => {
 		setIsTTY(false);
 		writeWranglerConfig({

packages/wrangler/src/d1/execute.ts

@@ -41,6 +41,31 @@ export type QueryResult = {
 	query?: string;
 };
 
+// Common SQLite Codes
+// See https://www.sqlite.org/rescode.html
+const SQLITE_RESULT_CODES = [
+	"SQLITE_ERROR",
+	"SQLITE_CONSTRAINT",
+	"SQLITE_MISMATCH",
+	"SQLITE_AUTH",
+];
+
+function isSqliteUserError(error: unknown): error is Error {
+	if (!(error instanceof Error)) {
+		return false;
+	}
+
+	const message = error.message.toUpperCase();
+
+	for (const code of SQLITE_RESULT_CODES) {
+		if (message.includes(code)) {
+			return true;
+		}
+	}
+
+	return false;
+}
+
 export const d1ExecuteCommand = createCommand({
 	metadata: {
 		description: "Execute a command or SQL file",
@@ -313,7 +338,13 @@ async function executeLocally({
 	try {
 		results = await db.batch(queries.map((query) => db.prepare(query)));
 	} catch (e: unknown) {
-		throw (e as { cause?: unknown })?.cause ?? e;
+		const cause = (e as { cause?: unknown })?.cause ?? e;
+
+		if (isSqliteUserError(cause)) {
+			throw new UserError(cause.message);
+		}
+
+		throw cause;
 	} finally {
 		await mf.dispose();
 	}