sugar-toolkit-gtk3/services/console/lib/purk/scripts/chaninfo.py

321 lines
11 KiB
Python
Raw Normal View History

2007-08-21 18:18:38 +02:00
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 (('&','&amp;'), ('<','&lt;'), ('>','&gt;')):
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__)