mcbot/cmds/xep.lua
author Mikael Berthe <mikael@lilotux.net>
Sat, 24 Apr 2010 10:45:00 +0200
changeset 39 237af42156a1
parent 30 e5e20eed6c58
child 66 d9c00a9fe9d5
permissions -rw-r--r--
Add checks after shcmd() calls
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
3
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     1
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     2
require "libs.shcmd"
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     3
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     4
-- The caching idea is borrowed from MattJ's riddim code,
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     5
-- and the function parse_xeps() as well.
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     6
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     7
local xeplisturl = "curl http://xmpp.org/extensions/xeps.xml"
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     8
22
4b360fe207e6 Minor description update
Mikael Berthe <mikael@lilotux.net>
parents: 16
diff changeset
     9
local xep = { ["desc"] = "Search XEP database" }
16
064a50911e05 Update command infrastructure
Mikael Berthe <mikael@lilotux.net>
parents: 11
diff changeset
    10
3
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    11
local xeps = {}
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    12
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    13
-- parse_xep() from Matthew Wild (aka MattJ)
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    14
local function parse_xeps(t)
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    15
if not t or t == "" then return nil end
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    16
local currxep = {}
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    17
    for b in string.gmatch(t,"<xep>(.-)</xep>") do
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    18
        for k,v in string.gmatch(b,"<(%w+)>(.-)</%1>") do
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    19
            currxep[k] = v
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    20
        end
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    21
        xeps[currxep.number] = { }
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    22
        for k, v in pairs(currxep) do xeps[currxep.number][k] = v end
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    23
    end
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    24
    return true
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    25
end
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    26
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    27
local function fetch_xeps ()
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    28
    local now = os.time()
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    29
    if os.difftime(now, xeps_updated_at) < 43200 then return end
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    30
    -- Let's get some fresh data
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    31
    local xepdata = shcmd(xeplisturl)
39
237af42156a1 Add checks after shcmd() calls
Mikael Berthe <mikael@lilotux.net>
parents: 30
diff changeset
    32
    if xepdata then
237af42156a1 Add checks after shcmd() calls
Mikael Berthe <mikael@lilotux.net>
parents: 30
diff changeset
    33
        xeps_updated_at = now
237af42156a1 Add checks after shcmd() calls
Mikael Berthe <mikael@lilotux.net>
parents: 30
diff changeset
    34
        parse_xeps(xepdata)
237af42156a1 Add checks after shcmd() calls
Mikael Berthe <mikael@lilotux.net>
parents: 30
diff changeset
    35
    end
3
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    36
end
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    37
11
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    38
-- Deeply inspired by MattJ's riddim bot
16
064a50911e05 Update command infrastructure
Mikael Berthe <mikael@lilotux.net>
parents: 11
diff changeset
    39
function xep.cmd (xepstr)
11
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    40
    if (xepstr) then
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    41
        xepstr = xepstr:gsub("[%s%?%.%!]+$", "")
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    42
        local n = xepstr:upper():match("^XEP[%-%s]?(%d+)$")
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    43
        if n then xepstr = n end
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    44
    end
3
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    45
    -- Check that xepnum is a valid number
11
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    46
    xepnum = tonumber(xepstr)
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    47
    if xepnum and (xepnum <= 0 or xepnum > 9999) then
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    48
        return nil, "What XEP?"
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    49
    end
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    50
    if not xepnum and not xepstr:match("^[%w%s%-_]+$") then
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    51
        return nil, "What XEP?  Please use a simple keyword..."
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    52
    end
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    53
3
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    54
    -- Download XEP list
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    55
    fetch_xeps()
11
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    56
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    57
    -- #1 The argument is a XEP number
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    58
    if xepnum then
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    59
        xepnum = tostring(xepnum)
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    60
        -- Expand to full 4 char number
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    61
        xepnum = string.rep("0", 4-xepnum:len())..xepnum
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    62
        local xep = xeps[tostring(xepnum)]
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    63
        if not xep then return nil, "Sorry, XEP-"..xepnum.." not found" end
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    64
        return "XEP-"..xep.number.." is \""..xep.name.."\"\n"..
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    65
               -- xep.type..", "..xep.status..", last updated "..xep.updated.."\n"..
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    66
               "URL: <http://xmpp.org/extensions/xep-"..xep.number..".html>"
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    67
    end
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    68
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    69
    -- #2 The argument is a keyword
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    70
    local r = ""
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    71
    xepstr = xepstr:lower()
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    72
    for n, xep in pairs(xeps) do
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    73
        if xep.name:lower():find(xepstr) then
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    74
            r = r.."XEP-"..xep.number..": "..xep.name.."\n"
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    75
        end
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    76
    end
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    77
    r = r:gsub("\n$", "")
4993fc6b1108 Improve xep command
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    78
    if r ~= "" then return r; end
30
e5e20eed6c58 Fix typo
Mikael Berthe <mikael@lilotux.net>
parents: 22
diff changeset
    79
    return nil, "Sorry, no matching XEP found"
3
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    80
end
508108deee63 Add command xep
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    81
16
064a50911e05 Update command infrastructure
Mikael Berthe <mikael@lilotux.net>
parents: 11
diff changeset
    82
mcbot_register_command("xep", xep)