aboutsummaryrefslogtreecommitdiff
path: root/modules/line_handler/message.py
blob: 7bc2e27e8366456753037f156914aacdf6692469 (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
from src import utils

def _from_self(server, direction, prefix):
    if direction == utils.Direction.SEND:
        if server.has_capability("echo-message"):
            return None
        else:
            return True
    else:
        if prefix:
            return server.is_own_nickname(prefix.nickname)
        else:
            return False

def privmsg(events, event):
    from_self = _from_self(event["server"], event["direction"],
        event.get("prefix", None))
    if from_self == None:
        return

    user = None
    if "prefix" in event and not from_self:
        user = event["server"].get_user(event["prefix"].nickname)

    message = event["args"][1]
    target = event["args"][0]

    # strip prefix_symbols from the start of target, for when people use
    # e.g. 'PRIVMSG +#channel :hi' which would send a message to only
    # voiced-or-above users
    statusmsg = []
    while target[0] in event["server"].prefix_symbols.keys():
        statusmsg.append(target[0])
        target = target[1:]

    channel = None
    if target[0] in event["server"].channel_types:
        if not target in event["server"].channels:
            return
        channel = event["server"].channels.get(target)

    action = False
    event_type = "message"
    ctcp_message = utils.irc.parse_ctcp(message)
    if ctcp_message:
        message = ctcp_message.message
        if ctcp_message.command == "ACTION":
            action = True
            message = ctcp_message.message
        else:
            event_type = "ctcp.%s" % ctcp_message.command

    if user and "account" in event["tags"]:
        user.identified_account = event["tags"]["account"]
        user.identified_account_id = event["server"].get_user(
            event["tags"]["account"]).get_id()

    kwargs = {"message": message, "message_split": message.split(" "),
        "server": event["server"], "tags": event["tags"],
        "action": action}

    direction = "send" if from_self else "received"
    context = "channel" if channel else "private"
    hook = events.on(direction).on(event_type).on(context)

    user_nickname = None
    if user:
        user_nickname = None if from_self else user.nickname

    if channel:
        hook.call(user=user, channel=channel, statusmsg=statusmsg, **kwargs)
        channel.buffer.add_message(user_nickname, message, action,
            event["tags"], user==None)
    elif event["server"].is_own_nickname(target):
        hook.call(user=user, **kwargs)
        user.buffer.add_message(user_nickname, message, action,
            event["tags"], False)
    elif from_self:
        # a message we've sent to a user
        user = event["server"].get_user(target)
        hook.call(user=user, **kwargs)
        user.buffer.add_message(user_nickname, message, action,
            event["tags"], True)

def notice(events, event):
    from_self = _from_self(event["server"], event["direction"],
        event.get("prefix", None))
    if from_self == None:
        return

    message = event["args"][1]
    target = event["args"][0]

    if "prefix" in event and (
            not event["prefix"] or
            not event["server"].name or
            event["prefix"].hostmask == event["server"].name or
            target == "*"):
        if event["prefix"]:
            event["server"].name = event["prefix"].hostmask

        events.on("received.server-notice").call(message=message,
            message_split=message.split(" "), server=event["server"])
    else:
        user = None
        if "prefix" in event and not from_self:
            user = event["server"].get_user(event["prefix"].nickname)

        channel = None
        if target[0] in event["server"].channel_types:
            channel = event["server"].channels.get(target)

        direction = "send" if from_self else "received"
        context = "channel" if channel else "private"
        hook = events.on(direction).on("notice").on(context)

        user_nickname = None
        if user:
            user_nickname = None if from_self else user.nickname

        kwargs = {"message": message, "message_split": message.split(" "),
            "server": event["server"], "tags": event["tags"]}

        if channel:
            hook.call(user=user, channel=channel, **kwargs)
            channel.buffer.add_notice(user_nickname, message, event["tags"],
                user==None)
        elif event["server"].is_own_nickname(target):
            hook.call(user=user, **kwargs)
            user.buffer.add_notice(user_nickname, message, event["tags"],
                False)
        elif from_self:
            # a notice we've sent to a user
            user = event["server"].get_user(target)
            hook.call(user=user, **kwargs)
            user.buffer.add_notice(user_nickname, message, event["tags"],
                True)

def tagmsg(events, event):
    from_self = _from_self(event["server"], event["direction"],
        event.get("prefix", None))
    if from_self == None:
        return

    user = None
    if "prefix" in event and not from_self:
        user = event["server"].get_user(event["prefix"].nickname)

    target = event["args"][0]
    channel = None
    if target[0] in event["server"].channel_types:
        channel = event["server"].channels.get(target)

    direction = "send" if from_self else "received"
    context = "channel" if channel else "private"
    hook = events.on(direction).on("tagmsg").on(context)

    kwargs = {"server": event["server"], "tags": event["tags"]}

    if channel:
        hook.call(user=user, channel=channel, **kwargs)
    elif event["server"].is_own_nickname(target):
        hook.call(user=user, **kwargs)
    elif from_self:
        user = event["server"].get_user(target)
        hook.call(user=user, **kwargs)