mcbot/cmds/mcabber_bts.lua
author Mikael Berthe <mikael@lilotux.net>
Sat, 24 Apr 2010 10:45:00 +0200
changeset 39 237af42156a1
parent 20 d6c602aaa231
child 45 d333cb35e79a
permissions -rw-r--r--
Add checks after shcmd() calls
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
6
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     1
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     2
require "libs.shcmd"
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     3
20
d6c602aaa231 Use aliases
Mikael Berthe <mikael@lilotux.net>
parents: 16
diff changeset
     4
local mcabber_bts = { ["desc"] = "Query mcabber (crew) BTS",
d6c602aaa231 Use aliases
Mikael Berthe <mikael@lilotux.net>
parents: 16
diff changeset
     5
                      ["aliases"] = { "bug", "issue" }
d6c602aaa231 Use aliases
Mikael Berthe <mikael@lilotux.net>
parents: 16
diff changeset
     6
                    }
16
064a50911e05 Update command infrastructure
Mikael Berthe <mikael@lilotux.net>
parents: 14
diff changeset
     7
6
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     8
local function get_issue_details (num)
8
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
     9
    local item
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    10
6
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    11
    num = tonumber(num);
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    12
    if not num then return nil end
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    13
    local issue = {}
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    14
    local btsurl = "http://bitbucket.org/McKael/mcabber-crew/issue/"
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    15
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    16
    issue.id = tostring(num)
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    17
    issue.url = btsurl..issue.id.."/"
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    18
    local data = shcmd("curl "..issue.url)
39
237af42156a1 Add checks after shcmd() calls
Mikael Berthe <mikael@lilotux.net>
parents: 20
diff changeset
    19
    if not data then return nil end
6
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    20
    local title = data:match("#"..issue.id.."%s*-%s*(.*)%s*&mdash;")
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    21
    if not title then return nil end
12
6428c88dc58a On more small update of mcabber_bts (&amp;)
Mikael Berthe <mikael@lilotux.net>
parents: 10
diff changeset
    22
    issue.title = title:gsub("&quot;", "\""):gsub("&amp;", "&")
14
5c3ef7c5ba6c Handle some more HTML tags in mcabber_bts
Mikael Berthe <mikael@lilotux.net>
parents: 13
diff changeset
    23
    issue.title = issue.title:gsub("&lt;", "<"):gsub("&gt;", ">")
13
4a7d06e8709f Fix previous commit :)
Mikael Berthe <mikael@lilotux.net>
parents: 12
diff changeset
    24
    issue.title = issue.title:gsub("&#39;", "'")
8
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    25
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    26
    -- Get Status
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    27
    item = data:match("Status:%s*</td>%s*<td [^>]*>(.-)</td>")
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    28
    if item then issue.status = item end
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    29
    -- Get Type
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    30
    item = data:match("Type:%s*</td>%s*<td [^>]*>(.-)</td>")
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    31
    if item then issue.type = item end
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    32
6
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    33
    return issue
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    34
end
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    35
16
064a50911e05 Update command infrastructure
Mikael Berthe <mikael@lilotux.net>
parents: 14
diff changeset
    36
function mcabber_bts.cmd (num)
6
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    37
    -- Check that xepnum is a valid number
7
c515e8218c87 Small optimization
Mikael Berthe <mikael@lilotux.net>
parents: 6
diff changeset
    38
    if num then num = num:gsub("[%s%?%.%!]+$", "") end
c515e8218c87 Small optimization
Mikael Berthe <mikael@lilotux.net>
parents: 6
diff changeset
    39
    num = tonumber(num)
6
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    40
    if not num or num < 1 then
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    41
        return nil, "Please provide an issue number"
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    42
    end
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    43
    local issue = get_issue_details(num)
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    44
    if not issue then
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    45
        return nil,"Sorry, couldn't find issue "..tostring(num)
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    46
    end
8
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    47
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    48
    local result = "Issue #"..issue.id.." -- "..issue.title.."\n"
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    49
    if issue.status then
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    50
        result = result .. "Status: " .. issue.status .. "\n"
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    51
    end
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    52
    if issue.type then
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    53
        result = result .. "Type: " .. issue.type .. "\n"
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    54
    end
db84d97cf13d Improve mcabber_bts (display status and type)
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    55
    return result .. "URL: <"..tostring(issue.url)..">"
6
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    56
end
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    57
7cf014d0d206 Add command bts/bug (mcabber_bts)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    58
mcbot_register_command("bts", mcabber_bts)