@intuned/runtime-dev 0.1.0-test.4 → 0.1.0-test.41

package/dist/commands/interface/run.js

@@ -1,15 +1,21 @@
 #!/usr/bin/env node
 "use strict";
 
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.runAutomationCLI = runAutomationCLI;
 var _commander = require("commander");
 var _dotenv = _interopRequireDefault(require("dotenv"));
 var _asyncLocalStorage = require("../../common/asyncLocalStorage");
 var net = _interopRequireWildcard(require("net"));
 var _zod = _interopRequireDefault(require("zod"));
 var _runApi = require("../../common/runApi");
-var _enums = require("src/runtime/enums");
+var _enums = require("../../runtime/enums");
 var _unixSocket = require("../common/utils/unixSocket");
 var _promises = require("timers/promises");
+var _jwtTokenManager = require("../../common/jwtTokenManager");
+var _formatZodError = require("../../common/formatZodError");
 function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
 function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
@@ -17,6 +23,7 @@ const startRunApiSchema = _zod.default.object({
   type: _zod.default.literal("start"),
   parameters: _runApi.runApiParametersSchema.extend({
     retrieveSession: _zod.default.boolean(),
+    functionsToken: _zod.default.string().optional(),
     context: _zod.default.object({
       jobId: _zod.default.string().optional(),
       jobRunId: _zod.default.string().optional(),
@@ -35,121 +42,136 @@ const abortRunApiSchema = _zod.default.object({
   type: _zod.default.literal("abort"),
   parameters: _zod.default.object({}).optional()
 });
-const inputSchema = _zod.default.union([startRunApiSchema, nextRunApiSchema, abortRunApiSchema]);
+const tokenUpdateSchema = _zod.default.object({
+  type: _zod.default.literal("tokenUpdate"),
+  parameters: _zod.default.object({
+    functionsToken: _zod.default.string()
+  })
+});
+const inputSchema = _zod.default.union([startRunApiSchema, nextRunApiSchema, abortRunApiSchema, tokenUpdateSchema]);
 _dotenv.default.config({
   path: `.env`
 });
-_commander.program.description("run user automation and communicate using unix socket").argument("<socket-path>", "path to unix socket").action(async socketPath => {
-  let context;
-  const throttleTime = 1_000;
-  let timeoutTimestamp = Date.now();
-  const client = net.createConnection(socketPath, () => {
-    console.log("connected to server!");
-  });
-  let generator = null;
-  const abortController = new AbortController();
-  client.on("end", () => {
-    console.log("Disconnected from socket");
-  });
-  client.on("error", err => {
-    var _generator;
-    console.error("Child connection error:", err);
-    void ((_generator = generator) === null || _generator === void 0 ? void 0 : _generator.throw(err).catch(() => undefined));
-  });
-  process.on("SIGINT", () => {
-    var _generator2;
-    console.log("Received SIGINT");
-    void ((_generator2 = generator) === null || _generator2 === void 0 ? void 0 : _generator2.throw(new Error("Interrupted")).catch(() => undefined));
-    client.end();
-    process.exit(1);
-  });
-  const jsonUnixSocket = new _unixSocket.JSONUnixSocket(client);
-  async function runGeneratorAndSendResult(next) {
-    if (!generator) return;
-    const result = await (0, _asyncLocalStorage.runWithContext)(context, () => generator.next(next));
-    if (result.done) {
-      const resultToSend = result.value.isOk() ? result.value.value : result.value.error.json;
-      const success = result.value.isOk();
-      jsonUnixSocket.sendJSON({
-        type: "done",
-        result: resultToSend,
-        success
-      });
-      return true;
-    } else {
-      jsonUnixSocket.sendJSON({
-        type: "yield",
-        result: result.value
-      });
-      return false;
-    }
-  }
-  for await (const data of jsonUnixSocket.receiveJSON()) {
-    const inputParseResult = inputSchema.safeParse(data);
-    if (!inputParseResult.success) {
-      console.error(inputParseResult.error.errors);
-      jsonUnixSocket.sendJSON({
-        type: "done",
-        result: {
-          error: "InvalidJSON",
-          message: "Invalid input",
-          details: {
-            errors: inputParseResult.error.errors
-          }
-        },
-        success: false
-      });
-      break;
-    }
-    const input = inputParseResult.data;
-    if (input.type === "abort") {
-      abortController.abort();
-      await (0, _promises.setTimeout)(10);
-      jsonUnixSocket.sendJSON({
-        type: "done",
-        result: null
-      });
-      break;
-    }
-    if (input.type === "start") {
-      const gen = (0, _runApi.runApiGenerator)({
-        ...input.parameters,
-        abortSignal: abortController.signal
-      });
-      generator = gen;
-      context = {
-        extendedPayloads: [],
-        runEnvironment: input.parameters.runOptions.environment === "deployed" ? _enums.RunEnvironment.DEPLOYED : _enums.RunEnvironment.IDE,
-        timeoutInfo: {
-          extendTimeoutCallback: async () => {
-            if (Date.now() - timeoutTimestamp < throttleTime) return;
-            timeoutTimestamp = Date.now();
-            jsonUnixSocket.sendJSON({
-              type: "extend"
-            });
-          }
-        },
-        ...(input.parameters.context ?? {}),
-        proxy: getProxyUrlFromRunOptions(input.parameters.runOptions)
+function runAutomationCLI(importFunction) {
+  _commander.program.description("run user automation and communicate using unix socket").argument("<socket-path>", "path to unix socket").action(async socketPath => {
+    let context;
+    const throttleTime = 1_000;
+    let timeoutTimestamp = Date.now();
+    const client = net.createConnection(socketPath);
+    let generator = null;
+    const abortController = new AbortController();
+    client.on("error", err => {
+      var _generator2;
+      void ((_generator2 = generator) === null || _generator2 === void 0 ? void 0 : _generator2.throw(err).catch(() => undefined));
+    });
+    process.on("SIGINT", () => {
+      var _generator3;
+      void ((_generator3 = generator) === null || _generator3 === void 0 ? void 0 : _generator3.throw(new Error("Interrupted")).catch(() => undefined));
+      client.end();
+      process.exit(1);
+    });
+    process.on("SIGTERM", () => {
+      var _generator4;
+      void ((_generator4 = generator) === null || _generator4 === void 0 ? void 0 : _generator4.throw(new Error("Interrupted")).catch(() => undefined));
+      client.end();
+      process.exit(1);
+    });
+    const jsonUnixSocket = new _unixSocket.JSONUnixSocket(client);
+    async function runGeneratorAndSendResult(next) {
+      const _generator = generator;
+      if (_generator == null) return {
+        done: true
       };
-      if (await runGeneratorAndSendResult()) {
-        break;
+      const result = await (0, _asyncLocalStorage.runWithContext)(context, () => _generator.next(next));
+      if (result.done) {
+        const resultToSend = result.value.isOk() ? result.value.value : result.value.error.json;
+        const success = result.value.isOk();
+        jsonUnixSocket.sendJSON({
+          type: "done",
+          result: resultToSend,
+          success
+        });
+      } else {
+        jsonUnixSocket.sendJSON({
+          type: "yield",
+          result: result.value
+        });
       }
-      continue;
+      return {
+        done: result.done ?? false
+      };
     }
-    if (input.type === "next") {
-      if (!generator) {
-        throw new Error("generator not started");
+    for await (const data of jsonUnixSocket.receiveJSON()) {
+      const inputParseResult = inputSchema.safeParse(data);
+      if (!inputParseResult.success) {
+        console.error("Validation error:", (0, _formatZodError.formatZodError)(inputParseResult.error));
+        jsonUnixSocket.sendJSON({
+          type: "done",
+          result: new _runApi.InternalInvalidInputError("Invalid input", (0, _formatZodError.formatZodError)(inputParseResult.error)).json,
+          success: false
+        });
+        break;
       }
-      if (await runGeneratorAndSendResult(input.parameters.value)) {
+      const input = inputParseResult.data;
+      if (input.type === "abort") {
+        abortController.abort();
+        await (0, _promises.setTimeout)(10);
+        jsonUnixSocket.sendJSON({
+          type: "done",
+          result: null
+        });
         break;
       }
-      continue;
+      if (input.type === "start") {
+        _jwtTokenManager.backendFunctionsTokenManager.token = input.parameters.functionsToken;
+        generator = (0, _runApi.runApiGenerator)({
+          ...input.parameters,
+          abortSignal: abortController.signal,
+          importFunction
+        });
+        context = {
+          extendedPayloads: [],
+          runEnvironment: input.parameters.runOptions.environment === "deployed" ? _enums.RunEnvironment.DEPLOYED : _enums.RunEnvironment.IDE,
+          timeoutInfo: {
+            extendTimeoutCallback: async () => {
+              if (Date.now() - timeoutTimestamp < throttleTime) return;
+              timeoutTimestamp = Date.now();
+              jsonUnixSocket.sendJSON({
+                type: "extend"
+              });
+            }
+          },
+          ...(input.parameters.context ?? {}),
+          proxy: getProxyUrlFromRunOptions(input.parameters.runOptions)
+        };
+        const {
+          done
+        } = await runGeneratorAndSendResult();
+        if (done) {
+          break;
+        }
+        continue;
+      }
+      if (input.type === "next") {
+        if (!generator) {
+          throw new Error("generator not started");
+        }
+        const {
+          done
+        } = await runGeneratorAndSendResult(input.parameters.value);
+        if (done) {
+          break;
+        }
+        continue;
+      }
+      if (input.type === "tokenUpdate") {
+        _jwtTokenManager.backendFunctionsTokenManager.token = input.parameters.functionsToken;
+      }
     }
-  }
-  process.exit(0);
-});
-_commander.program.parse(process.argv);
+    process.exit(0);
+  });
+  _commander.program.parse(process.argv);
+}
 function getProxyUrlFromRunOptions(runOptions) {
   if ((runOptions === null || runOptions === void 0 ? void 0 : runOptions.environment) !== "deployed") return undefined;
   const proxy = runOptions.proxy;
@@ -158,4 +180,7 @@ function getProxyUrlFromRunOptions(runOptions) {
   url.username = proxy.username;
   url.password = proxy.password;
   return url.toString();
+}
+if (require.main === module) {
+  runAutomationCLI();
 }

package/dist/commands/ts-check.js

@@ -5,17 +5,19 @@ var ts = _interopRequireWildcard(require("typescript"));
 var _commander = require("commander");
 var fs = _interopRequireWildcard(require("fs-extra"));
 var path = _interopRequireWildcard(require("path"));
-var _webTemplate = require("./common/utils/webTemplate");
+var _template = require("./common/utils/template");
 function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
 function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
-_commander.program.description("Check TypeScript types in the project").allowUnknownOption().action(async () => {
-  await (0, _webTemplate.moveWebTemplateFiles)();
+_commander.program.description("Check TypeScript types in the project").option("-t, --template <type>", "template to use", "WebTemplate").allowUnknownOption().action(async ({
+  template
+}) => {
+  await (0, _template.moveTemplateFiles)(template);
   const templateTsConfig = path.resolve(__dirname, "..", "..", "template.tsconfig.json");
-  await fs.copy(templateTsConfig, "./intuned/WebTemplate/tsconfig.json");
-  checkTypes();
+  await fs.copy(templateTsConfig, `./intuned/${template}/tsconfig.json`);
+  checkTypes(template);
 });
-function checkTypes() {
-  const configPath = ts.findConfigFile("./intuned/WebTemplate", ts.sys.fileExists, "tsconfig.json");
+function checkTypes(template) {
+  const configPath = ts.findConfigFile(`./intuned/${template}`, ts.sys.fileExists, "tsconfig.json");
   if (!configPath) {
     console.error("Could not find a valid 'tsconfig.json'.");
     process.exit(1);

package/dist/common/formatZodError.d.ts

@@ -0,0 +1,2 @@
+import { ZodError } from "zod";
+export declare function formatZodError(zodError: ZodError): string[];

package/dist/common/formatZodError.js

@@ -0,0 +1,18 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.formatZodError = formatZodError;
+function formatZodError(zodError) {
+  const formattedErrors = zodError.errors.map(error => {
+    const path = error.path.map(segment => {
+      return typeof segment === "number" ? `[${segment}]` : segment;
+    }).join(".");
+    if (path) {
+      return `${path} is invalid - ${error.message}`;
+    }
+    return error.message;
+  });
+  return formattedErrors;
+}

package/dist/common/getPlaywrightConstructs.js

@@ -151,7 +151,11 @@ async function loadSessionToContext({
 }) {
   let sessionToLoad;
   if (session.type === "state") {
-    sessionToLoad = session.state;
+    const state = session.state;
+    if (state === undefined || state === null) {
+      return;
+    }
+    sessionToLoad = state;
   } else {
     const fullPath = (0, _fileUtils.getFullPathInProject)(session.path);
     sessionToLoad = await fs.readJson(fullPath);

package/dist/common/runApi/errors.d.ts

@@ -1,5 +1,5 @@
 import { RunAutomationResponse } from "./types";
-import { RunErrorOptions } from "src/runtime/export";
+import { RunErrorOptions } from "../../runtime/export";
 export declare const apiNotFoundErrorCode = "APINotFoundError";
 export declare const invalidApiErrorCode = "InvalidAPIError";
 export declare const invalidCheckErrorCode = "InvalidCheckError";
@@ -9,7 +9,8 @@ export declare const authCheckNotFoundErrorCode = "AuthCheckNotFoundError";
 export declare const authCheckFailedErrorCode = "AuthCheckFailedError";
 export declare const maxLevelsExceededErrorCode = "MaxLevelsExceededError";
 export declare const automationError = "AutomationError";
-export declare const runAutomationErrorCodes: readonly ["APINotFoundError", "InvalidAPIError", "InvalidCheckError", "AbortedError", "AuthRequiredError", "AuthCheckNotFoundError", "AuthCheckFailedError", "MaxLevelsExceededError", "AutomationError"];
+export declare const internalInvalidInputErrorCode = "InternalInvalidInputError";
+export declare const runAutomationErrorCodes: readonly ["APINotFoundError", "InvalidAPIError", "InvalidCheckError", "AbortedError", "AuthRequiredError", "AuthCheckNotFoundError", "AuthCheckFailedError", "MaxLevelsExceededError", "AutomationError", "InternalInvalidInputError"];
 export type RunAutomationErrorCode = (typeof runAutomationErrorCodes)[number];
 export declare abstract class RunAutomationError<T = any> {
     code: RunAutomationErrorCode;
@@ -20,7 +21,7 @@ export declare abstract class RunAutomationError<T = any> {
     details?: T;
     get apiResponse(): RunAutomationResponse;
     get json(): {
-        code: "APINotFoundError" | "InvalidAPIError" | "InvalidCheckError" | "AbortedError" | "AuthRequiredError" | "AuthCheckNotFoundError" | "AuthCheckFailedError" | "MaxLevelsExceededError" | "AutomationError";
+        code: "APINotFoundError" | "InvalidAPIError" | "InvalidCheckError" | "AbortedError" | "AuthRequiredError" | "AuthCheckNotFoundError" | "AuthCheckFailedError" | "MaxLevelsExceededError" | "AutomationError" | "InternalInvalidInputError";
         details: T | undefined;
         cause: any;
     };
@@ -69,3 +70,7 @@ export declare class AutomationError extends RunAutomationError<{
     error: any;
     constructor(error: any);
 }
+export declare class InternalInvalidInputError extends RunAutomationError {
+    constructor(message: string, details?: any);
+    get apiResponse(): RunAutomationResponse;
+}

package/dist/common/runApi/errors.js

@@ -3,7 +3,7 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.runAutomationErrorCodes = exports.maxLevelsExceededErrorCode = exports.invalidCheckErrorCode = exports.invalidApiErrorCode = exports.automationError = exports.authRequiredErrorCode = exports.authCheckNotFoundErrorCode = exports.authCheckFailedErrorCode = exports.apiNotFoundErrorCode = exports.abortedErrorCode = exports.RunAutomationError = exports.MaxLevelsExceededError = exports.InvalidCheckError = exports.InvalidApiError = exports.AutomationError = exports.AuthRequiredError = exports.AuthCheckNotFoundError = exports.AuthCheckFailedError = exports.ApiNotFoundError = exports.AbortedError = void 0;
+exports.runAutomationErrorCodes = exports.maxLevelsExceededErrorCode = exports.invalidCheckErrorCode = exports.invalidApiErrorCode = exports.internalInvalidInputErrorCode = exports.automationError = exports.authRequiredErrorCode = exports.authCheckNotFoundErrorCode = exports.authCheckFailedErrorCode = exports.apiNotFoundErrorCode = exports.abortedErrorCode = exports.RunAutomationError = exports.MaxLevelsExceededError = exports.InvalidCheckError = exports.InvalidApiError = exports.InternalInvalidInputError = exports.AutomationError = exports.AuthRequiredError = exports.AuthCheckNotFoundError = exports.AuthCheckFailedError = exports.ApiNotFoundError = exports.AbortedError = void 0;
 var _runtime = require("../../runtime");
 var playwright = _interopRequireWildcard(require("@intuned/playwright-core"));
 function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
@@ -17,7 +17,8 @@ const authCheckNotFoundErrorCode = exports.authCheckNotFoundErrorCode = "AuthChe
 const authCheckFailedErrorCode = exports.authCheckFailedErrorCode = "AuthCheckFailedError";
 const maxLevelsExceededErrorCode = exports.maxLevelsExceededErrorCode = "MaxLevelsExceededError";
 const automationError = exports.automationError = "AutomationError";
-const runAutomationErrorCodes = exports.runAutomationErrorCodes = [apiNotFoundErrorCode, invalidApiErrorCode, invalidCheckErrorCode, abortedErrorCode, authRequiredErrorCode, authCheckNotFoundErrorCode, authCheckFailedErrorCode, maxLevelsExceededErrorCode, automationError];
+const internalInvalidInputErrorCode = exports.internalInvalidInputErrorCode = "InternalInvalidInputError";
+const runAutomationErrorCodes = exports.runAutomationErrorCodes = [apiNotFoundErrorCode, invalidApiErrorCode, invalidCheckErrorCode, abortedErrorCode, authRequiredErrorCode, authCheckNotFoundErrorCode, authCheckFailedErrorCode, maxLevelsExceededErrorCode, automationError, internalInvalidInputErrorCode];
 class RunAutomationError {
   wrapped = false;
   get apiResponse() {
@@ -92,6 +93,7 @@ class AuthRequiredError extends RunAutomationError {
     this.code = authRequiredErrorCode;
     this.statusCode = 401;
     this.message = "Authentication is required";
+    this.wrapped = true;
   }
 }
 exports.AuthRequiredError = AuthRequiredError;
@@ -110,6 +112,7 @@ class AuthCheckFailedError extends RunAutomationError {
     this.code = authCheckFailedErrorCode;
     this.statusCode = 401;
     this.message = "auth session check failed";
+    this.wrapped = true;
   }
   get apiResponse() {
     return {
@@ -169,4 +172,24 @@ class AutomationError extends RunAutomationError {
     };
   }
 }
-exports.AutomationError = AutomationError;
+exports.AutomationError = AutomationError;
+class InternalInvalidInputError extends RunAutomationError {
+  constructor(message, details) {
+    super();
+    this.code = internalInvalidInputErrorCode;
+    this.statusCode = 500;
+    this.message = message;
+    this.details = details;
+  }
+  get apiResponse() {
+    return {
+      status: 500,
+      body: {
+        error: "InternalServerError",
+        message: `Internal error: ${this.message}. Please report this issue to the Intuned team.`,
+        details: this.details
+      }
+    };
+  }
+}
+exports.InternalInvalidInputError = InternalInvalidInputError;

package/dist/common/runApi/index.d.ts

@@ -9,4 +9,4 @@ export declare function runApiGenerator<ResultType = any, _YieldType = any, _Nex
 }): AsyncGenerator<_YieldType, RunApiResult<ResultType, RunApiResultWithSessionOk>, _NextType>;
 export declare function runApiGenerator<ResultType = any, _YieldType = any, _NextType = any>(params: ExtendedRunApiParameters): AsyncGenerator<_YieldType, RunApiResult<ResultType>, _NextType>;
 export declare function runApi<ResultType = any>(params: ExtendedRunApiParameters): Promise<RunApiResult<ResultType>>;
-export declare function checkAuthSessionWithRetries(page: Page, context: BrowserContext, retries?: number): Promise<Result<boolean, RunAutomationError>>;
+export declare function checkAuthSessionWithRetries(page: Page, context: BrowserContext, importFunction: ExtendedRunApiParameters["importFunction"], retries?: number): Promise<Result<boolean, RunAutomationError>>;

package/dist/common/runApi/index.js

@@ -15,7 +15,6 @@ var _downloadDirectory = require("../../runtime/downloadDirectory");
 var _asyncLocalStorage = require("../asyncLocalStorage");
 var _fsExtra = _interopRequireWildcard(require("fs-extra"));
 var fs = _fsExtra;
-var _jwtTokenManager = require("../jwtTokenManager");
 var _contextStorageStateHelpers = require("../contextStorageStateHelpers");
 var _neverthrow = require("neverthrow");
 var _errors = require("./errors");
@@ -44,26 +43,32 @@ Object.keys(_types).forEach(function (key) {
     }
   });
 });
+var _formatZodError = require("../formatZodError");
+var _cleanEnvironmentVariables = require("../cleanEnvironmentVariables");
 function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
 function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
 async function* runApiGenerator({
   retrieveSession = false,
   abortSignal,
+  importFunction,
   ...input
 }) {
   let traceStarted = false;
+  const inputParseResult = _types.runApiParametersSchema.safeParse(input);
+  if (!inputParseResult.success) {
+    return (0, _neverthrow.err)(new _errors.InternalInvalidInputError("Input validation failed", (0, _formatZodError.formatZodError)(inputParseResult.error)));
+  }
   const {
     automationFunction,
     runOptions,
     tracing,
-    auth,
-    functionsToken
+    auth
   } = _types.runApiParametersSchema.parse(input);
-  _jwtTokenManager.backendFunctionsTokenManager.token = functionsToken;
-  const downloadsPath = (0, _downloadDirectory.getDownloadDirectoryPath)();
   let page;
   let context;
+  let downloadsPath;
   if (runOptions.environment === "deployed") {
+    downloadsPath = (0, _downloadDirectory.getDownloadDirectoryPath)();
     const {
       headless,
       proxy
@@ -87,10 +92,6 @@ async function* runApiGenerator({
       context
     } = await (0, _getPlaywrightConstructs.getPlaywrightConstructsForMode)(mode, cdpAddress, auth === null || auth === void 0 ? void 0 : auth.session));
   }
-  const executionContext = (0, _asyncLocalStorage.getExecutionContext)();
-  if (!executionContext) {
-    throw "";
-  }
   async function saveTraceIfNeeded({
     errorMessage
   }) {
@@ -122,18 +123,20 @@ async function* runApiGenerator({
     });
   });
   if (auth && auth.session.type === "state") {
-    await (0, _contextStorageStateHelpers.setContextStorageState)(context, auth.session.state);
+    const state = auth.session.state;
+    if (state === undefined || state === null) {
+      return (0, _neverthrow.err)(new _errors.AuthRequiredError());
+    }
+    await (0, _contextStorageStateHelpers.setContextStorageState)(context, state);
   }
   async function* runAutomation() {
     var _getExecutionContext;
     if (auth !== null && auth !== void 0 && auth.runCheck) {
       if (!auth.session) {
-        return (0, _neverthrow.err)({
-          code: "AuthRequiredError"
-        });
+        return (0, _neverthrow.err)(new _errors.AuthRequiredError());
       }
       console.log("Running auth check");
-      const authCheckResult = await checkAuthSessionWithRetries(page, context, 2);
+      const authCheckResult = await checkAuthSessionWithRetries(page, context, importFunction, 2);
       if (authCheckResult.isErr()) {
         const error = authCheckResult.error;
         if (["APINotFoundError", "InvalidAPIError"].includes(error.code)) {
@@ -142,20 +145,18 @@ async function* runApiGenerator({
         return authCheckResult;
       }
       if (!authCheckResult.value) {
-        return (0, _neverthrow.err)({
-          code: "AuthCheckFailedError"
-        });
+        return (0, _neverthrow.err)(new _errors.AuthCheckFailedError());
       }
     }
-    console.log("Running function", automationFunction.name);
-    const args = [...(automationFunction.params ? [automationFunction.params] : []), page, context];
-    const importResult = await importFunction(automationFunction.name);
-    if (importResult.isErr()) {
-      return importResult;
+    (0, _cleanEnvironmentVariables.cleanEnvironmentVariables)();
+    const args = [...(automationFunction.params !== undefined ? [automationFunction.params] : []), page, context];
+    const validatedModuleResult = await importUsingImportFunction(automationFunction.name, importFunction);
+    if (validatedModuleResult.isErr()) {
+      return validatedModuleResult;
     }
     let result;
-    if (importResult.value.type === "async-generator") {
-      const generator = importResult.value.generator(...args);
+    if (validatedModuleResult.value.type === "async-generator") {
+      const generator = validatedModuleResult.value.generator(...args);
       let next = undefined;
       while (true) {
         const generatorResult = await generator.next(...(next ? [next] : []));
@@ -167,7 +168,7 @@ async function* runApiGenerator({
         break;
       }
     } else {
-      result = await importResult.value.func(...args);
+      result = await validatedModuleResult.value.func(...args);
     }
     return (0, _neverthrow.ok)({
       result,
@@ -190,14 +191,15 @@ async function* runApiGenerator({
       return result.value;
     }
   } catch (error) {
-    console.log("run errored", error);
     return (0, _neverthrow.err)(new _errors.AutomationError(error));
   } finally {
     await saveTraceIfNeeded({
       errorMessage: "failed to save trace"
     });
     await context.close();
-    await fs.remove(downloadsPath);
+    if (downloadsPath !== undefined) {
+      await fs.remove(downloadsPath);
+    }
   }
 }
 async function runApi(params) {
@@ -211,13 +213,13 @@ async function runApi(params) {
   }
   return (0, _neverthrow.err)(new _errors.InvalidApiError("Expected API to be async function, got async generator"));
 }
-async function checkAuthSessionWithRetries(page, context, retries = 3) {
+async function checkAuthSessionWithRetries(page, context, importFunction, retries = 3) {
   if (retries === 0) {
     return (0, _neverthrow.ok)(false);
   }
   let tryNumber = 0;
   console.log("Checking auth session with retries", `${_constants.AUTH_SESSIONS_FOLDER_NAME}/check`);
-  const importResult = await importFunction(`${_constants.AUTH_SESSIONS_FOLDER_NAME}/check`);
+  const importResult = await importUsingImportFunction(`${_constants.AUTH_SESSIONS_FOLDER_NAME}/check`, importFunction);
   if (importResult.isErr()) {
     return (0, _neverthrow.err)(importResult.error);
   }
@@ -232,33 +234,9 @@ async function checkAuthSessionWithRetries(page, context, retries = 3) {
   }
   return (0, _neverthrow.ok)(false);
 }
-async function importFunction(path) {
-  const functionNameParts = path.split("/");
-  const functionNameDepth = functionNameParts.length;
+async function importUsingImportFunction(path, importFunction) {
   try {
-    let imported = undefined;
-    switch (functionNameDepth) {
-      case 1:
-        imported = await (specifier => new Promise(r => r(specifier)).then(s => _interopRequireWildcard(require(s))))(`${functionNameParts[0]}.ts`);
-        break;
-      case 2:
-        imported = await (specifier => new Promise(r => r(specifier)).then(s => _interopRequireWildcard(require(s))))(`${functionNameParts[0]}/${functionNameParts[1]}.ts`);
-        break;
-      case 3:
-        imported = await (specifier => new Promise(r => r(specifier)).then(s => _interopRequireWildcard(require(s))))(`${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}.ts`);
-        break;
-      case 4:
-        imported = await (specifier => new Promise(r => r(specifier)).then(s => _interopRequireWildcard(require(s))))(`${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}.ts`);
-        break;
-      case 5:
-        imported = await (specifier => new Promise(r => r(specifier)).then(s => _interopRequireWildcard(require(s))))(`${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}/${functionNameParts[4]}.ts`);
-        break;
-      case 6:
-        imported = await (specifier => new Promise(r => r(specifier)).then(s => _interopRequireWildcard(require(s))))(`${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}/${functionNameParts[4]}/${functionNameParts[5]}.ts`);
-        break;
-      default:
-        return (0, _neverthrow.err)(new _errors.MaxLevelsExceededError(5));
-    }
+    const imported = importFunction ? await importFunction(path) : await (specifier => new Promise(r => r(specifier)).then(s => _interopRequireWildcard(require(s))))(`./${path}.ts`);
     if (!imported || !imported.default || !imported.default.constructor) {
       return (0, _neverthrow.err)(new _errors.InvalidApiError("API file path does not have a default export"));
     }