core2026/tools/load-tests/src/trpc.ts

import type { LoadOperation } from './config.js';
import type { PhaseMetrics } from './metrics.js';

export interface TrpcRequest {
    url: string;
    init: RequestInit;
}

export const buildTrpcQuery = (
    baseUrl: string,
    trpcPath: string,
    operation: LoadOperation,
    token: string
): TrpcRequest => {
    const normalizedPath = trpcPath.endsWith('/') ? trpcPath.slice(0, -1) : trpcPath;
    const url = new URL(`${normalizedPath}/${operation.procedure}`, baseUrl);
    if (operation.input !== undefined) url.searchParams.set('input', JSON.stringify(operation.input));
    return {
        url: url.toString(),
        init: {
            method: 'GET',
            headers: { accept: 'application/json', authorization: `Bearer ${token}` },
        },
    };
};

const classifyTrpcPayload = (payload: unknown): string | null => {
    if (typeof payload !== 'object' || payload === null) return 'invalid-payload';
    if ('error' in payload) {
        const error = (payload as { error?: unknown }).error;
        if (typeof error === 'object' && error !== null && 'data' in error) {
            const data = (error as { data?: unknown }).data;
            if (
                typeof data === 'object' &&
                data !== null &&
                'code' in data &&
                typeof (data as { code?: unknown }).code === 'string'
            ) {
                const code = (data as { code: string }).code;
                return `trpc-${/^[A-Z_]+$/u.test(code) ? code.toLowerCase() : 'error'}`;
            }
        }
        return 'trpc-error';
    }
    return null;
};

const asRecord = (value: unknown): Record<string, unknown> | null =>
    typeof value === 'object' && value !== null && !Array.isArray(value) ? (value as Record<string, unknown>) : null;

const unwrapTrpcData = (payload: unknown): unknown => {
    const result = asRecord(asRecord(payload)?.result);
    const data = asRecord(result?.data);
    return data && 'json' in data ? data.json : result?.data;
};

export interface DashboardRevisions {
    context?: string;
    commandTable?: string;
    boardAccess?: string;
}

export interface DashboardObservation {
    revisions: DashboardRevisions;
    sourceRevisions: DashboardRevisions;
    resultKinds: string[];
    resultKindsBySlice: Partial<Record<keyof DashboardRevisions, string>>;
}

export const extractDashboardRevisions = (payload: unknown): DashboardObservation | null => {
    const data = asRecord(unwrapTrpcData(payload));
    if (!data) return null;
    const revisions: DashboardRevisions = {};
    const sourceRevisions: DashboardRevisions = {};
    const resultKinds: string[] = [];
    const resultKindsBySlice: DashboardObservation['resultKindsBySlice'] = {};
    for (const [wireName, outputName] of [
        ['context', 'context'],
        ['commandTable', 'commandTable'],
        ['boardAccess', 'boardAccess'],
    ] as const) {
        const slice = asRecord(data[wireName]);
        if (!slice) continue;
        if (typeof slice.revision === 'string' && /^[A-Za-z0-9_-]{22}$/u.test(slice.revision))
            revisions[outputName] = slice.revision;
        if (typeof slice.sourceRevision === 'string' && /^[A-Za-z0-9_-]{22}$/u.test(slice.sourceRevision)) {
            sourceRevisions[outputName] = slice.sourceRevision;
        }
        const resultKind =
            typeof slice.kind === 'string' && ['unchanged', 'snapshot', 'patch'].includes(slice.kind)
                ? slice.kind
                : 'other';
        resultKinds.push(resultKind);
        resultKindsBySlice[outputName] = resultKind;
    }
    return { revisions, sourceRevisions, resultKinds, resultKindsBySlice };
};

export const executeTrpcQuery = async (options: {
    baseUrl: string;
    trpcPath: string;
    operation: LoadOperation;
    token: string;
    signal: AbortSignal;
    metrics: PhaseMetrics;
}): Promise<DashboardObservation | undefined> => {
    const started = performance.now();
    let outcome: string | null;
    try {
        const request = buildTrpcQuery(options.baseUrl, options.trpcPath, options.operation, options.token);
        const response = await fetch(request.url, { ...request.init, signal: options.signal });
        if (!response.ok) {
            outcome = `http-${response.status}`;
            await response.body?.cancel();
        } else {
            const payload: unknown = await response.json();
            outcome = classifyTrpcPayload(payload);
            if (outcome === null && options.operation.procedure === 'dashboard.getContextBundleDelta') {
                const dashboard = extractDashboardRevisions(payload);
                for (const kind of dashboard?.resultKinds ?? [])
                    options.metrics.recordHttpResult(options.operation.name, kind);
                const input = asRecord(options.operation.input);
                const knownSource = asRecord(input?.knownSource);
                for (const slice of ['context', 'commandTable', 'boardAccess'] as const) {
                    const sourceRevision = dashboard?.sourceRevisions[slice];
                    if (sourceRevision !== undefined) options.metrics.httpSourceRevisionObserved += 1;
                    if (typeof knownSource?.[slice] === 'string') options.metrics.httpSourceRevisionKnownSent += 1;
                    if (
                        dashboard?.resultKindsBySlice[slice] === 'unchanged' &&
                        sourceRevision !== undefined &&
                        knownSource?.[slice] === sourceRevision
                    ) {
                        options.metrics.httpSourceRevisionMatchedUnchanged += 1;
                    }
                }
                options.metrics.recordHttp(options.operation.name, performance.now() - started, outcome);
                return dashboard ?? undefined;
            }
        }
    } catch (error) {
        if (options.signal.aborted) return;
        outcome = error instanceof TypeError ? 'network' : 'client';
    }
    options.metrics.recordHttp(options.operation.name, performance.now() - started, outcome);
    return undefined;
};