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

55 lines
1.4 KiB
Lua
Raw Normal View History

local imdb = {}
local HTTP = require('socket.http')
local URL = require('socket.url')
local JSON = require('dkjson')
local utilities = require('utilities')
imdb.command = 'imdb <query>'
function imdb:init(config)
imdb.triggers = utilities.triggers(self.info.username, config.cmd_pat):t('imdb', true).table
imdb.doc = [[```
]]..config.cmd_pat..[[imdb <query>
Returns an IMDb entry.
```]]
end
2015-07-03 00:15:52 +02:00
2016-05-27 02:26:30 +02:00
function imdb:action(msg, config)
2015-07-03 00:15:52 +02:00
local input = utilities.input(msg.text)
2015-07-03 00:15:52 +02:00
if not input then
if msg.reply_to_message and msg.reply_to_message.text then
input = msg.reply_to_message.text
else
utilities.send_message(self, msg.chat.id, imdb.doc, true, msg.message_id, true)
return
end
2015-07-03 00:15:52 +02:00
end
2015-09-22 01:34:02 +02:00
local url = 'http://www.omdbapi.com/?t=' .. URL.escape(input)
2015-07-03 00:15:52 +02:00
local jstr, res = HTTP.request(url)
if res ~= 200 then
2016-05-27 02:26:30 +02:00
utilities.send_reply(self, msg, config.errors.connection)
return
2015-07-03 00:15:52 +02:00
end
local jdat = JSON.decode(jstr)
2015-07-03 00:15:52 +02:00
if jdat.Response ~= 'True' then
2016-05-27 02:26:30 +02:00
utilities.send_reply(self, msg, config.errors.results)
return
2015-07-03 00:15:52 +02:00
end
local output = '*' .. jdat.Title .. ' ('.. jdat.Year ..')*\n'
output = output .. jdat.imdbRating ..'/10 | '.. jdat.Runtime ..' | '.. jdat.Genre ..'\n'
output = output .. '_' .. jdat.Plot .. '_\n'
2016-04-29 06:36:35 +02:00
output = output .. '[Read more.](http://imdb.com/title/' .. jdat.imdbID .. ')'
2015-07-03 00:15:52 +02:00
utilities.send_message(self, msg.chat.id, output, true, nil, true)
2015-07-03 00:15:52 +02:00
end
return imdb