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/youtube.lua

55 lines
1.6 KiB
Lua
Raw Normal View History

-- Thanks to @TiagoDanin for writing the original plugin.
local youtube = {}
local HTTPS = require('ssl.https')
local URL = require('socket.url')
local JSON = require('dkjson')
2016-06-07 06:31:34 +02:00
local utilities = require('otouto.utilities')
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>
Returns the top result from YouTube.
Alias: ]] .. config.cmd_pat .. 'yt'
end
youtube.command = 'youtube <query>'
2016-05-27 02:26:30 +02:00
function youtube:action(msg, config)
2016-08-14 04:46:18 +02:00
local input = utilities.input_from_msg(msg)
if not input then
utilities.send_reply(msg, youtube.doc, 'html')
2016-08-14 04:46:18 +02:00
return
end
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)
2016-08-14 04:46:18 +02:00
local jstr, res = HTTPS.request(url)
if res ~= 200 then
utilities.send_reply(msg, config.errors.connection)
2016-08-14 04:46:18 +02:00
return
end
2016-08-14 04:46:18 +02:00
local jdat = JSON.decode(jstr)
if jdat.pageInfo.totalResults == 0 then
utilities.send_reply(msg, config.errors.results)
2016-08-14 04:46:18 +02:00
return
end
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 .. ')'
utilities.send_message(msg.chat.id, output, false, nil, true)
end
return youtube