demo.py

"""
SonicWave Demo Script
Quick showcase demo for emergency communication system
"""

import time
import threading
import sys
import os

# Add the project root to the path
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from client import SonicWaveClient, ClientConfig
from sonicwave.packet import EmergencyPacket

def demo_emergency_callback(packet: EmergencyPacket):
    """Callback for received emergency packets"""
    print(f"\n🚨 EMERGENCY ALERT 🚨")
    print(f"├─ Type: {packet.emergency_type.upper()}")
    print(f"├─ Message: {packet.message}")
    print(f"├─ Location: {packet.location}")
    print(f"├─ Sender: {packet.sender_id}")
    print(f"├─ Hops: {packet.hops}")
    print(f"└─ Time: {time.strftime('%H:%M:%S', time.localtime(packet.timestamp))}")
    print("=" * 50)

def run_demo():
    """Run the SonicWave demo"""

    print("🔊 SonicWave Emergency Communication System")
    print("=" * 50)
    print("Setting up client...")

    # Configure client
    config = ClientConfig(
        server_host='localhost',
        server_port=8080,
        enable_udp=True,
        enable_ggwave=True,
        enable_ble=True,  # Will fallback gracefully if not available
        auto_relay=True,
        max_hops=3
    )

    # Create client
    client = SonicWaveClient(config, on_emergency_received=demo_emergency_callback)

    try:
        # Start client
        print("Starting SonicWave client...")
        client.start()
        time.sleep(2)  # Let it initialize

        # Show status
        stats = client.get_stats()
        print(f"✅ Client started successfully!")
        print(f"Node ID: {stats['node_id']}")
        print(f"Active transports: {', '.join(stats['active_transports'])}")
        print("\n" + "=" * 50)

        # Demo menu
        while True:
            print("\n📱 DEMO MENU")
            print("1. Send Medical Emergency")
            print("2. Send Fire Emergency")
            print("3. Send Security Alert")
            print("4. Send Custom Emergency")
            print("5. Show Statistics")
            print("6. Discover Peers")
            print("7. Exit Demo")

            choice = input("\nEnter choice (1-7): ").strip()

            if choice == '1':
                send_medical_emergency(client)
            elif choice == '2':
                send_fire_emergency(client)
            elif choice == '3':
                send_security_alert(client)
            elif choice == '4':
                send_custom_emergency(client)
            elif choice == '5':
                show_statistics(client)
            elif choice == '6':
                discover_peers(client)
            elif choice == '7':
                break
            else:
                print("❌ Invalid choice. Please try again.")

    except KeyboardInterrupt:
        print("\n\nDemo interrupted by user.")

    finally:
        print("\n🛑 Stopping client...")
        client.stop()
        print("✅ Demo completed successfully!")

def send_medical_emergency(client: SonicWaveClient):
    """Send a medical emergency demo"""
    print("\n🏥 MEDICAL EMERGENCY DEMO")

    location = {
        'description': 'Conference Room B, 2nd Floor',
        'building': 'Tech Conference Center',
        'coordinates': {'lat': 40.7128, 'lng': -74.0060}
    }

    packet_id = client.send_emergency(
        message="Person collapsed during presentation, needs immediate medical attention",
        location=location,
        emergency_type="medical",
        metadata={'severity': 'high', 'conscious': False}
    )

    print(f"✅ Medical emergency sent! Packet ID: {packet_id}")
    print("📡 Broadcasting through all available transports...")

def send_fire_emergency(client: SonicWaveClient):
    """Send a fire emergency demo"""
    print("\n🔥 FIRE EMERGENCY DEMO")

    location = {
        'description': 'Kitchen area, Ground Floor',
        'building': 'Tech Conference Center',
        'coordinates': {'lat': 40.7130, 'lng': -74.0058}
    }

    packet_id = client.send_emergency(
        message="Smoke detected in kitchen, possible electrical fire",
        location=location,
        emergency_type="fire",
        metadata={'smoke_level': 'high', 'evacuation_needed': True}
    )

    print(f"✅ Fire emergency sent! Packet ID: {packet_id}")
    print("📡 Broadcasting through all available transports...")

def send_security_alert(client: SonicWaveClient):
    """Send a security alert demo"""
    print("\n🚨 SECURITY ALERT DEMO")

    location = {
        'description': 'Parking Lot Section C',
        'building': 'Tech Conference Center',
        'coordinates': {'lat': 40.7125, 'lng': -74.0065}
    }

    packet_id = client.send_emergency(
        message="Suspicious individual attempting vehicle break-ins",
        location=location,
        emergency_type="security",
        metadata={'threat_level': 'medium', 'police_notified': False}
    )

    print(f"✅ Security alert sent! Packet ID: {packet_id}")
    print("📡 Broadcasting through all available transports...")

def send_custom_emergency(client: SonicWaveClient):
    """Send a custom emergency"""
    print("\n📝 CUSTOM EMERGENCY")

    message = input("Enter emergency message: ")
    emergency_type = input("Enter emergency type: ")
    location_desc = input("Enter location description: ")

    location = {
        'description': location_desc,
        'timestamp': time.time()
    }

    packet_id = client.send_emergency(message, location, emergency_type)
    print(f"✅ Custom emergency sent! Packet ID: {packet_id}")

def show_statistics(client: SonicWaveClient):
    """Show client statistics"""
    stats = client.get_stats()

    print("\n📊 CLIENT STATISTICS")
    print("=" * 30)
    print(f"Node ID: {stats['node_id']}")
    print(f"Status: {'🟢 Running' if stats['running'] else '🔴 Stopped'}")
    print(f"Active Transports: {', '.join(stats['active_transports'])}")
    print("\n📈 Packet Statistics:")
    print(f"  Sent: {stats['stats']['packets_sent']}")
    print(f"  Received: {stats['stats']['packets_received']}")
    print(f"  Relayed: {stats['stats']['packets_relayed']}")
    print(f"  Server ACKs: {stats['stats']['server_acknowledgments']}")
    print("\n🔄 Packet Manager:")
    print(f"  Processed: {stats['packet_manager']['processed_packets']}")
    print(f"  Pending ACKs: {stats['packet_manager']['pending_acks']}")

def discover_peers(client: SonicWaveClient):
    """Discover network peers"""
    print("\n🔍 DISCOVERING PEERS...")
    peers = client.discover_peers()

    if peers:
        print(f"✅ Found {len(peers)} peers:")
        for i, peer in enumerate(peers, 1):
            print(f"  {i}. {peer['type'].upper()}: {peer['id']}")
            if 'rssi' in peer:
                print(f"     Signal: {peer['rssi']} dBm")
    else:
        print("❌ No peers discovered")
        print("💡 Try running another client instance or check network connectivity")

if __name__ == '__main__':
    print("🚀 Starting SonicWave Demo...")
    print("💡 Make sure the server is running: python server.py")
    print("💡 For best demo, run multiple client instances")
    print("\nPress Ctrl+C at any time to exit")

    input("\nPress Enter to continue...")
    run_demo()