@cortexkit/aft-opencode 0.9.1 → 0.10.1

package/src/shared/rpc-server.ts

@@ -0,0 +1,135 @@
+import { mkdirSync, renameSync, unlinkSync, writeFileSync } from "node:fs";
+import { createServer, type IncomingMessage, type Server, type ServerResponse } from "node:http";
+import { dirname } from "node:path";
+import { log, warn } from "../logger.js";
+import { rpcPortFilePath } from "./rpc-utils.js";
+
+type RpcHandler = (params: Record<string, unknown>) => Promise<Record<string, unknown>>;
+
+export class AftRpcServer {
+  private server: Server | null = null;
+  private port = 0;
+  private handlers = new Map<string, RpcHandler>();
+  private portFilePath: string;
+
+  constructor(storageDir: string, directory: string) {
+    this.portFilePath = rpcPortFilePath(storageDir, directory);
+  }
+
+  /** Register an RPC method handler. */
+  handle(method: string, handler: RpcHandler): void {
+    this.handlers.set(method, handler);
+  }
+
+  /** Start the server on a random port, write port to disk. */
+  async start(): Promise<number> {
+    return new Promise((resolve, reject) => {
+      const server = createServer((req, res) => this.dispatch(req, res));
+
+      server.on("error", (err) => {
+        warn(`RPC server error: ${err.message}`);
+        reject(err);
+      });
+
+      server.listen(0, "127.0.0.1", () => {
+        const addr = server.address();
+        if (!addr || typeof addr === "string") {
+          reject(new Error("Failed to get server address"));
+          return;
+        }
+        this.port = addr.port;
+        this.server = server;
+
+        // Write port file atomically
+        try {
+          const dir = dirname(this.portFilePath);
+          mkdirSync(dir, { recursive: true });
+          const tmpPath = `${this.portFilePath}.tmp`;
+          writeFileSync(tmpPath, String(this.port), "utf-8");
+          renameSync(tmpPath, this.portFilePath);
+          log(`RPC server listening on 127.0.0.1:${this.port}`);
+        } catch (err) {
+          warn(`Failed to write RPC port file: ${err}`);
+        }
+
+        resolve(this.port);
+      });
+
+      // Don't keep the process alive just for the RPC server
+      server.unref();
+    });
+  }
+
+  /** Stop the server and clean up port file. */
+  stop(): void {
+    if (this.server) {
+      this.server.close();
+      this.server = null;
+    }
+    try {
+      unlinkSync(this.portFilePath);
+    } catch {
+      // ignore
+    }
+  }
+
+  private dispatch(req: IncomingMessage, res: ServerResponse): void {
+    const url = req.url ?? "";
+
+    if (req.method === "GET" && url === "/health") {
+      res.writeHead(200, { "Content-Type": "application/json" });
+      res.end(JSON.stringify({ ok: true, pid: process.pid }));
+      return;
+    }
+
+    if (req.method !== "POST" || !url.startsWith("/rpc/")) {
+      res.writeHead(404);
+      res.end("Not Found");
+      return;
+    }
+
+    const method = url.slice(5);
+    const handler = this.handlers.get(method);
+    if (!handler) {
+      res.writeHead(404, { "Content-Type": "application/json" });
+      res.end(JSON.stringify({ error: `Unknown method: ${method}` }));
+      return;
+    }
+
+    let body = "";
+    req.on("data", (chunk: Buffer) => {
+      body += chunk.toString();
+      if (body.length > 1_048_576) {
+        res.writeHead(413);
+        res.end("Request too large");
+        req.destroy();
+      }
+    });
+
+    req.on("end", () => {
+      let params: Record<string, unknown> = {};
+      try {
+        if (body.length > 0) {
+          params = JSON.parse(body);
+        }
+      } catch {
+        res.writeHead(400, { "Content-Type": "application/json" });
+        res.end(JSON.stringify({ error: "Invalid JSON" }));
+        return;
+      }
+
+      log(`RPC call: ${method} params=${JSON.stringify(params).slice(0, 200)}`);
+      handler(params)
+        .then((result) => {
+          log(`RPC result: ${method} => ${JSON.stringify(result).slice(0, 200)}`);
+          res.writeHead(200, { "Content-Type": "application/json" });
+          res.end(JSON.stringify(result));
+        })
+        .catch((err) => {
+          log(`RPC error: ${method} => ${err}`);
+          res.writeHead(500, { "Content-Type": "application/json" });
+          res.end(JSON.stringify({ error: String(err) }));
+        });
+    });
+  }
+}

package/src/shared/rpc-utils.ts

@@ -0,0 +1,21 @@
+import { createHash } from "node:crypto";
+import { join } from "node:path";
+
+/**
+ * Compute a stable hash for a project directory.
+ * Used to scope RPC port files per-project so multiple
+ * OpenCode Desktop instances don't overwrite each other.
+ */
+export function projectHash(directory: string): string {
+  // Normalize: strip trailing slashes
+  const normalized = directory.replace(/\/+$/, "");
+  return createHash("sha256").update(normalized).digest("hex").slice(0, 16);
+}
+
+/**
+ * Get the per-project RPC port file path.
+ */
+export function rpcPortFilePath(storageDir: string, directory: string): string {
+  const hash = projectHash(directory);
+  return join(storageDir, "rpc", hash, "port");
+}

package/src/shared/runtime.ts

@@ -0,0 +1,26 @@
+type BridgeLike = {
+  getBridge: (
+    directory: string,
+    sessionID: string,
+  ) => {
+    send: (command: string, params?: Record<string, unknown>) => Promise<Record<string, unknown>>;
+  };
+};
+
+const GLOBAL_KEY = "__AFT_SHARED_BRIDGE_POOL__";
+
+function getGlobalState(): { [GLOBAL_KEY]?: BridgeLike | null } {
+  return globalThis as { [GLOBAL_KEY]?: BridgeLike | null };
+}
+
+export function setSharedBridgePool(pool: BridgeLike): void {
+  getGlobalState()[GLOBAL_KEY] = pool;
+}
+
+export function getSharedBridgePool(): BridgeLike | null {
+  return getGlobalState()[GLOBAL_KEY] ?? null;
+}
+
+export function clearSharedBridgePool(): void {
+  getGlobalState()[GLOBAL_KEY] = null;
+}

package/src/shared/status.ts

@@ -0,0 +1,206 @@
+export interface AftStatusSnapshot {
+  version: string;
+  project_root: string | null;
+  features: {
+    format_on_edit: boolean;
+    validate_on_edit: string;
+    restrict_to_project_root: boolean;
+    experimental_search_index: boolean;
+    experimental_semantic_search: boolean;
+  };
+  search_index: {
+    status: string;
+    files: number | null;
+    trigrams: number | null;
+  };
+  semantic_index: {
+    status: string;
+    entries: number | null;
+    dimension: number | null;
+  };
+  disk: {
+    storage_dir: string | null;
+    trigram_disk_bytes: number;
+    semantic_disk_bytes: number;
+  };
+  lsp_servers: number;
+  symbol_cache: {
+    local_entries: number;
+    warm_entries: number;
+  };
+  storage_dir: string | null;
+}
+
+function asRecord(value: unknown): Record<string, unknown> {
+  return typeof value === "object" && value !== null ? (value as Record<string, unknown>) : {};
+}
+
+function readString(value: unknown, fallback = ""): string {
+  return typeof value === "string" ? value : fallback;
+}
+
+function readNullableString(value: unknown): string | null {
+  return typeof value === "string" ? value : null;
+}
+
+function readBoolean(value: unknown, fallback = false): boolean {
+  return typeof value === "boolean" ? value : fallback;
+}
+
+function readNumber(value: unknown, fallback = 0): number {
+  return typeof value === "number" && Number.isFinite(value) ? value : fallback;
+}
+
+function readOptionalNumber(value: unknown): number | null {
+  return typeof value === "number" && Number.isFinite(value) ? value : null;
+}
+
+function formatFlag(enabled: boolean): string {
+  return enabled ? "enabled" : "disabled";
+}
+
+function formatCount(value: number | null): string {
+  return value == null ? "—" : value.toLocaleString("en-US");
+}
+
+export function formatBytes(bytes: number): string {
+  if (!Number.isFinite(bytes) || bytes <= 0) return "0 B";
+  const units = ["B", "KB", "MB", "GB", "TB"];
+  let value = bytes;
+  let unitIndex = 0;
+
+  while (value >= 1024 && unitIndex < units.length - 1) {
+    value /= 1024;
+    unitIndex++;
+  }
+
+  const decimals = value >= 10 || unitIndex === 0 ? 0 : 1;
+  return `${value.toFixed(decimals)} ${units[unitIndex]}`;
+}
+
+export function coerceAftStatus(response: Record<string, unknown>): AftStatusSnapshot {
+  const features = asRecord(response.features);
+  const searchIndex = asRecord(response.search_index);
+  const semanticIndex = asRecord(response.semantic_index);
+  const disk = asRecord(response.disk);
+  const symbolCache = asRecord(response.symbol_cache);
+
+  return {
+    version: readString(response.version, "unknown"),
+    project_root: readNullableString(response.project_root),
+    features: {
+      format_on_edit: readBoolean(features.format_on_edit),
+      validate_on_edit: readString(features.validate_on_edit, "off"),
+      restrict_to_project_root: readBoolean(features.restrict_to_project_root),
+      experimental_search_index: readBoolean(features.experimental_search_index),
+      experimental_semantic_search: readBoolean(features.experimental_semantic_search),
+    },
+    search_index: {
+      status: readString(searchIndex.status, "unknown"),
+      files: readOptionalNumber(searchIndex.files),
+      trigrams: readOptionalNumber(searchIndex.trigrams),
+    },
+    semantic_index: {
+      status: readString(semanticIndex.status, "unknown"),
+      entries: readOptionalNumber(semanticIndex.entries),
+      dimension: readOptionalNumber(semanticIndex.dimension),
+    },
+    disk: {
+      storage_dir: readNullableString(disk.storage_dir),
+      trigram_disk_bytes: readNumber(disk.trigram_disk_bytes),
+      semantic_disk_bytes: readNumber(disk.semantic_disk_bytes),
+    },
+    lsp_servers: readNumber(response.lsp_servers),
+    symbol_cache: {
+      local_entries: readNumber(symbolCache.local_entries),
+      warm_entries: readNumber(symbolCache.warm_entries),
+    },
+    storage_dir: readNullableString(response.storage_dir),
+  };
+}
+
+export function formatStatusDialogMessage(status: AftStatusSnapshot): string {
+  const lines = [
+    `AFT version: ${status.version}`,
+    `Project root: ${status.project_root ?? "(not configured)"}`,
+    "",
+    "Enabled features",
+    `- format_on_edit: ${formatFlag(status.features.format_on_edit)}`,
+    `- experimental_search_index: ${formatFlag(status.features.experimental_search_index)}`,
+    `- experimental_semantic_search: ${formatFlag(status.features.experimental_semantic_search)}`,
+    "",
+    "Search index",
+    `- status: ${status.search_index.status}`,
+    `- files: ${formatCount(status.search_index.files)}`,
+    `- trigrams: ${formatCount(status.search_index.trigrams)}`,
+    "",
+    "Semantic index",
+    `- status: ${status.semantic_index.status}`,
+    `- entries: ${formatCount(status.semantic_index.entries)}`,
+  ];
+
+  if (status.semantic_index.dimension != null) {
+    lines.push(`- dimension: ${formatCount(status.semantic_index.dimension)}`);
+  }
+
+  lines.push(
+    "",
+    "Disk usage",
+    `- trigram index: ${formatBytes(status.disk.trigram_disk_bytes)}`,
+    `- semantic index: ${formatBytes(status.disk.semantic_disk_bytes)}`,
+    "",
+    "Runtime",
+    `- LSP servers: ${formatCount(status.lsp_servers)}`,
+    `- symbol cache: ${formatCount(status.symbol_cache.local_entries)} local / ${formatCount(status.symbol_cache.warm_entries)} warm`,
+  );
+
+  if (status.storage_dir ?? status.disk.storage_dir) {
+    lines.push(`- storage dir: ${status.storage_dir ?? status.disk.storage_dir}`);
+  }
+
+  return lines.join("\n");
+}
+
+export function formatStatusMarkdown(status: AftStatusSnapshot): string {
+  const lines = [
+    "## AFT Status",
+    "",
+    `- **Version:** \`${status.version}\``,
+    `- **Project root:** \`${status.project_root ?? "(not configured)"}\``,
+    "",
+    "### Enabled features",
+    `- \`format_on_edit\`: ${formatFlag(status.features.format_on_edit)}`,
+    `- \`experimental_search_index\`: ${formatFlag(status.features.experimental_search_index)}`,
+    `- \`experimental_semantic_search\`: ${formatFlag(status.features.experimental_semantic_search)}`,
+    "",
+    "### Search index",
+    `- **Status:** \`${status.search_index.status}\``,
+    `- **Files:** ${formatCount(status.search_index.files)}`,
+    `- **Trigrams:** ${formatCount(status.search_index.trigrams)}`,
+    "",
+    "### Semantic index",
+    `- **Status:** \`${status.semantic_index.status}\``,
+    `- **Entries:** ${formatCount(status.semantic_index.entries)}`,
+  ];
+
+  if (status.semantic_index.dimension != null) {
+    lines.push(`- **Dimension:** ${formatCount(status.semantic_index.dimension)}`);
+  }
+
+  lines.push(
+    "",
+    "### Disk usage",
+    `- **Trigram index:** ${formatBytes(status.disk.trigram_disk_bytes)}`,
+    `- **Semantic index:** ${formatBytes(status.disk.semantic_disk_bytes)}`,
+    "",
+    "### Runtime",
+    `- **LSP servers:** ${formatCount(status.lsp_servers)}`,
+    `- **Symbol cache:** ${formatCount(status.symbol_cache.local_entries)} local / ${formatCount(status.symbol_cache.warm_entries)} warm`,
+  );
+
+  if (status.storage_dir ?? status.disk.storage_dir) {
+    lines.push(`- **Storage dir:** \`${status.storage_dir ?? status.disk.storage_dir}\``);
+  }
+
+  return lines.join("\n");
+}

package/src/shared/tui-config.ts

@@ -0,0 +1,155 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { dirname, join } from "node:path";
+import { log } from "../logger.js";
+import { getOpenCodeConfigPaths } from "./opencode-config-dir.js";
+
+const PLUGIN_NAME = "@cortexkit/aft-opencode";
+const PLUGIN_ENTRY = `${PLUGIN_NAME}@latest`;
+
+function stripJsoncComments(text: string): string {
+  let result = "";
+  let i = 0;
+  let inString = false;
+  let escaped = false;
+
+  while (i < text.length) {
+    const ch = text[i];
+
+    if (inString) {
+      result += ch;
+      if (escaped) {
+        escaped = false;
+      } else if (ch === "\\") {
+        escaped = true;
+      } else if (ch === '"') {
+        inString = false;
+      }
+      i++;
+      continue;
+    }
+
+    if (ch === '"') {
+      inString = true;
+      result += ch;
+      i++;
+      continue;
+    }
+
+    if (ch === "/" && text[i + 1] === "/") {
+      i += 2;
+      while (i < text.length && text[i] !== "\n") i++;
+      continue;
+    }
+
+    if (ch === "/" && text[i + 1] === "*") {
+      i += 2;
+      while (i < text.length && !(text[i] === "*" && text[i + 1] === "/")) i++;
+      i += 2;
+      continue;
+    }
+
+    result += ch;
+    i++;
+  }
+
+  return result;
+}
+
+function stripTrailingCommas(text: string): string {
+  let result = "";
+  let i = 0;
+  let inString = false;
+  let escaped = false;
+
+  while (i < text.length) {
+    const ch = text[i];
+
+    if (inString) {
+      result += ch;
+      if (escaped) {
+        escaped = false;
+      } else if (ch === "\\") {
+        escaped = true;
+      } else if (ch === '"') {
+        inString = false;
+      }
+      i++;
+      continue;
+    }
+
+    if (ch === '"') {
+      inString = true;
+      result += ch;
+      i++;
+      continue;
+    }
+
+    if (ch === ",") {
+      let j = i + 1;
+      while (j < text.length && /\s/.test(text[j])) j++;
+      if (text[j] === "}" || text[j] === "]") {
+        i++;
+        continue;
+      }
+    }
+
+    result += ch;
+    i++;
+  }
+
+  return result;
+}
+
+function parseJsonc<T>(content: string): T {
+  return JSON.parse(stripTrailingCommas(stripJsoncComments(content))) as T;
+}
+
+function resolveTuiConfigPath(): string {
+  const configDir = getOpenCodeConfigPaths({ binary: "opencode" }).configDir;
+  const jsoncPath = join(configDir, "tui.jsonc");
+  const jsonPath = join(configDir, "tui.json");
+
+  if (existsSync(jsoncPath)) return jsoncPath;
+  if (existsSync(jsonPath)) return jsonPath;
+  return jsonPath;
+}
+
+export function ensureTuiPluginEntry(): boolean {
+  try {
+    const configPath = resolveTuiConfigPath();
+
+    let config: Record<string, unknown> = {};
+    if (existsSync(configPath)) {
+      config = parseJsonc<Record<string, unknown>>(readFileSync(configPath, "utf-8")) ?? {};
+    }
+
+    const plugins = Array.isArray(config.plugin)
+      ? config.plugin.filter((value): value is string => typeof value === "string")
+      : [];
+
+    if (
+      plugins.some(
+        (plugin) =>
+          plugin === PLUGIN_NAME ||
+          plugin.startsWith(`${PLUGIN_NAME}@`) ||
+          plugin.includes("opencode-plugin") ||
+          plugin.includes("aft-opencode"),
+      )
+    ) {
+      return false;
+    }
+
+    plugins.push(PLUGIN_ENTRY);
+    config.plugin = plugins;
+
+    mkdirSync(dirname(configPath), { recursive: true });
+    writeFileSync(configPath, `${JSON.stringify(config, null, 2)}\n`);
+    log(`[aft-plugin] added TUI plugin entry to ${configPath}`);
+    return true;
+  } catch (error) {
+    log(
+      `[aft-plugin] failed to update tui.json: ${error instanceof Error ? error.message : String(error)}`,
+    );
+    return false;
+  }
+}