J1147
/
ai-on-wechat
Bevaka 1
Stjärnmärk 0
Förgrening 0
Kod Ärenden 0 Pull-förfrågningar 0 Släpp 0 Wiki Aktiviteter
Du kan inte välja fler än 25 ämnen Ämnen måste starta med en bokstav eller siffra, kan innehålla bindestreck ('-') och vara max 35 tecken långa.
427 Incheckningar
2 Grenar
152MB
Träd: 8b73a74609
Grenar Taggar
${ item.name }
Skapa branchen ${ searchTerm }
från '8b73a74609'
${ noResults }
ai-on-wechat/plugins/plugin_manager.py

264 lines
11KB
Rå Blame Historik 

  1. # encoding:utf-8

  2. 

  3. import importlib

  4. import importlib.util

  5. import json

  6. import os

  7. import sys

  8. from common.singleton import singleton

  9. from common.sorted_dict import SortedDict

  10. from .event import *

  11. from common.log import logger

  12. from config import conf

  13. 

  14. 

  15. @singleton

  16. class PluginManager:

  17.     def __init__(self):

  18.         self.plugins = SortedDict(lambda k,v: v.priority,reverse=True)

  19.         self.listening_plugins = {}

  20.         self.instances = {}

  21.         self.pconf = {}

  22.         self.current_plugin_path = None

  23.         self.loaded = {}

  24. 

  25.     def register(self, name: str, desire_priority: int = 0, **kwargs):

  26.         def wrapper(plugincls):

  27.             plugincls.name = name

  28.             plugincls.priority = desire_priority

  29.             plugincls.desc = kwargs.get('desc')

  30.             plugincls.author = kwargs.get('author')

  31.             plugincls.path = self.current_plugin_path

  32.             plugincls.version = kwargs.get('version') if kwargs.get('version') != None else "1.0"

  33.             plugincls.namecn = kwargs.get('namecn') if kwargs.get('namecn') != None else name

  34.             plugincls.hidden = kwargs.get('hidden') if kwargs.get('hidden') != None else False

  35.             plugincls.enabled = True

  36.             if self.current_plugin_path == None:

  37.                 raise Exception("Plugin path not set")

  38.             self.plugins[name.upper()] = plugincls

  39.             logger.info("Plugin %s_v%s registered, path=%s" % (name, plugincls.version, plugincls.path))

  40.         return wrapper

  41. 

  42.     def save_config(self):

  43.         with open("./plugins/plugins.json", "w", encoding="utf-8") as f:

  44.             json.dump(self.pconf, f, indent=4, ensure_ascii=False)

  45. 

  46.     def load_config(self):

  47.         logger.info("Loading plugins config...")

  48. 

  49.         modified = False

  50.         if os.path.exists("./plugins/plugins.json"):

  51.             with open("./plugins/plugins.json", "r", encoding="utf-8") as f:

  52.                 pconf = json.load(f)

  53.                 pconf['plugins'] = SortedDict(lambda k,v: v["priority"],pconf['plugins'],reverse=True)

  54.         else:

  55.             modified = True

  56.             pconf = {"plugins": SortedDict(lambda k,v: v["priority"],reverse=True)}

  57.         self.pconf = pconf

  58.         if modified:

  59.             self.save_config()

  60.         return pconf

  61. 

  62.     def scan_plugins(self):

  63.         logger.info("Scaning plugins ...")

  64.         plugins_dir = "./plugins"

  65.         raws = [self.plugins[name] for name in self.plugins]

  66.         for plugin_name in os.listdir(plugins_dir):

  67.             plugin_path = os.path.join(plugins_dir, plugin_name)

  68.             if os.path.isdir(plugin_path):

  69.                 # 判断插件是否包含同名__init__.py文件

  70.                 main_module_path = os.path.join(plugin_path,"__init__.py")

  71.                 if os.path.isfile(main_module_path):

  72.                     # 导入插件

  73.                     import_path = "plugins.{}".format(plugin_name)

  74.                     try:

  75.                         self.current_plugin_path = plugin_path

  76.                         if plugin_path in self.loaded:

  77.                             if self.loaded[plugin_path] == None:

  78.                                 logger.info("reload module %s" % plugin_name)

  79.                                 self.loaded[plugin_path] = importlib.reload(sys.modules[import_path])

  80.                                 dependent_module_names = [name for name in sys.modules.keys() if name.startswith( import_path+ '.')]

  81.                                 for name in dependent_module_names:

  82.                                     logger.info("reload module %s" % name)

  83.                                     importlib.reload(sys.modules[name])

  84.                         else:

  85.                             self.loaded[plugin_path] = importlib.import_module(import_path)

  86.                         self.current_plugin_path = None

  87.                     except Exception as e:

  88.                         logger.exception("Failed to import plugin %s: %s" % (plugin_name, e))

  89.                         continue

  90.         pconf = self.pconf

  91.         news = [self.plugins[name] for name in self.plugins]

  92.         new_plugins = list(set(news) - set(raws))

  93.         modified = False

  94.         for name, plugincls in self.plugins.items():

  95.             rawname = plugincls.name

  96.             if rawname not in pconf["plugins"]:

  97.                 modified = True

  98.                 logger.info("Plugin %s not found in pconfig, adding to pconfig..." % name)

  99.                 pconf["plugins"][rawname] = {"enabled": plugincls.enabled, "priority": plugincls.priority}

  100.             else:

  101.                 self.plugins[name].enabled = pconf["plugins"][rawname]["enabled"]

  102.                 self.plugins[name].priority = pconf["plugins"][rawname]["priority"]

  103.                 self.plugins._update_heap(name) # 更新下plugins中的顺序

  104.         if modified:

  105.             self.save_config()

  106.         return new_plugins

  107. 

  108.     def refresh_order(self):

  109.         for event in self.listening_plugins.keys():

  110.             self.listening_plugins[event].sort(key=lambda name: self.plugins[name].priority, reverse=True)

  111. 

  112.     def activate_plugins(self): # 生成新开启的插件实例

  113.         failed_plugins = []

  114.         for name, plugincls in self.plugins.items():

  115.             if plugincls.enabled:

  116.                 if name not in self.instances:

  117.                     try:

  118.                         instance = plugincls()

  119.                     except Exception as e:

  120.                         logger.warn("Failed to init %s, diabled. %s" % (name, e))

  121.                         self.disable_plugin(name)

  122.                         failed_plugins.append(name)

  123.                         continue

  124.                     self.instances[name] = instance

  125.                     for event in instance.handlers:

  126.                         if event not in self.listening_plugins:

  127.                             self.listening_plugins[event] = []

  128.                         self.listening_plugins[event].append(name)

  129.         self.refresh_order()

  130.         return failed_plugins

  131. 

  132.     def reload_plugin(self, name:str):

  133.         name = name.upper()

  134.         if name in self.instances:

  135.             for event in self.listening_plugins:

  136.                 if name in self.listening_plugins[event]:

  137.                     self.listening_plugins[event].remove(name)

  138.             del self.instances[name]

  139.             self.activate_plugins()

  140.             return True

  141.         return False

  142.     

  143.     def load_plugins(self):

  144.         self.load_config()

  145.         self.scan_plugins()

  146.         pconf = self.pconf

  147.         logger.debug("plugins.json config={}".format(pconf))

  148.         for name,plugin in pconf["plugins"].items():

  149.             if name.upper() not in self.plugins:

  150.                 logger.error("Plugin %s not found, but found in plugins.json" % name)

  151.         self.activate_plugins()

  152. 

  153.     def emit_event(self, e_context: EventContext, *args, **kwargs):

  154.         if e_context.event in self.listening_plugins:

  155.             for name in self.listening_plugins[e_context.event]:

  156.                 if self.plugins[name].enabled and e_context.action == EventAction.CONTINUE:

  157.                     logger.debug("Plugin %s triggered by event %s" % (name,e_context.event))

  158.                     instance = self.instances[name]

  159.                     instance.handlers[e_context.event](e_context, *args, **kwargs)

  160.         return e_context

  161. 

  162.     def set_plugin_priority(self, name:str, priority:int):

  163.         name = name.upper()

  164.         if name not in self.plugins:

  165.             return False

  166.         if self.plugins[name].priority == priority:

  167.             return True

  168.         self.plugins[name].priority = priority

  169.         self.plugins._update_heap(name)

  170.         rawname = self.plugins[name].name

  171.         self.pconf["plugins"][rawname]["priority"] = priority

  172.         self.pconf["plugins"]._update_heap(rawname)

  173.         self.save_config()

  174.         self.refresh_order()

  175.         return True

  176. 

  177.     def enable_plugin(self, name:str):

  178.         name = name.upper()

  179.         if name not in self.plugins:

  180.             return False, "插件不存在"

  181.         if not self.plugins[name].enabled :

  182.             self.plugins[name].enabled = True

  183.             rawname = self.plugins[name].name

  184.             self.pconf["plugins"][rawname]["enabled"] = True

  185.             self.save_config()

  186.             failed_plugins = self.activate_plugins()

  187.             if name in failed_plugins:

  188.                 return False, "插件开启失败"

  189.             return True, "插件已开启"

  190.         return True, "插件已开启"

  191.     

  192.     def disable_plugin(self, name:str):

  193.         name = name.upper()

  194.         if name not in self.plugins:

  195.             return False

  196.         if self.plugins[name].enabled :

  197.             self.plugins[name].enabled = False

  198.             rawname = self.plugins[name].name

  199.             self.pconf["plugins"][rawname]["enabled"] = False

  200.             self.save_config()

  201.             return True

  202.         return True

  203.     

  204.     def list_plugins(self):

  205.         return self.plugins

  206.     

  207.     def install_plugin(self, repo:str):

  208.         try:

  209.             import common.package_manager as pkgmgr

  210.             pkgmgr.check_dulwich()

  211.         except Exception as e:

  212.             logger.error("Failed to install plugin, {}".format(e))

  213.             return False, "无法导入dulwich，安装插件失败"

  214.         import re

  215.         from dulwich import porcelain

  216. 

  217.         logger.info("clone git repo: {}".format(repo))

  218.         

  219.         match = re.match(r"^(https?:\/\/|git@)([^\/:]+)[\/:]([^\/:]+)\/(.+).git$", repo)

  220.         

  221.         if not match:

  222.             try:

  223.                 with open("./plugins/source.json","r") as f:

  224.                     source = json.load(f)

  225.                 if repo in source["repo"]:

  226.                     repo = source["repo"][repo]["url"]

  227.                     match = re.match(r"^(https?:\/\/|git@)([^\/:]+)[\/:]([^\/:]+)\/(.+).git$", repo)

  228.                     if not match:

  229.                         return False, "安装插件失败，source中的仓库地址不合法"

  230.                 else:

  231.                     return False, "安装插件失败，仓库地址不合法"

  232.             except Exception as e:

  233.                 logger.error("Failed to install plugin, {}".format(e))

  234.                 return False, "安装插件失败，请检查仓库地址是否正确"

  235.         dirname = os.path.join("./plugins",match.group(4))

  236.         try:

  237.             repo = porcelain.clone(repo, dirname, checkout=True)

  238.             if os.path.exists(os.path.join(dirname,"requirements.txt")):

  239.                 logger.info("detect requirements.txt，installing...")

  240.             pkgmgr.install_requirements(os.path.join(dirname,"requirements.txt"))

  241.             return True, "安装插件成功，请扫描插件或重启程序"

  242.         except Exception as e:

  243.             logger.error("Failed to install plugin, {}".format(e))

  244.             return False, "安装插件失败，"+str(e)

  245.         

  246.     def uninstall_plugin(self, name:str):

  247.         name = name.upper()

  248.         if name not in self.plugins:

  249.             return False, "插件不存在"

  250.         if name in self.instances:

  251.             self.disable_plugin(name)

  252.         dirname = self.plugins[name].path

  253.         try:

  254.             import shutil

  255.             shutil.rmtree(dirname)

  256.             rawname = self.plugins[name].name

  257.             del self.plugins[name]

  258.             del self.pconf["plugins"][rawname]

  259.             self.loaded[dirname] = None

  260.             self.save_config()

  261.             return True, "卸载插件成功"

  262.         except Exception as e:

  263.             logger.error("Failed to uninstall plugin, {}".format(e))

  264.             return False, "卸载插件失败，请手动删除文件夹完成卸载，"+str(e)