263 lines
15 KiB
Python
263 lines
15 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
|
|
from geeksbot.imports.utils import Paginator, Book
|
|
|
|
|
|
class Tickets(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@commands.command()
|
|
async def request(self, ctx, *, message=None):
|
|
if not ctx.guild:
|
|
await ctx.send('This command must be run from inside a guild.')
|
|
return
|
|
|
|
if not message:
|
|
await ctx.send('Please include a message containing your request')
|
|
return
|
|
|
|
if len(message) > 1000:
|
|
await ctx.send('Request is too long, please keep your request to less than 1000 characters.')
|
|
return
|
|
|
|
data = {
|
|
'author': ctx.author.id,
|
|
'message': ctx.message.id,
|
|
'channel': ctx.channel.id,
|
|
'content': message
|
|
}
|
|
msg_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/messages/{ctx.message.id}/wait/', headers=self.bot.auth_header)
|
|
if msg_resp.status == 404:
|
|
error = await msg_resp.json()
|
|
await ctx.send(error['details'])
|
|
return
|
|
|
|
resp = await self.bot.aio_session.post(f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/', headers=self.bot.auth_header, json=data)
|
|
|
|
if resp.status == 201:
|
|
admin_channel_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/channels/{ctx.guild.id}/admin/', headers=self.bot.auth_header)
|
|
request = await resp.json()
|
|
|
|
if admin_channel_resp.status == 200:
|
|
admin_chan_data = await admin_channel_resp.json()
|
|
msg = f''
|
|
admin_roles_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/guilds/{ctx.guild.id}/roles/admin/', headers=self.bot.auth_header)
|
|
if admin_roles_resp.status == 200:
|
|
admin_roles_data = await admin_roles_resp.json()
|
|
for role in admin_roles_data:
|
|
msg += f'{ctx.guild.get_role(int(role["id"])).mention} '
|
|
msg += f"New Request ID: {request['id']} " \
|
|
f"{ctx.author.mention} has requested assistance: \n" \
|
|
f"```{request['content']}``` \n" \
|
|
f"Requested at: {request['requested_at'].split('.')[0].replace('T', ' ')} GMT\n" \
|
|
f"In {ctx.guild.get_channel(int(request['channel'])).name}"
|
|
admin_chan = ctx.guild.get_channel(int(admin_chan_data['id']))
|
|
await admin_chan.send(msg)
|
|
await ctx.send(f'{ctx.author.mention} The admin have received your request.\n'
|
|
f'If you would like to update or close your request please reference Request ID `{request["id"]}`')
|
|
|
|
@commands.command(aliases=['comment'])
|
|
async def update(self, ctx, request_id=None, *, comment: str = None):
|
|
try:
|
|
request_id = int(request_id)
|
|
except ValueError:
|
|
await ctx.send("Please include the ID of the request you would like to update as the first thing after the command.")
|
|
return
|
|
|
|
if not comment:
|
|
await ctx.send("There is nothing to update since you didn't include a message.")
|
|
return
|
|
|
|
data = {
|
|
'author': ctx.author.id,
|
|
'content': comment
|
|
}
|
|
|
|
comment_resp = await self.bot.aio_session.post(f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/{request_id}/comments/', headers=self.bot.auth_header, json=data)
|
|
|
|
if comment_resp.status == 201:
|
|
comment = await comment_resp.json()
|
|
admin_channel_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/channels/{ctx.guild.id}/admin/',
|
|
headers=self.bot.auth_header)
|
|
|
|
if admin_channel_resp.status == 200:
|
|
admin_channel_data = await admin_channel_resp.json()
|
|
admin_channel = ctx.guild.get_channel(int(admin_channel_data['id']))
|
|
if admin_channel:
|
|
request_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/{request_id}/', headers=self.bot.auth_header)
|
|
pag = Paginator(self.bot, prefix='```md', suffix='```')
|
|
header = f'{ctx.author.mention} has commented on request {request_id}\n'
|
|
if request_resp.status == 200:
|
|
request = await request_resp.json()
|
|
requestor = ctx.guild.get_member(int(request["author"]))
|
|
header += (f'Original Request by {requestor.mention if requestor else "`User cannot be found`"}:\n'
|
|
f'```{request["content"]}```')
|
|
pag.set_header(header)
|
|
|
|
if request.get('comments'):
|
|
comments = request['comments']
|
|
for comment in comments:
|
|
author = ctx.guild.get_member(int(comment['author']))
|
|
pag.add(f'{author.display_name}: {comment["content"]}', keep_intact=True)
|
|
if ctx.author != requestor and requestor:
|
|
for page in pag.pages(page_headers=False):
|
|
await requestor.send(page)
|
|
book = Book(pag, (None, admin_channel, self.bot, ctx.message))
|
|
await book.create_book()
|
|
await ctx.send(f'{ctx.author.mention} Your comment has been added to the request.')
|
|
|
|
@commands.command(name='requests_list', aliases=['rl'])
|
|
async def _requests_list(self, ctx, closed: str = ''):
|
|
pag = Paginator(self.bot)
|
|
admin_roles_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/guilds/{ctx.guild.id}/roles/admin/', headers=self.bot.auth_header)
|
|
if admin_roles_resp.status == 200:
|
|
admin_roles_data = await admin_roles_resp.json()
|
|
admin_roles = [ctx.guild.get_role(int(role['id'])) for role in admin_roles_data]
|
|
if any([role in ctx.author.roles for role in admin_roles]):
|
|
requests_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/', headers=self.bot.auth_header)
|
|
if requests_resp.status == 200:
|
|
requests_data = await requests_resp.json()
|
|
requests_list = requests_data['requests'] if isinstance(requests_data, dict) else requests_data
|
|
while isinstance(requests_data, dict) and requests_data.get('next'):
|
|
requests_resp = await self.bot.aio_session.get(
|
|
requests_data['next'], headers=self.bot.auth_header)
|
|
if requests_resp.status == 200:
|
|
requests_data = await requests_resp.json()
|
|
requests_list.extend(requests_data['requests'] if isinstance(requests_data, dict) else requests_data)
|
|
for request in requests_list:
|
|
member = discord.utils.get(ctx.guild.members, id=int(request['author']))
|
|
title = (f"<{'Request ID':^20} {'Requested By':^20}>\n"
|
|
f"<{request['id']:^20} {member.display_name if member else 'None':^20}>")
|
|
orig_channel = ctx.guild.get_channel(int(request.get('channel')))
|
|
comments_count_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/{request["id"]}/comments/count/', headers=self.bot.auth_header)
|
|
pag.add(f"\n\n{title}\n"
|
|
f"{request['content']}\n\n"
|
|
f"Comments: {await comments_count_resp.text() if comments_count_resp.status == 200 else 0}\n"
|
|
f"Requested at: "
|
|
f"{request['requested_at'].split('.')[0].replace('T', ' ')} GMT\n"
|
|
f"In {orig_channel.name if orig_channel else 'N/A'}", keep_intact=True)
|
|
pag.add(f'\n\uFFF8\nThere are currently {len(requests_list)} requests open.')
|
|
else:
|
|
pag.add('There are no open requests for this guild.', keep_intact=True)
|
|
else:
|
|
requests_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/user/{ctx.author.id}/',
|
|
headers=self.bot.auth_header)
|
|
if requests_resp.status == 200:
|
|
requests_data = await requests_resp.json()
|
|
requests_list = requests_data['requests'] if isinstance(requests_data, dict) else requests_data
|
|
while isinstance(requests_data, dict) and requests_data.get('next'):
|
|
requests_resp = await self.bot.aio_session.get(
|
|
requests_data['next'], headers=self.bot.auth_header)
|
|
if requests_resp.status == 200:
|
|
requests_data = await requests_resp.json()
|
|
requests_list.extend(
|
|
requests_data['requests'] if isinstance(requests_data, dict) else requests_data)
|
|
for request in requests_list:
|
|
title = (f"<{'Request ID':^20}>\n"
|
|
f"<{request['id']:^20}>")
|
|
orig_channel = ctx.guild.get_channel(int(request.get('channel')))
|
|
comments_count_resp = await self.bot.aio_session.get(
|
|
f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/{request["id"]}/comments/count/',
|
|
headers=self.bot.auth_header)
|
|
pag.add(f"\n\n{title}\n"
|
|
f"{request['content']}\n\n"
|
|
f"Comments: {await comments_count_resp.text() if comments_count_resp.status == 200 else 0}\n"
|
|
f"Requested at: "
|
|
f"{request['requested_at'].split('.')[0].replace('T', ' ')} GMT\n"
|
|
f"In {orig_channel.name if orig_channel else 'N/A'}", keep_intact=True)
|
|
pag.add(f'\n\uFFF8\nYou currently have {len(requests_list)} requests open.')
|
|
else:
|
|
pag.add('You have no open requests for this guild.', keep_intact=True)
|
|
for page in pag.pages():
|
|
await ctx.send(page)
|
|
|
|
@commands.command()
|
|
async def close(self, ctx, *, ids=None):
|
|
if not ids:
|
|
await ctx.send('Please include at least one Request ID to close.')
|
|
return
|
|
|
|
admin = False
|
|
admin_roles_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/guilds/{ctx.guild.id}/roles/admin/',
|
|
headers=self.bot.auth_header)
|
|
if admin_roles_resp.status == 200:
|
|
admin_roles_data = await admin_roles_resp.json()
|
|
admin_roles = [ctx.guild.get_role(int(role['id'])) for role in admin_roles_data]
|
|
if any([role in ctx.author.roles for role in admin_roles]):
|
|
admin = True
|
|
|
|
ids = [id.strip() for id in ids.replace(' ', '').split(',')]
|
|
|
|
for id in ids:
|
|
request_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/{id}/', headers=self.bot.auth_header)
|
|
if request_resp.status == 200:
|
|
request = await request_resp.json()
|
|
requestor = ctx.guild.get_member(int(request['author']))
|
|
if requestor == ctx.author or admin:
|
|
data = {
|
|
'completed_by': ctx.author.id
|
|
}
|
|
delete_resp = await self.bot.aio_session.delete(f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/{id}/', headers=self.bot.auth_header, json=data)
|
|
if delete_resp.status == 202:
|
|
delete_data = await delete_resp.json()
|
|
if delete_data['completed']:
|
|
await ctx.send(f'Request {id} closed.')
|
|
await requestor.send(f'{ctx.author.display_name} has closed request {id} which was '
|
|
f'opened by you in the '
|
|
f'{ctx.guild.get_channel(int(request["channel"])).name} '
|
|
f'channel.'
|
|
f'```{request["content"]}```'
|
|
f'If there are any issues please open a new request.')
|
|
else:
|
|
await ctx.send('That is not your request to close.')
|
|
|
|
@commands.command()
|
|
async def view(self, ctx, id=None):
|
|
if not id:
|
|
await ctx.send('Please include the id of the request you would like to view.')
|
|
return
|
|
|
|
admin = False
|
|
admin_roles_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/guilds/{ctx.guild.id}/roles/admin/',
|
|
headers=self.bot.auth_header)
|
|
if admin_roles_resp.status == 200:
|
|
admin_roles_data = await admin_roles_resp.json()
|
|
admin_roles = [ctx.guild.get_role(int(role['id'])) for role in admin_roles_data]
|
|
if any([role in ctx.author.roles for role in admin_roles]):
|
|
admin = True
|
|
|
|
|
|
request_resp = await self.bot.aio_session.get(f'{self.bot.api_base}/messages/{ctx.guild.id}/requests/{id}/',
|
|
headers=self.bot.auth_header)
|
|
if request_resp.status == 200:
|
|
request = await request_resp.json()
|
|
requestor = ctx.guild.get_member(int(request['author']))
|
|
if requestor == ctx.author or admin:
|
|
pag = Paginator(self.bot, prefix='```md', suffix='```')
|
|
header = (f'Request {id} by {requestor.mention if requestor else "`User cannot be found`"}:\n'
|
|
f'```{request["content"]}```')
|
|
pag.set_header(header)
|
|
|
|
if request.get('comments'):
|
|
pag.add('Comments: \n')
|
|
comments = request['comments']
|
|
for comment in comments:
|
|
author = ctx.guild.get_member(int(comment['author']))
|
|
pag.add(f'{author.display_name}: {comment["content"]}', keep_intact=True)
|
|
else:
|
|
pag.add('No Comments')
|
|
|
|
if request.get('completed'):
|
|
closer = ctx.guild.get_member(int(request.get('completed_by')))
|
|
pag.add(f'Closed By: {closer.username}#{closer.discriminator}\n{request.get("completed_message")}')
|
|
book = Book(pag, (None, ctx.channel, self.bot, ctx.message))
|
|
await book.create_book()
|
|
else:
|
|
await ctx.send('That is not your request to close.')
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Tickets(bot))
|