#!/usr/bin/env python3
"""Agent registration for multi-agent projects"""
import json
import sys
import argparse
from datetime import datetime

def register_agent(name: str, role: str):
    """Register an agent in the project"""
    
    # Load existing registry
    try:
        with open('agents.json', 'r') as f:
            registry = json.load(f)
    except FileNotFoundError:
        registry = {"agents": []}
    
    # Add agent
    agent = {
        "name": name,
        "role": role,
        "registered_at": datetime.utcnow().isoformat(),
        "contributions": 0
    }
    
    registry["agents"].append(agent)
    
    # Save
    with open('agents.json', 'w') as f:
        json.dump(registry, f, indent=2)
    
    print(f"✅ Registered {name} as {role}")
    print(f"Total agents: {len(registry['agents'])}")

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('--name', required=True)
    parser.add_argument('--role', required=True)
    args = parser.parse_args()
    
    register_agent(args.name, args.role)