mirror of
https://github.com/aaPanel/aaPanel.git
synced 2026-08-17 21:25:47 +02:00
260 lines
10 KiB
Python
260 lines
10 KiB
Python
# coding: utf-8
|
|
# -------------------------------------------------------------------
|
|
# aapanel
|
|
# -------------------------------------------------------------------
|
|
# Copyright (c) 2015-2017 aapanel(http:#bt.cn) All rights reserved.
|
|
# -------------------------------------------------------------------
|
|
# Author: baozi <baozi@bt.cn>
|
|
# -------------------------------------------------------------------
|
|
# 新告警通道管理模块
|
|
# ------------------------------
|
|
import json
|
|
import glob
|
|
import os
|
|
import sys
|
|
|
|
if "/www/server/panel/class" not in sys.path:
|
|
sys.path.insert(0, "/www/server/panel/class")
|
|
import public
|
|
|
|
from mod.base import json_response
|
|
|
|
from mod.base.push_mod import PushManager, TaskConfig, TaskRecordConfig, TaskTemplateConfig, PushSystem
|
|
from mod.base.push_mod import update_mod_push_system, UPDATE_MOD_PUSH_FILE, load_task_template_by_file
|
|
from mod.base.msg import update_mod_push_msg
|
|
from mod.base.push_mod.rsync_push import load_rsync_template
|
|
from mod.base.push_mod.task_manager_push import load_task_manager_template
|
|
from mod.base.push_mod.load_push import load_load_template
|
|
|
|
# 内置模板目录
|
|
TEMPLATE_DIR = "/www/server/panel/mod/base/push_mod/templates/"
|
|
|
|
|
|
def update_mod():
|
|
files = os.path.join(TEMPLATE_DIR, "*_push_template.json")
|
|
template_files = sorted(glob.glob(files))
|
|
|
|
# 所有内置源模板, 不包括插件, 插件告警模板统一自身处理控制
|
|
all_source_ids = set()
|
|
for tpl_file in template_files:
|
|
if not os.path.isfile(tpl_file) or not tpl_file.endswith(".json"):
|
|
continue
|
|
try:
|
|
tpl_data = json.loads(public.readFile(tpl_file))
|
|
for tpl in tpl_data:
|
|
# 校验关键字段
|
|
if not tpl.get("id") or not tpl.get("load_cls") or not tpl.get("template"):
|
|
public.print_log(
|
|
f"Template file {tpl_file} missing required fields:"
|
|
f" id 模板唯一id / load_cls 模块/ template 前端表单控制"
|
|
)
|
|
continue
|
|
all_source_ids.add(str(tpl["id"]))
|
|
except Exception as e:
|
|
public.print_log(f"Failed to parse template file {tpl_file}: {e}")
|
|
|
|
# 启动都加载模板, merge操作
|
|
for tpl_file in template_files:
|
|
load_task_template_by_file(tpl_file)
|
|
|
|
# 动态加载模板,仅首次, 使用标志 UPDATE_MOD_PUSH_FILE 便于从其他调用处控制是否更新
|
|
if not os.path.exists(UPDATE_MOD_PUSH_FILE):
|
|
update_mod_push_msg()
|
|
load_rsync_template()
|
|
load_task_manager_template()
|
|
load_load_template()
|
|
update_mod_push_system()
|
|
|
|
|
|
update_mod()
|
|
del update_mod
|
|
|
|
|
|
class main(PushManager):
|
|
def get_task_list(self, get=None):
|
|
# 通道类型映射，包含模糊匹配规则
|
|
channel_map = {
|
|
# "wx_account": "wx_account",
|
|
"mail": "mail",
|
|
"webhook": "webhook",
|
|
"feishu": "feishu",
|
|
"dingding": "dingding",
|
|
# "短信": "sms",
|
|
"tg": "tg",
|
|
"discord": "discord",
|
|
}
|
|
try:
|
|
if get:
|
|
# 获取状态和关键词参数
|
|
status_filter = get.get("status", None)
|
|
keyword_filter = get.get("keyword", None)
|
|
else:
|
|
status_filter = ""
|
|
keyword_filter = ""
|
|
res = TaskConfig().config
|
|
|
|
# 按创建时间排序
|
|
res.sort(key=lambda x: x["create_time"])
|
|
# 读取发送者信息
|
|
sender_info = self.get_sender_info()
|
|
|
|
# 根据状态过滤任务
|
|
if status_filter:
|
|
res = [task for task in res if str(task["status"]).lower() == status_filter.lower()]
|
|
|
|
# 根据关键词过滤任务
|
|
if keyword_filter:
|
|
keyword_filter_lower = keyword_filter.lower()
|
|
filtered_res = []
|
|
for task in res:
|
|
task_match = False
|
|
if keyword_filter_lower == "Alert when the panel is logged in":
|
|
if task['keyword'] == "panel_login":
|
|
task_match = True
|
|
if keyword_filter_lower in task["title"].lower() or \
|
|
(task["task_data"].get("title") and keyword_filter_lower in task["task_data"][
|
|
"title"].lower()) or \
|
|
(task["time_rule"].get("send_interval") and keyword_filter_lower in str(
|
|
task["time_rule"]["send_interval"])) or \
|
|
(task["number_rule"].get("day_num") and keyword_filter_lower in str(
|
|
task["number_rule"]["day_num"])):
|
|
task_match = True
|
|
else:
|
|
for sender_id in task["sender"]:
|
|
sender = sender_info.get(sender_id, {})
|
|
sender_title = sender.get("data", {}).get("title", "").lower()
|
|
sender_type = sender.get("sender_type", "").lower()
|
|
if keyword_filter_lower in sender_title or \
|
|
keyword_filter_lower in sender_type:
|
|
task_match = True
|
|
break
|
|
# 检查关键词是否包含在通道类型的映射键中
|
|
for chinese_name, channel_type in channel_map.items():
|
|
if keyword_filter_lower in chinese_name.lower() and channel_type == sender_type:
|
|
task_match = True
|
|
break
|
|
if task_match:
|
|
filtered_res.append(task)
|
|
res = filtered_res
|
|
for i in res:
|
|
i['view_msg'] = self.get_view_msg_format(i)
|
|
|
|
return json_response(status=True, data=res)
|
|
except Exception as e:
|
|
import traceback
|
|
public.print_log(traceback.format_exc())
|
|
return json_response(status=True, data=str(e))
|
|
|
|
def get_sender_info(self):
|
|
sender_file = '/www/server/panel/data/mod_push_data/sender.json'
|
|
try:
|
|
with open(sender_file, 'r', encoding='utf-8') as f:
|
|
sender_data = json.load(f)
|
|
return {sender['id']: sender for sender in sender_data}
|
|
except Exception as e:
|
|
return {}
|
|
|
|
@staticmethod
|
|
def get_task_record(get):
|
|
page = 1
|
|
size = 10
|
|
try:
|
|
if hasattr(get, "page"):
|
|
page = int(get.page.strip())
|
|
if hasattr(get, "size"):
|
|
size = int(get.size.strip())
|
|
task_id = get.task_id.strip()
|
|
except (AttributeError, ValueError, TypeError):
|
|
return json_response(status=False, msg="Parameter error")
|
|
|
|
t = TaskRecordConfig(task_id)
|
|
t.config.sort(key=lambda x: x["create_time"])
|
|
page = max(page, 1)
|
|
size = max(size, 1)
|
|
count = len(t.config)
|
|
data = t.config[(page - 1) * size: page * size]
|
|
return json_response(status=True, data={
|
|
"count": count,
|
|
"list": data,
|
|
})
|
|
|
|
def clear_task_record(self, get):
|
|
try:
|
|
task_id = get.task_id.strip()
|
|
except (AttributeError, ValueError, TypeError):
|
|
return json_response(status=False, msg="Parameter error")
|
|
self.clear_task_record_by_task_id(task_id)
|
|
|
|
return json_response(status=True, msg="Cleared successfully")
|
|
|
|
@staticmethod
|
|
def remove_task_records(get):
|
|
try:
|
|
task_id = get.task_id.strip()
|
|
record_ids = set(json.loads(get.record_ids.strip()))
|
|
except (AttributeError, ValueError, TypeError):
|
|
return json_response(status=False, msg="Parameter error")
|
|
task_records = TaskRecordConfig(task_id)
|
|
for i in range(len(task_records.config) - 1, -1, -1):
|
|
if task_records.config[i]["id"] in record_ids:
|
|
del task_records.config[i]
|
|
|
|
task_records.save_config()
|
|
return json_response(status=True, msg="Cleared successfully")
|
|
|
|
@staticmethod
|
|
def get_task_template_list(get=None):
|
|
# 缓存
|
|
cache_key = 'mod_task:get_task_template_list'
|
|
cache = public.cache_get(cache_key)
|
|
if cache:
|
|
return json_response(status=True, data=cache)
|
|
|
|
res = []
|
|
p_sys = PushSystem()
|
|
for i in TaskTemplateConfig().config:
|
|
if not i['used']:
|
|
continue
|
|
|
|
to = p_sys.get_task_object(i["id"], i["load_cls"])
|
|
if not to:
|
|
continue
|
|
|
|
# 以下模板，只允许在安全模块中使用
|
|
if i['id'] in ['121', '122', '123', '124']:
|
|
continue
|
|
|
|
if str(i.get('id', '')) == '70':
|
|
# 废弃 ssl push
|
|
continue
|
|
|
|
t = to.filter_template(i["template"])
|
|
if not t:
|
|
continue
|
|
|
|
i["template"] = t
|
|
res.append(i)
|
|
|
|
res.sort(key=lambda x: x["title"])
|
|
# 缓存30s
|
|
public.cache_set(cache_key, res, 30)
|
|
return json_response(status=True, data=res)
|
|
|
|
@staticmethod
|
|
def get_view_msg_format(task: dict) -> str:
|
|
from mod.base.push_mod.rsync_push import ViewMsgFormat as Rv
|
|
from mod.base.push_mod.site_push import ViewMsgFormat as Sv
|
|
from mod.base.push_mod.task_manager_push import ViewMsgFormat as Tv
|
|
from mod.base.push_mod.database_push import ViewMsgFormat as Dv
|
|
from mod.base.push_mod.system_push import ViewMsgFormat as SSv
|
|
from mod.base.push_mod.load_push import ViewMsgFormat as Lv
|
|
from mod.base.push_mod.domain_blcheck_push import ViewMsgFormat as DBv
|
|
from mod.base.push_mod.safe_mod_push import ViewMsgFormat as SAv
|
|
|
|
list_obj = [Rv(), Sv(), Tv(), Dv(), SSv(), Lv(), DBv(), SAv()]
|
|
for i in list_obj:
|
|
res = i.get_msg(task)
|
|
if res is not None:
|
|
return res
|
|
return '<span>--</span>'
|