import os
import socket
import subprocess
import threading
from datetime import datetime
from sys import platform

import psutil
from speedtest import Speedtest

# 創(chuàng)建TCP套接字
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 綁定IP和端口
server_address = ('localhost', 8888)
server_socket.bind(server_address)

# 監(jiān)聽連接請求
server_socket.listen(5)


def handle_client(client_socket, client_address):
    while True:
        # 接收客戶端發(fā)送的數據
        data = client_socket.recv(1024)

        # 如果接收到空數據，則斷開連接
        if not data:
            client_socket.close()
            print(f"Connection with {client_address} closed")
            break

        # 處理接收到的數據
        request = data.decode('utf-8')
        if request == 'cpu':
            # 使用psutil庫獲取CPU使用率
            cpu_percent = psutil.cpu_percent(interval=1)
            response_data = f'CPU使用率：{cpu_percent}%'.encode('utf-8')
        elif request == 'memory':
            # 使用psutil庫獲取內存使用率
            memory_percent = psutil.virtual_memory().percent
            response_data = f'內存使用率：{memory_percent}%'.encode('utf-8')
        elif request == 'network':
            # 使用speedtest-cli庫獲取網絡帶寬信息
            st = Speedtest()
            download_speed = st.download()
            upload_speed = st.upload()
            response_data = f'下載速度：{download_speed / 1000000}Mbps，上傳速度：{upload_speed / 1000000}Mbps'.encode('utf-8')
        elif request == 'user':
            # 使用psutil庫獲取當前登錄用戶
            username = psutil.users()[0].name
            response_data = f'當前登錄用戶：{username}'.encode('utf-8')
        elif request == 'loadavg':
            # 使用os庫獲取系統(tǒng)負載情況
            load_avg = os.getloadavg()
            response_data = f'系統(tǒng)負載情況：{load_avg}'.encode('utf-8')
        elif request == 'time':
            # 使用datetime庫獲取當前時間
            current_time = datetime.datetime.now()
            response_data = f'當前時間：{current_time}'.encode('utf-8')
        elif request == 'process':
            # 使用psutil庫獲取進程列表
            process_list = []
            for process in psutil.process_iter(['pid', 'name', 'memory_info']):
                try:
                    process_list.append((process.info['pid'], process.info['name'], process.info['memory_info'].rss))
                except (psutil.AccessDenied, psutil.NoSuchProcess):
                    pass
            process_list.sort(key=lambda x: x[2], reverse=True)
            response_data = ''
            for i, (pid, name, memory) in enumerate(process_list[:10]):
                response_data += f'{i + 1}. 進程名稱：{name}，進程ID：{pid}，占用內存：{memory / 1024 / 1024:.2f}MB\n'
            response_data = response_data.encode('utf-8')
        elif request == 'system':
            # 使用platform庫獲取系統(tǒng)信息
            system_info = f'操作系統(tǒng)：{platform.system()} {platform.release()}\n處理器：{platform.processor()}\nPython版本：{platform.python_version()}'
            response_data = system_info.encode('utf-8')
        elif request == 'connection':
            # 使用psutil庫獲取網絡連接列表
            conn_list = []
            for conn in psutil.net_connections():
                if conn.status == psutil.CONN_ESTABLISHED:
                    conn_list.append((conn.laddr.ip, conn.laddr.port, conn.raddr.ip, conn.raddr.port, conn.pid))
            conn_list.sort(key=lambda x: x[4])
            response_data = ''
            for i, (laddr_ip, laddr_port, raddr_ip, raddr_port, pid) in enumerate(conn_list[:10]):
                response_data += f'{i + 1}. 本地地址：{laddr_ip}:{laddr_port}，遠程地址：{raddr_ip}:{raddr_port}，進程ID：{pid}\n'
            response_data = response_data.encode('utf-8')
        elif request == 'disk':
            # 使用psutil庫獲取磁盤使用情況
            disk_usage = psutil.disk_usage('/')
            disk_info = f'磁盤總容量：{disk_usage.total / 1024 / 1024 / 1024:.2f}GB，已用容量：{disk_usage.used / 1024 / 1024 / 1024:.2f}GB，可用容量：{disk_usage.free / 1024 / 1024 / 1024:.2f}GB'
            response_data = disk_info.encode('utf-8')
        elif request == 'load':
            # 使用psutil庫獲取系統(tǒng)負載
            load_avg = psutil.getloadavg()
            load_info = f'1分鐘內平均負載：{load_avg[0]:.2f}，5分鐘內平均負載：{load_avg[1]:.2f}，15分鐘內平均負載：{load_avg[2]:.2f}'
            response_data = load_info.encode('utf-8')
        elif request == 'thread':
            # 使用psutil庫獲取進程線程數
            thread_info = f'當前進程線程數：{psutil.Process().num_threads()}'
            response_data = thread_info.encode('utf-8')
        else:
            response_data = b'Invalid request'

        # 發(fā)送響應數據
        client_socket.sendall(response_data)


# 接收多個客戶端連接
while True:
    client_socket, client_address = server_socket.accept()
    print(f"New connection from {client_address}")
    # 創(chuàng)建新線程處理客戶端連接
    client_thread = threading.Thread(target=handle_client, args=(client_socket, client_address))
    client_thread.start()

import socket

# 創(chuàng)建TCP套接字
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 連接服務器
server_address = ('localhost', 8888)
client_socket.connect(server_address)

while True:
    # 發(fā)送請求數據給服務器
    request = input("請輸入要請求的數據類型（cpu/memory/network/user/loadavg/time/process/system/connection/disk/load/thread）：")
    client_socket.sendall(request.encode('utf-8'))

    # 接收服務器響應數據
    response_data = client_socket.recv(1024)

    # 處理接收到的數據
    response = response_data.decode('utf-8')
    print(response)

# 關閉客戶端套接字
client_socket.close()