Bot pro KSP Discord
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

38 lines
1.7 KiB

import discord
from discord.ext import commands
from discord.utils import get
import hrochobot.utils.data as data
MESSAGES_JSON = "messages"
def as_reply(message: str):
return "\n".join(map(lambda x: f"> {x}", message.split('\n')))
class Messages(commands.Cog):
def __init__(self, bot):
self.bot = bot
@discord.message_command()
async def forward(self, ctx, message: discord.Message):
messages_json = data.load_guild_data(ctx.guild.id, MESSAGES_JSON)
if "forward_channel" not in messages_json:
return await ctx.respond(f"Forwarding channel not set.", ephemeral=True)
forward_channel = get(ctx.guild.channels, id=messages_json["forward_channel"])
header_text = f"{ctx.author.mention} forwarded:\n{message.author.mention} wrote in {message.channel.mention}:\n"
await forward_channel.send(f"{header_text}{as_reply(message.clean_content)}")
return await ctx.respond("Message successfully forwarded.", ephemeral=True)
@discord.slash_command(description="Sets channel for forwarding.")
@discord.option("channel", discord.TextChannel, description="Channel for forwarding.")
@discord.default_permissions(administrator=True)
async def set_forward_channel(self, ctx, channel: discord.TextChannel):
messages_json = data.load_guild_data(ctx.guild.id, MESSAGES_JSON)
messages_json["forward_channel"] = channel.id
data.dump_guild_data(ctx.guild.id, MESSAGES_JSON, messages_json)
return await ctx.respond(f"Forwarding channel set to {channel.mention}.\n" + \
"Remember to set permissions for forwarding.", ephemeral=True)
def setup(bot):
bot.add_cog(Messages(bot))