2016-04-11 06:04:47 +02:00
|
|
|
local translate = {}
|
|
|
|
|
|
|
|
local HTTPS = require('ssl.https')
|
|
|
|
local URL = require('socket.url')
|
2016-04-15 21:07:23 +02:00
|
|
|
local JSON = require('dkjson')
|
2016-06-07 06:31:34 +02:00
|
|
|
local utilities = require('otouto.utilities')
|
2016-04-11 06:04:47 +02:00
|
|
|
|
|
|
|
translate.command = 'translate [text]'
|
2016-05-27 05:28:44 +02:00
|
|
|
|
|
|
|
function translate:init(config)
|
|
|
|
translate.triggers = utilities.triggers(self.info.username, config.cmd_pat):t('translate', true):t('tl', true).table
|
|
|
|
translate.doc = [[```
|
|
|
|
]]..config.cmd_pat..[[translate [text]
|
2016-01-08 14:44:37 +01:00
|
|
|
Translates input or the replied-to message into the bot's language.
|
|
|
|
```]]
|
2016-04-11 06:04:47 +02:00
|
|
|
end
|
2015-07-21 01:29:40 +02:00
|
|
|
|
2016-05-27 02:26:30 +02:00
|
|
|
function translate:action(msg, config)
|
2015-07-21 01:29:40 +02:00
|
|
|
|
2016-04-08 23:12:02 +02:00
|
|
|
local input = utilities.input(msg.text)
|
2015-11-25 03:22:04 +01:00
|
|
|
if not input then
|
|
|
|
if msg.reply_to_message and msg.reply_to_message.text then
|
|
|
|
input = msg.reply_to_message.text
|
|
|
|
else
|
2016-05-29 19:08:39 +02:00
|
|
|
utilities.send_message(self, msg.chat.id, translate.doc, true, msg.message_id, true)
|
2015-11-25 03:22:04 +01:00
|
|
|
return
|
|
|
|
end
|
2015-07-21 01:29:40 +02:00
|
|
|
end
|
2016-02-21 06:21:48 +01:00
|
|
|
|
2016-05-27 02:26:30 +02:00
|
|
|
local url = 'https://translate.yandex.net/api/v1.5/tr.json/translate?key=' .. config.yandex_key .. '&lang=' .. config.lang .. '&text=' .. URL.escape(input)
|
2016-02-21 06:21:48 +01:00
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
local str, res = HTTPS.request(url)
|
2015-07-21 01:29:40 +02:00
|
|
|
if res ~= 200 then
|
2016-05-27 02:26:30 +02:00
|
|
|
utilities.send_reply(self, msg, config.errors.connection)
|
2015-11-25 03:22:04 +01:00
|
|
|
return
|
2015-07-21 01:29:40 +02:00
|
|
|
end
|
2016-02-21 06:21:48 +01:00
|
|
|
|
2016-02-13 20:37:59 +01:00
|
|
|
local jdat = JSON.decode(str)
|
|
|
|
if jdat.code ~= 200 then
|
2016-05-27 02:26:30 +02:00
|
|
|
utilities.send_reply(self, msg, config.errors.connection)
|
2016-02-13 20:37:59 +01:00
|
|
|
return
|
|
|
|
end
|
2016-02-21 06:21:48 +01:00
|
|
|
|
2016-02-13 20:37:59 +01:00
|
|
|
local output = jdat.text[1]
|
2016-04-29 06:12:04 +02:00
|
|
|
output = '*Translation:*\n"' .. utilities.md_escape(output) .. '"'
|
2015-07-21 01:29:40 +02:00
|
|
|
|
2016-05-29 19:08:39 +02:00
|
|
|
utilities.send_reply(self, msg.reply_to_message or msg, output, true)
|
2015-07-21 01:29:40 +02:00
|
|
|
|
|
|
|
end
|
|
|
|
|
2016-04-11 06:04:47 +02:00
|
|
|
return translate
|