mirror of
http://shenjack.top:5100/shenjack/icalingua-python-bot.git
synced 2024-11-23 04:31:05 +08:00
remove ica-py
This commit is contained in:
parent
8dd5c0b7d4
commit
751402d5c6
GPG Key ID:
7B1134A979775551
|
|
@ -1,161 +0,0 @@
|
|||
import time
|
||||
import json
|
||||
import random
|
||||
import asyncio
|
||||
import argparse
|
||||
import traceback
|
||||
|
||||
from typing import Dict, List, Tuple, Any
|
||||
|
||||
import socketio
|
||||
from colorama import Fore
|
||||
from nacl.signing import SigningKey
|
||||
|
||||
from lib_not_dr.loggers import config
|
||||
|
||||
from data_struct import SendMessage, ReplyMessage, get_config, BotConfig, BotStatus
|
||||
from main import BOTCONFIG, _version_
|
||||
from router import route
|
||||
|
||||
logger = config.get_logger("icalingua")
|
||||
|
||||
sio: socketio.AsyncClient = socketio.AsyncClient()
|
||||
|
||||
|
||||
@sio.on("connect") # type: ignore
|
||||
def connect():
|
||||
logger.info(f"{Fore.GREEN}icalingua 已连接")
|
||||
|
||||
|
||||
@sio.on("requireAuth") # type: ignore
|
||||
async def require_auth(salt: str, versions: Dict[str, str]):
|
||||
logger.info(f"{Fore.BLUE}versions: {versions}\n{type(salt)}|{salt=}")
|
||||
# 准备数据
|
||||
sign = SigningKey(bytes.fromhex(BOTCONFIG.private_key))
|
||||
signature = sign.sign(bytes.fromhex(salt))
|
||||
|
||||
# 发送数据
|
||||
await sio.emit("auth", signature.signature)
|
||||
logger.info(f"{Fore.BLUE}send auth emit")
|
||||
|
||||
|
||||
@sio.on("auth") # type: ignore
|
||||
def auth(data: Dict[str, Any]):
|
||||
logger.info(f"auth: {data}")
|
||||
|
||||
|
||||
@sio.on("authFailed") # type: ignore
|
||||
async def auth_failed():
|
||||
logger.info(f"{Fore.RED}authFailed")
|
||||
await sio.disconnect()
|
||||
|
||||
|
||||
@sio.on("authSucceed") # type: ignore
|
||||
def auth_succeed():
|
||||
logger.info(f"{Fore.GREEN}authSucceed")
|
||||
|
||||
|
||||
@sio.on("connect_error") # type: ignore
|
||||
def connect_error(*args, **kwargs):
|
||||
logger.info(f"连接错误 {args}, {kwargs}")
|
||||
|
||||
|
||||
@sio.on("updateRoom") # type: ignore
|
||||
def update_room(data: Dict[str, Any]):
|
||||
logger.info(f"{Fore.CYAN}update_room: {data}")
|
||||
|
||||
|
||||
@sio.on("addMessage") # type: ignore
|
||||
async def add_message(data: Dict[str, Any]):
|
||||
logger.info(f"{Fore.MAGENTA}add_message: {data}")
|
||||
|
||||
is_self = data["message"]["senderId"] == BOTCONFIG.self_id
|
||||
|
||||
if not is_self:
|
||||
await route(data, sio)
|
||||
|
||||
|
||||
@sio.on("deleteMessage") # type: ignore
|
||||
def delete_message(message_id: str):
|
||||
logger.info(f"{Fore.MAGENTA}delete_message: {message_id}")
|
||||
|
||||
|
||||
@sio.on("setMessages") # type: ignore
|
||||
def set_messages(data: Dict[str, Any]):
|
||||
logger.info(
|
||||
f"{Fore.YELLOW}set_messages: {data}\nmessage_len: {len(data['messages'])}"
|
||||
)
|
||||
|
||||
|
||||
async def notice_startup(room_list: List[int]):
|
||||
for notice_room in BOTCONFIG.notice_room:
|
||||
if notice_room in room_list:
|
||||
notice_message = SendMessage(
|
||||
content=f"ica bot v{_version_}", room_id=notice_room
|
||||
)
|
||||
await sio.emit("sendMessage", notice_message.to_json())
|
||||
BotStatus.inited = True
|
||||
logger.info("inited", tag="notice room")
|
||||
else:
|
||||
logger.warn(f"未找到通知房间: {notice_room}", tag="notice room")
|
||||
await asyncio.sleep(random.randint(2, 5))
|
||||
|
||||
|
||||
@sio.on("setAllRooms") # type: ignore
|
||||
async def set_all_rooms(rooms: List[Dict[str, Any]]):
|
||||
BotStatus.running = True
|
||||
room_list: List[int] = [room.get("roomId") for room in rooms] # type: ignore
|
||||
if not BotStatus.inited:
|
||||
logger.info("initing...", tag="setAllRooms")
|
||||
logger.debug(f"room_list: {room_list}", tag="setAllRooms")
|
||||
if BOTCONFIG.notice_start:
|
||||
await notice_startup(room_list)
|
||||
if room_list != BotStatus.rooms:
|
||||
logger.info(f"{Fore.YELLOW}set_all_rooms: {rooms}\nlen: {len(rooms)}\n")
|
||||
BotStatus.rooms = room_list
|
||||
logger.info(f"更新房间: {room_list}", tag="setAllRooms")
|
||||
|
||||
|
||||
@sio.on("setAllChatGroups") # type: ignore
|
||||
def set_all_chat_groups(groups: List[Dict[str, Any]]):
|
||||
logger.info(f"{Fore.YELLOW}set_all_chat_groups: {groups}\nlen: {len(groups)}\n")
|
||||
|
||||
|
||||
@sio.on("notify") # type: ignore
|
||||
def notify(data: List[Tuple[str, Any]]):
|
||||
logger.info(f"notify: {data}")
|
||||
|
||||
|
||||
@sio.on("closeLoading") # type: ignore
|
||||
def close_loading(_):
|
||||
logger.info(f"{Fore.GREEN}close_loading")
|
||||
|
||||
|
||||
@sio.on("onlineData") # type: ignore
|
||||
def online_data(data: Dict[str, Any]):
|
||||
logger.info(f"{Fore.GREEN}online_data: {data}")
|
||||
|
||||
|
||||
@sio.on("*") # type: ignore
|
||||
def catch_all(event, data):
|
||||
logger.info(f"{Fore.RED}catch_all: {event}|{data}")
|
||||
|
||||
|
||||
async def main():
|
||||
"""
|
||||
while True:
|
||||
await self.eio.wait()
|
||||
await self.sleep(1) # give the reconnect task time to start up
|
||||
if not self._reconnect_task:
|
||||
break
|
||||
await self._reconnect_task
|
||||
if self.eio.state != 'connected':
|
||||
break
|
||||
"""
|
||||
try:
|
||||
await sio.connect(BOTCONFIG.host)
|
||||
await sio.wait()
|
||||
except KeyboardInterrupt:
|
||||
logger.info("KeyboardInterrupt")
|
||||
except Exception:
|
||||
logger.error(traceback.format_exc())
|
||||
|
|
@ -1,102 +0,0 @@
|
|||
from typing import List, Optional, Union, Literal, Tuple
|
||||
|
||||
import qtoml
|
||||
from lib_not_dr.types import Options
|
||||
|
||||
|
||||
class AtElement(Options):
|
||||
text: str
|
||||
id: Union[int, Literal['all']] = 'all'
|
||||
|
||||
|
||||
class ReplyMessage(Options):
|
||||
id: str
|
||||
username: str = ''
|
||||
content: str = ''
|
||||
files: list = []
|
||||
|
||||
def to_json(self) -> dict:
|
||||
return {
|
||||
'_id': self.id,
|
||||
'username': self.username,
|
||||
'content': self.content,
|
||||
'files': self.files
|
||||
}
|
||||
|
||||
|
||||
class SendMessage(Options):
|
||||
content: str
|
||||
room_id: Optional[int] = None
|
||||
room: Optional[int] = None # room id 和 room 二选一 ( 实际上直接填 room id 就行了 )
|
||||
file: None = None # TODO: 上传文件
|
||||
reply_to: Optional[ReplyMessage] = None # 源码 给了一个 any TODO: 回复消息
|
||||
b64_img: Optional[str] = None # TODO: 发送图片
|
||||
at: Optional[List[AtElement]] = [] # TODO: @某人
|
||||
sticker: Optional[None] = None # TODO: 发送表情
|
||||
message_type: Optional[str] = None # TODO: 消息类型
|
||||
|
||||
def to_json(self) -> dict:
|
||||
return {
|
||||
'content': self.content,
|
||||
'roomId': self.room_id,
|
||||
'room': self.room,
|
||||
'file': self.file,
|
||||
'replyMessage': self.reply_to.to_json() if self.reply_to else None,
|
||||
'b64img': self.b64_img,
|
||||
'at': self.at,
|
||||
'sticker': self.sticker,
|
||||
'messageType': self.message_type
|
||||
}
|
||||
|
||||
def to_content(self, content: str) -> "SendMessage":
|
||||
self.content = content
|
||||
return self
|
||||
|
||||
|
||||
class NewMessage(Options):
|
||||
sender_id: int
|
||||
sender_name: str
|
||||
room_id: int
|
||||
content: str
|
||||
msg_id: str
|
||||
data: dict
|
||||
|
||||
def init(self, **kwargs) -> None:
|
||||
data = kwargs.pop('data')
|
||||
|
||||
self.sender_name = data["message"]["username"]
|
||||
self.sender_id = data["message"]["senderId"]
|
||||
self.content = data["message"]["content"]
|
||||
self.room_id = data["roomId"]
|
||||
self.msg_id = data["message"]["_id"]
|
||||
|
||||
def is_self(self, self_id: int) -> bool:
|
||||
return self.sender_id == self_id
|
||||
|
||||
|
||||
class BotConfig(Options):
|
||||
name = 'icalingua bot config'
|
||||
# _check_filled = True
|
||||
private_key: str
|
||||
host: str
|
||||
self_id: int
|
||||
notice_room: List[int]
|
||||
notice_start: bool = False
|
||||
admin_list: List[int]
|
||||
py_plugin_path: str
|
||||
|
||||
def init(self, **kwargs) -> None:
|
||||
if self.notice_room is None:
|
||||
self.notice_start = False
|
||||
|
||||
|
||||
def get_config(config_path: str = 'config.toml') -> BotConfig:
|
||||
with open(config_path, 'r', encoding='utf-8') as f:
|
||||
config = qtoml.decoder.load(f)
|
||||
return BotConfig(**config)
|
||||
|
||||
|
||||
class BotStatus(Options):
|
||||
inited: bool = False
|
||||
running: bool = False
|
||||
rooms: List[int] = []
|
||||
|
|
@ -1,36 +0,0 @@
|
|||
import asyncio
|
||||
import argparse
|
||||
|
||||
# from lib_not_dr.types import Options
|
||||
from lib_not_dr.loggers import config
|
||||
|
||||
from data_struct import get_config, BotConfig, BotStatus
|
||||
|
||||
_version_ = "0.3.3"
|
||||
|
||||
logger = config.get_logger("bot")
|
||||
|
||||
BOTCONFIG: BotConfig = get_config()
|
||||
BotStatus = BotStatus()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# --debug
|
||||
# --config=config.toml
|
||||
# -n --no-notice
|
||||
parser = argparse.ArgumentParser(description=f"icalingua bot v{_version_}")
|
||||
parser.add_argument("-d", "--debug", action="store_true")
|
||||
parser.add_argument("-n", "--no-notice", action="store_true")
|
||||
parser.add_argument("-c", "--config", type=str)
|
||||
args = parser.parse_args()
|
||||
if args.debug:
|
||||
logger.global_level = 0
|
||||
if args.config:
|
||||
# global BOTCONFIG
|
||||
BOTCONFIG: BotConfig = get_config(args.config)
|
||||
if args.no_notice:
|
||||
BOTCONFIG.notice_start = False
|
||||
|
||||
from connect import main
|
||||
asyncio.run(main())
|
||||
|
||||
|
|
@ -1,98 +0,0 @@
|
|||
import time
|
||||
import json
|
||||
import aiohttp
|
||||
|
||||
from lib_not_dr.loggers import config
|
||||
|
||||
from data_struct import NewMessage, SendMessage
|
||||
|
||||
logger = config.get_logger("bmcl")
|
||||
|
||||
_version_ = "1.1.1"
|
||||
|
||||
|
||||
def format_data_size(data_bytes: float) -> str:
|
||||
data_lens = ["B", "KB", "MB", "GB", "TB"]
|
||||
data_len = "0B"
|
||||
for i in range(5):
|
||||
if data_bytes < 1024:
|
||||
data_bytes = round(data_bytes, 5)
|
||||
data_len = f"{data_bytes}{data_lens[i]}"
|
||||
break
|
||||
else:
|
||||
data_bytes /= 1024
|
||||
return data_len
|
||||
|
||||
|
||||
def format_hit_count(count: int) -> str:
|
||||
"""数据分段, 四位一个下划线
|
||||
|
||||
Args:
|
||||
count (int): 数据
|
||||
|
||||
Returns:
|
||||
str: 格式化后的数据
|
||||
1 -> 1
|
||||
1000 -> 1000
|
||||
10000 -> 1_0000
|
||||
100000 -> 10_0000
|
||||
1000000 -> 100_0000
|
||||
"""
|
||||
count_str = str(count)
|
||||
count_len = len(count_str)
|
||||
if count_len <= 4:
|
||||
return count_str
|
||||
else:
|
||||
return "_".join(count_str[i:i + 4] for i in range(0, count_len, 4))
|
||||
|
||||
|
||||
async def bmcl(sio, reply_msg: SendMessage, msg: NewMessage):
|
||||
req_time = time.time()
|
||||
# 记录请求时间
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.get(
|
||||
"https://bd.bangbang93.com/openbmclapi/metric/dashboard"
|
||||
) as response:
|
||||
if not response.status == 200 or response.reason != "OK":
|
||||
await sio.emit(
|
||||
"sendMessage",
|
||||
reply_msg.to_content(
|
||||
f"请求数据失败\n{response.status}"
|
||||
).to_json(),
|
||||
)
|
||||
logger.warn(
|
||||
f"数据请求失败, 请检查网络\n{response.status}",
|
||||
tag="bmclapi_dashboard",
|
||||
)
|
||||
return
|
||||
raw_data = await response.text()
|
||||
try:
|
||||
data = json.loads(raw_data)
|
||||
data_bytes: float = data["bytes"]
|
||||
data_hits: int = data["hits"]
|
||||
data_bandwidth: float = data["currentBandwidth"]
|
||||
load_str: float = data["load"] * 100
|
||||
online_node: int = data["currentNodes"]
|
||||
online_bandwidth: int = data["bandwidth"]
|
||||
data_len = format_data_size(data_bytes)
|
||||
hits_count = format_hit_count(data_hits)
|
||||
|
||||
report_msg = (
|
||||
f"OpenBMCLAPI 状态面板v{_version_} :\n"
|
||||
f"实时信息: {online_node} 带宽: {online_bandwidth}Mbps\n"
|
||||
f"负载: {load_str:.2f}% 带宽: {data_bandwidth:.2f}Mbps\n"
|
||||
f"当日请求: {hits_count} 数据量: {data_len}\n"
|
||||
f"请求时间: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(req_time))}\n"
|
||||
"数据源: https://bd.bangbang93.com/pages/dashboard"
|
||||
)
|
||||
await sio.emit(
|
||||
"sendMessage",
|
||||
reply_msg.to_content(report_msg).to_json()
|
||||
)
|
||||
|
||||
except (json.JSONDecodeError, AttributeError, ValueError) as e:
|
||||
await sio.emit(
|
||||
"sendMessage",
|
||||
reply_msg.to_content(f"返回数据解析错误\n{e}").to_json(),
|
||||
)
|
||||
logger.warn(f"返回数据解析错误\n{e}", tag="bmclapi_dashboard")
|
||||
|
|
@ -1,72 +0,0 @@
|
|||
import time
|
||||
import random
|
||||
import traceback
|
||||
|
||||
from main import BOTCONFIG
|
||||
|
||||
from colorama import Fore
|
||||
from lib_not_dr.loggers import config
|
||||
|
||||
logger = config.get_logger("safe_eval")
|
||||
|
||||
def safe_eval(code: str) -> str:
|
||||
try:
|
||||
# code = code.replace('help', '坏东西！\n')
|
||||
# code = code.replace('bytes', '坏东西！\n')
|
||||
# code = code.replace('encode', '坏东西！\n')
|
||||
# code = code.replace('decode', '坏东西！\n')
|
||||
# code = code.replace('compile', '屑的！\n')
|
||||
# code = code.replace('globals', '拿不到！\n')
|
||||
code = code.replace("os", "坏东西！\n")
|
||||
code = code.replace("sys", "坏东西！\n")
|
||||
# code = code.replace('input', '坏东西！\n')
|
||||
# code = code.replace('__', '啊哈！\n')
|
||||
# code = code.replace('import', '很坏！\n')
|
||||
code = code.replace(" kill", "别跑！\n")
|
||||
code = code.replace(" rm ", "别跑！\n")
|
||||
code = code.replace("exit", "好坏！\n")
|
||||
code = code.replace("eval", "啊哈！\n")
|
||||
code = code.replace("exec", "抓住！\n")
|
||||
start_time = time.time()
|
||||
try:
|
||||
import os
|
||||
import math
|
||||
import decimal
|
||||
|
||||
global_val = {
|
||||
"time": time,
|
||||
"math": math,
|
||||
"decimal": decimal,
|
||||
"random": random,
|
||||
"__import__": "<built-in function __import__>",
|
||||
"globals": "<built-in function globals>",
|
||||
"compile": "<built-in function compile>",
|
||||
"help": "<built-in function help>",
|
||||
"exit": "<built-in function exit>",
|
||||
"input": "<built-in function input>",
|
||||
"return": "别惦记你那个 return 了",
|
||||
"getattr": "<built-in function getattr>",
|
||||
"setattr": "<built-in function setattr>",
|
||||
}
|
||||
os.system = "不许"
|
||||
result = str(eval(code, global_val, {}))
|
||||
limit = 500
|
||||
if len(result) > limit:
|
||||
result = result[:limit]
|
||||
except:
|
||||
result = traceback.format_exc()
|
||||
end_time = time.time()
|
||||
result = result.replace(BOTCONFIG.private_key, "***")
|
||||
result = result.replace(BOTCONFIG.host, "***")
|
||||
|
||||
logger.info(f"{Fore.MAGENTA}safe_eval: {result}")
|
||||
|
||||
if result == "6" or result == 6:
|
||||
result = "他确实等于 6"
|
||||
|
||||
result = f"{code}\neval result:\n{result}\n耗时: {end_time - start_time} s"
|
||||
return result
|
||||
except:
|
||||
error = traceback.format_exc()
|
||||
result = f"error:\n{error}"
|
||||
return result
|
||||
|
|
@ -1,7 +0,0 @@
|
|||
lib-not-dr >= 0.3.13
|
||||
colorama
|
||||
qtoml
|
||||
pynacl
|
||||
|
||||
python-socketio
|
||||
aiohttp
|
||||
|
|
@ -1,62 +0,0 @@
|
|||
import random
|
||||
import asyncio
|
||||
|
||||
from lib_not_dr.loggers import config
|
||||
|
||||
from main import BOTCONFIG, _version_
|
||||
from data_struct import SendMessage, ReplyMessage, NewMessage
|
||||
|
||||
from plugins import bmcl, safe_eval
|
||||
|
||||
logger = config.get_logger("router")
|
||||
|
||||
async def route(data, sio):
|
||||
|
||||
is_self = data["message"]["senderId"] == BOTCONFIG.self_id
|
||||
sender_name = data["message"]["username"]
|
||||
sender_id = data["message"]["senderId"]
|
||||
content = data["message"]["content"]
|
||||
room_id = data["roomId"]
|
||||
msg_id = data["message"]["_id"]
|
||||
msg = NewMessage(data=data)
|
||||
|
||||
reply_msg = SendMessage(content="", room_id=room_id, reply_to=ReplyMessage(id=msg_id))
|
||||
|
||||
if content == "/bot":
|
||||
message = reply_msg.to_content(f"icalingua bot-python pong v{_version_}")
|
||||
await sio.emit("sendMessage", message.to_json())
|
||||
|
||||
elif content.startswith("=="):
|
||||
evals: str = content[2:]
|
||||
|
||||
result = safe_eval.safe_eval(evals)
|
||||
# whitelist = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', ' ', '.', '+', '-', '*', '/', '(', ')', '<',
|
||||
# '>', '=']
|
||||
# evals = evals.replace('**', '')
|
||||
# express = ''
|
||||
# for text in evals:
|
||||
# if text in whitelist:
|
||||
# express += text
|
||||
# if express == '':
|
||||
# result = '你在干嘛'
|
||||
# else:
|
||||
# result = str(eval(express))
|
||||
message = reply_msg.to_content(result)
|
||||
|
||||
await asyncio.sleep(random.random() * 2)
|
||||
await sio.emit("sendMessage", message.to_json())
|
||||
|
||||
elif content == "!!jrrp":
|
||||
randomer = random.Random(
|
||||
f'{sender_id}-{data["message"]["date"]}-jrrp-{_version_}'
|
||||
)
|
||||
result = randomer.randint(0, 50) + randomer.randint(0, 50)
|
||||
logger.info(f"{sender_name} 今日人品值为 {result}")
|
||||
message = reply_msg.to_content(f"{sender_name} 今日人品为 {result}")
|
||||
await asyncio.sleep(0.5)
|
||||
await sio.emit("sendMessage", message.to_json())
|
||||
|
||||
elif content == "/bmcl":
|
||||
await bmcl.bmcl(sio, reply_msg, msg)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user