@hueyexe/opencode-ensemble 0.1.0

package/src/tools/team-status.ts

@@ -0,0 +1,49 @@
+import type { ToolDeps } from "../types"
+import { findTeamBySession } from "../types"
+
+/**
+ * Execute the team_status tool. Shows team overview with member statuses and task summary.
+ */
+export async function executeTeamStatus(
+  deps: ToolDeps,
+  sessionId: string,
+): Promise<string> {
+  const teamInfo = findTeamBySession(deps.db, deps.registry, sessionId)
+  if (!teamInfo) throw new Error("This session is not in a team.")
+
+  const members = deps.db.query(
+    "SELECT name, session_id, agent, status, execution_status FROM team_member WHERE team_id = ? ORDER BY time_created ASC"
+  ).all(teamInfo.teamId) as Array<{
+    name: string; session_id: string; agent: string; status: string; execution_status: string
+  }>
+
+  const tasks = deps.db.query(
+    "SELECT status FROM team_task WHERE team_id = ?"
+  ).all(teamInfo.teamId) as Array<{ status: string }>
+
+  const lines: string[] = []
+  lines.push(`Team: ${teamInfo.teamName} (you are the ${teamInfo.role})`)
+  lines.push("")
+
+  if (members.length === 0) {
+    lines.push("No teammates spawned yet.")
+  } else {
+    lines.push("Members:")
+    for (const m of members) {
+      const statusIcon = m.status === "busy" ? "working" : m.status === "ready" ? "idle" : m.status
+      lines.push(`  ${m.name}  [${statusIcon}]  agent: ${m.agent}  session: ${m.session_id}`)
+    }
+  }
+
+  if (tasks.length > 0) {
+    const byStatus = new Map<string, number>()
+    for (const t of tasks) {
+      byStatus.set(t.status, (byStatus.get(t.status) ?? 0) + 1)
+    }
+    const parts = Array.from(byStatus.entries()).map(([s, n]) => `${n} ${s}`)
+    lines.push("")
+    lines.push(`Tasks: ${tasks.length} total (${parts.join(", ")})`)
+  }
+
+  return lines.join("\n")
+}

package/src/tools/team-tasks-add.ts

@@ -0,0 +1,49 @@
+import type { ToolDeps } from "../types"
+import { findTeamBySession } from "../types"
+import { generateId } from "../util"
+
+interface TaskInput {
+  content: string
+  priority: string
+  depends_on?: string[]
+}
+
+/**
+ * Execute the team_tasks_add tool. Adds tasks to the shared board.
+ * Tasks with unresolved dependencies are marked as 'blocked'.
+ */
+export async function executeTeamTasksAdd(
+  deps: ToolDeps,
+  args: { tasks: TaskInput[] },
+  sessionId: string,
+): Promise<string> {
+  const teamInfo = findTeamBySession(deps.db, deps.registry, sessionId)
+  if (!teamInfo) throw new Error("This session is not in a team.")
+
+  const ids: string[] = []
+  const now = Date.now()
+
+  for (const task of args.tasks) {
+    const id = generateId("task")
+    const depsJson = task.depends_on?.length ? JSON.stringify(task.depends_on) : null
+
+    // Determine initial status — blocked if has unresolved dependencies
+    let status = "pending"
+    if (task.depends_on?.length) {
+      const resolved = task.depends_on.every(depId => {
+        const dep = deps.db.query("SELECT status FROM team_task WHERE id = ? AND team_id = ?")
+          .get(depId, teamInfo.teamId) as { status: string } | null
+        return dep && (dep.status === "completed" || dep.status === "cancelled")
+      })
+      if (!resolved) status = "blocked"
+    }
+
+    deps.db.run(
+      "INSERT INTO team_task (id, team_id, content, status, priority, depends_on, time_created, time_updated) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
+      [id, teamInfo.teamId, task.content, status, task.priority, depsJson, now, now]
+    )
+    ids.push(id)
+  }
+
+  return `Added ${ids.length} task${ids.length !== 1 ? "s" : ""}: ${ids.join(", ")}`
+}

package/src/tools/team-tasks-complete.ts

@@ -0,0 +1,48 @@
+import type { ToolDeps } from "../types"
+import { findTeamBySession } from "../types"
+
+/**
+ * Execute the team_tasks_complete tool. Marks a task as completed
+ * and unblocks any dependent tasks.
+ */
+export async function executeTeamTasksComplete(
+  deps: ToolDeps,
+  args: { task_id: string },
+  sessionId: string,
+): Promise<string> {
+  const teamInfo = findTeamBySession(deps.db, deps.registry, sessionId)
+  if (!teamInfo) throw new Error("This session is not in a team.")
+
+  const task = deps.db.query("SELECT * FROM team_task WHERE id = ? AND team_id = ?")
+    .get(args.task_id, teamInfo.teamId) as Record<string, unknown> | null
+  if (!task) throw new Error(`Task "${args.task_id}" not found`)
+
+  const now = Date.now()
+  deps.db.run("UPDATE team_task SET status = 'completed', time_updated = ? WHERE id = ?", [now, args.task_id])
+
+  // Unblock dependent tasks
+  const allTasks = deps.db.query("SELECT id, depends_on, status FROM team_task WHERE team_id = ?")
+    .all(teamInfo.teamId) as Array<{ id: string; depends_on: string | null; status: string }>
+
+  let unblocked = 0
+  for (const t of allTasks) {
+    if (t.status !== "blocked" || !t.depends_on) continue
+    const depIds: string[] = JSON.parse(t.depends_on)
+    if (!depIds.includes(args.task_id)) continue
+
+    // Check if all dependencies are now resolved
+    const allResolved = depIds.every(depId => {
+      if (depId === args.task_id) return true // just completed
+      const dep = allTasks.find(d => d.id === depId)
+      return dep && (dep.status === "completed" || dep.status === "cancelled")
+    })
+
+    if (allResolved) {
+      deps.db.run("UPDATE team_task SET status = 'pending', time_updated = ? WHERE id = ?", [now, t.id])
+      unblocked++
+    }
+  }
+
+  const unblockedMsg = unblocked > 0 ? ` Unblocked ${unblocked} dependent task${unblocked !== 1 ? "s" : ""}.` : ""
+  return `Completed task: ${task.content}${unblockedMsg}`
+}

package/src/tools/team-tasks-list.ts

@@ -0,0 +1,23 @@
+import type { ToolDeps } from "../types"
+import { findTeamBySession } from "../types"
+
+/**
+ * Execute the team_tasks_list tool. Shows all tasks on the shared board.
+ */
+export async function executeTeamTasksList(
+  deps: ToolDeps,
+  sessionId: string,
+): Promise<string> {
+  const teamInfo = findTeamBySession(deps.db, deps.registry, sessionId)
+  if (!teamInfo) throw new Error("This session is not in a team.")
+
+  const tasks = deps.db.query(
+    "SELECT * FROM team_task WHERE team_id = ? ORDER BY time_created ASC"
+  ).all(teamInfo.teamId) as Array<Record<string, unknown>>
+
+  if (tasks.length === 0) return "No tasks on the board."
+
+  return tasks.map(t =>
+    `[${t.status}] ${t.content} (${t.id})${t.assignee ? ` → ${t.assignee}` : ""}${t.priority !== "medium" ? ` [${t.priority}]` : ""}`
+  ).join("\n")
+}

package/src/tools/team-view.ts

@@ -0,0 +1,23 @@
+import type { ToolDeps } from "../types"
+import { findTeamBySession } from "../types"
+
+/**
+ * Execute the team_view tool. Navigates the TUI to a teammate's session
+ * so the user can see what they're doing.
+ */
+export async function executeTeamView(
+  deps: ToolDeps,
+  args: { member: string },
+  sessionId: string,
+): Promise<string> {
+  const teamInfo = findTeamBySession(deps.db, deps.registry, sessionId)
+  if (!teamInfo) throw new Error("This session is not in a team.")
+
+  const member = deps.db.query("SELECT session_id, status, agent FROM team_member WHERE team_id = ? AND name = ?")
+    .get(teamInfo.teamId, args.member) as { session_id: string; status: string; agent: string } | null
+  if (!member) throw new Error(`Teammate "${args.member}" not found in team "${teamInfo.teamName}"`)
+
+  await deps.client.tui.selectSession({ sessionID: member.session_id })
+
+  return `Switched view to ${args.member}'s session (${member.session_id}). Use the session picker to return to the lead session.`
+}

package/src/types.ts

@@ -0,0 +1,82 @@
+import type { Database } from "bun:sqlite"
+import type { MemberRegistry, DescendantTracker } from "./state"
+
+/**
+ * Shared dependencies injected into every tool's execute function.
+ * This avoids global state and makes tools testable with mocks.
+ */
+export interface ToolDeps {
+  db: Database
+  registry: MemberRegistry
+  tracker: DescendantTracker
+  /** The OpenCode SDK client — used for session.create, promptAsync, abort, etc. */
+  client: PluginClient
+}
+
+/**
+ * Minimal interface for the OpenCode client methods we actually use.
+ * This is a subset of OpencodeClient, typed for our needs.
+ * Makes mocking trivial in tests.
+ */
+export interface PluginClient {
+  session: {
+    create(options: { body?: { parentID?: string; title?: string } }): Promise<{ data?: { id: string } }>
+    promptAsync(options: {
+      path: { id: string }
+      body: { parts: Array<{ type: "text"; text: string }>; model?: { providerID: string; modelID: string } }
+    }): Promise<unknown>
+    abort(options: { path: { id: string } }): Promise<unknown>
+    status(): Promise<{ data?: Record<string, { type: string }> }>
+  }
+  tui: {
+    showToast(options: {
+      title?: string
+      message?: string
+      variant?: "info" | "success" | "warning" | "error"
+      duration?: number
+    }): Promise<unknown>
+    selectSession(options: { sessionID?: string }): Promise<unknown>
+  }
+}
+
+/**
+ * Look up which team a session belongs to (as lead or member).
+ * Returns team info or undefined.
+ */
+export function findTeamBySession(
+  db: Database,
+  registry: MemberRegistry,
+  sessionId: string,
+): { teamId: string; teamName: string; role: "lead" | "member"; memberName?: string } | undefined {
+  // Check if session is a team member
+  const entry = registry.getBySession(sessionId)
+  if (entry) {
+    const team = db.query("SELECT name FROM team WHERE id = ? AND status = 'active'").get(entry.teamId) as { name: string } | null
+    if (team) return { teamId: entry.teamId, teamName: team.name, role: "member", memberName: entry.memberName }
+  }
+
+  // Check if session is a team lead
+  const leadTeam = db.query("SELECT id, name FROM team WHERE lead_session_id = ? AND status = 'active'").get(sessionId) as { id: string; name: string } | null
+  if (leadTeam) return { teamId: leadTeam.id, teamName: leadTeam.name, role: "lead" }
+
+  return undefined
+}
+
+/**
+ * Get the session ID for a recipient by name.
+ * "lead" resolves to the team's lead_session_id.
+ * Otherwise looks up the member registry.
+ */
+export function resolveRecipientSession(
+  db: Database,
+  registry: MemberRegistry,
+  teamId: string,
+  recipientName: string,
+): string | undefined {
+  if (recipientName === "lead") {
+    const team = db.query("SELECT lead_session_id FROM team WHERE id = ?").get(teamId) as { lead_session_id: string } | null
+    return team?.lead_session_id
+  }
+  const entry = registry.getByName(teamId, recipientName)
+  return entry?.sessionId
+}

package/src/util.ts

@@ -0,0 +1,35 @@
+const NAME_PATTERN = /^[a-z0-9]([a-z0-9-]*[a-z0-9])?$/
+
+let counter = 0
+
+/**
+ * Generate a time-sortable unique ID with the given prefix.
+ * Format: `{prefix}_{timestamp_hex}_{counter_hex}_{random_hex}`
+ */
+export function generateId(prefix: string): string {
+  const time = Date.now().toString(36)
+  const count = (counter++).toString(36).padStart(4, "0")
+  const rand = Math.random().toString(36).slice(2, 8)
+  return `${prefix}_${time}_${count}_${rand}`
+}
+
+/**
+ * Validate a team name. Returns an error message string if invalid, undefined if valid.
+ * Rules: 1-64 chars, lowercase alphanumeric with hyphens, no leading/trailing hyphens.
+ */
+export function validateTeamName(name: string): string | undefined {
+  if (name.length < 1 || name.length > 64) return "Team name must be 1-64 characters"
+  if (!NAME_PATTERN.test(name)) return "Team name must be lowercase alphanumeric with hyphens only"
+  return undefined
+}
+
+/**
+ * Validate a member name. Same rules as team name, plus "lead" is reserved.
+ * Returns an error message string if invalid, undefined if valid.
+ */
+export function validateMemberName(name: string): string | undefined {
+  if (name.toLowerCase() === "lead") return `Name "lead" is reserved`
+  if (name.length < 1 || name.length > 64) return "Member name must be 1-64 characters"
+  if (!NAME_PATTERN.test(name)) return "Member name must be lowercase alphanumeric with hyphens only"
+  return undefined
+}