iterate-ui-vite 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Connor White
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/index.d.ts

@@ -0,0 +1,30 @@
+import { Plugin } from 'vite';
+
+interface IteratePluginOptions {
+    /** Port for the iterate daemon (default: 4000) */
+    daemonPort?: number;
+    /** Disable the babel plugin that injects component names/source locations (default: false) */
+    disableBabelPlugin?: boolean;
+}
+/**
+ * Vite plugin for iterate.
+ *
+ * Usage:
+ * ```ts
+ * // vite.config.ts
+ * import { iterate } from 'iterate-ui-vite'
+ * export default defineConfig({
+ *   plugins: [iterate()]
+ * })
+ * ```
+ *
+ * Automatically:
+ * 1. Starts the iterate daemon when `vite dev` runs
+ * 2. Injects the overlay `<script>` into every HTML page
+ * 3. Proxies /__iterate__/* and /api/* to the daemon
+ * 4. Injects component name/source data attributes via babel
+ * 5. Cleans up daemon when dev server stops
+ */
+declare function iterate(options?: IteratePluginOptions): Plugin[];
+
+export { type IteratePluginOptions, iterate as default, iterate };

package/dist/index.js

@@ -0,0 +1,222 @@
+// src/index.ts
+import { spawn, execSync } from "child_process";
+import { createRequire } from "module";
+import { readFileSync } from "fs";
+import http from "http";
+function iterate(options = {}) {
+  const daemonPort = options.daemonPort ?? 4e3;
+  let daemon = null;
+  let overlayJS = null;
+  const plugins = [];
+  if (!options.disableBabelPlugin) {
+    let babelCore = null;
+    let babelPluginPath = null;
+    plugins.push({
+      name: "iterate:component-source",
+      apply: "serve",
+      enforce: "pre",
+      // Run before @vitejs/plugin-react transforms JSX
+      async configResolved(config) {
+        try {
+          babelCore = await import("@babel/core");
+          const require2 = createRequire(import.meta.url);
+          babelPluginPath = require2.resolve("iterate-ui-babel-plugin");
+        } catch {
+          console.warn("[iterate] Babel plugin setup failed \u2014 component names will not be available");
+        }
+      },
+      async transform(code, id) {
+        if (!babelCore || !babelPluginPath) return null;
+        if (!/\.[jt]sx$/.test(id)) return null;
+        if (id.includes("node_modules")) return null;
+        try {
+          const result = await babelCore.transformAsync(code, {
+            filename: id,
+            plugins: [
+              [babelPluginPath, { root: process.cwd() }],
+              // Need JSX syntax plugin to parse JSX without transforming it
+              ["@babel/plugin-syntax-jsx", {}],
+              ...id.endsWith(".tsx") ? [["@babel/plugin-syntax-typescript", { isTSX: true }]] : []
+            ],
+            parserOpts: {
+              plugins: [
+                "jsx",
+                ...id.endsWith(".tsx") ? ["typescript"] : []
+              ]
+            },
+            // Don't transform anything else — just inject attributes
+            presets: [],
+            sourceMaps: true,
+            configFile: false,
+            babelrc: false
+          });
+          if (!result?.code) return null;
+          return { code: result.code, map: result.map };
+        } catch (err) {
+          return null;
+        }
+      }
+    });
+  }
+  plugins.push({
+    name: "iterate",
+    apply: "serve",
+    // Only active during dev
+    configureServer(server) {
+      const repoRoot = getGitRoot() ?? server.config.root;
+      daemon = startDaemon(daemonPort, repoRoot);
+      server.middlewares.use("/__iterate__/overlay.js", (_req, res) => {
+        if (!overlayJS) {
+          try {
+            const require2 = createRequire(import.meta.url);
+            const overlayPath = require2.resolve("iterate-ui-overlay/standalone");
+            overlayJS = readFileSync(overlayPath, "utf-8");
+          } catch {
+            res.statusCode = 404;
+            res.end("Overlay bundle not found");
+            return;
+          }
+        }
+        res.setHeader("Content-Type", "application/javascript");
+        res.setHeader("Cache-Control", "no-cache");
+        res.end(overlayJS);
+      });
+      server.middlewares.use((req, res, next) => {
+        const url = req.url ?? "";
+        if (url.startsWith("/api/") || url.startsWith("/__iterate__/")) {
+          proxyRequest(req, res, daemonPort);
+          return;
+        }
+        next();
+      });
+      server.httpServer?.on("upgrade", (req, socket, head) => {
+        if (req.url === "/ws") {
+          const proxy = http.request(
+            {
+              hostname: "127.0.0.1",
+              port: daemonPort,
+              path: "/ws",
+              method: "GET",
+              headers: {
+                ...req.headers,
+                host: `127.0.0.1:${daemonPort}`
+              }
+            },
+            () => {
+            }
+          );
+          proxy.on("upgrade", (_proxyRes, proxySocket, proxyHead) => {
+            socket.write(
+              `HTTP/1.1 101 Switching Protocols\r
+Upgrade: websocket\r
+Connection: Upgrade\r
+Sec-WebSocket-Accept: ${_proxyRes.headers["sec-websocket-accept"]}\r
+\r
+`
+            );
+            if (proxyHead.length > 0) socket.write(proxyHead);
+            proxySocket.pipe(socket);
+            socket.pipe(proxySocket);
+            proxySocket.on("error", () => socket.destroy());
+            socket.on("error", () => proxySocket.destroy());
+          });
+          proxy.on("error", () => {
+            socket.destroy();
+          });
+          proxy.end();
+        }
+      });
+      server.httpServer?.on("close", () => {
+        stopDaemon(daemon, daemonPort);
+        daemon = null;
+      });
+    },
+    // Inject the overlay script into HTML
+    transformIndexHtml(html) {
+      const iterationName = JSON.stringify(process.env.ITERATE_ITERATION_NAME ?? "__original__");
+      return html.replace(
+        "</body>",
+        `<script>
+  window.__iterate_shell__ = { activeTool: 'select', activeIteration: ${iterationName}, daemonPort: ${daemonPort} };
+</script>
+<script src="/__iterate__/overlay.js" defer></script>
+</body>`
+      );
+    }
+  });
+  return plugins;
+}
+function getGitRoot() {
+  try {
+    return execSync("git rev-parse --show-toplevel", { encoding: "utf-8" }).trim();
+  } catch {
+    return null;
+  }
+}
+function startDaemon(port, cwd) {
+  const child = spawn(
+    process.execPath,
+    ["--input-type=module", "-e", `import { startDaemon } from "iterate-ui-daemon"; startDaemon({ port: ${port}, cwd: ${JSON.stringify(cwd)} });`],
+    {
+      cwd,
+      stdio: ["ignore", "pipe", "pipe"],
+      env: {
+        ...process.env,
+        ITERATE_PORT: String(port),
+        ITERATE_CWD: cwd,
+        NODE_NO_WARNINGS: "1"
+      }
+    }
+  );
+  child.stdout?.on("data", (data) => {
+    const msg = data.toString().trim();
+    if (msg) console.log(`[iterate] ${msg}`);
+  });
+  child.stderr?.on("data", (data) => {
+    const msg = data.toString().trim();
+    if (msg && !msg.includes("ExperimentalWarning")) {
+      console.error(`[iterate] ${msg}`);
+    }
+  });
+  child.on("exit", (code) => {
+    if (code !== 0 && code !== null) {
+      console.error(`[iterate] daemon exited with code ${code}`);
+    }
+  });
+  return child;
+}
+async function stopDaemon(child, port) {
+  try {
+    await fetch(`http://127.0.0.1:${port}/api/shutdown`, { method: "POST" });
+  } catch {
+    child?.kill("SIGTERM");
+  }
+}
+function proxyRequest(req, res, port) {
+  const proxyReq = http.request(
+    {
+      hostname: "127.0.0.1",
+      port,
+      path: req.url,
+      method: req.method,
+      headers: {
+        ...req.headers,
+        host: `127.0.0.1:${port}`
+      }
+    },
+    (proxyRes) => {
+      res.writeHead(proxyRes.statusCode ?? 500, proxyRes.headers);
+      proxyRes.pipe(res);
+    }
+  );
+  proxyReq.on("error", () => {
+    res.statusCode = 502;
+    res.end("iterate daemon not available");
+  });
+  req.pipe(proxyReq);
+}
+var index_default = iterate;
+export {
+  index_default as default,
+  iterate
+};

package/package.json

@@ -0,0 +1,53 @@
+{
+  "name": "iterate-ui-vite",
+  "version": "0.1.0",
+  "description": "iterate Vite plugin — auto-starts daemon and injects overlay",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/connorwhite-online/iterate",
+    "directory": "packages/vite"
+  },
+  "homepage": "https://iterate-ui.com",
+  "keywords": [
+    "iterate",
+    "ui",
+    "vite",
+    "plugin",
+    "ai",
+    "worktree"
+  ],
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "dependencies": {
+    "iterate-ui-overlay": "0.1.0",
+    "iterate-ui-babel-plugin": "0.1.0",
+    "iterate-ui-daemon": "0.1.0"
+  },
+  "peerDependencies": {
+    "vite": "^5.0.0 || ^6.0.0"
+  },
+  "devDependencies": {
+    "@babel/core": "^7.24.0",
+    "@types/babel__core": "^7.20.0",
+    "@types/node": "^22.0.0",
+    "tsup": "^8.3.0",
+    "typescript": "^5.7.0",
+    "vite": "^6.0.0"
+  },
+  "scripts": {
+    "build": "tsup src/index.ts --format esm --dts --external @babel/core",
+    "dev": "tsup src/index.ts --format esm --dts --watch --external @babel/core",
+    "clean": "rm -rf dist"
+  }
+}