diff --git a/languages/built-in/en.yml b/languages/built-in/en.yml
index b514a1c..591c298 100644
--- a/languages/built-in/en.yml
+++ b/languages/built-in/en.yml
@@ -344,6 +344,15 @@ apt_plugin_maintainer: Author
apt_plugin_size: size
apt_plugin_support: support period
apt_plugin_des_short: description
+apt_source_des: Add or delete or view the source of the plugin repository.
+apt_source_parameters: "[add/del] [source url]"
+apt_source_not_found: Something went wrong ~ The specified source does not exist.
+apt_source_header: "Apt Source List:"
+apt_source_add_success: Add success
+apt_source_del_success: Del success
+apt_source_add_failed: Add failed, this source already exists
+apt_source_add_invalid: Add failed, invalid source
+apt_source_del_failed: Del failed, this source does not exist
#prune
## prune
prune_des: Replying to a message with this command will delete all messages between the latest message and the message. Limit, 1000 messages based on message ID, more than 1000 messages may trigger the limit of deleting messages too quickly. (Non-group administrators only delete their own messages)
diff --git a/languages/built-in/zh-cn.yml b/languages/built-in/zh-cn.yml
index c157de0..f5f16eb 100644
--- a/languages/built-in/zh-cn.yml
+++ b/languages/built-in/zh-cn.yml
@@ -344,6 +344,15 @@ apt_plugin_maintainer: 作者
apt_plugin_size: 大小
apt_plugin_support: 支持周期
apt_plugin_des_short: 说明
+apt_source_des: "添加/删除/查看 自定义插件源"
+apt_source_parameters: "[add/del] [自定义软件源链接]"
+apt_source_not_found: 当前没有自定义插件源
+apt_source_header: "当前自定义插件源:"
+apt_source_add_success: 添加成功
+apt_source_del_success: 删除成功
+apt_source_add_failed: 添加失败，此源已存在
+apt_source_add_invalid: 添加失败，此源无效
+apt_source_del_failed: 删除失败，此源不存在
#prune
## prune
prune_des: 以此命令回复某条消息，将删除最新一条消息至该条消息之间的所有消息。限制：基于消息 ID 的 1000 条消息，大于 1000 条可能会触发删除消息过快限制。（非群组管理员只删除自己的消息）
diff --git a/pagermaid/__init__.py b/pagermaid/__init__.py
index e1c6348..e711937 100644
--- a/pagermaid/__init__.py
+++ b/pagermaid/__init__.py
@@ -21,8 +21,8 @@ from pagermaid.scheduler import scheduler
import pyromod.listen
from pyrogram import Client
-pgm_version = "1.4.0"
-pgm_version_code = 1400
+pgm_version = "1.4.1"
+pgm_version_code = 1401
CMD_LIST = {}
module_dir = __path__[0]
working_dir = getcwd()
diff --git a/pagermaid/common/plugin.py b/pagermaid/common/plugin.py
index 9cb600e..360575c 100644
--- a/pagermaid/common/plugin.py
+++ b/pagermaid/common/plugin.py
@@ -2,13 +2,16 @@ import contextlib
import json
import os
from pathlib import Path
-from typing import Optional, List
+from typing import Optional, List, Tuple, Dict
-from pydantic import BaseModel
+from pydantic import BaseModel, ValidationError
-from pagermaid import Config
+from pagermaid import Config, logs
+from pagermaid.enums import Message
from pagermaid.common.cache import cache
+from pagermaid.modules import plugin_list as active_plugins
from pagermaid.utils import client
+from pagermaid.services import sqlite
plugins_path = Path("plugins")
@@ -27,6 +30,11 @@ class LocalPlugin(BaseModel):
def disabled_path(self) -> Path:
return plugins_path / f"{self.name}.py.disabled"
+ @property
+ def load_status(self) -> bool:
+ """插件加载状态"""
+ return self.name in active_plugins
+
def remove(self):
with contextlib.suppress(FileNotFoundError):
os.remove(self.normal_path)
@@ -53,7 +61,8 @@ class RemotePlugin(LocalPlugin):
maintainer: str
size: str
supported: bool
- des: str
+ des: str = ""
+ des_short: str = ""
...
async def install(self) -> bool:
@@ -66,8 +75,61 @@ class RemotePlugin(LocalPlugin):
return False
-class PluginManager:
+class PluginRemote(BaseModel):
+ url: str
+ status: bool
+
+ @property
+ def text(self) -> str:
+ return f"{'✅' if self.status else '❌'} {self.url}"
+
+
+class PluginRemoteManager:
def __init__(self):
+ self.key = "plugins_remotes"
+
+ def get_remotes(self) -> List[PluginRemote]:
+ return [PluginRemote(**i) for i in sqlite.get(self.key, [])]
+
+ def set_remotes(self, remotes: List[PluginRemote]):
+ sqlite[self.key] = [i.dict() for i in remotes]
+
+ def add_remote(self, remote_url: str) -> bool:
+ remotes = self.get_remotes()
+ if not next(filter(lambda x: x.url == remote_url, remotes), None):
+ remotes.append(PluginRemote(url=remote_url, status=True))
+ self.set_remotes(remotes)
+ return True
+ return False
+
+ def remove_remote(self, remote_url: str) -> bool:
+ remotes = self.get_remotes()
+ if next(filter(lambda x: x.url == remote_url, remotes), None):
+ remotes = [i for i in remotes if i.url != remote_url]
+ self.set_remotes(remotes)
+ return True
+ return False
+
+ def disable_remote(self, remote_url: str) -> bool:
+ remotes = self.get_remotes()
+ if remote := next(filter(lambda x: x.url == remote_url, remotes), None):
+ remote.status = False
+ self.set_remotes(remotes)
+ return True
+ return False
+
+ def enable_remote(self, remote_url: str) -> bool:
+ remotes = self.get_remotes()
+ if remote := next(filter(lambda x: x.url == remote_url, remotes), None):
+ remote.status = True
+ self.set_remotes(remotes)
+ return True
+ return False
+
+
+class PluginManager:
+ def __init__(self, remote_manager: PluginRemoteManager):
+ self.remote_manager = remote_manager
self.version_map = {}
self.remote_version_map = {}
self.plugins: List[LocalPlugin] = []
@@ -134,25 +196,55 @@ class PluginManager:
def get_local_plugin(self, name: str) -> LocalPlugin:
return next(filter(lambda x: x.name == name, self.plugins), None)
- @cache()
- async def _load_remote_plugins(self) -> List[RemotePlugin]:
- plugin_list = await client.get(f"{Config.GIT_SOURCE}list.json")
- plugin_list = plugin_list.json()["list"]
- plugins = [
- RemotePlugin(
- **plugin,
- status=False,
- )
- for plugin in plugin_list
- ]
+ @staticmethod
+ async def fetch_remote_url(url: str) -> List[Dict]:
+ try:
+ data = await client.get(f"{url}list.json")
+ data.raise_for_status()
+ return data.json()["list"]
+ except Exception as e:
+ logs.error(f"获取远程插件列表失败: {e}")
+ raise e
+
+ async def load_remote_plugins_no_cache(self) -> List[RemotePlugin]:
+ remote_urls = [i.url for i in self.remote_manager.get_remotes()]
+ remote_urls.insert(0, Config.GIT_SOURCE)
+ plugins = []
+ plugins_name = []
+ for remote in remote_urls:
+ try:
+ plugin_list = await self.fetch_remote_url(remote)
+ except Exception as e:
+ logs.error(f"获取远程插件列表失败: {e}")
+ self.remote_manager.disable_remote(remote)
+ continue
+ self.remote_manager.enable_remote(remote)
+ for plugin in plugin_list:
+ try:
+ plugin_model = RemotePlugin(**plugin, status=False)
+ if plugin_model.name in plugins_name:
+ continue
+ plugins.append(plugin_model)
+ plugins_name.append(plugin_model.name)
+ except ValidationError:
+ logs.warning(f"远程插件 {plugin} 信息不完整")
+ continue
self.remote_plugins = plugins
- self.remote_version_map = {}
- for plugin in plugins:
- self.remote_version_map[plugin.name] = plugin.version
+ self.remote_version_map = {plugin.name: plugin.version for plugin in plugins}
return plugins
- async def load_remote_plugins(self) -> List[RemotePlugin]:
- plugin_list = await self._load_remote_plugins()
+ @cache()
+ async def load_remote_plugins_cache(self) -> List[RemotePlugin]:
+ return await self.load_remote_plugins_no_cache()
+
+ async def load_remote_plugins(
+ self, enable_cache: bool = True
+ ) -> List[RemotePlugin]:
+ plugin_list = (
+ await self.load_remote_plugins_cache()
+ if enable_cache
+ else await self.load_remote_plugins_no_cache()
+ )
for i in plugin_list:
i.status = self.get_plugin_load_status(i.name)
return plugin_list
@@ -187,5 +279,40 @@ class PluginManager:
updated_plugins.append(i)
return updated_plugins
+ @staticmethod
+ async def download_from_message(message: Message) -> str:
+ """Download a plugin from a message"""
+ reply = message.reply_to_message
+ file_path = None
+ if (
+ reply
+ and reply.document
+ and reply.document.file_name
+ and reply.document.file_name.endswith(".py")
+ ):
+ file_path = await message.reply_to_message.download()
+ elif (
+ message.document
+ and message.document.file_name
+ and message.document.file_name.endswith(".py")
+ ):
+ file_path = await message.download()
+ return file_path
-plugin_manager = PluginManager()
+ def get_plugins_status(
+ self,
+ ) -> Tuple[List[str], List[LocalPlugin], List[LocalPlugin]]:
+ """Get plugins status"""
+ all_local_plugins = self.plugins
+ disabled_plugins = []
+ inactive_plugins = []
+ for plugin in all_local_plugins:
+ if not plugin.status:
+ inactive_plugins.append(plugin)
+ elif not plugin.load_status:
+ disabled_plugins.append(plugin)
+ return active_plugins, disabled_plugins, inactive_plugins
+
+
+plugin_remote_manager = PluginRemoteManager()
+plugin_manager = PluginManager(plugin_remote_manager)
diff --git a/pagermaid/config.py b/pagermaid/config.py
index 6ddca7a..a197fc2 100644
--- a/pagermaid/config.py
+++ b/pagermaid/config.py
@@ -95,9 +95,9 @@ class Config:
)
try:
with open(
- f"languages{os.sep}built-in{os.sep}en.yml",
- "r",
- encoding="utf-8",
+ f"languages{os.sep}built-in{os.sep}en.yml",
+ "r",
+ encoding="utf-8",
) as f:
lang_default_dict = safe_load(f)
with open(
diff --git a/pagermaid/enums/__init__.py b/pagermaid/enums/__init__.py
index 2e45042..b442c2e 100644
--- a/pagermaid/enums/__init__.py
+++ b/pagermaid/enums/__init__.py
@@ -13,5 +13,5 @@ __all__ = [
"AsyncIOScheduler",
"SqliteDict",
"AsyncClient",
- "Logger"
+ "Logger",
]
diff --git a/pagermaid/modules/plugin.py b/pagermaid/modules/plugin.py
index 662ae92..f57e830 100644
--- a/pagermaid/modules/plugin.py
+++ b/pagermaid/modules/plugin.py
@@ -1,19 +1,17 @@
""" PagerMaid module to manage plugins. """
import contextlib
-import json
-from glob import glob
-from os import remove, chdir, path, sep
+from os import remove, path, sep
from os.path import exists
from re import search, I
from shutil import copyfile, move
-from pagermaid import log, working_dir, Config
-from pagermaid.common.plugin import plugin_manager
+from pagermaid import log, working_dir
+from pagermaid.common.plugin import plugin_remote_manager, plugin_manager
from pagermaid.common.reload import reload_all
+from pagermaid.enums import Message
from pagermaid.listener import listener
-from pagermaid.modules import plugin_list as active_plugins, __list_plugins
-from pagermaid.utils import upload_attachment, lang, Message, client
+from pagermaid.utils import upload_attachment, lang
def remove_plugin(name):
@@ -31,15 +29,6 @@ def move_plugin(file_path):
move(file_path, plugin_directory)
-def update_version(plugin_name, version):
- plugin_directory = f"{working_dir}{sep}plugins{sep}"
- with open(f"{plugin_directory}version.json", "r", encoding="utf-8") as f:
- version_json = json.load(f)
- version_json[plugin_name] = version
- with open(f"{plugin_directory}version.json", "w") as f:
- json.dump(version_json, f)
-
-
@listener(
is_plugin=False,
outgoing=True,
@@ -57,27 +46,19 @@ async def plugin(message: Message):
plugin_directory = f"{working_dir}{sep}plugins{sep}"
if message.parameter[0] == "install":
if len(message.parameter) == 1:
- message = await message.edit(lang("apt_processing"))
- file_path = None
- with contextlib.suppress(Exception):
- if reply:
- file_path = await reply.download()
- else:
- file_path = await message.download()
- if file_path is None or not file_path.endswith(".py"):
+ file_path = await plugin_manager.download_from_message(message)
+ if file_path is None:
await message.edit(lang("apt_no_py"))
- try:
- remove(str(file_path))
- except FileNotFoundError:
- pass
return
+ plugin_name = path.basename(file_path)[:-3]
+ plugin_manager.remove_plugin(plugin_name)
move_plugin(file_path)
await message.edit(
f"{lang('apt_name')}\n\n"
f"{lang('apt_plugin')} "
- f"{path.basename(file_path)[:-3]} {lang('apt_installed')}"
+ f"{plugin_name} {lang('apt_installed')}"
)
- await log(f"{lang('apt_install_success')} {path.basename(file_path)[:-3]}.")
+ await log(f"{lang('apt_install_success')} {plugin_name}.")
await reload_all()
elif len(message.parameter) >= 2:
await plugin_manager.load_remote_plugins()
@@ -134,27 +115,14 @@ async def plugin(message: Message):
await message.edit(lang("arg_error"))
elif message.parameter[0] == "status":
if len(message.parameter) == 1:
- inactive_plugins = sorted(__list_plugins())
- disabled_plugins = []
- if not len(inactive_plugins) == 0:
- for target_plugin in active_plugins:
- inactive_plugins.remove(target_plugin)
- chdir(f"plugins{sep}")
- for target_plugin in glob("*.py.disabled"):
- disabled_plugins += [f"{target_plugin[:-12]}"]
- chdir(f"..{sep}")
- active_plugins_string = ""
- inactive_plugins_string = ""
- disabled_plugins_string = ""
- for target_plugin in active_plugins:
- active_plugins_string += f"{target_plugin}, "
- active_plugins_string = active_plugins_string[:-2]
- for target_plugin in inactive_plugins:
- inactive_plugins_string += f"{target_plugin}, "
- inactive_plugins_string = inactive_plugins_string[:-2]
- for target_plugin in disabled_plugins:
- disabled_plugins_string += f"{target_plugin}, "
- disabled_plugins_string = disabled_plugins_string[:-2]
+ (
+ active_plugins,
+ disabled_plugins,
+ inactive_plugins,
+ ) = plugin_manager.get_plugins_status()
+ active_plugins_string = ", ".join(active_plugins)
+ inactive_plugins_string = ", ".join([i.name for i in inactive_plugins])
+ disabled_plugins_string = ", ".join([i.name for i in disabled_plugins])
if len(active_plugins) == 0:
active_plugins_string = f"`{lang('apt_no_running_plugins')}`"
if len(inactive_plugins) == 0:
@@ -207,17 +175,19 @@ async def plugin(message: Message):
elif exists(f"{plugin_directory}{file_name}.disabled"):
copyfile(f"{plugin_directory}{file_name}.disabled", file_name)
if exists(file_name):
- await message.edit(lang("apt_uploading"))
- await upload_attachment(
- file_name,
- message.chat.id,
- reply_id,
- thumb=f"pagermaid{sep}assets{sep}logo.jpg",
- caption=f"{lang('apt_name')}\n\n"
- f"PagerMaid-Pyro {message.parameter[1]} plugin.",
- )
- remove(file_name)
- await message.safe_delete()
+ try:
+ await message.edit(lang("apt_uploading"))
+ await upload_attachment(
+ file_name,
+ message.chat.id,
+ reply_id,
+ thumb=f"pagermaid{sep}assets{sep}logo.jpg",
+ caption=f"{lang('apt_name')}\n\n"
+ f"PagerMaid-Pyro {message.parameter[1]} plugin.",
+ )
+ await message.safe_delete()
+ finally:
+ remove(file_name)
else:
await message.edit(lang("apt_not_exist"))
else:
@@ -248,22 +218,12 @@ async def plugin(message: Message):
if len(message.parameter) == 1:
await message.edit(lang("apt_search_no_name"))
elif len(message.parameter) == 2:
+ await plugin_manager.load_remote_plugins()
search_result = []
plugin_name = message.parameter[1]
- plugin_list = await client.get(f"{Config.GIT_SOURCE}list.json")
- plugin_online = plugin_list.json()["list"]
- for i in plugin_online:
- if search(plugin_name, i["name"], I):
- search_result.extend(
- [
- "`"
- + i["name"]
- + "` / `"
- + i["version"]
- + "`\n "
- + i["des-short"]
- ]
- )
+ for i in plugin_manager.remote_plugins:
+ if search(plugin_name, i.name, I):
+ search_result.append(f"`{i.name}` / `{i.version}`\n {i.des_short}")
if len(search_result) == 0:
await message.edit(lang("apt_search_not_found"))
else:
@@ -277,24 +237,23 @@ async def plugin(message: Message):
if len(message.parameter) == 1:
await message.edit(lang("apt_search_no_name"))
elif len(message.parameter) == 2:
+ await plugin_manager.load_remote_plugins()
search_result = ""
plugin_name = message.parameter[1]
- plugin_list = await client.get(f"{Config.GIT_SOURCE}list.json")
- plugin_online = plugin_list.json()["list"]
- for i in plugin_online:
- if plugin_name == i["name"]:
- if i["supported"]:
+ for i in plugin_manager.remote_plugins:
+ if plugin_name == i.name:
+ if i.supported:
search_support = lang("apt_search_supporting")
else:
search_support = lang("apt_search_not_supporting")
search_result = (
- f"{lang('apt_plugin_name')}:`{i['name']}`\n"
- f"{lang('apt_plugin_ver')}:`Ver {i['version']}`\n"
- f"{lang('apt_plugin_section')}:`{i['section']}`\n"
- f"{lang('apt_plugin_maintainer')}:`{i['maintainer']}`\n"
- f"{lang('apt_plugin_size')}:`{i['size']}`\n"
+ f"{lang('apt_plugin_name')}:`{i.name}`\n"
+ f"{lang('apt_plugin_ver')}:`Ver {i.version}`\n"
+ f"{lang('apt_plugin_section')}:`{i.section}`\n"
+ f"{lang('apt_plugin_maintainer')}:`{i.maintainer}`\n"
+ f"{lang('apt_plugin_size')}:`{i.size}`\n"
f"{lang('apt_plugin_support')}:{search_support}\n"
- f"{lang('apt_plugin_des_short')}:{i['des-short']}"
+ f"{lang('apt_plugin_des_short')}:{i.des_short}"
)
break
if search_result == "":
@@ -307,20 +266,60 @@ async def plugin(message: Message):
return
message = await message.edit(lang("stats_loading"))
list_plugin = []
- with open(f"{plugin_directory}version.json", "r", encoding="utf-8") as f:
- version_json = json.load(f)
- plugin_list = await client.get(f"{Config.GIT_SOURCE}list.json")
- plugin_online = plugin_list.json()["list"]
- for key, value in version_json.items():
- if value == "0.0":
+ for key, value in plugin_manager.version_map.items():
+ if not value:
continue
- for i in plugin_online:
- if key == i["name"]:
- list_plugin.append(key)
- break
+ list_plugin.append(key)
if len(list_plugin) == 0:
await message.edit(lang("apt_why_not_install_a_plugin"))
else:
await message.edit(",apt install " + " ".join(list_plugin))
else:
await message.edit(lang("arg_error"))
+
+
+@listener(
+ is_plugin=False,
+ outgoing=True,
+ command="apt_source",
+ need_admin=True,
+ description=lang("apt_source_des"),
+ parameters=lang("apt_source_parameters"),
+)
+async def apt_source(message: Message):
+ if len(message.parameter) == 0:
+ remotes = plugin_remote_manager.get_remotes()
+ if len(remotes) == 0:
+ await message.edit(lang("apt_source_not_found"))
+ return
+ await message.edit(
+ f"{lang('apt_source_header')}\n\n" + "\n".join([i.text for i in remotes]),
+ disable_web_page_preview=True,
+ )
+ elif len(message.parameter) == 2:
+ url = message.parameter[1]
+ if not url.endswith("/"):
+ url += "/"
+ if message.parameter[0] == "add":
+ try:
+ status = await plugin_manager.fetch_remote_url(url)
+ except Exception:
+ status = False
+ if status:
+ if plugin_remote_manager.add_remote(url):
+ await message.edit(lang("apt_source_add_success"))
+ await plugin_manager.load_remote_plugins(enable_cache=False)
+ else:
+ await message.edit(lang("apt_source_add_failed"))
+ else:
+ await message.edit(lang("apt_source_add_invalid"))
+ elif message.parameter[0] == "del":
+ if plugin_remote_manager.remove_remote(url):
+ await message.edit(lang("apt_source_del_success"))
+ await plugin_manager.load_remote_plugins(enable_cache=False)
+ else:
+ await message.edit(lang("apt_source_del_failed"))
+ else:
+ await message.edit(lang("arg_error"))
+ else:
+ await message.edit(lang("arg_error"))
diff --git a/pagermaid/modules/sentry.py b/pagermaid/modules/sentry.py
index 9b7cd3c..743cdc2 100644
--- a/pagermaid/modules/sentry.py
+++ b/pagermaid/modules/sentry.py
@@ -30,7 +30,9 @@ def sentry_before_send(event, hint):
sentry_sdk_report_time = time()
sentry_sdk_git_hash = (
- run("git rev-parse HEAD", stdout=PIPE, shell=True, check=True).stdout.decode().strip()
+ run("git rev-parse HEAD", stdout=PIPE, shell=True, check=True)
+ .stdout.decode()
+ .strip()
)
sentry_sdk.init(
Config.SENTRY_API,
diff --git a/pagermaid/web/api/__init__.py b/pagermaid/web/api/__init__.py
index cb57d01..5b11baa 100644
--- a/pagermaid/web/api/__init__.py
+++ b/pagermaid/web/api/__init__.py
@@ -7,7 +7,10 @@ from pagermaid.web.api.ignore_groups import route as ignore_groups_route
from pagermaid.web.api.login import route as login_route
from pagermaid.web.api.plugin import route as plugin_route
from pagermaid.web.api.status import route as status_route
-from pagermaid.web.api.web_login import route as web_login_route, html_route as web_login_html_route
+from pagermaid.web.api.web_login import (
+ route as web_login_route,
+ html_route as web_login_html_route,
+)
__all__ = ["authentication", "base_api_router", "base_html_router"]
diff --git a/pagermaid/web/api/web_login.py b/pagermaid/web/api/web_login.py
index 2b4121e..5de21c8 100644
--- a/pagermaid/web/api/web_login.py
+++ b/pagermaid/web/api/web_login.py
@@ -20,7 +20,7 @@ class UserModel(BaseModel):
password: str
-class WebLogin():
+class WebLogin:
def __init__(self):
self.is_authorized = False
self.need_password = False
@@ -51,9 +51,7 @@ class WebLogin():
if not self.is_authorized:
return
if not await bot.storage.is_bot() and bot.takeout:
- bot.takeout_id = (
- await bot.invoke(InitTakeoutSession())
- ).id
+ bot.takeout_id = (await bot.invoke(InitTakeoutSession())).id
await bot.invoke(GetState())
bot.me = await bot.get_me()
diff --git a/pagermaid/web/pages/ignore_groups.py b/pagermaid/web/pages/ignore_groups.py
index b127c27..b513b06 100644
--- a/pagermaid/web/pages/ignore_groups.py
+++ b/pagermaid/web/pages/ignore_groups.py
@@ -5,7 +5,7 @@ from amis import (
CardsCRUD,
PageSchema,
Page,
- )
+)
card = Card(
header=Card.Header(
diff --git a/pyromod/methods/read_chat_history.py b/pyromod/methods/read_chat_history.py
index 80af626..8d1a34e 100644
--- a/pyromod/methods/read_chat_history.py
+++ b/pyromod/methods/read_chat_history.py
@@ -5,7 +5,7 @@ import pyrogram
async def read_chat_history(
- self: "pyrogram.Client", chat_id: Union[int, str], max_id: int = 0
+ self: "pyrogram.Client", chat_id: Union[int, str], max_id: int = 0
) -> bool:
peer = await self.resolve_peer(chat_id)
if isinstance(peer, pyrogram.raw.types.InputPeerChannel):