@tmustier/pi-agent-teams 0.1.2 → 0.2.0

package/extensions/teams/leader-messaging-commands.ts

@@ -0,0 +1,149 @@
+import type { ExtensionCommandContext } from "@mariozechner/pi-coding-agent";
+import { writeToMailbox } from "./mailbox.js";
+import { sanitizeName } from "./names.js";
+import { getTeamDir } from "./paths.js";
+import { TEAM_MAILBOX_NS } from "./protocol.js";
+import { ensureTeamConfig } from "./team-config.js";
+import type { TeamTask } from "./task-store.js";
+import type { TeammateRpc } from "./teammate-rpc.js";
+import type { TeamsStyle } from "./teams-style.js";
+import { formatMemberDisplayName, getTeamsStrings } from "./teams-style.js";
+
+export async function handleTeamSendCommand(opts: {
+	ctx: ExtensionCommandContext;
+	rest: string[];
+	teammates: Map<string, TeammateRpc>;
+	style: TeamsStyle;
+	renderWidget: () => void;
+}): Promise<void> {
+	const { ctx, rest, teammates, style, renderWidget } = opts;
+	const strings = getTeamsStrings(style);
+
+	const nameRaw = rest[0];
+	const msg = rest.slice(1).join(" ").trim();
+	if (!nameRaw || !msg) {
+		ctx.ui.notify("Usage: /team send <name> <msg...>", "error");
+		return;
+	}
+	const name = sanitizeName(nameRaw);
+	const t = teammates.get(name);
+	if (!t) {
+		ctx.ui.notify(`Unknown ${strings.memberTitle.toLowerCase()}: ${name}`, "error");
+		return;
+	}
+	if (t.status === "streaming") await t.followUp(msg);
+	else await t.prompt(msg);
+	ctx.ui.notify(`Sent to ${name}`, "info");
+	renderWidget();
+}
+
+export async function handleTeamSteerCommand(opts: {
+	ctx: ExtensionCommandContext;
+	rest: string[];
+	teammates: Map<string, TeammateRpc>;
+	style: TeamsStyle;
+	renderWidget: () => void;
+}): Promise<void> {
+	const { ctx, rest, teammates, style, renderWidget } = opts;
+	const strings = getTeamsStrings(style);
+
+	const nameRaw = rest[0];
+	const msg = rest.slice(1).join(" ").trim();
+	if (!nameRaw || !msg) {
+		ctx.ui.notify("Usage: /team steer <name> <msg...>", "error");
+		return;
+	}
+	const name = sanitizeName(nameRaw);
+	const t = teammates.get(name);
+	if (!t) {
+		ctx.ui.notify(`Unknown ${strings.memberTitle.toLowerCase()}: ${name}`, "error");
+		return;
+	}
+	await t.steer(msg);
+	ctx.ui.notify(`Steering sent to ${name}`, "info");
+	renderWidget();
+}
+
+export async function handleTeamDmCommand(opts: {
+	ctx: ExtensionCommandContext;
+	rest: string[];
+	leadName: string;
+	style: TeamsStyle;
+}): Promise<void> {
+	const { ctx, rest, leadName, style } = opts;
+	const strings = getTeamsStrings(style);
+
+	const nameRaw = rest[0];
+	const msg = rest.slice(1).join(" ").trim();
+	if (!nameRaw || !msg) {
+		ctx.ui.notify("Usage: /team dm <name> <msg...>", "error");
+		return;
+	}
+	const name = sanitizeName(nameRaw);
+	const teamId = ctx.sessionManager.getSessionId();
+	await writeToMailbox(getTeamDir(teamId), TEAM_MAILBOX_NS, name, {
+		from: leadName,
+		text: msg,
+		timestamp: new Date().toISOString(),
+	});
+	ctx.ui.notify(`DM queued for ${formatMemberDisplayName(style, name)}`, "info");
+}
+
+export async function handleTeamBroadcastCommand(opts: {
+	ctx: ExtensionCommandContext;
+	rest: string[];
+	teammates: Map<string, TeammateRpc>;
+	leadName: string;
+	style: TeamsStyle;
+	refreshTasks: () => Promise<void>;
+	getTasks: () => TeamTask[];
+	getTaskListId: () => string | null;
+}): Promise<void> {
+	const { ctx, rest, teammates, leadName, style, refreshTasks, getTasks, getTaskListId } = opts;
+	const strings = getTeamsStrings(style);
+
+	const msg = rest.join(" ").trim();
+	if (!msg) {
+		ctx.ui.notify("Usage: /team broadcast <msg...>", "error");
+		return;
+	}
+
+	const teamId = ctx.sessionManager.getSessionId();
+	const teamDir = getTeamDir(teamId);
+	const taskListId = getTaskListId();
+	const cfg = await ensureTeamConfig(teamDir, { teamId, taskListId: taskListId ?? teamId, leadName, style });
+
+	const recipients = new Set<string>();
+	for (const m of cfg.members) {
+		if (m.role === "worker") recipients.add(m.name);
+	}
+	for (const name of teammates.keys()) recipients.add(name);
+
+	// Include task owners (helps reach manual tmux workers not tracked as RPC teammates).
+	await refreshTasks();
+	for (const t of getTasks()) {
+		if (t.owner && t.owner !== leadName) recipients.add(t.owner);
+	}
+
+	const names = Array.from(recipients).sort();
+	if (names.length === 0) {
+		ctx.ui.notify(`No ${strings.memberTitle.toLowerCase()}s to broadcast to`, "warning");
+		return;
+	}
+
+	const ts = new Date().toISOString();
+	await Promise.all(
+		names.map((name) =>
+			writeToMailbox(teamDir, TEAM_MAILBOX_NS, name, {
+				from: leadName,
+				text: msg,
+				timestamp: ts,
+			}),
+		),
+	);
+
+	ctx.ui.notify(
+		`Broadcast queued for ${names.length} ${strings.memberTitle.toLowerCase()}(s): ${names.map((n) => formatMemberDisplayName(style, n)).join(", ")}`,
+		"info",
+	);
+}

package/extensions/teams/leader-plan-commands.ts

@@ -0,0 +1,96 @@
+import type { ExtensionCommandContext } from "@mariozechner/pi-coding-agent";
+import { writeToMailbox } from "./mailbox.js";
+import { sanitizeName } from "./names.js";
+import { getTeamDir } from "./paths.js";
+import { TEAM_MAILBOX_NS } from "./protocol.js";
+import type { TeamsStyle } from "./teams-style.js";
+import { formatMemberDisplayName, getTeamsStrings } from "./teams-style.js";
+
+export async function handleTeamPlanCommand(opts: {
+	ctx: ExtensionCommandContext;
+	rest: string[];
+	leadName: string;
+	style: TeamsStyle;
+	pendingPlanApprovals: Map<string, { requestId: string; name: string; taskId?: string }>;
+}): Promise<void> {
+	const { ctx, rest, leadName, style, pendingPlanApprovals } = opts;
+
+	const [planSub, ...planRest] = rest;
+	if (!planSub || planSub === "help") {
+		ctx.ui.notify(
+			[
+				"Usage:",
+				"  /team plan approve <name>",
+				"  /team plan reject <name> [feedback...]",
+			].join("\n"),
+			"info",
+		);
+		return;
+	}
+
+	if (planSub === "approve") {
+		const nameRaw = planRest[0];
+		if (!nameRaw) {
+			ctx.ui.notify("Usage: /team plan approve <name>", "error");
+			return;
+		}
+		const name = sanitizeName(nameRaw);
+		const pending = pendingPlanApprovals.get(name);
+		if (!pending) {
+			ctx.ui.notify(`No pending plan approval for ${name}`, "error");
+			return;
+		}
+
+		const teamId = ctx.sessionManager.getSessionId();
+		const teamDir = getTeamDir(teamId);
+		const ts = new Date().toISOString();
+		await writeToMailbox(teamDir, TEAM_MAILBOX_NS, name, {
+			from: leadName,
+			text: JSON.stringify({
+				type: "plan_approved",
+				requestId: pending.requestId,
+				from: leadName,
+				timestamp: ts,
+			}),
+			timestamp: ts,
+		});
+		pendingPlanApprovals.delete(name);
+		ctx.ui.notify(`Approved plan for ${formatMemberDisplayName(style, name)}`, "info");
+		return;
+	}
+
+	if (planSub === "reject") {
+		const nameRaw = planRest[0];
+		if (!nameRaw) {
+			ctx.ui.notify("Usage: /team plan reject <name> [feedback...]", "error");
+			return;
+		}
+		const name = sanitizeName(nameRaw);
+		const pending = pendingPlanApprovals.get(name);
+		if (!pending) {
+			ctx.ui.notify(`No pending plan approval for ${name}`, "error");
+			return;
+		}
+
+		const feedback = planRest.slice(1).join(" ").trim() || "Plan rejected";
+		const teamId = ctx.sessionManager.getSessionId();
+		const teamDir = getTeamDir(teamId);
+		const ts = new Date().toISOString();
+		await writeToMailbox(teamDir, TEAM_MAILBOX_NS, name, {
+			from: leadName,
+			text: JSON.stringify({
+				type: "plan_rejected",
+				requestId: pending.requestId,
+				from: leadName,
+				feedback,
+				timestamp: ts,
+			}),
+			timestamp: ts,
+		});
+		pendingPlanApprovals.delete(name);
+		ctx.ui.notify(`Rejected plan for ${formatMemberDisplayName(style, name)}: ${feedback}`, "info");
+		return;
+	}
+
+	ctx.ui.notify(`Unknown plan subcommand: ${planSub}`, "error");
+}

package/extensions/teams/leader-spawn-command.ts

@@ -0,0 +1,73 @@
+import type { ExtensionCommandContext, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import { pickComradeNames } from "./names.js";
+import type { TeammateRpc } from "./teammate-rpc.js";
+import type { TeamsStyle } from "./teams-style.js";
+import { formatMemberDisplayName, getTeamsStrings, isSovietStyle } from "./teams-style.js";
+
+export type ContextMode = "fresh" | "branch";
+export type WorkspaceMode = "shared" | "worktree";
+
+export type SpawnTeammateResult =
+	| {
+			ok: true;
+			name: string;
+			mode: ContextMode;
+			workspaceMode: WorkspaceMode;
+			note?: string;
+			warnings: string[];
+	  }
+	| { ok: false; error: string };
+
+export async function handleTeamSpawnCommand(opts: {
+	ctx: ExtensionCommandContext;
+	rest: string[];
+	teammates: Map<string, TeammateRpc>;
+	style: TeamsStyle;
+	spawnTeammate: (
+		ctx: ExtensionContext,
+		opts: { name: string; mode?: ContextMode; workspaceMode?: WorkspaceMode; planRequired?: boolean },
+	) => Promise<SpawnTeammateResult>;
+}): Promise<void> {
+	const { ctx, rest, teammates, style, spawnTeammate } = opts;
+	const strings = getTeamsStrings(style);
+
+	// Parse flags from any position
+	let nameRaw: string | undefined;
+	let mode: ContextMode = "fresh";
+	let workspaceMode: WorkspaceMode = "shared";
+	let planRequired = false;
+	for (const a of rest) {
+		if (a === "fresh" || a === "branch") mode = a;
+		else if (a === "shared" || a === "worktree") workspaceMode = a;
+		else if (a === "plan") planRequired = true;
+		else if (!nameRaw) nameRaw = a;
+	}
+
+	// Auto-pick a name only in soviet style.
+	if (!nameRaw) {
+		if (!isSovietStyle(style)) {
+			ctx.ui.notify("Usage: /team spawn <name> [fresh|branch] [shared|worktree] [plan]", "error");
+			return;
+		}
+		const taken = new Set(teammates.keys());
+		const picked = pickComradeNames(1, taken)[0];
+		if (!picked) {
+			ctx.ui.notify("Failed to pick a comrade name", "error");
+			return;
+		}
+		nameRaw = picked;
+	}
+
+	const res = await spawnTeammate(ctx, { name: nameRaw, mode, workspaceMode, planRequired });
+	if (!res.ok) {
+		ctx.ui.notify(res.error, "error");
+		return;
+	}
+
+	for (const w of res.warnings) ctx.ui.notify(w, "warning");
+	const displayName = formatMemberDisplayName(style, res.name);
+	ctx.ui.notify(
+		`${displayName} ${strings.joinedVerb} (${res.mode}${res.note ? ", " + res.note : ""} \u00b7 ${res.workspaceMode})`,
+		"info",
+	);
+}