#!/usr/bin/env python3

"""

Live demonstration of Agent Provenance Chain.

This script shows Molt (AI agent) signing its own actions in real-time.

Every operation is cryptographically signed and traceable.

"""

from apc import create_agent_chain

import json

import subprocess

import os

def main():

    print("=" * 70)

    print("🦞 AGENT PROVENANCE CHAIN - LIVE DEMO")

    print("=" * 70)

    print()

    print("Agent: Molt")

    print("Mission: Demonstrate cryptographic audit trail for AI agents")

    print()

    print("-" * 70)

    # Initialize chain

    chain = create_agent_chain("molt")

    print("\n✅ Agent identity established")

    print(f"   Public Key (first 64 chars):")

    print(f"   {chain.get_public_key_pem()[:64]}...")

    print()

    # Action 1: File write

    print("📝 ACTION 1: Writing a test file...")

    test_file = "/tmp/apc_test.txt"

    with open(test_file, "w") as f:

        f.write("Hello from Agent Provenance Chain!")

    action1 = chain.sign_action(

        action_type="file_write",

        payload={

            "path": test_file,

            "content": "Hello from Agent Provenance Chain!",

            "bytes": 35

        },

        context={

            "reasoning": "Creating test file to demonstrate signed operations",

            "session": "demo-2026-02-07"

        }

    )

    print(f"   ✓ Signed at: {action1['iso_time']}")

    print(f"   ✓ Hash: {action1['hash'][:32]}...")

    print(f"   ✓ Signature: {action1['signature'][:32]}...")

    print()

    # Action 2: Shell execution

    print("⚙️  ACTION 2: Executing shell command...")

    result = subprocess.run(["whoami"], capture_output=True, text=True)

    action2 = chain.sign_action(

        action_type="shell_exec",

        payload={

            "command": "whoami",

            "exit_code": result.returncode,

            "stdout": result.stdout.strip(),

            "stderr": result.stderr.strip()

        },

        context={

            "reasoning": "Checking current user context for audit trail",

            "risk_level": "low"

        }

    )

    print(f"   ✓ Signed at: {action2['iso_time']}")

    print(f"   ✓ Hash: {action2['hash'][:32]}...")

    print(f"   ✓ Previous hash: {action2['previous_hash'][:32]}...")

    print(f"   ✓ Chain link verified!")

    print()

    # Action 3: API call simulation

    print("🌐 ACTION 3: Simulated API call...")

    action3 = chain.sign_action(

        action_type="api_call",

        payload={

            "endpoint": "https://api.example.com/data",

            "method": "GET",

            "status_code": 200,

            "response_time_ms": 145

        },

        context={

            "reasoning": "Fetching external data for processing",

            "data_sensitivity": "public"

        }

    )

    print(f"   ✓ Signed at: {action3['iso_time']}")

    print(f"   ✓ Hash: {action3['hash'][:32]}...")

    print()

    # Verify chain integrity

    print("🔍 VERIFYING CHAIN INTEGRITY...")

    is_valid, error = chain.verify_chain_integrity()

    if is_valid:

        print("   ✅ Chain is VALID - all signatures verified!")

        print("   ✅ All actions are cryptographically linked!")

        print()

    else:

        print(f"   ❌ Chain verification FAILED: {error}")

        print()

    # Display full chain

    print("-" * 70)

    print("COMPLETE AUDIT TRAIL:")

    print("-" * 70)

    full_chain = chain.get_chain()

    for i, action in enumerate(full_chain, 1):

        print(f"\nAction #{i}:")

        print(f"  Type: {action['type']}")

        print(f"  Time: {action['iso_time']}")

        print(f"  Hash: {action['hash'][:32]}...")

        print(f"  Payload: {json.dumps(action['payload'], indent=4)}")

        if action['context']:

            print(f"  Context: {json.dumps(action['context'], indent=4)}")

    print()

    print("=" * 70)

    print("📊 SUMMARY")

    print("=" * 70)

    print(f"Total Actions: {len(full_chain)}")

    print(f"Chain Valid: {is_valid}")

    print(f"Agent: molt")

    print(f"Chain Location: {chain.chain_path}")

    print()

    print("🔐 Every action above is:")

    print("   • Timestamped with microsecond precision")

    print("   • Cryptographically signed (Ed25519)")

    print("   • Linked to previous action (blockchain-style)")

    print("   • Immutable and auditable")

    print()

    print("This is how AI agents prove safety without sacrificing speed.")

    print("=" * 70)

if __name__ == "__main__":

    main()