134 lines
4.8 KiB
TypeScript
134 lines
4.8 KiB
TypeScript
import { z } from "zod";
|
|
|
|
import {
|
|
parseEnvironment,
|
|
parseEnvironmentValues,
|
|
} from "../config/environment.js";
|
|
import { ConfigurationError } from "../domain/configurationErrors.js";
|
|
import { err, ok, type Result } from "../domain/result.js";
|
|
import {
|
|
effectiveClientServer,
|
|
type ClientConfigurationDocument,
|
|
} from "./ClientConfigurationDocument.js";
|
|
import { PRODUCT_IDENTITY } from "../identity.js";
|
|
import { compareUnicodeCodePoints } from "../domain/unicodeCodePointOrder.js";
|
|
import { dshDocument } from "./DshCordisPatch.js";
|
|
import type { McpToolTimeoutInputs } from "../mcpStartupPolicy.js";
|
|
import type { AnalysisProviderSelector } from "../contracts/providerSelection.js";
|
|
|
|
export type ProviderRequestBudgetResolver = (
|
|
analysisProvider: AnalysisProviderSelector,
|
|
environment: Readonly<Record<string, string>>,
|
|
) => Result<number | undefined, ConfigurationError>;
|
|
|
|
export interface ClientRegistrationEnvironmentDetails {
|
|
readonly values: Readonly<Record<string, string>>;
|
|
readonly timeoutInputs: McpToolTimeoutInputs;
|
|
}
|
|
|
|
/** Read explicit REA overrides without dropping peers when one value is invalid. */
|
|
export const clientRegistrationEnvironment = (
|
|
parsed: ClientConfigurationDocument | undefined,
|
|
overrides: Readonly<Record<string, string>> = {},
|
|
validation: "fragment" | "effective" = "effective",
|
|
): Result<Readonly<Record<string, string>>, ConfigurationError> => {
|
|
if (parsed?.dialect === "dsh")
|
|
return dshDocument(parsed.document).environment(overrides, validation);
|
|
const details = clientRegistrationEnvironmentDetails(
|
|
parsed,
|
|
overrides,
|
|
validation,
|
|
);
|
|
return details.ok ? ok(details.value.values) : details;
|
|
};
|
|
|
|
/** Retain original strings and the parser-owned timeout facts. */
|
|
export const clientRegistrationEnvironmentDetails = (
|
|
parsed: ClientConfigurationDocument | undefined,
|
|
overrides: Readonly<Record<string, string>> = {},
|
|
validation: "fragment" | "effective" = "effective",
|
|
resolveProviderRequestBudget?: ProviderRequestBudgetResolver,
|
|
): Result<ClientRegistrationEnvironmentDetails, ConfigurationError> => {
|
|
if (parsed?.dialect === "dsh") {
|
|
const environment = dshDocument(parsed.document).environment(
|
|
overrides,
|
|
validation,
|
|
);
|
|
if (!environment.ok) return environment;
|
|
return validatedDetails(
|
|
environment.value,
|
|
validation,
|
|
resolveProviderRequestBudget,
|
|
);
|
|
}
|
|
const key =
|
|
parsed?.dialect === "opencode" || parsed?.dialect === "opencode_v2"
|
|
? "environment"
|
|
: "env";
|
|
const registration =
|
|
parsed === undefined
|
|
? undefined
|
|
: effectiveClientServer(parsed, PRODUCT_IDENTITY.mcpServerKey);
|
|
const raw =
|
|
typeof registration === "object" && registration !== null
|
|
? Reflect.get(registration, key)
|
|
: undefined;
|
|
const existing = z
|
|
.record(z.string(), z.string())
|
|
.safeParse(raw === undefined ? {} : raw);
|
|
if (!existing.success)
|
|
return err(
|
|
new ConfigurationError("Invalid REA server environment", {
|
|
settings: existing.error.issues.map((issue) => ({
|
|
setting: [key, ...issue.path].join("."),
|
|
constraint:
|
|
"Server environment must be an object containing only string values",
|
|
})),
|
|
}),
|
|
);
|
|
// Validate known settings with the runtime's parser, but retain the original
|
|
// string overrides, including unknown client-specific keys and no defaults.
|
|
const environment = { ...existing.data, ...overrides };
|
|
return validatedDetails(
|
|
environment,
|
|
validation,
|
|
resolveProviderRequestBudget,
|
|
);
|
|
};
|
|
|
|
const validatedDetails = (
|
|
environment: Readonly<Record<string, string>>,
|
|
validation: "fragment" | "effective",
|
|
resolveProviderRequestBudget?: ProviderRequestBudgetResolver,
|
|
): Result<ClientRegistrationEnvironmentDetails, ConfigurationError> => {
|
|
const validated =
|
|
validation === "fragment"
|
|
? parseEnvironmentValues(environment)
|
|
: parseEnvironment(environment);
|
|
if (!validated.ok) return validated;
|
|
const providerRequestBudget =
|
|
resolveProviderRequestBudget?.(
|
|
validated.value.REA_ANALYSIS_PROVIDER,
|
|
environment,
|
|
) ?? ok(undefined);
|
|
if (!providerRequestBudget.ok) return providerRequestBudget;
|
|
return ok({
|
|
values: Object.fromEntries(
|
|
Object.entries(environment).sort(([left], [right]) =>
|
|
compareUnicodeCodePoints(left, right),
|
|
),
|
|
),
|
|
timeoutInputs: {
|
|
analysisProvider: validated.value.REA_ANALYSIS_PROVIDER,
|
|
ghidraStartupTimeoutMs: validated.value.REA_GHIDRA_STARTUP_TIMEOUT_MS,
|
|
hopperStartupTimeoutMs: validated.value.REA_HOPPER_STARTUP_TIMEOUT_MS,
|
|
...(validated.value.REA_JEB_MCP_TIMEOUT_MS === undefined
|
|
? {}
|
|
: { jebMcpTimeoutMs: validated.value.REA_JEB_MCP_TIMEOUT_MS }),
|
|
...(providerRequestBudget.value === undefined
|
|
? {}
|
|
: { providerRequestTimeoutMs: providerRequestBudget.value }),
|
|
},
|
|
});
|
|
};
|