woodburn-bot/support.py

279 lines
9.6 KiB
Python
Raw Normal View History

2024-02-07 23:24:31 +11:00
import datetime
import re
import discord
import json
import os
import dotenv
dotenv.load_dotenv()
TICKETS_FILE_PATH = '/mnt/tickets.json'
DISCORD_TOKEN = os.getenv('DISCORD_TOKEN')
intents = discord.Intents.default()
client = discord.Client(intents=intents)
def set_client(c):
global client
client = c
print("Client set")
if not os.path.exists(TICKETS_FILE_PATH):
with open(TICKETS_FILE_PATH, 'w') as f:
json.dump({"server": {}}, f)
def is_server_valid(server:str):
server = server
with open(TICKETS_FILE_PATH, 'r') as f:
ticketsData = json.load(f)
if server in ticketsData['server']:
return True
return False
async def create_ticket(user_id, server:str):
with open(TICKETS_FILE_PATH, 'r') as f:
ticketsData = json.load(f)
if server not in ticketsData['server']:
return "Server not found"
tickets = ticketsData['server'][server]['tickets']
if user_id != "admin":
for ticket in tickets:
if ticket['user_id'] == user_id and ticket['status'] == "open":
return "You already have an open ticket"
ticketNum = len(tickets) + 1
# Generate a ticket name
user = await client.fetch_user(int(user_id))
if user == None:
return "User not found"
# Create a new ticket channel
guild = await client.fetch_guild(int(server))
guild = client.get_guild(int(server))
ticketName = f"{ticketNum}-{user.name}-ticket"
category = discord.utils.get(guild.categories, id=int(ticketsData['server'][server]['category']))
if category == None:
return "Category not found"
# Create the ticket channel
ticketChannel = await guild.create_text_channel(ticketName, category=category)
# Remove the default permissions
await ticketChannel.set_permissions(guild.default_role, read_messages=False, send_messages=False)
# Add the user to the ticket channel
await ticketChannel.set_permissions(user, read_messages=True, send_messages=True)
# Add the modRole to the ticket channel
admin = guild.get_role(int(ticketsData['server'][server]['adminRole']))
await ticketChannel.set_permissions(admin, read_messages=True, send_messages=True)
# Add the ticket to the tickets list
tickets.append({
'user_id': user_id,
'channel_id': ticketChannel.id,
'status': "open",
'members': [user_id]
})
ticketsData['server'][server]['tickets'] = tickets
with open(TICKETS_FILE_PATH, 'w') as f:
json.dump(ticketsData, f)
await ticketChannel.send(f"G'day <@{user_id}>, your ticket has been created. Please let us know how we can help you.")
await ticketChannel.send(f"Commands: \n</adduser:1204746777639653386> <user> - Add a user to the ticket\n</removeuser:1204747212878520402> <user> - Remove a user from the ticket\n</renameticket:1204758890332291123> <new_name> - Rename the ticket\n</closeticket:1204742678005424131> - Close the ticket")
return f"Ticket <#{ticketChannel.id}> created"
async def ticketAddServer(server, category, adminRole,closedCategory):
with open(TICKETS_FILE_PATH, 'r') as f:
ticketsData = json.load(f)
ticketsData['server'][server] = {
'category': category,
'closedCategory': closedCategory,
'adminRole': adminRole,
'tickets': []
}
with open(TICKETS_FILE_PATH, 'w') as f:
json.dump(ticketsData, f)
return "Server added"
async def close_ticket(user_id,channel_id, server):
with open(TICKETS_FILE_PATH, 'r') as f:
ticketsData = json.load(f)
tickets = ticketsData['server'][server]['tickets']
validTicket = False
for ticket in tickets:
if str(ticket['channel_id']) == channel_id:
ticket['status'] = "closed"
validTicket = True
break
if not validTicket:
return "This ticket does not exist"
ticketsData['server'][server]['tickets'] = tickets
with open(TICKETS_FILE_PATH, 'w') as f:
json.dump(ticketsData, f)
# Remove read and send permissions for everyone
guild = await client.fetch_guild(int(server))
guild = client.get_guild(int(server))
ticketChannel = guild.get_channel(int(channel_id))
await ticketChannel.set_permissions(guild.default_role, read_messages=False, send_messages=False)
for member in ticket['members']:
user = await client.fetch_user(int(member))
await ticketChannel.set_permissions(user, read_messages=False, send_messages=False)
await ticketChannel.send(f"This ticket has been closed by <@{user_id}>")
# Move to the closed category
closedCategory = discord.utils.get(guild.categories, id=int(ticketsData['server'][server]['closedCategory']))
if closedCategory == None:
return "Category not found"
await ticketChannel.edit(category=closedCategory)
async def addMemberToTicket(user_id, channel_id, server):
guild = await client.fetch_guild(int(server))
guild = client.get_guild(int(server))
user = await fuzzyUser(user_id, guild)
if user == False:
return "User not found"
user_id = str(user.id)
with open(TICKETS_FILE_PATH, 'r') as f:
ticketsData = json.load(f)
tickets = ticketsData['server'][server]['tickets']
validTicket = False
for ticket in tickets:
if str(ticket['channel_id']) == channel_id:
if user_id in ticket['members']:
return "User already in ticket"
ticket['members'].append(user_id)
validTicket = True
break
if not validTicket:
return "This ticket does not exist"
ticketsData['server'][server]['tickets'] = tickets
with open(TICKETS_FILE_PATH, 'w') as f:
json.dump(ticketsData, f)
ticketChannel = guild.get_channel(int(channel_id))
await ticketChannel.set_permissions(user, read_messages=True, send_messages=True)
await ticketChannel.send(f"{user.mention} has been added to the ticket")
return "User added to ticket"
async def removeMemberFromTicket(user_id, channel_id, server):
guild = await client.fetch_guild(int(server))
guild = client.get_guild(int(server))
user = await fuzzyUser(user_id, guild)
if user == False:
return "User not found"
user_id = str(user.id)
with open(TICKETS_FILE_PATH, 'r') as f:
ticketsData = json.load(f)
tickets = ticketsData['server'][server]['tickets']
validTicket = False
for ticket in tickets:
if str(ticket['channel_id']) == channel_id:
if user_id not in ticket['members']:
return "User not found in ticket"
ticket['members'].remove(user_id)
validTicket = True
break
if not validTicket:
return "This ticket does not exist"
ticketsData['server'][server]['tickets'] = tickets
with open(TICKETS_FILE_PATH, 'w') as f:
json.dump(ticketsData, f)
ticketChannel = guild.get_channel(int(channel_id))
await ticketChannel.set_permissions(user, read_messages=False, send_messages=False)
await ticketChannel.send(f"{user.mention} has been removed from the ticket")
return "User removed from ticket"
async def reopen_ticket(user_id,channel_id,guild_id):
with open(TICKETS_FILE_PATH, 'r') as f:
ticketsData = json.load(f)
tickets = ticketsData['server'][guild_id]['tickets']
validTicket = False
for ticket in tickets:
if str(ticket['channel_id']) == channel_id:
if ticket['status'] == "open":
return "Ticket is already open"
ticket['status'] = "open"
validTicket = True
break
if not validTicket:
return "This ticket does not exist"
ticketsData['server'][guild_id]['tickets'] = tickets
with open(TICKETS_FILE_PATH, 'w') as f:
json.dump(ticketsData, f)
guild = await client.fetch_guild(int(guild_id))
guild = client.get_guild(int(guild_id))
channel = guild.get_channel(int(channel_id))
# Remove read and send permissions for everyone
await channel.set_permissions(guild.default_role, read_messages=False, send_messages=False)
for member in ticket['members']:
user = await client.fetch_user(int(member))
await channel.set_permissions(user, read_messages=True, send_messages=True)
await channel.send(f"This ticket has been reopened by <@{user_id}>")
# Move to the closed category
category = discord.utils.get(guild.categories, id=int(ticketsData['server'][guild_id]['category']))
if category == None:
return "Category not found"
await channel.edit(category=category)
return "Ticket reopened"
async def rename_ticket(user_id,channel_id,server_id,newName):
try:
guild = await client.fetch_guild(int(server_id))
guild = client.get_guild(int(server_id))
channel = guild.get_channel(int(channel_id))
await channel.edit(name=newName)
return "Ticket renamed"
except:
return "Error renaming ticket"
async def fuzzyUser(search, guild):
user = None
try:
user = await client.fetch_user(int(search))
except:
pass
if user == None:
user = discord.utils.get(guild.members, name=search)
if user == None:
user = discord.utils.get(guild.members, nick=search)
if user == None:
user = discord.utils.get(guild.members, global_name=search)
if user == None:
user = discord.utils.get(guild.members, global_name=search)
if user == None:
user = guild.get_member_named(search)
if user == None:
user = await guild.query_members(search)
if len(user) > 0:
user = user[0]
else:
return False
if user == None:
return False
return user