xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Annotate irc-mirror.py.

Browse source
This commit is contained in:
Tim Abbott 2016-09-10 12:19:08 -07:00
parent 33733e67ba
commit 38b62518af
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
bots/irc-mirror.py
View file

@ -9,27 +9,34 @@
from __future__ import print_function from __future__ import print_function
import irc.bot import irc.bot
import irc.strings import irc.strings
from irc.client import ip_numstr_to_quad, ip_quad_to_numstr from irc.client import ip_numstr_to_quad, ip_quad_to_numstr, Event, ServerConnection
import zulip import zulip
import optparse import optparse
if False: from typing import Any
IRC_DOMAIN = "irc.example.com" IRC_DOMAIN = "irc.example.com"
def zulip_sender(sender_string): def zulip_sender(sender_string):
# type: (str) -> str
nick = sender_string.split("!")[0] nick = sender_string.split("!")[0]
return nick + "@" + IRC_DOMAIN return nick + "@" + IRC_DOMAIN
class IRCBot(irc.bot.SingleServerIRCBot): class IRCBot(irc.bot.SingleServerIRCBot):
def __init__(self, channel, nickname, server, port=6667): def __init__(self, channel, nickname, server, port=6667):
# type: (irc.bot.Channel, str, str, int) -> None
irc.bot.SingleServerIRCBot.__init__(self, [(server, port)], nickname, nickname) irc.bot.SingleServerIRCBot.__init__(self, [(server, port)], nickname, nickname)
self.channel = channel self.channel = channel # type: irc.bot.Channel
def on_nicknameinuse(self, c, e): def on_nicknameinuse(self, c, e):
# type: (ServerConnection, Event) -> None
c.nick(c.get_nickname().replace("_zulip", "__zulip")) c.nick(c.get_nickname().replace("_zulip", "__zulip"))
def on_welcome(self, c, e): def on_welcome(self, c, e):
# type: (ServerConnection, Event) -> None
c.join(self.channel) c.join(self.channel)
def forward_to_irc(msg): def forward_to_irc(msg):
# type: (Dict[str, Any]) -> None
if msg["type"] == "stream": if msg["type"] == "stream":
send = lambda x: c.privmsg(msg["display_recipient"], x) send = lambda x: c.privmsg(msg["display_recipient"], x)
else: else:
@ -48,6 +55,7 @@ class IRCBot(irc.bot.SingleServerIRCBot):
# zulip_client.call_on_each_message(forward_to_irc) # zulip_client.call_on_each_message(forward_to_irc)
def on_privmsg(self, c, e): def on_privmsg(self, c, e):
# type: (ServerConnection, Event) -> None
content = e.arguments[0] content = e.arguments[0]
sender = zulip_sender(e.source) sender = zulip_sender(e.source)
if sender.endswith("_zulip@" + IRC_DOMAIN): if sender.endswith("_zulip@" + IRC_DOMAIN):
@ -62,6 +70,7 @@ class IRCBot(irc.bot.SingleServerIRCBot):
})) }))
def on_pubmsg(self, c, e): def on_pubmsg(self, c, e):
# type: (ServerConnection, Event) -> None
content = e.arguments[0] content = e.arguments[0]
stream = e.target stream = e.target
sender = zulip_sender(e.source) sender = zulip_sender(e.source)
@ -79,9 +88,11 @@ class IRCBot(irc.bot.SingleServerIRCBot):
})) }))
def on_dccmsg(self, c, e): def on_dccmsg(self, c, e):
# type: (ServerConnection, Event) -> None
c.privmsg("You said: " + e.arguments[0]) c.privmsg("You said: " + e.arguments[0])
def on_dccchat(self, c, e): def on_dccchat(self, c, e):
# type: (ServerConnection, Event) -> None
if len(e.arguments) != 2: if len(e.arguments) != 2:
return return
args = e.arguments[1].split() args = e.arguments[1].split()