321 lines
11 KiB
Python
321 lines
11 KiB
Python
|
import windows
|
||
|
|
|
||
|
|
def _justprefix(network, channel, nick):
|
||
|
|
fr, to = network.isupport["PREFIX"][1:].split(")")
|
||
|
|
|
||
|
|
for mode, prefix in zip(fr, to):
|
||
|
|
if mode in channel.nicks.get(nick, ''):
|
||
|
|
return prefix
|
||
|
|
|
||
|
|
return ''
|
||
|
|
|
||
|
|
def prefix(network, channelname, nick):
|
||
|
|
channel = getchan(network, channelname)
|
||
|
|
|
||
|
|
if channel:
|
||
|
|
nick = '%s%s' % (_justprefix(network, channel, nick), nick)
|
||
|
|
|
||
|
|
return nick
|
||
|
|
|
||
|
|
def escape(string):
|
||
|
|
for escapes in (('&','&'), ('<','<'), ('>','>')):
|
||
|
|
string = string.replace(*escapes)
|
||
|
|
return string
|
||
|
|
|
||
|
|
def sortkey(network, channelname, nick):
|
||
|
|
chanmodes, dummy = network.isupport["PREFIX"][1:].split(")")
|
||
|
|
nickmodes = mode(network, channelname, nick)
|
||
|
|
|
||
|
|
return '%s%s' % (''.join(str(int(mode not in nickmodes)) for mode in chanmodes), network.norm_case(nick))
|
||
|
|
|
||
|
|
def nicklist_add(network, channel, nick):
|
||
|
|
window = windows.get(windows.ChannelWindow, network, channel.name, core)
|
||
|
|
#window = core.window
|
||
|
|
if window:
|
||
|
|
window.nicklist.append(nick, escape(prefix(network, channel.name, nick)), sortkey(network, channel.name, nick))
|
||
|
|
|
||
|
|
def nicklist_del(network, channel, nick):
|
||
|
|
window = windows.get(windows.ChannelWindow, network, channel.name, core)
|
||
|
|
#window = core.window
|
||
|
|
if window:
|
||
|
|
try:
|
||
|
|
window.nicklist.remove(nick)
|
||
|
|
except ValueError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def setupListRightClick(e):
|
||
|
|
if isinstance(e.window, windows.ChannelWindow):
|
||
|
|
#if isinstance(core.window, windows.ChannelWindow):
|
||
|
|
#if e.data[0] in e.window.network.isupport["PREFIX"].split(")")[1]:
|
||
|
|
if e.data[0] in core.window.network.isupport["PREFIX"].split(")")[1]:
|
||
|
|
e.nick = e.data[1:]
|
||
|
|
else:
|
||
|
|
e.nick = e.data
|
||
|
|
|
||
|
|
def setupSocketConnect(e):
|
||
|
|
e.network.channels = {}
|
||
|
|
|
||
|
|
def setdownDisconnect(e):
|
||
|
|
e.network.channels = {}
|
||
|
|
|
||
|
|
class Channel(object):
|
||
|
|
def __init__(self, name):
|
||
|
|
self.name = name
|
||
|
|
self.nicks = {}
|
||
|
|
self.normal_nicks = {} # mapping of normal nicks to actual nicks
|
||
|
|
self.getting_names = False #are we between lines in a /names reply?
|
||
|
|
self.mode = ''
|
||
|
|
self.special_mode = {} #for limits, keys, and anything similar
|
||
|
|
self.topic = ''
|
||
|
|
self.got_mode = False #did we get at least one mode reply?
|
||
|
|
self.got_names = False #did we get at least one names reply?
|
||
|
|
|
||
|
|
def getchan(network, channel):
|
||
|
|
return hasattr(network, 'channels') and network.channels.get(network.norm_case(channel))
|
||
|
|
|
||
|
|
#return a list of channels you're on on the given network
|
||
|
|
def channels(network):
|
||
|
|
if not hasattr(network, 'channels'):
|
||
|
|
network.channels = {}
|
||
|
|
|
||
|
|
return list(network.channels)
|
||
|
|
|
||
|
|
#return True if you're on the channel
|
||
|
|
def ischan(network, channel):
|
||
|
|
return bool(getchan(network, channel))
|
||
|
|
|
||
|
|
#return True if the nick is on the channel
|
||
|
|
def ison(network, channel, nickname):
|
||
|
|
channel = getchan(network, channel)
|
||
|
|
return channel and network.norm_case(nickname) in channel.normal_nicks
|
||
|
|
|
||
|
|
#return a list of nicks on the given channel
|
||
|
|
def nicks(network, channel):
|
||
|
|
channel = getchan(network, channel)
|
||
|
|
|
||
|
|
if channel:
|
||
|
|
return channel.nicks
|
||
|
|
else:
|
||
|
|
return {}
|
||
|
|
|
||
|
|
#return the mode on the given channel
|
||
|
|
def mode(network, channel, nickname=''):
|
||
|
|
channel = getchan(network, channel)
|
||
|
|
|
||
|
|
if channel:
|
||
|
|
if nickname:
|
||
|
|
realnick = channel.normal_nicks.get(network.norm_case(nickname))
|
||
|
|
if realnick:
|
||
|
|
return channel.nicks[realnick]
|
||
|
|
|
||
|
|
else:
|
||
|
|
result = channel.mode
|
||
|
|
for m in channel.mode:
|
||
|
|
if m in channel.special_mode:
|
||
|
|
result += ' '+channel.special_mode[m]
|
||
|
|
return result
|
||
|
|
|
||
|
|
return ''
|
||
|
|
|
||
|
|
#return the topic on the given channel
|
||
|
|
def topic(network, channel):
|
||
|
|
channel = getchan(network, channel)
|
||
|
|
|
||
|
|
if channel:
|
||
|
|
return channel.topic
|
||
|
|
else:
|
||
|
|
return ''
|
||
|
|
|
||
|
|
def setupJoin(e):
|
||
|
|
print e
|
||
|
|
if e.source == e.network.me:
|
||
|
|
e.network.channels[e.network.norm_case(e.target)] = Channel(e.target)
|
||
|
|
e.network.raw('MODE '+e.target)
|
||
|
|
|
||
|
|
#if we wanted to be paranoid, we'd account for not being on the channel
|
||
|
|
channel = getchan(e.network,e.target)
|
||
|
|
channel.nicks[e.source] = ''
|
||
|
|
channel.normal_nicks[e.network.norm_case(e.source)] = e.source
|
||
|
|
|
||
|
|
if e.source == e.network.me:
|
||
|
|
#If the channel window already existed, and we're joining, then we
|
||
|
|
#didn't clear out the nicklist when we left. That means we have to clear
|
||
|
|
#it out now.
|
||
|
|
window = windows.get(windows.ChannelWindow, e.network, e.target, core)
|
||
|
|
#window = core.window
|
||
|
|
#print core
|
||
|
|
if window:
|
||
|
|
window.nicklist.clear()
|
||
|
|
|
||
|
|
nicklist_add(e.network, channel, e.source)
|
||
|
|
|
||
|
|
def setdownPart(e):
|
||
|
|
if e.source == e.network.me:
|
||
|
|
del e.network.channels[e.network.norm_case(e.target)]
|
||
|
|
else:
|
||
|
|
channel = getchan(e.network,e.target)
|
||
|
|
nicklist_del(e.network, channel, e.source)
|
||
|
|
del channel.nicks[e.source]
|
||
|
|
del channel.normal_nicks[e.network.norm_case(e.source)]
|
||
|
|
|
||
|
|
def setdownKick(e):
|
||
|
|
if e.target == e.network.me:
|
||
|
|
del e.network.channels[e.network.norm_case(e.channel)]
|
||
|
|
else:
|
||
|
|
channel = getchan(e.network,e.channel)
|
||
|
|
nicklist_del(e.network, channel, e.target)
|
||
|
|
del channel.nicks[e.target]
|
||
|
|
del channel.normal_nicks[e.network.norm_case(e.target)]
|
||
|
|
|
||
|
|
def setdownQuit(e):
|
||
|
|
#if paranoid: check if e.source is me
|
||
|
|
for channame in channels(e.network):
|
||
|
|
channel = getchan(e.network,channame)
|
||
|
|
if e.source in channel.nicks:
|
||
|
|
nicklist_del(e.network, channel, e.source)
|
||
|
|
del channel.nicks[e.source]
|
||
|
|
del channel.normal_nicks[e.network.norm_case(e.source)]
|
||
|
|
|
||
|
|
def setupMode(e):
|
||
|
|
channel = getchan(e.network,e.channel)
|
||
|
|
if channel:
|
||
|
|
user_modes = e.network.isupport['PREFIX'].split(')')[0][1:]
|
||
|
|
|
||
|
|
(list_modes,
|
||
|
|
always_parm_modes,
|
||
|
|
set_parm_modes,
|
||
|
|
normal_modes) = e.network.isupport['CHANMODES'].split(',')
|
||
|
|
|
||
|
|
list_modes += user_modes
|
||
|
|
|
||
|
|
mode_on = True #are we reading a + section or a - section?
|
||
|
|
params = e.text.split(' ')
|
||
|
|
|
||
|
|
for char in params.pop(0):
|
||
|
|
if char == '+':
|
||
|
|
mode_on = True
|
||
|
|
|
||
|
|
elif char == '-':
|
||
|
|
mode_on = False
|
||
|
|
|
||
|
|
else:
|
||
|
|
if char in user_modes:
|
||
|
|
#these are modes like op and voice
|
||
|
|
nickname = params.pop(0)
|
||
|
|
nicklist_del(e.network, channel, nickname)
|
||
|
|
if mode_on:
|
||
|
|
channel.nicks[nickname] += char
|
||
|
|
else:
|
||
|
|
channel.nicks[nickname] = channel.nicks[nickname].replace(char, '')
|
||
|
|
nicklist_add(e.network, channel, nickname)
|
||
|
|
|
||
|
|
elif char in list_modes:
|
||
|
|
#things like ban/unban
|
||
|
|
#FIXME: We don't keep track of those lists here, but we know
|
||
|
|
# when they're changed and how. Scriptors should be able to
|
||
|
|
# take advantage of this
|
||
|
|
params.pop(0)
|
||
|
|
|
||
|
|
elif char in always_parm_modes:
|
||
|
|
#these always have a parameter
|
||
|
|
param = params.pop(0)
|
||
|
|
|
||
|
|
if mode_on:
|
||
|
|
channel.special_mode[char] = param
|
||
|
|
else:
|
||
|
|
#account for unsetting modes that aren't set
|
||
|
|
channel.special_mode.pop(char, None)
|
||
|
|
|
||
|
|
elif char in set_parm_modes:
|
||
|
|
#these have a parameter only if they're being set
|
||
|
|
if mode_on:
|
||
|
|
channel.special_mode[char] = params.pop(0)
|
||
|
|
else:
|
||
|
|
#account for unsetting modes that aren't set
|
||
|
|
channel.special_mode.pop(char, None)
|
||
|
|
|
||
|
|
if char not in list_modes:
|
||
|
|
if mode_on:
|
||
|
|
channel.mode = channel.mode.replace(char, '')+char
|
||
|
|
else:
|
||
|
|
channel.mode = channel.mode.replace(char, '')
|
||
|
|
|
||
|
|
def setdownNick(e):
|
||
|
|
for channame in channels(e.network):
|
||
|
|
channel = getchan(e.network,channame)
|
||
|
|
if e.source in channel.nicks:
|
||
|
|
nicklist_del(e.network, channel, e.source)
|
||
|
|
del channel.normal_nicks[e.network.norm_case(e.source)]
|
||
|
|
channel.nicks[e.target] = channel.nicks[e.source]
|
||
|
|
del channel.nicks[e.source]
|
||
|
|
channel.normal_nicks[e.network.norm_case(e.target)] = e.target
|
||
|
|
nicklist_add(e.network, channel, e.target)
|
||
|
|
|
||
|
|
def setupTopic(e):
|
||
|
|
channel = getchan(e.network, e.target)
|
||
|
|
if channel:
|
||
|
|
channel.topic = e.text
|
||
|
|
|
||
|
|
def setupRaw(e):
|
||
|
|
if e.msg[1] == '353': #names reply
|
||
|
|
channel = getchan(e.network,e.msg[4])
|
||
|
|
if channel:
|
||
|
|
if not channel.getting_names:
|
||
|
|
channel.nicks.clear()
|
||
|
|
channel.normal_nicks.clear()
|
||
|
|
channel.getting_names = True
|
||
|
|
if not channel.got_names:
|
||
|
|
e.quiet = True
|
||
|
|
for nickname in e.msg[5].split(' '):
|
||
|
|
if nickname:
|
||
|
|
if not nickname[0].isalpha() and nickname[0] in e.network.prefixes:
|
||
|
|
n = nickname[1:]
|
||
|
|
channel.nicks[n] = e.network.prefixes[nickname[0]]
|
||
|
|
channel.normal_nicks[e.network.norm_case(n)] = n
|
||
|
|
else:
|
||
|
|
channel.nicks[nickname] = ''
|
||
|
|
channel.normal_nicks[e.network.norm_case(nickname)] = nickname
|
||
|
|
|
||
|
|
elif e.msg[1] == '366': #end of names reply
|
||
|
|
channel = getchan(e.network,e.msg[3])
|
||
|
|
if channel:
|
||
|
|
if not channel.got_names:
|
||
|
|
e.quiet = True
|
||
|
|
channel.got_names = True
|
||
|
|
channel.getting_names = False
|
||
|
|
|
||
|
|
window = windows.get(windows.ChannelWindow, e.network, e.msg[3], core)
|
||
|
|
if window:
|
||
|
|
window.nicklist.replace(
|
||
|
|
(nick, escape(prefix(e.network, channel.name, nick)), sortkey(e.network, channel.name, nick)) for nick in channel.nicks
|
||
|
|
)
|
||
|
|
|
||
|
|
elif e.msg[1] == '324': #channel mode is
|
||
|
|
channel = getchan(e.network,e.msg[3])
|
||
|
|
if channel:
|
||
|
|
if not channel.got_mode:
|
||
|
|
e.quiet = True
|
||
|
|
channel.got_mode = True
|
||
|
|
mode = e.msg[4]
|
||
|
|
params = e.msg[:4:-1]
|
||
|
|
list_modes, always_parm_modes, set_parm_modes, normal_modes = \
|
||
|
|
e.network.isupport['CHANMODES'].split(',')
|
||
|
|
parm_modes = always_parm_modes + set_parm_modes
|
||
|
|
channel.mode = e.msg[4]
|
||
|
|
channel.special_mode.clear()
|
||
|
|
for char in channel.mode:
|
||
|
|
if char in parm_modes:
|
||
|
|
channel.special_mode[char] = params.pop()
|
||
|
|
|
||
|
|
elif e.msg[1] == '331': #no topic
|
||
|
|
channel = getchan(e.network,e.msg[3])
|
||
|
|
if channel:
|
||
|
|
channel.topic = ''
|
||
|
|
|
||
|
|
elif e.msg[1] == '332': #channel topic is
|
||
|
|
channel = getchan(e.network,e.msg[3])
|
||
|
|
if channel:
|
||
|
|
channel.topic = e.text
|
||
|
|
|
||
|
|
#core.events.load(__name__)
|