const { InfractionTargetTypes, InfractionDictionary, InfractionColors } = require('../../../util/Constants.js'); const { Util } = require('../../../util/'); const Constants = { MaxCharacters: 1024, // Max embed description is 2048 characters, however some of those description characters are going to usernames, types, filler text, etc. RemovedInfractions: ['BAN', 'SOFTBAN', 'KICK'] }; class Infraction { constructor(client, data = {}) { this.client = client; this.type = data.type || null; this.id = null; this.case = null; this.message = data.message; //NOT REQUIRED this.arguments = data.arguments || {}; this.guildId = data.guild ? data.guild.id : null; this.guild = data.guild || null; this.channelId = data.channel ? data.channel.id : null; this.channel = data.channel || null; this.messageId = data.message ? data.message.id : null; this.message = data.message || null; this.targetId = data.target ? data.target.id : null; this.target = data.target || []; this.executorId = data.executor ? data.executor.id : null; this.executor = data.executor || null; this.targetType = data.targetType && InfractionTargetTypes.includes(data.targetType) ? data.targetType : 'USER'; this.resolved = false; this.duration = isNaN(data.duration) ? null : data.duration; //How long the action will last. Must be in milliseconds. this.callback = isNaN(data.duration) ? null : Date.now() + data.duration * 1000; //At what epoch(?) time it will callback. this.reason = data.reason.length ? data.reason : 'N/A'; this.points = data.points || 0; this.totalPoints = 0; this.expiration = data.expiration || 0; this.data = data.data || {}; //Miscellaneous data that may need to be saved for future use. this.flags = data.arguments ? Object.keys(data.arguments) : []; this.hyperlink = data.hyperlink || null; // To overwrite hyperlink (if it's from a callback) this.modlogMessageId = null; this.dmlogMessageId = null; this.modlogId = null; this.changes = []; this.timestamp = Date.now(); this._callbacked = Boolean(data._callbacked); this._fetched = Boolean(data); } async handle() { const { moderationLog, dmInfraction } = await this.guild.settings(); //Increment CaseId, should only be called if making a new infraction. this.guild._settings.caseId++; this.case = this.guild._settings.caseId; await this.guild._updateSettings({ caseId: this.case }); /* Logging */ if(moderationLog.channel) { if(moderationLog.infractions.includes(this.type)) { this._moderationLog = await this.client.resolver.resolveChannel(moderationLog.channel, true, this.guild); if(!this._moderationLog) return undefined; this.modlogId = this._moderationLog.id; try { this._logMessage = await this._moderationLog.send('', { embed: this._embed() }); this.modlogMessageId = this._logMessage.id; } catch(e) { console.error(e); } //eslint-disable-line no-empty } else { this.client.logger.debug(`Did not log infraction ${this.type} because it is not in the infractions.`); } } if(dmInfraction.enabled) { if(this.targetType === 'USER') { let message = dmInfraction.messages[this.type] || dmInfraction.messages.default; if(!message) message = ''; message = message .replace(/\{(guild|server)\}/ugim, this.guild.name) .replace(/\{user\}/ugim, this.target.tag) .replace(/\{infraction\}/ugim, this.dictionary.past) .replace(/\{from\|on\}/ugim, Constants.RemovedInfractions.includes(this.type) ? 'from' : 'on'); //add more if you want i should probably add a better system for this... try { const logMessage = await this.target.send(message, { embed: this._embed(true) }); this.dmlogMessageId = logMessage.id; } catch(e) {} //eslint-disable-line no-empty } } if(this.duration) { await this.client.moderationManager._handleExpirations([this.json]); } /* LMAOOOO PLEASE DONT JUDGE ME */ if(this.data.roles && this.data.roles.length > 0) { this.data.roleIds = this.data.roles.map((r) => r.id); this.data.roleNames = this.data.roles.map((r) => r.name); delete this.data.roles; } return this.save(); } async save() { return this.client.storageManager.mongodb.infractions.insertOne(this.json) .catch((error) => { this.client.logger.error(`There was an issue saving infraction data to the database.\n${error.stack || error}`); }); } hyperlink(bool = false) { if(bool) return `https://discord.com/channels/${this.guildId}/${this.modlogId}/${this.modlogMessageId}`; return `https://discord.com/channels/${this.guildId}/${this.channelId}/${this.messageId}`; } _embed(dm) { let description = ""; description += `${this.guild.format('INFRACTION_DESCRIPTION', { type: this.dictionary.past.toUpperCase(), moderator: `${Util.escapeMarkdown(this.executor.tag)}`, reason: Util.escapeMarkdown(this.reason.length > Constants.MaxCharacters ? `${this.reason.substring(0, Constants.MaxCharacters-3)}...` : this.reason, { italic: false, underline: false, strikethrough: false }) })}`; if(this.duration) { description += `\n${this.guild.format('INFRACTION_DESCRIPTIONDURATION', { duration: Util.duration(this.duration) })}`; } if(this.points && this.points > 0) { //TODO: Add expiration to INFRACTION_DESCRIPTIONPOINTS description += `\n${this.guild.format('INFRACTION_DESCRIPTIONPOINTS', { points: this.points, total: this.pointsTotal, expires: Util.duration(this.pointsExpiration) })}`; } if(this.description && this.description instanceof Function) { description += this.description(dm); } if(!this.silent && (this.message || this.hyperlink) && (dm && !Constants.RemovedInfractions.includes(this.type))) { description += `\n${this.guild.format('INFRACTION_DESCRIPTIONJUMPTO', { name: this.hyperlink ? 'Case' : 'Message', link: this.hyperlink ? this.hyperlink : `https://discord.com/channels/${this.guildId}/${this.channelId}/${this.messageId}` })}`; } const blah = { author: { name: `${this.target.display} (${this.targetId})`, icon_url: this.target.displayAvatarURL() //eslint-disable-line camelcase }, timestamp: this.timestamp, color: this.color, footer: { text: `》 Case ${this.case}` }, description }; return blah; } get json() { return { id: `${this.guildId}:${this.case}`, guild: this.guildId, channel: this.channelId, channelName: this.channel.display, message: this.messageId, executor: this.executorId, executorTag: this.executor.display, target: this.targetId, targetTag: this.target.display, targetType: this.targetType, type: this.type, case: this.case, timestamp: this.timestamp, duration: this.duration, callback: this.callback, reason: this.reason, data: this.data, flags: this.flags, modLogMessage: this.modLogMessageId, dmLogMessage: this.dmlogMessageId, modLogChannel: this.modlogId, resolved: this.resolved, changes: this.changes, _callbacked: this._callbacked || false }; } get targetIcon() { return this.targetType === 'USER' ? this.target.displayAvatarURL() : this.guild.iconURL(); } get actions() { const actions = []; for(const argument of Object.values(this.arguments)) { if(['silent', 'prune', 'force'].includes(argument)) actions.push(argument); } return actions; } get _reason() { let str = `[${this.type}][targetId:${this.target.id}] Executed by ${this.executor.tag} (${this.executor.id}) because: ${this.reason}`; if(str.length > 512) str = `${this.reason.substring(0, 509)}...`; return str; } get dictionary() { return InfractionDictionary[this.type]; } get color() { return InfractionColors[this.type]; } //Super Functions _succeed() { return { error: false, infraction: this }; } _fail(message, fatal = false) { return { error: true, infraction: this, reason: message, fatal }; } verify() { return this._verify(); } async _verify() { const { protection } = await this.guild.settings(); if (this.executor.id === this.guild.ownerID) return this._succeed(); if (this.guild && protection.enabled && this.targetType === 'USER') { const executor = await this.guild.members.fetch(this.executor.id).catch(); const target = await this.guild.members.fetch(this.target.id).catch(); if (!target) return this._succeed(); if(protection.type === 'position') { const executorHighest = executor.roles.highest; const targetHighest = target.roles.highest; if (executorHighest.comparePositionTo(targetHighest) < 0) { return this._fail('INFRACTION_PROTECTIONPOSITIONERROR'); } } else if(protection.type === 'role') { const contains = target.roles.cache.some((r) => protection.roles.includes(r.id)); if(contains) { return this._fail('INFRACTION_PROTECTIONROLEERROR'); } } } return this._succeed(); } // async fetch() { //Data from Mongodb (id-based data) // const data = await this.client.storageManager.mongodb.infractions.findOne({ id: this.id }); // if(!data) { // this.client.logger.error(`Case ${this.id} is missing infraction data in database.`); // return null; // } // if(data.guild) { // let guild = null; // try { // guild = await this.client.guilds.fetch(data.guild); // } catch(error) { // this.client.logger.error(`Unable to fetch guild: ${data.guild}\n${error.stack || error}`); // guild = null; // } // if(!guild) return null; // if(data.targets) { // this.targetIds = data.targets; // for(const target of data.targets) { // const fetchedTarget = await this._fetchTarget(target); // if(fetchedTarget) this.targets.push(fetchedTarget); // } // } // if(data.executor) { // this.executorId = data.executor; // const fetchedExecutor = await this._fetchTarget(data.executor, 'USER'); // if(fetchedExecutor) this.executor = fetchedExecutor; // } // } // this.type = data.type; // this.timestamp = data.timestamp; // this.duration = data.duration; // this.reason = data.reason; // this.channelId = data.channel; // this.resolved = data.resolved; // this._callbacked = data._callbacked; // this.dictionary = InfractionDictionary[this.type]; // this.color = InfractionColors[this.type]; // this.modlogMessageId = data.modlogMessageId; // this.dmlogMessageId = data.dmlogMessageId; // this._fetched = Boolean(data._fetched); // return this; // } // async _fetchTarget(target, type = null) { // type = type || this.targetType; // let fetched = null; // if(type === 'CHANNEL') { // fetched = await this.client.resolver.resolveChannel(target, true, this.guild); // } else if (type) { // fetched = await this.client.resolver.resolveMember(target, true, this.guild); // if(!fetched) { // fetched = await this.client.resolver.resolveUser(target, true); // } // } // return fetched || null; // } } module.exports = Infraction;