DiscordTickets/src/modules/archive.js

/**
 * 
 *  @name DiscordTickets
 *  @author eartharoid <contact@eartharoid.me>
 *  @license GNU-GPLv3
 * 
 */


const ChildLogger = require('leekslazylogger').ChildLogger;
const log = new ChildLogger();
const lineReader = require('line-reader');
const fs = require('fs');
const dtf = require('@eartharoid/dtf');
const config = require('../../user/' + require('../').config);
const fetch = require('node-fetch');

module.exports.add = (message) => {

	if(message.type !== 'DEFAULT') return;

	if (config.transcripts.text.enabled) { // text transcripts
		let path = `user/transcripts/text/${message.channel.id}.txt`,
			time = dtf('HH:mm:ss n_D MMM YY', message.createdAt),
			msg = message.cleanContent;
		message.attachments.each(a => msg += '\n' + a.url);
		let string = `[${time}] [${message.author.tag}] :> ${msg}`;
		fs.appendFileSync(path, string + '\n');
	}

	if (config.transcripts.web.enabled) { // web archives
		let raw = `user/transcripts/raw/${message.channel.id}.log`,
			json = `user/transcripts/raw/entities/${message.channel.id}.json`;

		let embeds = [];
		for (let embed in message.embeds)
			// embeds.push(message.embeds[embed]); // .toJSON()
			embeds[embed] = { ...message.embeds[embed] };

		// message
		fs.appendFileSync(raw, JSON.stringify({
			id: message.id,
			author: message.author.id,
			content: message.content, // do not use cleanContent!
			time: message.createdTimestamp,
			embeds: embeds,
			attachments: [...message.attachments.values()]
		}) + '\n');

		// channel entities
		if (!fs.existsSync(json))
			fs.writeFileSync(json, JSON.stringify({
				entities: {
					users: {},
					channels: {},
					roles: {}
				}
			})); // create new

		let data = JSON.parse(fs.readFileSync(json));

		if (!data.entities.users[message.author.id]) {
			data.entities.users[message.author.id] = {
				avatar: message.author.avatarURL(),
				username: message.author.username,
				discriminator: message.author.discriminator,
				displayName: message.member.displayName,
				color: message.member.displayColor,
				badge: message.author.bot ? 'bot' : null
			};
		}

		// mentions.users
		message.mentions.members.each(m => data.entities.users[m.id] = { // for mentions
			avatar: m.user.avatarURL(),
			username: m.user.username,
			discriminator: m.user.discriminator,
			displayName: m.user.displayName || m.user.username,
			color: m.displayColor === 0 ? null : m.displayColor,
			badge: m.user.bot ? 'bot' : null
		});

		message.mentions.channels.each(c => data.entities.channels[c.id] = { // for mentions only
			name: c.name
		});

		message.mentions.roles.each(r => data.entities.roles[r.id] = { // for mentions only
			name: r.name,
			color: r.color === 0 ? 7506394 : r.color
		});

		fs.writeFileSync(json, JSON.stringify(data));

	}
};

module.exports.export = (Ticket, channel) => new Promise((resolve, reject) => {	

	(async () => {
		let ticket = await Ticket.findOne({
			where: {
				channel: channel.id
			}
		});
		
		let raw = `user/transcripts/raw/${channel.id}.log`,
			json = `user/transcripts/raw/entities/${channel.id}.json`;

		if (!config.transcripts.web.enabled || !fs.existsSync(raw) || !fs.existsSync(json))
			return reject(false);
		
		let data = JSON.parse(fs.readFileSync(json));
	
		data.ticket = {
			id: ticket.id,
			name: channel.name,
			creator: ticket.creator,
			channel: channel.id,
			topic: channel.topic
		};

		data.messages = [];

		lineReader.eachLine(raw, line => {
			let message = JSON.parse(line);
			let index = data.messages.findIndex(m => m.id === message.id);
			if (index === -1)
				data.messages.push(message);
			else
				data.messages[index] = message;	
		}, () => {
			let endpoint = config.transcripts.web.server;
			if (endpoint[endpoint.length - 1] === '/')
				endpoint = endpoint.slice(0, -1);
			endpoint += `/${data.ticket.creator}/${data.ticket.channel}/?key=${process.env.ARCHIVES_KEY}`;
			fetch(endpoint, {
				method: 'post',
				body:    JSON.stringify(data),
				headers: { 'Content-Type': 'application/json' },
			})
				.then(res => res.json())
				.then(res => {
					if (res.status !== 200) {
						log.warn(res);
						return resolve(new Error(`${res.status} (${res.message})`));
					}

					log.success(`Uploaded ticket #${ticket.id} archive to server`);

					fs.unlinkSync(raw);
					fs.unlinkSync(json);

					resolve(res.url);
				});
		});	
	
	})();

});