# -*- coding: utf-8 -*-
#
# Copyright (c) 2010 by xt <xt@bash.no>
#
# 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 <http://www.gnu.org/licenses/>.
#
# This script colors nicks in IRC channels in the actual message
# not just in the prefix section.
#
#
# History:
# 2012-01-14, nesthib
# version 11: input_text_display hook and modifier to colorize nicks in input bar
# 2010-12-22, xt
# version 10: hook config option for updating blacklist
# 2010-12-20, xt
# version 0.9: hook new config option for weechat 0.3.4
# 2010-11-01, nils_2
# version 0.8: hook_modifier() added to communicate with rainbow_text
# 2010-10-01, xt
# version 0.7: changes to support non-irc-plugins
# 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 (based on ruby script by dominikh)
import weechat
import re
w = weechat
SCRIPT_NAME = "colorize_nicks"
SCRIPT_AUTHOR = "xt <xt@bash.no>"
SCRIPT_VERSION = "11"
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
"colorize_input" : 'off', # boolean
}
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
full_name = modifier_data.split(';')[1]
server = full_name.split('.')[0]
channel = '.'.join(full_name.split('.')[1:])
buffer = w.buffer_search('', full_name)
# Check if buffer has colorized nicks
if not buffer in colored_nicks:
return line
if channel in ignore_channels:
return line
try:
min_length = int(w.config_get_plugin('min_nick_length'))
except ValueError:
w.prnt('', '%sError with option min_nick_length, should be a integer' % weechat.prefix('error'))
w.config_set_plugin('min_nick_length', settings['min_nick_length'])
reset = w.color('reset')
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 colorize_input_cb(data, modifier, modifier_data, line):
''' Callback that does the colorizing in input '''
global ignore_nicks, ignore_channels, colored_nicks
try:
min_length = int(w.config_get_plugin('min_nick_length'))
except ValueError:
w.prnt('', '%sError with option min_nick_length, should be a integer' % weechat.prefix('error'))
w.config_set_plugin('min_nick_length', settings['min_nick_length'])
if w.config_get_plugin('colorize_input') == 'on':
pass
elif w.config_get_plugin('colorize_input') == 'off':
return line
else:
w.prnt('', '%sError with option colorize_input, should be on or off' % weechat.prefix('error'))
w.config_set_plugin('colorize_input', settings['colorize_input'])
return line
buffer = w.current_buffer()
# Check if buffer has colorized nicks
if not buffer in colored_nicks:
return line
channel = w.buffer_get_string(buffer,'name')
if channel in ignore_channels:
return line
reset = w.color('reset')
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(*args):
''' 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.buffer_get_string(pointer, 'localvar_server')
my_nick = w.buffer_get_string(pointer, 'localvar_nick')
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
def update_blacklist(*args):
global ignore_channels, ignore_nicks
if w.config_get_plugin('blacklist_channels'):
ignore_channels = w.config_get_plugin('blacklist_channels').split(',')
ignore_nicks = w.config_get_plugin('blacklist_nicks').split(',')
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)))
update_blacklist() # Set blacklist
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', '')
# Hook config for changing colors
w.hook_config('weechat.color.chat_nick_colors', 'populate_nicks', '')
# Hook for working togheter with other scripts (like colorize_lines)
w.hook_modifier('colorize_nicks', 'colorize_cb', '')
# Hook for modifying input
w.hook_modifier('250|input_text_display', 'colorize_input_cb', '')
# Hook for updating blacklist (this could be improved to use fnmatch)
weechat.hook_config('plugins.var.python.%s.blacklist*' %SCRIPT_NAME, 'update_blacklist', '')
Back to scripts |
Download