GitHub/PyLink
3
0
Fork 0
mirror of https://github.com/jlu5/PyLink.git synced 2025-01-26 04:04:22 +01:00
Code Issues Projects Releases Wiki Activity

relay: spawn users on the right pseudoservers, and use SJOIN to introduce them if possible

Browse Source
This commit is contained in:
James Lu 2015-07-13 16:09:22 -07:00
parent 67f5bbba4b
commit d8cb3f1973
1 changed files with 26 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
plugins/relay.py
View File

@ -14,6 +14,7 @@ from log import log
dbname = "pylinkrelay.db"
relayusers = defaultdict(dict)
relayservers = defaultdict(dict)
def normalizeNick(irc, netname, nick, separator="/"):
orig_nick = nick
@ -123,10 +124,27 @@ def relayJoins(irc, channel, users, ts, modes):
realname = userobj.realname
log.debug('Okay, spawning %s/%s everywhere', user, irc.name)
for name, remoteirc in utils.networkobjects.items():
nick = normalizeNick(remoteirc, irc.name, userobj.nick)
if name == irc.name:
# Don't relay things to their source network...
continue
try: # Spawn our pseudoserver first
relayservers[remoteirc.name][irc.name] = sid = \
remoteirc.proto.spawnServer(remoteirc, '%s.relay' % irc.name,
endburst=False)
# We want to wait a little bit for the remote IRCd to send their users,
# so we can join them as part of a burst on remote networks.
# Because IRC is asynchronous, we can't really control how long
# this will take.
endburst_timer = threading.Timer(0.5, remoteirc.proto.endburstServer,
args=(remoteirc, sid))
log.debug('(%s) Setting timer to BURST %s', remoteirc.name, sid)
endburst_timer.start()
except ValueError:
# Server already exists (raised by the protocol module).
sid = relayservers[remoteirc.name][irc.name]
log.debug('(%s) Have we bursted %s yet? %s', remoteirc.name, sid,
remoteirc.servers[sid].has_bursted)
nick = normalizeNick(remoteirc, irc.name, userobj.nick)
# If the user (stored here as {(netname, UID):
# {network1: UID1, network2: UID2}}) exists, don't spawn it
# again!
@ -135,11 +153,15 @@ def relayJoins(irc, channel, users, ts, modes):
u = userpair_index.get(remoteirc.name)
if u is None: # .get() returns None if not found
u = remoteirc.proto.spawnClient(remoteirc, nick, ident=ident,
host=host, realname=realname).uid
host=host, realname=realname,
server=sid).uid
remoteirc.users[u].remote = irc.name
log.debug('(%s) Spawning client %s (UID=%s)', irc.name, nick, u)
relayusers[(irc.name, userobj.uid)][remoteirc.name] = u
remoteirc.users[u].remote = irc.name
if not remoteirc.servers[sid].has_bursted:
# TODO: join users in batches with SJOIN, not one by one.
remoteirc.proto.sjoinServer(remoteirc, sid, channel, [('', u)], ts=ts)
else:
remoteirc.proto.joinClient(remoteirc, u, channel)
def removeChannel(irc, channel):