akari-bot/modules/server/__init__.py

import asyncio
import re

from core.builtins import Bot
from core.component import module
from core.dirty_check import check
from .server import server

s = module('server', alias='s', developers=['_LittleC_', 'OasisAkari'])


@s.handle('<ServerIP:Port> [-r] [-p] {{server.help}}',
          options_desc={'-r': '{server.option.r}', '-p': '{server.option.p}'})
async def main(msg: Bot.MessageSession):
    enabled_addon = msg.options.get('server_revoke')
    if enabled_addon is None:
        enabled_addon = True
    gather_list = []
    matchObj = re.match(r'(.*)[\s:](.*)', msg.parsed_msg["<ServerIP:Port>"], re.M | re.I)
    is_local_ip = False
    server_address = msg.parsed_msg["<ServerIP:Port>"]
    if matchObj:
        server_address = matchObj.group(1)
    if server_address == 'localhost':
        is_local_ip = True
    matchserip = re.match(r'(.*?)\.(.*?)\.(.*?)\.(.*?)', server_address)
    if matchserip:
        if matchserip.group(1) == '192':
            if matchserip.group(2) == '168':
                is_local_ip = True
        if matchserip.group(1) == '172':
            if 16 <= int(matchserip.group(2)) <= 31:
                is_local_ip = True
        if matchserip.group(1) == '10':
            if 0 <= int(matchserip.group(2)) <= 255:
                is_local_ip = True
        if matchserip.group(1) == '127':
            is_local_ip = True
        if matchserip.group(1) == '0':
            if matchserip.group(2) == '0':
                if matchserip.group(3) == '0':
                    is_local_ip = True
    if is_local_ip:
        return await msg.sendMessage(msg.locale.t('server.message.local_ip'))
    sm = ['j', 'b']
    for x in sm:
        gather_list.append(asyncio.ensure_future(s(
            msg, f'{msg.parsed_msg["<ServerIP:Port>"]}', msg.parsed_msg.get('-r', False), msg.parsed_msg.get('-p', False
                                                                                                             ), x,
            enabled_addon)))
    g = await asyncio.gather(*gather_list)
    if g == ['', '']:
        msg_ = msg.locale.t('server.message.none')
        if msg.Feature.delete and enabled_addon:
            msg_ += '\n' + msg.locale.t('server.message.revoke')
        send = await msg.sendMessage(msg_)
        if msg.Feature.delete and enabled_addon:
            await msg.sleep(90)
            await send.delete()
            await msg.finish()


@s.handle('revoke <enable|disable> {{server.help.revoke}}')
async def revoke(msg: Bot.MessageSession):
    if msg.parsed_msg.get('<enable|disable>') == 'enable':
        msg.data.edit_option('server_revoke', True)
        await msg.finish(msg.locale.t('server.message.revoke.enable'))
    elif msg.parsed_msg.get('<enable|disable>') == 'disable':
        msg.data.edit_option('server_revoke', False)
        await msg.finish(msg.locale.t('server.message.revoke.disable'))


async def s(msg: Bot.MessageSession, address, raw, showplayer, mode, enabled_addon):
    sendmsg = await server(msg, address, raw, showplayer, mode)
    if sendmsg != '':
        sendmsg = await check(sendmsg)
        for x in sendmsg:
            m = x['content']
            if msg.Feature.delete and enabled_addon:
                m += '\n' + msg.locale.t('server.message.revoke')
            send = await msg.sendMessage(m)
            if msg.Feature.delete and enabled_addon:
                await msg.sleep(90)
                await send.delete()
                await msg.finish()
    return sendmsg