mirror of
https://github.com/jxxghp/MoviePilot-Plugins.git
synced 2026-03-27 10:05:57 +00:00
260 lines
8.9 KiB
Python
260 lines
8.9 KiB
Python
import threading
|
||
|
||
from gotify import Gotify
|
||
|
||
from queue import Queue
|
||
from time import time, sleep
|
||
from typing import Any, List, Dict, Tuple
|
||
from urllib.parse import urlencode
|
||
|
||
from app.core.event import eventmanager, Event
|
||
from app.log import logger
|
||
from app.plugins import _PluginBase
|
||
from app.schemas.types import EventType, NotificationType
|
||
from app.utils.http import RequestUtils
|
||
|
||
|
||
class GotifyMsg(_PluginBase):
|
||
# 插件名称
|
||
plugin_name = "gotify消息通知"
|
||
# 插件描述
|
||
plugin_desc = "支持使用gotify发送消息通知。"
|
||
# 插件图标
|
||
plugin_icon = "https://raw.githubusercontent.com/gotify/logo/refs/heads/master/gotify-logo.png"
|
||
# 插件版本
|
||
plugin_version = "1.1"
|
||
# 插件作者
|
||
plugin_author = "lethargicScribe"
|
||
# 作者主页
|
||
author_url = "https://github.com/lethargicScribe"
|
||
# 插件配置项ID前缀
|
||
plugin_config_prefix = "gotifymsg_"
|
||
# 加载顺序
|
||
plugin_order = 25
|
||
# 可使用的用户级别
|
||
auth_level = 1
|
||
|
||
# 私有属性
|
||
_enabled = False
|
||
_server = None
|
||
_token = None
|
||
_msgtypes = []
|
||
|
||
# 消息处理线程
|
||
processing_thread = None
|
||
# 上次发送时间
|
||
last_send_time = 0
|
||
# 消息队列
|
||
message_queue = Queue()
|
||
# 消息发送间隔（秒）
|
||
send_interval = 5
|
||
# 退出事件
|
||
__event = threading.Event()
|
||
# gotify实例
|
||
gotify = None
|
||
|
||
def init_plugin(self, config: dict = None):
|
||
self.__event.clear()
|
||
if config:
|
||
self._enabled = config.get("enabled")
|
||
self._server = config.get("server")
|
||
self._token = config.get("token")
|
||
self._msgtypes = config.get("msgtypes") or []
|
||
|
||
if self._enabled and self._token and self._server:
|
||
# 初始化gotify客户端实例
|
||
self.gotify = Gotify(
|
||
base_url = self._server,
|
||
app_token = self._token,
|
||
)
|
||
# 启动处理队列的后台线程
|
||
self.processing_thread = threading.Thread(target=self.process_queue)
|
||
self.processing_thread.daemon = True
|
||
self.processing_thread.start()
|
||
|
||
def get_state(self) -> bool:
|
||
return self._enabled and (True if self._server and self._token else False)
|
||
|
||
@staticmethod
|
||
def get_command() -> List[Dict[str, Any]]:
|
||
pass
|
||
|
||
def get_api(self) -> List[Dict[str, Any]]:
|
||
pass
|
||
|
||
def get_form(self) -> Tuple[List[dict], Dict[str, Any]]:
|
||
"""
|
||
拼装插件配置页面，需要返回两块数据：1、页面配置；2、数据结构
|
||
"""
|
||
# 编历 NotificationType 枚举，生成消息类型选项
|
||
MsgTypeOptions = []
|
||
for item in NotificationType:
|
||
MsgTypeOptions.append({
|
||
"title": item.value,
|
||
"value": item.name
|
||
})
|
||
return [
|
||
{
|
||
'component': 'VForm',
|
||
'content': [
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'enabled',
|
||
'label': '启用插件',
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'server',
|
||
'label': '服务器',
|
||
'placeholder': 'https://gotify.example.com',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'token',
|
||
'label': 'gotify 令牌',
|
||
'placeholder': 'xxxxx',
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSelect',
|
||
'props': {
|
||
'multiple': True,
|
||
'chips': True,
|
||
'model': 'msgtypes',
|
||
'label': '消息类型',
|
||
'items': MsgTypeOptions
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
]
|
||
}
|
||
], {
|
||
"enabled": False,
|
||
'server': '',
|
||
'token': '',
|
||
'msgtypes': []
|
||
}
|
||
|
||
def get_page(self) -> List[dict]:
|
||
pass
|
||
|
||
@eventmanager.register(EventType.NoticeMessage)
|
||
def send(self, event: Event):
|
||
"""
|
||
消息发送事件，将消息加入队列
|
||
"""
|
||
if not self.get_state() or not event.event_data:
|
||
return
|
||
|
||
msg_body = event.event_data
|
||
# 验证消息的有效性
|
||
if not msg_body.get("title") and not msg_body.get("text"):
|
||
logger.warn("标题和内容不能同时为空")
|
||
return
|
||
|
||
# 将消息加入队列
|
||
self.message_queue.put(msg_body)
|
||
logger.info("消息已加入队列等待发送")
|
||
|
||
def process_queue(self):
|
||
"""
|
||
处理队列中的消息，按间隔时间发送
|
||
"""
|
||
while True:
|
||
if self.__event.is_set():
|
||
logger.info("消息发送线程正在退出...")
|
||
break
|
||
# 获取队列中的下一条消息
|
||
msg_body = self.message_queue.get()
|
||
|
||
# 检查是否满足发送间隔时间
|
||
current_time = time()
|
||
time_since_last_send = current_time - self.last_send_time
|
||
if time_since_last_send < self.send_interval:
|
||
sleep(self.send_interval - time_since_last_send)
|
||
|
||
# 处理消息内容
|
||
channel = msg_body.get("channel")
|
||
if channel:
|
||
continue
|
||
msg_type: NotificationType = msg_body.get("type")
|
||
title = msg_body.get("title")
|
||
text = msg_body.get("text")
|
||
|
||
# 检查消息类型是否已启用
|
||
if msg_type and self._msgtypes and msg_type.name not in self._msgtypes:
|
||
logger.info(f"消息类型 {msg_type.value} 未开启消息发送")
|
||
continue
|
||
|
||
# 尝试发送消息
|
||
try:
|
||
self.gotify.create_message(
|
||
text,
|
||
title=title,
|
||
priority=0,
|
||
)
|
||
except Exception as msg_e:
|
||
logger.error(f"gotify消息发送失败，{str(msg_e)}")
|
||
|
||
# 标记任务完成
|
||
self.message_queue.task_done()
|
||
|
||
def stop_service(self):
|
||
"""
|
||
退出插件
|
||
"""
|
||
self.__event.set()
|