Start huge refactor
Miikka Virtanen
parent
b82eb08505
commit
6d33e7adb4
10
.eslintrc
10
.eslintrc
|
|
@ -11,6 +11,14 @@
|
|||
},
|
||||
|
||||
"rules": {
|
||||
|
||||
"space-infix-ops": "error",
|
||||
"space-unary-ops": ["error", {
|
||||
"words": true,
|
||||
"nonwords": false,
|
||||
"overrides": {
|
||||
"!": true,
|
||||
"!!": true
|
||||
}
|
||||
}]
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
/.vscode
|
||||
/.idea
|
||||
/node_modules
|
||||
/config.json
|
||||
/blocked.json
|
||||
|
|
|
|||
|
|
@ -0,0 +1,2 @@
|
|||
*
|
||||
!/.gitignore
|
||||
601
index.js
601
index.js
|
|
@ -1,601 +0,0 @@
|
|||
const fs = require('fs');
|
||||
const http = require('http');
|
||||
const https = require('https');
|
||||
const url = require('url');
|
||||
const crypto = require('crypto');
|
||||
const publicIp = require('public-ip');
|
||||
const Eris = require('eris');
|
||||
const moment = require('moment');
|
||||
const mime = require('mime');
|
||||
const Queue = require('./queue');
|
||||
const config = require('./config');
|
||||
|
||||
const logServerPort = config.port || 8890;
|
||||
|
||||
const bot = new Eris.CommandClient(config.token, {}, {
|
||||
prefix: config.prefix || '!',
|
||||
ignoreSelf: true,
|
||||
ignoreBots: true,
|
||||
defaultHelpCommand: false,
|
||||
});
|
||||
|
||||
let modMailGuild;
|
||||
const modMailChannels = {};
|
||||
const messageQueue = new Queue();
|
||||
|
||||
const blockFile = `${__dirname}/blocked.json`;
|
||||
let blocked = [];
|
||||
|
||||
const logDir = `${__dirname}/logs`;
|
||||
const logFileFormatRegex = /^([0-9\-]+?)__([0-9]+?)__([0-9a-f]+?)\.txt$/;
|
||||
|
||||
const userMentionRegex = /^<@\!?([0-9]+?)>$/;
|
||||
|
||||
const attachmentDir = `${__dirname}/attachments`;
|
||||
|
||||
try {
|
||||
const blockedJSON = fs.readFileSync(blockFile, {encoding: 'utf8'});
|
||||
blocked = JSON.parse(blockedJSON);
|
||||
} catch(e) {
|
||||
fs.writeFileSync(blockFile, '[]');
|
||||
}
|
||||
|
||||
function saveBlocked() {
|
||||
fs.writeFileSync(blockFile, JSON.stringify(blocked, null, 4));
|
||||
}
|
||||
|
||||
/*
|
||||
* MODMAIL LOG UTILITY FUNCTIONS
|
||||
*/
|
||||
|
||||
function getLogFileInfo(logfile) {
|
||||
const match = logfile.match(logFileFormatRegex);
|
||||
if (! match) return null;
|
||||
|
||||
const date = moment.utc(match[1], 'YYYY-MM-DD-HH-mm-ss').format('YYYY-MM-DD HH:mm:ss');
|
||||
|
||||
return {
|
||||
filename: logfile,
|
||||
date: date,
|
||||
userId: match[2],
|
||||
token: match[3],
|
||||
};
|
||||
}
|
||||
|
||||
function getLogFilePath(logfile) {
|
||||
return `${logDir}/${logfile}`;
|
||||
}
|
||||
|
||||
function getLogFileUrl(logfile) {
|
||||
const info = getLogFileInfo(logfile);
|
||||
|
||||
return publicIp.v4().then(ip => {
|
||||
return `http://${ip}:${logServerPort}/logs/${info.token}`;
|
||||
});
|
||||
}
|
||||
|
||||
function getRandomLogFile(userId) {
|
||||
return new Promise(resolve => {
|
||||
crypto.randomBytes(16, (err, buf) => {
|
||||
const token = buf.toString('hex');
|
||||
const date = moment.utc().format('YYYY-MM-DD-HH-mm-ss');
|
||||
|
||||
resolve(`${date}__${userId}__${token}.txt`);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function findLogFile(token) {
|
||||
return new Promise(resolve => {
|
||||
fs.readdir(logDir, (err, files) => {
|
||||
for (const file of files) {
|
||||
if (file.endsWith(`__${token}.txt`)) {
|
||||
resolve(file);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
resolve(null);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function getLogsByUserId(userId) {
|
||||
return new Promise(resolve => {
|
||||
fs.readdir(logDir, (err, files) => {
|
||||
const logfileInfos = files
|
||||
.map(file => getLogFileInfo(file))
|
||||
.filter(info => info && info.userId === userId);
|
||||
|
||||
resolve(logfileInfos);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function getLogsWithUrlByUserId(userId) {
|
||||
return getLogsByUserId(userId).then(infos => {
|
||||
const urlPromises = infos.map(info => {
|
||||
return getLogFileUrl(info.filename).then(url => {
|
||||
info.url = url;
|
||||
return info;
|
||||
});
|
||||
});
|
||||
|
||||
return Promise.all(urlPromises).then(infos => {
|
||||
infos.sort((a, b) => {
|
||||
if (a.date > b.date) return 1;
|
||||
if (a.date < b.date) return -1;
|
||||
return 0;
|
||||
});
|
||||
|
||||
return infos;
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/*
|
||||
* Attachments
|
||||
*/
|
||||
|
||||
function getAttachmentPath(id) {
|
||||
return `${attachmentDir}/${id}`;
|
||||
}
|
||||
|
||||
function saveAttachment(attachment, tries = 0) {
|
||||
return new Promise((resolve, reject) => {
|
||||
if (tries > 3) {
|
||||
console.error('Attachment download failed after 3 tries:', attachment);
|
||||
reject('Attachment download failed after 3 tries');
|
||||
return;
|
||||
}
|
||||
|
||||
const filepath = getAttachmentPath(attachment.id);
|
||||
const writeStream = fs.createWriteStream(filepath);
|
||||
|
||||
https.get(attachment.url, (res) => {
|
||||
res.pipe(writeStream);
|
||||
writeStream.on('finish', () => {
|
||||
writeStream.close()
|
||||
resolve();
|
||||
});
|
||||
}).on('error', (err) => {
|
||||
fs.unlink(filepath);
|
||||
console.error('Error downloading attachment, retrying');
|
||||
resolve(saveAttachment(attachment));
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function saveAttachments(msg) {
|
||||
if (! msg.attachments || msg.attachments.length === 0) return Promise.resolve();
|
||||
return Promise.all(msg.attachments.map(saveAttachment));
|
||||
}
|
||||
|
||||
function getAttachmentUrl(id, desiredName) {
|
||||
if (desiredName == null) desiredName = 'file.bin';
|
||||
|
||||
return publicIp.v4().then(ip => {
|
||||
return `http://${ip}:${logServerPort}/attachments/${id}/${desiredName}`;
|
||||
});
|
||||
}
|
||||
|
||||
/*
|
||||
* MAIN FUNCTIONALITY
|
||||
*/
|
||||
|
||||
bot.on('ready', () => {
|
||||
modMailGuild = bot.guilds.find(g => g.id === config.mailGuildId);
|
||||
|
||||
if (! modMailGuild) {
|
||||
console.error('You need to set and invite me to the mod mail guild first!');
|
||||
process.exit(0);
|
||||
}
|
||||
|
||||
bot.editStatus(null, {name: config.status || 'Message me for help'});
|
||||
console.log('Bot started, listening to DMs');
|
||||
});
|
||||
|
||||
function getModmailChannelInfo(channel) {
|
||||
if (! channel.topic) return null;
|
||||
|
||||
const match = channel.topic.match(/^MODMAIL\|([0-9]+)\|(.*)$/);
|
||||
if (! match) return null;
|
||||
|
||||
return {
|
||||
userId: match[1],
|
||||
name: match[2],
|
||||
};
|
||||
}
|
||||
|
||||
// This function doesn't actually return the channel object, but an object like {id, _wasCreated}
|
||||
// This is because we can't trust the guild.channels list to actually contain our channel, even if it exists
|
||||
// (has to do with the api events' consistency), and we only cache IDs, so we return an object that can be
|
||||
// constructed from just the ID; we do this even if we find a matching channel so the returned object's signature is consistent
|
||||
function getModmailChannel(user, allowCreate = true) {
|
||||
// If the channel id's in the cache, use that
|
||||
if (modMailChannels[user.id]) {
|
||||
return Promise.resolve({
|
||||
id: modMailChannels[user.id],
|
||||
_wasCreated: false,
|
||||
});
|
||||
}
|
||||
|
||||
// Try to find a matching channel
|
||||
let candidate = modMailGuild.channels.find(c => {
|
||||
const info = getModmailChannelInfo(c);
|
||||
return info && info.userId === user.id;
|
||||
});
|
||||
|
||||
if (candidate) {
|
||||
// If we found a matching channel, return that
|
||||
return Promise.resolve({
|
||||
id: candidate.id,
|
||||
_wasCreated: false,
|
||||
});
|
||||
} else {
|
||||
// If one is not found, create and cache it
|
||||
if (! allowCreate) return Promise.resolve(null);
|
||||
|
||||
let cleanName = user.username.replace(/[^a-zA-Z0-9]/ig, '').toLowerCase().trim();
|
||||
if (cleanName === '') cleanName = 'unknown';
|
||||
|
||||
console.log(`[NOTE] Since no candidate was found, creating channel ${cleanName}-${user.discriminator}`);
|
||||
return modMailGuild.createChannel(`${cleanName}-${user.discriminator}`)
|
||||
.then(channel => {
|
||||
// This is behind a timeout because Discord was telling me the channel didn't exist after creation even though it clearly did
|
||||
// ¯\_(ツ)_/¯
|
||||
return new Promise(resolve => {
|
||||
const topic = `MODMAIL|${user.id}|${user.username}#${user.discriminator}`;
|
||||
setTimeout(() => resolve(channel.edit({topic: topic})), 200);
|
||||
});
|
||||
}, err => {
|
||||
console.error(`Error creating modmail channel for ${user.username}#${user.discriminator}!`);
|
||||
})
|
||||
.then(channel => {
|
||||
modMailChannels[user.id] = channel.id;
|
||||
|
||||
return {
|
||||
id: channel.id,
|
||||
_wasCreated: true,
|
||||
};
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
function formatAttachment(attachment) {
|
||||
let filesize = attachment.size || 0;
|
||||
filesize /= 1024;
|
||||
|
||||
return getAttachmentUrl(attachment.id, attachment.filename).then(attachmentUrl => {
|
||||
return `**Attachment:** ${attachment.filename} (${filesize.toFixed(1)}KB)\n${attachmentUrl}`;
|
||||
});
|
||||
}
|
||||
|
||||
function formatRelayedPM(msg) {
|
||||
let content = msg.content;
|
||||
|
||||
// Get a local URL for all attachments so we don't rely on discord's servers (which delete attachments when the channel/DM thread is deleted)
|
||||
const attachmentFormatPromise = msg.attachments.map(formatAttachment);
|
||||
return Promise.all(attachmentFormatPromise).then(formattedAttachments => {
|
||||
formattedAttachments.forEach(str => {
|
||||
content += `\n\n${str}`;
|
||||
});
|
||||
|
||||
return content;
|
||||
});
|
||||
}
|
||||
|
||||
function getTimestamp(date) {
|
||||
return moment.utc(date).format('HH:mm');
|
||||
}
|
||||
|
||||
// When we get a private message, create a modmail channel or reuse an existing one.
|
||||
// If the channel was not reused, assume it's a new modmail thread and send the user an introduction message.
|
||||
bot.on('messageCreate', (msg) => {
|
||||
if (! (msg.channel instanceof Eris.PrivateChannel)) return;
|
||||
if (msg.author.id === bot.user.id) return;
|
||||
|
||||
if (blocked.indexOf(msg.author.id) !== -1) return;
|
||||
|
||||
saveAttachments(msg);
|
||||
|
||||
// This needs to be queued, as otherwise if a user sent a bunch of messages initially and the createChannel endpoint is delayed, we might get duplicate channels
|
||||
messageQueue.add(() => {
|
||||
return getModmailChannel(msg.author).then(channel => {
|
||||
return formatRelayedPM(msg).then(content => {
|
||||
// Get previous modmail logs for this user
|
||||
// Show a note of them at the beginning of the thread for reference
|
||||
return getLogsByUserId(msg.author.id).then(logs => {
|
||||
if (channel._wasCreated) {
|
||||
if (logs.length > 0) {
|
||||
bot.createMessage(channel.id, `${logs.length} previous modmail logs with this user. Use !logs ${msg.author.id} for details.`);
|
||||
}
|
||||
|
||||
let creationNotificationMessage = `New modmail thread: <#${channel.id}>`;
|
||||
if (config.pingCreationNotification) creationNotificationMessage = `@here ${creationNotificationMessage}`;
|
||||
|
||||
bot.createMessage(modMailGuild.id, {
|
||||
content: creationNotificationMessage,
|
||||
disableEveryone: false,
|
||||
});
|
||||
|
||||
msg.channel.createMessage("Thank you for your message! Our mod team will reply to you here as soon as possible.").then(null, (err) => {
|
||||
bot.createMessage(modMailGuild.id, {
|
||||
content: `There is an issue sending messages to ${msg.author.username}#${msg.author.discriminator} (id ${msg.author.id}); consider messaging manually`
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
const timestamp = getTimestamp();
|
||||
bot.createMessage(channel.id, `[${timestamp}] « **${msg.author.username}#${msg.author.discriminator}:** ${content}`);
|
||||
}); // getLogsByUserId
|
||||
}); // formatRelayedPM
|
||||
}); // getModmailChannel
|
||||
}); // messageQueue.add
|
||||
});
|
||||
|
||||
// Edits in PMs
|
||||
bot.on('messageUpdate', (msg, oldMessage) => {
|
||||
if (! (msg.channel instanceof Eris.PrivateChannel)) return;
|
||||
if (msg.author.id === bot.user.id) return;
|
||||
|
||||
if (blocked.indexOf(msg.author.id) !== -1) return;
|
||||
|
||||
let oldContent = oldMessage.content;
|
||||
const newContent = msg.content;
|
||||
|
||||
if (oldContent == null) oldContent = '*Unavailable due to bot restart*';
|
||||
|
||||
getModmailChannel(msg.author, false).then(channel => {
|
||||
if (! channel) return;
|
||||
bot.createMessage(channel.id, `**The user edited their message:**\n**Before:** ${oldContent}\n**After:** ${newContent}`);
|
||||
});
|
||||
});
|
||||
|
||||
// "Bot was mentioned in #general-discussion"
|
||||
bot.on('messageCreate', msg => {
|
||||
if (msg.author.id === bot.user.id) return;
|
||||
if (blocked.indexOf(msg.author.id) !== -1) return;
|
||||
|
||||
if (msg.mentions.some(user => user.id === bot.user.id)) {
|
||||
bot.createMessage(modMailGuild.id, {
|
||||
content: `@here Bot mentioned in ${msg.channel.mention} by **${msg.author.username}#${msg.author.discriminator}**: "${msg.cleanContent}"`,
|
||||
disableEveryone: false,
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
// Mods can reply to modmail threads using !r or !reply
|
||||
// These messages get relayed back to the DM thread between the bot and the user
|
||||
// Attachments are shown as URLs
|
||||
bot.registerCommand('reply', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== modMailGuild.id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
const channelInfo = getModmailChannelInfo(msg.channel);
|
||||
if (! channelInfo) return;
|
||||
|
||||
saveAttachments(msg).then(() => {
|
||||
bot.getDMChannel(channelInfo.userId).then(dmChannel => {
|
||||
const roleId = msg.member.roles[0];
|
||||
const role = (roleId ? (modMailGuild.roles.get(roleId) || {}).name : '');
|
||||
const roleStr = (role ? `(${role}) ` : '');
|
||||
|
||||
let argMsg = args.join(' ').trim();
|
||||
let content = `**${roleStr}${msg.author.username}:** ${argMsg}`;
|
||||
|
||||
const sendMessage = (file, attachmentUrl) => {
|
||||
dmChannel.createMessage(content, file).then(() => {
|
||||
if (attachmentUrl) content += `\n\n**Attachment:** ${attachmentUrl}`;
|
||||
|
||||
const timestamp = getTimestamp();
|
||||
msg.channel.createMessage(`[${timestamp}] » ${content}`);
|
||||
}, (err) => {
|
||||
if (err.resp && err.resp.statusCode === 403) {
|
||||
msg.channel.createMessage(`Could not send reply; the user has likely blocked the bot`);
|
||||
} else if (err.resp) {
|
||||
msg.channel.createMessage(`Could not send reply; error code ${err.resp.statusCode}`);
|
||||
} else {
|
||||
msg.channel.createMessage(`Could not send reply: ${err.toString()}`);
|
||||
}
|
||||
});
|
||||
|
||||
msg.delete();
|
||||
};
|
||||
|
||||
if (msg.attachments.length > 0) {
|
||||
fs.readFile(getAttachmentPath(msg.attachments[0].id), (err, data) => {
|
||||
const file = {file: data, name: msg.attachments[0].filename};
|
||||
|
||||
getAttachmentUrl(msg.attachments[0].id, msg.attachments[0].filename).then(attachmentUrl => {
|
||||
sendMessage(file, attachmentUrl);
|
||||
});
|
||||
});
|
||||
} else {
|
||||
sendMessage();
|
||||
}
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
bot.registerCommandAlias('r', 'reply');
|
||||
|
||||
bot.registerCommand('close', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== modMailGuild.id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
const channelInfo = getModmailChannelInfo(msg.channel);
|
||||
if (! channelInfo) return;
|
||||
|
||||
msg.channel.createMessage('Saving logs and closing channel...');
|
||||
msg.channel.getMessages(10000).then(messages => {
|
||||
const log = messages.reverse().map(message => {
|
||||
const date = moment.utc(message.timestamp, 'x').format('YYYY-MM-DD HH:mm:ss');
|
||||
return `[${date}] ${message.author.username}#${message.author.discriminator}: ${message.content}`;
|
||||
}).join('\n') + '\n';
|
||||
|
||||
getRandomLogFile(channelInfo.userId).then(logfile => {
|
||||
fs.writeFile(getLogFilePath(logfile), log, {encoding: 'utf8'}, err => {
|
||||
getLogFileUrl(logfile).then(logurl => {
|
||||
const closeMessage = `Modmail thread with ${channelInfo.name} (${channelInfo.userId}) was closed by ${msg.author.mention}
|
||||
Logs: <${logurl}>`;
|
||||
|
||||
bot.createMessage(modMailGuild.id, closeMessage);
|
||||
|
||||
delete modMailChannels[channelInfo.userId];
|
||||
msg.channel.delete();
|
||||
});
|
||||
});
|
||||
})
|
||||
});
|
||||
});
|
||||
|
||||
bot.registerCommand('block', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== modMailGuild.id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
let userId;
|
||||
if (args[0]) {
|
||||
if (args[0].match(/^[0-9]+$/)) {
|
||||
userId = args[0];
|
||||
} else {
|
||||
let mentionMatch = args[0].match(userMentionRegex);
|
||||
if (mentionMatch) userId = mentionMatch[1];
|
||||
}
|
||||
} else {
|
||||
const modmailChannelInfo = getModmailChannelInfo(msg.channel);
|
||||
if (modmailChannelInfo) userId = modmailChannelInfo.userId;
|
||||
}
|
||||
|
||||
if (! userId) return;
|
||||
|
||||
blocked.push(userId);
|
||||
saveBlocked();
|
||||
msg.channel.createMessage(`Blocked <@${userId}> (id ${userId}) from modmail`);
|
||||
});
|
||||
|
||||
bot.registerCommand('unblock', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== modMailGuild.id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
let userId;
|
||||
if (args[0]) {
|
||||
if (args[0].match(/^[0-9]+$/)) {
|
||||
userId = args[0];
|
||||
} else {
|
||||
let mentionMatch = args[0].match(userMentionRegex);
|
||||
if (mentionMatch) userId = mentionMatch[1];
|
||||
}
|
||||
} else {
|
||||
const modmailChannelInfo = getModmailChannelInfo(msg.channel);
|
||||
if (modmailChannelInfo) userId = modmailChannelInfo.userId;
|
||||
}
|
||||
|
||||
if (! userId) return;
|
||||
|
||||
blocked.splice(blocked.indexOf(userId), 1);
|
||||
saveBlocked();
|
||||
msg.channel.createMessage(`Unblocked <@${userId}> (id ${userId}) from modmail`);
|
||||
});
|
||||
|
||||
bot.registerCommand('logs', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== modMailGuild.id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
let userId;
|
||||
if (args[0]) {
|
||||
if (args[0].match(/^[0-9]+$/)) {
|
||||
userId = args[0];
|
||||
} else {
|
||||
let mentionMatch = args[0].match(userMentionRegex);
|
||||
if (mentionMatch) userId = mentionMatch[1];
|
||||
}
|
||||
} else {
|
||||
const modmailChannelInfo = getModmailChannelInfo(msg.channel);
|
||||
if (modmailChannelInfo) userId = modmailChannelInfo.userId;
|
||||
}
|
||||
|
||||
if (! userId) return;
|
||||
|
||||
getLogsWithUrlByUserId(userId).then(infos => {
|
||||
let message = `**Log files for <@${userId}>:**\n`;
|
||||
|
||||
message += infos.map(info => {
|
||||
const formattedDate = moment.utc(info.date, 'YYYY-MM-DD HH:mm:ss').format('MMM Do [at] HH:mm [UTC]');
|
||||
return `\`${formattedDate}\`: <${info.url}>`;
|
||||
}).join('\n');
|
||||
|
||||
msg.channel.createMessage(message);
|
||||
});
|
||||
});
|
||||
|
||||
bot.on('channelCreate', channel => {
|
||||
console.log(`[NOTE] Got channel creation event for #${channel.name} (ID ${channel.id})`);
|
||||
});
|
||||
|
||||
bot.connect();
|
||||
|
||||
/*
|
||||
* MODMAIL LOG SERVER
|
||||
*/
|
||||
|
||||
function serveLogs(res, pathParts) {
|
||||
const token = pathParts[pathParts.length - 1];
|
||||
if (token.match(/^[0-9a-f]+$/) === null) return res.end();
|
||||
|
||||
findLogFile(token).then(logfile => {
|
||||
if (logfile === null) return res.end();
|
||||
|
||||
fs.readFile(getLogFilePath(logfile), {encoding: 'utf8'}, (err, data) => {
|
||||
if (err) {
|
||||
res.statusCode = 404;
|
||||
res.end('Log not found');
|
||||
return;
|
||||
}
|
||||
|
||||
res.setHeader('Content-Type', 'text/plain; charset=UTF-8');
|
||||
res.end(data);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function serveAttachments(res, pathParts) {
|
||||
const desiredFilename = pathParts[pathParts.length - 1];
|
||||
const id = pathParts[pathParts.length - 2];
|
||||
|
||||
if (id.match(/^[0-9]+$/) === null) return res.end();
|
||||
if (desiredFilename.match(/^[0-9a-z\._-]+$/i) === null) return res.end();
|
||||
|
||||
const attachmentPath = getAttachmentPath(id);
|
||||
fs.access(attachmentPath, (err) => {
|
||||
if (err) {
|
||||
res.statusCode = 404;
|
||||
res.end('Attachment not found');
|
||||
return;
|
||||
}
|
||||
|
||||
const filenameParts = desiredFilename.split('.');
|
||||
const ext = (filenameParts.length > 1 ? filenameParts[filenameParts.length - 1] : 'bin');
|
||||
const fileMime = mime.lookup(ext);
|
||||
|
||||
res.setHeader('Content-Type', fileMime);
|
||||
|
||||
const read = fs.createReadStream(attachmentPath);
|
||||
read.pipe(res);
|
||||
})
|
||||
}
|
||||
|
||||
const server = http.createServer((req, res) => {
|
||||
const parsedUrl = url.parse(`http://${req.url}`);
|
||||
const pathParts = parsedUrl.path.split('/').filter(v => v !== '');
|
||||
|
||||
if (parsedUrl.path.startsWith('/logs/')) serveLogs(res, pathParts);
|
||||
if (parsedUrl.path.startsWith('/attachments/')) serveAttachments(res, pathParts);
|
||||
});
|
||||
|
||||
server.listen(logServerPort);
|
||||
|
|
@ -0,0 +1,52 @@
|
|||
const fs = require('fs');
|
||||
const https = require('https');
|
||||
const config = require('../config');
|
||||
const utils = require('./utils');
|
||||
|
||||
const attachmentDir = config.attachmentDir || `${__dirname}/attachments`;
|
||||
|
||||
function getAttachmentPath(id) {
|
||||
return `${attachmentDir}/${id}`;
|
||||
}
|
||||
|
||||
function saveAttachment(attachment, tries = 0) {
|
||||
return new Promise((resolve, reject) => {
|
||||
if (tries > 3) {
|
||||
console.error('Attachment download failed after 3 tries:', attachment);
|
||||
reject('Attachment download failed after 3 tries');
|
||||
return;
|
||||
}
|
||||
|
||||
const filepath = getAttachmentPath(attachment.id);
|
||||
const writeStream = fs.createWriteStream(filepath);
|
||||
|
||||
https.get(attachment.url, (res) => {
|
||||
res.pipe(writeStream);
|
||||
writeStream.on('finish', () => {
|
||||
writeStream.close()
|
||||
resolve();
|
||||
});
|
||||
}).on('error', (err) => {
|
||||
fs.unlink(filepath);
|
||||
console.error('Error downloading attachment, retrying');
|
||||
resolve(saveAttachment(attachment));
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function saveAttachments(msg) {
|
||||
if (! msg.attachments || msg.attachments.length === 0) return Promise.resolve();
|
||||
return Promise.all(msg.attachments.map(saveAttachment));
|
||||
}
|
||||
|
||||
function getAttachmentUrl(id, desiredName) {
|
||||
if (desiredName == null) desiredName = 'file.bin';
|
||||
return utils.getSelfUrl(`attachments/${id}/${desiredName}`);
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
getAttachmentPath,
|
||||
saveAttachment,
|
||||
saveAttachments,
|
||||
getAttachmentUrl,
|
||||
};
|
||||
|
|
@ -0,0 +1,27 @@
|
|||
const jsonDb = require('./jsonDb');
|
||||
|
||||
function isBlocked(userId) {
|
||||
return jsonDb.get('blocked').then(blocked => {
|
||||
return blocked.indexOf(userId) !== -1;
|
||||
});
|
||||
}
|
||||
|
||||
function block(userId) {
|
||||
return jsonDb.get('blocked').then(blocked => {
|
||||
blocked.push(userId);
|
||||
return jsonDb.save('blocked', blocked);
|
||||
});
|
||||
}
|
||||
|
||||
function unblock(userId) {
|
||||
return jsonDb.get('blocked').then(blocked => {
|
||||
blocked.splice(blocked.indexOf(userId), 1);
|
||||
return jsonDb.save('blocked', blocked);
|
||||
});
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
isBlocked,
|
||||
block,
|
||||
unblock,
|
||||
};
|
||||
|
|
@ -0,0 +1,304 @@
|
|||
const Eris = require('eris');
|
||||
const moment = require('moment');
|
||||
const Queue = require('./queue');
|
||||
const config = require('../config');
|
||||
const utils = require('./utils');
|
||||
const blocked = require('./blocked');
|
||||
const threads = require('./threads');
|
||||
const logs = require('./logs');
|
||||
const attachments = require('./attachments');
|
||||
const webserver = require('./webserver');
|
||||
|
||||
const bot = new Eris.CommandClient(config.token, {}, {
|
||||
prefix: config.prefix || '!',
|
||||
ignoreSelf: true,
|
||||
ignoreBots: true,
|
||||
defaultHelpCommand: false,
|
||||
});
|
||||
|
||||
const messageQueue = new Queue();
|
||||
|
||||
bot.on('ready', () => {
|
||||
bot.editStatus(null, {name: config.status || 'Message me for help'});
|
||||
console.log('Bot started, listening to DMs');
|
||||
});
|
||||
|
||||
// "Bot was mentioned in #general-discussion"
|
||||
bot.on('messageCreate', msg => {
|
||||
if (msg.author.id === bot.user.id) return;
|
||||
|
||||
if (msg.mentions.some(user => user.id === bot.user.id)) {
|
||||
blocked.isBlocked(msg.author.id).then(isBlocked => {
|
||||
if (isBlocked) return;
|
||||
|
||||
bot.createMessage(utils.getModmailGuild(bot).id, {
|
||||
content: `@here Bot mentioned in ${msg.channel.mention} by **${msg.author.username}#${msg.author.discriminator}**: "${msg.cleanContent}"`,
|
||||
disableEveryone: false,
|
||||
});
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
// When we get a private message, forward the contents to the corresponding modmail thread
|
||||
bot.on('messageCreate', (msg) => {
|
||||
if (! (msg.channel instanceof Eris.PrivateChannel)) return;
|
||||
if (msg.author.id === bot.user.id) return;
|
||||
|
||||
blocked.isBlocked(msg.author.id).then(isBlocked => {
|
||||
if (isBlocked) return;
|
||||
|
||||
// Download and save copies of attachments in the background
|
||||
attachments.saveAttachments(msg);
|
||||
|
||||
let thread, logs;
|
||||
|
||||
messageQueue.add(() => {
|
||||
threads.getForUser(bot, msg.author)
|
||||
.then(userThread => {
|
||||
thread = userThread;
|
||||
return logs.getLogsByUserId(msg.author.id);
|
||||
})
|
||||
.then(userLogs => {
|
||||
logs = userLogs;
|
||||
return utils.formatUserDM(msg);
|
||||
})
|
||||
.then(content => {
|
||||
// If the thread does not exist and could not be created, send a warning about this to all mods so they can DM the user directly instead
|
||||
if (! thread) {
|
||||
let warningMessage = `
|
||||
@here Error creating modmail thread for ${msg.author.username}#${msg.author.discriminator} (${msg.author.id})!
|
||||
|
||||
Here's what their message contained:
|
||||
\`\`\`${content}\`\`\`
|
||||
`;
|
||||
|
||||
bot.createMessage(utils.getModmailGuild(bot).id, {
|
||||
content: `@here Error creating modmail thread for ${msg.author.username}#${msg.author.discriminator} (${msg.author.id})!`,
|
||||
disableEveryone: false,
|
||||
});
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
// If the thread was just created, do some extra stuff
|
||||
if (thread._wasCreated) {
|
||||
// Mention previous logs at the start of the thread
|
||||
if (logs.length > 0) {
|
||||
bot.createMessage(thread.channelId, `${logs.length} previous modmail logs with this user. Use !logs ${msg.author.id} for details.`);
|
||||
}
|
||||
|
||||
// Ping mods of the new thread
|
||||
let creationNotificationMessage = `New modmail thread: <#${channel.id}>`;
|
||||
if (config.pingCreationNotification) creationNotificationMessage = `@here ${creationNotificationMessage}`;
|
||||
|
||||
bot.createMessage(utils.getModmailGuild(bot).id, {
|
||||
content: creationNotificationMessage,
|
||||
disableEveryone: false,
|
||||
});
|
||||
|
||||
// Send an automatic reply to the user informing them of the successfully created modmail thread
|
||||
msg.channel.createMessage("Thank you for your message! Our mod team will reply to you here as soon as possible.").then(null, (err) => {
|
||||
bot.createMessage(modMailGuild.id, {
|
||||
content: `There is an issue sending messages to ${msg.author.username}#${msg.author.discriminator} (id ${msg.author.id}); consider messaging manually`
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
const timestamp = utils.getTimestamp();
|
||||
bot.createMessage(channel.id, `[${timestamp}] « **${msg.author.username}#${msg.author.discriminator}:** ${content}`);
|
||||
})
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
// Edits in DMs
|
||||
bot.on('messageUpdate', (msg, oldMessage) => {
|
||||
if (! (msg.channel instanceof Eris.PrivateChannel)) return;
|
||||
if (msg.author.id === bot.user.id) return;
|
||||
|
||||
blocked.isBlocked(msg.author.id).then(isBlocked => {
|
||||
if (isBlocked) return;
|
||||
|
||||
let oldContent = oldMessage.content;
|
||||
const newContent = msg.content;
|
||||
|
||||
if (oldContent == null) oldContent = '*Unavailable due to bot restart*';
|
||||
|
||||
threads.getForUser(bot, msg.author).then(thread => {
|
||||
if (! thread) return;
|
||||
|
||||
const editMessage = utils.disableLinkPreviews(`**The user edited their message:**\n**Before:** ${oldContent}\n**After:** ${newContent}`);
|
||||
|
||||
bot.createMessage(thread.channelId, editMessage);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
// Mods can reply to modmail threads using !r or !reply
|
||||
// These messages get relayed back to the DM thread between the bot and the user
|
||||
bot.registerCommand('reply', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== utils.getModmailGuild(bot).id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
threads.getByChannelId(msg.channel.id).then(thread => {
|
||||
if (! thread) return;
|
||||
|
||||
attachments.saveAttachments(msg).then(() => {
|
||||
bot.getDMChannel(thread.userId).then(dmChannel => {
|
||||
const roleId = msg.member.roles[0];
|
||||
const role = (roleId ? (modMailGuild.roles.get(roleId) || {}).name : '');
|
||||
const roleStr = (role ? `(${role}) ` : '');
|
||||
|
||||
let argMsg = args.join(' ').trim();
|
||||
let content = `**${roleStr}${msg.author.username}:** ${argMsg}`;
|
||||
|
||||
function sendMessage(file, attachmentUrl) {
|
||||
dmChannel.createMessage(content, file).then(() => {
|
||||
if (attachmentUrl) content += `\n\n**Attachment:** ${attachmentUrl}`;
|
||||
|
||||
const timestamp = getTimestamp();
|
||||
msg.channel.createMessage(`[${timestamp}] » ${content}`);
|
||||
}, (err) => {
|
||||
if (err.resp && err.resp.statusCode === 403) {
|
||||
msg.channel.createMessage(`Could not send reply; the user has likely blocked the bot`);
|
||||
} else if (err.resp) {
|
||||
msg.channel.createMessage(`Could not send reply; error code ${err.resp.statusCode}`);
|
||||
} else {
|
||||
msg.channel.createMessage(`Could not send reply: ${err.toString()}`);
|
||||
}
|
||||
});
|
||||
|
||||
msg.delete();
|
||||
};
|
||||
|
||||
// If the reply has an attachment, relay it as is
|
||||
if (msg.attachments.length > 0) {
|
||||
fs.readFile(attachments.getAttachmentPath(msg.attachments[0].id), (err, data) => {
|
||||
const file = {file: data, name: msg.attachments[0].filename};
|
||||
|
||||
getAttachmentUrl(msg.attachments[0].id, msg.attachments[0].filename).then(attachmentUrl => {
|
||||
sendMessage(file, attachmentUrl);
|
||||
});
|
||||
});
|
||||
} else {
|
||||
sendMessage();
|
||||
}
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
bot.registerCommandAlias('r', 'reply');
|
||||
|
||||
bot.registerCommand('close', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== modMailGuild.id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
threads.getByChannelId(msg.channel.id).then(thread => {
|
||||
if (! thread) return;
|
||||
|
||||
msg.channel.createMessage('Saving logs and closing channel...');
|
||||
msg.channel.getMessages(10000).then(messages => {
|
||||
const log = messages.reverse().map(msg => {
|
||||
const date = moment.utc(msg.timestamp, 'x').format('YYYY-MM-DD HH:mm:ss');
|
||||
return `[${date}] ${msg.author.username}#${msg.author.discriminator}: ${msg.content}`;
|
||||
}).join('\n') + '\n';
|
||||
|
||||
logs.getNewLogFile(thread.userId).then(logFilename => {
|
||||
logs.saveLogFile(logFilename, log)
|
||||
.then(() => getLogFileUrl(logFilename))
|
||||
.then(url => {
|
||||
const closeMessage = `Modmail thread with ${thread.username} (${thread.userId}) was closed by ${msg.author.mention}
|
||||
Logs: <${url}>`;
|
||||
|
||||
bot.createMessage(utils.getModmailGuild(bot).id, closeMessage);
|
||||
threads.close(thread.channelId).then(() => msg.channel.delete());
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
bot.registerCommand('block', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== utils.getModmailGuild(bot).id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
function block(userId) {
|
||||
blocked.block(userId).then(() => {
|
||||
msg.channel.createMessage(`Blocked <@${userId}> (id ${userId}) from modmail`);
|
||||
});
|
||||
}
|
||||
|
||||
if (args.length > 0) {
|
||||
const userId = utils.getUserMention(args.join(' '));
|
||||
if (! userId) return;
|
||||
block(userId);
|
||||
} else {
|
||||
// Calling !block without args in a modmail thread blocks the user of that thread
|
||||
threads.getByChannelId(msg.channel.id).then(thread => {
|
||||
if (! thread) return;
|
||||
block(userId);
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
bot.registerCommand('unblock', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== utils.getModmailGuild(bot).id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
function unblock(userId) {
|
||||
blocked.unblock(userId).then(() => {
|
||||
msg.channel.createMessage(`Unblocked <@${userId}> (id ${userId}) from modmail`);
|
||||
});
|
||||
}
|
||||
|
||||
if (args.length > 0) {
|
||||
const userId = utils.getUserMention(args.join(' '));
|
||||
if (! userId) return;
|
||||
unblock(userId);
|
||||
} else {
|
||||
// Calling !unblock without args in a modmail thread unblocks the user of that thread
|
||||
threads.getByChannelId(msg.channel.id).then(thread => {
|
||||
if (! thread) return;
|
||||
unblock(userId);
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
bot.registerCommand('logs', (msg, args) => {
|
||||
if (! msg.channel.guild) return;
|
||||
if (msg.channel.guild.id !== utils.getModmailGuild(bot).id) return;
|
||||
if (! msg.member.permission.has('manageRoles')) return;
|
||||
|
||||
function getLogs(userId) {
|
||||
getLogsWithUrlByUserId(userId).then(infos => {
|
||||
let message = `**Log files for <@${userId}>:**\n`;
|
||||
|
||||
message += infos.map(info => {
|
||||
const formattedDate = moment.utc(info.date, 'YYYY-MM-DD HH:mm:ss').format('MMM Do [at] HH:mm [UTC]');
|
||||
return `\`${formattedDate}\`: <${info.url}>`;
|
||||
}).join('\n');
|
||||
|
||||
msg.channel.createMessage(message);
|
||||
});
|
||||
}
|
||||
|
||||
if (args.length > 0) {
|
||||
const userId = utils.getUserMention(args.join(' '));
|
||||
if (! userId) return;
|
||||
getLogs(userId);
|
||||
} else {
|
||||
// Calling !logs without args in a modmail thread returns the logs of the user of that thread
|
||||
threads.getByChannelId(msg.channel.id).then(thread => {
|
||||
if (! thread) return;
|
||||
getLogs(userId);
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
bot.connect();
|
||||
webserver.run();
|
||||
|
|
@ -0,0 +1,67 @@
|
|||
const fs = require('fs');
|
||||
const path = require('path');
|
||||
|
||||
const dbDir = config.dbDir || `${__dirname}/db`;
|
||||
|
||||
const databases = {};
|
||||
|
||||
class JSONDB {
|
||||
constructor(path, def = {}, useCloneByDefault = true) {
|
||||
this.path = path;
|
||||
this.useCloneByDefault = useCloneByDefault;
|
||||
|
||||
this.load = new Promise(resolve => {
|
||||
fs.readFile(path, {encoding: 'utf8'}, (err, data) => {
|
||||
if (err) return resolve(def);
|
||||
|
||||
let unserialized;
|
||||
try { unserialized = JSON.parse(data); }
|
||||
catch (e) { unserialized = def; }
|
||||
|
||||
resolve(unserialized);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
get(clone) {
|
||||
if (clone == null) clone = this.useCloneByDefault;
|
||||
|
||||
return this.load.then(data => {
|
||||
if (clone) return JSON.parse(JSON.stringify(data));
|
||||
else return data;
|
||||
});
|
||||
}
|
||||
|
||||
save(newData) {
|
||||
const serialized = JSON.stringify(newData);
|
||||
this.load = new Promise((resolve, reject) => {
|
||||
fs.writeFile(this.path, serialized, {encoding: 'utf8'}, () => {
|
||||
resolve(newData);
|
||||
});
|
||||
});
|
||||
|
||||
return this.get();
|
||||
}
|
||||
}
|
||||
|
||||
function getDb(dbName, def) {
|
||||
if (! databases[dbName]) {
|
||||
const dbPath = path.resolve(dbDir, `${dbName}.json`);
|
||||
databases[dbName] = new JSONDB(dbPath, def);
|
||||
}
|
||||
|
||||
return databases[dbName];
|
||||
}
|
||||
|
||||
function get(dbName, def) {
|
||||
return getDb(dbName, def).get();
|
||||
}
|
||||
|
||||
function save(dbName, data) {
|
||||
return getDb(dbName, data).save(data);
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
get,
|
||||
save,
|
||||
};
|
||||
|
|
@ -0,0 +1,108 @@
|
|||
const fs = require('fs');
|
||||
const crypto = require('crypto');
|
||||
const moment = require('moment');
|
||||
const config = require('../config');
|
||||
|
||||
const logDir = config.logDir || `${__dirname}/logs`;
|
||||
const logFileFormatRegex = /^([0-9\-]+?)__([0-9]+?)__([0-9a-f]+?)\.txt$/;
|
||||
|
||||
function getLogFileInfo(logFilename) {
|
||||
const match = logFilename.match(logFileFormatRegex);
|
||||
if (! match) return null;
|
||||
|
||||
const date = moment.utc(match[1], 'YYYY-MM-DD-HH-mm-ss').format('YYYY-MM-DD HH:mm:ss');
|
||||
|
||||
return {
|
||||
filename: logFilename,
|
||||
date: date,
|
||||
userId: match[2],
|
||||
token: match[3],
|
||||
};
|
||||
}
|
||||
|
||||
function getLogFilePath(logFilename) {
|
||||
return `${logDir}/${logFilename}`;
|
||||
}
|
||||
|
||||
function getLogFileUrl(logFilename) {
|
||||
const info = getLogFileInfo(logFilename);
|
||||
return utils.getSelfUrl(`logs/${info.token}`);
|
||||
}
|
||||
|
||||
function getNewLogFile(userId) {
|
||||
return new Promise(resolve => {
|
||||
crypto.randomBytes(16, (err, buf) => {
|
||||
const token = buf.toString('hex');
|
||||
const date = moment.utc().format('YYYY-MM-DD-HH-mm-ss');
|
||||
|
||||
resolve(`${date}__${userId}__${token}.txt`);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function findLogFile(token) {
|
||||
return new Promise(resolve => {
|
||||
fs.readdir(logDir, (err, files) => {
|
||||
for (const file of files) {
|
||||
if (file.endsWith(`__${token}.txt`)) {
|
||||
resolve(file);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
resolve(null);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function getLogsByUserId(userId) {
|
||||
return new Promise(resolve => {
|
||||
fs.readdir(logDir, (err, files) => {
|
||||
const logfileInfos = files
|
||||
.map(file => getLogFileInfo(file))
|
||||
.filter(info => info && info.userId === userId);
|
||||
|
||||
resolve(logfileInfos);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function getLogsWithUrlByUserId(userId) {
|
||||
return getLogsByUserId(userId).then(infos => {
|
||||
const urlPromises = infos.map(info => {
|
||||
return getLogFileUrl(info.filename).then(url => {
|
||||
info.url = url;
|
||||
return info;
|
||||
});
|
||||
});
|
||||
|
||||
return Promise.all(urlPromises).then(infos => {
|
||||
infos.sort((a, b) => {
|
||||
if (a.date > b.date) return 1;
|
||||
if (a.date < b.date) return -1;
|
||||
return 0;
|
||||
});
|
||||
|
||||
return infos;
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function saveLogFile(logFilename, content) {
|
||||
return new Promise((resolve, reject) => {
|
||||
fs.writeFile(getLogFilePath(logFilename), content, {encoding: 'utf8'}, err => {
|
||||
if (err) return reject(err);
|
||||
resolve();
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
getLogFileInfo,
|
||||
getLogFilePath,
|
||||
getNewLogFile,
|
||||
findLogFile,
|
||||
getLogsByUserId,
|
||||
getLogsWithUrlByUserId,
|
||||
saveLogFile,
|
||||
};
|
||||
|
|
@ -6,7 +6,7 @@ class Queue {
|
|||
|
||||
add(fn) {
|
||||
this.queue.push(fn);
|
||||
if (!this.running) this.next();
|
||||
if (! this.running) this.next();
|
||||
}
|
||||
|
||||
next() {
|
||||
|
|
@ -18,7 +18,11 @@ class Queue {
|
|||
}
|
||||
|
||||
const fn = this.queue.shift();
|
||||
Promise.resolve(fn()).then(() => this.next());
|
||||
new Promise(resolve => {
|
||||
// Either fn() completes or the timeout of 10sec is reached
|
||||
Promise.resolve(fn()).then(resolve);
|
||||
setTimeout(resolve, 10000);
|
||||
}).then(() => this.next());
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,80 @@
|
|||
const Eris = require('eris');
|
||||
const utils = require('./utils');
|
||||
const jsonDb = require('./jsonDb');
|
||||
|
||||
/**
|
||||
* @typedef {Object} ModMailThread
|
||||
* @property {String} channelId
|
||||
* @property {String} userId
|
||||
* @property {String} username
|
||||
* @property {Boolean} _wasCreated
|
||||
*/
|
||||
|
||||
/**
|
||||
* Returns information about the modmail thread channel for the given user. We can't return channel objects
|
||||
* directly since they're not always available immediately after creation.
|
||||
* @param {Eris.Client} bot
|
||||
* @param {Eris.User} user
|
||||
* @param {Boolean} allowCreate
|
||||
* @returns {Promise<ThreadInfo>}
|
||||
*/
|
||||
function getForUser(bot, user, allowCreate = true) {
|
||||
return jsonDb.get('threads').then(threads => {
|
||||
const thread = threads.find(t => t.userId === user.id);
|
||||
if (thread) return thread;
|
||||
|
||||
// If we didn't find an existing modmail thread, attempt creating one
|
||||
if (! allowCreate) return null;
|
||||
|
||||
// Channel names are particularly picky about what characters they allow...
|
||||
let cleanName = user.username.replace(/[^a-zA-Z0-9]/ig, '').toLowerCase().trim();
|
||||
if (cleanName === '') cleanName = 'unknown';
|
||||
|
||||
const channelName = `${cleanName}-${user.discriminator}`;
|
||||
console.log(`[NOTE] Creating new thread channel ${channelName}`);
|
||||
|
||||
return utils.getModmailGuild(bot).createChannel(`${channelName}`)
|
||||
.then(channel => {
|
||||
const thread = {
|
||||
channelId: channel.id,
|
||||
userId: user.id,
|
||||
username: `${user.username}#${user.discriminator}`,
|
||||
};
|
||||
|
||||
const threads = jsonDb.get('threads');
|
||||
threads.push(thread);
|
||||
jsonDb.save('threads', threads);
|
||||
|
||||
thread._wasCreated = true;
|
||||
return thread;
|
||||
}, err => {
|
||||
console.error(`Error creating modmail channel for ${user.username}#${user.discriminator}!`);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* @param {String} channelId
|
||||
* @returns {Promise<ThreadInfo>}
|
||||
*/
|
||||
function getByChannelId(channelId) {
|
||||
return jsonDb.get('threads').then(threads => {
|
||||
return threads.find(t => t.userId === user.id);
|
||||
});
|
||||
}
|
||||
|
||||
function close(channelId) {
|
||||
return jsonDb.get('threads').then(threads => {
|
||||
const thread = threads.find(t => t.userId === user.id);
|
||||
if (! thread) return;
|
||||
|
||||
threads.splice(threads.indexOf(thread), 1);
|
||||
return jsonDb.save('threads', threads);
|
||||
});
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
getForUser,
|
||||
getByChannelId,
|
||||
close,
|
||||
};
|
||||
|
|
@ -0,0 +1,75 @@
|
|||
const moment = require('moment');
|
||||
const publicIp = require('public-ip');
|
||||
const config = require('../config');
|
||||
const utils = require('./utils');
|
||||
|
||||
function getModmailGuild(bot) {
|
||||
return bot.guilds.find(g => g.id === config.mailGuildId);
|
||||
}
|
||||
|
||||
function formatAttachment(attachment) {
|
||||
let filesize = attachment.size || 0;
|
||||
filesize /= 1024;
|
||||
|
||||
return utils.getAttachmentUrl(attachment.id, attachment.filename).then(attachmentUrl => {
|
||||
return `**Attachment:** ${attachment.filename} (${filesize.toFixed(1)}KB)\n${attachmentUrl}`;
|
||||
});
|
||||
}
|
||||
|
||||
function formatUserDM(msg) {
|
||||
let content = msg.content;
|
||||
|
||||
// Get a local URL for all attachments so we don't rely on discord's servers (which delete attachments when the channel/DM thread is deleted)
|
||||
const attachmentFormatPromise = msg.attachments.map(formatAttachment);
|
||||
return Promise.all(attachmentFormatPromise).then(formattedAttachments => {
|
||||
formattedAttachments.forEach(str => {
|
||||
content += `\n\n${str}`;
|
||||
});
|
||||
|
||||
return content;
|
||||
});
|
||||
}
|
||||
|
||||
const userMentionRegex = /^<@\!?([0-9]+?)>$/;
|
||||
|
||||
function getUserMention(str) {
|
||||
str = str.trim();
|
||||
|
||||
if (str.match(/^[0-9]+$/)) {
|
||||
// User ID
|
||||
return str;
|
||||
} else {
|
||||
let mentionMatch = str.match(userMentionRegex);
|
||||
if (mentionMatch) return mentionMatch[1];
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
function getTimestamp(date) {
|
||||
return moment.utc(date).format('HH:mm');
|
||||
}
|
||||
|
||||
function disableLinkPreviews(str) {
|
||||
return str.replace(/(^|[^<])(https?:\/\/\S+)/ig, '$1<$2>');
|
||||
}
|
||||
|
||||
function getSelfUrl(path) {
|
||||
if (config.url) {
|
||||
return Promise.resolve(`${config.url}/${path}`);
|
||||
} else {
|
||||
return publicIp.v4().then(ip => {
|
||||
return `http://${ip}:${logServerPort}/${path}`;
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
getModmailGuild,
|
||||
formatAttachment,
|
||||
formatUserDM,
|
||||
getUserMention,
|
||||
getTimestamp,
|
||||
disableLinkPreviews,
|
||||
getSelfUrl,
|
||||
};
|
||||
|
|
@ -0,0 +1,67 @@
|
|||
const http = require('http');
|
||||
const mime = require('mime');
|
||||
const config = require('../config');
|
||||
|
||||
const port = config.port || 8890;
|
||||
|
||||
function serveLogs(res, pathParts) {
|
||||
const token = pathParts[pathParts.length - 1];
|
||||
if (token.match(/^[0-9a-f]+$/) === null) return res.end();
|
||||
|
||||
findLogFile(token).then(logfile => {
|
||||
if (logfile === null) return res.end();
|
||||
|
||||
fs.readFile(getLogFilePath(logfile), {encoding: 'utf8'}, (err, data) => {
|
||||
if (err) {
|
||||
res.statusCode = 404;
|
||||
res.end('Log not found');
|
||||
return;
|
||||
}
|
||||
|
||||
res.setHeader('Content-Type', 'text/plain; charset=UTF-8');
|
||||
res.end(data);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function serveAttachments(res, pathParts) {
|
||||
const desiredFilename = pathParts[pathParts.length - 1];
|
||||
const id = pathParts[pathParts.length - 2];
|
||||
|
||||
if (id.match(/^[0-9]+$/) === null) return res.end();
|
||||
if (desiredFilename.match(/^[0-9a-z\._-]+$/i) === null) return res.end();
|
||||
|
||||
const attachmentPath = getAttachmentPath(id);
|
||||
fs.access(attachmentPath, (err) => {
|
||||
if (err) {
|
||||
res.statusCode = 404;
|
||||
res.end('Attachment not found');
|
||||
return;
|
||||
}
|
||||
|
||||
const filenameParts = desiredFilename.split('.');
|
||||
const ext = (filenameParts.length > 1 ? filenameParts[filenameParts.length - 1] : 'bin');
|
||||
const fileMime = mime.lookup(ext);
|
||||
|
||||
res.setHeader('Content-Type', fileMime);
|
||||
|
||||
const read = fs.createReadStream(attachmentPath);
|
||||
read.pipe(res);
|
||||
})
|
||||
}
|
||||
|
||||
function run() {
|
||||
const server = http.createServer((req, res) => {
|
||||
const parsedUrl = url.parse(`http://${req.url}`);
|
||||
const pathParts = parsedUrl.path.split('/').filter(v => v !== '');
|
||||
|
||||
if (parsedUrl.path.startsWith('/logs/')) serveLogs(res, pathParts);
|
||||
if (parsedUrl.path.startsWith('/attachments/')) serveAttachments(res, pathParts);
|
||||
});
|
||||
|
||||
server.listen(logServerPort);
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
run,
|
||||
};
|
||||
Loading…
Reference in New Issue