nonebot-plugin-l4d2-server 0.5.1__py3-none-any.whl → 0.5.3__py3-none-any.whl

nonebot_plugin_l4d2_server/l4d2_server/__init__.py

@@ -1,62 +1,62 @@
-import asyncio
-from rcon.source.proto import Packet,Type
-from rcon.source.async_rcon import communicate,close
-
-async def main(host):
-    host = '43.142.178.212'
-    port = 40003
-    password = '1145149191810'
-    encoding = 'utf-8'
-
-    # Connect to RCON server
-
-
-    # Main loop
-    while True:
-        # Read user input
-        command = input('> ')
-        if not command:
-            break
-
-   # 连接服务器
-    reader, writer = await asyncio.open_connection(host, port)
-    login = Packet.make_login(password, encoding=encoding)
-    response = await communicate(reader, writer, login)
-
-    # 等待 SERVERDATA_AUTH_RESPONSE 数据包
-    while response.type != Type.SERVERDATA_AUTH_RESPONSE:
-        response = await Packet.aread(reader)
-
-    if response.id == -1:
-        await close(writer)
-        raise WrongPassword()
-
-    # 循环接收用户输入并发送指令
-    while True:
-        try:
-            command = input('请输入指令：')
-            
-        except EOFError:
-            break
-        
-        if command=='停止':
-            break
-        # 发送指令
-        
-        command = f'say {command}'
-        request = Packet.make_command(command, encoding=encoding)
-        response = await communicate(reader, writer, request)
-
-        # if response.id != request.id:
-        #     raise SessionTimeout()
-
-        print(response.payload.decode(encoding, errors='ignore'))
-
-    # 断开连接
-    await close(writer)
-    
-class WrongPassword(Exception):
-    """Indicates a wrong password."""
-    
-class SessionTimeout(Exception):
+import asyncio
+from rcon.source.proto import Packet,Type
+from rcon.source.async_rcon import communicate,close
+
+async def main(host):
+    host = '43.142.178.212'
+    port = 40003
+    password = '1145149191810'
+    encoding = 'utf-8'
+
+    # Connect to RCON server
+
+
+    # Main loop
+    while True:
+        # Read user input
+        command = input('> ')
+        if not command:
+            break
+
+   # 连接服务器
+    reader, writer = await asyncio.open_connection(host, port)
+    login = Packet.make_login(password, encoding=encoding)
+    response = await communicate(reader, writer, login)
+
+    # 等待 SERVERDATA_AUTH_RESPONSE 数据包
+    while response.type != Type.SERVERDATA_AUTH_RESPONSE:
+        response = await Packet.aread(reader)
+
+    if response.id == -1:
+        await close(writer)
+        raise WrongPassword()
+
+    # 循环接收用户输入并发送指令
+    while True:
+        try:
+            command = input('请输入指令：')
+            
+        except EOFError:
+            break
+        
+        if command=='停止':
+            break
+        # 发送指令
+        
+        command = f'say {command}'
+        request = Packet.make_command(command, encoding=encoding)
+        response = await communicate(reader, writer, request)
+
+        # if response.id != request.id:
+        #     raise SessionTimeout()
+
+        print(response.payload.decode(encoding, errors='ignore'))
+
+    # 断开连接
+    await close(writer)
+    
+class WrongPassword(Exception):
+    """Indicates a wrong password."""
+    
+class SessionTimeout(Exception):
     """Indicates that the session timed out."""

nonebot_plugin_l4d2_server/l4d2_server/rcon.py

@@ -1,29 +1,29 @@
-from rcon.source import rcon
-import asyncio
-from pathlib import Path
-from ..config import l4_config,CHECK_FILE
-# from ..config import l4_rcon,l4_host,l4_port,l4_rcon
-
-
-async def rcon_server(PASSWORD:str,msg:str):
-    # response = await rcon(command=msg, host=l4_host, port=l4_port, passwd=PASSWORD,encoding='utf-8')
-    # return response
-    try:
-        response = await asyncio.wait_for(rcon(command=msg, host=l4_config.l4_ipall[CHECK_FILE]['host'], port = l4_config.l4_ipall[CHECK_FILE]['port'], passwd=PASSWORD), timeout=30)
-        return response
-    except asyncio.TimeoutError:
-        return '超时'
-
-async def read_server_cfg_rcon():
-    """如果没有输入rcon，尝试自动获取"""
-    if not l4_config.l4_ipall[CHECK_FILE]['rcon']:
-        cfg_server = Path(l4_config.l4_ipall[CHECK_FILE]['location'],'left4dead2/cfg/server.cfg')
-        with open(cfg_server,'r')as cfg:
-            content:str = cfg.read()
-            lines = content.split('\n')
-            for line in lines:
-                if line.startswith('rcon_password'):
-                    password = line.split(' ')[-1]
-                    password = password.strip('"')
-                    return password
+from rcon.source import rcon
+import asyncio
+from pathlib import Path
+from ..config import l4_config,CHECK_FILE
+# from ..config import l4_rcon,l4_host,l4_port,l4_rcon
+
+
+async def rcon_server(PASSWORD:str,msg:str):
+    # response = await rcon(command=msg, host=l4_host, port=l4_port, passwd=PASSWORD,encoding='utf-8')
+    # return response
+    try:
+        response = await asyncio.wait_for(rcon(command=msg, host=l4_config.l4_ipall[CHECK_FILE]['host'], port = l4_config.l4_ipall[CHECK_FILE]['port'], passwd=PASSWORD), timeout=30)
+        return response
+    except asyncio.TimeoutError:
+        return '超时'
+
+async def read_server_cfg_rcon():
+    """如果没有输入rcon，尝试自动获取"""
+    if not l4_config.l4_ipall[CHECK_FILE]['rcon']:
+        cfg_server = Path(l4_config.l4_ipall[CHECK_FILE]['location'],'left4dead2/cfg/server.cfg')
+        with open(cfg_server,'r')as cfg:
+            content:str = cfg.read()
+            lines = content.split('\n')
+            for line in lines:
+                if line.startswith('rcon_password'):
+                    password = line.split(' ')[-1]
+                    password = password.strip('"')
+                    return password
     return l4_config.l4_ipall[CHECK_FILE]['rcon']

nonebot_plugin_l4d2_server/l4d2_server/workshop.py

@@ -1,50 +1,50 @@
-import httpx
-from nonebot.log import logger
-try:
-    import ujson as json
-except:    
-    import json
-
-async def workshop_to_dict(msg:str):
-    """把创意工坊的id，转化为信息字典"""
-    i = await api_get_json(msg)
-    
-    # 处理是否是多地图文件
-    if i['file_url'] == i['preview_url']:
-        return await primary_map(i)
-    else:
-        return await only_map(i)
-
-    
-async def api_get_json(msg:str) ->dict:
-    url_serach = 'https://db.steamworkshopdownloader.io/prod/api/details/file'
-    data = f'[{msg}]'
-    headers = {
-        'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0'
-    }
-    data = httpx.post(url=url_serach,headers= headers,data=data).content.decode('utf-8')
-    logger.info(data)
-    out = {}
-    data = data[1:-1]
-    data = json.loads(data)
-    return data
-    
-
-async def only_map(i:dict):
-    """单地图下载"""
-    out = {}
-    out['名字'] = i['title']
-    out['游戏'] = i['app_name']
-    out['下载地址'] = i['file_url']
-    out['图片地址'] = i['preview_url']
-    out['细节'] = i['file_description']
-    return out
-
-async def primary_map(i):
-    """主地图返回多地图参数"""
-    map_list = []
-    map_list.append(i)
-    for one in i['children']:
-        map_list.append(await api_get_json(one['publishedfileid']))
-    return map_list
-
+import httpx
+from nonebot.log import logger
+try:
+    import ujson as json
+except:    
+    import json
+
+async def workshop_to_dict(msg:str):
+    """把创意工坊的id，转化为信息字典"""
+    i = await api_get_json(msg)
+    
+    # 处理是否是多地图文件
+    if i['file_url'] == i['preview_url']:
+        return await primary_map(i)
+    else:
+        return await only_map(i)
+
+    
+async def api_get_json(msg:str) ->dict:
+    url_serach = 'https://db.steamworkshopdownloader.io/prod/api/details/file'
+    data = f'[{msg}]'
+    headers = {
+        'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0'
+    }
+    data = httpx.post(url=url_serach,headers= headers,data=data).content.decode('utf-8')
+    logger.info(data)
+    out = {}
+    data = data[1:-1]
+    data = json.loads(data)
+    return data
+    
+
+async def only_map(i:dict):
+    """单地图下载"""
+    out = {}
+    out['名字'] = i['title']
+    out['游戏'] = i['app_name']
+    out['下载地址'] = i['file_url']
+    out['图片地址'] = i['preview_url']
+    out['细节'] = i['file_description']
+    return out
+
+async def primary_map(i):
+    """主地图返回多地图参数"""
+    map_list = []
+    map_list.append(i)
+    for one in i['children']:
+        map_list.append(await api_get_json(one['publishedfileid']))
+    return map_list
+

nonebot_plugin_l4d2_server/l4d2_update/__init__.py

@@ -0,0 +1,143 @@
+from typing import Any, Tuple
+
+from nonebot.log import logger
+from nonebot.matcher import Matcher
+from nonebot.params import RegexGroup
+from nonebot.permission import SUPERUSER
+from nonebot import get_bot, on_regex, get_driver, on_command
+from nonebot.adapters.onebot.v11 import Bot, MessageEvent, MessageSegment
+
+from ..utils import register_menu
+from ..rule import FullCommand
+from .draw_update_log import draw_update_log_img
+from .restart import restart_message, restart_genshinuid
+
+l4d_restart = on_command('l4重启', rule=FullCommand())
+get_update_log = on_command('更新记录', rule=FullCommand())
+l4d_update = on_regex(
+    r'^(l4d)(强行)?(强制)?(更新)$',
+    block=True,
+)
+
+
+driver = get_driver()
+
+
+@driver.on_bot_connect
+async def _():
+    logger.info('检查遗留信息...')
+    bot = get_bot()
+    update_log = await restart_message()
+    if update_log == {}:
+        return
+    if update_log['send_type'] == 'group':
+        await bot.call_api(
+            api='send_group_msg',
+            group_id=update_log['send_to'],
+            message=update_log['msg'],
+        )
+    else:
+        await bot.call_api(
+            api='send_private_msg',
+            user_id=update_log['send_to'],
+            message=update_log['msg'],
+        )
+    logger.info('遗留信息检查完毕!')
+
+
+@get_update_log.handle()
+@register_menu(
+    '更新记录',
+    '更新记录',
+    '查看插件最近的更新记录',
+    detail_des=(
+        '介绍：\n'
+        '查看插件最近的有效Git更新记录\n'
+        ' \n'
+        '指令：\n'
+        '- <ft color=(238,120,0)>更新记录</ft>'
+    ),
+)
+async def send_updatelog_msg(
+    matcher: Matcher,
+):
+    im = await draw_update_log_img(is_update=False)
+    logger.info('正在执行[更新记录]...')
+    if isinstance(im, str):
+        await matcher.finish(im)
+    elif isinstance(im, bytes):
+        await matcher.finish(MessageSegment.image(im))
+    else:
+        await matcher.finish('发生了未知错误,请联系管理员检查后台输出!')
+
+
+@l4d_restart.handle()
+@register_menu(
+    '重启Bot',
+    'l4重启',
+    '重启Bot框架',
+    trigger_method='超级用户指令',
+    detail_des=(
+        '介绍：\n' '重启Bot框架\n' ' \n' '指令：\n' '- <ft color=(238,120,0)>l4重启</ft>'
+    ),
+)
+async def send_restart_msg(
+    bot: Bot,
+    event: MessageEvent,
+    matcher: Matcher,
+):
+    if not await SUPERUSER(bot, event):
+        return
+    logger.warning('开始执行[重启]')
+    qid = event.user_id
+    if len(event.get_session_id().split('_')) == 3:
+        send_id = event.get_session_id().split('_')[1]
+        send_type = 'group'
+    else:
+        send_id = qid
+        send_type = 'private'
+    await matcher.send('正在执行[l4重启]...')
+    await restart_genshinuid(send_type, str(send_id))
+
+
+@l4d_update.handle()
+@register_menu(
+    '更新插件',
+    'l4更新',
+    '手动更新插件',
+    detail_des=(
+        '介绍：\n'
+        '手动更新插件（执行 git pull）\n'
+        '每加上一个可选参数，执行等级加1\n'
+        '当执行等级≥1时会还原上次更改，等级≥2时会清空暂存\n'
+        ' \n'
+        '指令：\n'
+        '- <ft color=(238,120,0)>l4d</ft>'
+        '<ft color=(125,125,125)>(强行)(强制)</ft>'
+        '<ft color=(238,120,0)>更新</ft>'
+    ),
+)
+async def send_update_msg(
+    bot: Bot,
+    event: MessageEvent,
+    matcher: Matcher,
+    args: Tuple[Any, ...] = RegexGroup(),
+):
+    if not await SUPERUSER(bot, event):
+        return
+
+    logger.info('[l4d更新] 正在执行 ...')
+    level = 2
+    if args[1] is None:
+        level -= 1
+    if args[2] is None:
+        level -= 1
+    logger.info(f'[l4d更新] 更新等级为{level}')
+    await matcher.send(f'开始执行[l4d更新], 执行等级为{level}')
+    im = await draw_update_log_img(level)
+    if isinstance(im, str):
+        await matcher.finish(im)
+    elif isinstance(im, bytes):
+        await matcher.finish(MessageSegment.image(im))
+    else:
+        await matcher.finish('发生了未知错误,请联系管理员检查后台输出!')

nonebot_plugin_l4d2_server/l4d2_update/draw_update_log.py

@@ -0,0 +1,41 @@
+from pathlib import Path
+from typing import Union
+
+# from PIL import Image, ImageDraw
+
+from .update import update_from_git
+# from ..l4d2_image.image import convert_img
+# from ..l4d2_image.image import get_color_bg
+# from ..utils.genshin_fonts.genshin_fonts import genshin_font_origin
+
+R_PATH = Path(__file__).parent
+TEXT_PATH = R_PATH / 'texture2d'
+
+# gs_font_30 = genshin_font_origin(30)
+black_color = (24, 24, 24)
+
+log_config = {
+    'key': '✨🐛🎨⚡🍱♻️',
+    'num': 18,
+}
+
+log_map = {'✨': 'feat', '🐛': 'bug', '🍱': 'bento', '⚡️': 'zap', '🎨': 'art'}
+
+
+async def draw_update_log_img(
+    level: int = 0,
+    repo_path: Union[str, Path, None] = None,
+    is_update: bool = True,
+) -> Union[bytes, str]:
+    log_list = await update_from_git(level, repo_path, log_config, is_update)
+    if len(log_list) == 0:
+        return '更新失败!更多错误信息请查看控制台...\n' \
+               '>> 可以尝试使用\n' \
+               '>> [gs强制更新](危险)\n' \
+               '>> [gs强行强制更新](超级危险)!'
+
+    result = 'L4D2Bot 更新记录\n\n'
+    for log in log_list:
+        result += f'- {log[2:]}\n'
+
+    return result

nonebot_plugin_l4d2_server/l4d2_update/restart.py

@@ -0,0 +1,67 @@
+import os
+import sys
+import json
+import time
+import platform
+import subprocess
+from pathlib import Path
+
+# from ..utils.db_operation.db_operation import config_check
+
+bot_start = Path().cwd() / 'bot.py'
+restart_sh_path = Path().cwd() / 'gs_restart.sh'
+update_log_path = Path(__file__).parent / 'update_log.json'
+
+_restart_sh = '''#!/bin/bash
+kill -9 {}
+{} &'''
+
+
+async def get_restart_sh(extra: str) -> str:
+    args = f'{extra} {str(bot_start.absolute())}'
+    return _restart_sh.format(str(bot_start.absolute()), args)
+
+
+async def restart_genshinuid(send_type: str, send_id: str) -> None:
+    # extra = ''
+    # if await config_check('UsePoetry'):
+    #     extra = 'poetry run '
+    extra = sys.executable
+    restart_sh = await get_restart_sh(extra)
+    if not restart_sh_path.exists():
+        with open(restart_sh_path, "w", encoding="utf8") as f:
+            f.write(restart_sh)
+        if platform.system() == 'Linux':
+            os.system(f'chmod +x {str(restart_sh_path)}')
+            os.system(f'chmod +x {str(bot_start)}')
+    now_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))
+    update_log = {
+        'type': 'restart',
+        'msg': '重启完成!',
+        'send_type': send_type,
+        'send_to': send_id,
+        'time': now_time,
+    }
+    with open(str(update_log_path), 'w', encoding='utf-8') as f:
+        json.dump(update_log, f)
+    if platform.system() == 'Linux':
+        os.execl(str(restart_sh_path), ' ')
+    else:
+        pid = os.getpid()
+        subprocess.Popen(
+            f'taskkill /F /PID {pid} & {extra} {bot_start}',
+            shell=True,
+        )
+
+
+async def restart_message() -> dict:
+    if update_log_path.exists():
+        with open(update_log_path, 'r', encoding='utf-8') as f:
+            update_log = json.load(f)
+        msg = f'{update_log["msg"]}\n重启时间:{update_log["time"]}'
+        update_log['msg'] = msg
+        os.remove(update_log_path)
+        os.remove(restart_sh_path)
+        return update_log
+    else:
+        return {}

nonebot_plugin_l4d2_server/l4d2_update/update.py

@@ -0,0 +1,65 @@
+from pathlib import Path
+from typing import List, Union
+
+import git
+from nonebot.log import logger
+from git.exc import GitCommandError
+
+
+async def update_from_git(
+    level: int = 0,
+    repo_path: Union[str, Path, None] = None,
+    log_config: dict = {
+        'key': '✨🐛',
+        'num': 7,
+    },
+    is_update: bool = True,
+) -> List[str]:
+    if repo_path is None:
+        repo_path = Path(__file__).parents[2]
+    repo = git.Repo(repo_path)  # type: ignore
+    o = repo.remotes.origin
+
+    if is_update:
+        # 清空暂存
+        if level >= 2:
+            logger.warning('[gs更新] 正在执行 git clean --xdf')
+            repo.git.clean('-xdf')
+        # 还原上次更改
+        if level >= 1:
+            logger.warning('[gs更新] 正在执行 git reset --hard')
+            repo.git.reset('--hard')
+
+        try:
+            pull_log = o.pull()
+            logger.info(f'[gs更新] {pull_log}')
+        except GitCommandError as e:
+            logger.warning(e)
+            return []
+
+    commits = list(repo.iter_commits(max_count=40))
+    log_list = []
+    for commit in commits:
+        if isinstance(commit.message, str):
+            for key in log_config['key']:
+                if key in commit.message:
+                    log_list.append(commit.message.replace('\n', ''))
+                    if len(log_list) >= log_config['num']:
+                        break
+    return log_list
+
+
+async def update_genshinuid(
+    level: int = 0, repo_path: Union[str, Path, None] = None
+) -> str:
+    log_list = await update_from_git(level, repo_path)
+    if len(log_list) == 0:
+        return (
+            '更新失败!更多错误信息请查看控制台...\n '
+            '>> 可以尝试使用\n '
+            '>> [gs强制更新](危险)\n '
+            '>> [gs强行强制更新](超级危险)!'
+        )
+    log = '\n'.join(log_list)
+    logger.info(f'[gs更新]\n{log}')
+    return f'更新成功!\n >> 最近有效更新为:\n{log}'