feat: add setSessionPartitionResolver

Author: samuelmaddock

packages/electron-chrome-extensions/src/browser/index.ts

@@ -20,6 +20,7 @@ import { ExtensionRouter } from './router'
 import { checkLicense, License } from './license'
 import { readLoadedExtensionManifest } from './manifest'
 import { PermissionsAPI } from './api/permissions'
+import { resolvePartition } from './partition'
 
 function checkVersion() {
   const electronVersion = process.versions.electron
@@ -105,8 +106,7 @@ export class ElectronChromeExtensions extends EventEmitter {
       }
 
       const partition = url?.searchParams.get('partition') || '_self'
-      const remoteSession =
-        partition === '_self' ? session : electronSession.fromPartition(partition)
+      const remoteSession = partition === '_self' ? session : resolvePartition(partition)
       const extensions = ElectronChromeExtensions.fromSession(remoteSession)
       if (!extensions) {
         return new Response(`ElectronChromeExtensions not found for "${partition}"`, {

packages/electron-chrome-extensions/src/browser/partition.ts

@@ -0,0 +1,19 @@
+import { session } from 'electron'
+
+type SessionPartitionResolver = (partition: string) => Electron.Session
+
+let resolvePartitionImpl: SessionPartitionResolver = (partition) => session.fromPartition(partition)
+
+/**
+ * Overrides the default `session.fromPartition()` behavior for retrieving Electron Sessions.
+ * This allows using custom identifiers (e.g., profile IDs) to find sessions, enabling features like
+ * `<browser-actions>` to work with non-standard session management schemes.
+ * @param handler A function that receives a string identifier and returns the corresponding Electron `Session`.
+ */
+export function setSessionPartitionResolver(resolver: SessionPartitionResolver) {
+  resolvePartitionImpl = resolver
+}
+
+export function resolvePartition(partition: string) {
+  return resolvePartitionImpl(partition)
+}

packages/electron-chrome-extensions/src/browser/router.ts

@@ -1,6 +1,8 @@
-import { app, ipcMain, session, Session, WebContents } from 'electron'
+import { app, ipcMain, Session } from 'electron'
 import debug from 'debug'
 
+import { resolvePartition } from './partition'
+
 // Shorten base64 encoded icons
 const shortenValues = (k: string, v: any) =>
   typeof v === 'string' && v.length > 128 ? v.substr(0, 128) + '...' : v
@@ -117,7 +119,7 @@ class RoutingDelegate {
     const ses =
       sessionPartition === DEFAULT_SESSION
         ? getSessionFromEvent(event)
-        : session.fromPartition(sessionPartition)
+        : resolvePartition(sessionPartition)
 
     const observer = this.sessionMap.get(ses)
 

packages/electron-chrome-extensions/src/index.ts

@@ -1 +1,2 @@
 export * from './browser'
+export { setSessionPartitionResolver } from './browser/partition'