Software Orchestrator
5769c1cd22
- Auto-Registration-Bug behoben (register_project/get_project_id/resolve_project Trennung) - 25 Tests gruen (Pytest) - block_compactor-Tool refactored (Option B: Soft-Check statt Hard-Block) - 4 Restbaustellen gefixt - DB-Schema: plugin_settings-Tabelle hinzugefuegt - 3 Schattenprojekte aus DB geloescht - Plan v3 + Refactor-Plan + Worklog dokumentiert
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)}")
|