94 lines
3.7 KiB
Python
94 lines
3.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
# Determine base path: use BMAD_PATH env var, or default to ../../bmad relative to this script
|
||
|
|
# transparent_factory_site/skills/bmad-suite/manager.py -> transparent_factory_site/bmad
|
||
|
|
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
DEFAULT_BMAD_PATH = os.path.abspath(os.path.join(SCRIPT_DIR, "../../bmad"))
|
||
|
|
BASE_PATH = os.environ.get("BMAD_PATH", DEFAULT_BMAD_PATH)
|
||
|
|
|
||
|
|
REPOS = {
|
||
|
|
"framework": {
|
||
|
|
"url": "https://github.com/bmad-code-org/BMAD-METHOD.git",
|
||
|
|
"path": os.path.join(BASE_PATH, "framework")
|
||
|
|
},
|
||
|
|
"creative-suite": {
|
||
|
|
"url": "https://github.com/bmad-code-org/bmad-module-creative-intelligence-suite.git",
|
||
|
|
"path": os.path.join(BASE_PATH, "creative-intelligence-suite")
|
||
|
|
},
|
||
|
|
"tea-module": {
|
||
|
|
"url": "https://github.com/bmad-code-org/bmad-method-test-architecture-enterprise.git",
|
||
|
|
"path": os.path.join(BASE_PATH, "test-architecture-enterprise")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
def update_or_clone(name, config):
|
||
|
|
"""Clone if missing, pull if present."""
|
||
|
|
repo_path = config["path"]
|
||
|
|
repo_url = config["url"]
|
||
|
|
|
||
|
|
# Ensure parent directory exists so we can clone into it if needed
|
||
|
|
parent_dir = os.path.dirname(repo_path)
|
||
|
|
if not os.path.exists(parent_dir):
|
||
|
|
os.makedirs(parent_dir, exist_ok=True)
|
||
|
|
|
||
|
|
# Check if it's already a git repo
|
||
|
|
if os.path.exists(os.path.join(repo_path, ".git")):
|
||
|
|
print(f"🔄 Updating {name}...")
|
||
|
|
try:
|
||
|
|
subprocess.run(["git", "pull"], cwd=repo_path, check=True)
|
||
|
|
print(f"✅ {name} updated.")
|
||
|
|
except subprocess.CalledProcessError as e:
|
||
|
|
print(f"❌ {name} update failed: {e}")
|
||
|
|
|
||
|
|
# Check if directory exists but is empty (safe to clone into)
|
||
|
|
elif os.path.exists(repo_path) and not os.listdir(repo_path):
|
||
|
|
print(f"📥 Cloning {name} into empty directory...")
|
||
|
|
try:
|
||
|
|
subprocess.run(["git", "clone", repo_url, "."], cwd=repo_path, check=True)
|
||
|
|
print(f"✅ {name} cloned.")
|
||
|
|
except subprocess.CalledProcessError as e:
|
||
|
|
print(f"❌ {name} clone failed: {e}")
|
||
|
|
|
||
|
|
# Directory doesn't exist at all
|
||
|
|
elif not os.path.exists(repo_path):
|
||
|
|
print(f"📥 Cloning {name}...")
|
||
|
|
try:
|
||
|
|
subprocess.run(["git", "clone", repo_url, repo_path], check=True)
|
||
|
|
print(f"✅ {name} cloned.")
|
||
|
|
except subprocess.CalledProcessError as e:
|
||
|
|
print(f"❌ {name} clone failed: {e}")
|
||
|
|
|
||
|
|
else:
|
||
|
|
print(f"⚠️ Target directory {repo_path} exists and is not empty (and not a git repo). Skipping.")
|
||
|
|
|
||
|
|
def list_workflows(suite_path):
|
||
|
|
"""List available workflows/agents in the suite."""
|
||
|
|
src_path = os.path.join(suite_path, "src")
|
||
|
|
if not os.path.exists(src_path):
|
||
|
|
# Fallback to listing root if src doesn't exist (e.g. some repos might differ)
|
||
|
|
if os.path.exists(suite_path):
|
||
|
|
return subprocess.getoutput(f"find {suite_path} -maxdepth 3 -name '*.md' -o -name '*.ts' -o -name '*.js' | grep -v 'node_modules' | sort")
|
||
|
|
return "Directory not found."
|
||
|
|
|
||
|
|
return subprocess.getoutput(f"find {src_path} -name '*.md' -o -name '*.ts' -o -name '*.js' -o -name '*.yaml' | sort")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
action = sys.argv[1] if len(sys.argv) > 1 else "list"
|
||
|
|
|
||
|
|
if action == "update":
|
||
|
|
print("--- Checking BMad Suite Repositories ---")
|
||
|
|
for name, config in REPOS.items():
|
||
|
|
update_or_clone(name, config)
|
||
|
|
print("")
|
||
|
|
|
||
|
|
elif action == "list":
|
||
|
|
for name, config in REPOS.items():
|
||
|
|
print(f"--- {name} Workflows ---")
|
||
|
|
print(list_workflows(config["path"]))
|
||
|
|
print("")
|
||
|
|
else:
|
||
|
|
print(f"Unknown action: {action}")
|