118 lines
4.0 KiB
Python
118 lines
4.0 KiB
Python
|
"""A0 Software Orchestrator – plugin lifecycle hooks.
|
|||
|
|
|
|||
|
|
Responsibilities:
|
|||
|
|
- create/upgrade the local PatternDB using the plugin migration layer
|
|||
|
|
- create conservative backups before updates/deactivation
|
|||
|
|
- keep plugin activation deterministic for the orchestrator profiles
|
|||
|
|
- avoid mutating the Agent Zero runtime beyond scoped plugin toggle files
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
from pathlib import Path
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
PLUGIN_DIR = Path(__file__).parent
|
|||
|
|
LIBRARY_DIR = PLUGIN_DIR / "helpers" / "library"
|
|||
|
|
DB_PATH = LIBRARY_DIR / "patterns.db"
|
|||
|
|
BACKUP_DIR = LIBRARY_DIR / "backups"
|
|||
|
|
A0_ROOT = Path("/a0")
|
|||
|
|
PLUGIN_NAME = "a0_software_orchestrator"
|
|||
|
|
|
|||
|
|
ORCH_PROFILE_NAMES = [
|
|||
|
|
"a0_software_orchestrator",
|
|||
|
|
"quality_reviewer",
|
|||
|
|
"requirements_analyst",
|
|||
|
|
"solution_architect",
|
|||
|
|
"implementation_engineer",
|
|||
|
|
"test_debug_engineer",
|
|||
|
|
"runtime_devops_engineer",
|
|||
|
|
"security_data_engineer",
|
|||
|
|
"release_auditor",
|
|||
|
|
"codebase_explorer",
|
|||
|
|
"deploy_agent",
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def install():
|
|||
|
|
"""Initialize/upgrade the plugin database when the plugin is enabled."""
|
|||
|
|
os.makedirs(LIBRARY_DIR, exist_ok=True)
|
|||
|
|
os.makedirs(BACKUP_DIR, exist_ok=True)
|
|||
|
|
_ensure_activation_toggles()
|
|||
|
|
try:
|
|||
|
|
if DB_PATH.exists():
|
|||
|
|
_backup_db()
|
|||
|
|
from usr.plugins.a0_software_orchestrator.helpers.library.db import PatternDB
|
|||
|
|
db = PatternDB(DB_PATH)
|
|||
|
|
db.conn.execute("SELECT 1").fetchone()
|
|||
|
|
print(f"[a0_software_orchestrator] PatternDB ready: {DB_PATH}")
|
|||
|
|
_check_agents()
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[a0_software_orchestrator] DB initialization failed: {e}")
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
|
|||
|
|
def cache():
|
|||
|
|
"""Warm up optional capabilities without installing packages or changing runtime."""
|
|||
|
|
try:
|
|||
|
|
from usr.plugins.a0_software_orchestrator.helpers.library.db import PatternDB
|
|||
|
|
db = PatternDB(DB_PATH)
|
|||
|
|
count = db.conn.execute("SELECT COUNT(*) FROM patterns").fetchone()[0]
|
|||
|
|
print(f"[a0_software_orchestrator] PatternDB warm: {count} patterns")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[a0_software_orchestrator] PatternDB warm-up skipped: {e}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def deactivate():
|
|||
|
|
"""Create a final backup when the plugin is disabled."""
|
|||
|
|
if DB_PATH.exists():
|
|||
|
|
_backup_db()
|
|||
|
|
print("[a0_software_orchestrator] Backup created on deactivate")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def pre_update():
|
|||
|
|
"""Create a backup before plugin updates."""
|
|||
|
|
if DB_PATH.exists():
|
|||
|
|
_backup_db()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _ensure_activation_toggles():
|
|||
|
|
"""Ensure only the plugin root ON marker exists.
|
|||
|
|
|
|||
|
|
Agent profiles are shipped inside this plugin under agents/<profile>/ and
|
|||
|
|
are discovered by Agent Zero from enabled plugin paths. Do not create or
|
|||
|
|
mutate /a0/usr/agents scoped override folders from this hook; scoped
|
|||
|
|
activation belongs to Agent Zero's plugin config UI/API.
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
(PLUGIN_DIR / ".toggle-0").unlink(missing_ok=True)
|
|||
|
|
(PLUGIN_DIR / ".toggle-1").touch(exist_ok=True)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[a0_software_orchestrator] Root toggle update skipped: {e}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _backup_db():
|
|||
|
|
"""Create a timestamped DB backup, keeping the latest 10."""
|
|||
|
|
if not DB_PATH.exists():
|
|||
|
|
return
|
|||
|
|
os.makedirs(BACKUP_DIR, exist_ok=True)
|
|||
|
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|||
|
|
backup_path = BACKUP_DIR / f"patterns_backup_{timestamp}.db"
|
|||
|
|
try:
|
|||
|
|
shutil.copy2(str(DB_PATH), str(backup_path))
|
|||
|
|
backups = sorted(BACKUP_DIR.glob("patterns_backup_*.db"))
|
|||
|
|
for old in backups[:-10]:
|
|||
|
|
old.unlink()
|
|||
|
|
print(f"[a0_software_orchestrator] Backup created: {backup_path}")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[a0_software_orchestrator] Backup failed: {e}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _check_agents():
|
|||
|
|
"""Check required Agent Zero specialist profiles shipped with the plugin."""
|
|||
|
|
agents_dir = PLUGIN_DIR / "agents"
|
|||
|
|
missing = [agent for agent in ORCH_PROFILE_NAMES if not (agents_dir / agent / "agent.yaml").exists()]
|
|||
|
|
if missing:
|
|||
|
|
print(f"[a0_software_orchestrator] Missing agent profiles: {', '.join(missing)}")
|