72 lines
2.3 KiB
TypeScript
72 lines
2.3 KiB
TypeScript
import type {
|
|
QqbotBangDreamCommandInput,
|
|
QqbotBangDreamCommandOutput,
|
|
QqbotBangDreamOperationKey,
|
|
} from '@/qqbot/plugins/bangDream/qqbot-bangdream.types';
|
|
import type {
|
|
QqbotBangDreamOperationHandlerName,
|
|
TsuguOperationDefinition,
|
|
} from '@/qqbot/plugins/bangDream/registry/operation-registry';
|
|
import {
|
|
createTsuguHookContext,
|
|
type TsuguHookRegistry,
|
|
} from '@/qqbot/plugins/bangDream/hook/hook-registry';
|
|
|
|
export type TsuguOperationPipelineOptions = {
|
|
executeHandler: (
|
|
handlerName: QqbotBangDreamOperationHandlerName,
|
|
input: QqbotBangDreamCommandInput,
|
|
) => Promise<QqbotBangDreamCommandOutput>;
|
|
hookRegistry: TsuguHookRegistry;
|
|
normalizeError: (error: unknown) => string;
|
|
resolveOperation: (
|
|
operationKey: QqbotBangDreamOperationKey,
|
|
) => TsuguOperationDefinition | undefined;
|
|
waitForReady: () => Promise<void>;
|
|
};
|
|
|
|
/**
|
|
* 串联 Tsugu 命令执行阶段：准备主数据、解析 operation、调用渲染 handler、输出 hook。
|
|
*/
|
|
export class TsuguOperationPipeline {
|
|
constructor(private readonly options: TsuguOperationPipelineOptions) {}
|
|
|
|
async run(
|
|
operationKey: QqbotBangDreamOperationKey,
|
|
input: QqbotBangDreamCommandInput,
|
|
) {
|
|
const context = createTsuguHookContext(operationKey, input);
|
|
await this.options.hookRegistry.beforeParse(context);
|
|
|
|
try {
|
|
context.stage = 'mainData';
|
|
await this.options.waitForReady();
|
|
|
|
context.stage = 'operation';
|
|
const operation = this.options.resolveOperation(operationKey);
|
|
if (!operation) {
|
|
throw new Error(`BangDream 插件能力不存在：${operationKey}`);
|
|
}
|
|
context.handlerName = operation.handlerName;
|
|
await this.options.hookRegistry.afterResolve(context);
|
|
|
|
context.stage = 'handler';
|
|
await this.options.hookRegistry.beforeRender(context);
|
|
const output = await this.options.executeHandler(
|
|
operation.handlerName,
|
|
input,
|
|
);
|
|
|
|
context.stage = 'output';
|
|
context.imageCount = output.imageCount;
|
|
context.query = output.query || context.query;
|
|
await this.options.hookRegistry.afterOutput(context);
|
|
return output;
|
|
} catch (error) {
|
|
const message = this.options.normalizeError(error);
|
|
await this.options.hookRegistry.onError(context, message);
|
|
throw new Error(message);
|
|
}
|
|
}
|
|
}
|