TotalFreedomBot/commands/miscellaneous.py

import asyncio
from datetime import datetime

import discord
from discord.ext import commands

from checks import is_dev, is_tf_developer
from functions import read_json, write_json


class Miscellaneous(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @is_dev()
    @commands.command()
    async def killbot(self, ctx):
        em = discord.Embed()
        em.description = 'Bot offline.'
        await ctx.send(embed=em)
        await self.bot.logout()
        return

    @is_tf_developer()
    @commands.command()
    async def telnetconfig(self, ctx, *args):
        em = discord.Embed()
        em.title = 'Telnet config'
        if not args or args[0] in ['reconnect', 'connect']:
            try:
                self.bot.telnet_object.connect()
                self.bot.telnet_object_2.connect()
            except Exception as e:
                em.description = f'Failed reconnection: {e}'
                em.colour = 0xFF0000
            else:
                em.description = 'Reconnection successful'
                em.colour = 0x00FF00
        elif args[0] == 'name':
            try:
                self.bot.telnet_object.session.close()
                self.bot.telnet_object_2.session.close()
                self.bot.telnet_object.connect(args[1])
                self.bot.telnet_object_2.connect(args[1])
                config = read_json('config')
                config['TELNET_USERNAME'] = self.bot.telnet_object.username
                write_json('config', config)
            except Exception as e:
                em.description = f'Failed config edit: {e}'
                em.colour = 0xFF0000
            else:
                em.description = 'Configuration successful'
                em.colour = 0x00FF00

        elif args[0] == 'test':
            command = ''
            for x in range(1, len(args)):
                command += f'{args[x]} '
            time_sent = str(datetime.utcnow().replace(microsecond=0))[11:]

            self.bot.telnet_object.session.write(
                bytes(command, 'ascii') + b"\r\n")
            self.bot.telnet_object.session.read_until(
                bytes(f'{time_sent} INFO]:', 'ascii'), 2)
            if ctx.channel == ctx.guild.get_channel(self.bot.server_chat):
                self.bot.telnet_object.session.read_until(
                    bytes('\r\n', 'ascii'), 2)
            next_line = self.bot.telnet_object.session.read_until(
                bytes('\r\n', 'ascii'), 2)
            em.description = f"Response from server: {next_line.decode('utf-8')}"
        else:
            em.description = f'Command **{args[0]}** not found.'
            em.colour = 0xFF0000

        await ctx.send(embed=em)

    @is_dev()
    @commands.command()
    async def debug(self, ctx, *, cmd):
        """Executes a line of code"""
        try:
            result = eval(cmd)
            if asyncio.iscoroutine(result):
                result = await result
            await ctx.send(f'''```py
{result}```''')
        except Exception as e:
            await ctx.send(f'''```py
{type(e).__name__}: {e}```''')


def setup(bot):
    bot.add_cog(Miscellaneous(bot))