# -*- coding: utf-8 -*- # # Copyright (c) 2010 by xt # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . # # This script colors nicks in IRC channels in the actual message # not just in the prefix section. # # # History: # 2010-07-29, xt # version 0.6: compile regexp as per patch from Chris quigybo@hotmail.com # 2010-07-19, xt # version 0.5: fix bug with incorrect coloring of own nick # 2010-06-02, xt # version 0.4: update to reflect API changes # 2010-03-26, xt # version 0.3: fix error with exception # 2010-03-24, xt # version 0.2: use ignore_channels when populating to increase performance. # 2010-02-03, xt # version 0.1: initial import weechat import re w = weechat SCRIPT_NAME = "colorize_nicks" SCRIPT_AUTHOR = "xt " SCRIPT_VERSION = "0.6" SCRIPT_LICENSE = "GPL" SCRIPT_DESC = "Use the weechat nick colors in the chat area" settings = { "blacklist_channels" : '', # comma separated list of channels (use short_name) "blacklist_nicks" : 'so,root', # comma separated list of nicks "min_nick_length" : '2', # length } VALID_NICK = r'([@~&!%+])?([-a-zA-Z0-9\[\]\\`_^\{|\}]+)' valid_nick_re = re.compile(VALID_NICK) PREFIX_COLORS = { '@' : 'nicklist_prefix1', '~' : 'nicklist_prefix1', '&' : 'nicklist_prefix1', '!' : 'nicklist_prefix1', '%' : 'nicklist_prefix2', '+' : 'nicklist_prefix3', } ignore_channels = [] ignore_nicks = [] # Dict with every nick on every channel with its color as lookup value colored_nicks = {} def colorize_cb(data, modifier, modifier_data, line): ''' Callback that does the colorizing, and returns new line if changed ''' global ignore_nicks, ignore_channels, colored_nicks if not 'irc_privmsg' in modifier_data: return line full_name = modifier_data.split(';')[1] server = full_name.split('.')[0] channel = '.'.join(full_name.split('.')[1:]) # Check that privmsg is in a channel and that that channel is not ignored if not w.info_get('irc_is_channel', channel) or channel in ignore_channels: return line min_length = int(w.config_get_plugin('min_nick_length')) reset = w.color('reset') buffer = w.buffer_search('', full_name) # Check if buffer has colorized nicks if not buffer in colored_nicks: return line for words in valid_nick_re.findall(line): prefix, nick = words[0], words[1] # Check that nick is not ignored and longer than minimum length if len(nick) < min_length or nick in ignore_nicks: continue if nick in colored_nicks[buffer]: nick_color = colored_nicks[buffer][nick] line = line.replace(nick, '%s%s%s' %(nick_color, nick, reset)) return line def populate_nicks(*kwargs): ''' Fills entire dict with all nicks weechat can see and what color it has assigned to it. ''' global colored_nicks colored_nicks = {} servers = w.infolist_get('irc_server', '', '') while w.infolist_next(servers): servername = w.infolist_string(servers, 'name') colored_nicks[servername] = {} my_nick = w.info_get('irc_nick', servername) channels = w.infolist_get('irc_channel', '', servername) while w.infolist_next(channels): pointer = w.infolist_pointer(channels, 'buffer') nicklist = w.infolist_get('nicklist', pointer, '') channelname = w.infolist_string(channels, 'name') if not pointer in colored_nicks: colored_nicks[pointer] = {} while w.infolist_next(nicklist): nick = w.infolist_string(nicklist, 'name') if nick == my_nick: nick_color = w.color(\ w.config_string(\ w.config_get('weechat.color.chat_nick_self'))) else: nick_color = w.info_get('irc_nick_color', nick) colored_nicks[pointer][nick] = nick_color w.infolist_free(nicklist) w.infolist_free(channels) w.infolist_free(servers) return w.WEECHAT_RC_OK def add_nick(data, signal, type_data): ''' Add nick to dict of colored nicks ''' global colored_nicks pointer, nick = type_data.split(',') if not pointer in colored_nicks: colored_nicks[pointer] = {} servername = w.infolist_get('buffer', pointer, '') w.infolist_next(servername) server = w.infolist_string(servername, 'name') w.infolist_free(servername) servername = server.split('.')[0] my_nick = w.info_get('irc_nick', servername) if nick == my_nick: nick_color = w.color(\ w.config_string(\ w.config_get('weechat.color.chat_nick_self'))) else: nick_color = w.info_get('irc_nick_color', nick) colored_nicks[pointer][nick] = nick_color return w.WEECHAT_RC_OK def remove_nick(data, signal, type_data): ''' Remove nick from dict with colored nicks ''' global colored_nicks pointer, nick = type_data.split(',') if pointer in colored_nicks and nick in colored_nicks[pointer]: del colored_nicks[pointer][nick] return w.WEECHAT_RC_OK if __name__ == "__main__": if w.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE, SCRIPT_DESC, "", ""): # Set default settings for option, default_value in settings.iteritems(): if not w.config_is_set_plugin(option): w.config_set_plugin(option, default_value) for key, value in PREFIX_COLORS.iteritems(): PREFIX_COLORS[key] = w.color(w.config_string(w.config_get('weechat.look.%s'%value))) ignore_channels = w.config_get_plugin('blacklist_channels').split(',') ignore_nicks = w.config_get_plugin('blacklist_nicks').split(',') populate_nicks() # Run it once to get data ready w.hook_signal('nicklist_nick_added', 'add_nick', '') w.hook_signal('nicklist_nick_removed', 'remove_nick', '') w.hook_modifier('weechat_print', 'colorize_cb', '')