Skip to content

Commit

Permalink
forgot the license code 💀
Browse files Browse the repository at this point in the history
  • Loading branch information
@NullyIsHere
NullyIsHere authored Jan 1, 2025
1 parent c10aefd commit b70ba2c
Showing 1 changed file with 42 additions and 18 deletions.
60 changes: 42 additions & 18 deletions cogs/badge.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,24 +17,25 @@
"""

import nextcord
from nextcord.ext import commands
from nextcord.ext import commands, tasks
from typing import Optional
from utils import log, langmgr, restrictions_legacy as r_legacy, slash as slash_handler
from enum import Enum
import aiohttp
import asyncio

restrictions_legacy = r_legacy.Restrictions()
language = langmgr.partial()
language.load()
slash = slash_handler.SlashHelper(language)

class UserRole(Enum):
# let values be None until set by langmgr
OWNER = language.get('owner','badge.roles')
ADMIN = language.get('admin','badge.roles')
MODERATOR = language.get('moderator','badge.roles')
TRUSTED = language.get('trusted','badge.roles')
BANNED = language.get('banned','badge.roles')
USER = language.get('user','badge.roles')
OWNER = language.get('owner', 'badge.roles')
ADMIN = language.get('admin', 'badge.roles')
MODERATOR = language.get('moderator', 'badge.roles')
TRUSTED = language.get('trusted', 'badge.roles')
BANNED = language.get('banned', 'badge.roles')
USER = language.get('user', 'badge.roles')

class Badge(commands.Cog, name=':medal: Badge'):
"""Badge contains commands that show you your role in Unifier."""
Expand All @@ -46,7 +47,7 @@ def __init__(self, bot):
language = self.bot.langmgr
self.embed_colors = {
UserRole.OWNER: (
self.bot.colors.greens_hair if self.bot.user.id==1187093090415149056 else self.bot.colors.unifier
self.bot.colors.greens_hair if self.bot.user.id == 1187093090415149056 else self.bot.colors.unifier
),
UserRole.ADMIN: nextcord.Color.green(),
UserRole.MODERATOR: nextcord.Color.purple(),
Expand All @@ -55,6 +56,18 @@ def __init__(self, bot):
UserRole.USER: nextcord.Color.blurple()
}
restrictions_legacy.attach_bot(self.bot)
self.unifier_team_data = {}
self.check_unifier_team.start()

@tasks.loop(minutes=30)
async def check_unifier_team(self):
url = "https://colab.unifierhq.org/data.json"
async with aiohttp.ClientSession() as session:
async with session.get(url) as response:
if response.status == 200:
self.unifier_team_data = await response.json()
else:
self.logger.warning(f"Failed to fetch Unifier team data: {response.status}")

@nextcord.slash_command(
description=language.desc('badge.badge'),
Expand All @@ -72,48 +85,57 @@ async def badge(
user_role = self.get_user_role(user.id)
embed = nextcord.Embed(
description=selector.fget("body", values={
'mention':f"<@{user.id}>",'role': user_role.value
'mention': f"<@{user.id}>", 'role': user_role.value
}),
color=self.embed_colors[user_role]
)
embed.set_author(
name=f'@{user.name}',
icon_url=user.avatar.url if user.avatar else None
)
if user_role==UserRole.BANNED:
if user_role == UserRole.BANNED:
embed.set_footer(text=selector.get("easter_egg"))


if str(user.id) in self.unifier_team_data:
team_member = self.unifier_team_data[str(user.id)]
embed.add_field(
name="Unifier Team",
value=f"{team_member['icon']} - **{team_member['displayname']}** is a **{team_member['role']}** in the Unifier team.",
inline=False
)

await ctx.send(embed=embed)

@commands.command(hidden=True,aliases=['trust'],description=language.desc('badge.verify'))
@commands.command(hidden=True, aliases=['trust'], description=language.desc('badge.verify'))
@restrictions_legacy.admin()
async def verify(self, ctx, user: nextcord.User):
selector = language.get_selector(ctx)

if user.id in self.bot.trusted_group:
return await ctx.send(f'{self.bot.ui_emojis.error} '+selector.fget("failed", values={'user': user.name}))
return await ctx.send(f'{self.bot.ui_emojis.error} ' + selector.fget("failed", values={'user': user.name}))

self.bot.trusted_group.append(user.id)

self.bot.db['trusted'] = self.bot.trusted_group
await self.bot.loop.run_in_executor(None, lambda: self.bot.db.save_data())

await ctx.send(f'{self.bot.ui_emojis.success} '+selector.fget("success", values={'user': user.name}))
await ctx.send(f'{self.bot.ui_emojis.success} ' + selector.fget("success", values={'user': user.name}))

@commands.command(hidden=True, aliases=['untrust'], description=language.desc('badge.unverify'))
@restrictions_legacy.admin()
async def unverify(self, ctx, user: nextcord.User):
selector = language.get_selector(ctx)

if not user.id in self.bot.trusted_group:
return await ctx.send(f'{self.bot.ui_emojis.error} '+selector.fget("failed", values={'user': user.name}))
return await ctx.send(f'{self.bot.ui_emojis.error} ' + selector.fget("failed", values={'user': user.name}))

self.bot.trusted_group.remove(user.id)

self.bot.db['trusted'] = self.bot.trusted_group
await self.bot.loop.run_in_executor(None, lambda: self.bot.db.save_data())

await ctx.send(f'{self.bot.ui_emojis.success} '+selector.fget("success", values={'user': user.name}))
await ctx.send(f'{self.bot.ui_emojis.success} ' + selector.fget("success", values={'user': user.name}))

def get_user_role(self, user_id):
if user_id == self.bot.owner or user_id in self.bot.other_owners:
Expand All @@ -132,6 +154,9 @@ def get_user_role(self, user_id):
async def cog_command_error(self, ctx, error):
await self.bot.exhandler.handle(ctx, error)

def setup(bot):
bot.add_cog(Badge(bot))

# Happy new 2025, leaving this as an easteregg, with love, ItsAsheer, green., summer., Lezetho, Arhan, Saphire, and arandomguy

# note from green to itsasheer:
Expand All @@ -141,5 +166,4 @@ async def cog_command_error(self, ctx, error):
# (jokes aside happy new year)

# ItsAsheer --> green. : Its bc new update to badge.py soon :eyes:, and bc its not a critical feature (imagine i broke bridge.py. We could create an easter egg file where we leave comments :)
def setup(bot):
bot.add_cog(Badge(bot))

0 comments on commit b70ba2c

Please sign in to comment.