searchbot-discord/extensions/core.py

360 lines
12 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
# tacibot core
# Handles all important main features of any bot.
'''Core File'''
import discord
import os
from discord.ext import commands
import time
import asyncio
import sys
import cpuinfo
import math
import psutil
from extensions.models.help import TaciHelpCommand
2020-03-19 18:16:22 +00:00
from typing import List, Optional
2020-03-02 18:11:15 +00:00
class Core(commands.Cog):
"""Provides all core features of a bot."""
def __init__(self, bot):
2020-03-02 18:11:15 +00:00
# Main Stuff
self.bot = bot
2020-03-02 20:47:58 +00:00
self.extensions_list = bot.extensions_list
2020-03-02 18:11:15 +00:00
self.emoji = "\U0001F4E6"
2020-03-02 18:11:15 +00:00
# Help Command
2020-03-02 04:58:05 +00:00
self._original_help_command = bot.help_command
2020-03-19 18:16:22 +00:00
if bot.custom_help:
bot.help_command = TaciHelpCommand()
2020-03-02 04:58:05 +00:00
bot.help_command.cog = self
2020-03-19 18:16:22 +00:00
def _create_tutorial(self, guild) -> str:
"""Creates the tutorial message."""
prefixes: str = f"`@{self.bot.user.name}`"
if self.bot.prefix:
others: str = ', '.join(f'`{p}`' for p in self.bot.prefix)
prefixes += f', {others}'
msg: str = (
f"**Hi!** Thanks for adding me to `{guild.name}`.\n\n"
f"I'm **{self.bot.user.name}** - _{self.bot.description}_\n\n"
f"My prefix{'es are' if self.bot.prefix else ' is'}: "
f"{prefixes}.\n\n"
"You may find more information with `help`.\n\n"
"You may also find our support server and repo with `about`"
)
if 'extensions.botlist' in self.extensions_list:
2020-03-20 15:21:49 +00:00
msg += ", or vote and review this bot with `vote`.\n\n"
else:
msg += ".\n\n"
msg += (
2020-03-19 18:16:22 +00:00
"_Please note that this bot may log errors, guild names, "
"command calls/contents, and the names of command users "
"for debug and maintenance purposes. "
"These logs are shared with nobody "
"other than those who help develop this bot. "
"If you do not agree to this, please remove this bot._\n\n"
"_You may recall this message at any time with `tutorial`._"
)
return msg
# function lifted from StackOverflow
def _humanbytes(self, B) -> str:
"""Return the given bytes as a human friendly KB, MB, GB, or TB string."""
B = float(B)
KB = float(1024)
MB = float(KB ** 2) # 1,048,576
GB = float(KB ** 3) # 1,073,741,824
TB = float(KB ** 4) # 1,099,511,627,776
if B < KB:
return '{0} {1}'.format(
B, 'Bytes' if 0 == B > 1 else 'Byte')
elif KB <= B < MB:
return '{0:.2f} KB'.format(B/KB)
elif MB <= B < GB:
return '{0:.2f} MB'.format(B/MB)
elif GB <= B < TB:
return '{0:.2f} GB'.format(B/GB)
elif TB <= B:
return '{0:.2f} TB'.format(B/TB)
2020-03-19 18:16:22 +00:00
else:
return 'ERROR'
@commands.command(aliases=['info', 'source', 'server'])
async def about(self, ctx):
"""Returns information about this bot."""
2020-03-20 15:32:26 +00:00
# Header
2020-03-20 15:24:03 +00:00
msg = (
2020-03-21 15:21:01 +00:00
f"__**{self.bot.user.name}**__ - _{self.bot.description}_\n"
"Written in `discord.py`.\n\n"
2020-03-20 15:24:03 +00:00
f"This instance by **{self.bot.appinfo.owner}.**\n\n"
)
2020-03-20 15:32:26 +00:00
# Repo and support server
2020-03-19 18:16:22 +00:00
if self.bot.repo:
msg += f"**Source Code:** _<{self.bot.repo}>_\n"
if self.bot.support_server:
msg += f"**Support Server:** _<{self.bot.support_server}>_\n\n"
2020-03-20 15:32:26 +00:00
# Properly handle blank prefixes
if ctx.prefix == '':
prefix: str = ''
else:
prefix: str = f"**{ctx.prefix}**"
# Footer
msg += (
f"_See {prefix}`help` for help, `invite` to add the bot, "
"and `stats` for statistics._"
)
# Sending
await ctx.send(msg)
@commands.command(aliases=['addbot', 'connect', 'join'])
async def invite(self, ctx):
"""Gets a link to invite this bot to your server."""
msg = (
"**Thanks for checking me out!**\n\n"
"Use the following link to add me:\n"
2020-03-20 15:32:26 +00:00
"*<https://discordapp.com/oauth2/authorize"
f"?client_id={self.bot.user.id}&scope=bot"
)
2020-03-19 18:16:22 +00:00
if self.bot.perms:
msg += f"&permissions={self.bot.perms}>*"
else:
msg += ">*"
await ctx.send(msg)
2020-03-19 18:16:22 +00:00
@commands.command()
async def tutorial(self, ctx):
"""Resends the tutorial message."""
if ctx.guild:
msg: str = self._create_tutorial(ctx.guild)
else:
msg: str = "**Cannot send tutorial in DMs!**"
await ctx.send(msg)
@commands.command()
async def stats(self, ctx):
"""Provides statistics on the bot itself."""
mem = psutil.virtual_memory()
currproc = psutil.Process(os.getpid())
total_ram = self._humanbytes(mem[0])
available_ram = self._humanbytes(mem[1])
usage = self._humanbytes(currproc.memory_info().rss)
msg = f"""
```
Total RAM: {total_ram}
Available RAM: {available_ram}
RAM used by bot: {usage}
2020-03-21 15:21:01 +00:00
Number of bot commands: {len(self.bot.commands)}
Number of extensions present: {len(self.bot.cogs)}
Guild count: {len(self.bot.guilds)}
```
"""
await ctx.send(msg)
@commands.command()
async def ping(self, ctx):
"""Checks the ping of the bot."""
before = time.monotonic()
pong = await ctx.send("...")
after = time.monotonic()
ping = (after - before) * 1000
await pong.edit(content="`PING discordapp.com {}ms`".format(int(ping)))
2020-03-19 18:16:22 +00:00
@commands.group(aliases=['extensions', 'ext'],
2020-03-02 20:47:58 +00:00
invoke_without_command=True)
@commands.is_owner()
2020-03-19 18:16:22 +00:00
async def extend(self, ctx, name: str = None):
2020-03-02 20:49:25 +00:00
"""Provides status of extensions and lets you hotswap extensions."""
2020-03-02 20:47:58 +00:00
# Provides status of extension
if name is not None:
status = "is" if name in self.extensions_list else "is not"
msg = f"**{name}** {status} currently loaded and/or existent."
2020-03-02 20:47:58 +00:00
# Handles empty calls
else:
msg = (
"**Nothing was provided!**\n\n"
"Please provide an extension name for status, "
"or provide a subcommand."
)
# Sends completed message
await ctx.send(msg)
@extend.command(aliases=['le', 'l'])
@commands.is_owner()
async def load(self, ctx, name: str):
"""Load an extension into the bot."""
m = await ctx.send(f'Loading {name}')
2020-03-02 20:47:58 +00:00
extension_name = f'extensions.{name}'
if extension_name not in self.extensions_list:
try:
self.bot.load_extension(extension_name)
2020-03-02 20:47:58 +00:00
self.extensions_list.append(extension_name)
await m.edit(content='Extension loaded.')
except Exception as e:
await m.edit(
content=f'Error while loading {name}\n`{type(e).__name__}: {e}`')
else:
await m.edit(content='Extension already loaded.')
2020-03-02 20:47:58 +00:00
@extend.command(aliases=["ule", "ul"])
@commands.is_owner()
async def unload(self, ctx, name: str):
"""Unload an extension from the bot."""
m = await ctx.send(f'Unloading {name}')
2020-03-02 20:47:58 +00:00
extension_name = f'extensions.{name}'
if extension_name in self.extensions_list:
self.bot.unload_extension(extension_name)
2020-03-02 20:47:58 +00:00
self.extensions_list.remove(extension_name)
await m.edit(content='Extension unloaded.')
else:
await m.edit(content='Extension not found or not loaded.')
2020-03-02 20:47:58 +00:00
@extend.command(aliases=["rle", "rl"])
@commands.is_owner()
async def reload(self, ctx, name: str):
"""Reload an extension of the bot."""
m = await ctx.send(f'Reloading {name}')
2020-03-02 20:47:58 +00:00
extension_name = f'extensions.{name}'
if extension_name in self.extensions_list:
self.bot.unload_extension(extension_name)
try:
self.bot.load_extension(extension_name)
await m.edit(content='Extension reloaded.')
except Exception as e:
2020-03-02 20:47:58 +00:00
self.extensions_list.remove(extension_name)
await m.edit(
content=f'Failed to reload extension\n`{type(e).__name__}: {e}`')
else:
await m.edit(content='Extension isn\'t loaded.')
@extend.command(name='list')
async def list_cmd(self, ctx):
"""Lists all extensions loaded by the bot."""
# Message Construction
msg = "**Loaded Extensions**\n\n"
msg += '\n'.join(f'`{e}`' for e in self.extensions_list)
msg += "\n\n_See the other subcommands of this command to manage them._"
# Message Sending
await ctx.send(msg)
@commands.command(aliases=['exit', 'reboot'])
@commands.is_owner()
async def restart(self, ctx):
"""Turns the bot off."""
2020-03-02 20:26:45 +00:00
await ctx.send(":zzz: **Restarting...**")
exit()
2020-03-02 18:34:27 +00:00
@commands.command()
@commands.is_owner()
async def leave(self, ctx):
"""Makes the bot leave the server this was called in."""
2020-03-19 18:16:22 +00:00
if ctx.guild:
await ctx.send(
"\U0001F4A8 **Leaving server.** "
2020-03-19 18:16:22 +00:00
"_If you want me back, add me or get an admin to._"
)
await ctx.guild.leave()
else:
await ctx.send(
2020-03-19 18:16:22 +00:00
"**Can't leave!** _This channel is not inside a guild._"
)
2020-03-02 18:34:27 +00:00
2020-03-02 04:58:05 +00:00
def cog_unload(self):
self.bot.help_command = self._original_help_command
2020-03-19 18:16:22 +00:00
@commands.Cog.listener()
async def on_guild_join(self, guild):
"""Sends owner notification and guild tutorial."""
# Prerequisites
guild_msg: str = self._create_tutorial(guild)
channel: Optional[discord.TextChannel] = None
owner: discord.Member = guild.owner
# Tutorial Message
# Get text channels
text_channels = []
for c in guild.channels:
if type(c) is discord.TextChannel:
text_channels.append(c)
# Sets channel to general if it exists
for c in guild.channels:
if c.name == 'general':
channel = c
# XXX This looks like garbage
# Else posts in first open channel
if not channel:
for c in guild.channels:
if c.permissions_for(guild.me).send_messages:
channel = c
# Send tutorial message
if channel:
await channel.send(guild_msg)
else:
guild_msg += (
"\n\n_I am sending this message to you as there were no "
"channels I could send messages to in your server. "
"Please give me send message permissions in the channels "
"You wish to use me in!_"
)
await guild.owner.send(guild_msg)
return # Ends here if there are no good channels to send to
# Owner Disclosure
# Message Building
owner_msg = (
"**Hi there!**\n\n"
f"I am **{self.bot.user.name}** - _{self.bot.description}_\n\n"
"I am messaging you to inform you I was added to your server, "
f"`{guild.name}`, by someone "
"with **Manage Server** permissions.\n\n"
f"I have sent a tutorial message to `{channel.name}` "
"describing how I may be used.\n\n"
"If you do not wish to have me there, "
"simply kick me from the server.\n\n"
"_Thanks for your time!_"
)
# Send owner disclosure
await guild.owner.send(owner_msg)
def setup(bot):
bot.add_cog(Core(bot))