@intuned/runtime-dev 0.1.0-test.8 → 1.0.0-udas.0

package/WebTemplate/utils.ts

@@ -1,7 +1,7 @@
 import * as playwright from "@intuned/playwright-core";
 import { Handler, Response } from "@tinyhttp/app";
 import * as path from "path";
-import { getExecutionContext } from "@intuned/runtime/dist/common/asyncLocalStorage";
+import { getExecutionContext } from "@intuned/runtime";
 import { setTimeout } from "timers/promises";
 
 export class FunctionNotFoundError extends Error {
@@ -208,3 +208,50 @@ export async function waitWithExtendableTimeout<T>({
     result: await promise,
   };
 }
+
+export async function importModule(path: string) {
+  const [folderName, ...functionNameParts] = path.split("/");
+  const functionNameDepth = functionNameParts.length;
+
+  // string literals should be inline
+  // currently we support only 5 levels of depth
+  // rollup dynamic import does not support multiple levels of dynamic imports so we need to specify the possible paths explicitly
+  try {
+    let imported: any = undefined;
+    switch (functionNameDepth) {
+      case 1:
+        imported = await import(`./${folderName}/${functionNameParts[0]}.ts`);
+        break;
+      case 2:
+        imported = await import(
+          `./${folderName}/${functionNameParts[0]}/${functionNameParts[1]}.ts`
+        );
+        break;
+      case 3:
+        imported = await import(
+          `./${folderName}/${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}.ts`
+        );
+        break;
+      case 4:
+        imported = await import(
+          `./${folderName}/${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}.ts`
+        );
+        break;
+      case 5:
+        imported = await import(
+          `./${folderName}/${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}/${functionNameParts[4]}.ts`
+        );
+        break;
+      default:
+        throw new Error(
+          "intuned supports maximum 5 levels of depth in the api folder"
+        );
+    }
+    return imported;
+  } catch (error: any) {
+    if (error.message.includes("Unknown variable dynamic import")) {
+      throw new FunctionNotFoundError("", path);
+    }
+    throw error;
+  }
+}

package/bin/intuned-ts-check

@@ -1,2 +1,2 @@
 #!/usr/bin/env node
-require("../dist/commands/ts-check.js");
+require("../dist/commands/ts-check.js");

package/dist/commands/api/run.js

@@ -16,6 +16,7 @@ var _Logger = require("../../common/Logger");
 var _nanoid = require("nanoid");
 var _chalk = _interopRequireDefault(require("chalk"));
 var _runApi = require("../../common/runApi");
+var _tsNodeImport = require("../common/tsNodeImport");
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 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; }
@@ -39,7 +40,7 @@ async function executeCLI(apiName, mode, inputData, options) {
   const runApiResult = await (0, _runApi.runApi)({
     automationFunction: {
       name: `api/${apiName}`,
-      params: inputData ?? {}
+      params: inputData
     },
     auth: authSessionPathToUse ? {
       session: {
@@ -49,17 +50,18 @@ async function executeCLI(apiName, mode, inputData, options) {
       runCheck: false
     } : undefined,
     runOptions: {
-      environment: "ide",
+      environment: "cdp",
       cdpAddress: options.cdpAddress,
       mode
-    }
+    },
+    importFunction: _tsNodeImport.tsNodeImport
   });
   if (runApiResult.isErr()) {
     if (runApiResult.error instanceof _runApi.AutomationError) {
       throw runApiResult.error.error;
     }
-    console.error("An error occurred while running the API", runApiResult.error);
-    return;
+    console.error(runApiResult.error);
+    throw new Error("An error occurred while running the API");
   }
   const {
     result,

package/dist/commands/auth-sessions/run-check.js

@@ -8,6 +8,7 @@ var _settings = require("../common/utils/settings");
 var _dotenv = _interopRequireDefault(require("dotenv"));
 var _constants = require("../../common/constants");
 var _runApi = require("../../common/runApi");
+var _tsNodeImport = require("../common/tsNodeImport");
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 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; }
@@ -31,7 +32,7 @@ _commander.program.description("run auth session check").option("--cdpAddress <c
       name: `${_constants.AUTH_SESSIONS_FOLDER_NAME}/check`
     },
     runOptions: {
-      environment: "ide",
+      environment: "cdp",
       mode,
       cdpAddress
     },
@@ -41,10 +42,15 @@ _commander.program.description("run auth session check").option("--cdpAddress <c
         path: authSessionPath
       },
       runCheck: false
-    }
+    },
+    importFunction: _tsNodeImport.tsNodeImport
   });
   if (runApiResult.isErr()) {
-    throw new Error(runApiResult.error);
+    if (runApiResult.error instanceof _runApi.AutomationError) {
+      throw runApiResult.error.error;
+    }
+    console.error(runApiResult.error);
+    throw new Error("Error running auth session check");
   }
   const result = runApiResult.value.result;
   console.log("check result", result);

package/dist/commands/auth-sessions/run-create.js

@@ -12,6 +12,7 @@ var _runApi = require("../../common/runApi");
 var _asyncLocalStorage = require("../../common/asyncLocalStorage");
 var _nanoid = require("nanoid");
 var _enums = require("../../runtime/enums");
+var _tsNodeImport = require("../common/tsNodeImport");
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 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; }
@@ -43,15 +44,16 @@ _commander.program.description("run auth session create").option("--cdpAddress <
   async function runCreate() {
     const generator = (0, _runApi.runApiGenerator)({
       automationFunction: {
-        name: "auth-sessions/create",
-        params: inputData ?? {}
+        name: `${_constants.AUTH_SESSIONS_FOLDER_NAME}/create`,
+        params: inputData
       },
       runOptions: {
-        environment: "ide",
+        environment: "cdp",
         mode,
         cdpAddress
       },
-      retrieveSession: true
+      retrieveSession: true,
+      importFunction: _tsNodeImport.tsNodeImport
     });
     let nextGeneratorParam = undefined;
     while (true) {
@@ -61,8 +63,11 @@ _commander.program.description("run auth session create").option("--cdpAddress <
       } = await generator.next(...(nextGeneratorParam ? [nextGeneratorParam] : []));
       if (done) {
         if (value.isErr()) {
-          console.log(value.error);
-          throw new Error(value.error);
+          if (value.error instanceof _runApi.AutomationError) {
+            throw value.error.error;
+          }
+          console.error(value.error);
+          throw new Error("Error while running create");
         }
         const fullState = value.value.session;
         if (pathToSave) {

package/dist/commands/build.js

@@ -10,7 +10,7 @@ var _pluginDynamicImportVars = _interopRequireDefault(require("@rollup/plugin-dy
 var _pluginJson = _interopRequireDefault(require("@rollup/plugin-json"));
 var fs = _interopRequireWildcard(require("fs-extra"));
 var path = _interopRequireWildcard(require("path"));
-var _webTemplate = require("./common/utils/webTemplate");
+var _template = require("./common/utils/template");
 var _dotenv = _interopRequireDefault(require("dotenv"));
 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; }
@@ -21,18 +21,20 @@ function isThirdPartyWarning(warning) {
   return false;
 }
 _dotenv.default.config();
-_commander.program.description("build the intuned server").addArgument(new _commander.Argument("<mode>", "mode of execution").choices(["vanilla", "playwright"]).default("playwright").argOptional()).argument("[outfile]", "output bundle", "./output/bundle_v2.js").allowUnknownOption().action(async (mode, outfile) => {
-  await (0, _webTemplate.moveWebTemplateFiles)();
-  const currentTemplateTsConfig = path.resolve(__dirname, "..", "..", "template.tsconfig.json");
-  await fs.copy(currentTemplateTsConfig, "./intuned/WebTemplate/tsconfig.json");
-  await build(outfile, mode);
+_commander.program.description("build the intuned server").option("-t, --template <type>", "template to use", "WebTemplate").addArgument(new _commander.Argument("<mode>", "mode of execution").choices(["vanilla", "playwright"]).default("playwright").argOptional()).argument("[outfile]", "output bundle", "./output/bundle_v2.js").allowUnknownOption().action(async (mode, outfile, {
+  template
+}) => {
+  await (0, _template.moveTemplateFiles)(template);
+  const currentTemplateTsConfig = path.resolve(path.dirname(__filename), "..", "..", "template.tsconfig.json");
+  await fs.copy(currentTemplateTsConfig, `./intuned/${template}/tsconfig.json`);
+  await build(outfile, mode, template);
 });
-async function build(outfile, mode) {
+async function build(outfile, mode, template) {
   let bundle = null;
   let buildFailed = false;
   try {
     bundle = await (0, _rollup.rollup)({
-      input: `./intuned/WebTemplate/index.${mode}.ts`,
+      input: `./intuned/${template}/index.${mode}.ts`,
       output: {
         globals: {
           crypto: "crypto"
@@ -42,13 +44,15 @@ async function build(outfile, mode) {
         exportConditions: ["node"],
         preferBuiltins: true
       }), (0, _pluginTypescript.default)({
-        tsconfig: "./intuned/WebTemplate/tsconfig.json"
+        tsconfig: `./intuned/${template}/tsconfig.json`
       }), (0, _pluginCommonjs.default)({
-        include: ["node_modules/**", "intuned/WebTemplate/**", "dist", "../typescript-sdk/**", "../typescript-runtime/**"],
+        include: ["node_modules/**", `intuned/${template}/**`, "dist", "../typescript-sdk/**", "../typescript-runtime/**"],
         extensions: [".js"],
         ignoreGlobal: false,
         sourceMap: false,
-        dynamicRequireTargets: ["node_modules/@intuned/runtime/**/*.js"]
+        dynamicRequireTargets: ["api/**/*.js", "auth-sessions/**/*.js"],
+        requireReturnsDefault: "auto",
+        transformMixedEsModules: true
       }), (0, _pluginDynamicImportVars.default)({
         include: ["**/*.js", "**/*.ts"],
         exclude: ["**/*.d.ts", "**/*.js.map"]
@@ -61,7 +65,7 @@ async function build(outfile, mode) {
     });
     console.log(`📦 Building ${outfile}`);
     const outfileFolder = path.dirname(outfile);
-    const assetsDir = path.resolve(__dirname, "..", "common", "assets");
+    const assetsDir = path.resolve(path.dirname(__filename), "..", "common", "assets");
     await fs.copy(assetsDir, `${outfileFolder}/assets`);
     await bundle.write({
       format: "cjs",

package/dist/commands/common/tsNodeImport.d.ts

@@ -0,0 +1 @@
+export declare function tsNodeImport(apiName: string): Promise<any>;

package/dist/commands/common/tsNodeImport.js

@@ -0,0 +1,20 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.tsNodeImport = tsNodeImport;
+var _fileUtils = require("./utils/fileUtils");
+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 tsNodeImport(apiName) {
+  require("ts-node").register({
+    transpileOnly: true,
+    compilerOptions: {
+      lib: ["dom", "es2020"]
+    }
+  });
+  const path = (0, _fileUtils.getFullPathInProject)(...apiName.split("/"));
+  const imported = await (specifier => new Promise(r => r(`${specifier}`)).then(s => _interopRequireWildcard(require(s))))(path);
+  return imported;
+}

package/dist/commands/common/utils/settings.js

@@ -6,14 +6,14 @@ Object.defineProperty(exports, "__esModule", {
 exports.getSettings = getSettings;
 var _fileUtils = require("./fileUtils");
 var _settingsSchema = require("../../../common/settingsSchema");
-var _fsExtra = require("fs-extra");
+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 getSettings() {
   const settingsFilePath = (0, _fileUtils.getFullPathInProject)("Intuned.json");
   console.log("loading settings");
-  const settings = await (0, _fsExtra.readJSON)(settingsFilePath);
-  if (settings) {
-    const parsed = _settingsSchema.settingsSchema.safeParse(settings);
-    console.log(parsed);
+  const settings = await (specifier => new Promise(r => r(`${specifier}`)).then(s => _interopRequireWildcard(require(s))))(settingsFilePath);
+  if (settings.default) {
+    const parsed = _settingsSchema.settingsSchema.safeParse(settings.default);
     if (parsed.success) {
       return parsed.data;
     } else {

package/dist/commands/common/utils/template.d.ts

@@ -0,0 +1,2 @@
+export type TemplateName = "WebTemplate" | "InterfaceTemplate";
+export declare const moveTemplateFiles: (templateName: TemplateName) => Promise<void>;

package/dist/commands/common/utils/{webTemplate.js → template.js}

@@ -3,18 +3,18 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.moveWebTemplateFiles = void 0;
+exports.moveTemplateFiles = void 0;
 var fs = _interopRequireWildcard(require("fs-extra"));
 var path = _interopRequireWildcard(require("path"));
 var _fileUtils = require("./fileUtils");
 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; }
-const moveWebTemplateFiles = async () => {
+const moveTemplateFiles = async templateName => {
   await fs.remove("./intuned");
   await fs.ensureDir("./intuned");
-  const currentFileLocation = path.resolve(__dirname, "..", "..", "..", "..", "WebTemplate");
-  await fs.copy(`${currentFileLocation}`, "./intuned/WebTemplate", {
-    filter: (src, dest) => {
+  const currentFileLocation = path.resolve(__dirname, "..", "..", "..", "..", templateName);
+  await fs.copy(`${currentFileLocation}`, `./intuned/${templateName}`, {
+    filter: (src, _) => {
       if (src.includes(".d.ts")) {
         return false;
       }
@@ -25,7 +25,7 @@ const moveWebTemplateFiles = async () => {
   const pathsIgnoreList = [(0, _fileUtils.getFullPathInProject)("intuned"), (0, _fileUtils.getFullPathInProject)("node_modules"), (0, _fileUtils.getFullPathInProject)("package.json"), (0, _fileUtils.getFullPathInProject)("yarn.lock"), (0, _fileUtils.getFullPathInProject)(".env")];
   const filesToCopy = filesAndFolders.filter(file => !pathsIgnoreList.includes(file.fullPath));
   for (const file of filesToCopy) {
-    await fs.copy(file.fullPath, `./intuned/WebTemplate/${file.name}`);
+    await fs.copy(file.fullPath, `./intuned/${templateName}/${file.name}`);
   }
 };
-exports.moveWebTemplateFiles = moveWebTemplateFiles;
+exports.moveTemplateFiles = moveTemplateFiles;

package/dist/commands/interface/run.d.ts

@@ -1,2 +1 @@
-#!/usr/bin/env node
-export {};
+export declare function runAutomationCLI(importFunction?: (path: string) => Promise<any>): void;

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");
+var _stream = require("stream");
 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,127 +42,157 @@ 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 = 60 * 1000;
+    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");
-      }
-      if (await runGeneratorAndSendResult(input.parameters.value)) {
-        break;
+    const stream = new _stream.PassThrough({
+      objectMode: true
+    });
+    async function messagesLoop() {
+      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;
+        }
+        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 === "tokenUpdate") {
+          _jwtTokenManager.backendFunctionsTokenManager.token = input.parameters.functionsToken;
+          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 === "standalone" ? _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)
+          };
+          stream.write({});
+          continue;
+        }
+        if (input.type === "next") {
+          if (!generator) {
+            throw new Error("generator not started");
+          }
+          stream.write({
+            value: input.parameters.value
+          });
+          continue;
+        }
       }
-      continue;
+      stream.end();
+      client.end();
     }
-  }
-  process.exit(0);
-});
-_commander.program.parse(process.argv);
+    void messagesLoop();
+    for await (const {
+      value
+    } of stream) {
+      const {
+        done
+      } = await runGeneratorAndSendResult(value);
+      if (done) break;
+    }
+    if (!client.closed) {
+      client.end();
+    }
+    process.exit(0);
+  });
+  _commander.program.parse(process.argv);
+}
 function getProxyUrlFromRunOptions(runOptions) {
-  if ((runOptions === null || runOptions === void 0 ? void 0 : runOptions.environment) !== "deployed") return undefined;
+  if ((runOptions === null || runOptions === void 0 ? void 0 : runOptions.environment) !== "standalone") return undefined;
   const proxy = runOptions.proxy;
   if (!proxy) return undefined;
   const url = new URL(proxy.server);
   url.username = proxy.username;
   url.password = proxy.password;
   return url.toString();
+}
+if (require.main === module) {
+  runAutomationCLI();
 }