|
|
|
@ -1,3 +1,4 @@
|
|
|
|
|
import io
|
|
|
|
|
import re
|
|
|
|
|
import time
|
|
|
|
|
import requests
|
|
|
|
@ -13,7 +14,7 @@ else:
|
|
|
|
|
NewMessage = TypeVar("NewMessage")
|
|
|
|
|
IcaClient = TypeVar("IcaClient")
|
|
|
|
|
|
|
|
|
|
_version_ = "2.1.2-rs"
|
|
|
|
|
_version_ = "2.2.0-rs"
|
|
|
|
|
|
|
|
|
|
def format_data_size(data_bytes: float) -> str:
|
|
|
|
|
data_lens = ["B", "KB", "MB", "GB", "TB"]
|
|
|
|
@ -82,7 +83,6 @@ def wrap_request(url: str, msg: NewMessage, client: IcaClient) -> Optional[dict]
|
|
|
|
|
return response.json()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def bmcl_dashboard(msg: NewMessage, client: IcaClient) -> None:
|
|
|
|
|
req_time = time.time()
|
|
|
|
|
# 记录请求时间
|
|
|
|
@ -111,56 +111,102 @@ def bmcl_dashboard(msg: NewMessage, client: IcaClient) -> None:
|
|
|
|
|
client.send_message(reply)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def parse_rank(data: dict) -> dict:
|
|
|
|
|
rank_data = {"hits": 0, "bytes": 0}
|
|
|
|
|
if "metric" in data:
|
|
|
|
|
rank_data["hits"] = data["metric"]["hits"]
|
|
|
|
|
rank_data["bytes"] = data["metric"]["bytes"]
|
|
|
|
|
return {
|
|
|
|
|
"name": data["name"],
|
|
|
|
|
"start": data["isEnabled"],
|
|
|
|
|
# "full": "全量" if "fullSize" in data else "分片",
|
|
|
|
|
# "version": data["version"] if "version" in data else "未知版本",
|
|
|
|
|
"owner": data["sponsor"]["name"] if "sponsor" in data else "未知",
|
|
|
|
|
"rank": rank_data
|
|
|
|
|
}
|
|
|
|
|
def check_is_full_data(data: list) -> bool:
|
|
|
|
|
return 'user' in data[0]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def bmcl_rank(msg: NewMessage, client: IcaClient, name: Optional[str]) -> None:
|
|
|
|
|
def display_rank_min(ranks: list, req_time) -> str:
|
|
|
|
|
cache = io.StringIO()
|
|
|
|
|
cache.write(f"bmclapi v{_version_}-排名({len(ranks)})")
|
|
|
|
|
if check_is_full_data(ranks):
|
|
|
|
|
cache.write("完整\n")
|
|
|
|
|
for rank in ranks:
|
|
|
|
|
cache.write('✅' if rank['isEnabled'] else '❌')
|
|
|
|
|
if 'fullSize' in rank:
|
|
|
|
|
cache.write('🌕' if rank['fullSize'] else '🌘')
|
|
|
|
|
cache.write(f"-{rank['index']+1:3}")
|
|
|
|
|
cache.write(f"|{rank['name']}\n")
|
|
|
|
|
else:
|
|
|
|
|
cache.write("无cookie\n")
|
|
|
|
|
for rank in ranks:
|
|
|
|
|
cache.write('✅' if rank['isEnabled'] else '❌')
|
|
|
|
|
cache.write(f"-{rank['index']+1:3}")
|
|
|
|
|
cache.write(f"|{rank['name']}\n")
|
|
|
|
|
cache.write(f"请求时间: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(req_time))}")
|
|
|
|
|
return cache.getvalue()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def display_rank_full(ranks: list, req_time) -> str:
|
|
|
|
|
cache = io.StringIO()
|
|
|
|
|
cache.write(f"bmclapi v{_version_}-排名({len(ranks)})")
|
|
|
|
|
if check_is_full_data(ranks):
|
|
|
|
|
cache.write("完整\n")
|
|
|
|
|
for rank in ranks:
|
|
|
|
|
# 基本信息
|
|
|
|
|
cache.write('✅' if rank['isEnabled'] else '❌')
|
|
|
|
|
if 'fullSize' in rank:
|
|
|
|
|
cache.write('🌕' if rank['fullSize'] else '🌘')
|
|
|
|
|
cache.write(f"|{rank['index']+1:3}|")
|
|
|
|
|
cache.write(f"{rank['name']}")
|
|
|
|
|
if 'version' in rank:
|
|
|
|
|
cache.write(f"|{rank['version']}")
|
|
|
|
|
cache.write('\n')
|
|
|
|
|
# 用户/赞助信息
|
|
|
|
|
if ('user' in rank) and (rank['user'] is not None):
|
|
|
|
|
cache.write(f"所有者:{rank['user']['name']}")
|
|
|
|
|
if 'sponsor' in rank:
|
|
|
|
|
cache.write(f"|赞助者:{rank['sponsor']['name']}")
|
|
|
|
|
if 'sponsor' in rank or ('user' in rank and rank['user'] is not None):
|
|
|
|
|
cache.write('\n')
|
|
|
|
|
# 数据信息
|
|
|
|
|
if 'metric' in rank:
|
|
|
|
|
hits = format_hit_count(rank['metric']['hits'])
|
|
|
|
|
data = format_data_size(rank['metric']['bytes'])
|
|
|
|
|
cache.write(f"hit/data|{hits}|{data}")
|
|
|
|
|
cache.write('\n')
|
|
|
|
|
else:
|
|
|
|
|
cache.write("无cookie\n")
|
|
|
|
|
for rank in ranks:
|
|
|
|
|
cache.write('✅' if rank['isEnabled'] else '❌')
|
|
|
|
|
cache.write(f"-{rank['index']+1:3}")
|
|
|
|
|
cache.write(f"|{rank['name']}|\n")
|
|
|
|
|
if 'sponsor' in rank:
|
|
|
|
|
cache.write(f"赞助者: {rank['sponsor']['name']}|")
|
|
|
|
|
if 'metric' in rank:
|
|
|
|
|
cache.write(f"hit/data|{format_hit_count(rank['metric']['hits'])}|{format_data_size(rank['metric']['bytes'])}\n")
|
|
|
|
|
cache.write(f"请求时间: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(req_time))}")
|
|
|
|
|
return cache.getvalue()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def bmcl_rank_general(msg, client):
|
|
|
|
|
req_time = time.time()
|
|
|
|
|
# 记录请求时间
|
|
|
|
|
rank_data = wrap_request("https://bd.bangbang93.com/openbmclapi/metric/rank", msg, client)
|
|
|
|
|
if rank_data is None:
|
|
|
|
|
return
|
|
|
|
|
ranks = [parse_rank(data) for data in rank_data]
|
|
|
|
|
if name is None:
|
|
|
|
|
# 全部排名
|
|
|
|
|
# 预处理数据
|
|
|
|
|
for i, r in enumerate(rank_data):
|
|
|
|
|
r['index'] = i
|
|
|
|
|
# 显示前3名
|
|
|
|
|
limit = 3
|
|
|
|
|
if len(ranks) < limit:
|
|
|
|
|
show_ranks = ranks
|
|
|
|
|
else:
|
|
|
|
|
show_ranks = ranks[:limit]
|
|
|
|
|
rank_msg = (
|
|
|
|
|
f"{'✅' if r['start'] else '❌'}名称: {r['name']}\n"
|
|
|
|
|
# f"-{rank['full']} \n"
|
|
|
|
|
# f"版本: {r['version']}\n"
|
|
|
|
|
f"赞助者: {r['owner']}|"
|
|
|
|
|
f"h/d {format_hit_count(r['rank']['hits'])}|{format_data_size(r['rank']['bytes'])}"
|
|
|
|
|
for r in show_ranks
|
|
|
|
|
)
|
|
|
|
|
rank_msg = "\n".join(rank_msg)
|
|
|
|
|
report_msg = (
|
|
|
|
|
f"OpenBMCLAPI 面板v{_version_}-排名\n{rank_msg}\n"
|
|
|
|
|
f"请求时间: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(req_time))}\n"
|
|
|
|
|
)
|
|
|
|
|
reply = msg.reply_with(report_msg)
|
|
|
|
|
ranks = rank_data[:3]
|
|
|
|
|
# ranks = rank_data
|
|
|
|
|
report_msg = display_rank_full(ranks, req_time)
|
|
|
|
|
client.info(report_msg)
|
|
|
|
|
reply = msg.reply_with(display_rank_full(ranks, req_time))
|
|
|
|
|
client.send_message(reply)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def bmcl_rank(msg: NewMessage, client: IcaClient, name: str) -> None:
|
|
|
|
|
req_time = time.time()
|
|
|
|
|
# 记录请求时间
|
|
|
|
|
rank_data = wrap_request("https://bd.bangbang93.com/openbmclapi/metric/rank", msg, client)
|
|
|
|
|
if rank_data is None:
|
|
|
|
|
return
|
|
|
|
|
else:
|
|
|
|
|
# 预处理数据
|
|
|
|
|
for i, r in enumerate(rank_data):
|
|
|
|
|
r['index'] = i
|
|
|
|
|
# 搜索是否有这个名字的节点
|
|
|
|
|
names = [r["name"].lower() for r in ranks]
|
|
|
|
|
names = [r["name"].lower() for r in rank_data]
|
|
|
|
|
try:
|
|
|
|
|
finds = [re.search(name.lower(), n) for n in names]
|
|
|
|
|
except re.error as e:
|
|
|
|
@ -172,38 +218,22 @@ def bmcl_rank(msg: NewMessage, client: IcaClient, name: Optional[str]) -> None:
|
|
|
|
|
client.send_message(reply)
|
|
|
|
|
return
|
|
|
|
|
# 如果找到 > 3 个节点, 则提示 不显示
|
|
|
|
|
counts = [True for find in finds if find]
|
|
|
|
|
counts = [f for f in finds if f]
|
|
|
|
|
ranks = [rank_data[i] for i, f in enumerate(finds) if f]
|
|
|
|
|
if len(counts) > 3:
|
|
|
|
|
if len(counts) > 10:
|
|
|
|
|
reply = msg.reply_with(f"搜索|{name}|到{len(counts)}个节点, 请用更精确的名字")
|
|
|
|
|
else:
|
|
|
|
|
# 4~10 个节点 只显示名称和次序
|
|
|
|
|
find_msg = [f"{'✅' if r['start'] else '❌'}{r['name']}-No.{i+1}" for i, r in enumerate(ranks) if finds[i]]
|
|
|
|
|
find_msg = "\n".join(find_msg)
|
|
|
|
|
report_msg = f"OpenBMCLAPI 面板v{_version_}-搜索|{name}|\n{find_msg}"
|
|
|
|
|
report_msg = display_rank_min(ranks, req_time)
|
|
|
|
|
reply = msg.reply_with(report_msg)
|
|
|
|
|
client.send_message(reply)
|
|
|
|
|
return
|
|
|
|
|
rank_msgs = []
|
|
|
|
|
for i, find in enumerate(finds):
|
|
|
|
|
if find:
|
|
|
|
|
rank = ranks[i]
|
|
|
|
|
rank_msg = (
|
|
|
|
|
f"{'✅' if rank['start'] else '❌'}名称: {rank['name']}-No.{i+1}\n"
|
|
|
|
|
# f"-{rank['full']} \n"
|
|
|
|
|
# f"版本: {rank['version']}\n"
|
|
|
|
|
f"赞助商: {rank['owner']}|"
|
|
|
|
|
f"h/d {format_hit_count(rank['rank']['hits'])}|{format_data_size(rank['rank']['bytes'])}"
|
|
|
|
|
)
|
|
|
|
|
rank_msgs.append(rank_msg)
|
|
|
|
|
rank_msgs = "\n".join(rank_msgs)
|
|
|
|
|
report_msg = f"OpenBMCLAPI 面板v{_version_}-排名\n{rank_msgs}"
|
|
|
|
|
reply = msg.reply_with(report_msg)
|
|
|
|
|
# 如果找到 <= 3 个节点, 则显示全部信息
|
|
|
|
|
report_msg = display_rank_full(ranks, req_time)
|
|
|
|
|
client.info(report_msg)
|
|
|
|
|
reply = msg.reply_with(report_msg)
|
|
|
|
|
client.send_message(reply)
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
help = """/bmcl -> dashboard
|
|
|
|
@ -219,11 +249,14 @@ help = """/bmcl -> dashboard
|
|
|
|
|
|
|
|
|
|
def on_message(msg: NewMessage, client: IcaClient) -> None:
|
|
|
|
|
if not (msg.is_from_self or msg.is_reply):
|
|
|
|
|
if '\n' in msg.content:
|
|
|
|
|
return
|
|
|
|
|
try:
|
|
|
|
|
if msg.content.startswith("/bmcl"):
|
|
|
|
|
if msg.content == "/bmcl":
|
|
|
|
|
bmcl_dashboard(msg, client)
|
|
|
|
|
elif msg.content == "/bmcl rank":
|
|
|
|
|
bmcl_rank(msg, client, None)
|
|
|
|
|
bmcl_rank_general(msg, client)
|
|
|
|
|
elif msg.content.startswith("/bmcl rank") and len(msg.content) > 11:
|
|
|
|
|
name = msg.content[11:]
|
|
|
|
|
bmcl_rank(msg, client, name)
|
|
|
|
@ -239,6 +272,10 @@ def on_message(msg: NewMessage, client: IcaClient) -> None:
|
|
|
|
|
if len(name) > 1:
|
|
|
|
|
name = name[1]
|
|
|
|
|
bmcl_rank(msg, client, name)
|
|
|
|
|
except: # noqa
|
|
|
|
|
report_msg = f"bmcl插件发生错误,请呼叫shenjack\n{traceback.format_exc()}"
|
|
|
|
|
reply = msg.reply_with(report_msg)
|
|
|
|
|
client.send_and_warn(reply)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def on_config() -> Tuple[str, str]:
|
|
|
|
|