major restructure

This commit is contained in:
buenosairesam
2026-01-20 05:31:26 -03:00
parent 27b32deba4
commit e4052374db
328 changed files with 1018 additions and 10018 deletions

526
build.py
View File

@@ -2,59 +2,54 @@
"""
Soleprint Build Tool
Builds the soleprint instance using modelgen for model generation.
All modes copy files (no symlinks) for Docker compatibility.
After editing source files, re-run `python build.py` to update gen/.
Generates soleprint instances from source + room config.
Usage:
python build.py # Build gen/standalone/
python build.py --cfg <room> # Build gen/<room>/
python build.py --all # Build all (standalone + rooms)
python build.py --deploy --output /path/ # Build for production
python build.py --models # Only regenerate models
python build.py # Build gen/standalone/
python build.py --cfg <room> # Build gen/<room>/
python build.py --all # Build all rooms
python build.py --output /path/ # Build to custom path
python build.py --models # Only regenerate models
Examples:
# Build standalone
python build.py
cd gen/standalone && .venv/bin/python run.py
Generated structure for standalone rooms:
gen/standalone/
artery/, atlas/, station/, main.py, ...
# With room config
python build.py --cfg myroom
cd gen/myroom && .venv/bin/python run.py
# Build all targets
python build.py --all
# Build for deployment
python build.py --deploy --output ../deploy/soleprint/
Generated structure for managed rooms:
gen/<room>/
<managed_name>/ # Copied repos + ctrl
link/ # Bridge code
soleprint/ # Soleprint instance
"""
import argparse
import json
import logging
import os
import shutil
import subprocess
import sys
from pathlib import Path
# SPR root is where this script lives
SPR_ROOT = Path(__file__).resolve().parent
# Configure logging
logging.basicConfig(
level=logging.INFO,
format="%(message)s",
)
logging.basicConfig(level=logging.INFO, format="%(message)s")
log = logging.getLogger(__name__)
def load_config(cfg_name: str | None) -> dict:
"""Load room config.json."""
room = cfg_name or "standalone"
config_path = SPR_ROOT / "cfg" / room / "config.json"
if config_path.exists():
return json.loads(config_path.read_text())
return {}
def ensure_dir(path: Path):
"""Create directory if it doesn't exist."""
path.mkdir(parents=True, exist_ok=True)
def copy_path(source: Path, target: Path):
def copy_path(source: Path, target: Path, quiet: bool = False):
"""Copy file or directory, resolving symlinks."""
if target.is_symlink():
target.unlink()
@@ -66,40 +61,153 @@ def copy_path(source: Path, target: Path):
if source.is_dir():
shutil.copytree(source, target, symlinks=False)
log.info(f" Copied: {target.name}/ ({count_files(target)} files)")
if not quiet:
log.info(f" {target.name}/")
else:
shutil.copy2(source, target)
log.info(f" Copied: {target.name}")
if not quiet:
log.info(f" {target.name}")
def count_files(path: Path) -> int:
"""Count files in directory recursively."""
return sum(1 for _ in path.rglob("*") if _.is_file())
def generate_models(output_dir: Path, cfg_name: str | None = None):
"""Generate models using modelgen tool.
def merge_into(source: Path, target: Path):
"""Merge source directory into target (additive, overwrites conflicts)."""
if not source.exists():
return
for item in source.rglob("*"):
if item.is_file():
rel = item.relative_to(source)
dest = target / rel
dest.parent.mkdir(parents=True, exist_ok=True)
shutil.copy2(item, dest)
Args:
output_dir: Directory where models/pydantic/__init__.py will be created
cfg_name: Room config name (e.g., 'amar'), or None for standalone
"""
room = cfg_name or "standalone"
def parse_gitignore(gitignore_path: Path) -> set[str]:
"""Parse .gitignore and return set of patterns to ignore."""
patterns = set()
if not gitignore_path.exists():
return patterns
for line in gitignore_path.read_text().splitlines():
line = line.strip()
# Skip comments and empty lines
if not line or line.startswith("#"):
continue
# Remove trailing slashes (directory indicators)
pattern = line.rstrip("/")
# Skip negation patterns (we don't support them)
if pattern.startswith("!"):
continue
patterns.add(pattern)
return patterns
def copy_repo(source: Path, target: Path):
"""Copy a repo directory, excluding .git and gitignored files."""
if not source.exists():
log.warning(f"Repo not found: {source}")
return False
# Always ignore these
always_ignore = {".git", "__pycache__", "node_modules", ".venv", "venv", ".env"}
# Parse .gitignore from repo root
gitignore_patterns = parse_gitignore(source / ".gitignore")
def ignore_patterns(directory, files):
ignored = set()
rel_dir = Path(directory).relative_to(source)
for f in files:
# Always ignore these
if f in always_ignore:
ignored.add(f)
continue
# Check gitignore patterns
rel_path = rel_dir / f if str(rel_dir) != "." else Path(f)
for pattern in gitignore_patterns:
# Simple pattern matching (filename or extension)
if pattern.startswith("*."):
# Extension pattern like *.pyc
if f.endswith(pattern[1:]):
ignored.add(f)
break
elif pattern == f or pattern == str(rel_path):
# Exact match
ignored.add(f)
break
elif "/" not in pattern and f == pattern:
# Simple name match anywhere
ignored.add(f)
break
return ignored
shutil.copytree(source, target, ignore=ignore_patterns, symlinks=False)
return True
def build_managed(output_dir: Path, cfg_name: str, config: dict):
"""Build managed folder with repos + ctrl."""
managed = config.get("managed", {})
managed_name = managed.get("name", cfg_name)
repos = managed.get("repos", {})
managed_dir = output_dir / managed_name
ensure_dir(managed_dir)
log.info(f"Building managed ({managed_name})...")
# Copy repos
for repo_name, repo_path in repos.items():
source = Path(repo_path)
target = managed_dir / repo_name
if copy_repo(source, target):
log.info(f" {repo_name}/")
# Copy ctrl from cfg/<room>/managed/ctrl/
room_cfg = SPR_ROOT / "cfg" / cfg_name
managed_ctrl = room_cfg / "managed" / "ctrl"
if managed_ctrl.exists():
ctrl_dir = managed_dir / "ctrl"
ensure_dir(ctrl_dir)
for item in managed_ctrl.iterdir():
copy_path(item, ctrl_dir / item.name)
def build_link(output_dir: Path, cfg_name: str):
"""Build link folder."""
room_cfg = SPR_ROOT / "cfg" / cfg_name
link_source = room_cfg / "link"
if not link_source.exists():
return
log.info("Building link...")
link_dir = output_dir / "link"
copy_path(link_source, link_dir)
def generate_models(output_dir: Path, room: str):
"""Generate models using modelgen tool."""
config_path = SPR_ROOT / "cfg" / room / "config.json"
if not config_path.exists():
log.warning(f"Config not found at {config_path}")
log.warning(f"Config not found: {config_path}")
return False
# Soleprint-specific: models go in models/pydantic/__init__.py
models_file = output_dir / "models" / "pydantic" / "__init__.py"
models_file.parent.mkdir(parents=True, exist_ok=True)
# Run modelgen as subprocess
cmd = [
sys.executable,
"-m",
"station.tools.modelgen",
"soleprint.station.tools.modelgen",
"from-config",
"--config",
str(config_path),
@@ -113,309 +221,143 @@ def generate_models(output_dir: Path, cfg_name: str | None = None):
return result.returncode == 0
def copy_cfg(output_dir: Path, cfg_name: str | None):
"""Copy configuration files to output directory.
Args:
output_dir: Target directory
cfg_name: Name of room config (e.g., 'amar'), or None for standalone
"""
room = cfg_name or "standalone"
def copy_cfg(output_dir: Path, room: str):
"""Copy room configuration files to output directory."""
room_cfg = SPR_ROOT / "cfg" / room
if not room_cfg.exists():
log.warning(f"Room config '{room}' not found at {room_cfg}")
log.warning(f"Room config not found: {room}")
return
log.info(f"\nCopying {room} room config...")
log.info(f"Copying {room} config...")
# Copy config.json to cfg/
# config.json -> cfg/
cfg_dir = output_dir / "cfg"
ensure_dir(cfg_dir)
room_config = room_cfg / "config.json"
if room_config.exists():
copy_path(room_config, cfg_dir / "config.json")
if (room_cfg / "config.json").exists():
copy_path(room_cfg / "config.json", cfg_dir / "config.json")
# Copy data/ to output data/
room_data = room_cfg / "data"
if room_data.exists():
log.info(f" Copying {room} data files...")
copy_path(room_data, output_dir / "data")
# data/ -> data/
if (room_cfg / "data").exists():
copy_path(room_cfg / "data", output_dir / "data")
# Copy .env.example to output root
env_example = room_cfg / ".env.example"
if env_example.exists():
copy_path(env_example, output_dir / ".env.example")
# .env.example
if (room_cfg / ".env.example").exists():
copy_path(room_cfg / ".env.example", output_dir / ".env.example")
# Copy room-specific databrowse depot if exists
room_databrowse = room_cfg / "databrowse" / "depot"
if room_databrowse.exists():
log.info(f" Copying {room} databrowse depot...")
target = output_dir / "station" / "monitors" / "databrowse" / "depot"
copy_path(room_databrowse, target)
# Copy room-specific tester tests if exists
room_tests = room_cfg / "tester" / "tests"
if room_tests.exists():
log.info(f" Copying {room} tester tests...")
target = output_dir / "station" / "tools" / "tester" / "tests"
copy_path(room_tests, target)
# Copy room-specific monitors if exists
room_monitors = room_cfg / "monitors"
if room_monitors.exists():
log.info(f" Copying {room} monitors...")
for monitor in room_monitors.iterdir():
if monitor.is_dir():
target = output_dir / "station" / "monitors" / monitor.name
copy_path(monitor, target)
# Copy room-specific models if exists
# Room-specific models
room_models = room_cfg / "models"
if room_models.exists():
log.info(f" Copying {room} models...")
target = output_dir / "models" / room
copy_path(room_models, target)
copy_path(room_models, output_dir / "models" / room)
# Copy room-specific soleprint config (docker-compose.yml, etc)
# Room-specific soleprint config (docker-compose.yml, etc)
room_soleprint = room_cfg / "soleprint"
if room_soleprint.exists():
log.info(f" Copying {room} soleprint config...")
for item in room_soleprint.iterdir():
copy_path(item, output_dir / item.name)
# Merge room-specific system configs (artery, atlas, station)
for system in ["artery", "atlas", "station"]:
room_system = room_cfg / system
if room_system.exists():
log.info(f" Merging {room} {system}...")
merge_into(room_system, output_dir / system)
def build_dev(output_dir: Path, cfg_name: str | None = None):
"""
Build for development using copies (Docker-compatible).
Structure:
gen/standalone/ or gen/<room>/
├── main.py
├── run.py
├── index.html
├── requirements.txt
├── Dockerfile
├── dataloader/
├── artery/
├── atlas/
├── station/
├── data/
├── cfg/
├── .env.example # From cfg/<room>/.env.example
└── models/ # Generated
After editing source files, re-run `python build.py` to update.
"""
log.info("\n=== Building DEV environment ===")
log.info(f"SPR root: {SPR_ROOT}")
log.info(f"Output: {output_dir}")
if cfg_name:
log.info(f"Room cfg: {cfg_name}")
ensure_dir(output_dir)
def build_soleprint(output_dir: Path, room: str):
"""Build soleprint folder with core + room config merged."""
soleprint = SPR_ROOT / "soleprint"
# Soleprint core files
log.info("\nCopying soleprint files...")
soleprint = SPR_ROOT / "soleprint"
copy_path(soleprint / "main.py", output_dir / "main.py")
copy_path(soleprint / "run.py", output_dir / "run.py")
copy_path(soleprint / "index.html", output_dir / "index.html")
copy_path(soleprint / "requirements.txt", output_dir / "requirements.txt")
log.info("Copying soleprint core...")
for name in [
"main.py",
"run.py",
"index.html",
"generate.html",
"requirements.txt",
"Dockerfile",
]:
if (soleprint / name).exists():
copy_path(soleprint / name, output_dir / name)
copy_path(soleprint / "dataloader", output_dir / "dataloader")
if (soleprint / "Dockerfile").exists():
copy_path(soleprint / "Dockerfile", output_dir / "Dockerfile")
# System directories
log.info("\nCopying systems...")
log.info("Copying systems...")
for system in ["artery", "atlas", "station"]:
source = SPR_ROOT / system
source = soleprint / system
if source.exists():
copy_path(source, output_dir / system)
# Config (includes data/ from room)
log.info("\nCopying config...")
copy_cfg(output_dir, cfg_name)
# Room config (includes merging room-specific artery/atlas/station)
copy_cfg(output_dir, room)
# Models (generated)
log.info("\nGenerating models...")
if not generate_models(output_dir, cfg_name):
log.warning("Model generation failed, you may need to run it manually")
# Generate models
log.info("Generating models...")
if not generate_models(output_dir, room):
log.warning("Model generation failed")
log.info("\n✓ Dev build complete!")
log.info(f"\nTo run:")
log.info(f" cd {output_dir}")
log.info(f" python3 -m venv .venv")
log.info(f" .venv/bin/pip install -r requirements.txt")
log.info(f" .venv/bin/python run.py # Single-port bare-metal dev")
if cfg_name:
log.info(
f"\nAfter editing source, rebuild with: python build.py --cfg {cfg_name}"
)
def build(output_dir: Path, cfg_name: str | None = None):
"""Build complete room instance."""
room = cfg_name or "standalone"
config = load_config(cfg_name)
managed = config.get("managed")
log.info(f"\n=== Building {room} ===")
ensure_dir(output_dir)
if managed:
# 3-folder structure: <managed>/, link/, soleprint/
build_managed(output_dir, room, config)
build_link(output_dir, room)
soleprint_name = config.get("framework", {}).get("name", "soleprint")
soleprint_dir = output_dir / soleprint_name
ensure_dir(soleprint_dir)
build_soleprint(soleprint_dir, room)
else:
log.info(f"\nAfter editing source, rebuild with: python build.py")
# Standalone: everything in output_dir
build_soleprint(output_dir, room)
log.info(f"\n✓ Built: {output_dir}")
def build_deploy(output_dir: Path, cfg_name: str | None = None):
"""
Build for deployment by copying all files (no symlinks).
"""
log.info("\n=== Building DEPLOY package ===")
log.info(f"SPR root: {SPR_ROOT}")
log.info(f"Output: {output_dir}")
if cfg_name:
log.info(f"Room cfg: {cfg_name}")
if output_dir.exists():
response = input(f"\nOutput directory exists. Overwrite? [y/N] ")
if response.lower() != "y":
log.info("Aborted.")
return
shutil.rmtree(output_dir)
ensure_dir(output_dir)
# Soleprint core files (copy)
log.info("\nCopying soleprint files...")
soleprint = SPR_ROOT / "soleprint"
copy_path(soleprint / "main.py", output_dir / "main.py")
copy_path(soleprint / "run.py", output_dir / "run.py")
copy_path(soleprint / "index.html", output_dir / "index.html")
copy_path(soleprint / "requirements.txt", output_dir / "requirements.txt")
copy_path(soleprint / "dataloader", output_dir / "dataloader")
if (soleprint / "Dockerfile").exists():
copy_path(soleprint / "Dockerfile", output_dir / "Dockerfile")
# System directories (copy)
log.info("\nCopying systems...")
for system in ["artery", "atlas", "station"]:
source = SPR_ROOT / system
if source.exists():
copy_path(source, output_dir / system)
# Config (includes data/ from room)
log.info("\nCopying config...")
copy_cfg(output_dir, cfg_name)
# Models (generate fresh) - pass output_dir, modelgen adds models/pydantic
log.info("\nGenerating models...")
if not generate_models(output_dir, cfg_name):
# Fallback: copy from gen if exists
room = cfg_name or "standalone"
existing = SPR_ROOT / "gen" / room / "models"
if existing.exists():
log.info(" Using existing models from gen/")
copy_path(existing, output_dir / "models")
# Copy schema.json for reference
log.info("\nCopying schema...")
copy_path(SPR_ROOT / "schema.json", output_dir / "schema.json")
# Create run script
run_script = output_dir / "start.sh"
run_script.write_text("""#!/bin/bash
# Soleprint runner
cd "$(dirname "$0")"
if [ ! -d ".venv" ]; then
echo "Creating virtual environment..."
python3 -m venv .venv
.venv/bin/pip install -r requirements.txt
fi
echo "Starting soleprint on http://localhost:12000"
.venv/bin/python main.py
""")
run_script.chmod(0o755)
log.info(" Created: start.sh")
total_files = count_files(output_dir)
log.info(f"\n✓ Deploy build complete! ({total_files} files)")
log.info(f"\nTo run:")
log.info(f" cd {output_dir}")
log.info(f" ./start.sh")
log.info(f"\nOr deploy to server:")
log.info(f" rsync -av {output_dir}/ server:/app/soleprint/")
log.info(f" ssh server 'cd /app/soleprint && ./start.sh'")
def build_models():
def build_models_only():
"""Only regenerate models."""
log.info("\n=== Generating models only ===")
output_dir = SPR_ROOT / "gen"
ensure_dir(output_dir)
if generate_models(output_dir):
log.info("\n✓ Models generated!")
log.info("Generating models...")
if generate_models(SPR_ROOT / "gen"):
log.info("✓ Models generated")
else:
log.error("Model generation failed")
sys.exit(1)
def main():
parser = argparse.ArgumentParser(
description="Soleprint Build Tool",
formatter_class=argparse.RawDescriptionHelpFormatter,
epilog=__doc__,
)
parser = argparse.ArgumentParser(description="Soleprint Build Tool")
parser.add_argument(
"--output",
"-o",
type=Path,
default=None,
help="Output directory (default: gen/standalone/ or gen/<cfg>/)",
)
parser.add_argument(
"--cfg",
"-c",
type=str,
default=None,
help="Room config to include (e.g., 'myroom')",
)
parser.add_argument(
"--all",
action="store_true",
help="Build all configs (standalone + all rooms in cfg/)",
)
parser.add_argument(
"--deploy",
action="store_true",
help="Build for deployment (creates start.sh, schema.json)",
)
parser.add_argument(
"--models",
action="store_true",
help="Only regenerate models",
)
parser.add_argument("--output", "-o", type=Path, help="Output directory")
parser.add_argument("--cfg", "-c", type=str, help="Room config name")
parser.add_argument("--all", action="store_true", help="Build all rooms")
parser.add_argument("--models", action="store_true", help="Only regenerate models")
args = parser.parse_args()
if args.models:
build_models()
elif args.deploy:
if not args.output:
log.error("--deploy requires --output")
sys.exit(1)
build_deploy(args.output.resolve(), args.cfg)
build_models_only()
elif args.all:
# Build standalone
build_dev(SPR_ROOT / "gen" / "standalone", None)
# Build all room configs
cfg_dir = SPR_ROOT / "cfg"
for room in cfg_dir.iterdir():
build(SPR_ROOT / "gen" / "standalone", None)
for room in (SPR_ROOT / "cfg").iterdir():
if room.is_dir() and room.name not in ("__pycache__", "standalone"):
build_dev(SPR_ROOT / "gen" / room.name, room.name)
build(SPR_ROOT / "gen" / room.name, room.name)
else:
# Determine output directory
if args.output:
output_dir = args.output.resolve()
elif args.cfg:
output_dir = SPR_ROOT / "gen" / args.cfg
else:
output_dir = SPR_ROOT / "gen" / "standalone"
build_dev(output_dir, args.cfg)
build(output_dir, args.cfg)
if __name__ == "__main__":