@pixelbyte-software/pixcode 1.54.11 → 1.55.0

package/server/modules/orchestration/a2a/task-dispatcher.ts

@@ -1,345 +0,0 @@
-import crypto from 'node:crypto';
-
-import { adapterRegistry } from '@/modules/orchestration/a2a/adapter-registry.js';
-import { a2aBus } from '@/modules/orchestration/a2a/bus.js';
-import { a2aTaskStore } from '@/modules/orchestration/a2a/task-store.js';
-import type {
-  BusEvent,
-  Message,
-  SubmitTaskInput,
-  Task,
-  TaskState,
-} from '@/modules/orchestration/a2a/types.js';
-import type { TaskHandle } from '@/modules/orchestration/a2a/adapters/abstract-a2a.adapter.js';
-import { portWatcher } from '@/modules/orchestration/preview/port-watcher.js';
-import type { PreviewArtifactData } from '@/modules/orchestration/preview/types.js';
-import {
-  type PermissionPolicy,
-  type PermissionPolicyContext,
-} from '@/modules/orchestration/security/permission-policy.js';
-import { workspaceManager } from '@/modules/orchestration/workspace/workspace-manager.js';
-import type {
-  WorkspaceHandle,
-  WorkspaceKind,
-  WorkspaceMetadata,
-} from '@/modules/orchestration/workspace/types.js';
-import { WorkspaceError } from '@/modules/orchestration/workspace/types.js';
-
-type RoutingHints = {
-  preferredAdapterId?: string;
-  preferredProvider?: string;
-  preferredSkillId?: string;
-};
-
-const TERMINAL: TaskState[] = ['completed', 'canceled', 'failed'];
-const MAX_TASKS = 1000;
-const activeWorkspaces = new Map<string, WorkspaceHandle>();
-const activeHandles = new Map<string, TaskHandle>();
-const previewStops = new Map<string, () => void>();
-const taskUnsubs = new Map<string, () => void>();
-const finalizingTasks = new Set<string>();
-
-function newId(prefix: string): string {
-  return `${prefix}_${crypto.randomBytes(8).toString('hex')}`;
-}
-
-function isTerminalTaskState(state: TaskState): boolean {
-  return TERMINAL.includes(state);
-}
-
-function readRoutingHints(value: unknown): RoutingHints {
-  if (!value || typeof value !== 'object') {
-    return {};
-  }
-
-  const source = value as Record<string, unknown>;
-  return {
-    preferredAdapterId:
-      typeof source.preferredAdapterId === 'string' ? source.preferredAdapterId : undefined,
-    preferredProvider:
-      typeof source.preferredProvider === 'string' ? source.preferredProvider : undefined,
-    preferredSkillId:
-      typeof source.preferredSkillId === 'string' ? source.preferredSkillId : undefined,
-  };
-}
-
-function readString(value: unknown): string | undefined {
-  return typeof value === 'string' && value.trim() ? value : undefined;
-}
-
-function readBoolean(value: unknown): boolean | undefined {
-  return typeof value === 'boolean' ? value : undefined;
-}
-
-function readObject(value: unknown): Record<string, unknown> | undefined {
-  return value && typeof value === 'object' ? value as Record<string, unknown> : undefined;
-}
-
-function readWorkspaceKind(value: unknown): WorkspaceKind | undefined {
-  return value === 'host' || value === 'worktree' || value === 'docker' ? value : undefined;
-}
-
-function workspaceMetadata(workspace: WorkspaceHandle, keepAfterCompletion?: boolean): WorkspaceMetadata {
-  return {
-    id: workspace.id,
-    kind: workspace.kind,
-    path: workspace.path,
-    baseRef: workspace.baseRef,
-    branchName: workspace.branchName,
-    keepAfterCompletion,
-  };
-}
-
-function evictOldestTerminalTask(): boolean {
-  for (const [taskId, task] of a2aTaskStore.entries()) {
-    if (!isTerminalTaskState(task.state)) continue;
-    const unsub = taskUnsubs.get(taskId);
-    if (unsub) {
-      unsub();
-      taskUnsubs.delete(taskId);
-    }
-    a2aTaskStore.delete(taskId);
-    return true;
-  }
-  return false;
-}
-
-async function finalizeTerminalTask(task: Task): Promise<void> {
-  if (finalizingTasks.has(task.id)) return;
-  finalizingTasks.add(task.id);
-
-  const stopPreview = previewStops.get(task.id);
-  if (stopPreview) {
-    stopPreview();
-    previewStops.delete(task.id);
-  }
-
-  const workspace = activeWorkspaces.get(task.id);
-  try {
-    if (workspace) {
-      const diff = await workspace.diff();
-      if (diff.trim()) {
-        a2aBus.publish({
-          kind: 'artifact',
-          taskId: task.id,
-          artifact: {
-            artifactId: newId('art'),
-            type: 'file-diff',
-            parts: [{ kind: 'text', text: diff }],
-            metadata: {
-              source: 'workspace-diff',
-              workspaceId: workspace.id,
-              workspaceKind: workspace.kind,
-              baseRef: workspace.baseRef,
-            },
-          },
-        });
-      }
-
-      const keepAfterCompletion = task.metadata?.workspace &&
-        typeof task.metadata.workspace === 'object' &&
-        readBoolean((task.metadata.workspace as Record<string, unknown>).keepAfterCompletion);
-      if (workspace.kind !== 'host' && keepAfterCompletion !== true) {
-        await workspace.destroy();
-      }
-    }
-  } catch (error) {
-    const message = error instanceof Error ? error.message : String(error);
-    task.metadata = {
-      ...task.metadata,
-      workspaceFinalizationError: message,
-    };
-    task.updatedAt = Date.now();
-    a2aTaskStore.set(task);
-  } finally {
-    activeWorkspaces.delete(task.id);
-    activeHandles.delete(task.id);
-    const unsub = taskUnsubs.get(task.id);
-    if (unsub) {
-      unsub();
-      taskUnsubs.delete(task.id);
-    }
-    finalizingTasks.delete(task.id);
-  }
-}
-
-function attachBusToTask(task: Task): void {
-  if (taskUnsubs.has(task.id)) return;
-
-  const unsubscribe = a2aBus.subscribe(task.id, (event: BusEvent) => {
-    if (event.kind === 'task-state') {
-      task.state = event.state;
-      if (event.error) task.error = event.error;
-      task.updatedAt = Date.now();
-      a2aTaskStore.set(task);
-      if (isTerminalTaskState(event.state)) {
-        void finalizeTerminalTask(task);
-      }
-    } else if (event.kind === 'message') {
-      task.history.push(event.message);
-      task.updatedAt = Date.now();
-      a2aTaskStore.set(task);
-    } else if (event.kind === 'artifact') {
-      task.artifacts.push(event.artifact);
-      task.updatedAt = Date.now();
-      a2aTaskStore.set(task);
-    }
-  });
-  taskUnsubs.set(task.id, unsubscribe);
-}
-
-export function getA2ATask(taskId: string): Task | undefined {
-  return a2aTaskStore.get(taskId);
-}
-
-export function listA2AAgentCards() {
-  return adapterRegistry.agentCards();
-}
-
-export async function submitA2ATask(input: SubmitTaskInput): Promise<Task> {
-  const metadata = input.metadata as Record<string, unknown> | undefined;
-  const routing = readRoutingHints(metadata?.routing);
-  const adapter = adapterRegistry.resolve(input.adapterId, routing);
-  if (!adapter) {
-    throw new Error(`ADAPTER_NOT_FOUND: ${input.adapterId}`);
-  }
-
-  if (a2aTaskStore.size >= MAX_TASKS && !evictOldestTerminalTask()) {
-    throw new Error(`TASK_LIMIT: task store at capacity (${MAX_TASKS})`);
-  }
-
-  const userMessage: Message = input.message;
-  const task: Task = {
-    id: newId('task'),
-    contextId: input.contextId,
-    state: 'submitted',
-    history: [{ ...userMessage, taskId: '' }],
-    artifacts: [],
-    metadata: {
-      ...metadata,
-      adapterId: adapter.id,
-      adapterSelector: input.adapterId,
-    },
-    createdAt: Date.now(),
-    updatedAt: Date.now(),
-  };
-  task.history[0] = { ...userMessage, taskId: task.id };
-  a2aTaskStore.set(task);
-  attachBusToTask(task);
-
-  const workspaceOptions = (metadata?.workspace && typeof metadata.workspace === 'object'
-    ? metadata.workspace
-    : {}) as Record<string, unknown>;
-  let workspace: WorkspaceHandle;
-  try {
-    workspace = await workspaceManager.create({
-      taskId: task.id,
-      projectPath: readString(workspaceOptions.projectPath) ?? process.cwd(),
-      kind: readWorkspaceKind(workspaceOptions.kind) ?? readWorkspaceKind(metadata?.isolation),
-      baseRef: readString(workspaceOptions.baseRef) ?? readString(metadata?.baseRef) ?? 'HEAD',
-      keepAfterCompletion: readBoolean(workspaceOptions.keepAfterCompletion),
-      metadata: workspaceOptions,
-    });
-  } catch (err) {
-    const workspaceError = err instanceof WorkspaceError ? err : undefined;
-    a2aBus.publish({
-      kind: 'task-state',
-      taskId: task.id,
-      state: 'failed',
-      error: {
-        code: workspaceError?.code ?? 'WORKSPACE_CREATE_FAILED',
-        message: err instanceof Error ? err.message : String(err),
-        details: workspaceError?.details,
-      },
-    });
-    return a2aTaskStore.get(task.id) ?? task;
-  }
-
-  activeWorkspaces.set(task.id, workspace);
-  task.metadata = {
-    ...task.metadata,
-    workspace: workspaceMetadata(workspace, readBoolean(workspaceOptions.keepAfterCompletion)),
-  };
-  task.updatedAt = Date.now();
-  a2aTaskStore.set(task);
-
-  previewStops.set(
-    task.id,
-    portWatcher.watch({
-      taskId: task.id,
-      workspace,
-      onPort: (event) => {
-        const data: PreviewArtifactData = {
-          url: event.url,
-          proxiedUrl: `/preview/${event.port}/`,
-          port: event.port,
-          host: event.host,
-          processName: event.processName,
-          confidence: event.confidence,
-        };
-        a2aBus.publish({
-          kind: 'artifact',
-          taskId: task.id,
-          artifact: {
-            artifactId: newId('art'),
-            type: 'preview-url',
-            parts: [{ kind: 'data', data: { ...data } }],
-            metadata: {
-              source: 'port-watcher',
-              workspaceId: workspace.id,
-            },
-          },
-        });
-      },
-    }),
-  );
-
-  try {
-    const handle = await adapter.submitTask(task, {
-      cwd: workspace.path,
-      workspace,
-      model: readString(metadata?.model),
-      permissionMode: readString(metadata?.permissionMode),
-      permissionPolicy: readObject(metadata?.permissionPolicy) as PermissionPolicy | undefined,
-      permissionPolicyContext: readObject(metadata?.permissionPolicyContext) as PermissionPolicyContext | undefined,
-      toolsSettings: readObject(metadata?.toolsSettings),
-    });
-    activeHandles.set(task.id, handle);
-    handle.finished.finally(() => {
-      activeHandles.delete(task.id);
-    }).catch(() => undefined);
-  } catch (err) {
-    a2aBus.publish({
-      kind: 'task-state',
-      taskId: task.id,
-      state: 'failed',
-      error: {
-        code: 'ADAPTER_SUBMIT_FAILED',
-        message: err instanceof Error ? err.message : String(err),
-      },
-    });
-  }
-
-  return a2aTaskStore.get(task.id) ?? task;
-}
-
-export async function cancelA2ATask(taskId: string, adapterId?: string): Promise<Task | undefined> {
-  const task = a2aTaskStore.get(taskId);
-  if (!task) return undefined;
-
-  const handle = activeHandles.get(taskId);
-  if (handle) {
-    await handle.cancel();
-    return a2aTaskStore.get(taskId) ?? task;
-  }
-
-  const resolvedAdapterId = adapterId ?? readString(task.metadata?.adapterId);
-  const adapter = resolvedAdapterId ? adapterRegistry.resolve(resolvedAdapterId) : undefined;
-  if (adapter) {
-    await adapter.cancelTask(task.id);
-    return a2aTaskStore.get(task.id) ?? task;
-  }
-
-  a2aBus.publish({ kind: 'task-state', taskId: task.id, state: 'canceled' });
-  return a2aTaskStore.get(task.id) ?? task;
-}

package/server/modules/orchestration/a2a/task-store.ts

@@ -1,178 +0,0 @@
-// server/modules/orchestration/a2a/task-store.ts
-// A2A task persistence backed by shared JsonStore for atomic writes and corruption recovery.
-// Exposes a Map-compatible surface (get/set/delete/size/entries/values) so routes.ts
-// can treat the store as a typed Map while still benefiting from durable JSON storage.
-
-import os from 'os';
-import path from 'path';
-
-import { JsonStore } from '@/database/json-store.js';
-import type { Message, Task, TaskState, TaskSummary } from '@/modules/orchestration/a2a/types.js';
-
-const filePath = path.join(os.homedir(), '.pixcode', 'a2a-tasks.json');
-
-export interface A2ATaskStoreOptions {
-  terminalTaskTtlMs?: number;
-}
-
-export interface ListOptions {
-  state?: TaskState;
-  contextId?: string;
-  adapterId?: string;
-  limit?: number;
-}
-
-export class A2ATaskStore {
-  private store: JsonStore;
-
-  constructor(_options?: A2ATaskStoreOptions) {
-    this.store = new JsonStore(filePath, { a2a_tasks: [] });
-  }
-
-  // ── Map-compatible API ─────────────────────────────────────────────────────
-
-  get(id: string): Task | undefined {
-    return this.store.findWhere('a2a_tasks', (t: Task) => t.id === id) ?? undefined;
-  }
-
-  set(task: Task): void {
-    const existing = this.get(task.id);
-    if (existing) {
-      this.store.updateWhere('a2a_tasks', (t: Task) => t.id === task.id, task);
-    } else {
-      this.store.insert('a2a_tasks', task, { autoId: false });
-    }
-  }
-
-  delete(id: string): void {
-    this.store.deleteWhere('a2a_tasks', (t: Task) => t.id === id);
-  }
-
-  get size(): number {
-    return (this.store.filterWhere('a2a_tasks', () => true) as Task[]).length;
-  }
-
-  *entries(): IterableIterator<[string, Task]> {
-    const tasks = this.store.filterWhere('a2a_tasks', () => true) as Task[];
-    for (const task of tasks) {
-      yield [task.id, task];
-    }
-  }
-
-  *values(): IterableIterator<Task> {
-    const tasks = this.store.filterWhere('a2a_tasks', () => true) as Task[];
-    for (const task of tasks) {
-      yield task;
-    }
-  }
-
-  // ── Query helpers ──────────────────────────────────────────────────────────
-
-  list(options: ListOptions = {}): Task[] {
-    const { state, contextId, adapterId, limit = 50 } = options;
-    let tasks = this.store.filterWhere('a2a_tasks', (t: Task) => {
-      if (state && t.state !== state) return false;
-      if (contextId && t.contextId !== contextId) return false;
-      if (adapterId && (t.metadata?.adapterId as string | undefined) !== adapterId) return false;
-      return true;
-    }) as Task[];
-    if (limit > 0) tasks = tasks.slice(0, limit);
-    return tasks;
-  }
-
-  summarize(task: Task): TaskSummary {
-    return {
-      id: task.id,
-      contextId: task.contextId,
-      state: task.state,
-      adapterId: task.metadata?.adapterId as string | undefined,
-      adapterSelector: task.metadata?.adapterSelector as string | undefined,
-      error: task.error,
-      createdAt: task.createdAt,
-      updatedAt: task.updatedAt,
-      messageCount: task.history?.length ?? 0,
-      artifactCount: task.artifacts?.length ?? 0,
-      lastMessage: task.history?.at(-1),
-    };
-  }
-
-  // ── Legacy named methods (kept for any future internal callers) ────────────
-
-  getTask(id: string): Task | undefined {
-    return this.get(id);
-  }
-
-  getTaskByIndex(index: number): Task | undefined {
-    const tasks = this.store.filterWhere('a2a_tasks', () => true) as Task[];
-    return tasks[index] ?? undefined;
-  }
-
-  getTasks(): Task[] {
-    return this.store.filterWhere('a2a_tasks', () => true) as Task[];
-  }
-
-  createTask(task: Task): void {
-    this.store.insert('a2a_tasks', task, { autoId: false });
-  }
-
-  updateTask(id: string, updates: Partial<Task>): void {
-    const existing = this.get(id);
-    if (!existing) return;
-    const updated = { ...existing, ...updates, id: existing.id };
-    this.store.updateWhere('a2a_tasks', (t: Task) => t.id === id, updated);
-  }
-
-  updateTaskState(id: string, state: TaskState): void {
-    this.updateTask(id, { status: { state } } as Partial<Task>);
-  }
-
-  deleteTask(id: string): void {
-    this.delete(id);
-  }
-
-  appendHistory(id: string, entry: { state: TaskState; timestamp: string }): void {
-    const task = this.get(id);
-    if (!task) return;
-    const history = [...(task.history ?? []), entry as unknown as Message];
-    this.updateTask(id, { history } as Partial<Task>);
-  }
-
-  addArtifact(id: string, artifact: NonNullable<Task['artifacts']>[number]): void {
-    const task = this.get(id);
-    if (!task) return;
-    const artifacts = [...(task.artifacts ?? []), artifact];
-    this.updateTask(id, { artifacts } as Partial<Task>);
-  }
-
-  addMessage(id: string, message: Message): void {
-    const task = this.get(id);
-    if (!task) return;
-    const history = [...(task.history ?? []), message];
-    this.updateTask(id, { history } as Partial<Task>);
-  }
-
-  updateMessage(id: string, messageId: string, updates: Partial<Message>): void {
-    const task = this.get(id);
-    if (!task?.history) return;
-    const history = task.history.map((m) =>
-      m.messageId === messageId ? { ...m, ...updates } : m,
-    );
-    this.updateTask(id, { history } as Partial<Task>);
-  }
-
-  getMessage(id: string, messageId: string): Message | undefined {
-    const task = this.get(id);
-    return task?.history?.find((m) => m.messageId === messageId);
-  }
-
-  getMessages(id: string): Message[] {
-    const task = this.get(id);
-    return task?.history ?? [];
-  }
-
-  clearMessages(id: string): void {
-    this.updateTask(id, { history: [] } as Partial<Task>);
-  }
-}
-
-export const a2aTaskStore = new A2ATaskStore();

package/server/modules/orchestration/a2a/types.ts

@@ -1,126 +0,0 @@
-// server/modules/orchestration/a2a/types.ts
-// A2A protocol v0.2 types — minimal surface used by pixcode.
-// See https://a2a-protocol.org for the full spec; this file
-// keeps only what the orchestrator actually exchanges.
-
-export type TaskState =
-  | 'submitted'
-  | 'working'
-  | 'input-required'
-  | 'completed'
-  | 'canceled'
-  | 'failed';
-
-export type Role = 'user' | 'agent';
-
-export type PartKind = 'text' | 'file' | 'data';
-
-export interface TextPart {
-  kind: 'text';
-  text: string;
-}
-
-export interface FilePart {
-  kind: 'file';
-  name: string;
-  mimeType?: string;
-  bytesBase64?: string;
-  uri?: string;
-}
-
-export interface DataPart {
-  kind: 'data';
-  data: Record<string, unknown>;
-}
-
-export type Part = TextPart | FilePart | DataPart;
-
-export interface Message {
-  messageId: string;
-  role: Role;
-  parts: Part[];
-  /** Required for task-scoped messages. Omit only for broadcast/standalone messages. */
-  taskId?: string;
-}
-
-export type ArtifactType =
-  | 'file-diff'
-  | 'command-output'
-  | 'preview-url'
-  | 'handoff-artifact'
-  | 'data';
-
-// Note: Artifact and AuthScheme use `type` discriminator (matches A2A
-// v0.2 wire format). Part and BusEvent use `kind` per the same spec.
-export interface Artifact {
-  artifactId: string;
-  type: ArtifactType;
-  parts: Part[];
-  metadata?: Record<string, unknown>;
-}
-
-export interface TaskError {
-  code: string;
-  message: string;
-  details?: Record<string, unknown>;
-}
-
-export interface Task {
-  id: string;
-  contextId?: string;
-  state: TaskState;
-  history: Message[];
-  artifacts: Artifact[];
-  error?: TaskError;
-  metadata?: Record<string, unknown>;
-  createdAt: number;
-  updatedAt: number;
-}
-
-export interface TaskSummary {
-  id: string;
-  contextId?: string;
-  state: TaskState;
-  adapterId?: string;
-  adapterSelector?: string;
-  error?: TaskError;
-  createdAt: number;
-  updatedAt: number;
-  messageCount: number;
-  artifactCount: number;
-  lastMessage?: Message;
-}
-
-export interface AgentSkill {
-  id: string;
-  description: string;
-  examples?: string[];
-}
-
-export type AuthScheme =
-  | { type: 'none' }
-  | { type: 'bearer' }
-  | { type: 'mtls' };
-
-export interface AgentCard {
-  name: string;
-  description: string;
-  url: string;
-  version: string;
-  capabilities: string[];
-  skills: AgentSkill[];
-  authentication: AuthScheme;
-}
-
-export interface SubmitTaskInput {
-  message: Message;
-  contextId?: string;
-  metadata?: Record<string, unknown>;
-  /** Adapter id, "auto", or "skill:<id>". Resolved by the adapter registry. */
-  adapterId: string;
-}
-
-export type BusEvent =
-  | { kind: 'task-state'; taskId: string; state: TaskState; error?: TaskError }
-  | { kind: 'message'; taskId: string; message: Message }
-  | { kind: 'artifact'; taskId: string; artifact: Artifact };