Mikubot-2/otouto/plugins/starwars-crawl.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