refactor(errors): route fatal app errors

This commit is contained in:
ZacharyZhang-NY committed 2026-04-24 03:31:39 -04:00
1 parent ead9475b71
commit 38e2807f6c
6 files changed
+84 -27

No files matched your search

+1 -21
View File
@@ -78,26 +78,6 @@ export function createBrowserActionService({
});
}
/**
* Returns the approval record for `approvalId`, or null if the record is
* missing. Isolates the "approval row disappeared" case so the caller does
* not need to nest a try/catch around the state machine.
*/
function getApprovalOrNull(approvalId: string): ReturnType<typeof db.core.approvals.getApprovalRecord> | null {
try {
return db.core.approvals.getApprovalRecord(approvalId);
} catch (err) {
const classified = classifyUnknownError(err, {
kind: "recoverable",
code: "browser_action.approval.lookup_failed",
message: `getApprovalRecord failed for ${approvalId}`,
context: { approvalId },
});
log.warn(`[${classified.code}]`, classified.message, classified.context);
return null;
}
}
async function dispatchBrowserAction(actionId: string, companyId: string) {
const action = db.core.social.getBrowserAction(actionId, companyId);
if (!action) {
@@ -107,7 +87,7 @@ export function createBrowserActionService({
throw new Error(`Browser action ${actionId} is ${action.status} and cannot be dispatched.`);
}
if (action.approvalId) {
const approval = getApprovalOrNull(action.approvalId);
const approval = db.core.approvals.getApprovalRecord(action.approvalId);
if (!approval || approval.companyId !== companyId) {
throw new Error(`Browser action approval ${action.approvalId} does not belong to company ${companyId}.`);
}
+18 -4
View File
@@ -1,19 +1,33 @@
import { getOAuthProvider } from "@mariozechner/pi-ai/oauth";
import { getEnvApiKey } from "@mariozechner/pi-ai";
import { getConnectorDefinition, isPiKnownProvider } from "@main/connectors";
import { UpstreamError } from "../errors/index.js";
import { CredentialStore } from "./credential-store";
import type { OAuthCredentialPayload, StoredCredentialPayload } from "./types";
const REFRESH_WINDOW_MS = 60_000;
export class CredentialError extends Error {
export type CredentialErrorCode =
| "CREDENTIAL_MISSING"
| "CREDENTIAL_EXPIRED"
| "CREDENTIAL_INVALID"
| "CREDENTIAL_REFRESH_FAILED";
export class CredentialError extends UpstreamError {
readonly connectorId: string;
constructor(
public readonly code: "CREDENTIAL_MISSING" | "CREDENTIAL_EXPIRED" | "CREDENTIAL_INVALID" | "CREDENTIAL_REFRESH_FAILED",
public readonly connectorId: string,
code: CredentialErrorCode,
connectorId: string,
message?: string,
) {
super(message ?? code);
super({
code,
message: message ?? code,
context: { connectorId },
});
this.name = "CredentialError";
this.connectorId = connectorId;
}
}
+19
View File
@@ -13,6 +13,22 @@ export interface Logger {
warn?: (...args: unknown[]) => void;
}
type FatalErrorRoute = (error: AppError) => never;
const defaultFatalErrorRoute: FatalErrorRoute = (error) => {
if (process.env.VITEST === "true" || process.env.NODE_ENV === "test") {
throw error;
}
process.stderr.write(`[fatal:${error.code}] ${error.message}\n`);
process.abort();
};
let fatalErrorRoute: FatalErrorRoute = defaultFatalErrorRoute;
export function setFatalErrorRouteForTesting(route: FatalErrorRoute | null): void {
fatalErrorRoute = route ?? defaultFatalErrorRoute;
}
/**
* Classify an unknown thrown value and emit a single structured log line.
* Returns the `AppError` so the caller can still branch on `err.kind`
@@ -46,5 +62,8 @@ export function logAppError(
} else if (typeof logger.warn === "function") {
logger.warn(...parts);
}
if (err.kind === "fatal") {
fatalErrorRoute(err);
}
return err;
}
+1 -1
View File
@@ -18,4 +18,4 @@ export {
type ErrorKind,
} from "./taxonomy.js";
export { logAppError, type Logger } from "./handlers.js";
export { logAppError, setFatalErrorRouteForTesting, type Logger } from "./handlers.js";
+30
View File
@@ -0,0 +1,30 @@
import { afterEach, describe, expect, it, vi } from "vitest";
import { logAppError, setFatalErrorRouteForTesting } from "../../src/main/errors";
import type { AppError } from "../../src/main/errors";
describe("AppError fatal route", () => {
afterEach(() => {
setFatalErrorRouteForTesting(null);
});
it("routes fatal errors through the process-level handler", () => {
const logger = { error: vi.fn(), warn: vi.fn() };
const routed: AppError[] = [];
const sentinel = new Error("fatal route called");
setFatalErrorRouteForTesting((error) => {
routed.push(error);
throw sentinel;
});
expect(() => logAppError(logger, new Error("sqlite corrupt"), {
kind: "fatal",
code: "db.sqlite.corrupt",
message: "SQLite profile is corrupt",
context: { profile: "test" },
})).toThrow(sentinel);
expect(routed[0]?.kind).toBe("fatal");
expect(routed[0]?.code).toBe("db.sqlite.corrupt");
expect(logger.error).toHaveBeenCalled();
});
});
+15 -1
View File
@@ -5,6 +5,7 @@ import { afterEach, beforeAll, beforeEach, describe, expect, it } from "vitest";
import { AppDatabase } from "../../src/main/database";
import { CredentialStore } from "../../src/main/credentials/credential-store";
import { CredentialResolver, CredentialError } from "../../src/main/credentials/credential-resolver";
import { AppError } from "../../src/main/errors";
beforeAll(() => {
process.env.AGENTCOMPANY_ALLOW_INSECURE_VAULT = "true";
@@ -56,7 +57,20 @@ describe("CredentialResolver", () => {
it("throws CREDENTIAL_MISSING when the env-backed key is unset", async () => {
delete process.env.XAI_API_KEY;
store.save("xai", { kind: "api_key", value: { source: "env", value: "XAI_API_KEY" } });
await expect(resolver.resolve("xai")).rejects.toBeInstanceOf(CredentialError);
let thrown: unknown;
try {
await resolver.resolve("xai");
} catch (error) {
thrown = error;
}
expect(thrown).toBeInstanceOf(CredentialError);
expect(thrown).toBeInstanceOf(AppError);
expect(thrown).toMatchObject({
kind: "upstream",
code: "CREDENTIAL_MISSING",
connectorId: "xai",
context: { connectorId: "xai" },
});
});
it("assembles AWS extraEnv for IAM auth mode", async () => {