import os
import re

from ping3 import ping

from AcrossPlatform.get_platform import GLOBAL_DIR
from LSZXNetWork.NetworkDriverLinux.wifi import DEFAULT_DEVICE, LAN_DEVICE
from LogRecord.log_recorder import GLOBAL_LOG

DEFAULT_IP_SEGMENT = r"192.168.[0-9]{1,3}.[0-9]{1,3}"
AP_CONF_DIR = os.path.join(GLOBAL_DIR, "LSZXNetWork/NetworkDriverLinux/start_wifi_ap.sh")
PATTERN_AP_CONF_DIR = os.path.join(GLOBAL_DIR, "LSZXNetWork/NetworkDriverLinux/pattern_start_wifi_ap.sh")
STOP_AP_CONF_DIR = os.path.join(GLOBAL_DIR, "LSZXNetWork/NetworkDriverLinux/stop_wifi_ap.sh")
PATTERN_STOP_AP_CONF_DIR = os.path.join(GLOBAL_DIR, "LSZXNetWork/NetworkDriverLinux/pattern_stop_wifi_ap.sh")
MASTER_IP = "192.168.137.1"
WLAN_DEVICE_TAG = "<wlan>"
LAN_DEVICE_TAG = "<lan>"
SSID_DEVICE_TAG = "<ssid>"
PASS_DEVICE_TAG = "<passphrase>"


# 获得同个网段下的设备
def get_machine_ip_address():
    echo = os.popen("arp -a | awk '{print $2}' | tr -d '()'").read()
    ip_list = re.findall(DEFAULT_IP_SEGMENT, echo)
    ip_list = list(filter(lambda x: not x.endswith("255") and not x.endswith("138.1"), ip_list))
    return ip_list


# 结束wifi
def stop_wifi_ap():
    with open(PATTERN_STOP_AP_CONF_DIR, "r") as file:
        conf = file.read()
    conf = conf.replace(WLAN_DEVICE_TAG, DEFAULT_DEVICE)
    conf = conf.replace(LAN_DEVICE_TAG, LAN_DEVICE)
    with open(STOP_AP_CONF_DIR, "w") as file:
        file.write(conf)
    os.popen(f"sudo bash {STOP_AP_CONF_DIR}").read()

def kill_port(port):
    os.popen(f"sudo fuser -k {port}/tcp").read()

def get_wifi_ap_status():
    try:
        if isinstance(ping("192.168.137.1", timeout=500, unit="ms"), float):
            return True
        else:
            return False
    except Exception as e:
        GLOBAL_LOG.write(f"AP状态获取出错，原因是：{e.args}。")
        return False


class WiFiAP:

    def __init__(self):
        stop_wifi_ap()
        self.ssid = None
        self.password = None

    def reboot(self):
        stop_wifi_ap()
        if self.ssid:
            self.set_ssid(ssid=self.ssid)
            self.set_password(password=self.password)
            self.start()

    def set_ssid(self, ssid):
        self.ssid = ssid

    def set_password(self, password):
        self.password = password

    def edit_ap_conf(self):
        with open(PATTERN_AP_CONF_DIR, "r") as file:
            conf = file.read()
        conf = conf.replace(WLAN_DEVICE_TAG, DEFAULT_DEVICE)
        conf = conf.replace(LAN_DEVICE_TAG, LAN_DEVICE)
        conf = conf.replace(SSID_DEVICE_TAG, self.ssid)
        conf = conf.replace(PASS_DEVICE_TAG, self.password)
        with open(AP_CONF_DIR, "w") as file:
            file.write(conf)

    @staticmethod
    def start_cmd():
        os.popen(f"sudo bash {AP_CONF_DIR}").read()

    def start(self):
        if self.ssid:
            self.edit_ap_conf()
            self.start_cmd()

    @staticmethod
    def stop():
        stop_wifi_ap()

    @staticmethod
    def kill_wifi():
        stop_wifi_ap()