@skybloxsystems/ticket-bot/Bot/node_modules/discord-html-transcripts/src/index.js

Main

const exportHtml = require('./exporthtml');
const discord    = require('discord.js');

/**
 * 
 * @param {discord.Collection<string, discord.Message> | discord.Message[]} messages 
 * @param {discord.Channel} channel
 * @param {Object} opts
 * @param {Boolean} opts.returnBuffer Whether to return a buffer of the html or a message attachment.
 * @param {String} opts.fileName The name of the file. [ONLY VALID WITH `returnBuffer` SET TO FALSE]
 * @returns {discord.MessageAttachment | Buffer} 
 */
module.exports.generateFromMessages = (messages, channel, opts={ returnBuffer: false }) => {
    if(!opts.returnBuffer) opts.returnBuffer = false;
    if(!opts.fileName) opts.fileName = 'transcript.html';
   
    // check if is array
    if(!Array.isArray(messages)) {
        throw new Error('Provided messages must be either an array or a collection of Messages.');
    }

    if(messages.length === 0) {
        return exportHtml(messages, opts);
    }

    // check if array contains discord messages
    if(!(messages[0] instanceof discord.Message)) {
        throw new Error('Provided messages does not contain valid Messages.');
    }

    return exportHtml(messages, channel, opts);
}

/**
 * 
 * @param {discord.Channel} channel 
 * @param {Object} opts
 * @param {Number} opts.limit Max amount of messages to fetch. Set to -1 to disable.
 * @param {Boolean} opts.returnBuffer Whether to return a buffer of the html or a message attachment.
 * @param {String} opts.fileName The name of the file. [ONLY VALID WITH `returnBuffer` SET TO FALSE]
 */
module.exports.createTranscript = async (channel, opts={ limit: -1 }) => {
    if(!opts.returnBuffer) opts.returnBuffer = false;
    if(!opts.fileName) opts.fileName = 'transcript.html';
    if(!opts.limit) opts.limit = -1;

    if(!(channel.isText)) {
        throw new Error('Provided channel must be a valid channel.');
    }

    if(!channel.isText()) {
        throw new Error('Provided channel must be a text channel.');
    }

    const sum_messages = [];
    var last_id;

    while (true) {
        const messages = await channel.messages.fetch({ limit: 100, before: last_id });
        sum_messages.push(...messages.values());
        last_id = messages.last().id;
    
        if (messages.size != 100 || (opts.limit > 0 && sum_messages.length >= opts.limit)) {
            break;
        }
    }

    return exportHtml(sum_messages, channel, opts)
}