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

nonebot_plugin_l4d2_server/{l4d2_image → l4_image}/vtfs.py

@@ -4,8 +4,10 @@ from nonebot.log import logger
 from PIL import Image as Image
 from srctools.vtf import VTF, ImageFormats
 
+# 以下部分测试喷漆失败，todo
 
-async def img_to_vtf(pic_byte: bytes, tag) -> BytesIO:
+
+async def img_to_vtf(pic_byte: bytes, tag: str) -> BytesIO:
     pic = BytesIO(pic_byte)
     pic = Image.open(pic).convert("RGBA")
     vtf_io = BytesIO()
@@ -19,7 +21,7 @@ async def img_to_vtf(pic_byte: bytes, tag) -> BytesIO:
     if tag == "覆盖":
         logger.info(tag)
         img2 = Image.new("RGBA", (1024, 1024), (255, 255, 255, 0))
-        r, g, b, a = pic.split()
+        _, _, _, a = pic.split()
         img2.paste(pic, mask=a)
         pic = pic.resize((1024, 1024))
     elif tag == "填充":

nonebot_plugin_l4d2_server/l4_request/__init__.py

@@ -0,0 +1,143 @@
+from typing import Dict, List, Optional, cast
+
+from nonebot.log import logger
+
+from ..config import server_all_path
+from ..l4_image import msg_to_image
+from ..utils.api.models import AllServer, NserverOut, OutServer
+from ..utils.utils import split_maohao
+from .draw_msg import draw_one_ip, get_much_server
+
+try:
+    import ujson as json
+except ImportError:
+    import json
+
+
+# 获取全部服务器信息
+ALLHOST: Dict[str, List[NserverOut]] = {}
+COMMAND = set()
+
+
+async def get_all_server_detail():
+    out_list: List[AllServer] = []
+    for group in ALLHOST:
+        msg_list = await get_group_detail(group)
+        if msg_list is None:
+            continue
+        active_server = 0
+        max_server = 0
+        active_player = 0
+        max_player = 0
+        for index, msg in enumerate(msg_list):
+            max_server = index + 1
+            if msg["server"].max_players != 0:
+                active_server += 1
+                active_player += msg["server"].player_count
+                max_player += msg["server"].max_players
+        data = {
+            "command": group,
+            "active_server": active_server,
+            "max_server": max_server,
+            "active_player": active_player,
+            "max_player": max_player,
+        }
+        out_list.append(cast(AllServer, data))
+
+    # to do作图，先用文字凑合
+    out_msg = ""
+    for one in out_list:
+        out_msg += f"{one['command']} | 服务器{one['active_server']}/{one['max_server']} | 玩家{one['active_player']}/{one['max_player']}\n"
+    return out_msg
+
+
+async def get_server_detail(
+    command: str,
+    _id: Optional[str] = None,
+    is_img: bool = True,
+):
+    server_json = ALLHOST.get(command)
+    logger.info(server_json)
+    if server_json is None:
+        logger.warning("未找到这个组")
+        return None
+
+    if _id is None:
+        # 输出组信息
+        logger.info(f"正在请求组服务器信息 {command}")
+        server_dict = await get_much_server(server_json, command)
+        if is_img:
+            out_msg = await msg_to_image(server_dict)
+        else:
+            out_msg = server_dict
+        return out_msg
+
+    # 返回单个
+    logger.info("正在请求单服务器信息")
+    out_msg = ""
+    for i in server_json:
+        if str(_id) == str(i["id"]):
+            out_msg = await draw_one_ip(i["host"], i["port"])
+            if is_img:
+                return cast(bytes, out_msg)
+            if not is_img:
+                return cast(List[OutServer], out_msg)
+    # print(out_msg)
+    return None
+
+
+async def get_group_detail(
+    command: str,
+):
+    server_json = ALLHOST.get(command)
+    logger.info(server_json)
+    if server_json is None:
+        logger.warning("未找到这个组")
+        return None
+
+    logger.info("正在请求组服务器信息")
+    return await get_much_server(server_json, command)
+
+
+async def get_ip_server(ip: str):
+    host, port = split_maohao(ip)
+    return await draw_one_ip(host, port)
+
+
+# 以下是重载ip
+def reload_ip():
+    global COMMAND
+    # print("正在读取json文件")
+    group_ip = []
+    for item in server_all_path.iterdir():
+        if item.is_file():
+            if item.name.endswith("json"):
+                json_data = json.loads(item.read_text(encoding="utf-8"))
+                group_server = cast(Dict[str, List[NserverOut]], json_data)
+
+                for group, group_ip in group_server.items():
+                    # 处理ip,host,port关系
+                    for one_ip in group_ip:
+                        if one_ip.get("ip"):
+                            if one_ip.get("host") and one_ip.get("port"):
+                                pass
+                            if one_ip.get("host") and not one_ip.get("port"):
+                                one_ip["port"] = 20715
+                            if not one_ip.get("host"):
+                                one_ip["host"], one_ip["port"] = split_maohao(
+                                    one_ip.get("ip"),
+                                )
+                        else:
+                            if one_ip.get("host") and one_ip.get("port"):
+                                one_ip["ip"] = f'{one_ip["host"]}:{one_ip["port"]}'
+                            if one_ip.get("host") and not one_ip.get("port"):
+                                one_ip["ip"] = f'{one_ip["host"]}:20715'
+                            else:
+                                logger.warning(f"{one_ip} 没有ip")
+
+                    ALLHOST.update({group: group_ip})
+                    COMMAND.add(group)
+                logger.success(f"成功加载 {item.name.split('.')[0]} {len(group_ip)}个")
+
+            if item.name.endswith("txt"):
+                """to do"""

nonebot_plugin_l4d2_server/l4_request/draw_msg.py

@@ -0,0 +1,83 @@
+import asyncio
+
+# from logging import log
+from typing import List, Tuple
+
+from ..config import config
+from ..utils.api.models import NserverOut, OutServer
+from ..utils.api.request import L4API
+
+
+async def draw_one_ip(host: str, port: int):
+    """输出单个ip"""
+    # 先用文字凑合
+    try:
+        ser_list = await L4API.a2s_info([(host, port)], is_player=True)
+    except asyncio.exceptions.TimeoutError:
+        return "服务器无响应"
+    player_msg = ""
+    one_server = ser_list[0][0]
+    one_player = ser_list[0][1]
+
+    if len(one_player):
+        max_duration_len = max(
+            [len(str(await convert_duration(i.duration))) for i in one_player],
+        )
+        max_score_len = max(len(str(i.score)) for i in one_player)
+
+        for player in one_player:
+            soc = "[{:>{}}]".format(player.score, max_score_len)
+            chines_dur = await convert_duration(player.duration)
+            dur = "{:^{}}".format(chines_dur, max_duration_len)
+
+            player_msg += f"{soc} | {dur} | {player.name} \n"
+    else:
+        player_msg = "服务器感觉很安静啊"
+
+    msg = f"""*{one_server.server_name}*
+游戏: {one_server.folder}
+地图: {one_server.map_name}
+人数: {one_server.player_count}/{one_server.max_players}"""
+    if one_server.ping is not None:
+        msg += f"""
+ping: {one_server.ping * 1000:.0f}ms
+{player_msg}"""
+    if config.l4_show_ip:
+        msg += f"""
+connect {host}:{port}"""
+    return msg
+
+
+async def get_much_server(server_json: List[NserverOut], command: str):
+    out_server: List[OutServer] = []
+    search_list: List[Tuple[str, int]] = []
+    for i in server_json:
+        search_list.append((i["host"], i["port"]))  # noqa: PERF401
+
+    all_server = await L4API.a2s_info(search_list, is_player=True)
+
+    for index, i in enumerate(all_server):
+        out_server.append(
+            {
+                "server": i[0],
+                "player": i[1],
+                "host": server_json[index]["host"],
+                "port": server_json[index]["port"],
+                "command": command,
+                "id_": server_json[index]["id"],
+            },
+        )
+
+    return out_server
+
+
+async def convert_duration(duration: float) -> str:
+    minutes, seconds = divmod(duration, 60)
+    hours, minutes = divmod(minutes, 60)
+    time_str = ""
+    if hours > 0:
+        time_str += f"{int(hours)}h "
+    if minutes > 0:
+        time_str += f"{int(minutes)}m "
+    time_str += f"{int(seconds)}s"
+    return time_str

nonebot_plugin_l4d2_server/utils/api/api.py

@@ -0,0 +1,10 @@
+# coding=utf-8
+
+
+anne_ban = "https://sb.trygek.com/"
+"""anne电信服网址，sourceban++"""
+anne_ser = f"{anne_ban}l4d_stats/"
+AnneRankApi = f"{anne_ser}ranking/index.php?type=coop"
+AnnePlayerApi = f"{anne_ser}ranking/player.php?steamid="
+# post
+AnneSearchApi = f"{anne_ser}/ranking/search.php"

nonebot_plugin_l4d2_server/utils/api/models.py

@@ -0,0 +1,137 @@
+# coding=utf-8
+from typing import List, Optional, TypedDict
+
+import a2s
+from pydantic import BaseModel
+
+
+class SourceBansInfo(BaseModel):
+    """source服务器信息"""
+
+    index: int
+    host: str
+    port: int
+
+
+class NserverOut(TypedDict):
+    id: int
+    ip: str
+    host: str
+    port: int
+    version: Optional[str]
+
+
+class OutServer(TypedDict):
+    server: a2s.SourceInfo
+    player: List[a2s.Player]
+    host: str
+    port: int
+    command: str
+    id_: int
+
+
+class AnnePlayer(TypedDict):
+    mode: str
+    name: str
+    source: str
+    playtime: str
+
+
+class AllServer(TypedDict):
+    command: str
+    active_server: int
+    max_server: int
+    active_player: int
+    max_player: int
+
+
+class AnneSearch(TypedDict):
+    steamid: str
+    rank: str
+    name: str
+    score: str
+    play_time: str
+    last_time: str
+
+
+class AnnePlayerDetail(TypedDict):
+    rank: int
+    source: str
+    avg_source: float
+    kills: str
+    kills_people: str
+    headshots: str
+    avg_headshots: float
+    map_play: str
+
+
+class AnnePlayerError(TypedDict):
+    mistake_shout: str
+    kill_friend: str
+    down_friend: str
+    abandon_friend: str
+    put_into: str
+    agitate_witch: str
+
+
+class AnnePlayerInfAvg(TypedDict):
+    avg_smoker: float
+    avg_boomer: float
+    avg_hunter: float
+    avg_charger: float
+    avg_spitter: float
+    avg_jockey: float
+    avg_tank: float
+
+
+class AnnePlayerSur(TypedDict):
+    map_clear: str
+    prefect_into: str
+    get_oil: str
+    ammo_arrange: str
+    adrenaline_give: str
+    pills_give: str
+    """给药"""
+    first_aid_give: str
+    """给包"""
+    friend_up: str
+    diss_friend: str
+    save_friend: str
+    protect_friend: str
+    pro_from_smoker: str
+    pro_from_hunter: str
+    pro_from_charger: str
+    pro_from_jockey: str
+    melee_charge: str
+    """刀牛"""
+    tank_kill: str
+    witch_instantly_kill: str
+    """秒妹"""
+
+
+class AnnePlayerInf(TypedDict):
+    sur_ace: str
+    sur_down: str
+    boommer_hit: str
+    hunter_prefect: str
+    hunter_success: str
+    tank_damage: str
+    charger_multiple: str
+
+
+class AnnePlayerInfo(TypedDict):
+    name: str
+    avatar: str
+    steamid: str
+    playtime: str
+    lasttime: str
+
+
+class AnnePlayer2(TypedDict):
+    kill_msg: str
+    info: AnnePlayerInfo
+    detail: AnnePlayerDetail
+    error: AnnePlayerError
+    inf_avg: AnnePlayerInfAvg
+    sur: AnnePlayerSur
+    inf: AnnePlayerInf