mcabbot.lua
author Mikael Berthe <mikael@lilotux.net>
Sun, 25 Nov 2012 19:29:32 +0100
changeset 65 2cefbe9f3ac6
parent 41 5c0a2db896f5
child 66 d9c00a9fe9d5
permissions -rw-r--r--
shortenurl: cosmetical change
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     1
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     2
local home = os.getenv("HOME")
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     3
package.path = package.path..
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     4
                ";"..home.."/.mcabber/lua/?.lua"..
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     5
                ";"..home.."/.mcabber/lua/mcbot/?.lua"
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     6
home = nil
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     7
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     8
require "mcbot.mcbot_engine"
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     9
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    10
local mcabbot = {}
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    11
mcabbot.nickname = "McBot"
40
95ba0699f365 Update bot's JID
Mikael Berthe <mikael@lilotux.net>
parents: 28
diff changeset
    12
mcabbot.jid = "mcbot@lilotux.net"
27
a166e9ab8065 Add command prefix (useful for MUC rooms)
Mikael Berthe <mikael@lilotux.net>
parents: 24
diff changeset
    13
mcabbot.cmdprefix = "%%"
0
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    14
24
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    15
local function hk_subscription (h)
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    16
    if h.type == "subscribe" then
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    17
        -- Let's authorize them...
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    18
        main.run("authorization allow "..h.jid)
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    19
        -- ... and subscribe to their own presence
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    20
        main.run("authorization request "..h.jid)
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    21
        return 1
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    22
    elseif h.type == "unsubscribe" or h.type == "unsubscribed" then
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    23
        main.run("authorization cancel "..h.jid)
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    24
    end
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    25
    return 0
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    26
end
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    27
1
cca972635e5e Provid more data to the bot functions
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    28
local function hk_message_in (h)
0
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    29
    local perso = false
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    30
    local muc = false
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    31
    local message = h.message
41
5c0a2db896f5 Ignore incoming error messages
Mikael Berthe <mikael@lilotux.net>
parents: 40
diff changeset
    32
5c0a2db896f5 Ignore incoming error messages
Mikael Berthe <mikael@lilotux.net>
parents: 40
diff changeset
    33
    if h.error and h.error == "true" then return end
5c0a2db896f5 Ignore incoming error messages
Mikael Berthe <mikael@lilotux.net>
parents: 40
diff changeset
    34
0
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    35
    if h.groupchat and h.groupchat == "true" then
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    36
        muc = true
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    37
    end
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    38
    if muc == true then
28
510849236c58 Fix command prefix usage
Mikael Berthe <mikael@lilotux.net>
parents: 27
diff changeset
    39
        if h.delayed == "" then
510849236c58 Fix command prefix usage
Mikael Berthe <mikael@lilotux.net>
parents: 27
diff changeset
    40
            if mcabbot.cmdprefix and message:starts(mcabbot.cmdprefix) then
510849236c58 Fix command prefix usage
Mikael Berthe <mikael@lilotux.net>
parents: 27
diff changeset
    41
                perso = true
510849236c58 Fix command prefix usage
Mikael Berthe <mikael@lilotux.net>
parents: 27
diff changeset
    42
            elseif h.attention and h.attention == "true" and
510849236c58 Fix command prefix usage
Mikael Berthe <mikael@lilotux.net>
parents: 27
diff changeset
    43
                   h.resource and h.resource ~= mcabbot.nickname then
0
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    44
                perso = true
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    45
            end
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    46
        end
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    47
    else
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    48
        if h.jid ~= mcabbot.jid then
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    49
            perso = true
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    50
        end
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    51
    end
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    52
    if perso == true then
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    53
        local cmd, msg
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    54
        if muc == true then
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    55
            cmd = "say_to -q "..h.jid.." "..h.resource..": "
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    56
        else
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    57
            cmd = "say_to -q "..h.jid.."/"..h.resource.." "
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    58
        end
1
cca972635e5e Provid more data to the bot functions
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    59
        mcabbot.message = h
27
a166e9ab8065 Add command prefix (useful for MUC rooms)
Mikael Berthe <mikael@lilotux.net>
parents: 24
diff changeset
    60
        local res, errmsg = process(message, mcabbot, muc, mcabbot.cmdprefix)
1
cca972635e5e Provid more data to the bot functions
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    61
        mcabbot.message = nil
0
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    62
        if res then
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    63
            msg = res
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    64
        elseif errmsg then
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    65
            msg = "! " .. errmsg
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    66
        end
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    67
        if (msg) then main.run(cmd..msg) end
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    68
    end
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    69
end
89add07d6fe4 Initial revision
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    70
1
cca972635e5e Provid more data to the bot functions
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    71
main.hook("hook-post-message-in", hk_message_in)
24
77a72b227d59 Handle subscription requests
Mikael Berthe <mikael@lilotux.net>
parents: 1
diff changeset
    72
main.hook("hook-subscription", hk_subscription)