Merge remote-tracking branch 'upstream/master' into dev

otouto/bindings.lua

@@ -22,11 +22,11 @@
 
 local bindings = {}
 
-local HTTPS = require('ssl.https')
-HTTPS.timeout = 10
-local JSON = require('dkjson')
+local https = require('ssl.https')
+https.timeout = 10
+local json = require('dkjson')
 local ltn12 = require('ltn12')
-local MP_ENCODE = require('multipart-post').encode
+local mp_encode = require('multipart-post').encode
 
 function bindings.init(token)
     bindings.BASE_URL = 'https://api.telegram.org/bot' .. token .. '/'
@@ -65,8 +65,8 @@ function bindings.request(method, parameters, file)
 		parameters = {''}
 	end
 	local response = {}
-	local body, boundary = MP_ENCODE(parameters)
-	local success, code = HTTPS.request{
+	local body, boundary = mp_encode(parameters)
+	local success, code = https.request{
 		url = bindings.BASE_URL .. method,
 		method = 'POST',
 		headers = {
@@ -81,7 +81,7 @@ function bindings.request(method, parameters, file)
 		print(method .. ': Connection error. [' .. code  .. ']')
 		return false, false
 	else
-		local result = JSON.decode(data)
+		local result = json.decode(data)
 		if not result then
 			return false, false
 		elseif result.ok then

otouto/plugins/voice_to_text.lua

@@ -0,0 +1,65 @@
+local vtt = {}
+
+vtt.triggers = {
+  '/nil'
+}
+
+local apikey = cred_data.witai_apikey
+local headers = {
+  ["Content-Type"] = "audio/mpeg3",
+  Authorization = "Bearer "..apikey
+}
+
+function vtt:pre_process(msg, config)
+  if not msg.voice then return msg end -- Ignore
+  local mime_type = msg.voice.mime_type
+  if mime_type ~= 'audio/ogg' then return msg end -- We only want to transcript voice messages
+
+  local file_id = msg.voice.file_id
+  local file_size = msg.voice.file_size
+  if file_size > 19922944 then
+	print('File is over 20 MB - can\'t download :(')
+	return msg
+  end
+  
+  utilities.send_typing(msg.chat.id, 'typing')
+  -- Saving file to the Telegram Cloud
+  local request = bindings.request('getFile', {
+		file_id = file_id
+  } )
+
+  local download_url = 'https://api.telegram.org/file/bot'..config.bot_api_key..'/'..request.result.file_path
+  local ogg_file_name = file_id..'.ogg'
+  local ogg_file = download_to_file(download_url, ogg_file_name)
+  
+  -- Convert to MP3
+  run_command('ffmpeg -loglevel panic -i /tmp/'..ogg_file_name..' -ac 1 -y /tmp/'..file_id..'.mp3')
+  os.remove('/tmp/'..ogg_file_name)
+  
+  -- Voice-To-Text via wit.ai
+  local req_url = 'https://api.wit.ai/speech'
+  local response_body = {}
+  local ok, response_code = https.request{
+    url = req_url,	
+    method = "POST",
+    headers = headers,
+    source = ltn12.source.file(io.open("/tmp/"..file_id..".mp3")),
+    sink = ltn12.sink.table(response_body)
+  }
+  os.remove("/tmp/"..file_id..".mp3")
+
+  if response_code ~= 200 then
+	return msg
+  end
+  vardump(response_body)
+  local out = json.decode(table.concat(response_body))
+  vardump(out)
+  utilities.send_reply(msg, out._text)
+    
+  return msg
+end
+
+function vtt:action(msg)
+end
+
+return vtt