akari-bot/modules/server/server.py

import re
import traceback

import aiohttp
import ujson as json

from core.builtins import ErrorMessage
from core.logger import Logger


async def server(msg, address, raw=False, showplayer=False, mode='j'):
    matchObj = re.match(r'(.*)[\s:](.*)', address, re.M | re.I)
    servers = []
    n = '\n'

    if matchObj:
        serip = matchObj.group(1)
        port1 = matchObj.group(2)
        port2 = matchObj.group(2)
    else:
        serip = address
        port1 = '25565'
        port2 = '19132'

    if mode == 'j':
        try:
            url = 'http://motd.wd-api.com/v1/java?host=' + serip + '&port=' + port1
            async with aiohttp.ClientSession() as session:
                async with session.get(url, timeout=aiohttp.ClientTimeout(total=20)) as req:
                    if req.status != 200:
                        Logger.error(await req.text())
                    else:
                        jejson = json.loads(await req.text())
                        try:
                            servers.append('[JE]')
                            if 'description' in jejson:
                                description = jejson['description']
                                if 'text' in description:
                                    servers.append(str(description['text']))
                                elif 'extra' in description:
                                    extra = description['extra']
                                    text = []
                                    for item in extra[:]:
                                        text.append(str(item['text']))
                                    servers.append(''.join(text))
                                else:
                                    servers.append(str(description))

                            if 'players' in jejson:
                                onlinesplayer = f"{msg.locale.t('server.message.player')}{str(jejson['players']['online'])} / {str(jejson['players']['max'])}"
                                servers.append(onlinesplayer)
                                if showplayer:
                                    playerlist = []
                                    if 'sample' in jejson['players']:
                                        for x in jejson['players']['sample']:
                                            playerlist.append(x['name'])
                                        servers.append(
                                            msg.locale.t('server.message.player.current') + '\n' + '\n'.join(playerlist))
                                    else:
                                        if jejson['players']['online'] == 0:
                                            servers.append(msg.locale.t('server.message.player.current.none'))
                            if 'version' in jejson:
                                versions = msg.locale.t('server.message.version') + jejson['version']['name']
                                servers.append(versions)
                            servers.append(serip + ':' + port1)
                        except Exception:
                            traceback.print_exc()
                            servers.append(str(ErrorMessage(msg.locale.t('server.message.error'))))
        except Exception:
            traceback.print_exc()
        if raw:
            return n.join(servers)
        return re.sub(r'§\w', "", n.join(servers))
    if mode == 'b':
        try:
            beurl = 'http://motd.wd-api.com/v1/bedrock?host=' + serip + '&port=' + port2
            async with aiohttp.ClientSession() as session2:
                async with session2.get(beurl, timeout=aiohttp.ClientTimeout(total=20)) as req:
                    if req.status != 200:
                        Logger.debug(await req.text())
                    else:
                        bemotd = await req.text()
                        bejson = json.loads(bemotd)
                        unpack_data = bejson['data'].split(';')
                        edition = unpack_data[0]
                        motd_1 = unpack_data[1]
                        version_name = unpack_data[3]
                        player_count = unpack_data[4]
                        max_players = unpack_data[5]
                        motd_2 = unpack_data[7]
                        game_mode = unpack_data[8]
                        bemsg = '[BE]\n' + \
                                motd_1 + ' - ' + motd_2 + \
                                '\n' + msg.locale.t('server.message.player') + player_count + '/' + max_players + \
                                '\n' + msg.locale.t('server.message.version') + edition + version_name + \
                                '\n' + msg.locale.t('server.message.gamemode') + game_mode
                        servers.append(bemsg)
                        servers.append(serip + ':' + port2)

        except Exception:
            traceback.print_exc()
        if raw:
            return n.join(servers)
        return re.sub(r'§\w', "", n.join(servers))