assistant-ui 0.0.119 → 0.0.121

package/dist/lib/cloud-setup-login.js

@@ -0,0 +1,137 @@
+import { abortable } from "./cloud-abort.js";
+import { validateCloudUrl } from "./cloud-url.js";
+import { randomUUID } from "node:crypto";
+import { StatewireClient, StatewireHttp } from "statewire";
+//#region src/lib/cloud-setup-login.ts
+const validInputs = (inputs) => Array.isArray(inputs) && inputs.every((input) => input !== null && typeof input === "object" && typeof input.id === "string" && typeof input.status === "string");
+const setupLoginUrl = (authorization, issuer, attempt = randomUUID()) => {
+	const url = new URL(authorization.verificationUriComplete);
+	const issuerOrigin = new URL(issuer).origin;
+	if (!["https://accounts.assistant-ui.com", "https://accounts.aui.dev"].includes(issuerOrigin) || url.origin !== issuerOrigin || !/^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i.test(attempt) || !/^[A-Z0-9-]{4,32}$/.test(authorization.userCode) || url.pathname !== "/device" || url.username || url.password || url.hash || url.searchParams.get("user_code") !== authorization.userCode || url.searchParams.getAll("user_code").length !== 1 || !Number.isFinite(authorization.expiresAt) || authorization.expiresAt <= Date.now() || authorization.expiresAt > Date.now() + 6e5) throw new Error("Accounts returned an invalid device approval link.");
+	const publicUrl = new URL("/device", url.origin);
+	publicUrl.searchParams.set("user_code", authorization.userCode);
+	publicUrl.searchParams.set("setup_attempt", attempt);
+	publicUrl.searchParams.set("setup_expires", String(authorization.expiresAt));
+	return publicUrl.href;
+};
+const connectCloudLoginSetup = async (url) => {
+	const target = validateCloudUrl(url);
+	const controller = new AbortController();
+	const client = new StatewireClient({
+		transport: StatewireHttp({ url: target }),
+		onError: (error) => controller.abort(error)
+	});
+	let unsubscribe = () => {};
+	let inputId;
+	let sessionId;
+	let pendingHref;
+	let pendingAsk;
+	let terminalResult;
+	let completion;
+	let disposed = false;
+	const reconcile = () => {
+		const state = client.state;
+		if (state && !validInputs(state.inputs)) {
+			controller.abort(/* @__PURE__ */ new Error("The setup returned invalid inputs."));
+			return Promise.resolve();
+		}
+		if (disposed || inputId === void 0 || terminalResult === void 0 || state?.id !== sessionId || !Array.isArray(state.inputs) || !["planning", "installing"].includes(state.status) || ["dismissed", "answered"].includes(state.inputs.find((input) => input.id === inputId)?.status ?? "")) return Promise.resolve();
+		completion ??= abortable(client.commands["checkout/answer"]({
+			inputId,
+			answer: terminalResult
+		}), AbortSignal.timeout(1e4));
+		return completion;
+	};
+	const checkSession = () => {
+		const next = client.state;
+		if (next && !validInputs(next.inputs)) {
+			controller.abort(/* @__PURE__ */ new Error("The setup returned invalid inputs."));
+			return;
+		}
+		if (next?.id === sessionId && Array.isArray(next.inputs) && pendingHref) inputId ??= next.inputs.find((input) => input.preset === "assistant-ui-cli-login" && input.help?.href === pendingHref)?.id;
+		if (next?.version !== 2 || next.id !== sessionId || !Array.isArray(next.inputs) || !["planning", "installing"].includes(next.status) || inputId !== void 0 && next.inputs.find((input) => input.id === inputId)?.status === "dismissed") controller.abort(/* @__PURE__ */ new Error("The setup login was cancelled."));
+		if (client.connection.status === "stopped") controller.abort(/* @__PURE__ */ new Error("The setup connection stopped."));
+		reconcile().catch(() => {});
+	};
+	const dispose = async () => {
+		try {
+			if (pendingAsk && terminalResult) {
+				await abortable(pendingAsk, AbortSignal.timeout(1e4)).catch(() => {});
+				await reconcile().catch(() => {});
+			}
+		} finally {
+			disposed = true;
+			unsubscribe();
+			client.dispose();
+		}
+	};
+	try {
+		let resolveReady;
+		const ready = new Promise((resolve) => {
+			resolveReady = resolve;
+		});
+		unsubscribe = client.subscribe(() => {
+			if (client.state !== void 0) resolveReady();
+			if (client.connection.status === "stopped") controller.abort(/* @__PURE__ */ new Error("The setup connection stopped."));
+		});
+		if (client.state !== void 0) resolveReady();
+		if (client.connection.status === "stopped") controller.abort(/* @__PURE__ */ new Error("The setup connection stopped."));
+		await abortable(ready, AbortSignal.any([controller.signal, AbortSignal.timeout(1e4)]));
+		const state = client.state;
+		if (state?.version !== 2 || !state.id || !validInputs(state.inputs) || !["planning", "installing"].includes(state.status)) throw new Error("Start the setup in the wizard before signing in the CLI.");
+		sessionId = state.id;
+		unsubscribe();
+		unsubscribe = client.subscribe(checkSession);
+		checkSession();
+	} catch (error) {
+		await dispose();
+		throw error;
+	}
+	const command = (run) => {
+		controller.signal.throwIfAborted();
+		return abortable(run(), AbortSignal.any([controller.signal, AbortSignal.timeout(1e4)]));
+	};
+	return {
+		signal: controller.signal,
+		async publish(authorization, issuer) {
+			controller.signal.throwIfAborted();
+			const href = setupLoginUrl(authorization, issuer);
+			pendingHref = href;
+			pendingAsk = client.commands["agent/ask"]({
+				kind: "text",
+				preset: "assistant-ui-cli-login",
+				prompt: "Sign in to Assistant Cloud",
+				help: {
+					summary: "Authorize the CLI in your browser, then return to this setup.",
+					href
+				}
+			}).then(async (result) => {
+				if (!disposed && client.state?.id === sessionId) {
+					inputId = result.inputId;
+					checkSession();
+					await reconcile().catch(() => {});
+				}
+				return result;
+			});
+			try {
+				await command(() => pendingAsk);
+				checkSession();
+				controller.signal.throwIfAborted();
+			} catch (error) {
+				terminalResult = controller.signal.aborted ? "cancelled" : "failed";
+				await reconcile().catch(() => {});
+				throw error;
+			}
+			return href;
+		},
+		async complete(result) {
+			terminalResult = result;
+			await reconcile();
+		},
+		dispose
+	};
+};
+//#endregion
+export { connectCloudLoginSetup, setupLoginUrl };
+
+//# sourceMappingURL=cloud-setup-login.js.map

package/dist/lib/cloud-setup-login.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloud-setup-login.js","names":[],"sources":["../../src/lib/cloud-setup-login.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { DeviceAuthorization } from \"aui-auth/device\";\nimport { StatewireClient, StatewireHttp } from \"statewire\";\nimport { abortable } from \"./cloud-abort\";\nimport { validateCloudUrl } from \"./cloud-url\";\n\ntype SetupState = {\n  version: number;\n  id: string | null;\n  status: string;\n  inputs: {\n    id: string;\n    status: string;\n    preset?: string;\n    help?: { href?: string };\n  }[];\n};\n\ntype SetupCommands = {\n  \"agent/ask\": (input: {\n    kind: \"text\";\n    preset: string;\n    prompt: string;\n    help: { summary: string; href: string };\n  }) => { inputId: string };\n  \"checkout/answer\": (answer: { inputId: string; answer: string }) => void;\n};\n\nconst validInputs = (inputs: unknown): inputs is SetupState[\"inputs\"] =>\n  Array.isArray(inputs) &&\n  inputs.every(\n    (input) =>\n      input !== null &&\n      typeof input === \"object\" &&\n      typeof input.id === \"string\" &&\n      typeof input.status === \"string\",\n  );\n\nexport const setupLoginUrl = (\n  authorization: DeviceAuthorization,\n  issuer: string,\n  attempt: string = randomUUID(),\n) => {\n  const url = new URL(authorization.verificationUriComplete);\n  const issuerOrigin = new URL(issuer).origin;\n  if (\n    ![\"https://accounts.assistant-ui.com\", \"https://accounts.aui.dev\"].includes(\n      issuerOrigin,\n    ) ||\n    url.origin !== issuerOrigin ||\n    !/^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i.test(\n      attempt,\n    ) ||\n    !/^[A-Z0-9-]{4,32}$/.test(authorization.userCode) ||\n    url.pathname !== \"/device\" ||\n    url.username ||\n    url.password ||\n    url.hash ||\n    url.searchParams.get(\"user_code\") !== authorization.userCode ||\n    url.searchParams.getAll(\"user_code\").length !== 1 ||\n    !Number.isFinite(authorization.expiresAt) ||\n    authorization.expiresAt <= Date.now() ||\n    authorization.expiresAt > Date.now() + 10 * 60 * 1000\n  ) {\n    throw new Error(\"Accounts returned an invalid device approval link.\");\n  }\n  const publicUrl = new URL(\"/device\", url.origin);\n  publicUrl.searchParams.set(\"user_code\", authorization.userCode);\n  publicUrl.searchParams.set(\"setup_attempt\", attempt);\n  publicUrl.searchParams.set(\"setup_expires\", String(authorization.expiresAt));\n  return publicUrl.href;\n};\n\nexport const connectCloudLoginSetup = async (url: string) => {\n  const target = validateCloudUrl(url);\n  const controller = new AbortController();\n  const client = new StatewireClient<SetupState | undefined, SetupCommands>({\n    transport: StatewireHttp({ url: target }),\n    onError: (error) => controller.abort(error),\n  });\n  let unsubscribe = () => {};\n  let inputId: string | undefined;\n  let sessionId: string;\n  let pendingHref: string | undefined;\n  let pendingAsk: Promise<{ inputId: string }> | undefined;\n  let terminalResult: \"signed-in\" | \"cancelled\" | \"failed\" | undefined;\n  let completion: Promise<void> | undefined;\n  let disposed = false;\n  const reconcile = () => {\n    const state = client.state;\n    if (state && !validInputs(state.inputs)) {\n      controller.abort(new Error(\"The setup returned invalid inputs.\"));\n      return Promise.resolve();\n    }\n    if (\n      disposed ||\n      inputId === undefined ||\n      terminalResult === undefined ||\n      state?.id !== sessionId ||\n      !Array.isArray(state.inputs) ||\n      ![\"planning\", \"installing\"].includes(state.status) ||\n      [\"dismissed\", \"answered\"].includes(\n        state.inputs.find((input) => input.id === inputId)?.status ?? \"\",\n      )\n    )\n      return Promise.resolve();\n    completion ??= abortable(\n      client.commands[\"checkout/answer\"]({ inputId, answer: terminalResult }),\n      AbortSignal.timeout(10_000),\n    );\n    return completion;\n  };\n  const checkSession = () => {\n    const next = client.state;\n    if (next && !validInputs(next.inputs)) {\n      controller.abort(new Error(\"The setup returned invalid inputs.\"));\n      return;\n    }\n    if (next?.id === sessionId && Array.isArray(next.inputs) && pendingHref) {\n      inputId ??= next.inputs.find(\n        (input) =>\n          input.preset === \"assistant-ui-cli-login\" &&\n          input.help?.href === pendingHref,\n      )?.id;\n    }\n    if (\n      next?.version !== 2 ||\n      next.id !== sessionId ||\n      !Array.isArray(next.inputs) ||\n      ![\"planning\", \"installing\"].includes(next.status) ||\n      (inputId !== undefined &&\n        next.inputs.find((input) => input.id === inputId)?.status ===\n          \"dismissed\")\n    ) {\n      controller.abort(new Error(\"The setup login was cancelled.\"));\n    }\n    if (client.connection.status === \"stopped\")\n      controller.abort(new Error(\"The setup connection stopped.\"));\n    void reconcile().catch(() => {});\n  };\n  const dispose = async () => {\n    try {\n      if (pendingAsk && terminalResult) {\n        await abortable(pendingAsk, AbortSignal.timeout(10_000)).catch(\n          () => {},\n        );\n        await reconcile().catch(() => {});\n      }\n    } finally {\n      disposed = true;\n      unsubscribe();\n      client.dispose();\n    }\n  };\n  try {\n    let resolveReady!: () => void;\n    const ready = new Promise<void>((resolve) => {\n      resolveReady = resolve;\n    });\n    unsubscribe = client.subscribe(() => {\n      if (client.state !== undefined) resolveReady();\n      if (client.connection.status === \"stopped\") {\n        controller.abort(new Error(\"The setup connection stopped.\"));\n      }\n    });\n    if (client.state !== undefined) resolveReady();\n    if (client.connection.status === \"stopped\") {\n      controller.abort(new Error(\"The setup connection stopped.\"));\n    }\n    await abortable(\n      ready,\n      AbortSignal.any([controller.signal, AbortSignal.timeout(10_000)]),\n    );\n    const state = client.state!;\n    if (\n      state?.version !== 2 ||\n      !state.id ||\n      !validInputs(state.inputs) ||\n      ![\"planning\", \"installing\"].includes(state.status)\n    ) {\n      throw new Error(\n        \"Start the setup in the wizard before signing in the CLI.\",\n      );\n    }\n    sessionId = state.id;\n    unsubscribe();\n    unsubscribe = client.subscribe(checkSession);\n    checkSession();\n  } catch (error) {\n    await dispose();\n    throw error;\n  }\n  const command = <T>(run: () => Promise<T>) => {\n    controller.signal.throwIfAborted();\n    return abortable(\n      run(),\n      AbortSignal.any([controller.signal, AbortSignal.timeout(10_000)]),\n    );\n  };\n  return {\n    signal: controller.signal,\n    async publish(authorization: DeviceAuthorization, issuer: string) {\n      controller.signal.throwIfAborted();\n      const href = setupLoginUrl(authorization, issuer);\n      pendingHref = href;\n      pendingAsk = client.commands[\"agent/ask\"]({\n        kind: \"text\",\n        preset: \"assistant-ui-cli-login\",\n        prompt: \"Sign in to Assistant Cloud\",\n        help: {\n          summary:\n            \"Authorize the CLI in your browser, then return to this setup.\",\n          href,\n        },\n      }).then(async (result) => {\n        if (!disposed && client.state?.id === sessionId) {\n          inputId = result.inputId;\n          checkSession();\n          await reconcile().catch(() => {});\n        }\n        return result;\n      });\n      try {\n        await command(() => pendingAsk!);\n        checkSession();\n        controller.signal.throwIfAborted();\n      } catch (error) {\n        terminalResult = controller.signal.aborted ? \"cancelled\" : \"failed\";\n        await reconcile().catch(() => {});\n        throw error;\n      }\n      return href;\n    },\n    async complete(result: \"signed-in\" | \"cancelled\" | \"failed\") {\n      terminalResult = result;\n      await reconcile();\n    },\n    dispose,\n  };\n};\n"],"mappings":";;;;;AA4BA,MAAM,eAAe,WACnB,MAAM,QAAQ,MAAM,KACpB,OAAO,OACJ,UACC,UAAU,QACV,OAAO,UAAU,YACjB,OAAO,MAAM,OAAO,YACpB,OAAO,MAAM,WAAW,QAC5B;AAEF,MAAa,iBACX,eACA,QACA,UAAkB,WAAW,MAC1B;CACH,MAAM,MAAM,IAAI,IAAI,cAAc,uBAAuB;CACzD,MAAM,eAAe,IAAI,IAAI,MAAM,CAAC,CAAC;CACrC,IACE,CAAC,CAAC,qCAAqC,0BAA0B,CAAC,CAAC,SACjE,YACF,KACA,IAAI,WAAW,gBACf,CAAC,yEAAyE,KACxE,OACF,KACA,CAAC,oBAAoB,KAAK,cAAc,QAAQ,KAChD,IAAI,aAAa,aACjB,IAAI,YACJ,IAAI,YACJ,IAAI,QACJ,IAAI,aAAa,IAAI,WAAW,MAAM,cAAc,YACpD,IAAI,aAAa,OAAO,WAAW,CAAC,CAAC,WAAW,KAChD,CAAC,OAAO,SAAS,cAAc,SAAS,KACxC,cAAc,aAAa,KAAK,IAAI,KACpC,cAAc,YAAY,KAAK,IAAI,IAAI,KAEvC,MAAM,IAAI,MAAM,oDAAoD;CAEtE,MAAM,YAAY,IAAI,IAAI,WAAW,IAAI,MAAM;CAC/C,UAAU,aAAa,IAAI,aAAa,cAAc,QAAQ;CAC9D,UAAU,aAAa,IAAI,iBAAiB,OAAO;CACnD,UAAU,aAAa,IAAI,iBAAiB,OAAO,cAAc,SAAS,CAAC;CAC3E,OAAO,UAAU;AACnB;AAEA,MAAa,yBAAyB,OAAO,QAAgB;CAC3D,MAAM,SAAS,iBAAiB,GAAG;CACnC,MAAM,aAAa,IAAI,gBAAgB;CACvC,MAAM,SAAS,IAAI,gBAAuD;EACxE,WAAW,cAAc,EAAE,KAAK,OAAO,CAAC;EACxC,UAAU,UAAU,WAAW,MAAM,KAAK;CAC5C,CAAC;CACD,IAAI,oBAAoB,CAAC;CACzB,IAAI;CACJ,IAAI;CACJ,IAAI;CACJ,IAAI;CACJ,IAAI;CACJ,IAAI;CACJ,IAAI,WAAW;CACf,MAAM,kBAAkB;EACtB,MAAM,QAAQ,OAAO;EACrB,IAAI,SAAS,CAAC,YAAY,MAAM,MAAM,GAAG;GACvC,WAAW,sBAAM,IAAI,MAAM,oCAAoC,CAAC;GAChE,OAAO,QAAQ,QAAQ;EACzB;EACA,IACE,YACA,YAAY,KAAA,KACZ,mBAAmB,KAAA,KACnB,OAAO,OAAO,aACd,CAAC,MAAM,QAAQ,MAAM,MAAM,KAC3B,CAAC,CAAC,YAAY,YAAY,CAAC,CAAC,SAAS,MAAM,MAAM,KACjD,CAAC,aAAa,UAAU,CAAC,CAAC,SACxB,MAAM,OAAO,MAAM,UAAU,MAAM,OAAO,OAAO,CAAC,EAAE,UAAU,EAChE,GAEA,OAAO,QAAQ,QAAQ;EACzB,eAAe,UACb,OAAO,SAAS,kBAAkB,CAAC;GAAE;GAAS,QAAQ;EAAe,CAAC,GACtE,YAAY,QAAQ,GAAM,CAC5B;EACA,OAAO;CACT;CACA,MAAM,qBAAqB;EACzB,MAAM,OAAO,OAAO;EACpB,IAAI,QAAQ,CAAC,YAAY,KAAK,MAAM,GAAG;GACrC,WAAW,sBAAM,IAAI,MAAM,oCAAoC,CAAC;GAChE;EACF;EACA,IAAI,MAAM,OAAO,aAAa,MAAM,QAAQ,KAAK,MAAM,KAAK,aAC1D,YAAY,KAAK,OAAO,MACrB,UACC,MAAM,WAAW,4BACjB,MAAM,MAAM,SAAS,WACzB,CAAC,EAAE;EAEL,IACE,MAAM,YAAY,KAClB,KAAK,OAAO,aACZ,CAAC,MAAM,QAAQ,KAAK,MAAM,KAC1B,CAAC,CAAC,YAAY,YAAY,CAAC,CAAC,SAAS,KAAK,MAAM,KAC/C,YAAY,KAAA,KACX,KAAK,OAAO,MAAM,UAAU,MAAM,OAAO,OAAO,CAAC,EAAE,WACjD,aAEJ,WAAW,sBAAM,IAAI,MAAM,gCAAgC,CAAC;EAE9D,IAAI,OAAO,WAAW,WAAW,WAC/B,WAAW,sBAAM,IAAI,MAAM,+BAA+B,CAAC;EAC7D,UAAe,CAAC,CAAC,YAAY,CAAC,CAAC;CACjC;CACA,MAAM,UAAU,YAAY;EAC1B,IAAI;GACF,IAAI,cAAc,gBAAgB;IAChC,MAAM,UAAU,YAAY,YAAY,QAAQ,GAAM,CAAC,CAAC,CAAC,YACjD,CAAC,CACT;IACA,MAAM,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC;GAClC;EACF,UAAU;GACR,WAAW;GACX,YAAY;GACZ,OAAO,QAAQ;EACjB;CACF;CACA,IAAI;EACF,IAAI;EACJ,MAAM,QAAQ,IAAI,SAAe,YAAY;GAC3C,eAAe;EACjB,CAAC;EACD,cAAc,OAAO,gBAAgB;GACnC,IAAI,OAAO,UAAU,KAAA,GAAW,aAAa;GAC7C,IAAI,OAAO,WAAW,WAAW,WAC/B,WAAW,sBAAM,IAAI,MAAM,+BAA+B,CAAC;EAE/D,CAAC;EACD,IAAI,OAAO,UAAU,KAAA,GAAW,aAAa;EAC7C,IAAI,OAAO,WAAW,WAAW,WAC/B,WAAW,sBAAM,IAAI,MAAM,+BAA+B,CAAC;EAE7D,MAAM,UACJ,OACA,YAAY,IAAI,CAAC,WAAW,QAAQ,YAAY,QAAQ,GAAM,CAAC,CAAC,CAClE;EACA,MAAM,QAAQ,OAAO;EACrB,IACE,OAAO,YAAY,KACnB,CAAC,MAAM,MACP,CAAC,YAAY,MAAM,MAAM,KACzB,CAAC,CAAC,YAAY,YAAY,CAAC,CAAC,SAAS,MAAM,MAAM,GAEjD,MAAM,IAAI,MACR,0DACF;EAEF,YAAY,MAAM;EAClB,YAAY;EACZ,cAAc,OAAO,UAAU,YAAY;EAC3C,aAAa;CACf,SAAS,OAAO;EACd,MAAM,QAAQ;EACd,MAAM;CACR;CACA,MAAM,WAAc,QAA0B;EAC5C,WAAW,OAAO,eAAe;EACjC,OAAO,UACL,IAAI,GACJ,YAAY,IAAI,CAAC,WAAW,QAAQ,YAAY,QAAQ,GAAM,CAAC,CAAC,CAClE;CACF;CACA,OAAO;EACL,QAAQ,WAAW;EACnB,MAAM,QAAQ,eAAoC,QAAgB;GAChE,WAAW,OAAO,eAAe;GACjC,MAAM,OAAO,cAAc,eAAe,MAAM;GAChD,cAAc;GACd,aAAa,OAAO,SAAS,YAAY,CAAC;IACxC,MAAM;IACN,QAAQ;IACR,QAAQ;IACR,MAAM;KACJ,SACE;KACF;IACF;GACF,CAAC,CAAC,CAAC,KAAK,OAAO,WAAW;IACxB,IAAI,CAAC,YAAY,OAAO,OAAO,OAAO,WAAW;KAC/C,UAAU,OAAO;KACjB,aAAa;KACb,MAAM,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC;IAClC;IACA,OAAO;GACT,CAAC;GACD,IAAI;IACF,MAAM,cAAc,UAAW;IAC/B,aAAa;IACb,WAAW,OAAO,eAAe;GACnC,SAAS,OAAO;IACd,iBAAiB,WAAW,OAAO,UAAU,cAAc;IAC3D,MAAM,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC;IAChC,MAAM;GACR;GACA,OAAO;EACT;EACA,MAAM,SAAS,QAA8C;GAC3D,iBAAiB;GACjB,MAAM,UAAU;EAClB;EACA;CACF;AACF"}

package/dist/lib/cloud-url.d.ts

@@ -0,0 +1,2 @@
+export declare const validateCloudUrl: (value: string) => string;
+//# sourceMappingURL=cloud-url.d.ts.map

package/dist/lib/cloud-url.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloud-url.d.ts","sourceRoot":"","sources":["../../src/lib/cloud-url.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,gBAAgB,GAAI,OAAO,MAAM,KAAG,MAkBhD,CAAC"}

package/dist/lib/cloud-url.js

@@ -0,0 +1,14 @@
+//#region src/lib/cloud-url.ts
+const validateCloudUrl = (value) => {
+	const url = new URL(value);
+	if (url.username || url.password || url.search || url.hash || url.protocol !== "https:" && !(url.protocol === "http:" && [
+		"localhost",
+		"127.0.0.1",
+		"[::1]"
+	].includes(url.hostname))) throw new Error("Cloud, accounts and setup URLs must use HTTPS or HTTP localhost for development.");
+	return value.replace(/\/+$/, "");
+};
+//#endregion
+export { validateCloudUrl };
+
+//# sourceMappingURL=cloud-url.js.map

package/dist/lib/cloud-url.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloud-url.js","names":[],"sources":["../../src/lib/cloud-url.ts"],"sourcesContent":["export const validateCloudUrl = (value: string): string => {\n  const url = new URL(value);\n  if (\n    url.username ||\n    url.password ||\n    url.search ||\n    url.hash ||\n    (url.protocol !== \"https:\" &&\n      !(\n        url.protocol === \"http:\" &&\n        [\"localhost\", \"127.0.0.1\", \"[::1]\"].includes(url.hostname)\n      ))\n  ) {\n    throw new Error(\n      \"Cloud, accounts and setup URLs must use HTTPS or HTTP localhost for development.\",\n    );\n  }\n  return value.replace(/\\/+$/, \"\");\n};\n"],"mappings":";AAAA,MAAa,oBAAoB,UAA0B;CACzD,MAAM,MAAM,IAAI,IAAI,KAAK;CACzB,IACE,IAAI,YACJ,IAAI,YACJ,IAAI,UACJ,IAAI,QACH,IAAI,aAAa,YAChB,EACE,IAAI,aAAa,WACjB;EAAC;EAAa;EAAa;CAAO,CAAC,CAAC,SAAS,IAAI,QAAQ,IAG7D,MAAM,IAAI,MACR,kFACF;CAEF,OAAO,MAAM,QAAQ,QAAQ,EAAE;AACjC"}

package/dist/lib/transform.js

@@ -2,8 +2,8 @@ import { SpawnExitError, SpawnSignalError, runSpawnCapture } from "./run-spawn.j
 import * as fs$1 from "node:fs";
 import path from "node:path";
 import { sync } from "glob";
-import debug from "debug";
 import { fileURLToPath } from "node:url";
+import debug from "debug";
 //#region src/lib/transform.ts
 const log = debug("codemod:transform");
 const __filename = fileURLToPath(import.meta.url);

package/dist/program.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"program.d.ts","sourceRoot":"","sources":["../src/program.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAWpC,wBAAgB,YAAY,YAc3B"}
+{"version":3,"file":"program.d.ts","sourceRoot":"","sources":["../src/program.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAYpC,wBAAgB,YAAY,YAe3B"}

package/dist/program.js

@@ -1,6 +1,7 @@
 import { add } from "./commands/add.js";
 import { agent } from "./commands/agent.js";
 import { create } from "./commands/create.js";
+import { cloud } from "./commands/cloud.js";
 import { doctor } from "./commands/doctor.js";
 import { info } from "./commands/info.js";
 import { init } from "./commands/init.js";
@@ -10,7 +11,7 @@ import { codemodCommand, upgradeCommand } from "./commands/upgrade.js";
 import { Command } from "commander";
 //#region src/program.ts
 function buildProgram() {
-	return new Command().name("assistant-ui").description("add components and dependencies to your project").addCommand(add).addCommand(create).addCommand(init).addCommand(mcp).addCommand(codemodCommand).addCommand(upgradeCommand).addCommand(update).addCommand(agent).addCommand(info).addCommand(doctor);
+	return new Command().name("assistant-ui").description("add components and dependencies to your project").addCommand(add).addCommand(create).addCommand(init).addCommand(mcp).addCommand(codemodCommand).addCommand(upgradeCommand).addCommand(update).addCommand(agent).addCommand(cloud).addCommand(info).addCommand(doctor);
 }
 //#endregion
 export { buildProgram };

package/dist/program.js.map

@@ -1 +1 @@
-{"version":3,"file":"program.js","names":[],"sources":["../src/program.ts"],"sourcesContent":["import { Command } from \"commander\";\nimport { create } from \"./commands/create\";\nimport { add } from \"./commands/add\";\nimport { codemodCommand, upgradeCommand } from \"./commands/upgrade\";\nimport { init } from \"./commands/init\";\nimport { update } from \"./commands/update\";\nimport { mcp } from \"./commands/mcp\";\nimport { agent } from \"./commands/agent\";\nimport { info } from \"./commands/info\";\nimport { doctor } from \"./commands/doctor\";\n\nexport function buildProgram() {\n  return new Command()\n    .name(\"assistant-ui\")\n    .description(\"add components and dependencies to your project\")\n    .addCommand(add)\n    .addCommand(create)\n    .addCommand(init)\n    .addCommand(mcp)\n    .addCommand(codemodCommand)\n    .addCommand(upgradeCommand)\n    .addCommand(update)\n    .addCommand(agent)\n    .addCommand(info)\n    .addCommand(doctor);\n}\n"],"mappings":";;;;;;;;;;;AAWA,SAAgB,eAAe;CAC7B,OAAO,IAAI,QAAQ,CAAC,CACjB,KAAK,cAAc,CAAC,CACpB,YAAY,iDAAiD,CAAC,CAC9D,WAAW,GAAG,CAAC,CACf,WAAW,MAAM,CAAC,CAClB,WAAW,IAAI,CAAC,CAChB,WAAW,GAAG,CAAC,CACf,WAAW,cAAc,CAAC,CAC1B,WAAW,cAAc,CAAC,CAC1B,WAAW,MAAM,CAAC,CAClB,WAAW,KAAK,CAAC,CACjB,WAAW,IAAI,CAAC,CAChB,WAAW,MAAM;AACtB"}
+{"version":3,"file":"program.js","names":[],"sources":["../src/program.ts"],"sourcesContent":["import { Command } from \"commander\";\nimport { create } from \"./commands/create\";\nimport { add } from \"./commands/add\";\nimport { codemodCommand, upgradeCommand } from \"./commands/upgrade\";\nimport { init } from \"./commands/init\";\nimport { update } from \"./commands/update\";\nimport { mcp } from \"./commands/mcp\";\nimport { agent } from \"./commands/agent\";\nimport { info } from \"./commands/info\";\nimport { doctor } from \"./commands/doctor\";\nimport { cloud } from \"./commands/cloud\";\n\nexport function buildProgram() {\n  return new Command()\n    .name(\"assistant-ui\")\n    .description(\"add components and dependencies to your project\")\n    .addCommand(add)\n    .addCommand(create)\n    .addCommand(init)\n    .addCommand(mcp)\n    .addCommand(codemodCommand)\n    .addCommand(upgradeCommand)\n    .addCommand(update)\n    .addCommand(agent)\n    .addCommand(cloud)\n    .addCommand(info)\n    .addCommand(doctor);\n}\n"],"mappings":";;;;;;;;;;;;AAYA,SAAgB,eAAe;CAC7B,OAAO,IAAI,QAAQ,CAAC,CACjB,KAAK,cAAc,CAAC,CACpB,YAAY,iDAAiD,CAAC,CAC9D,WAAW,GAAG,CAAC,CACf,WAAW,MAAM,CAAC,CAClB,WAAW,IAAI,CAAC,CAChB,WAAW,GAAG,CAAC,CACf,WAAW,cAAc,CAAC,CAC1B,WAAW,cAAc,CAAC,CAC1B,WAAW,MAAM,CAAC,CAClB,WAAW,KAAK,CAAC,CACjB,WAAW,KAAK,CAAC,CACjB,WAAW,IAAI,CAAC,CAChB,WAAW,MAAM;AACtB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assistant-ui",
-  "version": "0.0.119",
+  "version": "0.0.121",
   "description": "CLI for assistant-ui",
   "keywords": [
     "cli",
@@ -21,13 +21,15 @@
     "bin",
     "dist",
     "src",
-    "README.md"
+    "README.md",
+    "templates/cloud-harness"
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/agent-launcher": "^0.1.16",
+    "@assistant-ui/agent-launcher": "^0.1.17",
     "@clack/prompts": "^1.8.1",
-    "chalk": "^6.0.0",
+    "aui-auth": "^0.0.11",
+    "chalk": "^6.0.1",
     "cli-progress": "^3.12.0",
     "commander": "^15.0.0",
     "cross-spawn": "^7.0.6",
@@ -37,18 +39,20 @@
     "glob": "^13.0.6",
     "jscodeshift": "^17.4.0",
     "jsonc-parser": "^3.3.1",
-    "semver": "^7.8.5"
+    "react": "^19.3.0",
+    "semver": "^7.8.5",
+    "statewire": "^0.19.4"
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.29",
+    "@assistant-ui/x-buildutils": "0.0.30",
     "@types/cli-progress": "^3.11.6",
     "@types/cross-spawn": "^6.0.6",
     "@types/debug": "^4.1.13",
     "@types/jscodeshift": "^17.3.0",
-    "@types/node": "^26.5.1",
+    "@types/node": "^26.6.3",
     "@types/semver": "^7.8.0",
-    "@vitest/coverage-v8": "^5.0.1",
-    "vitest": "^5.0.1"
+    "@vitest/coverage-v8": "^5.0.3",
+    "vitest": "^5.0.3"
   },
   "publishConfig": {
     "access": "public",
@@ -64,10 +68,10 @@
     "url": "https://github.com/assistant-ui/assistant-ui/issues"
   },
   "scripts": {
-    "build": "aui-build",
-    "test": "vitest run",
+    "build": "aui-build && node scripts/copy-cloud-harness-template.mjs",
+    "test": "vitest run && node --test scripts/copy-cloud-harness-template.test.mjs",
+    "test:coverage": "vitest run --coverage --coverage.reporter=text --coverage.reporter=html --coverage.reporter=json-summary",
     "typecheck": "tsc --noEmit && tsc --noEmit -p tsconfig.test.json",
-    "test:watch": "vitest",
-    "test:coverage": "vitest run --coverage"
+    "test:watch": "vitest"
   }
 }

package/src/commands/cloud.test.ts

@@ -0,0 +1,238 @@
+import { mkdtemp, readFile, rm, writeFile } from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
+import { afterEach, describe, expect, it, vi } from "vitest";
+import { setupCloud, cloud } from "./cloud";
+
+const {
+  fetchOrganizations,
+  fetchProjects,
+  createProject,
+  cloudAccessToken,
+  provisionCloudHarness,
+  scaffold,
+  select,
+  password,
+} = vi.hoisted(() => ({
+  fetchOrganizations: vi.fn(),
+  fetchProjects: vi.fn(),
+  createProject: vi.fn(),
+  cloudAccessToken: vi.fn(),
+  provisionCloudHarness: vi.fn(),
+  scaffold: vi.fn(),
+  select: vi.fn(),
+  password: vi.fn(),
+}));
+
+vi.mock("aui-auth/accounts", async (importOriginal) => ({
+  ...(await importOriginal<typeof import("aui-auth/accounts")>()),
+  fetchOrganizations,
+  fetchProjects,
+  createProject,
+}));
+vi.mock("../lib/cloud-auth", async (importOriginal) => ({
+  ...(await importOriginal<typeof import("../lib/cloud-auth")>()),
+  cloudAccessToken,
+}));
+vi.mock("../lib/cloud-provision", async (importOriginal) => ({
+  ...(await importOriginal<typeof import("../lib/cloud-provision")>()),
+  provisionCloudHarness,
+}));
+vi.mock("./create", async (importOriginal) => ({
+  ...(await importOriginal<typeof import("./create")>()),
+  create: { parseAsync: scaffold },
+}));
+vi.mock("@clack/prompts", async (importOriginal) => ({
+  ...(await importOriginal<typeof import("@clack/prompts")>()),
+  select,
+  password,
+}));
+
+const directories: string[] = [];
+afterEach(async () => {
+  await Promise.all(
+    directories
+      .splice(0)
+      .map((directory) => rm(directory, { recursive: true, force: true })),
+  );
+});
+
+const fixture = async () => {
+  const directory = await mkdtemp(
+    path.join(os.tmpdir(), "assistant-ui-cloud-command-"),
+  );
+  directories.push(directory);
+  cloudAccessToken.mockResolvedValue("oauth-access");
+  fetchOrganizations.mockResolvedValue([
+    { id: "org-1", slug: "team", name: "Team" },
+  ]);
+  fetchProjects.mockResolvedValue([
+    { id: "project-1", slug: "chat", name: "Team chat" },
+  ]);
+  provisionCloudHarness.mockResolvedValue({
+    project_id: "project-1",
+    workspace_id: "shared-workspace",
+    harness_id: "harness-1",
+    harness_origin: "https://harness.example.com",
+    api_origin: "https://project.example.com",
+    api_key: "cloud-server-key",
+  });
+  scaffold.mockResolvedValue(undefined);
+  return directory;
+};
+
+const options = {
+  backendUrl: "http://localhost:3000/api/chat",
+  apiUrl: "https://cloud.example.com",
+  yes: true,
+  skipInstall: true,
+  accessCode: "hackathon-code",
+};
+
+describe("cloud setup", () => {
+  it("selects an existing project and scaffolds a shared harness with server-only credentials", async () => {
+    const directory = await fixture();
+    await setupCloud(directory, { ...options, org: "team", project: "chat" });
+    expect(createProject).not.toHaveBeenCalled();
+    expect(scaffold).toHaveBeenCalledWith(
+      expect.arrayContaining([
+        "cloud-harness",
+        "--skip-install",
+        "--no-skills",
+      ]),
+      { from: "user" },
+    );
+    expect(provisionCloudHarness).toHaveBeenCalledWith(
+      options.apiUrl,
+      "oauth-access",
+      expect.objectContaining({
+        org_id: "org-1",
+        project_id: "project-1",
+        access_code: "hackathon-code",
+        backend_urls: [],
+        allow_localhost: true,
+      }),
+    );
+    const env = await readFile(path.join(directory, ".env.local"), "utf8");
+    expect(env).toContain('ASSISTANT_API_KEY="cloud-server-key"');
+    expect(env).not.toContain("NEXT_PUBLIC_ASSISTANT_API_KEY");
+    expect(
+      await readFile(
+        path.join(directory, ".assistant-ui", "cloud.json"),
+        "utf8",
+      ),
+    ).not.toContain("cloud-server-key");
+  });
+
+  it("does not provision again or overwrite customizations when setup is repeated", async () => {
+    const directory = await fixture();
+    await setupCloud(directory, options);
+    const envFile = path.join(directory, ".env.local");
+    const custom = `${(await readFile(envFile, "utf8")).replace(/^ASSISTANT_API_KEY=/m, "export ASSISTANT_API_KEY=")}OPENAI_API_KEY=custom-key\nCUSTOM_SETTING=keep-me\n`;
+    await writeFile(envFile, custom);
+    const before = provisionCloudHarness.mock.calls.length;
+    await setupCloud(directory, options);
+    expect(provisionCloudHarness.mock.calls.length).toBe(before);
+    expect(await readFile(envFile, "utf8")).toBe(custom);
+  });
+
+  it("requires an explicit project when noninteractive users have several choices", async () => {
+    const directory = await fixture();
+    fetchProjects.mockResolvedValue([
+      { id: "one", slug: "one", name: "One" },
+      { id: "two", slug: "two", name: "Two" },
+    ]);
+    await expect(setupCloud(directory, options)).rejects.toThrow("--project");
+  });
+
+  it("reuses an organization created by an interrupted setup", async () => {
+    const directory = await fixture();
+    const before = fetchProjects.mock.calls.length;
+    await setupCloud(directory, {
+      ...options,
+      newOrg: "team",
+      project: "chat",
+    });
+    expect(fetchProjects.mock.calls.slice(before)).toEqual([
+      [expect.anything(), "oauth-access", { id: "org-1" }],
+    ]);
+    expect(fetchProjects).toHaveBeenCalledWith(
+      expect.anything(),
+      "oauth-access",
+      {
+        id: "org-1",
+      },
+    );
+    expect(provisionCloudHarness).toHaveBeenCalledWith(
+      options.apiUrl,
+      "oauth-access",
+      expect.objectContaining({ org_id: "org-1" }),
+    );
+  });
+
+  it("rejects a changed backend before silently reusing configured resources", async () => {
+    const directory = await fixture();
+    await setupCloud(directory, options);
+    const before = provisionCloudHarness.mock.calls.length;
+    await expect(
+      setupCloud(directory, {
+        ...options,
+        backendUrl: "https://app.example.com/api/chat",
+      }),
+    ).rejects.toThrow("backend allowlist in the Cloud dashboard");
+    expect(provisionCloudHarness.mock.calls.length).toBe(before);
+  });
+
+  it("rejects conflicting selection flags before logging in", async () => {
+    const directory = await fixture();
+    const before = cloudAccessToken.mock.calls.length;
+    await expect(
+      setupCloud(directory, { ...options, org: "team", newOrg: "other" }),
+    ).rejects.toThrow("Choose --org or --new-org");
+    expect(cloudAccessToken.mock.calls.length).toBe(before);
+  });
+
+  it("rejects URL credentials and invalid organization slugs before logging in", async () => {
+    const directory = await fixture();
+    const before = cloudAccessToken.mock.calls.length;
+    await expect(
+      setupCloud(directory, {
+        ...options,
+        backendUrl: "https://user:secret@app.example.com/api/chat",
+      }),
+    ).rejects.toThrow("credentials");
+    await expect(
+      setupCloud(directory, { ...options, newOrg: "Bad Org!" }),
+    ).rejects.toThrow("Organization slugs");
+    expect(cloudAccessToken.mock.calls.length).toBe(before);
+  });
+
+  it("stops before provisioning when organization selection is cancelled", async () => {
+    const directory = await fixture();
+    fetchOrganizations.mockResolvedValue([
+      { id: "one", slug: "one", name: "One" },
+      { id: "two", slug: "two", name: "Two" },
+    ]);
+    const prompts =
+      await vi.importActual<typeof import("@clack/prompts")>("@clack/prompts");
+    const cancelled = await prompts.select({
+      message: "Cancel test",
+      options: [{ value: "one", label: "One" }],
+      signal: AbortSignal.abort(),
+    });
+    select.mockResolvedValue(cancelled);
+    const before = provisionCloudHarness.mock.calls.length;
+    await expect(
+      setupCloud(directory, { ...options, yes: false }),
+    ).rejects.toThrow();
+    expect(provisionCloudHarness.mock.calls.length).toBe(before);
+  });
+
+  it("exposes login, logout, and setup as cloud commands", () => {
+    expect(cloud.commands.map((command) => command.name())).toEqual([
+      "login",
+      "logout",
+      "setup",
+    ]);
+  });
+});