nonebot-plugin-l4d2-server 0.6.6__py3-none-any.whl → 1.0.0a2__py3-none-any.whl

nonebot_plugin_l4d2_server/__init__.py

@@ -17,112 +17,29 @@
 
 from nonebot import get_driver, require
 
-require("nonebot_plugin_apscheduler")  # noqa: F401
-require("nonebot_plugin_saa")  # noqa: F401
-require("nonebot_plugin_htmlrender")  # noqa: F401
-require("nonebot_plugin_txt2img")  # noqa: F401
-scheduler = require("nonebot_plugin_apscheduler").scheduler
+require("nonebot_plugin_htmlrender")
+require("nonebot_plugin_alconna")
+require("nonebot_plugin_tortoise_orm")
 
-from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent
-from nonebot.log import logger
-from nonebot.matcher import Matcher
-from nonebot.plugin import PluginMetadata
-from nonebot.typing import T_State
+from nonebot.plugin import PluginMetadata, inherit_supported_adapters  # noqa: E402
 
-from .l4d2_data import sq_L4D2
-
-# from .l4d2_file.input_json import *
-from .l4d2_image.steam import url_to_byte_name
-from .l4d2_push import scheduler  # noqa: F401
-from .l4d2_utils.command import help_, search_api
-from .l4d2_utils.config import l4_config
-from .l4d2_utils.utils import upload_file
-from .l4d2_web import web, webUI  # noqa: F401
+from . import __main__ as __main__  # noqa: E402
+from .config import ConfigModel  # noqa: E402
+from .l4_help import __version__  # noqa: E402
 
 driver = get_driver()
 
-__version__ = "0.6.6"
+
 __plugin_meta__ = PluginMetadata(
     name="求生之路小助手",
     description="可用于管理求生之路查服和本地管理",
     usage="群内对有关求生之路的查询和操作",
+    config=ConfigModel,
     type="application",
     homepage="https://github.com/Agnes4m/nonebot_plugin_l4d2_server",
-    supported_adapters={"~onebot.v11"},
+    supported_adapters=inherit_supported_adapters("nonebot_plugin_alconna"),
     extra={
         "version": __version__,
         "author": "Agnes4m <Z735803792@163.com>",
     },
 )
-
-
-"""相当于启动就检查数据库"""
-
-
-@help_.handle()
-async def _(matcher: Matcher):
-    msg = """=====求生机器人帮助=====
-    1、电信服战绩查询【求生anne | Ranne[id/steamid/@]】
-    2、电信服绑定【求生绑定[id/steamid]】",
-    3、电信服状态查询【云xx】
-    4、创意工坊下载【创意工坊下载[物品id/链接]】
-    5、指定ip查询【求生ip[ip]】(可以是域名)
-    6、求生喷漆制作【求生喷漆】
-    7、本地服务器相关【l4地图上传】【l4地图改名】【l4地图删除】【l4路径】【l4插件】
-    8、ip导入（私聊发送json文件，格式参考下方链接中readme）
-    9、内置ip更新【l4公益服更新】【求生更新anne】
-    项目地址:https://github.com/Agnes4m/nonebot_plugin_l4d2_server
-    """
-    await matcher.finish(msg)
-
-
-@search_api.got("is_sure", prompt='如果需要上传，请发送 "yes"')
-async def _(matcher: Matcher, bot: Bot, event: GroupMessageEvent, state: T_State):
-    if not l4_config.l4_group_upload:
-        return
-    is_sure = str(state["is_sure"])
-    if is_sure == "yes":
-        data_dict: dict = state["maps"][0]
-        if isinstance(data_dict, dict):
-            logger.info("开始上传")
-            if l4_config.l4_only:
-                reu = await url_to_byte_name(data_dict["url"], "htp")
-            else:
-                reu = await url_to_byte_name(data_dict["url"])
-            if not reu:
-                return
-            data_file, file_name = reu
-            if data_file:
-                await matcher.send("获取地址成功，尝试上传")
-                await upload_file(bot, event, data_file, file_name)
-            else:
-                await search_api.send("出错了，原因是下载链接不存在")
-        else:
-            ...
-            # logger.info("开始上传")
-            # for data_one in data_dict:
-            #     reu = await url_to_byte_name(data_one["url"])
-            #     if not reu:
-            #         return
-            #     data_file, file_name = reu
-            #     await all_zip_to_one()
-            #     await upload_file(bot, event, data_file, file_name)
-    else:
-        await matcher.finish("已取消上传")
-
-
-# @reload_ip.handle()
-# async def _(matcher:Matcher):
-#     global matchers
-#     await matcher.send('正在重载ip，可能需要一点时间')
-#     for _, l4_matchers in matchers.items():
-#         for l4_matcher in l4_matchers:
-#             l4_matcher.destroy()
-#     await get_des_ip()
-#     await matcher.finish('已重载ip')
-
-
-@driver.on_shutdown
-async def close_db():
-    """关闭数据库"""
-    sq_L4D2._close()  # noqa: SLF001

nonebot_plugin_l4d2_server/__main__.py

@@ -0,0 +1,140 @@
+"""
+* Copyright (c) 2023, Agnes Digital
+*
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with this program.  If not, see <https://www.gnu.org/licenses/>.
+"""
+
+from typing import TYPE_CHECKING, List, Optional
+
+from nonebot.adapters import Message
+from nonebot.log import logger
+from nonebot.matcher import Matcher
+from nonebot.params import CommandArg, CommandStart, RawCommand
+from nonebot.plugin import on_command
+from nonebot_plugin_alconna import UniMessage
+
+from .config import config
+from .l4_help import get_l4d2_core_help
+from .l4_request import (
+    COMMAND,
+    get_all_server_detail,
+    get_ip_server,
+    get_server_detail,
+    reload_ip,
+)
+from .utils.api.request import L4API
+
+if TYPE_CHECKING:
+    from .utils.api.models import OutServer
+
+reload_ip()
+
+l4_help = on_command("l4help", aliases={"l4帮助", "l4d2帮助"})
+l4_request = on_command("anne", aliases=COMMAND, priority=10)
+l4_reload = on_command("l4reload", aliases={"l4刷新,l4重载"})
+l4_all = on_command("l4all", aliases={"l4全服"})
+l4_connect = on_command("connect", aliases={"l4连接"})
+l4_find_player = on_command("l4find", aliases={"l4查找"})
+
+
+@l4_help.handle()
+async def _(matcher: Matcher):
+    """帮助"""
+    logger.info("开始执行[l4d2帮助]")
+    im = await get_l4d2_core_help()
+    if isinstance(im, str):
+        await matcher.finish(im)
+    await UniMessage.image(raw=im).send()
+
+
+@l4_request.handle()
+async def _(
+    start: str = CommandStart(),
+    command: str = RawCommand(),
+    args: Message = CommandArg(),
+):
+    """例：
+    指令：  /橘5
+    start: /(command开头指令)
+    command: /橘(响应的全部指令)
+    args: 5(响应的指令后的数字)
+    """
+
+    if start:
+        command = command.replace(start, "")
+    if command == "anne":
+        command = "云"
+    _id: Optional[str] = args.extract_plain_text()
+    if _id is not None and not _id.isdigit() and _id:
+        return
+    if not _id:
+        _id = None
+    logger.info(f"组:{command} ;数字:{_id}")
+    msg = await get_server_detail(command, _id)
+    if msg is not None:
+        if isinstance(msg, str):
+            await UniMessage.text(msg).finish()
+        if isinstance(msg, bytes):
+            await UniMessage.image(raw=msg).finish()
+    else:
+        await UniMessage.text("服务器无响应").finish()
+
+
+@l4_find_player.handle()
+async def _(
+    args: Message = CommandArg(),
+):
+    msg: str = args.extract_plain_text().strip()
+    tag_list: List[str] = msg.split(" ", maxsplit=1)
+    if len(tag_list) < 2:
+        return await UniMessage.text("格式错误，正确格式：/l4find 组名 玩家名").finish()
+    group, name = tag_list
+    out: List[OutServer] = await get_server_detail(group, is_img=False)  # type: ignore
+    out_msg = "未找到玩家"
+    for one in out:
+        for player in one["player"]:
+            if name in player.name:
+                out_msg = await get_ip_server(f"{one['host']}:{one['port']}")
+
+    return await UniMessage.text(out_msg).finish()
+
+
+@l4_all.handle()
+async def _():
+    await UniMessage.text(await get_all_server_detail()).finish()
+
+
+@l4_connect.handle()
+async def _(args: Message = CommandArg()):
+    ip: Optional[str] = args.extract_plain_text()
+    if ip is not None:
+        await UniMessage.text(await get_ip_server(ip)).finish()
+
+
+# anne部分
+if config.l4_anne:
+    logger.info("加载anne功能")
+    from .l4_anne import *  # noqa: F403
+
+
+@l4_reload.handle()
+async def _(args: Message = CommandArg()):
+    arg = args.extract_plain_text().strip()
+    print(arg)
+    if not arg:
+        reload_ip()
+        logger.success("重载ip完成")
+        await UniMessage.text("重载ip完成").finish()
+    else:
+        print(await L4API.get_sourceban())

nonebot_plugin_l4d2_server/config.py

@@ -0,0 +1,35 @@
+from pathlib import Path
+
+from nonebot import get_plugin_config
+from pydantic import BaseModel
+
+DATAPATH = Path(__file__).parent.joinpath("data")
+DATAOUT = Path("data/L4D2")
+print(DATAOUT.absolute())
+server_all_path = DATAOUT / "l4d2"
+server_all_path.mkdir(parents=True, exist_ok=True)
+
+ICONPATH = DATAPATH / "icon"
+
+
+class ConfigModel(BaseModel):
+    l4_anne: bool = True
+    """是否启用anne电信服相关功能"""
+    l4_enable: bool = True
+    """是否全局启用求生功能"""
+    l4_image: bool = False
+    """是否启用图片"""
+    l4_connect: bool = True
+    """是否在查服命令后加入connect ip"""
+    l4_path: str = "data/L4D2"
+    """插件数据路径"""
+    l4_players: int = 4
+    """查询总图的时候展示的用户数量"""
+    l4_style: str = "default"
+    """图片风格"""
+    l4_font: str = str(Path(__file__).parent.joinpath("data/font/loli.ttf"))
+    """字体"""
+    l4_show_ip: bool = True
+
+
+config = get_plugin_config(ConfigModel)

nonebot_plugin_l4d2_server/l4_anne/__init__.py

@@ -0,0 +1,124 @@
+from nonebot import log as log, on_command  # noqa: N999
+from nonebot.adapters import Event, Message
+from nonebot.log import logger
+from nonebot.params import CommandArg
+from nonebot_plugin_alconna import UniMessage
+
+from ..utils.api.request import L4API
+from ..utils.database.models import SteamUser
+from ..utils.utils import get_message_at
+from .ranne import get_anne_rank_out
+
+anne_bind = on_command("Banne", aliases={"l4绑定", "anne绑定"}, priority=5, block=True)
+anne_search = on_command("Sanne", aliases={"anne搜索"})
+anne_rank = on_command("Ranne", aliases={"anne成绩"}, block=True, priority=1)
+
+anne_del = on_command(
+    "Danne",
+    aliases={"l4删除", "anne删除", "l4解绑", "anne解绑"},
+    priority=5,
+    block=True,
+)
+
+
+@anne_search.handle()
+async def _(args: Message = CommandArg()):
+    name: str = args.extract_plain_text().strip()
+    print(name)
+    user_list = await L4API.get_anne_steamid(name)
+    if user_list is None:
+        await UniMessage.text("未找到玩家").finish()
+    msg = f"---有{len(user_list)}个玩家---"
+    for index, user in enumerate(user_list, start=1):
+        if index >= 10:
+            break
+        msg += f"""
+        {index}. {user["name"]} | [{user["score"]}] | {user["play_time"]}
+        {user["steamid"]}
+        """
+    if msg:
+        await UniMessage.text("\n".join(msg.splitlines())).finish()
+    else:
+        await UniMessage.text("没有找到玩家").finish()
+
+
+@anne_bind.handle()
+async def _(ev: Event, args: Message = CommandArg()):
+    arg: str = args.extract_plain_text()
+    if not arg:
+        await UniMessage.text("虚空绑定？").finish()
+
+    user = await SteamUser.get_or_none(userid=int(ev.get_user_id()))
+    if user is None:
+        user = await SteamUser.create(userid=int(ev.get_user_id()))
+
+    if len(arg) == 17:
+        # steamid64
+        logger.info(f"SteamID64:{arg}")
+        user.SteamID64 = arg
+        msg = "绑定steamid64"
+
+    elif arg.startswith("STEAM_"):
+        # steamid
+        logger.info(f"SteamID:{arg}")
+        user.SteamID = arg
+        msg = "绑定steamid"
+
+    else:
+        # name
+        logger.info(f"Name:{arg}")
+        user.Name = arg
+        msg = "绑定名字"
+    await user.save()
+
+    return await UniMessage.text(f"{msg}成功").finish()
+
+
+@anne_del.handle()
+async def _(ev: Event):
+    if record := await SteamUser.get_or_none(userid=int(ev.get_user_id())):
+        logger.info(f"删除用户:{record}")
+        await record.delete()
+        await record.save()
+        return await UniMessage.text("删除成功").finish()
+    return await UniMessage.text("没有绑定信息呢").finish()
+
+
+@anne_rank.handle()
+async def _(ev: Event, args: Message = CommandArg()):
+    uid = await get_message_at(str(ev.json()))
+    if uid is None:
+        uid = int(int(ev.get_user_id()))
+    steamid = ""
+    arg: str = args.extract_plain_text().strip()
+    logger.info(f"arg:{arg}")
+    # 优先从数据库查询
+    if not arg:
+        msg = await SteamUser.get_or_none(userid=uid)
+        if msg is not None:
+            steamid = msg.SteamID
+            if not steamid:
+                name = msg.Name
+                if not name:
+                    await UniMessage.text("未绑定名字/steamid").finish()
+                msg_dict = await L4API.get_anne_steamid(name)
+                if not msg_dict:
+                    await UniMessage.text("绑定的昵称找不到呢").finish()
+                steamid = msg_dict[0]["steamid"]
+            logger.info(f"steamid:{steamid}")
+
+    # 再从arg中查找
+    else:
+        if arg.startswith("STEAM_"):
+            steamid = arg
+        else:
+            arg_dict = await L4API.get_anne_steamid(arg)
+            if not arg_dict:
+                await UniMessage.text("未找到该昵称玩家").finish()
+            steamid = arg_dict[0]["steamid"]
+    if not steamid:
+        await UniMessage.text("未找到玩家,请使用指令`l4搜索`查找").finish()
+    out_msg = await get_anne_rank_out(steamid)
+    if out_msg is None:
+        await UniMessage.text("未找到玩家").finish()
+    await UniMessage.text(out_msg).finish()

nonebot_plugin_l4d2_server/l4_anne/ranne.py

@@ -0,0 +1,18 @@
+# coding=utf-8
+# anne战绩查询
+from ..utils.api.request import L4API
+
+
+async def get_anne_rank_out(steamid: str):
+    msg = await L4API.get_anne_playerdetail(steamid)
+    if msg is None:
+        return None
+    return f"""电信anne查询结果：
+    昵称：{msg['info']['name']}
+    排名：{msg['detail']['rank']}
+    分数：{msg['detail']["source"]}
+    击杀：{msg['detail']["kills"]}
+    爆头率：{msg['detail']["avg_source"]}
+    时间：{msg['info']['playtime']}
+    上次：{msg['info']['lasttime']}
+    """

nonebot_plugin_l4d2_server/l4_help/Help.json

@@ -0,0 +1,102 @@
+{
+  "基础服务": {
+    "desc": "无需任何权限",
+    "data": [
+      {
+        "name": "(组)",
+        "desc": "返回一个组的服务器信息",
+        "eg": "/云",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      },
+      {
+        "name": "(组)(id)",
+        "desc": "查询单服信息",
+        "eg": "/云1",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      },
+      {
+        "name": "l4查找(组)(玩家)",
+        "desc": "查询用户所在服务器状态",
+        "eg": "/l4查找 云 爱丽数码",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      },
+      {
+        "name": "l4全服",
+        "desc": "返回可以查询的所有服务器组(未完成)",
+        "eg": "/l4全服",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      }
+    ]
+  },
+  "电信服anne服务": {
+    "desc": "电信anne服相关服务",
+    "data": [
+      {
+        "name": "Banne xx",
+        "desc": "绑定昵称/steamid",
+        "eg": "/Banne 爱丽数码",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      },
+      {
+        "name": "Ranne(玩家)",
+        "desc": "查询anne个人信息,没有参数查自己",
+        "eg": "/Ranne 爱丽数码",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      },
+      {
+        "name": "Sanne(玩家)",
+        "desc": "按昵称查询anne玩家信息",
+        "eg": "/Sanne 爱丽数码",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      },
+      {
+        "name": "Danne(玩家)",
+        "desc": "删除anne玩家信息",
+        "eg": "/Danne 爱丽数码",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      }
+    ]
+  },
+  "管理员服务": {
+    "desc": "管理可以使用的指令",
+    "data": [
+      {
+        "name": "l4刷新",
+        "desc": "重载存储的ip数据",
+        "eg": "l4刷新",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": true
+      }
+    ]
+  },
+  "其他服务": {
+    "desc": "一些杂项服务",
+    "data": [
+      {
+        "name": "connect",
+        "desc": "直接发送求生链接返回信息",
+        "eg": "connect 127.0.0.1:27015",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      }
+    ]
+  }
+}

nonebot_plugin_l4d2_server/l4_help/__init__.py

@@ -0,0 +1,57 @@
+from pathlib import Path
+from typing import Dict, Union
+
+import aiofiles
+from msgspec import json as msgjson
+from PIL import Image
+
+from ..config import ICONPATH
+from ..l4_image.convert import core_font
+from ..l4_image.model import PluginHelp
+from .draw import get_help
+
+__version__ = "1.0.0a2"
+TEXT_PATH = Path(__file__).parent / "texture2d"
+HELP_DATA = Path(__file__).parent / "Help.json"
+
+
+async def get_help_data() -> Union[Dict[str, PluginHelp], None]:
+    if HELP_DATA.exists():
+        async with aiofiles.open(HELP_DATA, "rb") as file:
+            return msgjson.decode(
+                await file.read(),
+                type=Dict[str, PluginHelp],
+            )
+    return None
+
+
+async def get_l4d2_core_help() -> Union[bytes, str]:
+    help_data = await get_help_data()
+    if help_data is None:
+        return "暂未找到帮助数据..."
+
+    bg_out = Image.open(TEXT_PATH / "bg.jpg")
+    bg_new = Image.new(
+        "RGBA",
+        (bg_out.width, bg_out.height),
+        (255, 255, 255, 100),
+    )
+    bg_out.paste(bg_new, None, bg_new)
+
+    return await get_help(
+        "L4插件",
+        f"版本号:{__version__}",
+        help_data,
+        bg_out,
+        Image.open(TEXT_PATH / "icon.png"),
+        Image.open(TEXT_PATH / "badge.png"),
+        Image.open(TEXT_PATH / "banner.png"),
+        Image.open(TEXT_PATH / "button.png"),
+        core_font,
+        is_dark=False,
+        column=4,
+        is_gaussian=False,
+        text_color=(23, 67, 91),
+        sub_c=(49, 110, 144),
+        icon_path=ICONPATH,
+    )