gologin-commonjs 2.1.1

package/gologin/src/gologin-api.js

@@ -0,0 +1,110 @@
+import puppeteer from 'puppeteer-core';
+
+import GoLogin from './gologin.js';
+
+export function getDefaultParams() {
+  return {
+    token: process.env.GOLOGIN_API_TOKEN,
+    profile_id: process.env.GOLOGIN_PROFILE_ID,
+    executablePath: process.env.GOLOGIN_EXECUTABLE_PATH,
+    autoUpdateBrowser: true,
+  };
+}
+
+const createLegacyGologin = ({ profileId, ...params }) => {
+  const defaults = getDefaultParams();
+  const mergedParams = {
+    ...defaults,
+    ...params,
+  };
+
+  mergedParams.profile_id = profileId ?? mergedParams.profile_id;
+
+  console.log({ mergedParams });
+
+  return new GoLogin(mergedParams);
+};
+
+const createdApis = [];
+
+export const delay = (ms = 250) => new Promise((res) => setTimeout(res, ms));
+
+export function GologinApi({ token }) {
+  if (!token) {
+    throw new Error('GoLogin API token is missing');
+  }
+
+  const browsers = [];
+  const legacyGls = [];
+
+  const launchLocal = async (params) => {
+    const legacyGologin = createLegacyGologin({
+      ...params,
+      token,
+    });
+
+    if (!params.profileId) {
+      const { id } = await legacyGologin.quickCreateProfile();
+      await legacyGologin.setProfileId(id);
+    }
+
+    const started = await legacyGologin.start();
+    const browser = await puppeteer.connect({
+      browserWSEndpoint: started.wsUrl,
+      ignoreHTTPSErrors: true,
+    });
+
+    browsers.push(browser);
+    legacyGls.push(legacyGologin);
+
+    return { browser };
+  };
+
+  const launchCloudProfile = async (params) => {
+    const profileParam = params.profileId
+      ? `&profile=${params.profileId}`
+      : '';
+
+    const geolocationParam = params.geolocation
+      ? `&geolocation=${params.geolocation}`
+      : '';
+
+    const browserWSEndpoint = `https://cloud.gologin.com/connect?token=${token}${profileParam}${geolocationParam}`;
+    const browser = await puppeteer.connect({
+      browserWSEndpoint,
+      ignoreHTTPSErrors: true,
+    });
+
+    browsers.push(browser);
+
+    return { browser };
+  };
+
+  const api = {
+    async launch(params = {}) {
+      if (params.cloud) {
+        return launchCloudProfile(params);
+      }
+
+      return launchLocal(params);
+    },
+
+    async exit(status = 0) {
+      Promise.allSettled(browsers.map((browser) => browser.close()));
+      Promise.allSettled(
+        legacyGls.map((gl) => gl.stopLocal({ posting: false })),
+      );
+      process.exit(status);
+    },
+
+    delay,
+  };
+
+  createdApis.push(api);
+
+  return api;
+}
+
+export function exitAll() {
+  Promise.allSettled(createdApis.map((api) => api.exit()));
+}