2015-11-25 03:22:04 +01:00
|
|
|
-- Thanks to @TiagoDanin for writing the original plugin.
|
2015-09-23 20:12:06 +02:00
|
|
|
|
2016-04-11 06:04:47 +02:00
|
|
|
local youtube = {}
|
|
|
|
|
|
|
|
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
|
|
|
|
2016-05-27 02:26:30 +02:00
|
|
|
function youtube:init(config)
|
2016-08-14 04:46:18 +02:00
|
|
|
assert(config.google_api_key,
|
|
|
|
'youtube.lua requires a Google API key from http://console.developers.google.com.'
|
|
|
|
)
|
2016-04-12 05:55:46 +02:00
|
|
|
|
2016-08-14 04:46:18 +02:00
|
|
|
youtube.triggers = utilities.triggers(self.info.username, config.cmd_pat):t('youtube', true):t('yt', true).table
|
|
|
|
youtube.doc = config.cmd_pat .. [[youtube <query>
|
2016-05-27 05:28:44 +02:00
|
|
|
Returns the top result from YouTube.
|
2016-07-25 11:03:35 +02:00
|
|
|
Alias: ]] .. config.cmd_pat .. 'yt'
|
2016-01-13 19:00:17 +01:00
|
|
|
end
|
|
|
|
|
2016-04-11 06:04:47 +02:00
|
|
|
youtube.command = 'youtube <query>'
|
2015-09-23 20:12:06 +02:00
|
|
|
|
2016-05-27 02:26:30 +02:00
|
|
|
function youtube:action(msg, config)
|
2015-11-25 03:22:04 +01:00
|
|
|
|
2016-08-14 04:46:18 +02:00
|
|
|
local input = utilities.input_from_msg(msg)
|
|
|
|
if not input then
|
2016-08-23 06:16:32 +02:00
|
|
|
utilities.send_reply(msg, youtube.doc, true)
|
2016-08-14 04:46:18 +02:00
|
|
|
return
|
|
|
|
end
|
2015-11-25 03:22:04 +01:00
|
|
|
|
2016-08-14 04:46:18 +02:00
|
|
|
local url = 'https://www.googleapis.com/youtube/v3/search?key=' .. config.google_api_key .. '&type=video&part=snippet&maxResults=4&q=' .. URL.escape(input)
|
2015-11-25 03:22:04 +01:00
|
|
|
|
2016-08-14 04:46:18 +02:00
|
|
|
local jstr, res = HTTPS.request(url)
|
|
|
|
if res ~= 200 then
|
2016-08-23 06:16:32 +02:00
|
|
|
utilities.send_reply(msg, config.errors.connection)
|
2016-08-14 04:46:18 +02:00
|
|
|
return
|
|
|
|
end
|
2015-11-25 03:22:04 +01:00
|
|
|
|
2016-08-14 04:46:18 +02:00
|
|
|
local jdat = JSON.decode(jstr)
|
|
|
|
if jdat.pageInfo.totalResults == 0 then
|
2016-08-23 06:16:32 +02:00
|
|
|
utilities.send_reply(msg, config.errors.results)
|
2016-08-14 04:46:18 +02:00
|
|
|
return
|
|
|
|
end
|
2015-11-25 03:22:04 +01:00
|
|
|
|
2016-08-14 04:46:18 +02:00
|
|
|
local vid_url = 'https://www.youtube.com/watch?v=' .. jdat.items[1].id.videoId
|
|
|
|
local vid_title = jdat.items[1].snippet.title
|
|
|
|
vid_title = vid_title:gsub('%(.+%)',''):gsub('%[.+%]','')
|
|
|
|
local output = '[' .. vid_title .. '](' .. vid_url .. ')'
|
2015-11-25 03:22:04 +01:00
|
|
|
|
2016-08-23 06:16:32 +02:00
|
|
|
utilities.send_message(msg.chat.id, output, false, nil, true)
|
2015-09-23 20:12:06 +02:00
|
|
|
|
|
|
|
end
|
|
|
|
|
2016-04-11 06:04:47 +02:00
|
|
|
return youtube
|