Mikubot-2/otouto/plugins/translate.lua

local translate = {}

local HTTPS = require('ssl.https')
local URL = require('socket.url')
local JSON = require('dkjson')
local utilities = require('otouto.utilities')

translate.command = 'translate [text]'

function translate:init(config)
    assert(config.yandex_key,
        'translate.lua requires a Yandex translate API key from http://tech.yandex.com/keys/get.'
    )

    translate.triggers = utilities.triggers(self.info.username, config.cmd_pat)
        :t('translate', true):t('tl', true).table
    translate.doc = config.cmd_pat .. [[translate [text]
Translates input or the replied-to message into the bot's language.]]
    translate.base_url = 'https://translate.yandex.net/api/v1.5/tr.json/translate?key=' .. config.yandex_key .. '&lang=' .. config.lang .. '&text=%s'
end

function translate:action(msg, config)
    local input = utilities.input_from_msg(msg)
    if not input then
        utilities.send_reply(msg, translate.doc, true)
        return
    end

    local url = translate.base_url:format(URL.escape(input))
    local jstr, code = HTTPS.request(url)
    if code ~= 200 then
        utilities.send_reply(msg, config.errors.connection)
        return
    end

    local data = JSON.decode(jstr)
    if data.code ~= 200 then
        utilities.send_reply(msg, config.errors.connection)
        return
    end

    utilities.send_reply(msg.reply_to_message or msg, utilities.style.enquote('Translation', data.text[1]), true)
end

return translate