79 lines
2.0 KiB
Lua
79 lines
2.0 KiB
Lua
-- Based on a plugin by matthewhesketh.
|
|
|
|
local HTTP = require('socket.http')
|
|
local JSON = require('dkjson')
|
|
local bindings = require('otouto.bindings')
|
|
local utilities = require('otouto.utilities')
|
|
|
|
local starwars = {}
|
|
|
|
function starwars:init(config)
|
|
starwars.triggers = utilities.triggers(self.info.username, config.cmd_pat)
|
|
:t('starwars', true):t('sw', true).table
|
|
starwars.doc = config.cmd_pat .. [[starwars <query>
|
|
Returns the opening crawl from the specified Star Wars film.
|
|
Alias: ]] .. config.cmd_pat .. 'sw'
|
|
starwars.command = 'starwars <query>'
|
|
starwars.base_url = 'http://swapi.co/api/films/'
|
|
end
|
|
|
|
local films_by_number = {
|
|
['phantom menace'] = 4,
|
|
['attack of the clones'] = 5,
|
|
['revenge of the sith'] = 6,
|
|
['new hope'] = 1,
|
|
['empire strikes back'] = 2,
|
|
['return of the jedi'] = 3,
|
|
['force awakens'] = 7
|
|
}
|
|
|
|
local corrected_numbers = {
|
|
4,
|
|
5,
|
|
6,
|
|
1,
|
|
2,
|
|
3,
|
|
7
|
|
}
|
|
|
|
function starwars:action(msg, config)
|
|
local input = utilities.input_from_msg(msg)
|
|
if not input then
|
|
utilities.send_reply(self, msg, starwars.doc, true)
|
|
return
|
|
end
|
|
|
|
bindings.sendChatAction(self, { chat_id = msg.chat.id, action = 'typing' } )
|
|
|
|
local film
|
|
if tonumber(input) then
|
|
input = tonumber(input)
|
|
film = corrected_numbers[input] or input
|
|
else
|
|
for title, number in pairs(films_by_number) do
|
|
if string.match(input, title) then
|
|
film = number
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
if not film then
|
|
utilities.send_reply(self, msg, config.errors.results)
|
|
return
|
|
end
|
|
|
|
local url = starwars.base_url .. film
|
|
local jstr, code = HTTP.request(url)
|
|
if code ~= 200 then
|
|
utilities.send_reply(self, msg, config.errors.connection)
|
|
return
|
|
end
|
|
|
|
local output = '*' .. JSON.decode(jstr).opening_crawl .. '*'
|
|
utilities.send_message(self, msg.chat.id, output, true, nil, true)
|
|
end
|
|
|
|
return starwars
|