3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-03-16 06:11:49 +00:00
zeppelin/src/plugins/Post.ts

171 lines
5 KiB
TypeScript
Raw Normal View History

import { Plugin, decorators as d } from "knub";
2019-01-15 03:39:39 +02:00
import { Channel, EmbedBase, Message, TextChannel } from "eris";
2019-01-15 03:04:47 +02:00
import { errorMessage, downloadFile } from "../utils";
2018-11-24 14:19:47 +02:00
import { GuildSavedMessages } from "../data/GuildSavedMessages";
2019-01-15 03:04:47 +02:00
import fs from "fs";
const fsp = fs.promises;
2019-01-15 03:58:58 +02:00
const COLOR_MATCH_REGEX = /^#?([0-9a-f]{6})$/;
export class PostPlugin extends Plugin {
2019-01-15 03:04:47 +02:00
public static pluginName = "post";
2018-11-24 14:19:47 +02:00
protected savedMessages: GuildSavedMessages;
onLoad() {
this.savedMessages = GuildSavedMessages.getInstance(this.guildId);
}
getDefaultOptions() {
return {
permissions: {
post: false,
edit: false
},
overrides: [
{
level: ">=100",
permissions: {
post: true,
edit: true
}
}
]
};
}
/**
2019-01-15 03:04:47 +02:00
* COMMAND: Post a message as the bot to the specified channel
*/
2019-01-15 03:04:47 +02:00
@d.command("post", "<channel:channel> [content:string$]")
@d.permission("post")
2019-01-15 03:04:47 +02:00
async postCmd(msg: Message, args: { channel: Channel; content?: string }) {
if (!(args.channel instanceof TextChannel)) {
msg.channel.createMessage(errorMessage("Channel is not a text channel"));
return;
}
2019-01-15 03:04:47 +02:00
const content = args.content || undefined;
let downloadedAttachment;
let file;
if (msg.attachments.length) {
downloadedAttachment = await downloadFile(msg.attachments[0].url);
file = {
name: msg.attachments[0].filename,
file: await fsp.readFile(downloadedAttachment.path)
};
}
if (content == null && file == null) {
msg.channel.createMessage(errorMessage("Text content or attachment required"));
return;
}
const createdMsg = await args.channel.createMessage(content, file);
await this.savedMessages.setPermanent(createdMsg.id);
2019-01-15 03:04:47 +02:00
if (downloadedAttachment) {
downloadedAttachment.deleteFn();
}
}
2019-01-15 03:58:58 +02:00
/**
* COMMAND: Post a message with an embed as the bot to the specified channel
*/
2019-01-15 03:39:39 +02:00
@d.command("post_embed", "<channel:channel>", {
options: [{ name: "title", type: "string" }, { name: "content", type: "string" }, { name: "color", type: "string" }]
})
@d.permission("post")
async postEmbedCmd(msg: Message, args: { channel: Channel; title?: string; content?: string; color?: string }) {
if (!(args.channel instanceof TextChannel)) {
msg.channel.createMessage(errorMessage("Channel is not a text channel"));
return;
}
if (!args.title && !args.content) {
msg.channel.createMessage(errorMessage("Title or content required"));
return;
}
let color = null;
if (args.color) {
2019-01-15 03:58:58 +02:00
const colorMatch = args.color.match(COLOR_MATCH_REGEX);
2019-01-15 03:39:39 +02:00
if (!colorMatch) {
msg.channel.createMessage(errorMessage("Invalid color specified, use hex colors"));
return;
}
color = parseInt(colorMatch[1], 16);
}
const embed: EmbedBase = {};
if (args.title) embed.title = args.title;
if (args.content) embed.description = args.content;
if (color) embed.color = color;
const createdMsg = await args.channel.createMessage({ embed });
await this.savedMessages.setPermanent(createdMsg.id);
}
/**
2019-01-15 03:58:58 +02:00
* COMMAND: Edit the specified message posted by the bot
*/
@d.command("edit", "<messageId:string> <content:string$>")
@d.permission("edit")
async editCmd(msg, args: { messageId: string; content: string }) {
const savedMessage = await this.savedMessages.find(args.messageId);
if (!savedMessage) {
msg.channel.createMessage(errorMessage("Unknown message"));
return;
}
if (savedMessage.user_id !== this.bot.user.id) {
msg.channel.createMessage(errorMessage("Message wasn't posted by me"));
return;
}
await this.bot.editMessage(savedMessage.channel_id, savedMessage.id, args.content);
}
2019-01-15 03:58:58 +02:00
/**
* COMMAND: Edit the specified message with an embed posted by the bot
*/
@d.command("edit_embed", "<messageId:string>", {
options: [{ name: "title", type: "string" }, { name: "content", type: "string" }, { name: "color", type: "string" }]
})
@d.permission("edit")
async editEmbedCmd(msg: Message, args: { messageId: string; title?: string; content?: string; color?: string }) {
const savedMessage = await this.savedMessages.find(args.messageId);
if (!savedMessage) {
msg.channel.createMessage(errorMessage("Unknown message"));
return;
}
if (!args.title && !args.content) {
msg.channel.createMessage(errorMessage("Title or content required"));
return;
}
let color = null;
if (args.color) {
const colorMatch = args.color.match(COLOR_MATCH_REGEX);
if (!colorMatch) {
msg.channel.createMessage(errorMessage("Invalid color specified, use hex colors"));
return;
}
color = parseInt(colorMatch[1], 16);
}
const embed: EmbedBase = savedMessage.data.embeds[0];
if (args.title) embed.title = args.title;
if (args.content) embed.description = args.content;
if (color) embed.color = color;
await this.bot.editMessage(savedMessage.channel_id, savedMessage.id, { embed });
}
}