aboutsummaryrefslogtreecommitdiffstats
path: root/weechat/python/colorize_nicks.py
diff options
context:
space:
mode:
authorJohannes Löthberg <johannes@kyriasis.com>2014-02-02 20:35:42 +0100
committerJohannes Löthberg <johannes@kyriasis.com>2014-02-02 22:01:01 +0100
commit7e7aa928072775240ff70ca61f2dd0e8a09242d8 (patch)
treee22d0a73612befd95744dec72770e74593ab47cd /weechat/python/colorize_nicks.py
downloaddotfiles-7e7aa928072775240ff70ca61f2dd0e8a09242d8.tar.xz
(Let's pretend that this is the) initial commit
Diffstat (limited to 'weechat/python/colorize_nicks.py')
-rw-r--r--weechat/python/colorize_nicks.py316
1 files changed, 316 insertions, 0 deletions
diff --git a/weechat/python/colorize_nicks.py b/weechat/python/colorize_nicks.py
new file mode 100644
index 0000000..d3730da
--- /dev/null
+++ b/weechat/python/colorize_nicks.py
@@ -0,0 +1,316 @@
+# -*- 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:
+# 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 = "14"
+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
+ "ignore_tags" : '', # comma separated list of tags to ignore. I.e. irc_join,irc_part,irc_quit
+ "greedy_matching" : 'on', # if off, then let's use lazy matching instead
+}
+
+
+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')
+
+ # 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_get_plugin('ignore_tags').split(','):
+ if i in tags_line:
+ 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
+ # 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_get_plugin('greedy_matching') == "on":
+ 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
+ elif w.config_get_plugin('greedy_matching') == "off":
+ if nick in colored_nicks[buffer]:
+ 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 str(type(match)) == "<type '_sre.SRE_Match'>":
+ 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
+
+ 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():
+ for option, default_value in list(settings.items()):
+ if not w.config_is_set_plugin(option):
+ w.config_set_plugin(option, default_value)
+
+ for key, value in list(PREFIX_COLORS.items()):
+# 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', '')