agent-ecosystem/src/main/services/team/runtimeTeammateMode.ts
2026-04-14 20:07:57 +03:00

79 lines
2.1 KiB
TypeScript

import { isTmuxRuntimeReadyForCurrentPlatform } from '@features/tmux-installer/main';
import { parseCliArgs } from '@shared/utils/cliArgsParser';
interface DesktopTeammateModeDecision {
injectedTeammateMode: 'tmux' | null;
forceProcessTeammates: boolean;
}
let tmuxAvailablePromise: Promise<boolean> | null = null;
function getExplicitTeammateMode(
rawExtraCliArgs: string | undefined
): 'auto' | 'tmux' | 'in-process' | null {
const tokens = parseCliArgs(rawExtraCliArgs);
for (let i = 0; i < tokens.length; i += 1) {
const token = tokens[i];
// eslint-disable-next-line security/detect-possible-timing-attacks -- parsing user-supplied CLI flags, not comparing secrets
if (token === '--teammate-mode') {
const next = tokens[i + 1];
if (next === 'auto' || next === 'tmux' || next === 'in-process') {
return next;
}
return null;
}
if (token.startsWith('--teammate-mode=')) {
const value = token.slice('--teammate-mode='.length);
if (value === 'auto' || value === 'tmux' || value === 'in-process') {
return value;
}
return null;
}
}
return null;
}
async function isTmuxAvailable(): Promise<boolean> {
if (!tmuxAvailablePromise) {
tmuxAvailablePromise = isTmuxRuntimeReadyForCurrentPlatform()
.then((value) => value)
.catch(() => false)
.finally(() => {
tmuxAvailablePromise = null;
});
}
return tmuxAvailablePromise;
}
export async function resolveDesktopTeammateModeDecision(
rawExtraCliArgs: string | undefined
): Promise<DesktopTeammateModeDecision> {
const explicitMode = getExplicitTeammateMode(rawExtraCliArgs);
if (explicitMode === 'tmux') {
return {
injectedTeammateMode: null,
forceProcessTeammates: true,
};
}
if (explicitMode === 'auto' || explicitMode === 'in-process') {
return {
injectedTeammateMode: null,
forceProcessTeammates: false,
};
}
if (!(await isTmuxAvailable())) {
return {
injectedTeammateMode: null,
forceProcessTeammates: false,
};
}
return {
injectedTeammateMode: 'tmux',
forceProcessTeammates: true,
};
}