@blackbelt-technology/pi-agent-dashboard 0.2.0

package/packages/shared/src/openspec-poller.ts

@@ -0,0 +1,96 @@
+/**
+ * Polls the openspec CLI to gather change data for the session's project.
+ * Uses async child processes to avoid blocking the event loop.
+ */
+import { spawnSync, execFile } from "node:child_process";
+import { promisify } from "node:util";
+import type { OpenSpecData, OpenSpecChange, OpenSpecArtifact } from "./types.js";
+
+const execFileAsync = promisify(execFile);
+const EMPTY_DATA: OpenSpecData = { initialized: false, changes: [] };
+
+/** Synchronous version — only used by bridge extension where async isn't practical */
+function runOpenSpecSync(args: string[], cwd: string): unknown | null {
+  try {
+    const result = spawnSync("openspec", args, {
+      cwd,
+      encoding: "utf-8",
+      stdio: ["pipe", "pipe", "pipe"],
+      timeout: 10_000,
+    });
+    if (result.status !== 0 || !result.stdout) return null;
+    return JSON.parse(result.stdout);
+  } catch {
+    return null;
+  }
+}
+
+/** Async version — non-blocking, used by server */
+async function runOpenSpecAsync(args: string[], cwd: string): Promise<unknown | null> {
+  try {
+    const { stdout } = await execFileAsync("openspec", args, {
+      cwd,
+      encoding: "utf-8",
+      timeout: 10_000,
+    });
+    if (!stdout) return null;
+    return JSON.parse(stdout);
+  } catch {
+    return null;
+  }
+}
+
+function buildOpenSpecData(
+  listResult: { changes?: Array<{ name: string; status: string; completedTasks: number; totalTasks: number }> } | null,
+  statusResults: Map<string, { artifacts?: Array<{ id: string; status: string }> } | null>,
+): OpenSpecData {
+  if (!listResult || !Array.isArray(listResult.changes)) {
+    return EMPTY_DATA;
+  }
+
+  const changes: OpenSpecChange[] = listResult.changes.map((c) => {
+    const statusResult = statusResults.get(c.name) ?? null;
+    const artifacts: OpenSpecArtifact[] = (statusResult?.artifacts ?? []).map((a) => ({
+      id: a.id,
+      status: (a.status === "done" ? "done" : a.status === "ready" ? "ready" : "blocked") as OpenSpecArtifact["status"],
+    }));
+
+    return {
+      name: c.name,
+      status: (c.status === "complete" ? "complete" : c.status === "in-progress" ? "in-progress" : "no-tasks") as OpenSpecChange["status"],
+      completedTasks: c.completedTasks ?? 0,
+      totalTasks: c.totalTasks ?? 0,
+      artifacts,
+    };
+  });
+
+  return { initialized: true, changes };
+}
+
+/** Synchronous poll — blocks event loop. Used by bridge extension. */
+export function pollOpenSpec(cwd: string): OpenSpecData {
+  const listResult = runOpenSpecSync(["list", "--json"], cwd) as any;
+  if (!listResult || !Array.isArray(listResult.changes)) return EMPTY_DATA;
+
+  const statusResults = new Map<string, any>();
+  for (const c of listResult.changes) {
+    statusResults.set(c.name, runOpenSpecSync(["status", "--change", c.name, "--json"], cwd));
+  }
+  return buildOpenSpecData(listResult, statusResults);
+}
+
+/** Async poll — non-blocking. Used by server directory service. */
+export async function pollOpenSpecAsync(cwd: string): Promise<OpenSpecData> {
+  const listResult = await runOpenSpecAsync(["list", "--json"], cwd) as any;
+  if (!listResult || !Array.isArray(listResult.changes)) return EMPTY_DATA;
+
+  // Run all status queries in parallel
+  const entries = await Promise.all(
+    listResult.changes.map(async (c: any): Promise<[string, any]> => {
+      const status = await runOpenSpecAsync(["status", "--change", c.name, "--json"], cwd);
+      return [c.name, status];
+    }),
+  );
+  const statusResults = new Map<string, any>(entries);
+  return buildOpenSpecData(listResult, statusResults);
+}

package/packages/shared/src/protocol.ts

@@ -0,0 +1,369 @@
+/**
+ * Extension ↔ Server WebSocket protocol messages.
+ */
+import type { DashboardEvent, CommandInfo, FlowInfo, SessionSource, ImageContent, FileEntry, TurnUsage, ContextUsage, ModelInfo, PiSessionInfo, OpenSpecPhase, RoleInfo } from "./types.js";
+
+// ── Extension → Server ──────────────────────────────────────────────
+
+export interface SessionRegisterMessage {
+  type: "session_register";
+  sessionId: string;
+  cwd: string;
+  name?: string;
+  source: SessionSource;
+  model?: string;
+  thinkingLevel?: string;
+  sessionFile?: string;
+  sessionDir?: string;
+  firstMessage?: string;
+  /** True when this is a fresh session start (not a reconnection) */
+  isNew?: boolean;
+  /** Number of conversation entries — used by server to skip event wipe on reconnect */
+  eventCount?: number;
+  /** OS process ID of the pi agent — used for force-kill escalation */
+  pid?: number;
+}
+
+export interface SessionUnregisterMessage {
+  type: "session_unregister";
+  sessionId: string;
+}
+
+export interface ProcessMetrics {
+  /** RSS in bytes */
+  rss: number;
+  /** Heap used in bytes */
+  heapUsed: number;
+  /** Heap total in bytes */
+  heapTotal: number;
+  /** CPU usage percent since last heartbeat (0-100+) */
+  cpuPercent: number;
+  /** Event loop max delay in ms since last heartbeat */
+  eventLoopMaxMs?: number;
+  /** System load average (1 min) */
+  loadAvg1m: number;
+}
+
+export interface SessionHeartbeatMessage {
+  type: "session_heartbeat";
+  sessionId: string;
+  /** Process metrics from the pi agent process */
+  metrics?: ProcessMetrics;
+}
+
+export interface EventForwardMessage {
+  type: "event_forward";
+  sessionId: string;
+  event: DashboardEvent;
+}
+
+export interface CommandsListMessage {
+  type: "commands_list";
+  sessionId: string;
+  commands: CommandInfo[];
+}
+
+export interface FlowsListMessage {
+  type: "flows_list";
+  sessionId: string;
+  flows: FlowInfo[];
+}
+
+export interface ExtensionUiRequestMessage {
+  type: "extension_ui_request";
+  sessionId: string;
+  requestId: string;
+  method: string;
+  params: Record<string, unknown>;
+}
+
+// StatsUpdateMessage removed — server extracts stats directly from forwarded turn_end events
+
+export interface FilesListMessage {
+  type: "files_list";
+  sessionId: string;
+  query: string;
+  files: FileEntry[];
+}
+
+export interface GitInfoUpdateMessage {
+  type: "git_info_update";
+  sessionId: string;
+  gitBranch: string;
+  gitBranchUrl?: string;
+  gitPrNumber?: number;
+  gitPrUrl?: string;
+}
+
+// OpenSpecUpdateMessage removed — server polls directly via DirectoryService
+
+export interface ModelsListMessage {
+  type: "models_list";
+  sessionId: string;
+  models: ModelInfo[];
+}
+
+export interface SessionNameUpdateMessage {
+  type: "session_name_update";
+  sessionId: string;
+  name: string;
+}
+
+export interface SessionsListExtensionMessage {
+  type: "sessions_list";
+  sessionId: string;
+  cwd: string;
+  sessions: PiSessionInfo[];
+}
+
+// SessionHistorySyncMessage removed — server reads history directly via DirectoryService
+// OpenSpecActivityUpdateMessage removed — server detects OpenSpec activity from forwarded events
+
+export interface ModelUpdateMessage {
+  type: "model_update";
+  sessionId: string;
+  model: string;
+  thinkingLevel?: string;
+}
+
+export interface ReplayCompleteMessage {
+  type: "replay_complete";
+  sessionId: string;
+}
+
+export interface FirstMessageUpdateMessage {
+  type: "first_message_update";
+  sessionId: string;
+  firstMessage: string;
+}
+
+export interface RolesListMessage {
+  type: "roles_list";
+  sessionId: string;
+  roles: Record<string, string>;
+  presets: Array<{ name: string; roles: Record<string, string> }>;
+  activePreset: string | null;
+}
+
+export interface ExtensionUiDismissMessage {
+  type: "extension_ui_dismiss";
+  sessionId: string;
+  requestId: string;
+}
+
+export interface SpawnNewSessionMessage {
+  type: "spawn_new_session";
+  sessionId: string;
+  cwd: string;
+}
+
+export interface ProcessInfo {
+  pid: number;
+  pgid: number;
+  command: string;
+  elapsedMs: number;
+}
+
+export interface ProcessListMessage {
+  type: "process_list";
+  sessionId: string;
+  processes: ProcessInfo[];
+}
+
+// LoadSessionEventsResultMessage and LoadSessionEventsErrorMessage removed — server loads directly
+
+export type ExtensionToServerMessage =
+  | SessionRegisterMessage
+  | SessionUnregisterMessage
+  | SessionHeartbeatMessage
+  | EventForwardMessage
+  | CommandsListMessage
+  | FlowsListMessage
+  | ExtensionUiRequestMessage
+  | FilesListMessage
+  | GitInfoUpdateMessage
+  | SessionNameUpdateMessage
+  | ModelsListMessage
+  | ModelUpdateMessage
+  | SessionsListExtensionMessage
+  | ExtensionUiDismissMessage
+  | ReplayCompleteMessage
+  | FirstMessageUpdateMessage
+  | RolesListMessage
+  | SpawnNewSessionMessage
+  | ProcessListMessage;
+
+// ── Server → Extension ──────────────────────────────────────────────
+
+export interface SendPromptToExtensionMessage {
+  type: "send_prompt";
+  sessionId: string;
+  text: string;
+  images?: ImageContent[];
+}
+
+export interface AbortToExtensionMessage {
+  type: "abort";
+  sessionId: string;
+}
+
+export interface RequestCommandsMessage {
+  type: "request_commands";
+  sessionId: string;
+}
+
+export interface RequestStateSyncMessage {
+  type: "request_state_sync";
+  sessionId: string;
+}
+
+export interface ListFilesMessage {
+  type: "list_files";
+  sessionId: string;
+  query: string;
+}
+
+// OpenSpecRefreshMessage removed — server refreshes directly via DirectoryService
+
+export interface RenameSessionExtensionMessage {
+  type: "rename_session";
+  sessionId: string;
+  name: string;
+}
+
+export interface RequestModelsMessage {
+  type: "request_models";
+  sessionId: string;
+}
+
+export interface SetThinkingLevelMessage {
+  type: "set_thinking_level";
+  sessionId: string;
+  level: string;
+}
+
+export interface ListSessionsExtensionMessage {
+  type: "list_sessions";
+  sessionId: string;
+  cwd: string;
+}
+
+export interface SetModelMessage {
+  type: "set_model";
+  sessionId: string;
+  provider: string;
+  modelId: string;
+}
+
+export interface ShutdownExtensionMessage {
+  type: "shutdown";
+  sessionId: string;
+}
+
+export interface FlowControlExtensionMessage {
+  type: "flow_control";
+  sessionId: string;
+  action: "abort" | "toggle_autonomous" | "dismiss_summary";
+}
+
+// LoadSessionEventsMessage removed — server loads directly via DirectoryService
+
+export interface HeartbeatAckMessage {
+  type: "heartbeat_ack";
+}
+
+export interface RequestFlowsRefreshMessage {
+  type: "request_flows_refresh";
+  sessionId: string;
+}
+
+export interface CredentialsUpdatedMessage {
+  type: "credentials_updated";
+}
+
+export interface FlowManagementExtensionMessage {
+  type: "flow_management";
+  sessionId: string;
+  action: "run" | "new" | "edit" | "delete";
+  flowName?: string;
+  task?: string;
+  description?: string;
+}
+
+export interface ArchitectPromptResponseExtensionMessage {
+  type: "architect_prompt_response";
+  sessionId: string;
+  promptId: string;
+  answer?: string;
+  cancelled?: boolean;
+}
+
+export interface RoleSetExtensionMessage {
+  type: "role_set";
+  sessionId: string;
+  role: string;
+  modelId: string;
+}
+
+export interface RolePresetLoadExtensionMessage {
+  type: "role_preset_load";
+  sessionId: string;
+  presetName: string;
+}
+
+export interface RolePresetSaveExtensionMessage {
+  type: "role_preset_save";
+  sessionId: string;
+  presetName: string;
+}
+
+export interface RolePresetDeleteExtensionMessage {
+  type: "role_preset_delete";
+  sessionId: string;
+  presetName: string;
+}
+
+export interface RequestRolesMessage {
+  type: "request_roles";
+  sessionId: string;
+}
+
+export interface KillProcessMessage {
+  type: "kill_process";
+  sessionId: string;
+  pgid: number;
+}
+
+export interface ExtensionUiResponseMessage {
+  type: "extension_ui_response";
+  sessionId: string;
+  requestId: string;
+  result?: unknown;
+  cancelled?: boolean;
+}
+
+export type ServerToExtensionMessage =
+  | SendPromptToExtensionMessage
+  | AbortToExtensionMessage
+  | ExtensionUiResponseMessage
+  | RequestCommandsMessage
+  | RequestStateSyncMessage
+  | ListFilesMessage
+  | RenameSessionExtensionMessage
+  | RequestModelsMessage
+  | SetThinkingLevelMessage
+  | ListSessionsExtensionMessage
+  | SetModelMessage
+  | ShutdownExtensionMessage
+  | FlowControlExtensionMessage
+  | HeartbeatAckMessage
+  | RequestFlowsRefreshMessage
+  | CredentialsUpdatedMessage
+  | FlowManagementExtensionMessage
+  | ArchitectPromptResponseExtensionMessage
+  | RoleSetExtensionMessage
+  | RolePresetLoadExtensionMessage
+  | RolePresetSaveExtensionMessage
+  | RolePresetDeleteExtensionMessage
+  | RequestRolesMessage
+  | KillProcessMessage;

package/packages/shared/src/resolve-jiti.ts

@@ -0,0 +1,43 @@
+/**
+ * Resolve the jiti register hook from pi's process context.
+ *
+ * The bridge extension runs inside pi's Node.js process. process.argv[1]
+ * points to pi's CLI entry (e.g., pi-coding-agent/dist/cli.js). Since
+ * jiti is a dependency of pi-coding-agent, createRequire(process.argv[1])
+ * can resolve it directly.
+ */
+
+import { createRequire } from "node:module";
+import { realpathSync } from "node:fs";
+import path from "node:path";
+
+const JITI_PACKAGES = [
+  "@mariozechner/jiti",
+  "@oh-my-pi/jiti",
+];
+
+/**
+ * Returns the absolute path to jiti's register hook (lib/jiti-register.mjs).
+ * Uses process.argv[1] (pi's entry point) to anchor module resolution.
+ */
+export function resolveJitiImport(): string {
+  const anchor = process.argv[1];
+  if (anchor) {
+    try {
+      // Resolve symlinks — process.argv[1] may be a symlink (e.g., bin/pi → dist/cli.js)
+      const resolved = realpathSync(anchor);
+      const req = createRequire(resolved);
+      for (const jiti of JITI_PACKAGES) {
+        try {
+          const pkgJson = req.resolve(`${jiti}/package.json`);
+          return path.join(path.dirname(pkgJson), "lib", "jiti-register.mjs");
+        } catch { /* next */ }
+      }
+    } catch { /* fall through */ }
+  }
+
+  throw new Error(
+    "Cannot find pi's TypeScript loader (jiti). " +
+    "Is @mariozechner/pi-coding-agent or @oh-my-pi/pi-coding-agent installed?"
+  );
+}