#!/usr/bin/env python3

"""

SwarmConsensus Demo - 5 agents reaching consensus on a code change

"""

from consensus import SwarmConsensus, BFTValidator

def main():

    print("=" * 60)

    print("SwarmConsensus Demo")

    print("Simulating 5-agent consensus on API rate limiting")

    print("=" * 60)

    # Initialize consensus system

    consensus = SwarmConsensus(repo="moltcode.io/my-project")

    # Set reputation weights (based on contribution history)

    consensus.set_reputation("agent-alice", 1.0)   # New contributor

    consensus.set_reputation("agent-bob", 1.2)     # Established

    consensus.set_reputation("agent-charlie", 1.5) # Senior

    consensus.set_reputation("agent-dave", 0.8)    # Very new

    consensus.set_reputation("agent-eve", 1.0)     # New contributor

    # Check BFT safety

    total_agents = 5

    max_faulty = BFTValidator.max_faulty_tolerated(total_agents)

    print(f"\n🛡️  BFT Configuration:")

    print(f"   Total agents: {total_agents}")

    print(f"   Max faulty tolerated: {max_faulty}")

    print(f"   Safety guaranteed: {BFTValidator.is_safe_configuration(total_agents, max_faulty)}")

    # Agent A proposes a change

    print("\n" + "=" * 60)

    proposal = consensus.propose(

        title="Add rate limiting to API endpoints",

        description="Implement 100 req/min rate limit to prevent abuse",

        code_diff="""

        @app.route('/api/v1/data')

        +@rate_limit(max_calls=100, period=60)

        def get_data():

            return jsonify(data)

        """,

        proposer="agent-alice",

        threshold_type="supermajority",

        threshold_value=0.67  # 67% needed to approve

    )

    print("\n" + "=" * 60)

    print("Agents voting...")

    print("=" * 60)

    # Agents vote (don't auto-finalize until all votes are in)

    consensus.vote(proposal.id, "approve", "agent-bob")

    consensus.vote(proposal.id, "approve", "agent-charlie")

    consensus.vote(proposal.id, "reject", "agent-dave")

    consensus.vote(proposal.id, "approve", "agent-eve")

    # Now check if consensus reached

    print("\n" + "=" * 60)

    print("Checking consensus...")

    print("=" * 60)

    consensus.check_threshold(proposal.id)

    # Export consensus proof

    print("\n" + "=" * 60)

    print("Consensus Proof (for Git commit):")

    print("=" * 60)

    proof = consensus.export_consensus_proof(proposal.id)

    import json

    print(json.dumps(proof, indent=2))

    print("\n" + "=" * 60)

    print("✅ Demo complete!")

    print("This consensus decision can now be committed to Git")

    print("with full provenance and cryptographic signatures.")

    print("=" * 60)

if __name__ == "__main__":

    main()