290 lines
10 KiB
Python
290 lines
10 KiB
Python
import discord
|
|
import requests
|
|
|
|
from checks import *
|
|
from discord.ext import commands
|
|
from datetime import datetime
|
|
from functions import *
|
|
from unicode import *
|
|
|
|
|
|
class ServerCommands(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@commands.command()
|
|
@is_liaison()
|
|
async def eventhost(self, ctx, user: discord.Member):
|
|
'Add or remove event host role - liaison only'
|
|
eventhostrole = ctx.guild.get_role(event_host)
|
|
if eventhostrole in user.roles:
|
|
await user.remove_roles(eventhostrole)
|
|
await ctx.send(f'```Succesfully took {eventhostrole.name} from {user.name}```')
|
|
else:
|
|
await user.add_roles(eventhostrole)
|
|
await ctx.send(f'```Succesfully added {eventhostrole.name} to {user.name}```')
|
|
|
|
@commands.command()
|
|
@is_creative_designer()
|
|
async def masterbuilder(self, ctx, user: discord.Member):
|
|
'Add or remove master builder role - ECD only'
|
|
master_builder_role = ctx.guild.get_role(master_builder)
|
|
if master_builder_role in user.roles:
|
|
await user.remove_roles(master_builder_role)
|
|
await ctx.send(f'```Succesfully took {master_builder_role.name} from {user.name}```')
|
|
else:
|
|
await user.add_roles(master_builder_role)
|
|
await ctx.send(f'```Succesfully added {master_builder_role.name} to {user.name}```')
|
|
|
|
@commands.command()
|
|
@is_staff()
|
|
async def serverban(self, ctx, user: discord.Member):
|
|
'Add or remove server banned role'
|
|
serverbannedrole = ctx.guild.get_role(server_banned)
|
|
if serverbannedrole in user.roles:
|
|
await user.remove_roles(serverbannedrole)
|
|
await ctx.send(f'Took Server Banned role from {user.name}')
|
|
else:
|
|
await user.add_roles(serverbannedrole)
|
|
await ctx.send(f'Added Server Banned role to {user.name}')
|
|
|
|
@commands.command()
|
|
@is_staff()
|
|
async def start(self, ctx):
|
|
'Starts the server'
|
|
em = discord.Embed()
|
|
try:
|
|
attempt = hit_endpoint('start')
|
|
except Exception as e:
|
|
em.title = 'Command error'
|
|
em.colour = 0xFF0000
|
|
em.description = 'Something went wrong'
|
|
print(f'Error while starting server: {e}')
|
|
await ctx.send(embed=em)
|
|
else:
|
|
if 'error' in attempt.lower():
|
|
em.title = 'Command error'
|
|
em.colour = 0xFF0000
|
|
em.description = f'{attempt}'
|
|
await ctx.send(embed=em)
|
|
else:
|
|
em.title = 'Success'
|
|
em.colour = 0x00FF00
|
|
em.description = f'{attempt}'
|
|
await ctx.send(embed=em)
|
|
|
|
@commands.command()
|
|
async def uptime(self, ctx):
|
|
"Returns the uptime of the VPS"
|
|
em = discord.Embed()
|
|
em.title = 'VPS Uptime Information'
|
|
em.description = hit_endpoint('uptime')
|
|
await ctx.send(embed=em)
|
|
|
|
@commands.command()
|
|
@is_staff()
|
|
async def stop(self, ctx):
|
|
'Stops the server'
|
|
em = discord.Embed()
|
|
try:
|
|
attempt = hit_endpoint('stop')
|
|
except Exception as e:
|
|
em.title = 'Command error'
|
|
em.colour = 0xFF0000
|
|
em.description = 'Something went wrong'
|
|
print(f'Error while stopping server: {e}')
|
|
await ctx.send(embed=em)
|
|
else:
|
|
if 'error' in attempt.lower():
|
|
em.title = 'Command error'
|
|
em.colour = 0xFF0000
|
|
em.description = f'{attempt}'
|
|
await ctx.send(embed=em)
|
|
else:
|
|
em.title = 'Success'
|
|
em.colour = 0x00FF00
|
|
em.description = f'{attempt}'
|
|
await ctx.send(embed=em)
|
|
|
|
@commands.command(aliases=['adminconsole', 'ac'])
|
|
@is_staff()
|
|
async def telnet(self, ctx, *args):
|
|
'mv, gtfo, kick, mute or warn from discord'
|
|
em = discord.Embed()
|
|
command = ''
|
|
for arg in args:
|
|
command += f'{arg} '
|
|
try:
|
|
if args[0] in ['mute', 'stfu', 'gtfo', 'ban', 'unban', 'unmute', 'smite', 'noob', 'tban', 'tempban', 'warn', 'mv', 'kick', 'cc', 'say']:
|
|
self.bot.telnet_object.session.write(
|
|
bytes(command, 'ascii') + b"\r\n")
|
|
elif args[0] == 'slconfig':
|
|
if args[1] not in ['add', 'remove']:
|
|
raise no_permission(['IS_SENIOR_ADMIN'])
|
|
else:
|
|
self.bot.telnet_object.session.write(
|
|
bytes(command, 'ascii') + b"\r\n")
|
|
else:
|
|
raise no_permission(['IS_SENIOR_ADMIN'])
|
|
except Exception as e:
|
|
em.title = 'Command error'
|
|
em.colour = 0xFF0000
|
|
em.description = f'{e}'
|
|
await ctx.send(embed=em)
|
|
else:
|
|
em.title = 'Success'
|
|
em.colour = 0x00FF00
|
|
em.description = 'Command sent.'
|
|
await ctx.send(embed=em)
|
|
|
|
@commands.command()
|
|
@is_senior()
|
|
async def kill(self, ctx):
|
|
'Kills the server'
|
|
em = discord.Embed()
|
|
try:
|
|
attempt = hit_endpoint('kill')
|
|
except Exception as e:
|
|
em.title = 'Command error'
|
|
em.colour = 0xFF0000
|
|
em.description = 'Something went wrong'
|
|
print(f'Error while killing server: {e}')
|
|
await ctx.send(embed=em)
|
|
else:
|
|
if 'error' in attempt.lower():
|
|
em.title = 'Command error'
|
|
em.colour = 0xFF0000
|
|
em.description = '{attempt}'
|
|
print(f'Error while killing server: {e}')
|
|
await ctx.send(embed=em)
|
|
else:
|
|
em.title = 'Success'
|
|
em.colour = 0x00FF00
|
|
em.description = f'{attempt}'
|
|
await ctx.send(embed=em)
|
|
|
|
@commands.command()
|
|
@is_staff()
|
|
async def restart(self, ctx):
|
|
'Restarts the server'
|
|
em = discord.Embed()
|
|
try:
|
|
self.bot.telnet_object.session.write(
|
|
bytes('restart', 'ascii') + b"\r\n")
|
|
except Exception as e:
|
|
em.title = 'Command error'
|
|
em.colour = 0xFF0000
|
|
em.description = 'Something went wrong'
|
|
print(f'Error while restarting server: {e}')
|
|
await ctx.send(embed=em)
|
|
else:
|
|
em.title = 'Success'
|
|
em.colour = 0x00FF00
|
|
em.description = 'Server restarting.'
|
|
await ctx.send(embed=em)
|
|
|
|
@commands.command()
|
|
@is_senior()
|
|
async def console(self, ctx, *, command):
|
|
'Send a command as console'
|
|
'''await ctx.send(f'```:[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: {ctx.author.name} issued server command: /{command}```')'''
|
|
em = discord.Embed()
|
|
try:
|
|
self.bot.telnet_object.session.write(
|
|
bytes(command, 'ascii') + b"\r\n")
|
|
except Exception as e:
|
|
em.title = 'Command error'
|
|
em.colour = 0xFF0000
|
|
em.description = f'{e}'
|
|
await ctx.send(embed=em)
|
|
else:
|
|
em.title = 'Success'
|
|
em.colour = 0x00FF00
|
|
em.description = 'Command sent.'
|
|
await ctx.send(embed=em)
|
|
|
|
@commands.command(aliases=['status'])
|
|
async def state(self, ctx):
|
|
'Gets the current status of the Server'
|
|
em = discord.Embed()
|
|
try:
|
|
requests.get(
|
|
"http://play.totalfreedom.me:28966/list?json=true", timeout=5).json()
|
|
except:
|
|
em.description = 'Server is offline'
|
|
em.colour = 0xFF0000
|
|
else:
|
|
em.description = 'Server is online'
|
|
em.colour = 0x00FF00
|
|
await ctx.send(embed=em)
|
|
|
|
@commands.command(name='list')
|
|
async def online(self, ctx):
|
|
'Gives a list of online players.'
|
|
em = discord.Embed()
|
|
em.title = "Player List"
|
|
try:
|
|
json = requests.get(
|
|
"http://play.totalfreedom.me:28966/list?json=true", timeout=5).json()
|
|
if json["online"] == 0:
|
|
em.description = "There are no online players"
|
|
else:
|
|
em.description = f"There are {json['online']} / {json['max']} online players"
|
|
ranks = list(json.keys())
|
|
for rank in ranks:
|
|
if rank not in ['max', 'online'] and json[rank]:
|
|
rank = rank.split('_')
|
|
for word in range(len(rank)):
|
|
rank[word] = rank[word].capitalize()
|
|
em = format_list_entry(
|
|
em, json[rank], f'{" ".join(rank)}')
|
|
|
|
except:
|
|
em.description = 'Server is offline'
|
|
await ctx.send(embed=em)
|
|
|
|
@commands.command()
|
|
async def ip(self, ctx):
|
|
'Returns the server IP'
|
|
await ctx.send(embed=discord.Embed(description='play.totalfreedom.me', title='Server IP'))
|
|
# pass #discordSRV responds already.
|
|
|
|
@commands.command()
|
|
@is_staff()
|
|
async def archivereports(self, ctx):
|
|
"""Archive all in-game reports older than 24 hours"""
|
|
count = 0
|
|
reports_channel = self.bot.get_channel(reports_channel_id)
|
|
archived_reports_channel = self.bot.get_channel(
|
|
archived_reports_channel_id)
|
|
await ctx.channel.trigger_typing()
|
|
async for report in reports_channel.history(limit=100):
|
|
try:
|
|
embed = report.embeds[0]
|
|
except:
|
|
await report.delete()
|
|
time = embed.timestamp
|
|
difference = datetime.now() - time
|
|
if difference.days >= 0:
|
|
await report.delete()
|
|
await archived_reports_channel.send("Message archived because it is older than 24 hours", embed=embed)
|
|
count += 1
|
|
await ctx.send("Archived **{}** reports that are older than 24 hours".format(count))
|
|
|
|
@commands.command()
|
|
@is_mod_or_has_perms()
|
|
async def fixreports(self, ctx):
|
|
await ctx.channel.trigger_typing()
|
|
reports_channel = self.bot.get_channel(reports_channel_id)
|
|
messages = await reports_channel.history(limit=500).flatten()
|
|
fixed = 0
|
|
for message in messages:
|
|
if len(message.reactions) == 0 and message.author == message.guild.me:
|
|
await message.add_reaction(clipboard)
|
|
fixed += 1
|
|
await ctx.send(f'Fixed **{fixed}** reports')
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(ServerCommands(bot))
|