Function makeActiveWorkDeadline

packages/plugins/mcp/src/sdk/invoke.ts:63–119

(
  timeoutMs: number = MCP_ACTIVE_WORK_TIMEOUT_MS,
)

};

export const makeActiveWorkDeadline = (
  timeoutMs: number = MCP_ACTIVE_WORK_TIMEOUT_MS,
): ActiveWorkDeadline => {
  const controller = new AbortController();
  let remainingMs = timeoutMs;
  let pendingElicitations = 0;
  let startedAt: number | undefined;
  let timer: ReturnType<typeof setTimeout> | undefined;

  const stopTimer = (): void => {
    if (timer === undefined || startedAt === undefined) return;
    clearTimeout(timer);
    timer = undefined;
    remainingMs = Math.max(0, remainingMs - (Date.now() - startedAt));
    startedAt = undefined;
  };

  const abortForTimeout = (): void => {
    timer = undefined;
    startedAt = undefined;
    // oxlint-disable-next-line executor/no-error-constructor -- boundary: AbortSignal consumers need a stable timeout reason
    controller.abort(new Error("MCP tool invocation exceeded its active-work deadline"));
  };

  const startTimer = (): void => {
    if (controller.signal.aborted || pendingElicitations > 0) return;
    if (remainingMs <= 0) {
      abortForTimeout();
      return;
    }
    startedAt = Date.now();
    timer = setTimeout(() => {
      remainingMs = 0;
      abortForTimeout();
    }, remainingMs);
  };

  startTimer();

  return {
    signal: controller.signal,
    pause: () => {
      pendingElicitations += 1;
      if (pendingElicitations === 1) stopTimer();
    },
    resume: () => {
      if (pendingElicitations === 0) return;
      pendingElicitations -= 1;
      if (pendingElicitations === 0) startTimer();
    },
    dispose: () => {
      stopTimer();
      // oxlint-disable-next-line executor/no-error-constructor -- boundary: disposing the scoped signal must interrupt SDK work
      controller.abort(new Error("MCP tool invocation was disposed"));
    },
  };
};