Carlos Florêncio
86 lines
3.2 KiB
TypeScript
86 lines
3.2 KiB
TypeScript
// Executor types match models.ExecutorType in apps/backend/internal/task/models/models.go.
|
|
|
|
export type CleanupSummary = {
|
|
lines: string[];
|
|
};
|
|
|
|
// mock_remote is test-only — intentionally absent so it falls through to GENERIC_LINE.
|
|
type KnownExecutor = "local" | "worktree" | "local_docker" | "remote_docker" | "sprites" | "ssh";
|
|
|
|
const SINGLE: Record<KnownExecutor, string> = {
|
|
local: "The agent ran directly in your repo - your files, branch, and folder are not touched.",
|
|
worktree:
|
|
"The task's git worktree and its branch will be deleted. Your main repo and other branches are not affected.",
|
|
local_docker:
|
|
"The Docker container running this task will be stopped and removed. Your host repo is not touched.",
|
|
remote_docker: "The remote Docker container running this task will be stopped and removed.",
|
|
sprites:
|
|
"The Sprites cloud sandbox for this task will be destroyed. Any uncommitted work inside it will be lost.",
|
|
ssh: "The task directory on the remote host will be removed (best-effort). Your local repo is not touched.",
|
|
};
|
|
|
|
const GENERIC_LINE = "Any running agent sessions will be stopped.";
|
|
|
|
function normalize(executorType: string | null | undefined): KnownExecutor | null {
|
|
if (!executorType) return null;
|
|
const key = executorType.toLowerCase();
|
|
if (key in SINGLE) return key as KnownExecutor;
|
|
return null;
|
|
}
|
|
|
|
/** Single-task variant. */
|
|
export function getCleanupSummary(executorType: string | null | undefined): CleanupSummary {
|
|
const known = normalize(executorType);
|
|
if (!known) return { lines: [GENERIC_LINE] };
|
|
return { lines: [SINGLE[known], GENERIC_LINE] };
|
|
}
|
|
|
|
const PLURAL: Partial<Record<KnownExecutor, (n: number) => string>> = {
|
|
local: (n) => `${n} local ${pl(n, "task", "tasks")} - your repo folders won't be touched.`,
|
|
worktree: (n) =>
|
|
`${n} ${pl(n, "worktree", "worktrees")} and ${pl(n, "its branch", "their branches")} will be deleted.`,
|
|
local_docker: (n) => `${n} Docker ${pl(n, "container", "containers")} will be removed.`,
|
|
remote_docker: (n) => `${n} remote Docker ${pl(n, "container", "containers")} will be removed.`,
|
|
sprites: (n) =>
|
|
`${n} Sprites ${pl(n, "sandbox", "sandboxes")} will be destroyed (uncommitted work lost).`,
|
|
ssh: (n) =>
|
|
`${n} remote task ${pl(n, "directory", "directories")} will be removed (best-effort).`,
|
|
};
|
|
|
|
function pl(n: number, one: string, many: string): string {
|
|
return n === 1 ? one : many;
|
|
}
|
|
|
|
/** Bulk variant — groups N tasks by executor type and emits one line per group. */
|
|
export function getBulkCleanupSummary(
|
|
executorTypes: Array<string | null | undefined>,
|
|
): CleanupSummary {
|
|
if (executorTypes.length === 0) return { lines: [GENERIC_LINE] };
|
|
|
|
const counts = new Map<KnownExecutor, number>();
|
|
for (const t of executorTypes) {
|
|
const known = normalize(t);
|
|
if (!known) continue;
|
|
counts.set(known, (counts.get(known) ?? 0) + 1);
|
|
}
|
|
|
|
const order: KnownExecutor[] = [
|
|
"worktree",
|
|
"local_docker",
|
|
"remote_docker",
|
|
"sprites",
|
|
"ssh",
|
|
"local",
|
|
];
|
|
|
|
const lines: string[] = [];
|
|
for (const key of order) {
|
|
const count = counts.get(key);
|
|
if (!count) continue;
|
|
const fmt = PLURAL[key];
|
|
if (fmt) lines.push(fmt(count));
|
|
}
|
|
lines.push(GENERIC_LINE);
|
|
return { lines };
|
|
}
|