using System;
using System.Diagnostics;
using System.Threading;

namespace DevEnvBandwidthMonitor
{
    class Program
    {
        static void Main(string[] args)
        {
            Console.WriteLine("開(kāi)發(fā)環(huán)境帶寬監(jiān)控 - 墨夶出品");
            Console.WriteLine("正在初始化監(jiān)控...請(qǐng)稍等");
            
            // 1. 檢查性能計(jì)數(shù)器是否可用
            if (!PerformanceCounterCategory.Exists("Network Interface"))
            {
                Console.WriteLine("錯(cuò)誤：性能計(jì)數(shù)器'Network Interface'不可用。");
                Console.WriteLine("請(qǐng)確保已啟用網(wǎng)絡(luò)接口性能計(jì)數(shù)器。");
                Console.WriteLine("解決方案：運(yùn)行 'lodctr /R' 以重置性能計(jì)數(shù)器。");
                return;
            }
            
            // 2. 獲取所有網(wǎng)絡(luò)接口實(shí)例
            string[] interfaceNames = PerformanceCounterCategory.GetInstancesNames("Network Interface");
            
            // 3. 遍歷每個(gè)網(wǎng)絡(luò)接口，選擇第一個(gè)活動(dòng)接口
            string activeInterface = null;
            foreach (string interfaceName in interfaceNames)
            {
                if (NetworkInterface.GetByName(interfaceName).OperationalStatus == OperationalStatus.Up)
                {
                    activeInterface = interfaceName;
                    break;
                }
            }
            
            if (activeInterface == null)
            {
                Console.WriteLine("錯(cuò)誤：未找到活動(dòng)網(wǎng)絡(luò)接口。");
                return;
            }
            
            Console.WriteLine($"正在監(jiān)控活動(dòng)接口: {activeInterface}");
            
            // 4. 創(chuàng)建性能計(jì)數(shù)器
            PerformanceCounter sentCounter = new PerformanceCounter("Network Interface", "Bytes Sent/sec", activeInterface);
            PerformanceCounter receivedCounter = new PerformanceCounter("Network Interface", "Bytes Received/sec", activeInterface);
            
            // 5. 開(kāi)始監(jiān)控
            StartMonitoring(sentCounter, receivedCounter, activeInterface);
        }

        static void StartMonitoring(PerformanceCounter sentCounter, PerformanceCounter receivedCounter, string interfaceName)
        {
            // 1. 創(chuàng)建計(jì)時(shí)器，每秒更新一次
            Timer timer = new Timer(state =>
            {
                // 2. 獲取當(dāng)前帶寬
                double sentMbps = (sentCounter.NextValue() * 8) / 1000000.0;
                double receivedMbps = (receivedCounter.NextValue() * 8) / 1000000.0;
                
                // 3. 打印結(jié)果
                Console.WriteLine($"[{DateTime.Now:HH:mm:ss}] {interfaceName} | 發(fā)送: {sentMbps:F2} Mbps | 接收: {receivedMbps:F2} Mbps");
                
                // 4. 如果帶寬異常，發(fā)出警報(bào)
                if (sentMbps > 50 || receivedMbps > 50)
                {
                    Console.ForegroundColor = ConsoleColor.Red;
                    Console.WriteLine("警告：帶寬異常高！可能有流量風(fēng)暴！");
                    Console.ResetColor();
                }
                
            }, null, 0, 1000);
            
            Console.WriteLine("監(jiān)控已啟動(dòng)。按任意鍵停止...");
            Console.ReadKey();
        }
    }
}