46 lines
1.5 KiB
Lua
46 lines
1.5 KiB
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(self, 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(self, msg, config.errors.connection)
|
|
return
|
|
end
|
|
|
|
local data = JSON.decode(jstr)
|
|
if data.code ~= 200 then
|
|
utilities.send_reply(self, msg, config.errors.connection)
|
|
return
|
|
end
|
|
|
|
utilities.send_reply(self, msg.reply_to_message or msg, utilities.style.enquote('Translation', data.text[1]), true)
|
|
end
|
|
|
|
return translate
|