'use strict'; const fs = require('fs'); const path = require('path'); const stream = require('stream'); const fetch = require('node-fetch'); const { Error: DiscordError, TypeError } = require('../errors'); const { browser } = require('../util/Constants'); const Util = require('../util/Util'); /** * The DataResolver identifies different objects and tries to resolve a specific piece of information from them. * @private */ class DataResolver { constructor() { throw new Error(`The ${this.constructor.name} class may not be instantiated.`); } /** * Data that can be resolved to give an invite code. This can be: * * An invite code * * An invite URL * @typedef {string} InviteResolvable */ /** * Data that can be resolved to give an template code. This can be: * * A template code * * A template URL * @typedef {string} GuildTemplateResolvable */ /** * Resolves the string to a code based on the passed regex. * @param {string} data The string to resolve * @param {RegExp} regex The RegExp used to extract the code * @returns {string} */ static resolveCode(data, regex) { const match = regex.exec(data); return match ? match[1] || data : data; } /** * Resolves InviteResolvable to an invite code. * @param {InviteResolvable} data The invite resolvable to resolve * @returns {string} */ static resolveInviteCode(data) { return this.resolveCode(data, /discord(?:(?:app)?\.com\/invite|\.gg(?:\/invite)?)\/([\w-]{2,255})/i); } /** * Resolves GuildTemplateResolvable to a template code. * @param {GuildTemplateResolvable} data The template resolvable to resolve * @returns {string} */ static resolveGuildTemplateCode(data) { return this.resolveCode(data, /discord(?:app)?\.(?:com\/template|new)\/([\w-]{2,255})/i); } /** * Resolves a Base64Resolvable, a string, or a BufferResolvable to a Base 64 image. * @param {BufferResolvable|Base64Resolvable} image The image to be resolved * @returns {Promise} */ static async resolveImage(image) { if (!image) return null; if (typeof image === 'string' && image.startsWith('data:')) { return image; } const file = await this.resolveFileAsBuffer(image); return DataResolver.resolveBase64(file); } /** * Data that resolves to give a Base64 string, typically for image uploading. This can be: * * A Buffer * * A base64 string * @typedef {Buffer|string} Base64Resolvable */ /** * Resolves a Base64Resolvable to a Base 64 image. * @param {Base64Resolvable} data The base 64 resolvable you want to resolve * @returns {?string} */ static resolveBase64(data) { if (Buffer.isBuffer(data)) return `data:image/jpg;base64,${data.toString('base64')}`; return data; } /** * Data that can be resolved to give a Buffer. This can be: * * A Buffer * * The path to a local file * * A URL * @typedef {string|Buffer} BufferResolvable */ /** * @external Stream * @see {@link https://nodejs.org/api/stream.html} */ /** * Resolves a BufferResolvable to a Buffer or a Stream. * @param {BufferResolvable|Stream} resource The buffer or stream resolvable to resolve * @returns {Promise} */ static async resolveFile(resource) { if (!browser && Buffer.isBuffer(resource)) return resource; if (browser && resource instanceof ArrayBuffer) return Util.convertToBuffer(resource); // eslint-disable-next-line no-undef if (browser && resource instanceof Blob) return resource; if (resource instanceof stream.Readable) return resource; if (typeof resource === 'string') { if (/^https?:\/\//.test(resource)) { const res = await fetch(resource); return browser ? res.blob() : res.body; } else if (!browser) { return new Promise((resolve, reject) => { const file = path.resolve(resource); fs.stat(file, (err, stats) => { if (err) return reject(err); if (!stats.isFile()) return reject(new DiscordError('FILE_NOT_FOUND', file)); return resolve(fs.createReadStream(file)); }); }); } } throw new TypeError('REQ_RESOURCE_TYPE'); } /** * Resolves a BufferResolvable to a Buffer. * @param {BufferResolvable|Stream} resource The buffer or stream resolvable to resolve * @returns {Promise} */ static async resolveFileAsBuffer(resource) { const file = await this.resolveFile(resource); if (Buffer.isBuffer(file)) return file; const buffers = []; for await (const data of file) buffers.push(data); return Buffer.concat(buffers); } } module.exports = DataResolver;