#!/usr/bin/env python3
"""
REChain Benchmark Runner
Version: 4.1.10+1160
"""

import argparse
import sys
import time
from pathlib import Path

class BenchmarkRunner:
    def __init__(self):
        self.results_dir = Path(__file__).parent / "results"
        self.results_dir.mkdir(exist_ok=True)
    
    def run_all(self):
        """Run all benchmark suites"""
        suites = [
            ("Application", self.run_application),
            ("Matrix", self.run_matrix),
            ("Blockchain", self.run_blockchain),
            ("IPFS", self.run_ipfs),
            ("AI", self.run_ai),
        ]
        
        for name, suite_func in suites:
            print(f"\n{'='*60}")
            print(f"Running {name} Benchmarks")
            print(f"{'='*60}")
            start = time.time()
            suite_func()
            elapsed = time.time() - start
            print(f"\n{name} benchmarks completed in {elapsed:.2f}s")
    
    def run_application(self):
        """Run application benchmarks"""
        from scripts.benchmark_startup import BenchmarkStartup
        from scripts.benchmark_memory import BenchmarkMemory
        
        BenchmarkStartup().run()
        BenchmarkMemory().run()
    
    def run_matrix(self):
        """Run Matrix protocol benchmarks"""
        from matrix.benchmark_sync import SyncBenchmark
        from matrix.benchmark_messages import MessageBenchmark
        
        SyncBenchmark().run()
        MessageBenchmark().run()
    
    def run_blockchain(self):
        """Run blockchain benchmarks"""
        from blockchain.benchmark_wallet import WalletBenchmark
        from blockchain.benchmark_transactions import TransactionBenchmark
        
        WalletBenchmark().run()
        TransactionBenchmark().run()
    
    def run_ipfs(self):
        """Run IPFS benchmarks"""
        from ipfs.benchmark_upload import UploadBenchmark
        from ipfs.benchmark_download import DownloadBenchmark
        
        UploadBenchmark().run()
        DownloadBenchmark().run()
    
    def run_ai(self):
        """Run AI services benchmarks"""
        from ai.benchmark_services import AIBenchmark
        
        AIBenchmark().run()

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="REChain Benchmark Runner")
    parser.add_argument("--all", action="store_true", help="Run all benchmarks")
    parser.add_argument("--application", action="store_true", help="Run application benchmarks")
    parser.add_argument("--matrix", action="store_true", help="Run Matrix benchmarks")
    parser.add_argument("--blockchain", action="store_true", help="Run blockchain benchmarks")
    parser.add_argument("--ipfs", action="store_true", help="Run IPFS benchmarks")
    parser.add_argument("--ai", action="store_true", help="Run AI benchmarks")
    
    args = parser.parse_args()
    
    runner = BenchmarkRunner()
    
    if args.all or not any([args.application, args.matrix, 
                           args.blockchain, args.ipfs, args.ai]):
        runner.run_all()
    else:
        if args.application:
            runner.run_application()
        if args.matrix:
            runner.run_matrix()
        if args.blockchain:
            runner.run_blockchain()
        if args.ipfs:
            runner.run_ipfs()
        if args.ai:
            runner.run_ai()