155 lines
6.1 KiB
Python
155 lines
6.1 KiB
Python
import discord
|
|
from utils.emoji import CROSS, TICK, ZWARNING
|
|
from discord.ext import commands
|
|
from discord.ui import LayoutView, TextDisplay, Separator, Container, ActionRow, Button
|
|
import aiosqlite
|
|
from utils.Tools import *
|
|
from utils.cv2 import CV2, build_container
|
|
|
|
|
|
|
|
|
|
|
|
class ConfirmView(LayoutView):
|
|
def __init__(self, ctx):
|
|
super().__init__(timeout=60)
|
|
self.ctx = ctx
|
|
self.value = None
|
|
|
|
self.yes_btn = Button(label="Confirm", style=discord.ButtonStyle.green)
|
|
self.no_btn = Button(label="Cancel", style=discord.ButtonStyle.red)
|
|
|
|
self.yes_btn.callback = self.confirm_callback
|
|
self.no_btn.callback = self.cancel_callback
|
|
|
|
container = build_container(
|
|
TextDisplay("**Confirm Action**"),
|
|
Separator(visible=True),
|
|
TextDisplay(self._desc),
|
|
ActionRow(self.yes_btn, self.no_btn),
|
|
)
|
|
self.add_item(container)
|
|
|
|
@property
|
|
def _desc(self):
|
|
return ""
|
|
|
|
async def confirm_callback(self, interaction: discord.Interaction):
|
|
if interaction.user != self.ctx.author:
|
|
return await interaction.response.send_message("You cannot interact with this confirmation.", ephemeral=True)
|
|
self.value = True
|
|
await interaction.response.defer()
|
|
self.stop()
|
|
|
|
async def cancel_callback(self, interaction: discord.Interaction):
|
|
if interaction.user != self.ctx.author:
|
|
return await interaction.response.send_message("You cannot interact with this confirmation.", ephemeral=True)
|
|
self.value = False
|
|
await interaction.response.defer()
|
|
self.stop()
|
|
|
|
|
|
class SetConfirmView(ConfirmView):
|
|
def __init__(self, ctx, user):
|
|
self._user = user
|
|
super().__init__(ctx)
|
|
|
|
@property
|
|
def _desc(self):
|
|
return f"**Are you sure you want to set {self._user.mention} as the Extra Owner?**"
|
|
|
|
|
|
class ResetConfirmView(ConfirmView):
|
|
@property
|
|
def _desc(self):
|
|
return "**Are you sure you want to reset the Extra Owner?**"
|
|
|
|
|
|
class Extraowner(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.bot.loop.create_task(self.initialize_db())
|
|
|
|
async def initialize_db(self):
|
|
self.db = await aiosqlite.connect('db/anti.db')
|
|
await self.db.execute('''
|
|
CREATE TABLE IF NOT EXISTS extraowners (
|
|
guild_id INTEGER PRIMARY KEY,
|
|
owner_id INTEGER
|
|
)
|
|
''')
|
|
await self.db.commit()
|
|
|
|
@commands.hybrid_command(name='extraowner', aliases=["owner"], help="Adds Extraowner to the server")
|
|
@blacklist_check()
|
|
@ignore_check()
|
|
@commands.cooldown(1, 10, commands.BucketType.user)
|
|
@commands.max_concurrency(1, per=commands.BucketType.default, wait=False)
|
|
@commands.guild_only()
|
|
async def extraowner(self, ctx, option: str = None, user: discord.Member = None):
|
|
guild_id = ctx.guild.id
|
|
|
|
if ctx.guild.member_count < 2:
|
|
return await ctx.send(view=CV2(f"{CROSS} Error", "Your Server Doesn't Meet My 30 Member Criteria"))
|
|
|
|
Ray = ['870179991462236170','767979794411028491']
|
|
if ctx.author.id != ctx.guild.owner_id and str(ctx.author.id) not in Ray:
|
|
return await ctx.send(view=CV2(f"{ZWARNING} Access Denied", "Only Server Owner Can Run This Command"))
|
|
|
|
if option is None:
|
|
pre = ctx.prefix
|
|
info_text = (
|
|
"Extraowners can adjust server antinuke settings & manage whitelist events, "
|
|
"so careful consideration is essential before assigning it to someone.\n\n"
|
|
f"**Extraowner Set** — `{pre}extraowner set @user`\n"
|
|
f"**Extraowner Reset** — `{pre}extraowner reset`\n"
|
|
f"**Extraowner View** — `{pre}extraowner view`"
|
|
)
|
|
return await ctx.reply(view=CV2("__Extra Owner__", info_text))
|
|
|
|
if option.lower() == 'set':
|
|
if user is None or user.bot:
|
|
return await ctx.reply(view=CV2(f"{CROSS} Error", "Please Provide a Valid User Mention or ID to Set as Extra Owner!"))
|
|
|
|
view = SetConfirmView(ctx, user)
|
|
message = await ctx.reply(view=view)
|
|
await view.wait()
|
|
|
|
if view.value is None:
|
|
await message.edit(view=CV2("⏰ Timed Out", "Confirmation timed out."))
|
|
elif view.value:
|
|
await self.db.execute('INSERT OR REPLACE INTO extraowners (guild_id, owner_id) VALUES (?, ?)', (guild_id, user.id))
|
|
await self.db.commit()
|
|
await message.edit(view=CV2(f"{TICK} Success", f"Added {user.mention} As Extraowner"))
|
|
else:
|
|
await message.edit(view=CV2(f"{CROSS} Cancelled", "Action cancelled."))
|
|
|
|
elif option.lower() == 'reset':
|
|
async with self.db.execute('SELECT owner_id FROM extraowners WHERE guild_id = ?', (guild_id,)) as cursor:
|
|
row = await cursor.fetchone()
|
|
|
|
if not row:
|
|
await ctx.reply(view=CV2(f"{CROSS} Error", "No extra owner has been designated for this guild."))
|
|
else:
|
|
view = ResetConfirmView(ctx)
|
|
message = await ctx.reply(view=view)
|
|
await view.wait()
|
|
|
|
if view.value is None:
|
|
await message.edit(view=CV2("⏰ Timed Out", "Confirmation timed out."))
|
|
elif view.value:
|
|
await self.db.execute('DELETE FROM extraowners WHERE guild_id = ?', (guild_id,))
|
|
await self.db.commit()
|
|
await message.edit(view=CV2(f"{TICK} Success", "Disabled Extraowner Configuration!"))
|
|
else:
|
|
await message.edit(view=CV2(f"{CROSS} Cancelled", "Action cancelled."))
|
|
|
|
elif option.lower() == 'view':
|
|
async with self.db.execute('SELECT owner_id FROM extraowners WHERE guild_id = ?', (guild_id,)) as cursor:
|
|
row = await cursor.fetchone()
|
|
|
|
if not row:
|
|
await ctx.reply(view=CV2(f"{CROSS} Error", "No extra owner is currently assigned."))
|
|
else:
|
|
await ctx.reply(view=CV2("Extra Owner", f"Current Extraowner is <@{row[0]}>"))
|