94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
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))
|