mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-17 12:31:20 +00:00
8b921a8ded
Use /a0/tmp/playwright as the Browser plugin Chromium cache and Docker install target while preserving full Chromium installs. Add startup migration cleanup for retired usr Playwright caches, update Browser status/runtime references and docs, and cover migration behavior with focused regressions.
130 lines
4.1 KiB
Python
130 lines
4.1 KiB
Python
from __future__ import annotations
|
|
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
from helpers import files, plugins, yaml as yaml_helper
|
|
from plugins._browser.helpers.config import (
|
|
PLUGIN_NAME,
|
|
browser_runtime_config,
|
|
normalize_browser_config,
|
|
)
|
|
from plugins._browser.helpers.playwright import (
|
|
find_playwright_binary,
|
|
get_playwright_cache_dir,
|
|
get_retired_playwright_cache_dirs,
|
|
)
|
|
from plugins._browser.helpers.runtime import close_all_runtimes_sync
|
|
|
|
|
|
def _load_saved_browser_config(project_name: str = "", agent_profile: str = "") -> dict:
|
|
entries = plugins.find_plugin_assets(
|
|
plugins.CONFIG_FILE_NAME,
|
|
plugin_name=PLUGIN_NAME,
|
|
project_name=project_name,
|
|
agent_profile=agent_profile,
|
|
only_first=True,
|
|
)
|
|
path = entries[0].get("path", "") if entries else ""
|
|
if path and files.exists(path):
|
|
return files.read_file_json(path) or {}
|
|
|
|
plugin_dir = plugins.find_plugin_dir(PLUGIN_NAME)
|
|
default_path = (
|
|
files.get_abs_path(plugin_dir, plugins.CONFIG_DEFAULT_FILE_NAME)
|
|
if plugin_dir
|
|
else ""
|
|
)
|
|
if default_path and files.exists(default_path):
|
|
return yaml_helper.loads(files.read_file(default_path)) or {}
|
|
|
|
return {}
|
|
|
|
|
|
def get_plugin_config(default=None, **kwargs):
|
|
return normalize_browser_config(default)
|
|
|
|
|
|
def save_plugin_config(settings=None, project_name="", agent_profile="", **kwargs):
|
|
normalized = normalize_browser_config(settings)
|
|
current = normalize_browser_config(
|
|
_load_saved_browser_config(project_name=project_name, agent_profile=agent_profile)
|
|
)
|
|
if browser_runtime_config(normalized) != browser_runtime_config(current):
|
|
close_all_runtimes_sync()
|
|
return normalized
|
|
|
|
|
|
def cleanup_playwright_cache() -> dict:
|
|
primary = Path(get_playwright_cache_dir())
|
|
retired_dirs = [
|
|
path for path in get_retired_playwright_cache_dirs() if path.resolve() != primary.resolve()
|
|
]
|
|
result = {"primary": str(primary), "migrated": "", "removed": [], "errors": []}
|
|
|
|
if find_playwright_binary(primary):
|
|
_remove_cache_dirs(retired_dirs, result)
|
|
return result
|
|
|
|
source = _best_playwright_cache(retired_dirs)
|
|
if not source:
|
|
return result
|
|
|
|
backup = _next_backup_path(primary) if primary.exists() else None
|
|
try:
|
|
if backup:
|
|
primary.rename(backup)
|
|
primary.parent.mkdir(parents=True, exist_ok=True)
|
|
shutil.move(str(source), str(primary))
|
|
result["migrated"] = str(source)
|
|
except Exception as exc:
|
|
if backup and backup.exists() and not primary.exists():
|
|
backup.rename(primary)
|
|
result["errors"].append(f"Failed to migrate {source} to {primary}: {exc}")
|
|
return result
|
|
|
|
if not find_playwright_binary(primary):
|
|
result["errors"].append(f"Migrated Playwright cache is not valid: {primary}")
|
|
if backup:
|
|
result["errors"].append(f"Previous primary Playwright cache retained at {backup}")
|
|
return result
|
|
|
|
if backup:
|
|
_remove_cache_dirs([backup], result)
|
|
_remove_cache_dirs(retired_dirs, result)
|
|
return result
|
|
|
|
|
|
def _best_playwright_cache(candidates: list[Path]) -> Path | None:
|
|
valid = [path for path in candidates if path.is_dir() and find_playwright_binary(path)]
|
|
if not valid:
|
|
return None
|
|
|
|
def modified_at(path: Path) -> float:
|
|
binary = find_playwright_binary(path)
|
|
try:
|
|
return binary.stat().st_mtime if binary else path.stat().st_mtime
|
|
except OSError:
|
|
return 0
|
|
|
|
return max(valid, key=modified_at)
|
|
|
|
|
|
def _next_backup_path(path: Path) -> Path:
|
|
backup = path.with_name(f"{path.name}.migration-backup")
|
|
counter = 2
|
|
while backup.exists():
|
|
backup = path.with_name(f"{path.name}.migration-backup-{counter}")
|
|
counter += 1
|
|
return backup
|
|
|
|
|
|
def _remove_cache_dirs(paths: list[Path], result: dict) -> None:
|
|
for path in paths:
|
|
if not path.exists():
|
|
continue
|
|
try:
|
|
shutil.rmtree(path)
|
|
result["removed"].append(str(path))
|
|
except Exception as exc:
|
|
result["errors"].append(f"Failed to remove Playwright cache {path}: {exc}")
|