import os
import sys
import subprocess
import socket
import argparse
import json
import time
from datetime import datetime
import threading
import requests
from urllib.parse import urlparse

class NetworkConnectivityChecker:
    def __init__(self):
        self.results = []
        self.errors = []
        self.monitoring = False
        
    def ping_host(self, host, count=4, timeout=3):
        """Ping測(cè)試"""
        try:
            # 根據(jù)操作系統(tǒng)選擇ping命令
            if os.name == 'nt':  # Windows
                cmd = ['ping', '-n', str(count), '-w', str(timeout * 1000), host]
            else:  # Linux/macOS
                cmd = ['ping', '-c', str(count), '-W', str(timeout), host]
                
            result = subprocess.run(cmd, capture_output=True, text=True, timeout=timeout * count + 5)
            
            if result.returncode == 0:
                # 解析ping輸出獲取統(tǒng)計(jì)信息
                output = result.stdout
                packet_loss = 0
                avg_rtt = 0
                
                if os.name == 'nt':  # Windows
                    # Windows ping輸出解析
                    for line in output.split('\n'):
                        if 'Lost' in line:
                            # 匹配 "Lost = X (X% loss)"
                            import re
                            match = re.search(r'\((\d+)% loss\)', line)
                            if match:
                                packet_loss = int(match.group(1))
                        elif 'Average' in line:
                            # 匹配 "Average = Xms"
                            import re
                            match = re.search(r'Average = (\d+)ms', line)
                            if match:
                                avg_rtt = int(match.group(1))
                else:  # Linux/macOS
                    # Unix-like系統(tǒng)ping輸出解析
                    for line in output.split('\n'):
                        if 'packet loss' in line:
                            # 匹配 "X packets transmitted, Y received, Z% packet loss"
                            import re
                            match = re.search(r'(\d+)% packet loss', line)
                            if match:
                                packet_loss = int(match.group(1))
                        elif 'rtt' in line:
                            # 匹配 "rtt min/avg/max/mdev = X/Y/Z/W ms"
                            import re
                            match = re.search(r'rtt [^=]+= [^/]+/([^/]+)/', line)
                            if match:
                                avg_rtt = float(match.group(1))
                                
                return {
                    'host': host,
                    'status': 'reachable',
                    'packet_loss': packet_loss,
                    'avg_rtt': avg_rtt,
                    'output': output
                }
            else:
                return {
                    'host': host,
                    'status': 'unreachable',
                    'error': result.stderr or 'Host unreachable',
                    'output': result.stdout
                }
                
        except subprocess.TimeoutExpired:
            return {
                'host': host,
                'status': 'timeout',
                'error': f'Ping timeout after {timeout * count + 5} seconds'
            }
        except Exception as e:
            return {
                'host': host,
                'status': 'error',
                'error': str(e)
            }
            
    def check_port(self, host, port, timeout=3):
        """端口檢查"""
        try:
            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            sock.settimeout(timeout)
            
            result = sock.connect_ex((host, port))
            sock.close()
            
            if result == 0:
                return {
                    'host': host,
                    'port': port,
                    'status': 'open'
                }
            else:
                return {
                    'host': host,
                    'port': port,
                    'status': 'closed',
                    'error': f'Connection refused (code: {result})'
                }
                
        except socket.gaierror as e:
            return {
                'host': host,
                'port': port,
                'status': 'dns_error',
                'error': f'DNS resolution failed: {e}'
            }
        except Exception as e:
            return {
                'host': host,
                'port': port,
                'status': 'error',
                'error': str(e)
            }
            
    def resolve_dns(self, hostname):
        """DNS解析測(cè)試"""
        try:
            ip_addresses = socket.getaddrinfo(hostname, None)
            ips = []
            for addr_info in ip_addresses:
                ip = addr_info[4][0]
                if ip not in ips:
                    ips.append(ip)
                    
            return {
                'hostname': hostname,
                'status': 'resolved',
                'ip_addresses': ips
            }
        except socket.gaierror as e:
            return {
                'hostname': hostname,
                'status': 'unresolved',
                'error': str(e)
            }
        except Exception as e:
            return {
                'hostname': hostname,
                'status': 'error',
                'error': str(e)
            }
            
    def http_check(self, url, timeout=10):
        """HTTP連通性測(cè)試"""
        try:
            parsed_url = urlparse(url)
            if not parsed_url.scheme:
                url = 'http://' + url
                
            response = requests.get(url, timeout=timeout)
            
            return {
                'url': url,
                'status': 'accessible',
                'status_code': response.status_code,
                'response_time': response.elapsed.total_seconds(),
                'content_length': len(response.content)
            }
        except requests.exceptions.Timeout:
            return {
                'url': url,
                'status': 'timeout',
                'error': f'Request timeout after {timeout} seconds'
            }
        except requests.exceptions.ConnectionError as e:
            return {
                'url': url,
                'status': 'connection_error',
                'error': str(e)
            }
        except requests.exceptions.RequestException as e:
            return {
                'url': url,
                'status': 'error',
                'error': str(e)
            }
        except Exception as e:
            return {
                'url': url,
                'status': 'error',
                'error': str(e)
            }
            
    def check_target(self, target):
        """檢查單個(gè)目標(biāo)"""
        result = {
            'target': target,
            'timestamp': datetime.now().isoformat(),
            'tests': {}
        }
        
        # 判斷目標(biāo)類型
        if target.startswith('http://') or target.startswith('https://'):
            # HTTP測(cè)試
            result['tests']['http'] = self.http_check(target)
        elif ':' in target and not target.replace(':', '').replace('.', '').isdigit():
            # 主機(jī)名:端口格式
            host, port = target.rsplit(':', 1)
            if port.isdigit():
                result['tests']['ping'] = self.ping_host(host)
                result['tests']['port'] = self.check_port(host, int(port))
                result['tests']['dns'] = self.resolve_dns(host)
            else:
                result['tests']['ping'] = self.ping_host(target)
                result['tests']['dns'] = self.resolve_dns(target)
        elif target.replace('.', '').isdigit() or not target.replace('-', '').replace('.', '').isdigit():
            # IP地址或主機(jī)名
            result['tests']['ping'] = self.ping_host(target)
            result['tests']['dns'] = self.resolve_dns(target)
        else:
            # 默認(rèn)進(jìn)行ping測(cè)試
            result['tests']['ping'] = self.ping_host(target)
            
        return result
        
    def check_multiple_targets(self, targets):
        """檢查多個(gè)目標(biāo)"""
        print(f"開始檢查 {len(targets)} 個(gè)目標(biāo)")
        print("=" * 60)
        
        self.results = []
        
        for target in targets:
            print(f"檢查目標(biāo): {target}")
            result = self.check_target(target)
            self.results.append(result)
            
            # 顯示結(jié)果
            tests = result['tests']
            for test_type, test_result in tests.items():
                status_icons = {
                    'reachable': '?',
                    'open': '?',
                    'resolved': '?',
                    'accessible': '?',
                    'unreachable': '?',
                    'closed': '?',
                    'unresolved': '?',
                    'timeout': '?',
                    'error': '??',
                    'connection_error': '?',
                    'dns_error': '?'
                }
                
                icon = status_icons.get(test_result['status'], '?')
                print(f"  {test_type.upper()}: {icon} {test_result['status']}")
                
                if 'error' in test_result:
                    print(f"    錯(cuò)誤: {test_result['error']}")
                elif test_type == 'ping':
                    if 'packet_loss' in test_result:
                        print(f"    丟包率: {test_result['packet_loss']}%")
                    if 'avg_rtt' in test_result and test_result['avg_rtt'] > 0:
                        print(f"    平均延遲: {test_result['avg_rtt']}ms")
                elif test_type == 'port':
                    print(f"    端口: {test_result['port']}")
                elif test_type == 'http':
                    if 'status_code' in test_result:
                        print(f"    狀態(tài)碼: {test_result['status_code']}")
                    if 'response_time' in test_result:
                        print(f"    響應(yīng)時(shí)間: {test_result['response_time']:.3f}s")
                elif test_type == 'dns':
                    if 'ip_addresses' in test_result:
                        print(f"    IP地址: {', '.join(test_result['ip_addresses'])}")
                        
            print()
            
        return self.results
        
    def generate_report(self):
        """生成檢查報(bào)告"""
        report = []
        report.append("=" * 80)
        report.append("網(wǎng)絡(luò)連通性檢查報(bào)告")
        report.append("=" * 80)
        report.append(f"檢查時(shí)間: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
        report.append(f"檢查目標(biāo)數(shù): {len(self.results)}")
        report.append("")
        
        # 統(tǒng)計(jì)信息
        stats = {
            'reachable': 0,
            'unreachable': 0,
            'timeout': 0,
            'error': 0
        }
        
        for result in self.results:
            for test_result in result['tests'].values():
                status = test_result['status']
                if status in stats:
                    stats[status] += 1
                elif status not in ['open', 'resolved', 'accessible']:
                    stats['error'] += 1
                    
        report.append("狀態(tài)統(tǒng)計(jì):")
        report.append(f"  可達(dá): {stats['reachable']}")
        report.append(f"  不可達(dá): {stats['unreachable']}")
        report.append(f"  超時(shí): {stats['timeout']}")
        report.append(f"  錯(cuò)誤: {stats['error']}")
        report.append("")
        
        # 詳細(xì)結(jié)果
        report.append("詳細(xì)檢查結(jié)果:")
        report.append("-" * 50)
        for result in self.results:
            report.append(f"目標(biāo): {result['target']}")
            for test_type, test_result in result['tests'].items():
                status_display = {
                    'reachable': '? 可達(dá)',
                    'open': '? 開放',
                    'resolved': '? 已解析',
                    'accessible': '? 可訪問',
                    'unreachable': '? 不可達(dá)',
                    'closed': '? 關(guān)閉',
                    'unresolved': '? 未解析',
                    'timeout': '? 超時(shí)',
                    'error': '?? 錯(cuò)誤',
                    'connection_error': '? 連接錯(cuò)誤',
                    'dns_error': '? DNS錯(cuò)誤'
                }
                
                status_text = status_display.get(test_result['status'], test_result['status'])
                report.append(f"  {test_type.upper()}: {status_text}")
                
                if 'error' in test_result:
                    report.append(f"    錯(cuò)誤: {test_result['error']}")
                elif test_type == 'ping':
                    if 'packet_loss' in test_result:
                        report.append(f"    丟包率: {test_result['packet_loss']}%")
                    if 'avg_rtt' in test_result and test_result['avg_rtt'] > 0:
                        report.append(f"    平均延遲: {test_result['avg_rtt']}ms")
                elif test_type == 'port':
                    report.append(f"    端口: {test_result['port']}")
                elif test_type == 'http':
                    if 'status_code' in test_result:
                        report.append(f"    狀態(tài)碼: {test_result['status_code']}")
                    if 'response_time' in test_result:
                        report.append(f"    響應(yīng)時(shí)間: {test_result['response_time']:.3f}s")
                elif test_type == 'dns':
                    if 'ip_addresses' in test_result:
                        report.append(f"    IP地址: {', '.join(test_result['ip_addresses'])}")
            report.append("")
            
        # 錯(cuò)誤信息
        if self.errors:
            report.append("錯(cuò)誤信息:")
            report.append("-" * 50)
            for error in self.errors[:10]:  # 只顯示前10個(gè)錯(cuò)誤
                report.append(f"  {error}")
            if len(self.errors) > 10:
                report.append(f"  ... 還有 {len(self.errors) - 10} 個(gè)錯(cuò)誤")
            report.append("")
            
        report.append("=" * 80)
        return "\n".join(report)
        
    def save_report(self, filename):
        """保存報(bào)告到文件"""
        try:
            report = self.generate_report()
            with open(filename, 'w', encoding='utf-8') as f:
                f.write(report)
            print(f"報(bào)告已保存到: {filename}")
            return True
        except Exception as e:
            print(f"保存報(bào)告時(shí)出錯(cuò): {e}")
            return False
            
    def save_json_report(self, filename):
        """保存JSON格式報(bào)告"""
        try:
            report_data = {
                'check_time': datetime.now().isoformat(),
                'results': self.results,
                'errors': self.errors
            }
            
            with open(filename, 'w', encoding='utf-8') as f:
                json.dump(report_data, f, indent=2, ensure_ascii=False)
            print(f"JSON報(bào)告已保存到: {filename}")
            return True
        except Exception as e:
            print(f"保存JSON報(bào)告時(shí)出錯(cuò): {e}")
            return False
            
    def continuous_monitor(self, targets, interval=60, max_checks=None):
        """持續(xù)監(jiān)控網(wǎng)絡(luò)連通性"""
        check_count = 0
        
        try:
            while True:
                print(f"\n[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] 開始第 {check_count + 1} 次檢查")
                self.check_multiple_targets(targets)
                
                # 顯示簡要狀態(tài)
                success_count = 0
                total_tests = 0
                
                for result in self.results:
                    for test_result in result['tests'].values():
                        total_tests += 1
                        if test_result['status'] in ['reachable', 'open', 'resolved', 'accessible']:
                            success_count += 1
                            
                print(f"成功測(cè)試: {success_count}/{total_tests}")
                
                check_count += 1
                
                # 檢查是否達(dá)到最大檢查次數(shù)
                if max_checks and check_count >= max_checks:
                    break
                    
                # 等待下次檢查
                time.sleep(interval)
                
        except KeyboardInterrupt:
            print("\n\n停止監(jiān)控...")

def main():
    parser = argparse.ArgumentParser(description="網(wǎng)絡(luò)連通性檢查工具")
    parser.add_argument("targets", nargs='*', help="要檢查的目標(biāo)（主機(jī)名、IP地址、URL等）")
    parser.add_argument("-f", "--file", help="從文件讀取目標(biāo)列表")
    parser.add_argument("-o", "--output", help="保存報(bào)告到文件")
    parser.add_argument("-j", "--json", help="保存JSON報(bào)告到文件")
    parser.add_argument("-m", "--monitor", type=int, help="持續(xù)監(jiān)控間隔（秒）")
    parser.add_argument("--max-checks", type=int, help="最大檢查次數(shù)")
    parser.add_argument("--timeout", type=int, default=3, help="網(wǎng)絡(luò)操作超時(shí)時(shí)間（秒，默認(rèn)：3）")
    
    args = parser.parse_args()
    
    # 檢查目標(biāo)
    targets = []
    if args.targets:
        targets.extend(args.targets)
    if args.file:
        try:
            with open(args.file, 'r', encoding='utf-8') as f:
                file_targets = [line.strip() for line in f if line.strip()]
                targets.extend(file_targets)
        except Exception as e:
            print(f"讀取目標(biāo)文件時(shí)出錯(cuò): {e}")
            sys.exit(1)
            
    if not targets:
        print("請(qǐng)指定要檢查的目標(biāo)")
        parser.print_help()
        sys.exit(1)
        
    # 檢查requests庫
    try:
        import requests
    except ImportError:
        print("警告: 缺少requests庫，HTTP測(cè)試功能將不可用")
        print("安裝命令: pip install requests")
        
    try:
        checker = NetworkConnectivityChecker()
        
        # 檢查網(wǎng)絡(luò)連通性
        if args.monitor:
            # 持續(xù)監(jiān)控模式
            checker.continuous_monitor(targets, args.monitor, args.max_checks)
        else:
            # 單次檢查模式
            checker.check_multiple_targets(targets)
            
            # 生成報(bào)告
            if args.output:
                checker.save_report(args.output)
            elif args.json:
                checker.save_json_report(args.json)
            else:
                print(checker.generate_report())
                
    except KeyboardInterrupt:
        print("\n\n用戶中斷操作")
    except Exception as e:
        print(f"程序執(zhí)行出錯(cuò): {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()

# HTTP測(cè)試需要
pip install requests

# 檢查單個(gè)目標(biāo)
python network_checker.py google.com

# 檢查多個(gè)目標(biāo)
python network_checker.py google.com github.com 8.8.8.8

# 檢查特定端口
python network_checker.py google.com:443 github.com:22

# 檢查HTTP服務(wù)
python network_checker.py https://www.google.com http://github.com

# 從文件讀取目標(biāo)列表
python network_checker.py -f targets.txt

# 保存報(bào)告到文件
python network_checker.py google.com github.com -o network_report.txt

# 保存JSON報(bào)告
python network_checker.py google.com github.com -j network_report.json

# 持續(xù)監(jiān)控（每30秒檢查一次）
python network_checker.py google.com github.com -m 30

# 持續(xù)監(jiān)控最多10次
python network_checker.py google.com github.com -m 30 --max-checks 10

python network_checker.py google.com github.com 8.8.8.8

python network_checker.py google.com github.com -m 60 --max-checks 10 -o status.log

python network_checker.py -f targets.txt -o full_report.txt -j report.json