#!/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)