Class MiddlewarePipeline

ai/ai.ts:2358–2386

class MiddlewarePipeline {
  private middlewares: Middleware[] = [];

  use(middleware: Middleware): void {
    this.middlewares.push(middleware);
  }

  async execute<T>(
    input: T,
    finalHandler: (input: T, token?: AbortToken) => Promise<any>,
    token?: AbortToken,
  ): Promise<any> {
    const exec = async (
      idx: number,
      curInput: T,
      curToken?: AbortToken,
    ): Promise<any> => {
      if (idx >= this.middlewares.length)
        return await finalHandler(curInput, curToken);
      const mw = this.middlewares[idx];
      return await mw.process(
        curInput,
        (nextInput, nextToken) => exec(idx + 1, nextInput, nextToken),
        curToken,
      );
    };
    return await exec(0, input, token);
  }
}

class TimeoutMiddleware implements Middleware {
  private configManagerPromise: Promise<ConfigManager>;