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/miku/plugins/translate.lua

142 lines
4.5 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('miku.utilities')
local mime = require("mime")
require("./miku/plugins/pasteee")
translate.command = 'translate [Text]'
function translate:init(config)
translate.triggers = {
"^/translate ([%w]+),([%a]+) (.+)",
"^/translate (to%:)([%w]+) (.+)",
"^/translate (.+)",
"^/getlanguages$",
"^/(whatlang) (.+)"
}
translate.doc = [[*
]]..config.cmd_pat..[[translate* _[Text]_: Übersetze Text in deutsch
*]]..config.cmd_pat..[[translate* to:Zielsprache _[Text]_: Übersetze Text in Zielsprache
*]]..config.cmd_pat..[[translate* Quellsprache,Zielsprache _[Text]_: Übersetze Text von beliebiger Sprache in beliebige Sprache
*]]..config.cmd_pat..[[getlanguages*: Postet alle verfügbaren Sprachcodes
*]]..config.cmd_pat..[[whatlang* _[Text]_: Gibt erkannte Sprache zurück]]
end
local bing_key = cred_data.bing_key
local accountkey = mime.b64(bing_key..':'..bing_key)
function translate:translate(source_lang, target_lang, text)
if not target_lang then target_lang = 'de' end
local url = 'https://api.datamarket.azure.com/Bing/MicrosoftTranslator/Translate?$format=json&Text=%27'..URL.escape(text)..'%27&To=%27'..target_lang..'%27&From=%27'..source_lang..'%27'
local response_body = {}
local request_constructor = {
url = url,
method = "GET",
sink = ltn12.sink.table(response_body),
headers = {
Authorization = "Basic "..accountkey
}
}
local ok, response_code, response_headers, response_status_line = https.request(request_constructor)
if not ok or response_code ~= 200 then return 'Ein Fehler ist aufgetreten.' end
local trans = json.decode(table.concat(response_body)).d.results[1].Text
return trans
end
function translate:detect_language(text)
local url = 'https://api.datamarket.azure.com/Bing/MicrosoftTranslator/Detect?$format=json&Text=%27'..URL.escape(text)..'%27'
local response_body = {}
local request_constructor = {
url = url,
method = "GET",
sink = ltn12.sink.table(response_body),
headers = {
Authorization = "Basic "..accountkey
}
}
local ok, response_code, response_headers, response_status_line = https.request(request_constructor)
if not ok or response_code ~= 200 then return 'en' end
local language = json.decode(table.concat(response_body)).d.results[1].Code
print('Erkannte Sprache: '..language)
return language
end
function translate:get_all_languages()
local url = 'https://api.datamarket.azure.com/Bing/MicrosoftTranslator/GetLanguagesForTranslation?$format=json'
local response_body = {}
local request_constructor = {
url = url,
method = "GET",
sink = ltn12.sink.table(response_body),
headers = {
Authorization = "Basic "..accountkey
}
}
local ok, response_code, response_headers, response_status_line = https.request(request_constructor)
if not ok or response_code ~= 200 then return 'Ein Fehler ist aufgetreten.' end
local lang_table = json.decode(table.concat(response_body)).d.results
local languages = ""
for i in pairs(lang_table) do
languages = languages..lang_table[i].Code..'\n'
end
local link = upload(languages)
return '[Sprachliste auf Paste.ee ansehen]('..link..')'
end
function translate:action(msg, config, matches)
utilities.send_typing(self, msg.chat.id, 'typing')
if matches[1] == '/getlanguages' then
utilities.send_reply(self, msg, translate:get_all_languages(), true)
return
end
if matches[1] == 'whatlang' and matches[2] then
local text = matches[2]
local lang = translate:detect_language(text)
utilities.send_reply(self, msg, 'Erkannte Sprache: '..lang, true)
return
end
-- Third pattern
if #matches == 1 then
print("First")
local text = matches[1]
local language = translate:detect_language(text)
utilities.send_reply(self, msg, translate:translate(language, nil, text))
return
end
-- Second pattern
if #matches == 3 and matches[1] == "to:" then
print("Second")
local target = matches[2]
local text = matches[3]
local language = translate:detect_language(text)
utilities.send_reply(self, msg, translate:translate(language, target, text))
return
end
-- First pattern
if #matches == 3 then
print("Third")
local source = matches[1]
local target = matches[2]
local text = matches[3]
utilities.send_reply(self, msg, translate:translate(source, target, text))
return
end
end
return translate