70 lines
2.7 KiB
TypeScript
70 lines
2.7 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';
|
|
|
|
export interface GSDToolsRuntime {
|
|
registry: ReturnType<typeof createRegistry>;
|
|
executionPolicy: QueryExecutionPolicy;
|
|
nativeHotpathAdapter: QueryNativeHotpathAdapter;
|
|
}
|
|
|
|
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>;
|
|
}): 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,
|
|
);
|
|
|
|
return { registry, executionPolicy, nativeHotpathAdapter };
|
|
}
|