143 lines
4.2 KiB
Python
143 lines
4.2 KiB
Python
import json
|
||
import random
|
||
import asyncio
|
||
from datetime import datetime, timedelta
|
||
from nonebot import on_notice, on_message, get_bot
|
||
from nonebot.adapters.onebot.v11 import (
|
||
GroupIncreaseNoticeEvent,
|
||
MessageEvent,
|
||
Message,
|
||
Bot,
|
||
)
|
||
from nonebot.permission import SUPERUSER
|
||
|
||
BAN_FILE = "zhenxun/plugins/nb_new/ban.json"
|
||
GROUP_FILE = "zhenxun/plugins/nb_new/group.json"
|
||
VERIFY_TIMEOUT = 300 # 5分钟
|
||
LEVEL_LIMIT = 20 # QQ等级限制
|
||
|
||
# 内存缓存
|
||
verify_data = {} # user_id -> {"code": xxx, "group_id": xxx, "expire": time}
|
||
|
||
# 初始化配置文件
|
||
def load_json(path, default):
|
||
try:
|
||
with open(path, "r", encoding="utf-8") as f:
|
||
return json.load(f)
|
||
except:
|
||
with open(path, "w", encoding="utf-8") as f:
|
||
json.dump(default, f, indent=2, ensure_ascii=False)
|
||
return default
|
||
|
||
|
||
def save_json(path, data):
|
||
with open(path, "w", encoding="utf-8") as f:
|
||
json.dump(data, f, indent=2, ensure_ascii=False)
|
||
|
||
|
||
ban_list = load_json(BAN_FILE, [])
|
||
group_list = load_json(GROUP_FILE, [记得改成要启用的群聊]) # 默认群
|
||
|
||
# 新人进群事件
|
||
new_member = on_notice(priority=5, block=False)
|
||
|
||
|
||
@new_member.handle()
|
||
async def _(bot: Bot, event: GroupIncreaseNoticeEvent):
|
||
user_id = event.user_id
|
||
group_id = event.group_id
|
||
|
||
if group_id not in group_list:
|
||
return
|
||
|
||
# 如果在ban名单中
|
||
if user_id in ban_list:
|
||
await bot.call_api("set_group_kick", group_id=group_id, user_id=user_id)
|
||
return
|
||
|
||
# 检查QQ等级
|
||
try:
|
||
profile = await bot.get_stranger_info(user_id=user_id, no_cache=True)
|
||
level = profile.get("level", 0)
|
||
if level < LEVEL_LIMIT:
|
||
await bot.call_api("set_group_kick", group_id=group_id, user_id=user_id)
|
||
return
|
||
except:
|
||
pass
|
||
|
||
# 禁言
|
||
await bot.call_api("set_group_ban", group_id=group_id, user_id=user_id, duration=3600)
|
||
|
||
# 生成验证码
|
||
code = str(random.randint(100000, 999999))
|
||
verify_data[user_id] = {
|
||
"code": code,
|
||
"group_id": group_id,
|
||
"expire": datetime.now() + timedelta(seconds=VERIFY_TIMEOUT),
|
||
}
|
||
|
||
try:
|
||
await bot.send_private_msg(user_id=user_id, message=f"请在5分钟内回复验证码：{code}\n回复错误或超时将被踢出群聊。")
|
||
except:
|
||
await bot.send_group_msg(group_id=group_id, message=f"[CQ:at,qq={user_id}] 请在私聊完成验证")
|
||
|
||
|
||
# 私聊消息监听
|
||
verify_reply = on_message(priority=5, block=False)
|
||
|
||
|
||
@verify_reply.handle()
|
||
async def _(bot: Bot, event: MessageEvent):
|
||
user_id = event.user_id
|
||
msg = str(event.get_message()).strip()
|
||
|
||
if user_id not in verify_data:
|
||
return
|
||
|
||
data = verify_data[user_id]
|
||
group_id = data["group_id"]
|
||
correct_code = data["code"]
|
||
|
||
# 超时
|
||
if datetime.now() > data["expire"]:
|
||
await bot.call_api("set_group_kick", group_id=group_id, user_id=user_id)
|
||
ban_list.append(user_id)
|
||
save_json(BAN_FILE, ban_list)
|
||
verify_data.pop(user_id, None)
|
||
return
|
||
|
||
# 验证
|
||
if msg == correct_code:
|
||
await bot.call_api("set_group_ban", group_id=group_id, user_id=user_id, duration=0)
|
||
await bot.send_group_msg(group_id=group_id, message=f"[CQ:at,qq={user_id}] 验证成功，欢迎加入群聊")
|
||
verify_data.pop(user_id, None)
|
||
else:
|
||
await bot.call_api("set_group_kick", group_id=group_id, user_id=user_id)
|
||
ban_list.append(user_id)
|
||
save_json(BAN_FILE, ban_list)
|
||
verify_data.pop(user_id, None)
|
||
|
||
|
||
# 定时任务：清理超时
|
||
async def check_timeout(bot: Bot):
|
||
while True:
|
||
now = datetime.now()
|
||
to_remove = []
|
||
for uid, data in verify_data.items():
|
||
if now > data["expire"]:
|
||
try:
|
||
await bot.call_api("set_group_kick", group_id=data["group_id"], user_id=uid)
|
||
except:
|
||
pass
|
||
ban_list.append(uid)
|
||
save_json(BAN_FILE, ban_list)
|
||
to_remove.append(uid)
|
||
for uid in to_remove:
|
||
verify_data.pop(uid, None)
|
||
await asyncio.sleep(60)
|
||
|
||
|
||
@new_member.handle()
|
||
async def start_check(bot: Bot, event: GroupIncreaseNoticeEvent):
|
||
asyncio.create_task(check_timeout(bot))
|