@kubb/studio 5.2.7 → 5.3.0

package/dist/index.cjs

@@ -2,14 +2,14 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const require_rolldown_runtime = require("./rolldown-runtime-qbf5tadS.cjs");
 const require_protocol = require("./protocol.cjs");
 let node_util = require("node:util");
+let node_process = require("node:process");
+node_process = require_rolldown_runtime.__toESM(node_process, 1);
+let node_child_process = require("node:child_process");
 let node_fs_promises = require("node:fs/promises");
 let node_path = require("node:path");
 node_path = require_rolldown_runtime.__toESM(node_path, 1);
-let node_child_process = require("node:child_process");
 let ofetch = require("ofetch");
 let node_crypto = require("node:crypto");
-let node_process = require("node:process");
-node_process = require_rolldown_runtime.__toESM(node_process, 1);
 let unstorage = require("unstorage");
 let unstorage_drivers_fs = require("unstorage/drivers/fs");
 unstorage_drivers_fs = require_rolldown_runtime.__toESM(unstorage_drivers_fs, 1);
@@ -411,14 +411,21 @@ async function loadOrCreateFallbackSecret() {
 	return secret;
 }
 /**
+* Hashes a stable secret into the machine token Studio expects, so a host can derive one from its
+* own identity without duplicating `getMachineToken`'s SHA-256 step.
+*/
+function machineTokenFrom(secret) {
+	return (0, node_crypto.hash)("sha256", secret);
+}
+/**
 * Returns the machine token derived from the `KUBB_AGENT_SECRET` environment variable.
 * Falls back to a generated secret persisted in the runtime storage if the env var is not set.
 * The token is hashed with SHA-256.
 */
 async function getMachineToken() {
-	if (node_process.default.env.KUBB_AGENT_SECRET) return (0, node_crypto.hash)("sha256", node_process.default.env.KUBB_AGENT_SECRET);
+	if (node_process.default.env.KUBB_AGENT_SECRET) return machineTokenFrom(node_process.default.env.KUBB_AGENT_SECRET);
 	fallbackSecretPromise ??= loadOrCreateFallbackSecret();
-	return (0, node_crypto.hash)("sha256", await fallbackSecretPromise);
+	return machineTokenFrom(await fallbackSecretPromise);
 }
 //#endregion
 //#region src/api.ts
@@ -617,9 +624,34 @@ async function waitForJob({ studioUrl, token, id, timeoutMs = 6e4 }) {
 		await new Promise((resolve) => setTimeout(resolve, 1e3));
 	}
 }
+/**
+* Creates or reuses a CI agent (`POST /api/agents`), keyed by `(organization, machineToken)`.
+* Authenticates via `x-api-key`. Reusing the same `machineToken` reuses the same agent instead of
+* consuming a new one from the organization's agent limit.
+*/
+async function createAgent({ studioUrl, token, name, machineToken }) {
+	try {
+		return await (0, ofetch.ofetch)(`${studioUrl}/api/agents`, {
+			method: "POST",
+			headers: { "x-api-key": token },
+			body: {
+				name,
+				machineToken
+			}
+		});
+	} catch (error) {
+		if (error instanceof ofetch.FetchError) {
+			const upgradeUrl = error.data?.data?.upgradeUrl;
+			const detail = responseMessage(error.data) ?? getErrorMessage(error);
+			const hint = upgradeUrl ? ` Agent limit reached; upgrade at ${upgradeUrl}.` : "";
+			throw new Error(`Failed to create a Kubb Studio agent: ${detail}${hint}`, { cause: error });
+		}
+		throw error;
+	}
+}
 //#endregion
 //#region package.json
-var version = "5.2.7";
+var version = "5.3.0";
 //#endregion
 //#region src/hooks.ts
 /**
@@ -2497,11 +2529,13 @@ async function pollForPairingToken({ studioUrl = agentDefaults.studioUrl, sessio
 //#endregion
 exports.InvalidAgentTokenError = InvalidAgentTokenError;
 exports.PairingCanceledError = PairingCanceledError;
+exports.createAgent = createAgent;
 exports.createClient = createClient;
 exports.createFileStorage = createFileStorage;
 exports.createJob = createJob;
 exports.createJobId = require_protocol.createJobId;
 exports.defaultStudioUrl = defaultStudioUrl;
+exports.machineTokenFrom = machineTokenFrom;
 exports.pollForPairingToken = pollForPairingToken;
 exports.runConnection = runConnection;
 exports.setStorage = setStorage;