329 lines
13 KiB
Python
329 lines
13 KiB
Python
|
# -*- 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:
|
||
|
|
# 2014-09-17, holomorph
|
||
|
|
# version 16: use weechat config facilities
|
||
|
|
# clean unused, minor linting, some simplification
|
||
|
|
# 2014-05-05, holomorph
|
||
|
|
# version 15: fix python2-specific re.search check
|
||
|
|
# 2013-01-29, nils_2
|
||
|
|
# version 14: make script compatible with Python 3.x
|
||
|
|
# 2012-10-19, ldvx
|
||
|
|
# version 13: Iterate over every word to prevent incorrect colorization of
|
||
|
|
# nicks. Added option greedy_matching.
|
||
|
|
# 2012-04-28, ldvx
|
||
|
|
# version 12: added ignore_tags to avoid colorizing nicks if tags are present
|
||
|
|
# 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)
|
||
|
|
#
|
||
|
|
# Known issues: nicks will not get colorized if they begin with a character
|
||
|
|
# such as ~ (which some irc networks do happen to accept)
|
||
|
|
|
||
|
|
import weechat
|
||
|
|
import re
|
||
|
|
w = weechat
|
||
|
|
|
||
|
|
SCRIPT_NAME = "colorize_nicks"
|
||
|
|
SCRIPT_AUTHOR = "xt <xt@bash.no>"
|
||
|
|
SCRIPT_VERSION = "16"
|
||
|
|
SCRIPT_LICENSE = "GPL"
|
||
|
|
SCRIPT_DESC = "Use the weechat nick colors in the chat area"
|
||
|
|
|
||
|
|
VALID_NICK = r'([@~&!%+])?([-a-zA-Z0-9\[\]\\`_^\{|\}]+)'
|
||
|
|
valid_nick_re = re.compile(VALID_NICK)
|
||
|
|
ignore_channels = []
|
||
|
|
ignore_nicks = []
|
||
|
|
|
||
|
|
# Dict with every nick on every channel with its color as lookup value
|
||
|
|
colored_nicks = {}
|
||
|
|
|
||
|
|
CONFIG_FILE_NAME = "colorize_nicks"
|
||
|
|
|
||
|
|
# config file and options
|
||
|
|
colorize_config_file = ""
|
||
|
|
colorize_config_option = {}
|
||
|
|
|
||
|
|
def colorize_config_init():
|
||
|
|
'''
|
||
|
|
Initialization of configuration file.
|
||
|
|
Sections: look.
|
||
|
|
'''
|
||
|
|
global colorize_config_file, colorize_config_option
|
||
|
|
colorize_config_file = weechat.config_new(CONFIG_FILE_NAME,
|
||
|
|
"colorize_config_reload_cb", "")
|
||
|
|
if colorize_config_file == "":
|
||
|
|
return
|
||
|
|
|
||
|
|
# section "look"
|
||
|
|
section_look = weechat.config_new_section(
|
||
|
|
colorize_config_file, "look", 0, 0, "", "", "", "", "", "", "", "", "", "")
|
||
|
|
if section_look == "":
|
||
|
|
weechat.config_free(colorize_config_file)
|
||
|
|
return
|
||
|
|
colorize_config_option["blacklist_channels"] = weechat.config_new_option(
|
||
|
|
colorize_config_file, section_look, "blacklist_channels",
|
||
|
|
"string", "Comma separated list of channels", "", 0, 0,
|
||
|
|
"", "", 0, "", "", "", "", "", "")
|
||
|
|
colorize_config_option["blacklist_nicks"] = weechat.config_new_option(
|
||
|
|
colorize_config_file, section_look, "blacklist_nicks",
|
||
|
|
"string", "Comma separated list of nicks", "", 0, 0,
|
||
|
|
"so,root", "so,root", 0, "", "", "", "", "", "")
|
||
|
|
colorize_config_option["min_nick_length"] = weechat.config_new_option(
|
||
|
|
colorize_config_file, section_look, "min_nick_length",
|
||
|
|
"integer", "Minimum length nick to colorize", "",
|
||
|
|
2, 20, "", "", 0, "", "", "", "", "", "")
|
||
|
|
colorize_config_option["colorize_input"] = weechat.config_new_option(
|
||
|
|
colorize_config_file, section_look, "colorize_input",
|
||
|
|
"boolean", "Whether to colorize input", "", 0,
|
||
|
|
0, "off", "off", 0, "", "", "", "", "", "")
|
||
|
|
colorize_config_option["ignore_tags"] = weechat.config_new_option(
|
||
|
|
colorize_config_file, section_look, "ignore_tags",
|
||
|
|
"string", "Comma separated list of tags to ignore; i.e. irc_join,irc_part,irc_quit", "", 0, 0,
|
||
|
|
"", "", 0, "", "", "", "", "", "")
|
||
|
|
colorize_config_option["greedy_matching"] = weechat.config_new_option(
|
||
|
|
colorize_config_file, section_look, "greedy_matching",
|
||
|
|
"boolean", "If off, then use lazy matching instead", "", 0,
|
||
|
|
0, "on", "on", 0, "", "", "", "", "", "")
|
||
|
|
|
||
|
|
def colorize_config_read():
|
||
|
|
''' Read configuration file. '''
|
||
|
|
global colorize_config_file
|
||
|
|
return weechat.config_read(colorize_config_file)
|
||
|
|
|
||
|
|
def colorize_nick_color(nick, my_nick):
|
||
|
|
''' Retrieve nick color from weechat. '''
|
||
|
|
if nick == my_nick:
|
||
|
|
return w.color(w.config_string(w.config_get('weechat.color.chat_nick_self')))
|
||
|
|
else:
|
||
|
|
return w.info_get('irc_nick_color', nick)
|
||
|
|
|
||
|
|
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]
|
||
|
|
channel = '.'.join(full_name.split('.')[1:])
|
||
|
|
|
||
|
|
buffer = w.buffer_search('', full_name)
|
||
|
|
# Check if buffer has colorized nicks
|
||
|
|
if buffer not in colored_nicks:
|
||
|
|
return line
|
||
|
|
|
||
|
|
if channel in ignore_channels:
|
||
|
|
return line
|
||
|
|
|
||
|
|
min_length = w.config_integer(colorize_config_option['min_nick_length'])
|
||
|
|
reset = w.color('reset')
|
||
|
|
|
||
|
|
# Don't colorize if the ignored tag is present in message
|
||
|
|
tags_line = modifier_data.rsplit(';')
|
||
|
|
if len(tags_line) >= 3:
|
||
|
|
tags_line = tags_line[2].split(',')
|
||
|
|
for i in w.config_string(colorize_config_option['ignore_tags']).split(','):
|
||
|
|
if i in tags_line:
|
||
|
|
return line
|
||
|
|
|
||
|
|
for words in valid_nick_re.findall(line):
|
||
|
|
nick = words[1]
|
||
|
|
# Check that nick is not ignored and longer than minimum length
|
||
|
|
if len(nick) < min_length or nick in ignore_nicks:
|
||
|
|
continue
|
||
|
|
# Check that nick is in the dictionary colored_nicks
|
||
|
|
if nick in colored_nicks[buffer]:
|
||
|
|
nick_color = colored_nicks[buffer][nick]
|
||
|
|
|
||
|
|
# Let's use greedy matching. Will check against every word in a line.
|
||
|
|
if w.config_boolean(colorize_config_option['greedy_matching']):
|
||
|
|
for word in line.split():
|
||
|
|
if nick in word:
|
||
|
|
# Is there a nick that contains nick and has a greater lenght?
|
||
|
|
# If so let's save that nick into var biggest_nick
|
||
|
|
biggest_nick = ""
|
||
|
|
for i in colored_nicks[buffer]:
|
||
|
|
if nick in i and nick != i and len(i) > len(nick):
|
||
|
|
if i in word:
|
||
|
|
# If a nick with greater len is found, and that word
|
||
|
|
# also happens to be in word, then let's save this nick
|
||
|
|
biggest_nick = i
|
||
|
|
# If there's a nick with greater len, then let's skip this
|
||
|
|
# As we will have the chance to colorize when biggest_nick
|
||
|
|
# iterates being nick.
|
||
|
|
if len(biggest_nick) > 0 and biggest_nick in word:
|
||
|
|
pass
|
||
|
|
elif len(word) < len(biggest_nick) or len(biggest_nick) == 0:
|
||
|
|
new_word = word.replace(nick, '%s%s%s' % (nick_color, nick, reset))
|
||
|
|
line = line.replace(word, new_word)
|
||
|
|
# Let's use lazy matching for nick
|
||
|
|
else:
|
||
|
|
nick_color = colored_nicks[buffer][nick]
|
||
|
|
# The two .? are in case somebody writes "nick:", "nick,", etc
|
||
|
|
# to address somebody
|
||
|
|
regex = r"(\A|\s).?(%s).?(\Z|\s)" % re.escape(nick)
|
||
|
|
match = re.search(regex, line)
|
||
|
|
if match is not None:
|
||
|
|
new_line = line[:match.start(2)] + nick_color+nick+reset + line[match.end(2):]
|
||
|
|
line = new_line
|
||
|
|
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
|
||
|
|
|
||
|
|
min_length = w.config_integer(colorize_config_option['min_nick_length'])
|
||
|
|
|
||
|
|
if not w.config_boolean(colorize_config_option['colorize_input']):
|
||
|
|
return line
|
||
|
|
|
||
|
|
buffer = w.current_buffer()
|
||
|
|
# Check if buffer has colorized nicks
|
||
|
|
if buffer not 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):
|
||
|
|
nick = 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, '')
|
||
|
|
|
||
|
|
if pointer not in colored_nicks:
|
||
|
|
colored_nicks[pointer] = {}
|
||
|
|
|
||
|
|
while w.infolist_next(nicklist):
|
||
|
|
nick = w.infolist_string(nicklist, 'name')
|
||
|
|
nick_color = colorize_nick_color(nick, my_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 pointer not in colored_nicks:
|
||
|
|
colored_nicks[pointer] = {}
|
||
|
|
|
||
|
|
my_nick = w.buffer_get_string(pointer, 'localvar_nick')
|
||
|
|
nick_color = colorize_nick_color(nick, my_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):
|
||
|
|
''' Set the blacklist for channels and nicks. '''
|
||
|
|
global ignore_channels, ignore_nicks
|
||
|
|
ignore_channels = w.config_string(colorize_config_option['blacklist_channels']).split(',')
|
||
|
|
ignore_nicks = w.config_string(colorize_config_option['blacklist_nicks']).split(',')
|
||
|
|
return w.WEECHAT_RC_OK
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if w.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE,
|
||
|
|
SCRIPT_DESC, "", ""):
|
||
|
|
colorize_config_init()
|
||
|
|
colorize_config_read()
|
||
|
|
|
||
|
|
# Run once to get data ready
|
||
|
|
update_blacklist()
|
||
|
|
populate_nicks()
|
||
|
|
|
||
|
|
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('%s.look.blacklist*' % SCRIPT_NAME, 'update_blacklist', '')
|