import json
import git.exc
from git.repo import Repo
import os
import sys
import requests
from model.provider.provider import Provider
import json
import util.plugin_util as putil
import importlib
from pip._internal import main as pipmain

PLATFORM_QQCHAN = 'qqchan'
PLATFORM_GOCQ = 'gocq'

# 指令功能的基类，通用的（不区分语言模型）的指令就在这实现
class Command:
    def __init__(self, provider: Provider):
        self.provider = Provider

    def get_plugin_modules(self):
        plugins = []
        try:
            if os.path.exists("addons/plugins"):
                plugins = putil.get_modules("addons/plugins")
                return plugins
            elif os.path.exists("QQChannelChatGPT/addons/plugins"):
                plugins = putil.get_modules("QQChannelChatGPT/addons/plugins")
                return plugins
            else:
                return None
        except BaseException as e:
            raise e

    def check_command(self, message, role, platform, message_obj):
        # 插件
        try:
            plugins = self.get_plugin_modules()
            if plugins != None:
                # print(f"[DEBUG] 当前加载的插件：{plugins}")
                for p in plugins:
                    # print(f"[Debug] 当前缓存的插件：{self.cached_plugins}")
                    try:
                        if p in self.cached_plugins:
                            module = self.cached_plugins[p]["module"]
                            obj = self.cached_plugins[p]["clsobj"]
                        else:
                            module = __import__("addons.plugins." + p + "." + p, fromlist=[p])
                            cls = putil.get_classes(p, module)
                            obj = getattr(module, cls[0])()
                            self.cached_plugins[p] = {
                                "module": module,
                                "clsobj": obj
                            }
                        hit, res = obj.run(message, role, platform, message_obj)
                        if hit:
                            return True, res
                    except BaseException as e:
                        print(f"[Debug] 加载{p}插件出现问题，原因{str(e)}")
        except BaseException as e:
            print(f"[Debug] 插件加载出现问题，原因: {str(e)}\n已安装插件: {plugins}\n如果你没有相关装插件的想法, 请直接忽略此报错, 不影响其他功能的运行。")

        if self.command_start_with(message, "nick"):
            return True, self.set_nick(message, platform)
        
        if self.command_start_with(message, "plugin"):
            return True, self.plugin_oper(message, role)
        
        return False, None
    
    '''
    插件指令
    '''
    def plugin_oper(self, message: str, role: str):
        l = message.split(" ")
        if len(l) < 2:
            return True, "【插件指令】示例：\n安装插件: \nplugin i 插件Github地址\n卸载插件: \nplugin i 插件名 \n重载插件: \nplugin reload\n查看插件列表\nplugin l", "plugin"
        else:
            ppath = ""
            if os.path.exists("addons/plugins"):
                ppath = "addons/plugins"
            elif os.path.exists("QQChannelChatGPT/addons/plugins"):
                ppath = "QQChannelChatGPT/addons/plugins"
            else:
                return False, "未找到插件目录", "plugin"
            if l[1] == "i":
                if role != "admin":
                    return False, f"你的身份组{role}没有权限安装插件", "plugin"
                try:
                    # 得到url的最后一段
                    d = l[2].split("/")[-1]
                    # 创建文件夹
                    plugin_path = os.path.join(ppath, d)
                    os.mkdir(plugin_path)
                    Repo.clone_from(l[2],to_path=plugin_path,branch='master')

                    # 读取插件的requirements.txt
                    if os.path.exists(os.path.join(plugin_path, "requirements.txt")):
                        with open(os.path.join(plugin_path, "requirements.txt"), "r", encoding="utf-8") as f:
                            for line in f.readlines():
                                mm = os.system(f"pip3 install {line.strip()}")
                                if mm != 0:
                                    return False, "插件依赖安装失败，需要您手动pip安装对应插件的依赖。", "plugin"

                    return True, "插件拉取成功~", "plugin"
                except BaseException as e:
                    return False, f"拉取插件失败，原因: {str(e)}", "plugin"
            elif l[1] == "d":
                if role != "admin":
                    return False, f"你的身份组{role}没有权限删除插件", "plugin"
                try:
                    os.remove(os.path.join(ppath, l[2]))
                    if l[2] in self.cached_plugins:
                        del self.cached_plugins[l[2]]
                    return True, "插件卸载成功~", "plugin"
                except BaseException as e:
                    return False, f"卸载插件失败，原因: {str(e)}", "plugin"
            elif l[1] == "l":
                try:
                    return True, "已安装的插件: \n" + "\n".join(os.listdir(ppath)) + "\n使用plugin v 插件名 查看插件帮助（如果有的话）", "plugin"
                except BaseException as e:
                    return False, f"获取插件列表失败，原因: {str(e)}", "plugin"
            elif l[1] == "v":
                try:
                    if l[2] in os.listdir(ppath):
                        # 获取Readme
                        if os.path.exists(os.path.join(ppath, l[2], "README.md")):
                            with open(os.path.join(ppath, l[2], "README.md"), "r", encoding="utf-8") as f:
                                readme = f.read()
                        else:
                            readme = "暂无帮助（未找到此插件的README.md）"
                        return True, readme, "plugin"
                except BaseException as e:
                    return False, f"获取插件版本失败，原因: {str(e)}", "plugin"
            elif l[1] == "reload":
                if role != "admin":
                    return False, f"你的身份组{role}没有权限重载插件", "plugin"
                try:
                    for pm in self.cached_plugins:
                        module = self.cached_plugins[pm]["module"]
                        cls = putil.get_classes(pm, module)
                        obj = getattr(module, cls[0])()
                        self.cached_plugins[pm] = {
                            "module": module,
                            "clsobj": obj
                        }
                    return True, "插件重载成功！", "plugin"
                except BaseException as e:
                    return False, f"插件重载失败，原因: {str(e)}", "plugin"


    '''
    nick: 存储机器人的昵称
    '''
    def set_nick(self, message: str, platform: str):
        if platform == PLATFORM_GOCQ:
            l = message.split(" ")
            if len(l) == 1:
                return True, "【设置机器人昵称】示例：\n支持多昵称\nnick 昵称1 昵称2 昵称3", "nick"
            nick = l[1:]
            self.general_command_storer("nick_qq", nick)
            return True, f"设置成功！现在你可以叫我这些昵称来提问我啦~", "nick"
        elif platform == PLATFORM_QQCHAN:
            nick = message.split(" ")[2]
            return False, "QQ频道平台不支持为机器人设置昵称。", "nick"
    
    """
    存储指令结果到cmd_config.json
    """
    def general_command_storer(self, key, value):
        if not os.path.exists("cmd_config.json"):
            config = {}
        else:
            with open("cmd_config.json", "r", encoding="utf-8") as f:
                config = json.load(f)
        config[key] = value
        with open("cmd_config.json", "w", encoding="utf-8") as f:
            json.dump(config, f, indent=4, ensure_ascii=False)
            f.flush()

    
    def general_commands(self):
        return {
            "help": "帮助",
            "keyword": "设置关键词/关键指令回复",
            "update": "更新面板",
            "update latest": "更新到最新版本",
            "update r": "重启机器人",
            "reset": "重置会话",
            "nick": "设置机器人昵称",
            "/bing": "切换到bing模型",
            "/gpt": "切换到OpenAI ChatGPT API",
            "/revgpt": "切换到网页版ChatGPT",
            "/bing 问题": "临时使用一次bing模型进行会话",
            "/gpt 问题": "临时使用一次OpenAI ChatGPT API进行会话",
            "/revgpt 问题": "临时使用一次网页版ChatGPT进行会话",
            "plugin": "插件安装、卸载和重载"
        }
    
    def help_messager(self, commands: dict):
        try:
            resp = requests.get("https://soulter.top/channelbot/notice.json").text
            notice = json.loads(resp)["notice"]
        except BaseException as e:
            notice = ""
        msg = "Github项目名QQChannelChatGPT, 有问题提交issue, 欢迎Star\n【指令列表】\n"
        for key, value in commands.items():
            msg += key + ": " + value + "\n"
        msg += notice
        return msg
    
    # 接受可变参数
    def command_start_with(self, message: str, *args):
        for arg in args:
            if message.startswith(arg) or message.startswith('/'+arg):
                return True
        return False
    
    # keyword: 关键字
    def keyword(self, message: str, role: str):
        if role != "admin":
            return True, "你没有权限使用该指令", "keyword"

        l = message.split(" ")

        if len(l) < 3:
            return True, "【设置关键词回复】示例：\nkeyword hi 你好\n当发送hi的时候会回复你好\nkeyword /hi 你好\n当发送/hi时会回复你好\n删除关键词: keyword d hi\n删除hi关键词的回复", "keyword"
        
        del_mode = False
        if l[1] == "d":
            print("删除关键词: "+l[2])
            del_mode = True    

        try:
            if os.path.exists("keyword.json"):
                with open("keyword.json", "r", encoding="utf-8") as f:
                    keyword = json.load(f)
                    if del_mode:
                        # 删除关键词
                        if l[2] not in keyword:
                            return False, "该关键词不存在", "keyword"
                        else: del keyword[l[2]]
                    else:
                        keyword[l[1]] = l[2]
            else:
                if del_mode:
                    return False, "该关键词不存在", "keyword"
                keyword = {l[1]: l[2]}
            with open("keyword.json", "w", encoding="utf-8") as f:
                print("设置指令: "+l[1]+" -> "+l[2])
                json.dump(keyword, f, ensure_ascii=False, indent=4)
                f.flush()
            if del_mode:
                return True, "删除成功: "+l[2], "keyword"
            return True, "设置成功: "+l[1]+" -> "+l[2], "keyword"
        except BaseException as e:
            return False, "设置失败: "+str(e), "keyword"
    
    def update(self, message: str, role: str):
        if role != "admin":
            return True, "你没有权限使用该指令", "keyword"
        l = message.split(" ")
        if len(l) == 1:
            # 得到本地版本号和最新版本号
            try:
                repo = Repo()
            except git.exc.InvalidGitRepositoryError:
                repo = Repo(path="QQChannelChatGPT")
            now_commit = repo.head.commit

            # 得到远程3条commit列表, 包含commit信息
            origin = repo.remotes.origin
            origin.fetch()
            commits = list(repo.iter_commits('master', max_count=3))
            commits_log = ''
            index = 1
            for commit in commits:
                if commit.message.endswith("\n"):
                    commits_log += f"[{index}] {commit.message}-----------\n"
                else:
                    commits_log += f"[{index}] {commit.message}\n-----------\n"
                index+=1
            remote_commit_hash = origin.refs.master.commit.hexsha[:6]

            return True, f"当前版本: {now_commit.hexsha[:6]}\n最新版本: {remote_commit_hash}\n\n3条commit(非最新):\n{str(commits_log)}\n使用update latest更新至最新版本\n", "update"
        else:
            if l[1] == "latest":
                pash_tag = ""
                try:
                    try:
                        repo = Repo()
                    except git.exc.InvalidGitRepositoryError:
                        repo = Repo(path="QQChannelChatGPT")
                        pash_tag = "QQChannelChatGPT"+os.sep
                    repo.remotes.origin.pull()

                    return True, "更新成功~是否重启？输入update r重启（重启指令不返回任何确认信息）。", "update"
                    
                except BaseException as e:
                    return False, "更新失败: "+str(e), "update"
            if l[1] == "r":
                py = sys.executable
                os.execl(py, py, *sys.argv)


    def reset(self):
        return False
    
    def set(self):
        return False
    
    def unset(self):
        return False
    
    def key(self):
        return False
    
    def help(self):
        return False
    
    def status(self):
        return False
    
    def token(self):
        return False
    
    def his(self):
        return False
    
    def draw(self):
        return False