68 lines
3.2 KiB
Python
68 lines
3.2 KiB
Python
from typing import Union
|
|
import discord
|
|
from redbot.core import commands, app_commands
|
|
|
|
# pylint: disable=arguments-differ
|
|
|
|
class Send(commands.Cog):
|
|
"""Allows you to send messages as the bot account."""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
async def send_to_target(self, target: Union[discord.Member, discord.TextChannel], interaction: discord.Interaction, message: str, secondary_message: str = None):
|
|
if isinstance(target, discord.Member):
|
|
target_type = "member"
|
|
elif isinstance(target, discord.TextChannel):
|
|
target_type = "textchannel"
|
|
try:
|
|
await target.send(message)
|
|
if secondary_message is not None:
|
|
await target.send(secondary_message)
|
|
await interaction.response.send_message(content=f"Message sent to {target.mention}!\nMessage contents:\n```{message}```\n```{secondary_message}```", ephemeral=True)
|
|
else:
|
|
await interaction.response.send_message(content=f"Message sent to {target.mention}!\nMessage contents:\n```{message}```", ephemeral=True)
|
|
except (discord.HTTPException, discord.Forbidden):
|
|
if target_type == "member":
|
|
await interaction.response.send_message(content="That user has their direct messages closed!", ephemeral=True)
|
|
elif target_type == "textchannel":
|
|
await interaction.response.send_message(content="I cannot access that channel!", ephemeral=True)
|
|
|
|
class MessageModal(discord.ui.Modal, title="Sending message..."):
|
|
def __init__(self, target):
|
|
super().__init__()
|
|
self.target = target
|
|
message = discord.ui.TextInput(
|
|
label="Message Content",
|
|
placeholder="I'm contacting you about your cars extended warranty...",
|
|
style=discord.TextStyle.paragraph,
|
|
max_length=1750
|
|
)
|
|
secondary_message = discord.ui.TextInput(
|
|
label="Secondary Message Content",
|
|
placeholder="Typically used for images/image links.",
|
|
style=discord.TextStyle.short,
|
|
required=False,
|
|
max_length=200
|
|
)
|
|
|
|
async def on_submit(self, interaction: discord.Interaction):
|
|
await Send.send_to_target(self, self.target, interaction, self.message, self.secondary_message)
|
|
|
|
send = app_commands.Group(name="send", description="Send a message as the bot user!")
|
|
|
|
@send.command(name="user", description="Sends a direct message to a user.")
|
|
async def user(self, interaction: discord.Interaction, member: discord.Member, message: str = None):
|
|
"""Sends a direct message to a user."""
|
|
if message:
|
|
await Send.send_to_target(self, member, interaction, message)
|
|
else:
|
|
await interaction.response.send_modal(Send.MessageModal(member))
|
|
|
|
@send.command(name="channel", description="Sends a message to a channel.")
|
|
async def channel(self, interaction: discord.Interaction, channel: discord.TextChannel, message: str = None):
|
|
"""Sends a message to a channel."""
|
|
if message:
|
|
await Send.send_to_target(self, channel, interaction, message)
|
|
else:
|
|
await interaction.response.send_modal(Send.MessageModal(channel))
|