next-anteater 0.1.0 → 0.1.2

package/bin/cli.mjs

@@ -0,0 +1,12 @@
+#!/usr/bin/env node
+
+const command = process.argv[2];
+
+if (command === "uninstall") {
+  const { main } = await import("../lib/uninstall.mjs");
+  await main();
+} else {
+  // Default to setup (handles "setup" arg or no arg)
+  const { main } = await import("../lib/setup.mjs");
+  await main();
+}

package/bin/uninstall-anteater.mjs

@@ -0,0 +1,7 @@
+#!/usr/bin/env node
+import { main } from "../lib/uninstall.mjs";
+
+main().catch((err) => {
+  console.error(`\n  \x1b[31mError:\x1b[39m ${err.message}\n`);
+  process.exit(1);
+});

package/lib/scaffold.mjs

@@ -432,6 +432,95 @@ jobs:
 `;
 }
 
+/**
+ * Generate the /api/anteater/runs route handler for multi-run discovery.
+ */
+export function generateRunsRoute({ isTypeScript }) {
+  const ext = isTypeScript ? "ts" : "js";
+  const TS = isTypeScript;
+  const lines = [];
+  const add = (s) => lines.push(s);
+
+  add('import { NextResponse } from "next/server";');
+  if (TS) add('import type { AnteaterRun, AnteaterRunsResponse } from "@anteater/next";');
+  add("");
+  add("function getRepo()" + (TS ? ": string | undefined" : "") + " {");
+  add("  if (process.env.ANTEATER_GITHUB_REPO) return process.env.ANTEATER_GITHUB_REPO;");
+  add("  const owner = process.env.VERCEL_GIT_REPO_OWNER;");
+  add("  const slug = process.env.VERCEL_GIT_REPO_SLUG;");
+  add("  if (owner && slug) return `${owner}/${slug}`;");
+  add("  return undefined;");
+  add("}");
+  add("");
+  add("export async function GET() {");
+  add("  const repo = getRepo();");
+  add("  const token = process.env.GITHUB_TOKEN;");
+  add("  if (!repo || !token) {");
+  add("    return NextResponse.json" + (TS ? "<AnteaterRunsResponse>" : "") + "({ runs: [], deploymentId: process.env.VERCEL_DEPLOYMENT_ID });");
+  add("  }");
+  add("");
+  add("  const gh = (url" + (TS ? ": string" : "") + ") =>");
+  add("    fetch(url, {");
+  add("      headers: {");
+  add("        Authorization: `Bearer ${token}`,");
+  add('        Accept: "application/vnd.github+json",');
+  add('        "X-GitHub-Api-Version": "2022-11-28",');
+  add("      },");
+  add('      cache: "no-store",');
+  add("    });");
+  add("");
+  add("  try {");
+  add("    const owner = repo.split(\"/\")[0];");
+  add("    const refsRes = await gh(`https://api.github.com/repos/${repo}/git/matching-refs/heads/anteater/`);");
+  add("    const refs" + (TS ? ": Array<{ ref: string }>" : "") + " = refsRes.ok ? await refsRes.json() : [];");
+  add("");
+  add("    const prsRes = await gh(`https://api.github.com/repos/${repo}/pulls?state=all&per_page=20&sort=created&direction=desc`);");
+  add("    const allPrs" + (TS ? ": any[]" : "") + " = prsRes.ok ? await prsRes.json() : [];");
+  add("    const anteaterPrs = allPrs.filter((pr" + (TS ? ": any" : "") + ") => pr.head.ref.startsWith(\"anteater/\"));");
+  add("    const prByBranch = new Map(anteaterPrs.map((pr" + (TS ? ": any" : "") + ") => [pr.head.ref, pr]));");
+  add("");
+  add("    const runs" + (TS ? ": AnteaterRun[]" : "") + " = [];");
+  add("    const branchNames = refs.map((r) => r.ref.replace(\"refs/heads/\", \"\"));");
+  add("    for (const pr of anteaterPrs) {");
+  add("      if (!branchNames.includes(pr.head.ref)) branchNames.push(pr.head.ref);");
+  add("    }");
+  add("");
+  add("    for (const branch of branchNames) {");
+  add("      const pr = prByBranch.get(branch);");
+  add("      const parts = branch.split(\"-\");");
+  add("      const requestId = parts[parts.length - 1] || \"\";");
+  add("      const mode = branch.includes(\"friend-\") ? \"copy\" : \"prod\";");
+  add("      let step" + (TS ? ": string" : "") + ";");
+  add("");
+  add("      if (pr?.merged_at) {");
+  add("        const mergedAgo = Date.now() - new Date(pr.merged_at).getTime();");
+  add("        if (mergedAgo > 300000) continue;");
+  add("        step = mergedAgo > 150000 ? \"done\" : \"redeploying\";");
+  add("      } else if (pr?.state === \"closed\") {");
+  add("        continue;");
+  add("      } else if (pr) {");
+  add("        step = \"merging\";");
+  add("      } else {");
+  add("        step = \"working\";");
+  add("      }");
+  add("");
+  add("      const prompt = pr?.title?.replace(/^anteater:\\\\s*/i, \"\") || \"Starting...\";");
+  add("      runs.push({ branch, requestId, prompt, step, mode });");
+  add("      if (runs.length >= 5) break;");
+  add("    }");
+  add("");
+  add("    return NextResponse.json" + (TS ? "<AnteaterRunsResponse>" : "") + "({ runs, deploymentId: process.env.VERCEL_DEPLOYMENT_ID });");
+  add("  } catch {");
+  add("    return NextResponse.json" + (TS ? "<AnteaterRunsResponse>" : "") + "({ runs: [], deploymentId: process.env.VERCEL_DEPLOYMENT_ID });");
+  add("  }");
+  add("}");
+
+  return {
+    filename: `route.${ext}`,
+    content: lines.join("\n") + "\n",
+  };
+}
+
 /**
  * Generate the AI apply script.
  */
@@ -606,6 +695,14 @@ export async function scaffoldFiles(cwd, options) {
     results.push(join(routeDir, route.filename));
   }
 
+  // Runs API route (multi-run discovery)
+  const runsRoute = generateRunsRoute(options);
+  const runsDir = options.isAppRouter ? "app/api/anteater/runs" : "pages/api/anteater/runs";
+  const runsPath = join(cwd, runsDir, runsRoute.filename);
+  if (await writeIfNotExists(runsPath, runsRoute.content)) {
+    results.push(join(runsDir, runsRoute.filename));
+  }
+
   // GitHub Action workflow
   const workflowPath = join(cwd, ".github/workflows/anteater.yml");
   if (await writeIfNotExists(workflowPath, generateWorkflow(options))) {

package/lib/uninstall.mjs

@@ -0,0 +1,112 @@
+/**
+ * anteater uninstall — Removes all scaffolded Anteater files from the project.
+ */
+import { readFile, writeFile, rm, access } from "node:fs/promises";
+import { join } from "node:path";
+import { bold, green, red, dim, ok, fail, heading, blank } from "./ui.mjs";
+
+const cwd = process.cwd();
+
+async function fileExists(path) {
+  try {
+    await access(path);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+async function removeFile(path, label) {
+  if (await fileExists(path)) {
+    await rm(path, { recursive: true });
+    ok(`Removed ${label}`);
+    return true;
+  }
+  return false;
+}
+
+async function unpatchLayout() {
+  for (const layoutFile of ["app/layout.tsx", "app/layout.js"]) {
+    const fullPath = join(cwd, layoutFile);
+    if (!(await fileExists(fullPath))) continue;
+
+    let content = await readFile(fullPath, "utf-8");
+    if (!content.includes("AnteaterBar")) return false;
+
+    // Remove import lines
+    content = content.replace(/^.*AnteaterBar.*\n/gm, "");
+    // Remove component usage
+    content = content.replace(/^\s*<AnteaterBar[^]*?\/>\s*\n?/gm, "");
+    content = content.replace(/^\s*<AnteaterBarWrapper\s*\/>\s*\n?/gm, "");
+
+    await writeFile(fullPath, content, "utf-8");
+    ok(`Unpatched ${layoutFile}`);
+    return true;
+  }
+  return false;
+}
+
+async function removeDependency() {
+  const pkgPath = join(cwd, "package.json");
+  if (!(await fileExists(pkgPath))) return false;
+
+  const pkg = JSON.parse(await readFile(pkgPath, "utf-8"));
+  let changed = false;
+
+  if (pkg.dependencies?.["@anteater/next"]) {
+    delete pkg.dependencies["@anteater/next"];
+    changed = true;
+  }
+  if (pkg.devDependencies?.["@anteater/next"]) {
+    delete pkg.devDependencies["@anteater/next"];
+    changed = true;
+  }
+
+  if (changed) {
+    await writeFile(pkgPath, JSON.stringify(pkg, null, 2) + "\n", "utf-8");
+    ok("Removed @anteater/next from package.json");
+  }
+  return changed;
+}
+
+export async function main() {
+  console.log();
+  console.log(`  ${bold("\u{1F41C} Anteater Uninstall")}`);
+  console.log(`  ${"\u2500".repeat(20)}`);
+  blank();
+
+  heading("Removing scaffolded files");
+
+  let removed = 0;
+
+  // Config file
+  if (await removeFile(join(cwd, "anteater.config.ts"), "anteater.config.ts")) removed++;
+  if (await removeFile(join(cwd, "anteater.config.js"), "anteater.config.js")) removed++;
+
+  // API routes
+  if (await removeFile(join(cwd, "app/api/anteater"), "app/api/anteater/")) removed++;
+  if (await removeFile(join(cwd, "pages/api/anteater"), "pages/api/anteater/")) removed++;
+
+  // Wrapper component
+  if (await removeFile(join(cwd, "components/anteater-bar-wrapper.tsx"), "components/anteater-bar-wrapper.tsx")) removed++;
+  if (await removeFile(join(cwd, "components/anteater-bar-wrapper.js"), "components/anteater-bar-wrapper.js")) removed++;
+
+  // GitHub workflow
+  if (await removeFile(join(cwd, ".github/workflows/anteater.yml"), ".github/workflows/anteater.yml")) removed++;
+
+  // Unpatch layout
+  heading("Cleaning up layout");
+  await unpatchLayout();
+
+  // Remove dependency
+  heading("Removing dependency");
+  await removeDependency();
+
+  blank();
+  if (removed > 0) {
+    console.log(`  ${green("\u2713")} Anteater uninstalled. Run ${dim("npx next-anteater setup")} to reinstall.`);
+  } else {
+    console.log(`  ${dim("No Anteater files found — nothing to remove.")}`);
+  }
+  console.log();
+}

package/package.json

@@ -1,9 +1,11 @@
 {
   "name": "next-anteater",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "AI-powered live editing for your Next.js app",
   "bin": {
-    "anteater": "./bin/setup-anteater.mjs"
+    "anteater": "bin/setup-anteater.mjs",
+    "anteater-uninstall": "bin/uninstall-anteater.mjs",
+    "next-anteater": "bin/cli.mjs"
   },
   "type": "module",
   "files": ["bin/", "lib/"],