Skip to content

Commit

Permalink
updating the loa endpoints please help 🙏
Browse files Browse the repository at this point in the history
  • Loading branch information
@MikeyUsersREC
MikeyUsersREC committed Nov 30, 2024
1 parent 4376c75 commit f4f1751
Show file tree
Hide file tree
Showing 4 changed files with 212 additions and 17 deletions.
39 changes: 37 additions & 2 deletions events/on_loa_accept.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ def __init__(self, bot: commands.Bot):
self.bot = bot

@commands.Cog.listener()
async def on_loa_accept(self, s_loa: dict, role_ids: list[int]):
async def on_loa_accept(self, s_loa: dict, role_ids: list[int], accepted_by: int):
guild = self.bot.get_guild(s_loa["guild_id"])
try:
user = await guild.fetch_member(int(s_loa["user_id"]))
Expand All @@ -27,14 +27,49 @@ async def on_loa_accept(self, s_loa: dict, role_ids: list[int]):
pass


loa_roles = list(filter(lambda x: x is not None, [discord.utils.get(identifier) for identifier in role_ids]))
settings = await self.bot.settings.find_by_id(guild.id)
loa_roles = list(filter(lambda x: x is not None, [discord.utils.get(guild.roles, id=identifier) for identifier in role_ids]))
for rl in loa_roles:
if rl not in user.roles:
try:
await user.add_roles(rl)
except discord.HTTPException:
pass

msg = s_loa["message_id"]
loa_channel_id = settings.get("staff_management", {}).get("channel", 0)
if not loa_channel_id:
return
loa_channel = guild.get_channel(loa_channel_id) or await guild.fetch_channel(loa_channel_id)
messg = None
try:
messg = await loa_channel.fetch_message(msg)
except:
pass
if not messg:
return

embed = messg.embeds[0]
embed.title = (
f"<:success:1163149118366040106> {s_loa['type']} Accepted"
)
embed.colour = GREEN_COLOR
try:
accepted_by_user = guild.get_member(accepted_by) or await guild.fetch_member(accepted_by)
except:
pass

embed.set_footer(text=f"Accepted by {accepted_by_user.name if accepted_by_user else 'n/a'}")

await messg.edit(
embed=embed,
view=None
)

view_item = await self.bot.views.db.find_one({"message_id": messg.id})

await self.bot.views.delete_by_id(view_item["_id"])


async def setup(bot):
await bot.add_cog(OnLOAAccept(bot))
35 changes: 34 additions & 1 deletion events/on_loa_deny.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ def __init__(self, bot: commands.Bot):
self.bot = bot

@commands.Cog.listener()
async def on_loa_deny(self, s_loa: dict):
async def on_loa_deny(self, s_loa: dict, denied_by: int):
guild = self.bot.get_guild(s_loa["guild_id"])
try:
user = await guild.fetch_member(int(s_loa["user_id"]))
Expand All @@ -27,6 +27,39 @@ async def on_loa_deny(self, s_loa: dict):
except:
pass

settings = await self.bot.settings.find_by_id(guild.id)
msg = s_loa["message_id"]
loa_channel_id = settings.get("staff_management", {}).get("channel", 0)
if not loa_channel_id:
return
loa_channel = guild.get_channel(loa_channel_id) or await guild.fetch_channel(loa_channel_id)
messg = None
try:
messg = await loa_channel.fetch_message(msg)
except:
pass
if not messg:
return



embed = messg.embeds[0]
embed.title = f"{s_loa['type']} Denied"
embed.colour = BLANK_COLOR
try:
accepted_by_user = guild.get_member(denied_by) or await guild.fetch_member(denied_by)
except:
pass

embed.set_footer(text=f"Denied by {accepted_by_user.name if accepted_by_user else 'n/a'}")

await messg.edit(embed=embed, view=None)

view_item = await self.bot.views.db.find_one({"message_id": messg.id})
await self.bot.views.delete_by_id(view_item.id)




async def setup(bot):
await bot.add_cog(OnLOADeny(bot))
7 changes: 1 addition & 6 deletions menus.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -911,20 +911,15 @@ async def accept(self, interaction: discord.Interaction, button: discord.ui.Butt
):
await generalised_interaction_check_failure(interaction.followup)
return
for item in self.children:
item.disabled = True
await interaction.message.edit(view=self)

for item in self.children:
item.disabled = True
if item.label == "Accept":
item.label = "Accepted"
else:
self.remove_item(item)
await interaction.message.edit(view=self)
s_loa = None
# # # # print(t(t(t(t(self)
# # # # print(t(t(t(t(self.bot)

for loa in await self.bot.loas.get_all():
if (
loa["message_id"] == interaction.message.id
Expand Down
148 changes: 140 additions & 8 deletions utils/api.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,13 +13,13 @@
from erm import Bot, management_predicate, is_staff, staff_predicate, staff_check, management_check, admin_check
from typing import Annotated
from decouple import config
from menus import LOAMenu
from utils.constants import BLANK_COLOR, GREEN_COLOR
from utils.utils import secure_logging
from utils.utils import get_elapsed_time, secure_logging
from pydantic import BaseModel

from utils.timestamp import td_format
# from helpers import MockContext
from utils.utils import tokenGenerator
from utils.utils import tokenGenerator, system_code_gen
import logging

logger = logging.getLogger(__name__)
Expand Down Expand Up @@ -292,6 +292,127 @@ async def POST_deny_application(
except Exception as e:
raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")

async def POST_send_loa(
self,
authorization: Annotated[str | None, Header()],
request: Request
):
if not authorization:
raise HTTPException(status_code=401, detail="Invalid authorization")

if not await validate_authorization(self.bot, authorization):
raise HTTPException(status_code=401, detail="Invalid or expired authorization.")

json_data = await request.json()
s_loa = json_data.get("loa") # OID -> dict { loa spec }
s_loa_item = await self.bot.loas.find_by_id(s_loa)
schema = s_loa_item
guild = self.bot.get_guild(schema["guild_id"]) or await self.bot.fetch_guild(schema["guild_id"])
try:
author = guild.get_member(schema["user_id"]) or await guild.get_member(schema["user_id"])
except:
raise HTTPException(status_code=400, detail="Invalid author")

request_type = schema['type']
settings = await self.bot.settings.find_by_id(guild.id)
management_roles = settings.get('staff_management', {}).get('management_role', [])
loa_roles = settings.get('staff_management').get(f'{request_type.lower()}_role')

embed = discord.Embed(
title=f"{request_type} Request",
color=BLANK_COLOR
)
embed.set_author(
name=guild.name,
icon_url=guild.icon.url if guild.icon else ''
)

past_author_notices = [item async for item in self.bot.loas.db.find({
"guild_id": guild.id,
"user_id": author.id,
"accepted": True,
"denied": False,
"expired": True,
"type": request_type.upper()
})]

shifts = []
storage_item = [
i
async for i in self.bot.shift_management.shifts.db.find(
{"UserID": author.id, "Guild": guild.id}
)
]

for s in storage_item:
if s["EndEpoch"] != 0:
shifts.append(s)

total_seconds = sum([get_elapsed_time(i) for i in shifts])

embed.add_field(
name="Staff Information",
value=(
f"> **Staff Member:** {author.mention}\n"
f"> **Top Role:** {author.top_role.name}\n"
f"> **Past {request_type}s:** {len(past_author_notices)}\n"
f"> **Shift Time:** {td_format(datetime.timedelta(seconds=total_seconds))}"
),
inline=False
)

embed.add_field(
name="Request Information",
value=(
f"> **Type:** {request_type}\n"
f"> **Reason:** {schema['reason']}\n"
f"> **Starts At:** <t:{schema.get('started_at', int(schema['_id'].split('_')[2]))}>\n"
f"> **Ends At:** <t:{schema['expiry']}>"
)
)

view = LOAMenu(
self.bot,
management_roles,
loa_roles,
schema,
author.id,
(code := system_code_gen())
)

staff_channel = settings.get("staff_management").get("channel")
staff_channel = discord.utils.get(guild.channels, id=staff_channel)

msg = await staff_channel.send(
embed=embed,
view=view
)
schema['message_id'] = msg.id
await self.bot.views.insert(
{
"_id": code,
"args": [
'SELF',
management_roles,
loa_roles,
schema,
author.id,
code
],
"view_type": 'LOAMenu',
'message_id': msg.id
}
)

ns = schema
del ns["_id"]
await self.bot.update_one({
"_id": schema["_id"]
}, ns)

return 200


async def POST_accept_loa(
self,
authorization: Annotated[str | None, Header()],
Expand All @@ -304,10 +425,17 @@ async def POST_accept_loa(
raise HTTPException(status_code=401, detail="Invalid or expired authorization.")

json_data = await request.json()
s_loa = json_data.get("loa") # dict { loa spec }
roles = json_data.get("roles") # list[int]
s_loa = json_data.get("loa") # oid -> dict { loa spec }
accepted_by = json_data.get("accepted_by")
s_loa_item = await self.bot.loas.find_by_id(s_loa)
s_loa = s_loa_item

self.bot.dispatch("loa_accept", s_loa=s_loa, role_ids=roles)
# fetch the actual accept roles
guild_id = s_loa["guild_id"]
config = await self.bot.settings.find_by_id(guild_id) or {}
roles = config.get("staff_management", {}).get(f"{s_loa['type']}_role", []) or []

self.bot.dispatch("loa_accept", s_loa=s_loa, role_ids=roles, accepted_by=accepted_by)

return 200

Expand All @@ -323,8 +451,12 @@ async def POST_deny_loa(
raise HTTPException(status_code=401, detail="Invalid or expired authorization.")

json_data = await request.json()
s_loa = json_data.get("loa") # dict { loa spec }
self.bot.dispatch("loa_deny", s_loa=s_loa)
s_loa = json_data.get("loa")
denied_by = json_data.get("denied_by")
s_loa_item = await self.bot.loas.find_by_id(s_loa)
s_loa = s_loa_item

self.bot.dispatch("loa_deny", s_loa=s_loa, denied_by=denied_by)

return 200

Expand Down

0 comments on commit f4f1751

Please sign in to comment.