9f760114bd
All help messages and many other things moved from markdown to html. Eventually, I'd like only things made from user input to use markdown. cats.lua, rmspic.lua, and dilbert.lua moved to sendPhoto with URL. xkcd.lua and apod.lua not moved to retain formatting. Probably a lot of other stuff that I forget about. I should commit more often.
55 lines
1.6 KiB
Lua
55 lines
1.6 KiB
Lua
-- Thanks to @TiagoDanin for writing the original plugin.
|
|
|
|
local youtube = {}
|
|
|
|
local HTTPS = require('ssl.https')
|
|
local URL = require('socket.url')
|
|
local JSON = require('dkjson')
|
|
local utilities = require('otouto.utilities')
|
|
|
|
function youtube:init(config)
|
|
assert(config.google_api_key,
|
|
'youtube.lua requires a Google API key from http://console.developers.google.com.'
|
|
)
|
|
|
|
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>'
|
|
|
|
function youtube:action(msg, config)
|
|
|
|
local input = utilities.input_from_msg(msg)
|
|
if not input then
|
|
utilities.send_reply(msg, youtube.doc, 'html')
|
|
return
|
|
end
|
|
|
|
local url = 'https://www.googleapis.com/youtube/v3/search?key=' .. config.google_api_key .. '&type=video&part=snippet&maxResults=4&q=' .. URL.escape(input)
|
|
|
|
local jstr, res = HTTPS.request(url)
|
|
if res ~= 200 then
|
|
utilities.send_reply(msg, config.errors.connection)
|
|
return
|
|
end
|
|
|
|
local jdat = JSON.decode(jstr)
|
|
if jdat.pageInfo.totalResults == 0 then
|
|
utilities.send_reply(msg, config.errors.results)
|
|
return
|
|
end
|
|
|
|
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
|