@kubb/agent 4.27.4 → 4.28.1

package/.output/server/chunks/nitro/nitro.mjs

@@ -1,10 +1,15 @@
-import process from 'node:process';globalThis._importMeta_=globalThis._importMeta_||{url:"file:///_entry.js",env:process.env};import http, { Server as Server$2 } from 'node:http';
+import process from 'node:process';globalThis._importMeta_=globalThis._importMeta_||{url:"file:///_entry.js",env:process.env};import urlNode from 'node:url'
+import pathNode from 'node:path'
+const __filename = urlNode.fileURLToPath(globalThis._importMeta_.url)
+const __dirname = pathNode.dirname(__filename)
+import http, { Server as Server$2 } from 'node:http';
 import https, { Server as Server$1 } from 'node:https';
 import { EventEmitter } from 'node:events';
 import { Buffer as Buffer$1 } from 'node:buffer';
-import fs$2, { promises, existsSync, readFileSync, mkdirSync, writeFileSync } from 'node:fs';
-import path$2, { resolve, dirname, join, relative, normalize } from 'node:path';
-import nodeCrypto, { createHash } from 'node:crypto';
+import fs$2, { promises, existsSync, readFileSync, writeFileSync } from 'node:fs';
+import path$2, { resolve, dirname, relative, join, normalize } from 'node:path';
+import anymatch from 'anymatch';
+import { createHash, randomBytes } from 'node:crypto';
 import process$1, { version as version$2 } from 'node:process';
 import { orderBy } from 'natural-orderby';
 import { merge as merge$1, mergeDeep, isPlainObject as isPlainObject$1, uniqueWith, isDeepEqual, isNumber, isFunction, difference } from 'remeda';
@@ -2597,11 +2602,11 @@ function defineDriver$1(factory) {
   return factory;
 }
 
-const DRIVER_NAME$1 = "memory";
+const DRIVER_NAME$2 = "memory";
 const memory = defineDriver$1(() => {
   const data = /* @__PURE__ */ new Map();
   return {
-    name: DRIVER_NAME$1,
+    name: DRIVER_NAME$2,
     getInstance: () => data,
     hasItem(key) {
       return data.has(key);
@@ -3150,6 +3155,118 @@ async function rmRecursive(dir) {
   );
 }
 
+const PATH_TRAVERSE_RE$1 = /\.\.:|\.\.$/;
+const DRIVER_NAME$1 = "fs";
+const unstorage_47drivers_47fs = defineDriver((userOptions = {}) => {
+  if (!userOptions.base) {
+    throw createRequiredError(DRIVER_NAME$1, "base");
+  }
+  const base = resolve(userOptions.base);
+  const ignore = anymatch(
+    userOptions.ignore || ["**/node_modules/**", "**/.git/**"]
+  );
+  const r = (key) => {
+    if (PATH_TRAVERSE_RE$1.test(key)) {
+      throw createError(
+        DRIVER_NAME$1,
+        `Invalid key: ${JSON.stringify(key)}. It should not contain .. segments`
+      );
+    }
+    const resolved = join(base, key.replace(/:/g, "/"));
+    return resolved;
+  };
+  let _watcher;
+  const _unwatch = async () => {
+    if (_watcher) {
+      await _watcher.close();
+      _watcher = void 0;
+    }
+  };
+  return {
+    name: DRIVER_NAME$1,
+    options: userOptions,
+    flags: {
+      maxDepth: true
+    },
+    hasItem(key) {
+      return existsSync(r(key));
+    },
+    getItem(key) {
+      return readFile(r(key), "utf8");
+    },
+    getItemRaw(key) {
+      return readFile(r(key));
+    },
+    async getMeta(key) {
+      const { atime, mtime, size, birthtime, ctime } = await promises.stat(r(key)).catch(() => ({}));
+      return { atime, mtime, size, birthtime, ctime };
+    },
+    setItem(key, value) {
+      if (userOptions.readOnly) {
+        return;
+      }
+      return writeFile(r(key), value, "utf8");
+    },
+    setItemRaw(key, value) {
+      if (userOptions.readOnly) {
+        return;
+      }
+      return writeFile(r(key), value);
+    },
+    removeItem(key) {
+      if (userOptions.readOnly) {
+        return;
+      }
+      return unlink(r(key));
+    },
+    getKeys(_base, topts) {
+      return readdirRecursive(r("."), ignore, topts?.maxDepth);
+    },
+    async clear() {
+      if (userOptions.readOnly || userOptions.noClear) {
+        return;
+      }
+      await rmRecursive(r("."));
+    },
+    async dispose() {
+      if (_watcher) {
+        await _watcher.close();
+      }
+    },
+    async watch(callback) {
+      if (_watcher) {
+        return _unwatch;
+      }
+      const { watch } = await import('chokidar');
+      await new Promise((resolve2, reject) => {
+        const watchOptions = {
+          ignoreInitial: true,
+          ...userOptions.watchOptions
+        };
+        if (!watchOptions.ignored) {
+          watchOptions.ignored = [];
+        } else if (Array.isArray(watchOptions.ignored)) {
+          watchOptions.ignored = [...watchOptions.ignored];
+        } else {
+          watchOptions.ignored = [watchOptions.ignored];
+        }
+        watchOptions.ignored.push(ignore);
+        _watcher = watch(base, watchOptions).on("ready", () => {
+          resolve2();
+        }).on("error", reject).on("all", (eventName, path) => {
+          path = relative(base, path);
+          if (eventName === "change" || eventName === "add") {
+            callback("update", path);
+          } else if (eventName === "unlink") {
+            callback("remove", path);
+          }
+        });
+      });
+      return _unwatch;
+    }
+  };
+});
+
 const PATH_TRAVERSE_RE = /\.\.:|\.\.$/;
 const DRIVER_NAME = "fs-lite";
 const unstorage_47drivers_47fs_45lite = defineDriver((opts = {}) => {
@@ -3220,6 +3337,7 @@ const storage = createStorage({});
 
 storage.mount('/assets', assets);
 
+storage.mount('kubb', unstorage_47drivers_47fs({"driver":"fs","base":"./.kubb/cache"}));
 storage.mount('data', unstorage_47drivers_47fs_45lite({"driver":"fsLite","base":"./.data/kv"}));
 
 function useStorage(base = "") {
@@ -5891,25 +6009,14 @@ function serializePluginOptions(options) {
 function isCommandMessage(msg) {
   return msg.type === "command";
 }
-
-const CONFIG_DIR = path$2.join(os$1.homedir(), ".kubb");
-const CONFIG_FILE = path$2.join(CONFIG_DIR, "config.json");
-function loadAgentConfig() {
-  try {
-    const content = readFileSync(CONFIG_FILE, "utf-8");
-    return JSON.parse(content);
-  } catch (_error) {
-    return { sessions: {} };
-  }
+function isPongMessage(msg) {
+  return msg.type === "pong";
 }
-function saveAgentConfig(config) {
-  try {
-    mkdirSync(CONFIG_DIR, { recursive: true });
-    writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2), "utf-8");
-  } catch (_error) {
-    logger.warn("Failed to save agent config");
-  }
+
+function getSessionKey(token) {
+  return `sessions:${createHash("sha512").update(token).digest("hex")}`;
 }
+
 function isSessionValid(session) {
   try {
     const expiresAt = new Date(session.expiresAt);
@@ -5919,60 +6026,17 @@ function isSessionValid(session) {
     return false;
   }
 }
-function getCachedSession(agentToken) {
-  var _a;
-  try {
-    const config = loadAgentConfig();
-    const sessionKey = hashAgentToken(agentToken);
-    const session = (_a = config.sessions) == null ? void 0 : _a[sessionKey];
-    if (!session) {
-      return null;
-    }
-    if (!isSessionValid(session)) {
-      deleteCachedSession(agentToken);
-      return null;
-    }
-    return session;
-  } catch (_error) {
-    logger.warn("Failed to get cached session");
-    return null;
-  }
+
+function generateToken() {
+  return randomBytes(32).toString("hex");
 }
-function cacheSession(agentToken, session) {
-  try {
-    const config = loadAgentConfig();
-    const sessionKey = hashAgentToken(agentToken);
-    if (!config.sessions) {
-      config.sessions = {};
-    }
-    config.sessions[sessionKey] = {
-      ...session,
-      storedAt: (/* @__PURE__ */ new Date()).toISOString()
-    };
-    saveAgentConfig(config);
-    logger.success("Cached agent session");
-  } catch (_error) {
-    logger.warn("Failed to cache session");
-  }
+function generateSecureToken(id = generateToken()) {
+  return createHash("sha256").update(id).digest("hex");
 }
-function deleteCachedSession(agentToken) {
-  var _a;
-  try {
-    const config = loadAgentConfig();
-    const sessionKey = hashAgentToken(agentToken);
-    if ((_a = config.sessions) == null ? void 0 : _a[sessionKey]) {
-      delete config.sessions[sessionKey];
-      saveAgentConfig(config);
-    }
-  } catch (_error) {
-    logger.warn("Failed to delete cached session");
+function generateMachineToken() {
+  if (process.env.KUBB_AGENT_SECRET) {
+    return generateSecureToken(process.env.KUBB_AGENT_SECRET);
   }
-}
-function hashAgentToken(token) {
-  return createHash("sha512").update(token).digest("hex");
-}
-
-function getMachineId() {
   const interfaces = os$1.networkInterfaces();
   const macs = [];
   for (const name in interfaces) {
@@ -5984,30 +6048,43 @@ function getMachineId() {
   }
   const hostname = os$1.hostname();
   const rawId = macs.join(",") + hostname;
-  return nodeCrypto.createHash("sha256").update(rawId).digest("hex");
+  return generateSecureToken(rawId);
 }
 
 async function createAgentSession({ token, studioUrl, noCache }) {
-  const cachedSession = !noCache ? getCachedSession(token) : null;
-  if (cachedSession) {
-    logger.success("Using cached agent session");
-    return cachedSession;
+  const storage = useStorage("kubb");
+  if (!noCache) {
+    const sessionKey = getSessionKey(token);
+    const stored = await storage.getItem(sessionKey);
+    if (stored && isSessionValid(stored)) {
+      logger.success("Using cached agent session");
+      return stored;
+    }
+    if (stored) {
+      await storage.removeItem(sessionKey);
+    }
   }
   const connectUrl = `${studioUrl}/api/agent/session/create`;
   try {
+    const machineToken = generateMachineToken();
     const data = await $fetch(connectUrl, {
       method: "POST",
       headers: {
         Authorization: `Bearer ${token}`
       },
-      body: { machineId: getMachineId() }
+      body: { machineToken }
     });
-    if (data && !noCache) {
-      cacheSession(token, data);
-    }
     if (!data) {
       throw new Error("No data available for agent session");
     }
+    if (!noCache) {
+      const sessionKey = getSessionKey(token);
+      await storage.setItem(sessionKey, {
+        ...data,
+        storedAt: (/* @__PURE__ */ new Date()).toISOString()
+      });
+      logger.success("Cached agent session");
+    }
     logger.success("Agent session created");
     return data;
   } catch (error) {
@@ -6016,7 +6093,7 @@ async function createAgentSession({ token, studioUrl, noCache }) {
 }
 async function registerAgent({ token, studioUrl }) {
   var _a, _b;
-  const machineId = getMachineId();
+  const machineToken = generateMachineToken();
   const registerUrl = `${studioUrl}/api/agent/register`;
   try {
     await $fetch(registerUrl, {
@@ -6024,7 +6101,7 @@ async function registerAgent({ token, studioUrl }) {
       headers: {
         Authorization: `Bearer ${token}`
       },
-      body: { machineId }
+      body: { machineToken }
     });
     logger.success("Agent registered with Studio");
   } catch (error) {
@@ -6079,7 +6156,7 @@ var BaseGenerator = (_a$3 = class {
 function isInputPath(config) {
   return typeof (config == null ? void 0 : config.input) === "object" && config.input !== null && "path" in config.input;
 }
-var version$1 = "4.27.4";
+var version$1 = "4.28.1";
 function getDiagnosticInfo() {
   return {
     nodeVersion: version$2,
@@ -12419,9 +12496,13 @@ parseSchemaObject_fn = function({ schema: _schemaObject, name, parentName, rootN
       }
     }, ...baseItems];
   }
+  if (schemaObject.type === "string" && schemaObject.contentMediaType === "application/octet-stream") {
+    baseItems.unshift({ keyword: schemaKeywords.blob });
+    return baseItems;
+  }
   if (schemaObject.format) {
     if (schemaObject.type === "integer" && schemaObject.format === "int64") {
-      baseItems.unshift({ keyword: schemaKeywords.bigint });
+      baseItems.unshift({ keyword: options.integerType === "number" ? schemaKeywords.integer : schemaKeywords.bigint });
       return baseItems;
     }
     if (schemaObject.type === "integer" && schemaObject.format === "int32") {
@@ -12634,6 +12715,27 @@ parseSchemaObject_fn = function({ schema: _schemaObject, name, parentName, rootN
   }));
   return (await Promise.all(writeTasks)).flat();
 }, _b);
+function getRequestBody(operationSchema) {
+  var _a2, _b2;
+  const requestBody = (_b2 = (_a2 = operationSchema == null ? void 0 : operationSchema.operation) == null ? void 0 : _a2.schema) == null ? void 0 : _b2.requestBody;
+  if (!requestBody || "$ref" in requestBody) return;
+  return requestBody;
+}
+function isRequestBodyRequired(operationSchema) {
+  const requestBody = getRequestBody(operationSchema);
+  return !!requestBody && requestBody.required === true;
+}
+function withRequiredRequestBodySchema(operationSchema) {
+  if (!operationSchema || !isRequestBodyRequired(operationSchema)) return operationSchema;
+  if (Array.isArray(operationSchema.schema.required) && operationSchema.schema.required.length > 0) return operationSchema;
+  return {
+    ...operationSchema,
+    schema: {
+      ...operationSchema.schema,
+      required: ["__kubb_required_request_body__"]
+    }
+  };
+}
 
 function getDefaultBanner({ title, description, version, config }) {
   try {
@@ -13377,6 +13479,18 @@ var OperationGenerator = (_a = class extends BaseGenerator {
       var _a3, _b2;
       return ((_a3 = item.statusCode) == null ? void 0 : _a3.toString().startsWith("4")) || ((_b2 = item.statusCode) == null ? void 0 : _b2.toString().startsWith("5"));
     });
+    const request = withRequiredRequestBodySchema(requestSchema ? {
+      name: this.context.UNSTABLE_NAMING ? resolveName(transformers_default.pascalCase(`${operationId} RequestData`)) : resolveName(transformers_default.pascalCase(`${operationId} ${operation.method === "get" ? "queryRequest" : "mutationRequest"}`)),
+      description: (_a2 = operation.schema.requestBody) == null ? void 0 : _a2.description,
+      operation,
+      operationName,
+      schema: requestSchema,
+      keys: resolveKeys(requestSchema),
+      keysToOmit: (_b = resolveKeys(requestSchema)) == null ? void 0 : _b.filter((key) => {
+        var _a3, _b2;
+        return (_b2 = (_a3 = requestSchema.properties) == null ? void 0 : _a3[key]) == null ? void 0 : _b2.readOnly;
+      })
+    } : void 0);
     return {
       pathParams: pathParamsSchema ? {
         name: resolveName(transformers_default.pascalCase(`${operationId} PathParams`)),
@@ -13399,18 +13513,7 @@ var OperationGenerator = (_a = class extends BaseGenerator {
         schema: headerParamsSchema,
         keys: resolveKeys(headerParamsSchema)
       } : void 0,
-      request: requestSchema ? {
-        name: this.context.UNSTABLE_NAMING ? resolveName(transformers_default.pascalCase(`${operationId} RequestData`)) : resolveName(transformers_default.pascalCase(`${operationId} ${operation.method === "get" ? "queryRequest" : "mutationRequest"}`)),
-        description: (_a2 = operation.schema.requestBody) == null ? void 0 : _a2.description,
-        operation,
-        operationName,
-        schema: requestSchema,
-        keys: resolveKeys(requestSchema),
-        keysToOmit: (_b = resolveKeys(requestSchema)) == null ? void 0 : _b.filter((key) => {
-          var _a3, _b2;
-          return (_b2 = (_a3 = requestSchema.properties) == null ? void 0 : _a3[key]) == null ? void 0 : _b2.readOnly;
-        })
-      } : void 0,
+      request,
       response: {
         name: this.context.UNSTABLE_NAMING ? resolveName(transformers_default.pascalCase(`${operationId} ResponseData`)) : resolveName(transformers_default.pascalCase(`${operationId} ${operation.method === "get" ? "queryResponse" : "mutationResponse"}`)),
         operation,
@@ -213376,13 +213479,14 @@ const pluginTs = definePlugin((options) => {
   const { output = {
     path: "types",
     barrelType: "named"
-  }, group, exclude = [], include, override = [], enumType = "asConst", enumKeyCasing = "none", enumSuffix = "enum", dateType = "string", unknownType = "any", optionalType = "questionToken", arrayType = "array", emptySchemaType = unknownType, syntaxType = "type", transformers: transformers$1 = {}, mapper = {}, paramsCasing, generators = [typeGenerator].filter(Boolean), contentType, UNSTABLE_NAMING } = options;
+  }, group, exclude = [], include, override = [], enumType = "asConst", enumKeyCasing = "none", enumSuffix = "enum", dateType = "string", integerType = "bigint", unknownType = "any", optionalType = "questionToken", arrayType = "array", emptySchemaType = unknownType, syntaxType = "type", transformers: transformers$1 = {}, mapper = {}, paramsCasing, generators = [typeGenerator].filter(Boolean), contentType, UNSTABLE_NAMING } = options;
   return {
     name: pluginTsName,
     options: {
       output,
       transformers: transformers$1,
       dateType,
+      integerType,
       optionalType,
       arrayType,
       enumType,
@@ -216383,7 +216487,9 @@ function QueryOptions$4({ name, clientName, dataReturnType, typeSchemas, paramsC
        ${enabledText}
        queryKey,
        queryFn: async ({ signal }) => {
-          config.signal = signal
+          if (!config.signal) {
+            config.signal = signal
+          }
           return ${clientName}(${clientParams.toCall({})})
        },
       })
@@ -216673,7 +216779,9 @@ function InfiniteQueryOptions$1({ name, clientName, initialPageParam, cursorPara
        ${enabledText}
        queryKey,
        queryFn: async ({ signal, pageParam }) => {
-          config.signal = signal
+          if (!config.signal) {
+            config.signal = signal
+          }
           ${infiniteOverrideParams}
           return ${clientName}(${clientParams.toCall()})
        },
@@ -216696,7 +216804,9 @@ function InfiniteQueryOptions$1({ name, clientName, initialPageParam, cursorPara
        ${enabledText}
        queryKey,
        queryFn: async ({ signal }) => {
-          config.signal = signal
+          if (!config.signal) {
+            config.signal = signal
+          }
           return ${clientName}(${clientParams.toCall()})
        },
        ${queryOptions.join(",\n")}
@@ -217351,7 +217461,9 @@ function SuspenseInfiniteQueryOptions({ name, clientName, initialPageParam, curs
        ${enabledText}
        queryKey,
        queryFn: async ({ signal, pageParam }) => {
-          config.signal = signal
+          if (!config.signal) {
+            config.signal = signal
+          }
           ${infiniteOverrideParams}
           return ${clientName}(${clientParams.toCall()})
        },
@@ -217374,7 +217486,9 @@ function SuspenseInfiniteQueryOptions({ name, clientName, initialPageParam, curs
        ${enabledText}
        queryKey,
        queryFn: async ({ signal }) => {
-          config.signal = signal
+          if (!config.signal) {
+            config.signal = signal
+          }
           return ${clientName}(${clientParams.toCall()})
        },
        ${queryOptions.join(",\n")}
@@ -219465,7 +219579,9 @@ function QueryOptions$3({ name, clientName, typeSchemas, paramsCasing, paramsTyp
       ${enabledText}
        queryKey,
        queryFn: async ({ signal }) => {
-          config.signal = signal
+          if (!config.signal) {
+            config.signal = signal
+          }
           return ${clientName}(${clientParams.toCall()})
        },
       })
@@ -220476,7 +220592,9 @@ function QueryOptions$2({ name, clientName, typeSchemas, paramsCasing, paramsTyp
       ${enabledText}
        queryKey,
        queryFn: async ({ signal }) => {
-          config.signal = signal
+          if (!config.signal) {
+            config.signal = signal
+          }
           return ${clientName}(${clientParams.toCall()})
        },
       })
@@ -222382,7 +222500,9 @@ function QueryOptions({ name, clientName, dataReturnType, typeSchemas, paramsCas
       ${enabledText}
        queryKey,
        queryFn: async ({ signal }) => {
-          config.signal = signal
+          if (!config.signal) {
+            config.signal = signal
+          }
           return ${clientName}(${clientParams.toCall({ transformName(name$1) {
         return `toValue(${name$1})`;
       } })})
@@ -222693,7 +222813,9 @@ function InfiniteQueryOptions({ name, clientName, initialPageParam, cursorParam,
        ${enabledText}
        queryKey,
        queryFn: async ({ signal, pageParam }) => {
-          config.signal = signal
+          if (!config.signal) {
+            config.signal = signal
+          }
           ${infiniteOverrideParams}
           return ${clientName}(${clientParams.toCall({ transformName(name$1) {
         return `toValue(${name$1})`;
@@ -223808,25 +223930,25 @@ const pluginVueQuery = definePlugin((options) => {
   };
 });
 
-const pluginRegistry = {
-  "@kubb/plugin-client": pluginClient,
-  "@kubb/plugin-cypress": pluginCypress,
-  "@kubb/plugin-faker": pluginFaker,
-  "@kubb/plugin-mcp": pluginMcp,
-  "@kubb/plugin-msw": pluginMsw,
-  "@kubb/plugin-oas": pluginOas,
-  "@kubb/plugin-react-query": pluginReactQuery,
-  "@kubb/plugin-redoc": pluginRedoc,
-  "@kubb/plugin-solid-query": pluginSolidQuery,
-  "@kubb/plugin-svelte-query": pluginSvelteQuery,
-  "@kubb/plugin-swr": pluginSwr,
-  "@kubb/plugin-ts": pluginTs,
-  "@kubb/plugin-vue-query": pluginVueQuery,
-  "@kubb/plugin-zod": pluginZod
-};
+const pluginRegistry = /* @__PURE__ */ new Map([
+  ["@kubb/plugin-client", pluginClient],
+  ["@kubb/plugin-cypress", pluginCypress],
+  ["@kubb/plugin-faker", pluginFaker],
+  ["@kubb/plugin-mcp", pluginMcp],
+  ["@kubb/plugin-msw", pluginMsw],
+  ["@kubb/plugin-oas", pluginOas],
+  ["@kubb/plugin-react-query", pluginReactQuery],
+  ["@kubb/plugin-redoc", pluginRedoc],
+  ["@kubb/plugin-solid-query", pluginSolidQuery],
+  ["@kubb/plugin-svelte-query", pluginSvelteQuery],
+  ["@kubb/plugin-swr", pluginSwr],
+  ["@kubb/plugin-ts", pluginTs],
+  ["@kubb/plugin-vue-query", pluginVueQuery],
+  ["@kubb/plugin-zod", pluginZod]
+]);
 function resolvePlugins(plugins) {
   return plugins.map(({ name, options }) => {
-    const factory = pluginRegistry[name];
+    const factory = pluginRegistry.get(name);
     if (typeof factory !== "function") {
       throw new Error(`Plugin "${name}" is not supported. Supported plugins: ${Object.keys(pluginRegistry).join(", ")}`);
     }
@@ -223979,7 +224101,7 @@ function setupEventsStream(ws, events) {
   });
 }
 
-var version = "4.27.4";
+var version = "4.28.1";
 
 const _zcw7I4pYH8OiCfaDcjy_x7I6IH1tLDQR3W_yRZgP6E = defineNitroPlugin(async (nitro) => {
   const studioUrl = process$1.env.KUBB_STUDIO_URL || "https://studio.kubb.dev";
@@ -223996,6 +224118,7 @@ const _zcw7I4pYH8OiCfaDcjy_x7I6IH1tLDQR3W_yRZgP6E = defineNitroPlugin(async (nit
   }
   const resolvedConfigPath = path$2.isAbsolute(configPath) ? configPath : path$2.resolve(root, configPath);
   const events = new AsyncEventEmitter();
+  const storage = useStorage("kubb");
   async function loadConfig() {
     const result = await getCosmiConfig(resolvedConfigPath);
     const configs = await getConfigs(result.config, {});
@@ -224048,7 +224171,7 @@ const _zcw7I4pYH8OiCfaDcjy_x7I6IH1tLDQR3W_yRZgP6E = defineNitroPlugin(async (nit
   }
   async function connectToStudio() {
     try {
-      const { sessionToken, wsUrl } = await createAgentSession({
+      const { sessionToken, wsUrl, isSandbox } = await createAgentSession({
         noCache,
         token,
         studioUrl
@@ -224057,10 +224180,13 @@ const _zcw7I4pYH8OiCfaDcjy_x7I6IH1tLDQR3W_yRZgP6E = defineNitroPlugin(async (nit
       const onError = async () => {
         cleanup();
         logger.error(`Failed to connect to Kubb Studio on "${wsUrl}"`);
-        const cachedSession = !noCache ? getCachedSession(token) : null;
-        if (cachedSession) {
-          deleteCachedSession(token);
-          await reconnectToStudio();
+        if (!noCache) {
+          const sessionKey = getSessionKey(token);
+          const stored = await storage.getItem(sessionKey);
+          if (stored && isSessionValid(stored)) {
+            await storage.removeItem(sessionKey);
+            await reconnectToStudio();
+          }
         }
       };
       const onOpen = () => {
@@ -224075,7 +224201,7 @@ const _zcw7I4pYH8OiCfaDcjy_x7I6IH1tLDQR3W_yRZgP6E = defineNitroPlugin(async (nit
             studioUrl,
             token
           }).catch(async () => {
-            deleteCachedSession(token);
+            await storage.removeItem(getSessionKey(token));
             await reconnectToStudio();
           });
         }
@@ -224096,23 +224222,35 @@ const _zcw7I4pYH8OiCfaDcjy_x7I6IH1tLDQR3W_yRZgP6E = defineNitroPlugin(async (nit
       }, 3e4);
       setupEventsStream(ws, events);
       ws.addEventListener("message", async (message) => {
-        var _a, _b, _c;
+        var _a, _b, _c, _d;
         try {
           const data = JSON.parse(message.data);
+          if (isPongMessage(data)) {
+            logger.info("Received pong from Studio");
+            return;
+          }
           if (isCommandMessage(data)) {
             if (data.command === "generate") {
               const config = await loadConfig();
               const studioConfig = readStudioConfig(resolvedConfigPath);
               const patch = (_a = data.payload) != null ? _a : studioConfig;
               const resolvedPlugins = (patch == null ? void 0 : patch.plugins) ? resolvePlugins(patch.plugins) : void 0;
+              if (allowWrite && isSandbox) {
+                logger.warn("Agent is running in a sandbox environment, write will be disabled");
+              }
+              if ((patch == null ? void 0 : patch.input) && !isSandbox) {
+                logger.warn("Input override via payload is only supported in sandbox mode and will be ignored");
+              }
+              const inputOverride = isSandbox ? { data: (_b = patch.input) != null ? _b : "" } : void 0;
               await generate({
                 config: {
                   ...config,
+                  input: inputOverride != null ? inputOverride : config.input,
                   plugins: resolvedPlugins != null ? resolvedPlugins : config.plugins,
                   root,
                   output: {
                     ...config.output,
-                    write: allowWrite
+                    write: isSandbox ? false : allowWrite
                   }
                 },
                 events
@@ -224130,11 +224268,11 @@ const _zcw7I4pYH8OiCfaDcjy_x7I6IH1tLDQR3W_yRZgP6E = defineNitroPlugin(async (nit
                   version,
                   configPath,
                   permissions: {
-                    allowAll,
-                    allowWrite
+                    allowAll: isSandbox ? false : allowWrite,
+                    allowWrite: isSandbox ? false : allowWrite
                   },
                   config: {
-                    plugins: (_b = config.plugins) == null ? void 0 : _b.map((plugin) => ({
+                    plugins: (_c = config.plugins) == null ? void 0 : _c.map((plugin) => ({
                       name: `@kubb/${plugin.name}`,
                       options: serializePluginOptions(plugin.options)
                     }))
@@ -224146,11 +224284,11 @@ const _zcw7I4pYH8OiCfaDcjy_x7I6IH1tLDQR3W_yRZgP6E = defineNitroPlugin(async (nit
           }
           logger.warn(`Unknown message type from Kubb Studio: ${message.data}`);
         } catch (error) {
-          logger.error(`[unhandledRejection] ${(_c = error == null ? void 0 : error.message) != null ? _c : error}`);
+          logger.error(`[unhandledRejection] ${(_d = error == null ? void 0 : error.message) != null ? _d : error}`);
         }
       });
     } catch (error) {
-      deleteCachedSession(token);
+      await storage.removeItem(getSessionKey(token));
       logger.error(`Something went wrong ${error}`);
       await reconnectToStudio();
     }