micode 0.7.0 → 0.7.2

package/src/tools/ast-grep/index.ts

@@ -0,0 +1,189 @@
+import { spawn, which } from "bun";
+import { tool } from "@opencode-ai/plugin/tool";
+
+/**
+ * Check if ast-grep CLI (sg) is available on the system.
+ * Returns installation instructions if not found.
+ */
+export async function checkAstGrepAvailable(): Promise<{ available: boolean; message?: string }> {
+  const sgPath = which("sg");
+  if (sgPath) {
+    return { available: true };
+  }
+  return {
+    available: false,
+    message:
+      "ast-grep CLI (sg) not found. AST-aware search/replace will not work.\n" +
+      "Install with one of:\n" +
+      "  npm install -g @ast-grep/cli\n" +
+      "  cargo install ast-grep --locked\n" +
+      "  brew install ast-grep",
+  };
+}
+
+const LANGUAGES = [
+  "c",
+  "cpp",
+  "csharp",
+  "css",
+  "dart",
+  "elixir",
+  "go",
+  "haskell",
+  "html",
+  "java",
+  "javascript",
+  "json",
+  "kotlin",
+  "lua",
+  "php",
+  "python",
+  "ruby",
+  "rust",
+  "scala",
+  "sql",
+  "swift",
+  "tsx",
+  "typescript",
+  "yaml",
+] as const;
+
+interface Match {
+  file: string;
+  range: { start: { line: number; column: number }; end: { line: number; column: number } };
+  text: string;
+  replacement?: string;
+}
+
+async function runSg(args: string[]): Promise<{ matches: Match[]; error?: string }> {
+  try {
+    const proc = spawn(["sg", ...args], {
+      stdout: "pipe",
+      stderr: "pipe",
+    });
+
+    const [stdout, stderr, exitCode] = await Promise.all([
+      new Response(proc.stdout).text(),
+      new Response(proc.stderr).text(),
+      proc.exited,
+    ]);
+
+    if (exitCode !== 0 && !stdout.trim()) {
+      if (stderr.includes("No files found")) {
+        return { matches: [] };
+      }
+      return { matches: [], error: stderr.trim() || `Exit code ${exitCode}` };
+    }
+
+    if (!stdout.trim()) return { matches: [] };
+
+    try {
+      const matches = JSON.parse(stdout) as Match[];
+      return { matches };
+    } catch {
+      return { matches: [], error: "Failed to parse output" };
+    }
+  } catch (e) {
+    const err = e as Error;
+    if (err.message?.includes("ENOENT")) {
+      return {
+        matches: [],
+        error:
+          "ast-grep CLI not found. Install with:\n" +
+          "  npm install -g @ast-grep/cli\n" +
+          "  cargo install ast-grep --locked\n" +
+          "  brew install ast-grep",
+      };
+    }
+    return { matches: [], error: err.message };
+  }
+}
+
+function formatMatches(matches: Match[], isDryRun = false): string {
+  if (matches.length === 0) return "No matches found";
+
+  const MAX = 100;
+  const truncated = matches.length > MAX;
+  const shown = matches.slice(0, MAX);
+
+  const lines = shown.map((m) => {
+    const loc = `${m.file}:${m.range.start.line}:${m.range.start.column}`;
+    const text = m.text.length > 100 ? `${m.text.slice(0, 100)}...` : m.text;
+    if (isDryRun && m.replacement) {
+      return `${loc}\n  - ${text}\n  + ${m.replacement}`;
+    }
+    return `${loc}: ${text}`;
+  });
+
+  if (truncated) {
+    lines.unshift(`Found ${matches.length} matches (showing first ${MAX}):`);
+  }
+
+  return lines.join("\n");
+}
+
+export const ast_grep_search = tool({
+  description:
+    "Search code patterns using AST-aware matching. " +
+    "Use meta-variables: $VAR (single node), $$$ (multiple nodes). " +
+    "Patterns must be complete AST nodes. " +
+    "Examples: 'console.log($MSG)', 'def $FUNC($$$):', 'async function $NAME($$$)'",
+  args: {
+    pattern: tool.schema.string().describe("AST pattern with meta-variables"),
+    lang: tool.schema.enum(LANGUAGES).describe("Target language"),
+    paths: tool.schema.array(tool.schema.string()).optional().describe("Paths to search"),
+  },
+  execute: async (args) => {
+    const sgArgs = ["run", "-p", args.pattern, "--lang", args.lang, "--json=compact"];
+    if (args.paths?.length) {
+      sgArgs.push(...args.paths);
+    } else {
+      sgArgs.push(".");
+    }
+
+    const result = await runSg(sgArgs);
+    if (result.error) return `Error: ${result.error}`;
+    return formatMatches(result.matches);
+  },
+});
+
+export const ast_grep_replace = tool({
+  description:
+    "Replace code patterns with AST-aware rewriting. " +
+    "Dry-run by default. Use meta-variables in rewrite to preserve matched content. " +
+    "Example: pattern='console.log($MSG)' rewrite='logger.info($MSG)'",
+  args: {
+    pattern: tool.schema.string().describe("AST pattern to match"),
+    rewrite: tool.schema.string().describe("Replacement pattern"),
+    lang: tool.schema.enum(LANGUAGES).describe("Target language"),
+    paths: tool.schema.array(tool.schema.string()).optional().describe("Paths to search"),
+    apply: tool.schema.boolean().optional().describe("Apply changes (default: false, dry-run)"),
+  },
+  execute: async (args) => {
+    const sgArgs = ["run", "-p", args.pattern, "-r", args.rewrite, "--lang", args.lang, "--json=compact"];
+
+    if (args.apply) {
+      sgArgs.push("--update-all");
+    }
+
+    if (args.paths?.length) {
+      sgArgs.push(...args.paths);
+    } else {
+      sgArgs.push(".");
+    }
+
+    const result = await runSg(sgArgs);
+    if (result.error) return `Error: ${result.error}`;
+
+    const isDryRun = !args.apply;
+    const output = formatMatches(result.matches, isDryRun);
+
+    if (isDryRun && result.matches.length > 0) {
+      return `${output}\n\n(Dry run - use apply=true to apply changes)`;
+    }
+    if (args.apply && result.matches.length > 0) {
+      return `Applied ${result.matches.length} replacements:\n${output}`;
+    }
+    return output;
+  },
+});

package/src/tools/background-task/manager.ts

@@ -0,0 +1,397 @@
+import type { PluginInput } from "@opencode-ai/plugin";
+import type {
+  BackgroundTask,
+  BackgroundTaskInput,
+  SessionCreateResponse,
+  SessionStatusResponse,
+  SessionMessagesResponse,
+} from "./types";
+
+const POLL_INTERVAL_MS = 2000;
+const TASK_TTL_MS = 60 * 60 * 1000; // 1 hour
+
+function generateTaskId(): string {
+  const chars = "abcdefghijklmnopqrstuvwxyz0123456789";
+  let result = "bg_";
+  for (let i = 0; i < 8; i++) {
+    result += chars.charAt(Math.floor(Math.random() * chars.length));
+  }
+  return result;
+}
+
+function formatDuration(start: Date, end?: Date): string {
+  const ms = (end || new Date()).getTime() - start.getTime();
+  const seconds = Math.floor(ms / 1000);
+  const minutes = Math.floor(seconds / 60);
+
+  if (minutes > 0) {
+    return `${minutes}m ${seconds % 60}s`;
+  }
+  return `${seconds}s`;
+}
+
+export class BackgroundTaskManager {
+  private tasks: Map<string, BackgroundTask> = new Map();
+  private notifications: Map<string, BackgroundTask[]> = new Map();
+  private pollingInterval?: ReturnType<typeof setInterval>;
+  private ctx: PluginInput;
+
+  constructor(ctx: PluginInput) {
+    this.ctx = ctx;
+  }
+
+  async launch(input: BackgroundTaskInput): Promise<BackgroundTask> {
+    const taskId = generateTaskId();
+
+    // Create new session for background task
+    const sessionResp = await this.ctx.client.session.create({
+      body: {},
+      query: { directory: this.ctx.directory },
+    });
+
+    const sessionData = sessionResp as SessionCreateResponse;
+    const sessionID = sessionData.data?.id;
+
+    if (!sessionID) {
+      throw new Error("Failed to create background session");
+    }
+
+    const task: BackgroundTask = {
+      id: taskId,
+      sessionID,
+      parentSessionID: input.parentSessionID,
+      parentMessageID: input.parentMessageID,
+      description: input.description,
+      prompt: input.prompt,
+      agent: input.agent,
+      status: "running",
+      startedAt: new Date(),
+      progress: {
+        toolCalls: 0,
+        lastUpdate: new Date(),
+      },
+    };
+
+    this.tasks.set(taskId, task);
+
+    // Fire-and-forget prompt
+    this.ctx.client.session
+      .prompt({
+        path: { id: sessionID },
+        body: {
+          parts: [{ type: "text", text: input.prompt }],
+          agent: input.agent,
+        },
+        query: { directory: this.ctx.directory },
+      })
+      .catch((error) => {
+        console.error(`[background-task] Failed to prompt session ${sessionID}:`, error);
+        task.status = "error";
+        task.error = error instanceof Error ? error.message : String(error);
+        task.completedAt = new Date();
+        this.markForNotification(task);
+      });
+
+    // Start polling if not already
+    this.startPolling();
+
+    return task;
+  }
+
+  async cancel(taskId: string): Promise<boolean> {
+    const task = this.tasks.get(taskId);
+    if (!task || task.status !== "running") {
+      return false;
+    }
+
+    try {
+      // Fire-and-forget abort
+      this.ctx.client.session
+        .abort({
+          path: { id: task.sessionID },
+          query: { directory: this.ctx.directory },
+        })
+        .catch((error) => {
+          console.error(`[background-task] Failed to abort session ${task.sessionID}:`, error);
+        });
+
+      task.status = "cancelled";
+      task.completedAt = new Date();
+      this.markForNotification(task);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  async cancelAll(): Promise<number> {
+    let cancelled = 0;
+    for (const task of this.tasks.values()) {
+      if (task.status === "running") {
+        if (await this.cancel(task.id)) {
+          cancelled++;
+        }
+      }
+    }
+    return cancelled;
+  }
+
+  getTask(taskId: string): BackgroundTask | undefined {
+    return this.tasks.get(taskId);
+  }
+
+  getAllTasks(): BackgroundTask[] {
+    return Array.from(this.tasks.values());
+  }
+
+  /**
+   * Poll all running tasks and update their status.
+   * Called by background_list to ensure fresh status.
+   */
+  async refreshTaskStatus(): Promise<void> {
+    const runningTasks = this.getRunningTasks();
+    if (runningTasks.length === 0) return;
+
+    try {
+      // Get all session statuses in one call
+      const resp = await this.ctx.client.session.status({
+        query: { directory: this.ctx.directory },
+      });
+
+      const statusResp = resp as SessionStatusResponse;
+      const statusMap = statusResp.data || {};
+
+      for (const task of runningTasks) {
+        const sessionStatus = statusMap[task.sessionID];
+        const statusType = sessionStatus?.type;
+
+        // Store last known session status for debugging
+        (task as BackgroundTask & { _sessionStatus?: string })._sessionStatus = statusType;
+
+        if (statusType === "idle" || statusType === undefined) {
+          // Session is idle OR not in status map (likely finished and cleaned up)
+          // Try to get result - if successful, mark as completed
+          const result = await this.fetchTaskResult(task);
+          if (result !== undefined || statusType === "idle") {
+            task.status = "completed";
+            task.completedAt = new Date();
+            task.result = result;
+          }
+          // If result is undefined and statusType is undefined, keep waiting
+          // (might be a timing issue with status propagation)
+        }
+      }
+    } catch (error) {
+      console.error("[background-task] Failed to refresh task status:", error);
+      // Don't mark all tasks as error - they may still be running
+    }
+  }
+
+  getRunningTasks(): BackgroundTask[] {
+    return this.getAllTasks().filter((t) => t.status === "running");
+  }
+
+  async getTaskResult(taskId: string): Promise<string | undefined> {
+    const task = this.tasks.get(taskId);
+    if (!task || task.status === "running") {
+      return undefined;
+    }
+
+    if (task.result) {
+      return task.result;
+    }
+
+    const result = await this.fetchTaskResult(task);
+    if (result !== undefined) {
+      task.result = result;
+    }
+    return result;
+  }
+
+  /**
+   * Fetch result from session messages without checking task status.
+   * Used during polling to check if a session has completed.
+   */
+  private async fetchTaskResult(task: BackgroundTask): Promise<string | undefined> {
+    try {
+      const resp = await this.ctx.client.session.messages({
+        path: { id: task.sessionID },
+        query: { directory: this.ctx.directory },
+      });
+
+      const messagesResp = resp as SessionMessagesResponse;
+      const messages = messagesResp.data || [];
+      const lastAssistant = [...messages].reverse().find((m) => m.info?.role === "assistant");
+
+      if (lastAssistant) {
+        const textParts = lastAssistant.parts?.filter((p) => p.type === "text") || [];
+        return textParts.map((p) => p.text || "").join("\n");
+      }
+    } catch (error) {
+      console.error(`[background-task] Failed to fetch result for task ${task.id}:`, error);
+    }
+
+    return undefined;
+  }
+
+  formatTaskStatus(task: BackgroundTask): string {
+    const duration = formatDuration(task.startedAt, task.completedAt);
+    const status = task.status.toUpperCase();
+
+    let output = `## Task: ${task.description}\n\n`;
+    output += `| Field | Value |\n|-------|-------|\n`;
+    output += `| ID | ${task.id} |\n`;
+    output += `| Status | ${status} |\n`;
+    output += `| Agent | ${task.agent} |\n`;
+    output += `| Duration | ${duration} |\n`;
+
+    if (task.progress) {
+      output += `| Tool Calls | ${task.progress.toolCalls} |\n`;
+      if (task.progress.lastTool) {
+        output += `| Last Tool | ${task.progress.lastTool} |\n`;
+      }
+    }
+
+    if (task.error) {
+      output += `\n### Error\n${task.error}\n`;
+    }
+
+    return output;
+  }
+
+  private startPolling(): void {
+    if (this.pollingInterval) return;
+
+    this.pollingInterval = setInterval(() => {
+      this.pollRunningTasks();
+    }, POLL_INTERVAL_MS);
+  }
+
+  private stopPolling(): void {
+    if (this.pollingInterval) {
+      clearInterval(this.pollingInterval);
+      this.pollingInterval = undefined;
+    }
+  }
+
+  private cleanupOldTasks(): void {
+    const now = Date.now();
+    for (const [taskId, task] of this.tasks) {
+      // Only cleanup completed/cancelled/error tasks
+      if (task.status === "running") continue;
+
+      const completedAt = task.completedAt?.getTime() || 0;
+      if (now - completedAt > TASK_TTL_MS) {
+        this.tasks.delete(taskId);
+      }
+    }
+  }
+
+  private async pollRunningTasks(): Promise<void> {
+    // Cleanup old completed tasks to prevent memory leak
+    this.cleanupOldTasks();
+
+    const runningTasks = this.getRunningTasks();
+
+    if (runningTasks.length === 0) {
+      this.stopPolling();
+      return;
+    }
+
+    try {
+      // Get all session statuses in one call
+      const resp = await this.ctx.client.session.status({
+        query: { directory: this.ctx.directory },
+      });
+
+      const statusResp = resp as SessionStatusResponse;
+      const statusMap = statusResp.data || {};
+
+      for (const task of runningTasks) {
+        const sessionStatus = statusMap[task.sessionID];
+        const statusType = sessionStatus?.type;
+
+        console.log(`[background-task] Poll ${task.id}: session=${task.sessionID} status=${statusType}`);
+
+        if (statusType === "idle" || statusType === undefined) {
+          // Session is idle OR not in status map (likely finished and cleaned up)
+          // Try to get result - if successful, mark as completed
+          const result = await this.fetchTaskResult(task);
+          if (result !== undefined || statusType === "idle") {
+            task.status = "completed";
+            task.completedAt = new Date();
+            task.result = result;
+            this.markForNotification(task);
+
+            await this.ctx.client.tui
+              .showToast({
+                body: {
+                  title: "Background Task Complete",
+                  message: task.description,
+                  variant: "success",
+                  duration: 5000,
+                },
+              })
+              .catch((error) => {
+                console.error(`[background-task] Failed to show toast for task ${task.id}:`, error);
+              });
+          }
+          // If result is undefined and statusType is undefined, keep waiting
+        }
+      }
+    } catch (error) {
+      console.error("[background-task] Failed to poll tasks:", error);
+      // Don't mark tasks as error - they may still be running, just can't check
+    }
+  }
+
+  private markForNotification(task: BackgroundTask): void {
+    const existing = this.notifications.get(task.parentSessionID) || [];
+    existing.push(task);
+    this.notifications.set(task.parentSessionID, existing);
+  }
+
+  getPendingNotifications(parentSessionID: string): BackgroundTask[] {
+    const notifications = this.notifications.get(parentSessionID) || [];
+    this.notifications.delete(parentSessionID);
+    return notifications;
+  }
+
+  handleEvent(event: { type: string; properties?: unknown }): void {
+    const props = event.properties as Record<string, unknown> | undefined;
+
+    // Track tool usage for progress
+    if (event.type === "message.part.updated") {
+      const info = props?.info as Record<string, unknown> | undefined;
+      const sessionID = info?.sessionID as string | undefined;
+      const partType = info?.type as string | undefined;
+
+      if (sessionID && partType === "tool_use") {
+        for (const task of this.tasks.values()) {
+          if (task.sessionID === sessionID && task.status === "running") {
+            if (!task.progress) {
+              task.progress = { toolCalls: 0, lastUpdate: new Date() };
+            }
+            task.progress.toolCalls++;
+            task.progress.lastTool = (info?.name as string) || undefined;
+            task.progress.lastUpdate = new Date();
+            break;
+          }
+        }
+      }
+    }
+
+    // Cleanup on session delete
+    if (event.type === "session.deleted") {
+      const sessionInfo = props?.info as { id?: string } | undefined;
+      if (sessionInfo?.id) {
+        for (const task of this.tasks.values()) {
+          if (task.sessionID === sessionInfo.id && task.status === "running") {
+            task.status = "cancelled";
+            task.completedAt = new Date();
+          }
+        }
+      }
+    }
+  }
+}