#!/usr/bin/env python3
"""
System Tracker - Monitor machine health and performance
"""

import psutil
import time
import argparse
from datetime import datetime
from process_monitor import ProcessMonitor


class SystemTracker:
    def __init__(self):
        self.start_time = time.time()
        self.process_monitor = ProcessMonitor()

    def get_cpu_usage(self):
        """Get current CPU usage percentage"""
        return psutil.cpu_percent(interval=1, percpu=False)

    def get_memory_info(self):
        """Get memory usage statistics"""
        mem = psutil.virtual_memory()
        return {
            'total': mem.total,
            'available': mem.available,
            'percent': mem.percent,
            'used': mem.used
        }

    def get_disk_usage(self):
        """Get disk usage statistics"""
        disk = psutil.disk_usage('/')
        return {
            'total': disk.total,
            'used': disk.used,
            'free': disk.free,
            'percent': disk.percent
        }

    def get_network_stats(self):
        """Get network I/O statistics"""
        net = psutil.net_io_counters()
        return {
            'bytes_sent': net.bytes_sent,
            'bytes_recv': net.bytes_recv,
            'packets_sent': net.packets_sent,
            'packets_recv': net.packets_recv
        }

    def display_stats(self):
        """Display all system statistics"""
        print(f"\n{'='*50}")
        print(f"System Tracker - {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
        print(f"{'='*50}\n")

        print(f"CPU Usage: {self.get_cpu_usage()}%")

        mem = self.get_memory_info()
        print(f"Memory: {mem['percent']}% ({mem['used'] / (1024**3):.2f}GB / {mem['total'] / (1024**3):.2f}GB)")

        disk = self.get_disk_usage()
        print(f"Disk: {disk['percent']}% ({disk['used'] / (1024**3):.2f}GB / {disk['total'] / (1024**3):.2f}GB)")

        net = self.get_network_stats()
        print(f"Network: Sent {net['bytes_sent'] / (1024**2):.2f}MB | Recv {net['bytes_recv'] / (1024**2):.2f}MB")

        self.process_monitor.display_processes()


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='System Tracker - Monitor machine health')
    parser.add_argument('-c', '--continuous', action='store_true', help='Run continuously')
    parser.add_argument('-i', '--interval', type=int, default=5, help='Update interval in seconds')
    args = parser.parse_args()

    tracker = SystemTracker()

    if args.continuous:
        try:
            while True:
                tracker.display_stats()
                time.sleep(args.interval)
        except KeyboardInterrupt:
            print("\n\nTracker stopped by user")
    else:
        tracker.display_stats()