Unstable Kitsune
07b436ed8c
- Updated `.gitignore` to include `cython_debug/`. - Introduced `DataManager` class for data retention policies. - Added logging functionality for server events in `logging.py`. - Refactored `hiring.py` and improved order submission in `kofishop.py`. - Added new cogs: `MORS`, `ServiceReview`, and `StaffMsg`. - Updated `info.json` for new cogs and modified `__init__.py` files for setup.
176 lines
8.0 KiB
Python
176 lines
8.0 KiB
Python
import discord
|
|
from redbot.core import commands, Config
|
|
from redbot.core.bot import Red
|
|
from typing import Optional
|
|
|
|
# --- Modals for the Commands ---
|
|
|
|
class OrderModal(discord.ui.Modal, title="Commission/Shop Order"):
|
|
def __init__(self, cog: "KofiShop"):
|
|
super().__init__()
|
|
self.cog = cog
|
|
|
|
comm_type = discord.ui.TextInput(label="What type of commission/item is this?")
|
|
payment_status = discord.ui.TextInput(label="Is this free or paid?")
|
|
description = discord.ui.TextInput(label="Please describe your request.", style=discord.TextStyle.paragraph)
|
|
questions = discord.ui.TextInput(label="Any questions for the artist?", style=discord.TextStyle.paragraph, required=False)
|
|
|
|
async def on_submit(self, interaction: discord.Interaction):
|
|
if not interaction.guild:
|
|
return await interaction.response.send_message("This must be used in a server.", ephemeral=True)
|
|
|
|
order_channel_id = await self.cog.config.guild(interaction.guild).order_channel()
|
|
if not order_channel_id:
|
|
return await interaction.response.send_message("The order channel has not been set by an admin.", ephemeral=True)
|
|
|
|
order_channel = interaction.guild.get_channel(order_channel_id)
|
|
if not isinstance(order_channel, discord.TextChannel):
|
|
return await interaction.response.send_message("The configured order channel is invalid.", ephemeral=True)
|
|
|
|
embed = discord.Embed(
|
|
title="New Order Placed",
|
|
description=f"Submitted by {interaction.user.mention}",
|
|
color=0x00ff00 # Green for new orders
|
|
)
|
|
embed.add_field(name="Item/Commission Type", value=self.comm_type.value, inline=False)
|
|
embed.add_field(name="Payment Status", value=self.payment_status.value, inline=False)
|
|
embed.add_field(name="Description", value=self.description.value, inline=False)
|
|
if self.questions.value:
|
|
embed.add_field(name="Questions", value=self.questions.value, inline=False)
|
|
|
|
try:
|
|
await order_channel.send(embed=embed)
|
|
await interaction.response.send_message("Your order has been successfully submitted!", ephemeral=True)
|
|
except discord.Forbidden:
|
|
await interaction.response.send_message("I don't have permission to send messages in the order channel.", ephemeral=True)
|
|
|
|
class ReviewModal(discord.ui.Modal, title="Leave a Review"):
|
|
def __init__(self, cog: "KofiShop"):
|
|
super().__init__()
|
|
self.cog = cog
|
|
|
|
item_name = discord.ui.TextInput(label="What item/commission are you reviewing?")
|
|
rating = discord.ui.TextInput(label="Rating (e.g., 10/10)")
|
|
review_text = discord.ui.TextInput(label="Your Review", style=discord.TextStyle.paragraph)
|
|
|
|
async def on_submit(self, interaction: discord.Interaction):
|
|
if not interaction.guild:
|
|
return await interaction.response.send_message("This must be used in a server.", ephemeral=True)
|
|
|
|
review_channel_id = await self.cog.config.guild(interaction.guild).review_channel()
|
|
if not review_channel_id:
|
|
return await interaction.response.send_message("The review channel has not been set by an admin.", ephemeral=True)
|
|
|
|
review_channel = interaction.guild.get_channel(review_channel_id)
|
|
if not isinstance(review_channel, discord.TextChannel):
|
|
return await interaction.response.send_message("The configured review channel is invalid.", ephemeral=True)
|
|
|
|
embed = discord.Embed(
|
|
title=f"New Review for: {self.item_name.value}",
|
|
description=f"Submitted by {interaction.user.mention}",
|
|
color=0xadd8e6 # Pastel Blue
|
|
)
|
|
embed.add_field(name="Rating", value=self.rating.value, inline=False)
|
|
embed.add_field(name="Review", value=self.review_text.value, inline=False)
|
|
|
|
try:
|
|
await review_channel.send(embed=embed)
|
|
await interaction.response.send_message("Thank you! Your review has been submitted.", ephemeral=True)
|
|
except discord.Forbidden:
|
|
await interaction.response.send_message("I don't have permission to send messages in the review channel.", ephemeral=True)
|
|
|
|
|
|
class KofiShop(commands.Cog):
|
|
"""
|
|
A cog to manage Ko-fi shop orders and reviews.
|
|
"""
|
|
|
|
def __init__(self, bot: Red):
|
|
self.bot = bot
|
|
self.config = Config.get_conf(self, identifier=5566778899, force_registration=True)
|
|
|
|
default_guild = {
|
|
"order_channel": None,
|
|
"review_channel": None,
|
|
"waitlist_channel": None
|
|
}
|
|
self.config.register_guild(**default_guild)
|
|
|
|
# --- Commands ---
|
|
@commands.hybrid_command()
|
|
@commands.guild_only()
|
|
async def order(self, ctx: commands.Context):
|
|
"""Place an order for a shop or commission item."""
|
|
if not ctx.interaction:
|
|
return
|
|
# We pass `self` (the cog instance) to the modal
|
|
await ctx.interaction.response.send_modal(OrderModal(self))
|
|
|
|
@commands.hybrid_command()
|
|
@commands.guild_only()
|
|
async def review(self, ctx: commands.Context):
|
|
"""Leave a review for a completed shop or commission item."""
|
|
if not ctx.interaction:
|
|
return
|
|
await ctx.interaction.response.send_modal(ReviewModal(self))
|
|
|
|
@commands.hybrid_command() # type: ignore
|
|
@commands.guild_only()
|
|
@commands.admin_or_permissions(manage_guild=True)
|
|
async def waitlist(self, ctx: commands.Context, user: discord.Member, *, item: str):
|
|
"""Add a user and their requested item to the waitlist."""
|
|
if not ctx.guild or not isinstance(ctx.channel, discord.TextChannel):
|
|
return await ctx.send("This command must be used in a server's text channel.", ephemeral=True)
|
|
|
|
waitlist_channel_id = await self.config.guild(ctx.guild).waitlist_channel()
|
|
if not waitlist_channel_id:
|
|
return await ctx.send("The waitlist channel has not been set by an admin.", ephemeral=True)
|
|
|
|
waitlist_channel = ctx.guild.get_channel(waitlist_channel_id)
|
|
if not isinstance(waitlist_channel, discord.TextChannel):
|
|
return await ctx.send("The configured waitlist channel is invalid.", ephemeral=True)
|
|
|
|
message = f"**{item}** ིྀ {user.mention} ✧ in {ctx.channel.mention}"
|
|
|
|
try:
|
|
await waitlist_channel.send(message)
|
|
await ctx.send(f"{user.mention} has been added to the waitlist for '{item}'.", ephemeral=True)
|
|
except discord.Forbidden:
|
|
await ctx.send(f"I don't have permission to send messages in the waitlist channel.", ephemeral=True)
|
|
|
|
# --- Settings Commands ---
|
|
@commands.group(aliases=["kset"]) # type: ignore
|
|
@commands.guild_only()
|
|
@commands.admin_or_permissions(manage_guild=True)
|
|
async def kofiset(self, ctx: commands.Context):
|
|
"""
|
|
Configure the KofiShop settings.
|
|
"""
|
|
pass
|
|
|
|
@kofiset.command(name="orderchannel")
|
|
async def kofiset_order(self, ctx: commands.Context, channel: discord.TextChannel):
|
|
"""Set the channel where new orders will be sent."""
|
|
if not ctx.guild: return
|
|
await self.config.guild(ctx.guild).order_channel.set(channel.id)
|
|
await ctx.send(f"Order channel has been set to {channel.mention}.")
|
|
|
|
@kofiset.command(name="reviewchannel")
|
|
async def kofiset_review(self, ctx: commands.Context, channel: discord.TextChannel):
|
|
"""Set the channel where new reviews will be sent."""
|
|
if not ctx.guild: return
|
|
await self.config.guild(ctx.guild).review_channel.set(channel.id)
|
|
await ctx.send(f"Review channel has been set to {channel.mention}.")
|
|
|
|
@kofiset.command(name="waitlistchannel")
|
|
async def kofiset_waitlist(self, ctx: commands.Context, channel: discord.TextChannel):
|
|
"""Set the channel where waitlist notifications will be sent."""
|
|
if not ctx.guild: return
|
|
await self.config.guild(ctx.guild).waitlist_channel.set(channel.id)
|
|
await ctx.send(f"Waitlist channel has been set to {channel.mention}.")
|
|
|
|
|
|
async def setup(bot: Red):
|
|
await bot.add_cog(KofiShop(bot))
|
|
|