@locusai/server 0.1.0

package/src/task-processor.ts

@@ -0,0 +1,304 @@
+import type { Database } from "bun:sqlite";
+import { readFileSync } from "node:fs";
+
+export interface ProcessorConfig {
+  ciPresetsPath: string;
+  repoPath: string;
+}
+
+interface RawTask {
+  id: number;
+  title: string;
+  description: string;
+  labels: string;
+  acceptanceChecklist: string;
+}
+
+export class TaskProcessor {
+  constructor(
+    private db: Database,
+    private config: ProcessorConfig
+  ) {}
+
+  async onStatusChanged(taskId: string, from: string, to: string) {
+    console.log(`[TaskProcessor] Task ${taskId} moved from ${from} to ${to}`);
+
+    if (to === "IN_PROGRESS") {
+      await this.handleInProgress(taskId);
+    } else if (to === "VERIFICATION") {
+      await this.handleVerification(taskId);
+    }
+  }
+
+  private async handleInProgress(taskId: string) {
+    try {
+      const rawTask = this.db
+        .prepare("SELECT * FROM tasks WHERE id = ?")
+        .get(taskId) as RawTask | undefined;
+      if (!rawTask) return;
+
+      const labels = JSON.parse(rawTask.labels || "[]");
+
+      // 1. Create a "Technical Implementation Draft" artifact
+      await this.createTechnicalDraft(
+        taskId,
+        rawTask.title,
+        rawTask.description
+      );
+
+      // 2. Update acceptance checklist if empty
+      await this.initChecklist(taskId, rawTask.acceptanceChecklist);
+
+      // 3. (Optional) Trigger CI if requested via labels
+      const presets = JSON.parse(
+        readFileSync(this.config.ciPresetsPath, "utf-8")
+      );
+      if (presets.quick && labels.includes("auto-ci")) {
+        await this.runCi(taskId, "quick");
+      }
+
+      // 4. Create git branch
+      const slug = this.slugify(rawTask.title);
+      const branchName = `task/${taskId}-${slug}`;
+      await this.createBranch(branchName);
+    } catch (err) {
+      console.error(
+        "[TaskProcessor] Failed to process In Progress transition:",
+        err
+      );
+    }
+  }
+
+  private async createTechnicalDraft(
+    taskId: string,
+    title: string,
+    description: string
+  ) {
+    const draftContent = `
+# Implementation Plan: ${title}
+
+## Objective
+${description || "No description provided."}
+
+## Suggested Steps
+1. [ ] Analyze current codebase for related components.
+2. [ ] Research potential side effects of the change.
+3. [ ] Implement the core logic.
+4. [ ] Run automated tests to verify.
+5. [ ] Perform manual verification.
+
+## Quality Gates
+- [ ] Code follows project style guidelines.
+- [ ] No regression in existing functionality.
+- [ ] Documentation updated if necessary.
+
+---
+*This is an automatically generated draft to kickstart the task.*
+`.trim();
+
+    this.db
+      .prepare(
+        `
+      INSERT INTO artifacts (taskId, type, title, contentText, createdBy, createdAt)
+      VALUES (?, ?, ?, ?, ?, ?)
+    `
+      )
+      .run(
+        taskId,
+        "TECH_DRAFT",
+        `Draft: ${title}`,
+        draftContent,
+        "system",
+        Date.now()
+      );
+  }
+
+  private async initChecklist(taskId: string, currentChecklistJson: string) {
+    const current = JSON.parse(currentChecklistJson || "[]");
+    if (current.length > 0) return; // Don't overwrite existing checklist
+
+    const defaultChecklist = [
+      { id: "step-1", text: "Research & Planning", done: false },
+      { id: "step-2", text: "Implementation", done: false },
+      { id: "step-3", text: "Testing & Verification", done: false },
+    ];
+
+    this.db
+      .prepare(
+        "UPDATE tasks SET acceptanceChecklist = ?, updatedAt = ? WHERE id = ?"
+      )
+      .run(JSON.stringify(defaultChecklist), Date.now(), taskId);
+
+    this.db
+      .prepare(
+        "INSERT INTO events (taskId, type, payload, createdAt) VALUES (?, ?, ?, ?)"
+      )
+      .run(
+        taskId,
+        "CHECKLIST_INITIALIZED",
+        JSON.stringify({ itemCount: defaultChecklist.length }),
+        Date.now()
+      );
+  }
+
+  private async runCi(taskId: string, preset: string) {
+    const presets = JSON.parse(
+      readFileSync(this.config.ciPresetsPath, "utf-8")
+    );
+    const commands = presets[preset];
+    if (!commands) return;
+
+    let allOk = true;
+    let combinedOutput = "";
+
+    for (const cmd of commands) {
+      try {
+        if (/[;&|><$`\n]/.test(cmd)) throw new Error("Invalid command");
+
+        const proc = Bun.spawn(cmd.split(" "), {
+          cwd: this.config.repoPath,
+          stdout: "pipe",
+          stderr: "pipe",
+        });
+
+        const stdout = await new Response(proc.stdout).text();
+        const stderr = await new Response(proc.stderr).text();
+        const exitCode = await proc.exited;
+
+        combinedOutput += `\n> ${cmd}\n${stdout}${stderr}\n`;
+        if (exitCode !== 0) allOk = false;
+      } catch (err: unknown) {
+        allOk = false;
+        const message = err instanceof Error ? err.message : String(err);
+        combinedOutput += `\n> ${cmd}\nError: ${message}\n`;
+      }
+    }
+
+    const summary = allOk ? "All checks passed" : "Some checks failed";
+    const now = Date.now();
+
+    // Save CI_OUTPUT artifact
+    this.db
+      .prepare(`
+      INSERT INTO artifacts (taskId, type, title, contentText, createdBy, createdAt)
+      VALUES (?, ?, ?, ?, ?, ?)
+    `)
+      .run(
+        taskId,
+        "CI_OUTPUT",
+        `Auto-CI: ${preset}`,
+        combinedOutput,
+        "system",
+        now
+      );
+
+    // Record CI_RAN event
+    this.db
+      .prepare(
+        "INSERT INTO events (taskId, type, payload, createdAt) VALUES (?, ?, ?, ?)"
+      )
+      .run(
+        taskId,
+        "CI_RAN",
+        JSON.stringify({
+          preset,
+          ok: allOk,
+          summary,
+          source: "auto-processor",
+        }),
+        now
+      );
+  }
+
+  private async handleVerification(taskId: string) {
+    try {
+      const rawTask = this.db
+        .prepare("SELECT * FROM tasks WHERE id = ?")
+        .get(taskId) as RawTask | undefined;
+      if (!rawTask) return;
+
+      const slug = this.slugify(rawTask.title);
+      const branchName = `task/${taskId}-${slug}`;
+
+      // Create PR (Assumes agent has pushed the branch)
+      await this.createPullRequest(
+        branchName,
+        rawTask.title,
+        rawTask.description
+      );
+    } catch (err) {
+      console.error(
+        "[TaskProcessor] Failed to process Verification transition:",
+        err
+      );
+    }
+  }
+
+  private slugify(text: string): string {
+    return text
+      .toLowerCase()
+      .replace(/[^a-z0-9]+/g, "-")
+      .replace(/^-|-$/g, "");
+  }
+
+  private async createBranch(branchName: string) {
+    console.log(`[TaskProcessor] Creating branch ${branchName}...`);
+    try {
+      const checkProc = Bun.spawn(
+        ["git", "show-ref", "--verify", `refs/heads/${branchName}`],
+        {
+          cwd: this.config.repoPath,
+          stdout: "ignore",
+          stderr: "ignore",
+        }
+      );
+      const exists = (await checkProc.exited) === 0;
+
+      if (!exists) {
+        // Create branch from HEAD but DO NOT checkout
+        // This is safe for parallel agents working in other directories
+        await Bun.spawn(["git", "branch", branchName], {
+          cwd: this.config.repoPath,
+          stdout: "ignore",
+          stderr: "pipe",
+        }).exited;
+      }
+    } catch (error) {
+      console.error(`[TaskProcessor] Git branch operation failed: ${error}`);
+    }
+  }
+
+  private async createPullRequest(
+    branchName: string,
+    title: string,
+    description: string
+  ) {
+    console.log(`[TaskProcessor] Creating PR for ${branchName}...`);
+    try {
+      // Create PR using gh cli (headless)
+      // We assume the branch has been pushed by the agent
+      await Bun.spawn(
+        [
+          "gh",
+          "pr",
+          "create",
+          "--title",
+          `[Task] ${title}`,
+          "--body",
+          description || "Task implementation",
+          "--head",
+          branchName,
+          "--base",
+          "master",
+        ],
+        {
+          cwd: this.config.repoPath,
+          stdout: "ignore",
+          stderr: "pipe",
+        }
+      ).exited;
+    } catch (error) {
+      console.error(`[TaskProcessor] PR creation failed: ${error}`);
+    }
+  }
+}

package/tsconfig.json

@@ -0,0 +1,8 @@
+{
+  "extends": "../../tsconfig.base.json",
+  "compilerOptions": {
+    "outDir": "dist",
+    "rootDir": "src"
+  },
+  "include": ["src"]
+}