import menu_utils as utils
import json
from mqtt_device import class_comm_mqtt_interface, class_comm_mqtt_thread
from device_conf import class_comm_device_config
import string
from enum import Enum
import math
from print_color import *

#创建mqtt 消息属性字典, 
class class_mqtt_info_object (class_comm_mqtt_interface):
    def __init__(self, device_name):
        class_comm_mqtt_interface.__init__(self)
        self.mqtt_dict = {}

        device_config_info = __import__(device_name)

        self.device_config : class_comm_device_config = device_config_info.comm_device_config
        self.menu_description = self.device_config.get_menu_caption_info() 

        menu_process_list = []
        for item_dict in self.menu_description :
            if "menu" in item_dict.keys() :
                menu_object = item_dict["menu"]
                self.__create_mqtt_dict__(menu_object, menu_process_list)

    #创建mqtt索引, 用于定位mqtt消息相关的菜单属性, 内部函数, 外部不要调用
    def __create_mqtt_dict__(self, menu_object, menu_process_list: list): #创建菜单对应的各项mqtt信息，包含菜单内部的子菜单的各项mqtt信息
        if menu_object == None :
            return
        if menu_object in menu_process_list :              #防止递归进入无限循环
            return 

        menu_process_list.append(menu_object)              #添加到处理队列中，可以防止menu_object重复处理

        for menu_item in menu_object :                          #遍历菜单内部的各个菜单项
            if "mqtt" in menu_item.keys() :                     #该项菜单具有mqtt属性
                mqtt_info_name = menu_item["mqtt"]              #获取mqtt的信息名称， mqtt的消息中包含该信息时，可快速定位该 菜单项
                if mqtt_info_name not in self.mqtt_dict.keys() :  #mqtt信息需要具有唯一性, 
                    self.mqtt_dict[mqtt_info_name] = menu_item  #把mqtt信息与 菜单项 进行(key, value)绑定. key = mqtt信息, value为菜单项
                else :
                    print_error_msg("Error, 菜单项有相同的mqtt字段(%s)"%(mqtt_info_name))

            if "sub_menu" in menu_item.keys() :
                sub_menu_name = menu_item["sub_menu"]
                if isinstance(sub_menu_name, Enum):
                    sub_menu_name = sub_menu_name.name
 
                sub_menu_object = self.search_menu_group_object(sub_menu_name)
                self.__create_mqtt_dict__(sub_menu_object, menu_process_list)    #递归调用, 直到处理完所有子菜单项

    #通过菜单名搜索到菜单项
    def search_menu_item(self, mqtt_info_name : string) :
        if mqtt_info_name in self.mqtt_dict.keys() :
            return self.mqtt_dict[mqtt_info_name]
        else :
            return None

    #通过菜单名字来查找菜单组对象, 每个菜单组包含若干个菜单项
    def search_menu_group_object(self, menu_group_name) :
        menu_group_object = None
        for item_dict in self.menu_description:
            menu_item_name = utils.dict_or_object_get_attr(item_dict, "name", None)
            if menu_item_name == menu_group_name :
                menu_group_object = utils.dict_or_object_get_attr(item_dict, "menu", None)
                break
        return menu_group_object
 

#定义mqtt信息组：所有设备的mqtt信息 都集中于该类
class class_mqtt_info_object_group():
    def __init__(self):
        self.device_mqtt_info_dict : class_mqtt_info_object = {}
    
    def create_mqtt_info_object(self, device_name) ->class_mqtt_info_object:
        try :
            mqtt_info_object = None 
            
            if device_name not in self.device_mqtt_info_dict.keys() :
                new_mqtt_info = class_mqtt_info_object(device_name)
                self.device_mqtt_info_dict[device_name] = new_mqtt_info
            mqtt_info_object = self.get_mqtt_info_object(device_name)
        except Exception as e:
            print_error_msg(str(e))
            mqtt_info_object = None
        finally :
            return mqtt_info_object
    
    def get_mqtt_info_object(self, device_name) ->class_mqtt_info_object:
        matched_mqtt_info = None
        if device_name in self.device_mqtt_info_dict.keys() :
            matched_mqtt_info = self.device_mqtt_info_dict[device_name]
        
        return matched_mqtt_info

    def search_mqtt_menu_item_info(self, device_name, mqtt_info_name) :
        if mqtt_info_name in self.device_mqtt_info_dict.keys() :
            mqtt_info_object : class_mqtt_info_object = self.device_mqtt_info_dict[device_name]
            menu_item =  mqtt_info_object.search_menu_item(mqtt_info_name)
            return menu_item
        return None