browser-workspace 0.4.2 → 0.4.9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (73) hide show
  1. package/README.md +15 -1
  2. package/SKILL.md +22 -21
  3. package/bin/__pycache__/browser-workspacecpython-311.pyc +0 -0
  4. package/package.json +1 -1
  5. package/platform_runner.py +11 -2
  6. package/platforms/chatgpt/actions/_lifecycle_common.py +73 -20
  7. package/platforms/chatgpt/actions/_new_turn.py +1 -0
  8. package/platforms/chatgpt/actions/_readiness.py +29 -3
  9. package/platforms/chatgpt/actions/_submit_thread.py +20 -3
  10. package/platforms/chatgpt/actions/_temporary_submit.py +22 -2
  11. package/platforms/chatgpt/actions/_thread_turn.py +31 -2
  12. package/platforms/chatgpt/tests/test_contract.py +114 -1
  13. package/platforms/iyf/AIRPLAY.md +25 -0
  14. package/platforms/iyf/actions/_find.py +18 -0
  15. package/platforms/iyf/actions/_open.py +10 -0
  16. package/platforms/iyf/manifest.yaml +11 -0
  17. package/platforms/iyf/safari-airplay +57 -0
  18. package/pyproject.toml +6 -2
  19. package/session_client.py +10 -0
  20. package/session_daemon.py +91 -3
  21. package/src/browser_workspace.egg-info/PKG-INFO +2 -1
  22. package/src/browser_workspace.egg-info/SOURCES.txt +91 -15
  23. package/src/browser_workspace.egg-info/entry_points.txt +2 -0
  24. package/src/browser_workspace.egg-info/requires.txt +1 -0
  25. package/src/browser_workspace.egg-info/top_level.txt +1 -1
  26. package/src/browser_workspace_app/__init__.py +1 -0
  27. package/src/browser_workspace_app/agent-workspace/agent_helpers.py +718 -0
  28. package/src/browser_workspace_app/cli.py +236 -0
  29. package/src/browser_workspace_app/extension/manifest.json +26 -0
  30. package/src/browser_workspace_app/extension/service-worker.mjs +126 -0
  31. package/src/browser_workspace_app/extension/workspace-manager.mjs +680 -0
  32. package/src/browser_workspace_app/extension/workspace-page.mjs +7 -0
  33. package/src/browser_workspace_app/extension/workspace.html +8 -0
  34. package/src/browser_workspace_app/interaction-skills/connection.md +52 -0
  35. package/src/browser_workspace_app/interaction-skills/cookies.md +3 -0
  36. package/src/browser_workspace_app/interaction-skills/cross-origin-iframes.md +3 -0
  37. package/src/browser_workspace_app/interaction-skills/dialogs.md +64 -0
  38. package/src/browser_workspace_app/interaction-skills/downloads.md +3 -0
  39. package/src/browser_workspace_app/interaction-skills/drag-and-drop.md +3 -0
  40. package/src/browser_workspace_app/interaction-skills/dropdowns.md +3 -0
  41. package/src/browser_workspace_app/interaction-skills/iframes.md +3 -0
  42. package/src/browser_workspace_app/interaction-skills/network-requests.md +3 -0
  43. package/src/browser_workspace_app/interaction-skills/print-as-pdf.md +3 -0
  44. package/src/browser_workspace_app/interaction-skills/profile-sync.md +90 -0
  45. package/src/browser_workspace_app/interaction-skills/screenshots.md +17 -0
  46. package/src/browser_workspace_app/interaction-skills/scrolling.md +22 -0
  47. package/src/browser_workspace_app/interaction-skills/shadow-dom.md +3 -0
  48. package/src/browser_workspace_app/interaction-skills/tabs.md +73 -0
  49. package/src/browser_workspace_app/interaction-skills/uploads.md +1 -0
  50. package/src/browser_workspace_app/interaction-skills/viewport.md +16 -0
  51. package/src/browser_workspace_app/licenses/browser-harness-MIT.txt +21 -0
  52. package/src/browser_workspace_app/managed_chrome.py +244 -0
  53. package/src/browser_workspace_app/node/admin.mjs +44 -0
  54. package/src/browser_workspace_app/package.json +36 -0
  55. package/src/browser_workspace_app/pyproject.toml +33 -0
  56. package/src/browser_workspace_app/session_client.py +117 -0
  57. package/src/browser_workspace_app/session_daemon.py +431 -0
  58. package/src/browser_harness/__init__.py +0 -2
  59. package/src/browser_harness/__pycache__/__init__.cpython-311.pyc +0 -0
  60. package/src/browser_harness/_ipc.py +0 -201
  61. package/src/browser_harness/admin.py +0 -1588
  62. package/src/browser_harness/auth.py +0 -546
  63. package/src/browser_harness/daemon.py +0 -894
  64. package/src/browser_harness/helpers.py +0 -668
  65. package/src/browser_harness/macos.py +0 -131
  66. package/src/browser_harness/mcp_cli.py +0 -16
  67. package/src/browser_harness/paths.py +0 -49
  68. package/src/browser_harness/recorder.py +0 -343
  69. package/src/browser_harness/run.py +0 -413
  70. package/src/browser_harness/telemetry.py +0 -308
  71. package/src/browser_harness/video-template.html +0 -970
  72. package/src/browser_harness/video.py +0 -749
  73. package/src/browser_harness/video_render.py +0 -524
@@ -0,0 +1,44 @@
1
+ import net from "node:net";
2
+ import os from "node:os";
3
+ import path from "node:path";
4
+
5
+ function socketPath() {
6
+ if (process.env.BROWSER_WORKSPACE_TEST_MODE === "1" && process.env.BROWSER_WORKSPACE_SESSION_SOCKET) return process.env.BROWSER_WORKSPACE_SESSION_SOCKET;
7
+ return path.join(os.homedir(), ".config", "browser-workspace", "runtime", "session.sock");
8
+ }
9
+
10
+ function request(payload) {
11
+ return new Promise((resolve, reject) => {
12
+ const socket = net.createConnection({ path: socketPath() });
13
+ let buffer = "";
14
+ let settled = false;
15
+ const fail = (error) => { if (!settled) { settled = true; reject(error); } };
16
+ socket.setEncoding("utf8");
17
+ socket.on("connect", () => socket.write(`${JSON.stringify(payload)}\n`));
18
+ socket.on("data", chunk => {
19
+ buffer += chunk;
20
+ const newline = buffer.indexOf("\n");
21
+ if (newline < 0 || settled) return;
22
+ settled = true;
23
+ socket.end();
24
+ try {
25
+ const result = JSON.parse(buffer.slice(0, newline));
26
+ if (result && result.error) reject(new Error(result.error));
27
+ else resolve(result);
28
+ } catch (error) { reject(error); }
29
+ });
30
+ socket.on("error", fail);
31
+ socket.on("end", () => { if (!settled) fail(new Error("browser-workspace daemon closed without a response")); });
32
+ });
33
+ }
34
+
35
+ export async function ensureWorkspace(name, size = 5) {
36
+ if (typeof name !== "string" || !name.trim()) throw new TypeError("workspace name must be a non-empty string");
37
+ if (!Number.isInteger(size) || size < 1) throw new TypeError("workspace size must be a positive integer");
38
+ return request({ op: "workspace-create", name: name.trim(), pool_size: size });
39
+ }
40
+
41
+ export async function deleteWorkspace(name, { force = false } = {}) {
42
+ if (typeof name !== "string" || !name.trim()) throw new TypeError("workspace name must be a non-empty string");
43
+ return request({ op: "workspace-delete", name: name.trim(), force: Boolean(force) });
44
+ }
@@ -0,0 +1,36 @@
1
+ {
2
+ "name": "browser-workspace",
3
+ "version": "0.4.8",
4
+ "description": "Agent browser workspace with persistent sessions, tab leasing, snapshots, and platform workflows.",
5
+ "bin": {
6
+ "browser-workspace": "bin/browser-workspace"
7
+ },
8
+ "files": [
9
+ "bin/",
10
+ "node/",
11
+ "src/",
12
+ "agent-workspace/agent_helpers.py",
13
+ "interaction-skills/",
14
+ "platforms/",
15
+ "licenses/",
16
+ "platform_runner.py",
17
+ "session_client.py",
18
+ "session_daemon.py",
19
+ "pyproject.toml",
20
+ "README.md",
21
+ "SKILL.md",
22
+ "PRIVACY.md",
23
+ "managed_chrome.py",
24
+ "extension/"
25
+ ],
26
+ "engines": {
27
+ "node": ">=20"
28
+ },
29
+ "scripts": {
30
+ "test": "python3 -m pytest -q test && node --test test/workspace-manager.test.mjs"
31
+ },
32
+ "repository": {
33
+ "type": "git",
34
+ "url": "git+https://github.com/lalalic/browser-workspace.git"
35
+ }
36
+ }
@@ -0,0 +1,33 @@
1
+ [build-system]
2
+ requires = ["setuptools==84.0.0"]
3
+ build-backend = "setuptools.build_meta"
4
+
5
+ [project]
6
+ name = "browser-workspace"
7
+ version = "0.4.8"
8
+ description = "Agent browser workspace with persistent Python sessions, tab leasing, snapshots, and platform workflows."
9
+ requires-python = ">=3.11"
10
+ dependencies = [
11
+ "browser-harness==0.1.13",
12
+ "cdp-use==1.4.5",
13
+ "fetch-use==0.4.0",
14
+ "pillow==12.3.0",
15
+ "PyYAML==6.0.2",
16
+ "websockets==15.0.1",
17
+ ]
18
+
19
+ [project.scripts]
20
+ browser-workspace = "browser_workspace_app.cli:main"
21
+
22
+ [tool.setuptools]
23
+ package-dir = {"" = "src"}
24
+
25
+ [tool.setuptools.packages.find]
26
+ where = ["src"]
27
+
28
+ [tool.setuptools.package-data]
29
+ browser_workspace_app = ["**/*"]
30
+
31
+ [tool.pytest.ini_options]
32
+ pythonpath = ["src"]
33
+ addopts = "--import-mode=importlib"
@@ -0,0 +1,117 @@
1
+ #!/usr/bin/env python3
2
+ from __future__ import annotations
3
+ import fcntl, json, os, shutil, socket, subprocess, sys, time
4
+ from pathlib import Path
5
+
6
+ ROOT=Path(__file__).resolve().parent
7
+ RUNTIME_DIR=Path.home()/'.config/browser-workspace/runtime'
8
+ SOCKET_PATH=(Path(os.environ['BROWSER_WORKSPACE_SESSION_SOCKET']) if os.environ.get('BROWSER_WORKSPACE_TEST_MODE')=='1' and os.environ.get('BROWSER_WORKSPACE_SESSION_SOCKET') else RUNTIME_DIR/'session.sock')
9
+ HARNESS_RUNTIME_SPEC='browser-harness @ git+https://github.com/lalalic/browser-harness.git@80b74e116b9126f5159c84e5e05a750573e5f192'
10
+
11
+ def package_version():
12
+ try:
13
+ return json.loads((ROOT/'package.json').read_text()).get('version','dev')
14
+ except Exception:
15
+ return 'dev'
16
+
17
+ def runtime_source():
18
+ configured=os.environ.get('BROWSER_WORKSPACE_RUNTIME_SOURCE')
19
+ if configured: return Path(configured)
20
+ if not (ROOT/'package.json').exists(): return ROOT
21
+ target=RUNTIME_DIR/'packages'/package_version()
22
+ marker=target/'.ready'
23
+ if marker.exists(): return target
24
+ target.parent.mkdir(parents=True,exist_ok=True)
25
+ tmp=target.with_name(target.name+'.tmp')
26
+ shutil.rmtree(tmp,ignore_errors=True)
27
+ tmp.mkdir(parents=True)
28
+ for name in ['pyproject.toml','session_daemon.py','session_client.py','platform_runner.py','managed_chrome.py','src','agent-workspace','interaction-skills','platforms','extension','licenses']:
29
+ source=ROOT/name
30
+ if not source.exists(): continue
31
+ destination=tmp/name
32
+ if source.is_dir(): shutil.copytree(source,destination)
33
+ else: shutil.copy2(source,destination)
34
+ # A published wheel installs directly under browser_workspace_app/ without
35
+ # a nested src/ tree. Reconstruct the build layout only in the runtime
36
+ # cache, never in the repository or published source tree.
37
+ if not (tmp/'src').exists():
38
+ package_root = tmp/'src'/'browser_workspace_app'
39
+ package_root.parent.mkdir(parents=True, exist_ok=True)
40
+ shutil.copytree(ROOT, package_root, ignore=shutil.ignore_patterns('__pycache__', '*.pyc', '.DS_Store'))
41
+ (tmp/'.ready').write_text(package_version())
42
+ shutil.rmtree(target,ignore_errors=True)
43
+ tmp.rename(target)
44
+ return target
45
+
46
+ SOURCE=runtime_source()
47
+ PYTHON=Path(os.environ.get('BROWSER_WORKSPACE_SESSION_PYTHON',RUNTIME_DIR/'venv'/package_version()/'bin/python'))
48
+
49
+
50
+
51
+ def ensure_runtime_python():
52
+ if PYTHON.exists():
53
+ return PYTHON
54
+ RUNTIME_DIR.mkdir(parents=True, exist_ok=True)
55
+ lock_path=RUNTIME_DIR/'runtime-bootstrap.lock'
56
+ with open(lock_path,'a+b') as lock:
57
+ fcntl.flock(lock,fcntl.LOCK_EX)
58
+ if PYTHON.exists():
59
+ return PYTHON
60
+ uv=shutil.which('uv')
61
+ if uv:
62
+ subprocess.run([uv,'venv','--python','3.11',str(PYTHON.parent.parent)],check=True)
63
+ subprocess.run([uv,'pip','install','--python',str(PYTHON),'-e',str(SOURCE)],check=True)
64
+ subprocess.run([uv,'pip','install','--python',str(PYTHON),'--reinstall',HARNESS_RUNTIME_SPEC],check=True)
65
+ return PYTHON
66
+ if sys.version_info < (3,11):
67
+ raise RuntimeError(
68
+ 'Browser Workspace needs Python 3.11+ for first-use bootstrap. '
69
+ 'Install uv or Python 3.11+, then retry.'
70
+ )
71
+ subprocess.run([sys.executable,'-m','venv',str(PYTHON.parent.parent)],check=True)
72
+ subprocess.run([str(PYTHON),'-m','pip','install','-e',str(SOURCE)],check=True)
73
+ subprocess.run([str(PYTHON),'-m','pip','install','--force-reinstall',HARNESS_RUNTIME_SPEC],check=True)
74
+ return PYTHON
75
+
76
+ def request(payload, ensure=True):
77
+ if ensure: ensure_daemon()
78
+ s=socket.socket(socket.AF_UNIX,socket.SOCK_STREAM)
79
+ try: s.connect(str(SOCKET_PATH))
80
+ except Exception:
81
+ s.close(); raise
82
+ with s:
83
+ f=s.makefile('rwb'); f.write((json.dumps(payload,ensure_ascii=False)+'\n').encode()); f.flush(); line=f.readline()
84
+ if not line: raise RuntimeError('browser-workspace session daemon closed without a response')
85
+ return json.loads(line)
86
+
87
+ def ping():
88
+ try: return request({'op':'ping'},ensure=False)
89
+ except Exception: return None
90
+
91
+ def owner_is_running():
92
+ """An exclusive owner lock means a live daemon already exists."""
93
+ owner_path=RUNTIME_DIR/'session.lock'
94
+ with open(owner_path,'a+b') as owner:
95
+ try:
96
+ fcntl.flock(owner,fcntl.LOCK_EX|fcntl.LOCK_NB)
97
+ except BlockingIOError:
98
+ return True
99
+ finally:
100
+ try: fcntl.flock(owner,fcntl.LOCK_UN)
101
+ except OSError: pass
102
+ return False
103
+
104
+ def ensure_daemon():
105
+ p=ping()
106
+ if p and p.get('ok'): return p
107
+ RUNTIME_DIR.mkdir(parents=True,exist_ok=True)
108
+ if owner_is_running():
109
+ raise RuntimeError('browser-workspace daemon is alive but its socket is unreachable; refusing duplicate startup')
110
+ runtime_python=ensure_runtime_python()
111
+ log=open(RUNTIME_DIR/'session-daemon.log','ab',buffering=0)
112
+ subprocess.Popen([str(runtime_python),str(SOURCE/'session_daemon.py')],stdin=subprocess.DEVNULL,stdout=log,stderr=log,start_new_session=True,close_fds=True)
113
+ deadline=time.time()+15
114
+ while time.time()<deadline:
115
+ time.sleep(.1); p=ping()
116
+ if p and p.get('ok'): return p
117
+ raise RuntimeError(f'browser-workspace session daemon did not start; see {RUNTIME_DIR / "session-daemon.log"}')
@@ -0,0 +1,431 @@
1
+ #!/usr/bin/env python3
2
+ import ast, atexit, contextlib, io, json, os, secrets, signal, socket, sys, threading, time, traceback
3
+ from pathlib import Path
4
+
5
+ ROOT = Path(__file__).resolve().parent
6
+ sys.path.insert(0, str(ROOT/'src'))
7
+ os.environ.setdefault('BH_HOME', str(Path.home()/'.config/browser-workspace/browser-core'))
8
+ # Preserve the canonical single Browser Harness daemon socket during migration.
9
+ # This is runtime-state compatibility only; Browser Workspace owns the code/venv.
10
+ os.environ.setdefault('BH_RUNTIME_DIR', str(Path.home()/'.config/browser-workspace/browser-core-runtime'))
11
+ os.environ.setdefault('BU_NAME', 'browser-workspace')
12
+ os.environ.setdefault('BH_AGENT_WORKSPACE', str(ROOT/'agent-workspace'))
13
+ os.environ.setdefault('BH_WORKSPACE_NAME', 'Harness')
14
+
15
+ from managed_chrome import ensure_managed_chrome
16
+ MANAGED_CHROME = ensure_managed_chrome()
17
+ from browser_harness.admin import ensure_daemon, _process_start_time
18
+ from browser_harness import _ipc as harness_ipc
19
+ from browser_harness import helpers as bh
20
+
21
+ HARNESS_OWNER_TOKEN = secrets.token_hex(24)
22
+ HARNESS_OWNER_PID = os.getpid()
23
+ HARNESS_OWNER_STARTED = _process_start_time(HARNESS_OWNER_PID)
24
+ HARNESS_OWNER_PIPE = None
25
+
26
+
27
+ def stop_owned_harness():
28
+ """Only a harness matching this daemon generation may stop."""
29
+ try:
30
+ conn, token = harness_ipc.connect(os.environ.get('BU_NAME', 'browser-workspace'), timeout=1)
31
+ with conn:
32
+ response = harness_ipc.request(conn, token, {'meta': 'owner_shutdown', 'owner_token': HARNESS_OWNER_TOKEN})
33
+ if isinstance(response, dict) and response.get('ok'):
34
+ # Allow the harness its graceful CDP/IPC cleanup before EOF of
35
+ # our owner pipe forces the crash-only exit path.
36
+ deadline = time.monotonic() + 2
37
+ while time.monotonic() < deadline and harness_ipc.ping(os.environ.get('BU_NAME', 'browser-workspace'), timeout=0.1):
38
+ time.sleep(0.05)
39
+ except (OSError, ValueError, TimeoutError):
40
+ pass
41
+
42
+ DEFAULT_WORKSPACE = os.environ.get('BH_WORKSPACE_NAME','Harness')
43
+ RUNTIME_DIR = Path.home()/'.config/browser-workspace/runtime'
44
+ SOCKET_PATH = (Path(os.environ['BROWSER_WORKSPACE_SESSION_SOCKET']) if os.environ.get('BROWSER_WORKSPACE_TEST_MODE')=='1' and os.environ.get('BROWSER_WORKSPACE_SESSION_SOCKET') else RUNTIME_DIR/'session.sock')
45
+ OWNER_LOCK_PATH = RUNTIME_DIR/'session.lock'
46
+ SESSIONS = {}
47
+ SESSIONS_LOCK = threading.Lock()
48
+ BROWSER_LOCK = threading.RLock()
49
+
50
+ SESSION_RECONCILE_SECONDS = 30
51
+
52
+
53
+ def is_reclaimed(info):
54
+ url = info.get('url', '')
55
+ if not url.startswith('chrome-extension://'):
56
+ return False
57
+ try:
58
+ from urllib.parse import urlparse, parse_qs
59
+ parsed = urlparse(url)
60
+ return parsed.path == '/workspace.html' and parse_qs(parsed.query).get('role') == ['idle']
61
+ except ValueError:
62
+ return False
63
+
64
+
65
+ def reconcile_sessions():
66
+ """Drop records for completely released leases, never release those tabs again."""
67
+ with BROWSER_LOCK:
68
+ try:
69
+ targets = bh.cdp('Target.getTargets')['targetInfos']
70
+ except Exception:
71
+ return []
72
+ pages = {i['targetId']: i for i in targets if i.get('type') == 'page'}
73
+ removed = []
74
+ with SESSIONS_LOCK:
75
+ for sid, session in list(SESSIONS.items()):
76
+ if session.lock.locked():
77
+ continue
78
+ # A leased root may have opened child tabs: keep any live child.
79
+ if all(tid not in pages or is_reclaimed(pages[tid])
80
+ for tid in session.owned_target_ids):
81
+ del SESSIONS[sid]
82
+ removed.append(sid)
83
+ return removed
84
+
85
+
86
+ def reconciliation_loop():
87
+ while True:
88
+ time.sleep(SESSION_RECONCILE_SECONDS)
89
+ reconcile_sessions()
90
+
91
+
92
+ def compilable_module_body(body):
93
+ """Return a module body that remains valid after trailing-result extraction.
94
+
95
+ Submitted sessions commonly contain ordinary imports before a future import.
96
+ Python accepts that source form, but recompiling an AST after removing the
97
+ trailing result expression requires future imports to be moved first.
98
+ """
99
+ leading = []
100
+ if body and isinstance(body[0], ast.Expr) and isinstance(body[0].value, ast.Constant) and isinstance(body[0].value.value, str):
101
+ leading.append(body.pop(0))
102
+ future = [node for node in body if isinstance(node, ast.ImportFrom) and node.module == '__future__']
103
+ if not future:
104
+ return leading + body
105
+ rest = [node for node in body if node not in future]
106
+ return leading + future + rest
107
+
108
+
109
+ class SessionBrowserProxy:
110
+ def __init__(self, session): self._session=session
111
+ def __getattr__(self, name): return getattr(bh, name)
112
+ def new_tab(self, url='about:blank'): return self._session._new_tab(url)
113
+ def close_tab(self, target=None): return self._session._close_tab(target)
114
+ def cdp(self, method, **kwargs): return self._session._cdp(method, **kwargs)
115
+
116
+ class Session:
117
+ def __init__(self, sid, workspace, target_id, workspace_supported):
118
+ self.session_id=sid; self.workspace=workspace; self.target_id=target_id; self.workspace_supported=bool(workspace_supported)
119
+ self.created_at=time.time(); self.last_used_at=self.created_at
120
+ self.owned_target_ids={target_id}
121
+ self.lock=threading.Lock()
122
+ self.browser=SessionBrowserProxy(self)
123
+ self.namespace={'__name__':'__browser_workspace_session__','bh':self.browser,'target_id':target_id}
124
+ for name in dir(bh):
125
+ if not name.startswith('_'): self.namespace[name]=getattr(bh,name)
126
+ self.namespace['new_tab']=self._new_tab
127
+ self.namespace['close_tab']=self._close_tab
128
+ self.namespace['cdp']=self._cdp
129
+
130
+ def _remember_target(self, tid):
131
+ if tid:
132
+ self.owned_target_ids.add(tid)
133
+ self.target_id=tid
134
+ self.namespace['target_id']=tid
135
+ return tid
136
+
137
+ def _new_tab(self, url='about:blank'):
138
+ return self._remember_target(bh.new_tab(url))
139
+
140
+ def _close_tab(self, target=None):
141
+ wanted=target
142
+ if isinstance(wanted, dict): wanted=wanted.get('targetId') or wanted.get('target_id')
143
+ if wanted is None: wanted=bh.current_tab().get('targetId')
144
+ result=bh.close_tab(target)
145
+ if wanted: self.owned_target_ids.discard(wanted)
146
+ return result
147
+
148
+ def _cdp(self, method, **kwargs):
149
+ if method == 'Target.createTarget':
150
+ raise RuntimeError(
151
+ 'Target.createTarget is disabled in Browser Workspace sessions; '
152
+ 'use new_tab(url) so the tab is owned by the active session/workspace'
153
+ )
154
+ result=bh.cdp(method, **kwargs)
155
+ if method == 'Target.closeTarget':
156
+ self.owned_target_ids.discard(kwargs.get('targetId'))
157
+ return result
158
+
159
+ def with_workspace(name, supported=None):
160
+ class Scope:
161
+ def __enter__(self):
162
+ self.name_token=bh.workspace_set_name(name)
163
+ self.support_token=bh.workspace_set_supported(supported) if supported is not None else None
164
+ return self
165
+ def __exit__(self,*_):
166
+ if self.support_token is not None: bh.workspace_reset_supported(self.support_token)
167
+ bh.workspace_reset_name(self.name_token)
168
+ return Scope()
169
+
170
+ def execute(source, namespace):
171
+ out,err=io.StringIO(),io.StringIO(); value=None
172
+ try:
173
+ tree=ast.parse(source,mode='exec'); body=list(tree.body)
174
+ tail=body.pop() if body and isinstance(body[-1],ast.Expr) else None
175
+ body=compilable_module_body(body)
176
+ with contextlib.redirect_stdout(out), contextlib.redirect_stderr(err):
177
+ if body: exec(compile(ast.Module(body=body,type_ignores=[]),'<browser-workspace-session>','exec'),namespace,namespace)
178
+ if tail is not None:
179
+ value=eval(compile(ast.Expression(tail.value),'<browser-workspace-session>','eval'),namespace,namespace); namespace['_']=value
180
+ result={'ok':True,'stdout':out.getvalue(),'stderr':err.getvalue()}
181
+ if value is not None:
182
+ try: json.dumps(value); result['value']=value
183
+ except (TypeError,ValueError): result['value_repr']=repr(value)
184
+ return result
185
+ except BaseException as exc:
186
+ traceback.print_exc(file=err)
187
+ return {'ok':False,'stdout':out.getvalue(),'stderr':err.getvalue(),'error':f'{type(exc).__name__}: {exc}'}
188
+
189
+ def start_session(workspace=None, url=None):
190
+ workspace=(workspace or DEFAULT_WORKSPACE).strip()
191
+ if not workspace: raise ValueError('workspace must not be empty')
192
+ sid=secrets.token_hex(8)
193
+ start_url=(url or 'about:blank').strip()
194
+ extension=bh.workspace_extension_status()
195
+ supported=bool(extension.get('supported'))
196
+ with BROWSER_LOCK, with_workspace(workspace, supported): tid=bh.new_tab(start_url)
197
+ s=Session(sid,workspace,tid,supported)
198
+ with SESSIONS_LOCK: SESSIONS[sid]=s
199
+ result={'session_id':sid,'workspace':workspace,'target_id':tid,'url':start_url,'workspace_supported':supported}
200
+ if not supported:
201
+ result['warning']='Browser Workspace extension is not installed; workspace grouping/leases are unavailable. Browser automation will continue without workspace isolation.'
202
+ result['extension_url']=extension.get('extensionUrl')
203
+ return result
204
+
205
+
206
+ def absorb_owned_children(session):
207
+ """Adopt popup/child page targets whose opener ancestry belongs to this session."""
208
+ try:
209
+ infos=bh.cdp('Target.getTargets').get('targetInfos', [])
210
+ except Exception:
211
+ return
212
+ pages={i.get('targetId'): i for i in infos if i.get('type') == 'page' and i.get('targetId')}
213
+ changed=True
214
+ while changed:
215
+ changed=False
216
+ for tid, info in pages.items():
217
+ if tid in session.owned_target_ids:
218
+ continue
219
+ opener=info.get('openerId')
220
+ if opener and opener in session.owned_target_ids:
221
+ session.owned_target_ids.add(tid)
222
+ changed=True
223
+
224
+ def close_owned_tabs(session):
225
+ """Close/release every live tab owned by the session, children before parents."""
226
+ absorb_owned_children(session)
227
+ errors=[]
228
+ # Close descendants/current tabs first. Repeatedly inspect live targets so already
229
+ # closed tabs disappear harmlessly and opener relationships remain usable.
230
+ while session.owned_target_ids:
231
+ try:
232
+ infos=bh.cdp('Target.getTargets').get('targetInfos', [])
233
+ except Exception:
234
+ infos=[]
235
+ live={i.get('targetId'): i for i in infos if i.get('type') == 'page' and i.get('targetId')}
236
+ owned_live=[tid for tid in session.owned_target_ids if tid in live]
237
+ if not owned_live:
238
+ session.owned_target_ids.clear()
239
+ break
240
+ parent_ids={live[tid].get('openerId') for tid in owned_live}
241
+ leaves=[tid for tid in owned_live if tid not in parent_ids]
242
+ targets=leaves or owned_live
243
+ for tid in targets:
244
+ try:
245
+ if session.workspace_supported:
246
+ try:
247
+ # Workspace-owned session tabs are leases. Release them back
248
+ # to the persistent pool instead of destroying the Chrome tab
249
+ # group when the final leased tab is stopped. Popup/child
250
+ # targets that are outside the workspace are still closed.
251
+ bh.close_tab(tid)
252
+ except RuntimeError as exc:
253
+ if 'outside workspace' not in str(exc):
254
+ raise
255
+ bh.cdp('Target.closeTarget', targetId=tid)
256
+ else:
257
+ bh.cdp('Target.closeTarget', targetId=tid)
258
+ except Exception as exc:
259
+ errors.append(f'{tid}: {type(exc).__name__}: {exc}')
260
+ finally:
261
+ session.owned_target_ids.discard(tid)
262
+ return errors
263
+
264
+ def get_session(sid):
265
+ with SESSIONS_LOCK: s=SESSIONS.get(sid)
266
+ if not s: raise KeyError(f'unknown session_id: {sid}')
267
+ return s
268
+
269
+ def exec_session(sid, code):
270
+ s=get_session(sid)
271
+ s.last_used_at=time.time()
272
+ with s.lock, BROWSER_LOCK, with_workspace(s.workspace, s.workspace_supported):
273
+ bh.switch_tab(s.target_id,activate=False)
274
+ r=execute(code,s.namespace)
275
+ absorb_owned_children(s)
276
+ try:
277
+ cur=bh.current_tab(); s.target_id=cur.get('targetId') or s.target_id; s.namespace['target_id']=s.target_id
278
+ except Exception: pass
279
+ return {'session_id':sid,'workspace':s.workspace,'target_id':s.target_id,**r}
280
+
281
+ def list_sessions():
282
+ with SESSIONS_LOCK:
283
+ sessions=list(SESSIONS.values())
284
+ return {
285
+ 'session_count': len(sessions),
286
+ 'sessions': [
287
+ {
288
+ 'session_id': s.session_id,
289
+ 'workspace': s.workspace,
290
+ 'target_id': s.target_id,
291
+ 'created_at': s.created_at,
292
+ 'last_used_at': s.last_used_at,
293
+ }
294
+ for s in sessions
295
+ ],
296
+ }
297
+
298
+ def stop_all_sessions():
299
+ with SESSIONS_LOCK:
300
+ ids=list(SESSIONS)
301
+ stopped=[]
302
+ errors=[]
303
+ for sid in ids:
304
+ try:
305
+ stopped.append(stop_session(sid))
306
+ except Exception as exc:
307
+ errors.append({'session_id':sid,'error':f'{type(exc).__name__}: {exc}'})
308
+ return {'stopped_count':len(stopped),'error_count':len(errors),'stopped':stopped,'errors':errors}
309
+
310
+ def stop_session(sid):
311
+ with SESSIONS_LOCK: s=SESSIONS.pop(sid,None)
312
+ if not s: raise KeyError(f'unknown session_id: {sid}')
313
+ release_error=None
314
+ with s.lock, BROWSER_LOCK, with_workspace(s.workspace, s.workspace_supported):
315
+ absorb_owned_children(s)
316
+ closed_count=len(s.owned_target_ids)
317
+ errors=close_owned_tabs(s)
318
+ if errors: release_error='; '.join(errors)
319
+ return {'session_id':sid,'workspace':s.workspace,'target_id':s.target_id,'closed_tabs':closed_count,'release_error':release_error}
320
+
321
+ SOCKET_IDENTITY = None
322
+
323
+ def remove_owned_socket():
324
+ """Never unlink a socket replaced by another owner."""
325
+ if SOCKET_IDENTITY is None:
326
+ return
327
+ try:
328
+ st = SOCKET_PATH.lstat()
329
+ if (st.st_dev, st.st_ino) == SOCKET_IDENTITY:
330
+ SOCKET_PATH.unlink()
331
+ except FileNotFoundError:
332
+ pass
333
+
334
+ def cleanup():
335
+ with SESSIONS_LOCK: ids=list(SESSIONS)
336
+ for sid in ids:
337
+ try: stop_session(sid)
338
+ except Exception: pass
339
+ stop_owned_harness()
340
+ # The write end is close-on-exec; its final close also detects SIGKILL
341
+ # and parent re-exec without relying on a recycled numeric PID.
342
+ if HARNESS_OWNER_PIPE is not None:
343
+ try: os.close(HARNESS_OWNER_PIPE)
344
+ except OSError: pass
345
+ try: remove_owned_socket()
346
+ except Exception: pass
347
+ atexit.register(cleanup)
348
+
349
+ def create_workspace(name, pool_size=5):
350
+ if not isinstance(name, str) or not name.strip():
351
+ raise ValueError('workspace name must be a non-empty string')
352
+ size=int(pool_size)
353
+ if size < 1:
354
+ raise ValueError('workspace pool size must be at least 1')
355
+ with BROWSER_LOCK:
356
+ result=bh.workspace_create(name.strip(), size)
357
+ return {
358
+ 'name': result.get('name', name.strip()),
359
+ 'poolSize': result.get('poolSize', size),
360
+ 'maxCapacity': result.get('maxCapacity', result.get('poolSize', size)),
361
+ 'initialized': result.get('initialized', True),
362
+ 'workspace_supported': True,
363
+ }
364
+
365
+ def delete_workspace(name, force=False):
366
+ if not isinstance(name, str) or not name.strip():
367
+ raise ValueError('workspace name must be a non-empty string')
368
+ with BROWSER_LOCK:
369
+ return bh.workspace_delete(name.strip(), bool(force))
370
+
371
+ def handle(req):
372
+ op=req.get('op')
373
+ if op=='ping': return {'ok':True,'pid':os.getpid(),'session_count':len(SESSIONS),'managed_chrome':MANAGED_CHROME}
374
+ if op=='list': return list_sessions()
375
+ if op=='start': return start_session(req.get('workspace'), req.get('url'))
376
+ if op=='exec': return exec_session(req['session_id'],req.get('code',''))
377
+ if op=='stop': return stop_session(req['session_id'])
378
+ if op=='stop-all': return stop_all_sessions()
379
+ if op=='workspace-create': return create_workspace(req['name'], req.get('pool_size', 5))
380
+ if op=='workspace-delete': return delete_workspace(req['name'], req.get('force', False))
381
+ raise ValueError(f'unknown op: {op}')
382
+
383
+ def serve_conn(conn):
384
+ with conn:
385
+ f=conn.makefile('rwb')
386
+ line=f.readline()
387
+ if not line: return
388
+ try: result=handle(json.loads(line))
389
+ except Exception as exc: result={'error':f'{type(exc).__name__}: {exc}'}
390
+ f.write((json.dumps(result,ensure_ascii=False)+'\n').encode()); f.flush()
391
+
392
+ def main():
393
+ global SOCKET_IDENTITY
394
+ RUNTIME_DIR.mkdir(parents=True,exist_ok=True)
395
+ try: os.chmod(RUNTIME_DIR,0o700)
396
+ except Exception: pass
397
+ import fcntl
398
+ owner=open(OWNER_LOCK_PATH,'a+b')
399
+ try:
400
+ fcntl.flock(owner,fcntl.LOCK_EX|fcntl.LOCK_NB)
401
+ except BlockingIOError:
402
+ owner.close(); return
403
+ global HARNESS_OWNER_PIPE
404
+ read_fd, write_fd = os.pipe() if os.name == 'posix' else (None, None)
405
+ HARNESS_OWNER_PIPE = write_fd
406
+ try:
407
+ owner_env = {
408
+ 'BH_OWNER_PID': str(HARNESS_OWNER_PID),
409
+ 'BH_OWNER_STARTED': str(HARNESS_OWNER_STARTED or ''),
410
+ 'BH_OWNER_TOKEN': HARNESS_OWNER_TOKEN,
411
+ }
412
+ if read_fd is not None:
413
+ owner_env['BH_OWNER_PIPE_FD'] = str(read_fd)
414
+ ensure_daemon(env=owner_env)
415
+ finally:
416
+ if read_fd is not None:
417
+ os.close(read_fd)
418
+ def terminate(_signum, _frame):
419
+ raise SystemExit(0)
420
+ signal.signal(signal.SIGTERM, terminate)
421
+ signal.signal(signal.SIGINT, terminate)
422
+ # Lock ownership is required before changing the shared socket path.
423
+ SOCKET_PATH.unlink(missing_ok=True)
424
+ srv=socket.socket(socket.AF_UNIX,socket.SOCK_STREAM); srv.bind(str(SOCKET_PATH)); os.chmod(SOCKET_PATH,0o600); SOCKET_IDENTITY=(SOCKET_PATH.stat().st_dev, SOCKET_PATH.stat().st_ino); srv.listen(32)
425
+ threading.Thread(target=reconciliation_loop, daemon=True).start()
426
+ try:
427
+ while True:
428
+ conn,_=srv.accept(); threading.Thread(target=serve_conn,args=(conn,),daemon=True).start()
429
+ finally:
430
+ srv.close(); owner.close()
431
+ if __name__=='__main__': main()
@@ -1,2 +0,0 @@
1
- """Browser Harness core package."""
2
-