# encoding: utf-8
# @author: 原凯峰
# @contact: 2894340009@qq.com
# @software: pycharm
# @file: Connector.py
# @time: 2024/6/19 9:20
# @desc:

import socket
import threading
import sys
from http.server import HTTPServer, BaseHTTPRequestHandler

from Server import *

# 全局变量
sockets = {}
socket_lock = threading.Lock()
exit_flag = False

# 维护连接的函数
def KeepConnection(sock, ip, port):
    """
    @author: 原凯峰
    @param sock:
    @param ip:
    @param port:
    @return:
    """
    while not exit_flag:
        data = sock.recv(1024)
        if not data:
            print(f"Gateway disconnected from {ip}:{port}")
            break
        print(f"Received from gateway ({ip}:{port}): {data.decode()}")

    with socket_lock:
        del sockets[sock]

# 连接到网关的函数
def ConnectToGateway(ip, port):
    """
    @author: 原凯峰
    @param ip:
    @param port:
    @return:
    """
    global exit_flag
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.connect((ip, port))

    with socket_lock:
        sockets[sock] = (ip, port)

    print(f"Connected to gateway at {ip}:{port}")

    threading.Thread(target=KeepConnection, args=(sock, ip, port)).start()


def StartHttpServer():
    """
    @author: 原凯峰
    @return:
    """
    global exit_flag
    server_address = ('', 8080)
    httpd = HTTPServer(server_address, CORSHTTPRequestHandler)
    print(f'Starting httpd server on port 8080...')
    try:
        while not exit_flag:
            httpd.handle_request()  # 使用 handle_request 替代 serve_forever 以便可以检查 exit_flag
    except KeyboardInterrupt:
        pass
    finally:
        httpd.server_close()
        print("HTTP server stopped")


# 发送消息到特定网关的函数
def SendMessageToGateway(ip, port, message):
    """
    @author: 原凯峰
    @param ip:
    @param port:
    @param message:
    @return:
    """
    with socket_lock:
        # 查找特定的网关连接
        for sock, (stored_ip, stored_port) in sockets.items():
            if stored_ip == ip and stored_port == port:
                try:
                    # 发送消息，这里假设消息是字符串格式
                    sock.sendall(message.encode('utf-8'))
                    print(f"Message sent to {ip}:{port}: {message}")
                except socket.error as e:
                    print(f"Error sending message to {ip}:{port}: {e}")
                return  # 找到并发送消息后退出函数

    # 如果没有找到连接
    print(f"No connection found for {ip}:{port}")

# 示例：发送消息到IP地址和端口
# 假设我们已经连接到了这个网关


def main():
    global exit_flag
    exit_flag = False  # 初始化exit_flag
    t = threading.Thread(target=StartHttpServer)
    t.start()
    while not exit_flag:
        command = input("Enter command (/connect_gateway/exit): ")
        # if command == "start_http_server":
        #     t = threading.Thread(target=StartHttpServer)
        #     t.start()
        if command == "connect_gateway":
            ip = input("Enter the gateway IP address: ")
            port = int(input("Enter the gateway port: "))
            ConnectToGateway(ip, port)
        elif command == "exit":
            exit_flag = True
            print("Exiting program...")
            break
        else:
            print("Unknown command")

# StartHttpServer函数修改

if __name__ == "__main__":
    main()