@stubber/virtual-worker 1.4.2 → 1.5.0

package/app/commands/browser/browser_manage_sessions.js

@@ -0,0 +1,25 @@
+import { create_error_conceptual } from "#app/functions/create_error_conceptual.js";
+import { clear_chromium_sessions, list_chromium_sessions } from "../../helpers/get_chromium_page.js";
+
+/**
+ *
+ * @param {Object} params
+ * @param {string} params.operation - The operation to perform: "list" or "delete"
+ * @param {string} [params.session_ids] - The session IDs to delete (required for "delete" operation)
+ */
+export const browser_manage_sessions = async (params, stubber_context) => {
+  const { operation: operation, session_ids } = params || {};
+
+  if (operation === "list") {
+    return await list_chromium_sessions();
+  } else if (operation === "delete") {
+    if (!session_ids || !Array.isArray(session_ids)) {
+      return create_error_conceptual({
+        message: "session_ids is required and must be an array for 'delete' operation.",
+      });
+    }
+    return await clear_chromium_sessions(session_ids);
+  } else {
+    return create_error_conceptual({ message: "Invalid operation. Must be 'list' or 'delete'." });
+  }
+};

package/app/commands/index.js

@@ -7,6 +7,7 @@ import { browser_press_key } from "./browser/browser_press_key.js";
 import { browser_extract_html } from "./browser/browser_extract_html.js";
 import { browser_screenshot } from "./browser/browser_screenshot.js";
 import { browser_get_clipboard_text } from "./browser/browser_get_clipboard_text.js";
+import { browser_manage_sessions } from "./browser/browser_manage_sessions.js";
 import { cli_run } from "./cli/cli_run.js";
 
 import { upload_files } from "./file-server/upload_files.js";
@@ -24,6 +25,7 @@ const all_commands = {
   browser_extract_html,
   browser_screenshot,
   browser_get_clipboard_text,
+  browser_manage_sessions,
 
   cli_run,
 

package/app/helpers/get_chromium_page.js

@@ -31,11 +31,12 @@ export const get_chromium_page = async (params, stubber_context) => {
   }
 
   // console.log(`get_chromium_page: session_id=${session_id}, slow_mo=${slow_mo}`);
+  const headless = params?.headless ?? process.env.HEADLESS_CHROMIUM ?? false;
 
   // Launch the browser once and reuse it
   if (!browser) {
     browser = await chromium.launch({
-      headless: false,
+      headless: headless === true || headless === "true",
       // eslint-disable-next-line id-match
       slowMo: slow_mo,
     });
@@ -65,3 +66,35 @@ export const get_chromium_page = async (params, stubber_context) => {
 
   return create_success({ message: "Successfully retrieved Chromium page.", payload: page });
 };
+
+export const list_chromium_sessions = () => {
+  return create_success({
+    message: "Successfully listed Chromium sessions.",
+    payload: {
+      sessions: Object.keys(contexts),
+    },
+  });
+};
+
+export const clear_chromium_sessions = async (session_ids) => {
+  try {
+    let cleared_sessions = [];
+    for (const session_id of session_ids) {
+      const context_entry = contexts[session_id];
+      if (context_entry) {
+        await context_entry.context.close();
+        delete contexts[session_id];
+        cleared_sessions.push(session_id);
+      }
+    }
+    return create_success({
+      message: "Successfully cleared specified Chromium sessions.",
+      payload: {
+        cleared_sessions,
+        sessions: Object.keys(contexts),
+      },
+    });
+  } catch (error) {
+    return create_error_technical({ message: "Failed to clear Chromium sessions.", error });
+  }
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stubber/virtual-worker",
-  "version": "1.4.2",
+  "version": "1.5.0",
   "description": "Template to easily create a node app and keep development standards",
   "main": "app.js",
   "directories": {