mirror of
https://gitlab.com/bignutty/labscore.git
synced 2025-06-07 21:53:07 -04:00
72 lines
No EOL
3.4 KiB
JavaScript
72 lines
No EOL
3.4 KiB
JavaScript
const { googleVisionOcr, googleTranslate } = require("../../../labscore/api");
|
||
const { TRANSLATE_LANGUAGES, TRANSLATE_LANGUAGE_MAPPINGS } = require("../../../labscore/constants");
|
||
const { getRecentImage } = require("../../../labscore/utils/attachment");
|
||
const { createEmbed } = require("../../../labscore/utils/embed");
|
||
const { codeblock, icon, pill } = require("../../../labscore/utils/markdown");
|
||
const { editOrReply } = require("../../../labscore/utils/message");
|
||
const { STATICS } = require("../../../labscore/utils/statics");
|
||
const { isSupported, getCodeFromAny } = require("../../../labscore/utils/translate");
|
||
|
||
const { Permissions } = require("detritus-client/lib/constants");
|
||
module.exports = {
|
||
name: 'ocrtr',
|
||
label: 'to',
|
||
aliases: ["octr"],
|
||
metadata: {
|
||
description: 'Uses Optical Character Recognition to translate text in an image.',
|
||
description_short: 'Image text recognition + translation.',
|
||
examples: ['ocrtr en'],
|
||
category: 'utils',
|
||
usage: 'ocrtr <language> [-from <origin language>]'
|
||
},
|
||
args: [
|
||
{name: 'from', default: 'auto', type: 'string', help: "Language to translate from"}
|
||
],
|
||
permissionsClient: [Permissions.EMBED_LINKS, Permissions.SEND_MESSAGES, Permissions.READ_MESSAGE_HISTORY, Permissions.USE_EXTERNAL_EMOJIS],
|
||
run: async (context, args) => {
|
||
context.triggerTyping();
|
||
|
||
if(!args.to) args.to = "en"
|
||
if(args.to.startsWith("-to")) args.to = args.to.replace("-to ", "")
|
||
|
||
if(!isSupported(args.to)) return editOrReply(context, createEmbed("warning", context, "Invalid language (to)."))
|
||
if(!isSupported(args.from)) return editOrReply(context, createEmbed("warning", context, "Invalid language (from)."))
|
||
|
||
args.to = getCodeFromAny(args.to)
|
||
args.from = getCodeFromAny(args.from)
|
||
|
||
let image = await getRecentImage(context, 50)
|
||
if (!image) return editOrReply(context, { embeds: [createEmbed("warning", context, "No images found.")] })
|
||
|
||
let ocr;
|
||
try{
|
||
ocr = await googleVisionOcr(context, image)
|
||
}catch(e){
|
||
return editOrReply(context, { embeds: [createEmbed("error", context, "Unable to retrieve Google Vision API response.")] })
|
||
}
|
||
|
||
if(ocr.response.body.status == 1) return editOrReply(context, { embeds: [createEmbed("warning", context, ocr.response.body.text)] })
|
||
|
||
try{
|
||
let translate = await googleTranslate(context, ocr.response.body.text, args.to, args.from)
|
||
|
||
let fromFlag = TRANSLATE_LANGUAGE_MAPPINGS[translate.response.body.language.from || args.from] || ''
|
||
let toFlag = TRANSLATE_LANGUAGE_MAPPINGS[translate.response.body.language.to] || ''
|
||
|
||
return editOrReply(context, createEmbed("default", context, {
|
||
description: `${icon("locale")} ${fromFlag} ${pill(TRANSLATE_LANGUAGES[translate.response.body.language.from || args.from])} ${icon("arrow_right")} ${toFlag} ${pill(TRANSLATE_LANGUAGES[translate.response.body.language.to])}\n${codeblock("ansi", [translate.response.body.translation])}`,
|
||
thumbnail: {
|
||
url: image
|
||
},
|
||
footer: {
|
||
iconUrl: STATICS.google,
|
||
text: `Google Cloud Vision • ${context.application.name}`
|
||
}
|
||
}))
|
||
}catch(e){
|
||
console.log(e)
|
||
if(e.response?.body?.status && e.response.body.status == 2) return editOrReply(context, {embeds:[createEmbed("error", context, `Unable to translate text.`)]})
|
||
return editOrReply(context, {embeds:[createEmbed("error", context, `Something went wrong.`)]})
|
||
}
|
||
}
|
||
}; |