84 lines
3.1 KiB
TypeScript
84 lines
3.1 KiB
TypeScript
import type { GSDEventStream } from './event-stream.js';
|
|
import { createRegistry } from './query/index.js';
|
|
import { GSDTransport } from './gsd-transport.js';
|
|
import { QueryExecutionPolicy } from './query-execution-policy.js';
|
|
import { QuerySubprocessAdapter } from './query-subprocess-adapter.js';
|
|
import { QueryNativeDirectAdapter } from './query-native-direct-adapter.js';
|
|
import { QueryNativeHotpathAdapter } from './query-native-hotpath-adapter.js';
|
|
import { formatQueryRawOutput } from './query-raw-output-projection.js';
|
|
import { createQueryNativeErrorFactory, createQueryToolsErrorFactory } from './query-tools-error-factory.js';
|
|
import { QueryRuntimeBridge, type RuntimeBridgeOptions } from './query-runtime-bridge.js';
|
|
|
|
export interface GSDToolsRuntime {
|
|
bridge: QueryRuntimeBridge;
|
|
}
|
|
|
|
export function createGSDToolsRuntime(opts: {
|
|
projectDir: string;
|
|
gsdToolsPath: string;
|
|
timeoutMs: number;
|
|
workstream?: string;
|
|
eventStream?: GSDEventStream;
|
|
sessionId?: string;
|
|
shouldUseNativeQuery: () => boolean;
|
|
execJsonFallback: (legacyCommand: string, legacyArgs: string[]) => Promise<unknown>;
|
|
execRawFallback: (legacyCommand: string, legacyArgs: string[]) => Promise<string>;
|
|
strictSdk?: boolean;
|
|
allowFallbackToSubprocess?: boolean;
|
|
onDispatchEvent?: RuntimeBridgeOptions['onDispatchEvent'];
|
|
}): GSDToolsRuntime {
|
|
const registry = createRegistry(opts.eventStream, opts.sessionId);
|
|
|
|
const queryToolsErrorFactory = createQueryToolsErrorFactory();
|
|
|
|
const subprocessAdapter = new QuerySubprocessAdapter({
|
|
projectDir: opts.projectDir,
|
|
gsdToolsPath: opts.gsdToolsPath,
|
|
timeoutMs: opts.timeoutMs,
|
|
workstream: opts.workstream,
|
|
...queryToolsErrorFactory,
|
|
});
|
|
|
|
const nativeErrorFactory = createQueryNativeErrorFactory(opts.timeoutMs);
|
|
|
|
const nativeDirectAdapter = new QueryNativeDirectAdapter({
|
|
timeoutMs: opts.timeoutMs,
|
|
dispatch: (registryCommand, registryArgs) => registry.dispatch(registryCommand, registryArgs, opts.projectDir),
|
|
...nativeErrorFactory,
|
|
});
|
|
|
|
const transport = new GSDTransport(registry, {
|
|
dispatchNative: (request) => nativeDirectAdapter.dispatchResult(
|
|
request.legacyCommand,
|
|
request.legacyArgs,
|
|
request.registryCommand,
|
|
request.registryArgs,
|
|
),
|
|
execSubprocessJson: (legacyCommand, legacyArgs) => subprocessAdapter.execJson(legacyCommand, legacyArgs),
|
|
execSubprocessRaw: (legacyCommand, legacyArgs) => subprocessAdapter.execRaw(legacyCommand, legacyArgs),
|
|
formatNativeRaw: (registryCommand, data) => formatQueryRawOutput(registryCommand, data),
|
|
});
|
|
|
|
const executionPolicy = new QueryExecutionPolicy(transport);
|
|
const nativeHotpathAdapter = new QueryNativeHotpathAdapter(
|
|
opts.shouldUseNativeQuery,
|
|
nativeDirectAdapter,
|
|
opts.execJsonFallback,
|
|
opts.execRawFallback,
|
|
);
|
|
|
|
const bridge = new QueryRuntimeBridge(
|
|
registry,
|
|
executionPolicy,
|
|
nativeHotpathAdapter,
|
|
opts.shouldUseNativeQuery,
|
|
{
|
|
strictSdk: opts.strictSdk,
|
|
allowFallbackToSubprocess: opts.allowFallbackToSubprocess,
|
|
onDispatchEvent: opts.onDispatchEvent,
|
|
},
|
|
);
|
|
|
|
return { bridge };
|
|
}
|