import re
import time

import serial
from serial import Serial

from utils.excel_reader import ExcelHelper


class ModemPool:
    BAUDRATE = 115200
    my_phone = "my_phone"
    phone_number_position = 10

    def __init__(self, port_list: list):
        self._port_list = port_list
        self._serial_list = []
        self._excel_helper = ExcelHelper()
        for port in self._port_list:
            ser = serial.Serial(port, self.BAUDRATE, timeout=1)
            self._serial_list.append(ser)

    def reset_all_modems(self):
        for ser in self._serial_list:
            self._send_command("AT+CFUN=1,1\r", ser)
            # send_command("AT+RESET\r", ser)
        #     wait for 20 second, so that the modem can init all the sims
        time.sleep(20)

    def get_raw_phone_number(self, slot_position):
        for index, ser in enumerate(self._serial_list):
            print("will get phone number for slot({}) SIM({}), port:{}".format(slot_position, index + 1, ser.port))
            self._select_sim_storage(ser)
            msg = self._execut_USSD_cmd("AT+CUSD=1, *132#\r", ser)
            if "Unfortunately" in str(msg):
                print("error for for slot({}) SIM({}), port:{}".format(slot_position, index + 1, ser.port))
                return
            #         find phone number
            match = re.search(r'33\d{9}', str(msg))
            phone_number = match.group(0)
            print("phone is " + phone_number)
            if phone_number:
                self._excel_helper.write_phone(phone_number)
            #         write the number to sim card's phonebook
            cmd = f'AT+CPBW={self.phone_number_position},\"{phone_number}\"\r'
            self._send_command(cmd, ser, wait_time_in_s=2)
            self.get_own_number(ser)

    def get_own_number(self, ser: Serial):
        print("saved phone number: " + str(self._send_command(f'AT+CPBR={self.phone_number_position}\r', ser)))

    def _select_sim_storage(self, ser):
        # use SIM Card storage
        cmd_sm = "AT+CPBS=\"SM\"\r"
        self._send_command(cmd_sm, ser)

    def _send_command(self, cmd: str, ser, wait_time_in_s: int = 0) -> bytes:
        print("send command {}".format(cmd))
        ser.write(cmd.encode())
        msg = ser.read(100)
        count = 0
        while 'OK' not in str(msg) and count < wait_time_in_s:
            time.sleep(1)
            count = count + 1
            msg = ser.read(100)
        # msg = ser.read(100)
        print(msg)
        return msg

    def _execut_USSD_cmd(self, cmd, ser) -> str:
        # the timeout for ussd command can be 120 s in mac
        return str(self._send_command(cmd, ser, 120))