@intuned/runtime-dev 0.1.0-test.11 → 0.1.0-test.13

package/dist/common/runApi/index.js

@@ -2,252 +2,238 @@ import { getDownloadDirectoryPath } from "../../runtime/downloadDirectory";
 import { getExecutionContext } from "../asyncLocalStorage";
 import * as fs from "fs-extra";
 import { backendFunctionsTokenManager } from "../jwtTokenManager";
-import { getContextStorageState, setContextStorageState } from "../contextStorageStateHelpers";
+import { getContextStorageState, setContextStorageState, } from "../contextStorageStateHelpers";
 import { remove } from "fs-extra";
 import { ok, err } from "neverthrow";
-import { AbortedError, ApiNotFoundError, AutomationError, InvalidApiError, InvalidCheckError, MaxLevelsExceededError } from "./errors";
+import { AbortedError, ApiNotFoundError, AutomationError, InvalidApiError, InvalidCheckError, MaxLevelsExceededError, } from "./errors.js";
 import { AUTH_SESSIONS_FOLDER_NAME } from "../constants";
-import { getPlaywrightConstructsForMode, getProductionPlaywrightConstructs } from "../getPlaywrightConstructs";
-import { runApiParametersSchema } from "./types";
-export * from "./types";
-export * from "./errors";
-export async function* runApiGenerator({
-  retrieveSession = false,
-  abortSignal,
-  ...input
-}) {
-  let traceStarted = false;
-  const {
-    automationFunction,
-    runOptions,
-    tracing,
-    auth,
-    functionsToken
-  } = runApiParametersSchema.parse(input);
-  backendFunctionsTokenManager.token = functionsToken;
-  const downloadsPath = getDownloadDirectoryPath();
-  let page;
-  let context;
-  if (runOptions.environment === "deployed") {
-    const {
-      headless,
-      proxy
-    } = runOptions;
-    ({
-      page,
-      context
-    } = await getProductionPlaywrightConstructs({
-      headless,
-      proxy,
-      downloadsPath,
-      storageState: auth === null || auth === void 0 ? void 0 : auth.session
-    }));
-  } else {
-    const {
-      mode,
-      cdpAddress
-    } = runOptions;
-    ({
-      page,
-      context
-    } = await getPlaywrightConstructsForMode(mode, cdpAddress, auth === null || auth === void 0 ? void 0 : auth.session));
-  }
-  const executionContext = getExecutionContext();
-  if (!executionContext) {
-    throw "";
-  }
-  async function saveTraceIfNeeded({
-    errorMessage
-  }) {
-    if (!tracing.enabled || !traceStarted) {
-      return;
+import { getPlaywrightConstructsForMode, getProductionPlaywrightConstructs, } from "../getPlaywrightConstructs";
+import { runApiParametersSchema, } from "./types.js";
+export * from "./types.js";
+export * from "./errors.js";
+export async function* runApiGenerator({ retrieveSession = false, abortSignal, ...input }) {
+    let traceStarted = false;
+    const { automationFunction, runOptions, tracing, auth, functionsToken } = runApiParametersSchema.parse(input);
+    backendFunctionsTokenManager.token = functionsToken;
+    const downloadsPath = getDownloadDirectoryPath();
+    let page;
+    let context;
+    if (runOptions.environment === "deployed") {
+        const { headless, proxy } = runOptions;
+        ({ page, context } = await getProductionPlaywrightConstructs({
+            headless,
+            proxy,
+            downloadsPath,
+            storageState: auth?.session,
+        }));
     }
-    try {
-      await context.tracing.stop({
-        path: tracing.filePath
-      });
-    } catch (error) {
-      console.log(errorMessage, error === null || error === void 0 ? void 0 : error.message);
-      await remove(tracing.filePath);
-    }
-  }
-  if (tracing.enabled) {
-    await context.tracing.start({
-      screenshots: true,
-      snapshots: true,
-      sources: true
-    });
-    traceStarted = true;
-  }
-  const abortSymbol = Symbol("abort");
-  const abortPromise = new Promise(resolve => {
-    if (!abortSignal) return;
-    abortSignal.addEventListener("abort", () => {
-      resolve(abortSymbol);
-    });
-  });
-  if (auth && auth.session.type === "state") {
-    await setContextStorageState(context, auth.session.state);
-  }
-  async function* runAutomation() {
-    var _getExecutionContext;
-    if (auth !== null && auth !== void 0 && auth.runCheck) {
-      if (!auth.session) {
-        return err({
-          code: "AuthRequiredError"
+    else {
+        const { mode, cdpAddress } = runOptions;
+        ({ page, context } = await getPlaywrightConstructsForMode(mode, cdpAddress, auth?.session));
+    }
+    const executionContext = getExecutionContext();
+    if (!executionContext) {
+        throw ""; // todo
+    }
+    async function saveTraceIfNeeded({ errorMessage }) {
+        if (!tracing.enabled || !traceStarted) {
+            return;
+        }
+        try {
+            await context.tracing.stop({ path: tracing.filePath });
+        }
+        catch (error) {
+            console.log(errorMessage, error?.message);
+            await remove(tracing.filePath);
+        }
+    }
+    if (tracing.enabled) {
+        await context.tracing.start({
+            screenshots: true,
+            snapshots: true,
+            sources: true,
+        });
+        traceStarted = true;
+    }
+    const abortSymbol = Symbol("abort");
+    const abortPromise = new Promise((resolve) => {
+        if (!abortSignal)
+            return;
+        abortSignal.addEventListener("abort", () => {
+            resolve(abortSymbol);
         });
-      }
-      console.log("Running auth check");
-      const authCheckResult = await checkAuthSessionWithRetries(page, context, 2);
-      if (authCheckResult.isErr()) {
-        const error = authCheckResult.error;
-        if (["APINotFoundError", "InvalidAPIError"].includes(error.code)) {
-          return err(new InvalidCheckError(`Auth session check function failed`, error));
+    });
+    if (auth && auth.session.type === "state") {
+        await setContextStorageState(context, auth.session.state);
+    }
+    async function* runAutomation() {
+        if (auth?.runCheck) {
+            if (!auth.session) {
+                return err({ code: "AuthRequiredError" });
+            }
+            console.log("Running auth check");
+            const authCheckResult = await checkAuthSessionWithRetries(page, context, 2);
+            if (authCheckResult.isErr()) {
+                const error = authCheckResult.error;
+                if (["APINotFoundError", "InvalidAPIError"].includes(error.code)) {
+                    return err(new InvalidCheckError(`Auth session check function failed`, error));
+                }
+                return authCheckResult;
+            }
+            if (!authCheckResult.value) {
+                return err({
+                    code: "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;
+        }
+        let result;
+        if (importResult.value.type === "async-generator") {
+            const generator = importResult.value.generator(...args);
+            let next = undefined;
+            while (true) {
+                const generatorResult = await generator.next(...(next ? [next] : []));
+                if (!generatorResult.done) {
+                    next = yield generatorResult.value;
+                    continue;
+                }
+                result = generatorResult.value;
+                break;
+            }
         }
-        return authCheckResult;
-      }
-      if (!authCheckResult.value) {
-        return err({
-          code: "AuthCheckFailedError"
+        else {
+            result = await importResult.value.func(...args);
+        }
+        return ok({
+            result,
+            extendedPayloads: getExecutionContext()?.extendedPayloads,
+            session: retrieveSession
+                ? await getContextStorageState(context)
+                : undefined,
         });
-      }
     }
-    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;
-    }
-    let result;
-    if (importResult.value.type === "async-generator") {
-      const generator = importResult.value.generator(...args);
-      let next = undefined;
-      while (true) {
-        const generatorResult = await generator.next(...(next ? [next] : []));
-        if (!generatorResult.done) {
-          next = yield generatorResult.value;
-          continue;
+    try {
+        const generator = runAutomation();
+        let next = undefined;
+        while (true) {
+            const result = await Promise.race([
+                generator.next(await next),
+                abortPromise,
+            ]);
+            if (result === abortSymbol) {
+                return err(new AbortedError());
+            }
+            if (!result.done) {
+                next = yield result.value;
+                continue;
+            }
+            return result.value;
         }
-        result = generatorResult.value;
-        break;
-      }
-    } else {
-      result = await importResult.value.func(...args);
-    }
-    return ok({
-      result,
-      extendedPayloads: (_getExecutionContext = getExecutionContext()) === null || _getExecutionContext === void 0 ? void 0 : _getExecutionContext.extendedPayloads,
-      session: retrieveSession ? await getContextStorageState(context) : undefined
-    });
-  }
-  try {
-    const generator = runAutomation();
-    let next = undefined;
-    while (true) {
-      const result = await Promise.race([generator.next(await next), abortPromise]);
-      if (result === abortSymbol) {
-        return err(new AbortedError());
-      }
-      if (!result.done) {
-        next = yield result.value;
-        continue;
-      }
-      return result.value;
-    }
-  } catch (error) {
-    console.log("run errored", error);
-    return err(new AutomationError(error));
-  } finally {
-    await saveTraceIfNeeded({
-      errorMessage: "failed to save trace"
-    });
-    await context.close();
-    await fs.remove(downloadsPath);
-  }
+    }
+    catch (error) {
+        console.log("run errored", error);
+        return err(new AutomationError(error));
+    }
+    finally {
+        await saveTraceIfNeeded({ errorMessage: "failed to save trace" });
+        await context.close();
+        await fs.remove(downloadsPath);
+    }
 }
 export async function runApi(params) {
-  const generator = runApiGenerator(params);
-  const {
-    value,
-    done
-  } = await generator.next();
-  if (done) {
-    return value;
-  }
-  return err(new InvalidApiError("Expected API to be async function, got async generator"));
+    const generator = runApiGenerator(params);
+    const { value, done } = await generator.next();
+    if (done) {
+        return value;
+    }
+    return err(new InvalidApiError("Expected API to be async function, got async generator"));
 }
 export async function checkAuthSessionWithRetries(page, context, retries = 3) {
-  if (retries === 0) {
+    if (retries === 0) {
+        return ok(false);
+    }
+    let tryNumber = 0;
+    console.log("Checking auth session with retries", `${AUTH_SESSIONS_FOLDER_NAME}/check`);
+    const importResult = await importFunction(`${AUTH_SESSIONS_FOLDER_NAME}/check`);
+    if (importResult.isErr()) {
+        return err(importResult.error);
+    }
+    if (importResult.value.type !== "async") {
+        return err(new InvalidCheckError("Check function is not an async function"));
+    }
+    const check = importResult.value.func;
+    while (retries > tryNumber) {
+        const result = await check(page, context);
+        if (result)
+            return ok(true);
+        tryNumber++;
+    }
     return ok(false);
-  }
-  let tryNumber = 0;
-  console.log("Checking auth session with retries", `${AUTH_SESSIONS_FOLDER_NAME}/check`);
-  const importResult = await importFunction(`${AUTH_SESSIONS_FOLDER_NAME}/check`);
-  if (importResult.isErr()) {
-    return err(importResult.error);
-  }
-  if (importResult.value.type !== "async") {
-    return err(new InvalidCheckError("Check function is not an async function"));
-  }
-  const check = importResult.value.func;
-  while (retries > tryNumber) {
-    const result = await check(page, context);
-    if (result) return ok(true);
-    tryNumber++;
-  }
-  return ok(false);
 }
 async function importFunction(path) {
-  const functionNameParts = path.split("/");
-  const functionNameDepth = functionNameParts.length;
-  try {
-    let imported = undefined;
-    switch (functionNameDepth) {
-      case 1:
-        imported = await import(`../../../${functionNameParts[0]}.ts`);
-        break;
-      case 2:
-        imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}.ts`);
-        break;
-      case 3:
-        imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}.ts`);
-        break;
-      case 4:
-        imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}.ts`);
-        break;
-      case 5:
-        imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}/${functionNameParts[4]}.ts`);
-        break;
-      case 6:
-        imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}/${functionNameParts[4]}/${functionNameParts[5]}.ts`);
-        break;
-      default:
-        return err(new MaxLevelsExceededError(5));
-    }
-    if (!imported || !imported.default || !imported.default.constructor) {
-      return err(new InvalidApiError("API file path does not have a default export"));
-    }
-    if (imported.default.constructor.name === "AsyncGeneratorFunction") {
-      return ok({
-        type: "async-generator",
-        generator: imported.default
-      });
-    }
-    if (imported.default.constructor.name === "AsyncFunction") {
-      return ok({
-        type: "async",
-        func: imported.default
-      });
-    }
-    return err(new InvalidApiError("API file path does not have a default async function/generator export"));
-  } catch (error) {
-    if (error.message.includes("Unknown variable dynamic import")) {
-      return err(new ApiNotFoundError(path));
-    }
-    if ((error === null || error === void 0 ? void 0 : error.code) === "ERR_MODULE_NOT_FOUND") {
-      console.log("API not found", error);
-      return err(new ApiNotFoundError(path));
-    }
-    throw error;
-  }
-}
+    const 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 = undefined;
+        switch (functionNameDepth) {
+            case 1:
+                imported = await import(`../../../${functionNameParts[0]}.ts`);
+                break;
+            case 2:
+                imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}.ts`);
+                break;
+            case 3:
+                imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}.ts`);
+                break;
+            case 4:
+                imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}.ts`);
+                break;
+            case 5:
+                imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}/${functionNameParts[4]}.ts`);
+                break;
+            case 6:
+                imported = await import(`../../../${functionNameParts[0]}/${functionNameParts[1]}/${functionNameParts[2]}/${functionNameParts[3]}/${functionNameParts[4]}/${functionNameParts[5]}.ts`);
+                break;
+            default:
+                return err(new MaxLevelsExceededError(5));
+        }
+        if (!imported || !imported.default || !imported.default.constructor) {
+            return err(new InvalidApiError("API file path does not have a default export"));
+        }
+        if (imported.default.constructor.name === "AsyncGeneratorFunction") {
+            return ok({
+                type: "async-generator",
+                generator: imported.default,
+            });
+        }
+        if (imported.default.constructor.name === "AsyncFunction") {
+            return ok({
+                type: "async",
+                func: imported.default,
+            });
+        }
+        return err(new InvalidApiError("API file path does not have a default async function/generator export"));
+    }
+    catch (error) {
+        if (error.message.includes("Unknown variable dynamic import")) {
+            return err(new ApiNotFoundError(path));
+        }
+        if (error?.code === "ERR_MODULE_NOT_FOUND") {
+            console.log("API not found", error);
+            return err(new ApiNotFoundError(path));
+        }
+        throw error;
+    }
+}