This repository has been archived on 2021-04-24. You can view files and clone it, but cannot push or open issues or pull requests.
Mikubot-2/otouto/plugins/translate.lua

52 lines
1.4 KiB
Lua
Raw Normal View History

local translate = {}
local HTTPS = require('ssl.https')
local URL = require('socket.url')
local JSON = require('dkjson')
local utilities = require('utilities')
translate.command = 'translate [text]'
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]
Translates input or the replied-to message into the bot's language.
```]]
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
local input = utilities.input(msg.text)
if not input then
if msg.reply_to_message and msg.reply_to_message.text then
input = msg.reply_to_message.text
else
utilities.send_message(self, msg.chat.id, translate.doc, true, msg.message_id, true)
return
end
2015-07-21 01:29:40 +02:00
end
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)
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)
return
2015-07-21 01:29:40 +02:00
end
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-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
utilities.send_reply(self, msg.reply_to_message or msg, output, true)
2015-07-21 01:29:40 +02:00
end
return translate