xiao/commands/analyze/ocr.js

const Command = require('../../structures/Command');
const { createWorker } = require('tesseract.js');
const { reactIfAble } = require('../../util/Util');
const { LOADING_EMOJI_ID, SUCCESS_EMOJI_ID } = process.env;

module.exports = class OcrCommand extends Command {
	constructor(client) {
		super(client, {
			name: 'ocr',
			aliases: ['tesseract', 'optical-character-recognition'],
			group: 'analyze',
			memberName: 'ocr',
			description: 'Performs Optical Character Recognition on an image.',
			throttling: {
				usages: 1,
				duration: 60
			},
			args: [
				{
					key: 'image',
					prompt: 'What image would you like to perform OCR on?',
					type: 'image',
					default: msg => msg.author.displayAvatarURL({ format: 'png', size: 2048 })
				}
			]
		});
	}

	async run(msg, { image }) {
		await reactIfAble(msg, this.client.user, LOADING_EMOJI_ID, '💬');
		const worker = createWorker();
		await worker.load();
		await worker.loadLanguage('eng');
		await worker.initialize('eng');
		const { data: { text } } = await worker.recognize(image);
		await worker.terminate();
		await reactIfAble(msg, this.client.user, SUCCESS_EMOJI_ID, '✅');
		if (text.length > 2000) {
			return msg.reply('The result was over 2000 characters, so here\'s a TXT file.', {
				files: [{ attachment: Buffer.from(text), name: 'ocr.txt' }]
			});
		}
		return msg.reply(text);
	}
};