201 lines
6.6 KiB
Python
201 lines
6.6 KiB
Python
import json
|
||
import re
|
||
from typing import Optional, Union, List, Tuple, Any
|
||
|
||
from app.core.context import MediaInfo, Context
|
||
from app.core.config import settings
|
||
from app.log import logger
|
||
from app.modules import _ModuleBase
|
||
from app.modules.slack.slack import Slack
|
||
|
||
|
||
class SlackModule(_ModuleBase):
|
||
slack: Slack = None
|
||
|
||
def init_module(self) -> None:
|
||
self.slack = Slack()
|
||
|
||
def stop(self):
|
||
self.slack.stop()
|
||
|
||
def init_setting(self) -> Tuple[str, Union[str, bool]]:
|
||
return "MESSAGER", "slack"
|
||
|
||
def message_parser(self, body: Any, form: Any, args: Any) -> Optional[dict]:
|
||
"""
|
||
解析消息内容，返回字典，注意以下约定值：
|
||
userid: 用户ID
|
||
username: 用户名
|
||
text: 内容
|
||
:param body: 请求体
|
||
:param form: 表单
|
||
:param args: 参数
|
||
:return: 消息内容、用户ID
|
||
"""
|
||
"""
|
||
# 消息
|
||
{
|
||
'client_msg_id': '',
|
||
'type': 'message',
|
||
'text': 'hello',
|
||
'user': '',
|
||
'ts': '1670143568.444289',
|
||
'blocks': [{
|
||
'type': 'rich_text',
|
||
'block_id': 'i2j+',
|
||
'elements': [{
|
||
'type': 'rich_text_section',
|
||
'elements': [{
|
||
'type': 'text',
|
||
'text': 'hello'
|
||
}]
|
||
}]
|
||
}],
|
||
'team': '',
|
||
'client': '',
|
||
'event_ts': '1670143568.444289',
|
||
'channel_type': 'im'
|
||
}
|
||
# 快捷方式
|
||
{
|
||
"type": "shortcut",
|
||
"token": "XXXXXXXXXXXXX",
|
||
"action_ts": "1581106241.371594",
|
||
"team": {
|
||
"id": "TXXXXXXXX",
|
||
"domain": "shortcuts-test"
|
||
},
|
||
"user": {
|
||
"id": "UXXXXXXXXX",
|
||
"username": "aman",
|
||
"team_id": "TXXXXXXXX"
|
||
},
|
||
"callback_id": "shortcut_create_task",
|
||
"trigger_id": "944799105734.773906753841.38b5894552bdd4a780554ee59d1f3638"
|
||
}
|
||
# 按钮点击
|
||
{
|
||
"type": "block_actions",
|
||
"team": {
|
||
"id": "T9TK3CUKW",
|
||
"domain": "example"
|
||
},
|
||
"user": {
|
||
"id": "UA8RXUSPL",
|
||
"username": "jtorrance",
|
||
"team_id": "T9TK3CUKW"
|
||
},
|
||
"api_app_id": "AABA1ABCD",
|
||
"token": "9s8d9as89d8as9d8as989",
|
||
"container": {
|
||
"type": "message_attachment",
|
||
"message_ts": "1548261231.000200",
|
||
"attachment_id": 1,
|
||
"channel_id": "CBR2V3XEX",
|
||
"is_ephemeral": false,
|
||
"is_app_unfurl": false
|
||
},
|
||
"trigger_id": "12321423423.333649436676.d8c1bb837935619ccad0f624c448ffb3",
|
||
"client": {
|
||
"id": "CBR2V3XEX",
|
||
"name": "review-updates"
|
||
},
|
||
"message": {
|
||
"bot_id": "BAH5CA16Z",
|
||
"type": "message",
|
||
"text": "This content can't be displayed.",
|
||
"user": "UAJ2RU415",
|
||
"ts": "1548261231.000200",
|
||
...
|
||
},
|
||
"response_url": "https://hooks.slack.com/actions/AABA1ABCD/1232321423432/D09sSasdasdAS9091209",
|
||
"actions": [
|
||
{
|
||
"action_id": "WaXA",
|
||
"block_id": "=qXel",
|
||
"text": {
|
||
"type": "plain_text",
|
||
"text": "View",
|
||
"emoji": true
|
||
},
|
||
"value": "click_me_123",
|
||
"type": "button",
|
||
"action_ts": "1548426417.840180"
|
||
}
|
||
]
|
||
}
|
||
"""
|
||
# 校验token
|
||
token = args.get("token")
|
||
if not token or token != settings.API_TOKEN:
|
||
return None
|
||
try:
|
||
msg_json: dict = json.loads(body)
|
||
except Exception as err:
|
||
logger.error(f"解析Slack消息失败：{err}")
|
||
return None
|
||
if msg_json:
|
||
if msg_json.get("type") == "message":
|
||
userid = msg_json.get("user")
|
||
text = msg_json.get("text")
|
||
username = msg_json.get("user")
|
||
elif msg_json.get("type") == "block_actions":
|
||
userid = msg_json.get("user", {}).get("id")
|
||
text = msg_json.get("actions")[0].get("value")
|
||
username = msg_json.get("user", {}).get("name")
|
||
elif msg_json.get("type") == "event_callback":
|
||
userid = msg_json.get('event', {}).get('user')
|
||
text = re.sub(r"<@[0-9A-Z]+>", "", msg_json.get("event", {}).get("text"), flags=re.IGNORECASE).strip()
|
||
username = ""
|
||
elif msg_json.get("type") == "shortcut":
|
||
userid = msg_json.get("user", {}).get("id")
|
||
text = msg_json.get("callback_id")
|
||
username = msg_json.get("user", {}).get("username")
|
||
else:
|
||
return None
|
||
logger.info(f"收到Slack消息：userid={userid}, username={username}, text={text}")
|
||
return {
|
||
"userid": userid,
|
||
"username": username,
|
||
"text": text
|
||
}
|
||
return None
|
||
|
||
def post_message(self, title: str,
|
||
text: str = None, image: str = None,
|
||
userid: Union[str, int] = None) -> Optional[bool]:
|
||
"""
|
||
发送消息
|
||
:param title: 标题
|
||
:param text: 内容
|
||
:param image: 图片
|
||
:param userid: 用户ID
|
||
:return: 成功或失败
|
||
"""
|
||
return self.slack.send_msg(title=title, text=text, image=image, userid=userid)
|
||
|
||
def post_medias_message(self, title: str, items: List[MediaInfo],
|
||
userid: Union[str, int] = None) -> Optional[bool]:
|
||
"""
|
||
发送媒体信息选择列表
|
||
:param title: 标题
|
||
:param items: 消息列表
|
||
:param userid: 用户ID
|
||
:return: 成功或失败
|
||
"""
|
||
return self.slack.send_meidas_msg(title=title, medias=items, userid=userid)
|
||
|
||
def post_torrents_message(self, title: str, items: List[Context],
|
||
mediainfo: MediaInfo = None,
|
||
userid: Union[str, int] = None) -> Optional[bool]:
|
||
"""
|
||
发送种子信息选择列表
|
||
:param title: 标题
|
||
:param items: 消息列表
|
||
:param mediainfo: 媒体信息
|
||
:param userid: 用户ID
|
||
:return: 成功或失败
|
||
"""
|
||
return self.slack.send_torrents_msg(title=title, torrents=items,
|
||
userid=userid)
|