browser-workspace 0.4.0

package/bin/browser-workspace

@@ -0,0 +1,75 @@
+#!/usr/bin/env python3
+from __future__ import annotations
+import argparse, json, os, sys
+from pathlib import Path
+ROOT=Path(__file__).resolve().parents[1]
+VENV_PY=ROOT/'.venv'/'bin'/'python'
+if VENV_PY.exists() and Path(sys.executable).resolve()!=VENV_PY.resolve() and os.environ.get('BROWSER_WORKSPACE_VENV_REEXEC')!='1':
+    env=dict(os.environ); env['BROWSER_WORKSPACE_VENV_REEXEC']='1'
+    os.execve(str(VENV_PY),[str(VENV_PY),str(Path(__file__).resolve()),*sys.argv[1:]],env)
+sys.path.insert(0,str(ROOT))
+from session_client import SOCKET_PATH, ensure_daemon
+from platform_runner import run_platform_action
+
+def emit(x): print(json.dumps(x,ensure_ascii=False))
+
+def status():
+    daemon=ensure_daemon()
+    helper=(ROOT/'node'/'admin.mjs').resolve()
+    return {
+        'root': str(ROOT.resolve()),
+        'cli': str(Path(__file__).resolve()),
+        'admin_helper': str(helper),
+        'managed_chrome': daemon.get('managed_chrome') if daemon else None,
+        'daemon': {
+            'running': bool(daemon and daemon.get('ok')),
+            'pid': daemon.get('pid') if daemon else None,
+            'session_count': daemon.get('session_count') if daemon else None,
+            'socket': str(SOCKET_PATH),
+        },
+    }
+
+def main():
+    p=argparse.ArgumentParser(prog='browser-workspace')
+    sp=p.add_subparsers(dest='group',required=True)
+    sp.add_parser('status')
+    session=sp.add_parser('session'); ssp=session.add_subparsers(dest='cmd',required=True)
+    start=ssp.add_parser('start'); start.add_argument('--workspace', help=argparse.SUPPRESS); start.add_argument('--url')
+    ssp.add_parser('list')
+    ex=ssp.add_parser('exec'); ex.add_argument('session_id'); ex.add_argument('--code-file')
+    stop=ssp.add_parser('stop'); stop.add_argument('session_id', nargs='?'); stop.add_argument('--all', action='store_true', dest='stop_all')
+    platform=sp.add_parser('platform'); psp=platform.add_subparsers(dest='pcmd',required=True)
+    profile=psp.add_parser('profile'); profile.add_argument('platform')
+    source=psp.add_parser('source'); source.add_argument('platform'); source.add_argument('action'); source.add_argument('--config')
+    run=psp.add_parser('run'); run.add_argument('platform'); run.add_argument('action'); run.add_argument('--config'); run.add_argument('--url'); run.add_argument('--session-id')
+    args=p.parse_args()
+    if args.group=='status': emit(status()); return
+    if args.group=='session' and args.cmd=='list':
+        from session_client import request
+        emit(request({'op':'list'})); return
+    if args.group=='session' and args.cmd=='start':
+        from session_client import request
+        emit(request({'op':'start','workspace':args.workspace,'url':args.url})); return
+    if args.group=='session' and args.cmd=='exec':
+        from session_client import request
+        if args.code_file: code=Path(args.code_file).read_text()
+        elif not sys.stdin.isatty(): code=sys.stdin.read()
+        else: p.error('session exec reads Python from stdin (use a heredoc) or --code-file')
+        r=request({'op':'exec','session_id':args.session_id,'code':code}); emit(r); raise SystemExit(0 if r.get('ok',False) else 1)
+    if args.group=='session' and args.cmd=='stop':
+        from session_client import request
+        if args.stop_all:
+            if args.session_id: p.error('session stop accepts either SESSION_ID or --all, not both')
+            emit(request({'op':'stop-all'})); return
+        if not args.session_id: p.error('session stop requires SESSION_ID or --all')
+        emit(request({'op':'stop','session_id':args.session_id})); return
+    if args.group=='platform' and args.pcmd=='source':
+        from platform_runner import action_path, prepare_action
+        path=action_path(args.platform,args.action); config=Path(args.config) if args.config else None
+        sys.stdout.write(prepare_action(path,config)); return
+    if args.group=='platform' and args.pcmd=='profile':
+        from platform_runner import platform_profile
+        emit(platform_profile(args.platform)); return
+    if args.group=='platform' and args.pcmd=='run':
+        r=run_platform_action(args.platform,args.action,config_path=args.config,url=args.url,session_id=args.session_id); emit(r); raise SystemExit(0 if (r.get('result') or {}).get('ok',False) else 1)
+if __name__=='__main__': main()

package/extension/manifest.json

@@ -0,0 +1,26 @@
+{
+  "manifest_version": 3,
+  "name": "Browser Workspace",
+  "version": "0.3.1",
+  "description": "Own named Chrome tab-group workspaces and lazy reusable tab pools for browser agents.",
+  "key": "MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAv7NU4eqCIkv793eZGTna5ntNKx42cj7wFWD0b+tEhGWHgXuYMDKXHuvtf1249vozaGGpDPFtH3d5uk0QNCl5yFA+kdL5RgTyg5qIpdFT4NET5Y8HSALxSjy27U4+Lf7XwYkQdSuClK7S9KxUI7OE9bloiDaSWbs9XjEBoeUGALXgqsIMgBVbyQ2RuMUo9mGpTreA+Tp2ATivU5ZnwmOuhJzniefbyAST/U6LY4VK/7jijnHLbfmTmCyi3xHWD2b8l9A7IsDBR1eRk6XY0YqZlC/H2wSmcUUQjhsgpiQK32iwwyNFspKgH4ReRPoO6RSuatUS8+J+/xevwCBF5eBq1QIDAQAB",
+  "permissions": [
+    "storage",
+    "tabs",
+    "tabGroups",
+    "alarms",
+    "webNavigation",
+    "contextMenus"
+  ],
+  "host_permissions": [
+    "http://*/*",
+    "https://*/*"
+  ],
+  "background": {
+    "service_worker": "service-worker.mjs",
+    "type": "module"
+  },
+  "action": {
+    "default_title": "Browser Workspace"
+  }
+}

package/extension/service-worker.mjs

@@ -0,0 +1,126 @@
+import { WorkspaceManager } from "./workspace-manager.mjs";
+
+const manager = new WorkspaceManager(chrome);
+const RECLAIM_ALARM = "browser-workspace.reclaim-stale";
+const closingWindowIds = new Set();
+
+chrome.alarms.create(RECLAIM_ALARM, { periodInMinutes: 5 });
+
+manager.restoreConfiguredWorkspaces().catch((error) => {
+  console.warn("Browser Workspace failed to restore configured workspaces", error);
+});
+
+const RELEASE_ALL_MENU_ID = "browser-workspace.release-all";
+
+async function ensureContextMenus() {
+  try {
+    await chrome.contextMenus.remove(RELEASE_ALL_MENU_ID);
+  } catch {
+    // Menu does not exist yet.
+  }
+  chrome.contextMenus.create({
+    id: RELEASE_ALL_MENU_ID,
+    title: "Release all tabs",
+    contexts: ["action"],
+  });
+}
+
+chrome.runtime.onInstalled.addListener(() => {
+  ensureContextMenus().catch((error) => {
+    console.warn("Browser Workspace failed to create extension menu", error);
+  });
+});
+
+chrome.runtime.onStartup.addListener(() => {
+  ensureContextMenus().catch((error) => {
+    console.warn("Browser Workspace failed to create extension menu", error);
+  });
+});
+
+ensureContextMenus().catch((error) => {
+  console.warn("Browser Workspace failed to initialize extension menu", error);
+});
+
+chrome.contextMenus.onClicked.addListener((info) => {
+  if (info.menuItemId !== RELEASE_ALL_MENU_ID) return;
+  manager.releaseAll().catch((error) => {
+    console.warn("Browser Workspace failed to release all tabs", error);
+  });
+});
+
+globalThis.browserWorkspaceManagerRpc = async (request) => {
+  try {
+    return { ok: true, result: await manager.rpc(request) };
+  } catch (error) {
+    return {
+      ok: false,
+      error: {
+        message: error?.message || String(error),
+        name: error?.name || "Error",
+      },
+    };
+  }
+};
+
+chrome.runtime.onMessage.addListener((request, _sender, sendResponse) => {
+  globalThis.browserWorkspaceManagerRpc(request).then(sendResponse);
+  return true;
+});
+
+chrome.tabs.onCreated.addListener((tab) => {
+  manager.inheritWorkspaceForCreatedTab(tab).catch((error) => {
+    console.warn("Browser Workspace failed to inherit child tab group", error);
+  });
+
+  setTimeout(() => {
+    manager.reconcileCreatedTab(tab.id).catch((error) => {
+      if (!/No tab with id|Invalid tab ID/i.test(String(error?.message || error))) {
+        console.warn("Browser Workspace failed delayed created-tab reconciliation", error);
+      }
+    });
+  }, 150);
+});
+
+chrome.webNavigation.onCreatedNavigationTarget.addListener((details) => {
+  manager.inheritNavigationTarget(details).catch((error) => {
+    console.warn("Browser Workspace failed to inherit navigation target", error);
+  });
+});
+
+chrome.tabs.onUpdated.addListener((_tabId, changeInfo, tab) => {
+  if (!Object.prototype.hasOwnProperty.call(changeInfo, "groupId")) return;
+  manager.inheritWorkspaceForCreatedTab(tab).catch((error) => {
+    console.warn("Browser Workspace failed to reconcile tab group ownership", error);
+  });
+});
+
+chrome.tabs.onActivated.addListener((activeInfo) => {
+  manager.noteActivatedTab(activeInfo).catch((error) => {
+    console.warn("Browser Workspace failed to record active tab context", error);
+  });
+});
+
+chrome.tabs.onRemoved.addListener((_tabId, removeInfo) => {
+  if (!removeInfo?.isWindowClosing) return;
+  closingWindowIds.add(removeInfo.windowId);
+  setTimeout(() => closingWindowIds.delete(removeInfo.windowId), 1000);
+});
+
+chrome.tabGroups.onRemoved.addListener((group) => {
+  // Chrome also emits onRemoved while moving a group between windows, and
+  // while a whole window is closing. Defer one turn so a moved group can
+  // reappear, and preserve persisted workspaces across normal window closes.
+  setTimeout(() => {
+    if (closingWindowIds.has(group.windowId)) return;
+    manager.removeDeletedGroup(group).catch((error) => {
+      console.warn("Browser Workspace failed to remove deleted workspace", error);
+    });
+  }, 100);
+});
+
+chrome.alarms.onAlarm.addListener((alarm) => {
+  if (alarm.name !== RECLAIM_ALARM) return;
+  manager.reclaimStaleTabs().catch((error) => {
+    console.warn("Browser Workspace failed to reclaim stale tabs", error);
+  });
+});