path: root/services/console/lib/purk/scripts/chaninfo.py

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__)