import logging
from model.AiCat import AiCat
from model.Clear import Clear
import toml

class private:
   def __init__(self, msg):
       self.user_id = msg.user_id
       self.message_id = msg.message_id
       self.message_type = msg.message_type
       self.raw_message = msg.raw_message
       self.sender = msg.sender
       self.message = msg.message
       self.self_id = msg.self_id
       self.time = msg.time

   def main(self):
       texts = [seg['data']['text'].strip() for seg in self.message if seg['type'] == 'text']
       full_text = ' '.join(texts).strip()
       permission = self.check_permission()
       if permission is None:
           return "服务器繁忙，请稍后再逝"
       elif permission:
           return self.menu(full_text)
       else:
           return "此bot未在该群启用"

   def check_permission(self):
       try:
           with open("./config.toml", "r", encoding="utf-8") as f:
               config = toml.load(f)
               allowed_users = config.get("allowed_users", [])
       except Exception as e:
           logging.error(str(e))
           return None

           # 检查当前群是否在允许列表中
       if allowed_users == "all":
           return True
       elif self.user_id in allowed_users:
           return True
       else:
           return False

   def menu(self,command):
        if command.startswith("/help"):
            return " 直接输入聊天内容即可 \n /help -- 获取帮助 \n /clear [群号 / private:Q号]  (all 为全部，不填为本 群/用户)  \n /status -- 查看bot状态 "
        elif command.startswith(" /cat"):#  留此指令接口为了方便通过审核
            # 排除 " /cat "" /cat"未传参情况
            parts = command.split(" /cat ", 1)
            if len(parts) > 1:
                chat_content = parts[1].strip()
                if chat_content:
                    cat = AiCat(chat_content,self.user_id , f"private:{self.user_id}")
                    answer = AiCat.main(cat)
                    return answer
                else:
                    return "你似乎没有提供想和我聊的内容喵~  \n 格式：/cat <提问内容>"
            else:
                return "你似乎没有提供想和我聊的内容喵~  \n 格式：/cat <提问内容>"
        elif command.startswith("/clear"):
            parts = command.split("/clear ", 1)
            if len(parts) > 1:
                group_id = parts[1].strip()
                if group_id:
                    clear = Clear(self.user_id,group_id)
                    return clear.main()
                else:
                    clear = Clear(self.user_id, f"private:{self.user_id}")
                    return  clear.main()
            else:
                clear = Clear(self.user_id, f"private:{self.user_id}")
                return clear.main()
        elif command.startswith(" /status"):
            return "---猫娘 QBOT---\n Q bot 运行正常 \n 版本: 2.0 pre \n © 融玩文化 | 无尽创意MCUNC"
        elif command.startswith(" /"):
            return "指令不存在，输入/help查看帮助"
        elif command == "":
            return "你似乎没有提供想和我聊的内容喵~  \n 直接输入聊天内容即可"
        elif command is None:
            return "你似乎没有提供想和我聊的内容喵~  \n 直接输入聊天内容即可"
        else:
            cat = AiCat(command, self.user_id,f"private:{self.user_id}")
            answer = AiCat.main(cat)
            return answer