@pixelbyte-software/pixcode 1.55.1 → 1.55.3

package/server/modules/tasks/task-routes.ts

@@ -0,0 +1,245 @@
+// ============================================================
+// Task Routes — Express API endpoints for task management
+// ============================================================
+
+import type { Router, Request, Response } from 'express';
+import express from 'express';
+
+import {
+  createTask,
+  getTask,
+  getTasksByProject,
+  getAllTasks,
+  updateTask,
+  deleteTask,
+  getLogs,
+  getPendingInteractions,
+  addLog,
+} from './task-store.js';
+import { enqueue } from './task-queue.js';
+import { cancelTask, taskEvents } from './task-runner.js';
+import { submitAnswer } from './user-interaction.js';
+import { getMaxConcurrent, setMaxConcurrent } from './task-queue.js';
+import { listRoles } from './role-presets.js';
+import type { CreateTaskInput, AgentType, TaskRole, TaskPriority } from './task-types.js';
+
+export function taskRouter(): Router {
+  const router = express.Router();
+
+  // ============================================================
+  // GET /api/tasks — List all tasks
+  // ============================================================
+  router.get('/', (req: Request, res: Response) => {
+    const limit = parseInt(String(req.query.limit)) || 100;
+    const projectId = String(req.query.projectId);
+    const tasks = projectId ? getTasksByProject(projectId) : getAllTasks(limit);
+    res.json({ tasks });
+  });
+
+  // ============================================================
+  // GET /api/tasks/:id — Get a single task
+  // ============================================================
+  router.get('/:id', (req: Request, res: Response) => {
+    const task = getTask(String(req.params.id));
+    if (!task) {
+      res.status(404).json({ error: 'Task not found' });
+      return;
+    }
+    res.json({ task });
+  });
+
+  // ============================================================
+  // POST /api/tasks — Create a new task
+  // ============================================================
+  router.post('/', async (req: Request, res: Response) => {
+    try {
+      const {
+        projectId,
+        title,
+        prompt,
+        agentType,
+        model,
+        role,
+        priority,
+        predecessorTaskId,
+        continueSession,
+        maxBudgetUsd,
+        interactionTimeout,
+        thinkingEnabled,
+        permissionMode,
+      } = req.body;
+
+      if (!projectId || !title || !prompt || !agentType) {
+        res.status(400).json({
+          error: 'Missing required fields: projectId, title, prompt, agentType',
+        });
+        return;
+      }
+
+      const validAgents: AgentType[] = ['claude-code', 'codex', 'gemini', 'qwen', 'opencode'];
+      if (!validAgents.includes(agentType)) {
+        res.status(400).json({ error: `Invalid agentType. Valid: ${validAgents.join(', ')}` });
+        return;
+      }
+
+      const input: CreateTaskInput = {
+        projectId,
+        title,
+        prompt,
+        agentType,
+        model,
+        role: role as TaskRole,
+        priority: priority as TaskPriority,
+        predecessorTaskId,
+        continueSession,
+        maxBudgetUsd,
+        interactionTimeout,
+        thinkingEnabled,
+        permissionMode,
+      };
+
+      const task = createTask(input);
+      await enqueue(task.id);
+      taskEvents.emit('task:enqueued', task.id);
+
+      res.status(201).json({ task });
+    } catch (err: any) {
+      res.status(500).json({ error: err.message });
+    }
+  });
+
+  // ============================================================
+  // POST /api/tasks/:id/cancel — Cancel a task
+  // ============================================================
+  router.post('/:id/cancel', async (req: Request, res: Response) => {
+    const success = await cancelTask(String(req.params.id));
+    if (!success) {
+      res.status(404).json({ error: 'Task not found' });
+      return;
+    }
+    res.json({ success: true });
+  });
+
+  // ============================================================
+  // DELETE /api/tasks/:id — Delete a task
+  // ============================================================
+  router.delete('/:id', (req: Request, res: Response) => {
+    const success = deleteTask(String(req.params.id));
+    if (!success) {
+      res.status(404).json({ error: 'Task not found' });
+      return;
+    }
+    res.json({ success: true });
+  });
+
+  // ============================================================
+  // GET /api/tasks/:id/logs — Get task logs
+  // ============================================================
+  router.get('/:id/logs', (req: Request, res: Response) => {
+    const limit = parseInt(String(req.query.limit)) || 200;
+    const logs = getLogs(String(req.params.id), limit);
+    res.json({ logs });
+  });
+
+  // ============================================================
+  // GET /api/tasks/:id/interactions — Get pending interactions
+  // ============================================================
+  router.get('/:id/interactions', (req: Request, res: Response) => {
+    const interactions = getPendingInteractions(String(req.params.id));
+    res.json({ interactions });
+  });
+
+  // ============================================================
+  // POST /api/tasks/interactions/:interactionId/answer — Answer a question
+  // ============================================================
+  router.post('/interactions/:interactionId/answer', (req: Request, res: Response) => {
+    const { answer } = req.body;
+    if (!answer) {
+      res.status(400).json({ error: 'Missing required field: answer' });
+      return;
+    }
+    const success = submitAnswer(String(req.params.interactionId), answer);
+    if (!success) {
+      res.status(404).json({ error: 'Interaction not found or already answered' });
+      return;
+    }
+    res.json({ success: true });
+  });
+
+  // ============================================================
+  // GET /api/tasks/meta/roles — List available roles
+  // ============================================================
+  router.get('/meta/roles', (req: Request, res: Response) => {
+    res.json({ roles: listRoles() });
+  });
+
+  // ============================================================
+  // GET /api/tasks/meta/agents — List available agents
+  // ============================================================
+  router.get('/meta/agents', (req: Request, res: Response) => {
+    res.json({
+      agents: [
+        { value: 'claude-code', label: 'Claude Code', provider: 'anthropic' },
+        { value: 'codex', label: 'Codex CLI', provider: 'openai' },
+        { value: 'gemini', label: 'Gemini CLI', provider: 'google' },
+        { value: 'qwen', label: 'Qwen Code', provider: 'alibaba' },
+        { value: 'opencode', label: 'OpenCode', provider: 'anomaly' },
+      ],
+    });
+  });
+
+  // ============================================================
+  // GET /api/tasks/meta/config — Get task system config
+  // ============================================================
+  router.get('/meta/config', (req: Request, res: Response) => {
+    res.json({
+      maxConcurrent: getMaxConcurrent(),
+    });
+  });
+
+  // ============================================================
+  // POST /api/tasks/meta/config — Set task system config
+  // ============================================================
+  router.post('/meta/config', (req: Request, res: Response) => {
+    const { maxConcurrent } = req.body;
+    if (typeof maxConcurrent === 'number' && maxConcurrent > 0) {
+      setMaxConcurrent(maxConcurrent);
+    }
+    res.json({ maxConcurrent: getMaxConcurrent() });
+  });
+
+  // ============================================================
+  // WebSocket event stream (SSE) — /api/tasks/events
+  // ============================================================
+  router.get('/events', (req: Request, res: Response) => {
+    res.writeHead(200, {
+      'Content-Type': 'text/event-stream',
+      'Cache-Control': 'no-cache',
+      Connection: 'keep-alive',
+    });
+
+    const send = (data: any) => {
+      res.write(`data: ${JSON.stringify(data)}\n\n`);
+    };
+
+    const handlers: Record<string, (data: any) => void> = {
+      'task:status': (data) => send({ type: 'task:status', ...data }),
+      'task:log': (data) => send({ type: 'task:log', ...data }),
+      'task:completed': (data) => send({ type: 'task:completed', ...data }),
+      'task:failed': (data) => send({ type: 'task:failed', ...data }),
+      'task:cancelled': (data) => send({ type: 'task:cancelled', ...data }),
+    };
+
+    for (const [event, handler] of Object.entries(handlers)) {
+      taskEvents.on(event, handler);
+    }
+
+    req.on('close', () => {
+      for (const [event, handler] of Object.entries(handlers)) {
+        taskEvents.off(event, handler);
+      }
+    });
+  });
+
+  return router;
+}

package/server/modules/tasks/task-runner.ts

@@ -0,0 +1,219 @@
+// ============================================================
+// Task Runner — Dispatches tasks to the right agent runner
+// ============================================================
+
+import { updateTask, addLog, getTask } from './task-store.js';
+import { createWorktree, removeWorktree, getChangedFiles } from './git-worktree.js';
+import { askUser, cancelTaskInteractions } from './user-interaction.js';
+import { buildSummaryPrompt } from './prompt-builder.js';
+import { claudeRunner } from './runners/claude-runner.js';
+import { codexRunner } from './runners/codex-runner.js';
+import { createSpawnRunner } from './runners/spawn-runner.js';
+import type { Task, AgentType } from './task-types.js';
+import type { AgentRunner, RunContext } from './runners/types.js';
+
+// Runner registry
+const runners: Partial<Record<AgentType, AgentRunner>> = {
+  'claude-code': claudeRunner,
+  'codex': codexRunner,
+  'gemini': createSpawnRunner('gemini'),
+  'qwen': createSpawnRunner('qwen'),
+  'opencode': createSpawnRunner('opencode'),
+};
+
+// Track running tasks for cancellation
+const runningTasks = new Map<string, AbortController>();
+
+export function getAbortController(taskId: string): AbortController | undefined {
+  return runningTasks.get(taskId);
+}
+
+// Event emitter for WebSocket updates
+import { EventEmitter } from 'node:events';
+export const taskEvents = new EventEmitter();
+
+// ============================================================
+// Resolve project path from projectId
+// ============================================================
+
+export async function resolveProjectPath(projectId: string): Promise<string> {
+  // Try to import pixcode's project service
+  try {
+    const { getProjects } = await import('../../projects.js');
+    const projects = await getProjects();
+    const project = projects.find((p: any) => p.name === projectId || p.id === projectId);
+    if (project) {
+      return project.fullPath || project.path || projectId;
+    }
+  } catch {
+    // Fall through to treating projectId as a path
+  }
+  return projectId;
+}
+
+// ============================================================
+// Generate summary using Claude SDK
+// ============================================================
+
+async function generateSummary(task: Task, projectPath: string): Promise<string> {
+  try {
+    const { query } = await import('@anthropic-ai/claude-agent-sdk');
+    const summaryPrompt = buildSummaryPrompt(task);
+
+    const response = await query({
+      prompt: summaryPrompt,
+      options: {
+        cwd: task.worktreePath || projectPath,
+        maxTurns: 3,
+        model: 'claude-sonnet-4-20250514',
+        permissionMode: 'plan' as any,
+      },
+    });
+
+    let summary = '';
+    for await (const message of response) {
+      if (message.type === 'result' && message.subtype === 'success') {
+        const success = message as any;
+        summary = success.result || '';
+      }
+    }
+    return summary.slice(0, 2000);
+  } catch {
+    // Fallback: simple summary
+    return `Task "${task.title}" completed with ${task.agentType}.`;
+  }
+}
+
+// ============================================================
+// Run a single task
+// ============================================================
+
+export async function runTask(task: Task, projectPath: string): Promise<void> {
+  const abortController = new AbortController();
+  runningTasks.set(task.id, abortController);
+
+  const logTask = (level: 'info' | 'warn' | 'error' | 'debug', message: string) => {
+    addLog(task.id, level, message);
+    taskEvents.emit('task:log', { taskId: task.id, level, message });
+  };
+
+  const broadcastStatus = (status: string) => {
+    taskEvents.emit('task:status', { taskId: task.id, status });
+  };
+
+  try {
+    // Update status to RUNNING
+    updateTask(task.id, {
+      status: 'RUNNING',
+      startedAt: new Date().toISOString(),
+    });
+    broadcastStatus('RUNNING');
+
+    // Create git worktree for isolation
+    let worktreePath: string | undefined;
+    let branchName: string | undefined;
+    try {
+      const wt = createWorktree(task, projectPath);
+      worktreePath = wt.worktreePath;
+      branchName = wt.branchName;
+      updateTask(task.id, { worktreePath, branchName });
+    } catch (err: any) {
+      logTask('warn', `Could not create worktree: ${err.message}. Running in main directory.`);
+    }
+
+    // Get the right runner
+    const runner = runners[task.agentType];
+    if (!runner) {
+      throw new Error(`No runner registered for agent type: ${task.agentType}`);
+    }
+
+    // Build context
+    const ctx: RunContext = {
+      task: { ...task, worktreePath, branchName },
+      projectPath,
+      abortController,
+      logTask,
+      broadcastStatus,
+      askUser: (question: string, options?: string[]) =>
+        askUser(task.id, question, options, task.interactionTimeout),
+    };
+
+    // Run the agent
+    const { result, costUsd, tokenCount, sessionId } = await runner.run(ctx);
+
+    // Get changed files
+    const changedFiles = getChangedFiles({ ...task, worktreePath }, projectPath);
+
+    // Generate summary
+    logTask('info', 'Generating task summary...');
+    const summary = await generateSummary({ ...task, worktreePath }, projectPath);
+
+    // Update task with results
+    updateTask(task.id, {
+      status: 'COMPLETED',
+      result: result || 'Completed',
+      summary,
+      costUsd: costUsd || 0,
+      tokenCount: tokenCount || { input: 0, output: 0 },
+      sessionId,
+      changedFiles,
+      completedAt: new Date().toISOString(),
+    });
+    broadcastStatus('COMPLETED');
+
+    logTask('info', `Task completed. Cost: $${(costUsd || 0).toFixed(4)}, Files changed: ${changedFiles.length}`);
+
+    taskEvents.emit('task:completed', {
+      taskId: task.id,
+      summary,
+      costUsd: costUsd || 0,
+      changedFiles,
+    });
+
+    // Clean up worktree (optional — keep for review)
+    // removeWorktree(updatedTask, projectPath);
+
+  } catch (err: any) {
+    logTask('error', `Task failed: ${err.message}`);
+
+    updateTask(task.id, {
+      status: 'FAILED',
+      result: err.message,
+      completedAt: new Date().toISOString(),
+    });
+    broadcastStatus('FAILED');
+
+    taskEvents.emit('task:failed', { taskId: task.id, error: err.message });
+  } finally {
+    runningTasks.delete(task.id);
+  }
+}
+
+// ============================================================
+// Cancel a running task
+// ============================================================
+
+export async function cancelTask(taskId: string): Promise<boolean> {
+  const task = getTask(taskId);
+  if (!task) return false;
+
+  // Abort the running process
+  const abortController = runningTasks.get(taskId);
+  if (abortController) {
+    abortController.abort();
+  }
+
+  // Cancel pending interactions
+  cancelTaskInteractions(taskId);
+
+  // Update status
+  updateTask(taskId, {
+    status: 'CANCELLED',
+    completedAt: new Date().toISOString(),
+  });
+
+  taskEvents.emit('task:status', { taskId, status: 'CANCELLED' });
+  taskEvents.emit('task:cancelled', { taskId });
+
+  return true;
+}

package/server/modules/tasks/task-store.ts

@@ -0,0 +1,223 @@
+// ============================================================
+// Task Store — JSON-backed persistent storage for tasks
+// ============================================================
+
+import fs from 'node:fs';
+import path from 'node:path';
+import { createRequire } from 'node:module';
+import { randomUUID } from 'node:crypto';
+
+import { findAppRoot, getModuleDir } from '@/utils/runtime-paths.js';
+
+import type { Task, TaskInteraction, TaskLog, TaskStatus, CreateTaskInput } from './task-types.js';
+
+const require = createRequire(import.meta.url);
+const __dirname = getModuleDir(import.meta.url);
+const APP_ROOT = findAppRoot(__dirname);
+
+const TASKS_FILE = process.env.PIXCODE_TASKS_DB || path.join(APP_ROOT, 'tasks.json');
+
+interface TaskStoreData {
+  _version: number;
+  tasks: Task[];
+  interactions: TaskInteraction[];
+  logs: TaskLog[];
+}
+
+function loadStore(): TaskStoreData {
+  try {
+    if (fs.existsSync(TASKS_FILE)) {
+      const raw = fs.readFileSync(TASKS_FILE, 'utf-8');
+      const data = JSON.parse(raw);
+      if (!data.tasks) data.tasks = [];
+      if (!data.interactions) data.interactions = [];
+      if (!data.logs) data.logs = [];
+      return data;
+    }
+  } catch (err) {
+    console.error('[task-store] Failed to load tasks.json:', err);
+  }
+  return { _version: 1, tasks: [], interactions: [], logs: [] };
+}
+
+function saveStore(data: TaskStoreData): void {
+  try {
+    const dir = path.dirname(TASKS_FILE);
+    if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+    const tmp = TASKS_FILE + '.tmp';
+    fs.writeFileSync(tmp, JSON.stringify(data, null, 2));
+    fs.renameSync(tmp, TASKS_FILE);
+  } catch (err) {
+    console.error('[task-store] Failed to save tasks.json:', err);
+  }
+}
+
+// In-memory cache — loaded once, kept in sync
+let storeData = loadStore();
+
+// Rate-limit saves to avoid hammering disk on every log write
+let saveTimer: NodeJS.Timeout | null = null;
+function scheduleSave(): void {
+  if (saveTimer) return;
+  saveTimer = setTimeout(() => {
+    saveTimer = null;
+    saveStore(storeData);
+  }, 500);
+}
+
+// Force immediate save (for critical state changes)
+function flushSave(): void {
+  if (saveTimer) {
+    clearTimeout(saveTimer);
+    saveTimer = null;
+  }
+  saveStore(storeData);
+}
+
+// ============================================================
+// Task CRUD
+// ============================================================
+
+export function createTask(input: CreateTaskInput): Task {
+  const now = new Date().toISOString();
+  const task: Task = {
+    id: randomUUID(),
+    projectId: input.projectId,
+    title: input.title,
+    prompt: input.prompt,
+    status: 'PENDING',
+    agentType: input.agentType,
+    model: input.model,
+    role: input.role || 'fullstack',
+    priority: input.priority || 'normal',
+    predecessorTaskId: input.predecessorTaskId,
+    continueSession: input.continueSession || false,
+    maxBudgetUsd: input.maxBudgetUsd,
+    interactionTimeout: input.interactionTimeout || 300_000,
+    thinkingEnabled: input.thinkingEnabled || false,
+    permissionMode: input.permissionMode,
+    costUsd: 0,
+    tokenCount: { input: 0, output: 0 },
+    createdAt: now,
+  };
+  storeData.tasks.push(task);
+  flushSave();
+  return task;
+}
+
+export function getTask(id: string): Task | undefined {
+  return storeData.tasks.find((t) => t.id === id);
+}
+
+export function getTasksByProject(projectId: string): Task[] {
+  return storeData.tasks
+    .filter((t) => t.projectId === projectId)
+    .sort((a, b) => new Date(b.createdAt).getTime() - new Date(a.createdAt).getTime());
+}
+
+export function getAllTasks(limit = 100): Task[] {
+  return storeData.tasks
+    .slice()
+    .sort((a, b) => new Date(b.createdAt).getTime() - new Date(a.createdAt).getTime())
+    .slice(0, limit);
+}
+
+export function updateTask(id: string, updates: Partial<Task>): Task | undefined {
+  const task = storeData.tasks.find((t) => t.id === id);
+  if (!task) return undefined;
+  Object.assign(task, updates, { updatedAt: new Date().toISOString() });
+  flushSave();
+  return task;
+}
+
+export function deleteTask(id: string): boolean {
+  const idx = storeData.tasks.findIndex((t) => t.id === id);
+  if (idx === -1) return false;
+  storeData.tasks.splice(idx, 1);
+  // Also remove related interactions and logs
+  storeData.interactions = storeData.interactions.filter((i) => i.taskId !== id);
+  storeData.logs = storeData.logs.filter((l) => l.taskId !== id);
+  flushSave();
+  return true;
+}
+
+export function getTasksByStatus(status: TaskStatus): Task[] {
+  return storeData.tasks.filter((t) => t.status === status);
+}
+
+export function getPredecessor(task: Task): Task | undefined {
+  if (!task.predecessorTaskId) return undefined;
+  return getTask(task.predecessorTaskId);
+}
+
+// ============================================================
+// Task Interactions
+// ============================================================
+
+export function createInteraction(taskId: string, question: string, options?: string[]): TaskInteraction {
+  const interaction: TaskInteraction = {
+    id: randomUUID(),
+    taskId,
+    question,
+    options,
+    status: 'pending',
+    createdAt: new Date().toISOString(),
+  };
+  storeData.interactions.push(interaction);
+  flushSave();
+  return interaction;
+}
+
+export function getPendingInteractions(taskId: string): TaskInteraction[] {
+  return storeData.interactions.filter(
+    (i) => i.taskId === taskId && i.status === 'pending',
+  );
+}
+
+export function answerInteraction(interactionId: string, answer: string): TaskInteraction | undefined {
+  const interaction = storeData.interactions.find((i) => i.id === interactionId);
+  if (!interaction) return undefined;
+  interaction.answer = answer;
+  interaction.status = 'answered';
+  interaction.answeredAt = new Date().toISOString();
+  flushSave();
+  return interaction;
+}
+
+export function timeoutInteraction(interactionId: string): void {
+  const interaction = storeData.interactions.find((i) => i.id === interactionId);
+  if (interaction && interaction.status === 'pending') {
+    interaction.status = 'timeout';
+    flushSave();
+  }
+}
+
+// ============================================================
+// Task Logs
+// ============================================================
+
+export function addLog(taskId: string, level: TaskLog['level'], message: string): void {
+  storeData.logs.push({
+    id: randomUUID(),
+    taskId,
+    level,
+    message,
+    timestamp: new Date().toISOString(),
+  });
+  scheduleSave();
+}
+
+export function getLogs(taskId: string, limit = 200): TaskLog[] {
+  return storeData.logs
+    .filter((l) => l.taskId === taskId)
+    .sort((a, b) => new Date(a.timestamp).getTime() - new Date(b.timestamp).getTime())
+    .slice(-limit);
+}
+
+// ============================================================
+// Utility
+// ============================================================
+
+export function getTaskStorePath(): string {
+  return TASKS_FILE;
+}