weechat - change colors, color nicks, go-script
- Change the colours of the nicks, the time format - Colour the nicks in the text - Add go-script to fuzzy go to buffers
This commit is contained in:
parent
68c2e0b11c
commit
3df25d3dda
1
.gitignore
vendored
1
.gitignore
vendored
@ -2,6 +2,7 @@
|
||||
# weechat
|
||||
weechat/irc.conf
|
||||
weechat/logs/
|
||||
weechat/xfer/
|
||||
weechat/weechat.log
|
||||
weechat/script/plugins.xml.gz
|
||||
|
||||
|
11
weechat/colorize_nicks.conf
Normal file
11
weechat/colorize_nicks.conf
Normal file
@ -0,0 +1,11 @@
|
||||
#
|
||||
# colorize_nicks.conf -- weechat v1.0.1
|
||||
#
|
||||
|
||||
[look]
|
||||
blacklist_channels = ""
|
||||
blacklist_nicks = "so,root"
|
||||
colorize_input = off
|
||||
greedy_matching = on
|
||||
ignore_tags = ""
|
||||
min_nick_length = 2
|
@ -22,10 +22,17 @@ python.anotify.show_upgrade_ended = "on"
|
||||
python.anotify.sticky = "off"
|
||||
python.anotify.sticky_away = "on"
|
||||
python.check_license = "off"
|
||||
python.lnotify.highlight = "on"
|
||||
python.lnotify.icon = "weechat"
|
||||
python.lnotify.notify_away = "off"
|
||||
python.lnotify.query = "on"
|
||||
python.go.auto_jump = "off"
|
||||
python.go.color_name = "black,cyan"
|
||||
python.go.color_name_highlight = "red,cyan"
|
||||
python.go.color_name_highlight_selected = "red,brown"
|
||||
python.go.color_name_selected = "black,brown"
|
||||
python.go.color_number = "yellow,magenta"
|
||||
python.go.color_number_selected = "yellow,red"
|
||||
python.go.message = "Go to: "
|
||||
python.go.short_name = "off"
|
||||
python.go.sort = "number,beginning"
|
||||
python.go.use_core_instead_weechat = "off"
|
||||
python.shortenurl.color = "red"
|
||||
python.shortenurl.ignore_list = "http://is.gd,http://tinyurl.com"
|
||||
python.shortenurl.short_own = "off"
|
||||
@ -34,3 +41,14 @@ python.shortenurl.urllength = "120"
|
||||
ruby.check_license = "off"
|
||||
|
||||
[desc]
|
||||
python.go.auto_jump = "automatically jump to buffer when it is uniquely selected (default: "off")"
|
||||
python.go.color_name = "color for buffer name (not selected) (default: "black,cyan")"
|
||||
python.go.color_name_highlight = "color for highlight in buffer name (not selected) (default: "red,cyan")"
|
||||
python.go.color_name_highlight_selected = "color for highlight in a selected buffer name (default: "red,brown")"
|
||||
python.go.color_name_selected = "color for a selected buffer name (default: "black,brown")"
|
||||
python.go.color_number = "color for buffer number (not selected) (default: "yellow,magenta")"
|
||||
python.go.color_number_selected = "color for selected buffer number (default: "yellow,red")"
|
||||
python.go.message = "message to display before list of buffers (default: "Go to: ")"
|
||||
python.go.short_name = "display and search in short names instead of buffer name (default: "off")"
|
||||
python.go.sort = "comma-separated list of keys to sort buffers (the order is important, sorts are performed in the given order): name = sort by name (or short name), (default: "number,beginning")"
|
||||
python.go.use_core_instead_weechat = "use name "core" instead of "weechat" for core buffer (default: "off")"
|
||||
|
1
weechat/python/autoload/colorize_nicks.py
Symbolic link
1
weechat/python/autoload/colorize_nicks.py
Symbolic link
@ -0,0 +1 @@
|
||||
../colorize_nicks.py
|
1
weechat/python/autoload/go.py
Symbolic link
1
weechat/python/autoload/go.py
Symbolic link
@ -0,0 +1 @@
|
||||
../go.py
|
328
weechat/python/colorize_nicks.py
Normal file
328
weechat/python/colorize_nicks.py
Normal file
@ -0,0 +1,328 @@
|
||||
# -*- 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', '')
|
490
weechat/python/go.py
Normal file
490
weechat/python/go.py
Normal file
@ -0,0 +1,490 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
#
|
||||
# Copyright (C) 2009-2014 Sébastien Helleu <flashcode@flashtux.org>
|
||||
# Copyright (C) 2010 m4v <lambdae2@gmail.com>
|
||||
# Copyright (C) 2011 stfn <stfnmd@googlemail.com>
|
||||
#
|
||||
# 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/>.
|
||||
#
|
||||
|
||||
#
|
||||
# History:
|
||||
#
|
||||
# 2014-05-12, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 2.0: add help on options, replace option "sort_by_activity" by
|
||||
# "sort" (add sort by name and first match at beginning of
|
||||
# name and by number), PEP8 compliance
|
||||
# 2012-11-26, Nei <anti.teamidiot.de>
|
||||
# version 1.9: add auto_jump option to automatically go to buffer when it
|
||||
# is uniquely selected
|
||||
# 2012-09-17, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 1.8: fix jump to non-active merged buffers (jump with buffer name
|
||||
# instead of number)
|
||||
# 2012-01-03 nils_2 <weechatter@arcor.de>
|
||||
# version 1.7: add option use_core_instead_weechat
|
||||
# 2012-01-03, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 1.6: make script compatible with Python 3.x
|
||||
# 2011-08-24, stfn <stfnmd@googlemail.com>:
|
||||
# version 1.5: /go with name argument jumps directly to buffer
|
||||
# Remember cursor position in buffer input
|
||||
# 2011-05-31, Elián Hanisch <lambdae2@gmail.com>:
|
||||
# version 1.4: Sort list of buffers by activity.
|
||||
# 2011-04-25, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 1.3: add info "go_running" (used by script input_lock.rb)
|
||||
# 2010-11-01, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 1.2: use high priority for hooks to prevent conflict with other
|
||||
# plugins/scripts (WeeChat >= 0.3.4 only)
|
||||
# 2010-03-25, Elián Hanisch <lambdae2@gmail.com>:
|
||||
# version 1.1: use a space for match the end of a string
|
||||
# 2009-11-16, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 1.0: add new option for displaying short names
|
||||
# 2009-06-15, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 0.9: fix typo in /help go with command /key
|
||||
# 2009-05-16, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 0.8: search buffer by number, fix bug when window is split
|
||||
# 2009-05-03, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 0.7: eat tab key (do not complete input, just move buffer
|
||||
# pointer)
|
||||
# 2009-05-02, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 0.6: sync with last API changes
|
||||
# 2009-03-22, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 0.5: update modifier signal name for input text display,
|
||||
# fix arguments for function string_remove_color
|
||||
# 2009-02-18, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 0.4: do not hook command and init options if register failed
|
||||
# 2009-02-08, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 0.3: case insensitive search for buffers names
|
||||
# 2009-02-08, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 0.2: add help about Tab key
|
||||
# 2009-02-08, Sébastien Helleu <flashcode@flashtux.org>:
|
||||
# version 0.1: initial release
|
||||
#
|
||||
|
||||
"""
|
||||
Quick jump to buffers.
|
||||
(this script requires WeeChat 0.3.0 or newer)
|
||||
"""
|
||||
|
||||
from __future__ import print_function
|
||||
|
||||
SCRIPT_NAME = 'go'
|
||||
SCRIPT_AUTHOR = 'Sébastien Helleu <flashcode@flashtux.org>'
|
||||
SCRIPT_VERSION = '2.0'
|
||||
SCRIPT_LICENSE = 'GPL3'
|
||||
SCRIPT_DESC = 'Quick jump to buffers'
|
||||
|
||||
SCRIPT_COMMAND = 'go'
|
||||
|
||||
IMPORT_OK = True
|
||||
|
||||
try:
|
||||
import weechat
|
||||
except ImportError:
|
||||
print('This script must be run under WeeChat.')
|
||||
print('Get WeeChat now at: http://www.weechat.org/')
|
||||
IMPORT_OK = False
|
||||
|
||||
import re
|
||||
|
||||
# script options
|
||||
SETTINGS = {
|
||||
'color_number': (
|
||||
'yellow,magenta',
|
||||
'color for buffer number (not selected)'),
|
||||
'color_number_selected': (
|
||||
'yellow,red',
|
||||
'color for selected buffer number'),
|
||||
'color_name': (
|
||||
'black,cyan',
|
||||
'color for buffer name (not selected)'),
|
||||
'color_name_selected': (
|
||||
'black,brown',
|
||||
'color for a selected buffer name'),
|
||||
'color_name_highlight': (
|
||||
'red,cyan',
|
||||
'color for highlight in buffer name (not selected)'),
|
||||
'color_name_highlight_selected': (
|
||||
'red,brown',
|
||||
'color for highlight in a selected buffer name'),
|
||||
'message': (
|
||||
'Go to: ',
|
||||
'message to display before list of buffers'),
|
||||
'short_name': (
|
||||
'off',
|
||||
'display and search in short names instead of buffer name'),
|
||||
'sort': (
|
||||
'number,beginning',
|
||||
'comma-separated list of keys to sort buffers '
|
||||
'(the order is important, sorts are performed in the given order): '
|
||||
'name = sort by name (or short name), ',
|
||||
'hotlist = sort by hotlist order, '
|
||||
'number = first match a buffer number before digits in name, '
|
||||
'beginning = first match at beginning of names (or short names); '
|
||||
'the default sort of buffers is by numbers'),
|
||||
'use_core_instead_weechat': (
|
||||
'off',
|
||||
'use name "core" instead of "weechat" for core buffer'),
|
||||
'auto_jump': (
|
||||
'off',
|
||||
'automatically jump to buffer when it is uniquely selected'),
|
||||
}
|
||||
|
||||
# hooks management
|
||||
HOOK_COMMAND_RUN = {
|
||||
'input': ('/input *', 'go_command_run_input'),
|
||||
'buffer': ('/buffer *', 'go_command_run_buffer'),
|
||||
'window': ('/window *', 'go_command_run_window'),
|
||||
}
|
||||
hooks = {}
|
||||
|
||||
# input before command /go (we'll restore it later)
|
||||
saved_input = ''
|
||||
saved_input_pos = 0
|
||||
|
||||
# last user input (if changed, we'll update list of matching buffers)
|
||||
old_input = None
|
||||
|
||||
# matching buffers
|
||||
buffers = []
|
||||
buffers_pos = 0
|
||||
|
||||
|
||||
def go_option_enabled(option):
|
||||
"""Checks if a boolean script option is enabled or not."""
|
||||
return weechat.config_string_to_boolean(weechat.config_get_plugin(option))
|
||||
|
||||
|
||||
def go_info_running(data, info_name, arguments):
|
||||
"""Returns "1" if go is running, otherwise "0"."""
|
||||
return '1' if 'modifier' in hooks else '0'
|
||||
|
||||
|
||||
def go_unhook_one(hook):
|
||||
"""Unhook something hooked by this script."""
|
||||
global hooks
|
||||
if hook in hooks:
|
||||
weechat.unhook(hooks[hook])
|
||||
del hooks[hook]
|
||||
|
||||
|
||||
def go_unhook_all():
|
||||
"""Unhook all."""
|
||||
go_unhook_one('modifier')
|
||||
for hook in HOOK_COMMAND_RUN:
|
||||
go_unhook_one(hook)
|
||||
|
||||
|
||||
def go_hook_all():
|
||||
"""Hook command_run and modifier."""
|
||||
global hooks
|
||||
priority = ''
|
||||
version = weechat.info_get('version_number', '') or 0
|
||||
# use high priority for hook to prevent conflict with other plugins/scripts
|
||||
# (WeeChat >= 0.3.4 only)
|
||||
if int(version) >= 0x00030400:
|
||||
priority = '2000|'
|
||||
for hook, value in HOOK_COMMAND_RUN.items():
|
||||
if hook not in hooks:
|
||||
hooks[hook] = weechat.hook_command_run(
|
||||
'%s%s' % (priority, value[0]),
|
||||
value[1], '')
|
||||
if 'modifier' not in hooks:
|
||||
hooks['modifier'] = weechat.hook_modifier(
|
||||
'input_text_display_with_cursor', 'go_input_modifier', '')
|
||||
|
||||
|
||||
def go_start(buf):
|
||||
"""Start go on buffer."""
|
||||
global saved_input, saved_input_pos, old_input, buffers_pos
|
||||
go_hook_all()
|
||||
saved_input = weechat.buffer_get_string(buf, 'input')
|
||||
saved_input_pos = weechat.buffer_get_integer(buf, 'input_pos')
|
||||
weechat.buffer_set(buf, 'input', '')
|
||||
old_input = None
|
||||
buffers_pos = 0
|
||||
|
||||
|
||||
def go_end(buf):
|
||||
"""End go on buffer."""
|
||||
global saved_input, saved_input_pos, old_input
|
||||
go_unhook_all()
|
||||
weechat.buffer_set(buf, 'input', saved_input)
|
||||
weechat.buffer_set(buf, 'input_pos', str(saved_input_pos))
|
||||
old_input = None
|
||||
|
||||
|
||||
def go_match_beginning(buf, string):
|
||||
"""Check if a string matches the beginning of buffer name/short name."""
|
||||
if not string:
|
||||
return False
|
||||
esc_str = re.escape(string)
|
||||
if re.search(r'^#?' + esc_str, buf['name']) \
|
||||
or re.search(r'^#?' + esc_str, buf['short_name']):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def go_now(buf, args):
|
||||
"""Go to buffer specified by args."""
|
||||
listbuf = go_matching_buffers(args)
|
||||
if not listbuf:
|
||||
return
|
||||
|
||||
# prefer buffer that matches at beginning (if option is enabled)
|
||||
if 'beginning' in weechat.config_get_plugin('sort').split(','):
|
||||
for index in range(len(listbuf)):
|
||||
if go_match_beginning(listbuf[index], args):
|
||||
weechat.command(buf,
|
||||
'/buffer ' + str(listbuf[index]['full_name']))
|
||||
return
|
||||
|
||||
# jump to first buffer in matching buffers by default
|
||||
weechat.command(buf, '/buffer ' + str(listbuf[0]['full_name']))
|
||||
|
||||
|
||||
def go_cmd(data, buf, args):
|
||||
"""Command "/go": just hook what we need."""
|
||||
global hooks
|
||||
if args:
|
||||
go_now(buf, args)
|
||||
elif 'modifier' in hooks:
|
||||
go_end(buf)
|
||||
else:
|
||||
go_start(buf)
|
||||
return weechat.WEECHAT_RC_OK
|
||||
|
||||
|
||||
def go_matching_buffers(strinput):
|
||||
"""Return a list with buffers matching user input."""
|
||||
global buffers_pos
|
||||
listbuf = []
|
||||
if len(strinput) == 0:
|
||||
buffers_pos = 0
|
||||
strinput = strinput.lower()
|
||||
infolist = weechat.infolist_get('buffer', '', '')
|
||||
while weechat.infolist_next(infolist):
|
||||
short_name = weechat.infolist_string(infolist, 'short_name')
|
||||
if go_option_enabled('short_name'):
|
||||
name = weechat.infolist_string(infolist, 'short_name')
|
||||
else:
|
||||
name = weechat.infolist_string(infolist, 'name')
|
||||
if name == 'weechat' and go_option_enabled('use_core_instead_weechat'):
|
||||
name = 'core'
|
||||
number = weechat.infolist_integer(infolist, 'number')
|
||||
full_name = weechat.infolist_string(infolist, 'full_name')
|
||||
if not full_name:
|
||||
full_name = '%s.%s' % (
|
||||
weechat.infolist_string(infolist, 'plugin_name'),
|
||||
weechat.infolist_string(infolist, 'name'))
|
||||
pointer = weechat.infolist_pointer(infolist, 'pointer')
|
||||
matching = name.lower().find(strinput) >= 0
|
||||
if not matching and strinput[-1] == ' ':
|
||||
matching = name.lower().endswith(strinput.strip())
|
||||
if not matching and strinput.isdigit():
|
||||
matching = str(number).startswith(strinput)
|
||||
if len(strinput) == 0 or matching:
|
||||
listbuf.append({
|
||||
'number': number,
|
||||
'short_name': short_name,
|
||||
'name': name,
|
||||
'full_name': full_name,
|
||||
'pointer': pointer,
|
||||
})
|
||||
weechat.infolist_free(infolist)
|
||||
|
||||
# sort buffers
|
||||
hotlist = []
|
||||
infolist = weechat.infolist_get('hotlist', '', '')
|
||||
while weechat.infolist_next(infolist):
|
||||
hotlist.append(
|
||||
weechat.infolist_pointer(infolist, 'buffer_pointer'))
|
||||
weechat.infolist_free(infolist)
|
||||
last_index_hotlist = len(hotlist)
|
||||
|
||||
def _sort_name(buf):
|
||||
"""Sort buffers by name (or short name)."""
|
||||
return buf['name']
|
||||
|
||||
def _sort_hotlist(buf):
|
||||
"""Sort buffers by hotlist order."""
|
||||
try:
|
||||
return hotlist.index(buf['pointer'])
|
||||
except ValueError:
|
||||
# not in hotlist, always last.
|
||||
return last_index_hotlist
|
||||
|
||||
def _sort_match_number(buf):
|
||||
"""Sort buffers by match on number."""
|
||||
return 0 if str(buf['number']) == strinput else 1
|
||||
|
||||
def _sort_match_beginning(buf):
|
||||
"""Sort buffers by match at beginning."""
|
||||
return 0 if go_match_beginning(buf, strinput) else 1
|
||||
|
||||
funcs = {
|
||||
'name': _sort_name,
|
||||
'hotlist': _sort_hotlist,
|
||||
'number': _sort_match_number,
|
||||
'beginning': _sort_match_beginning,
|
||||
}
|
||||
|
||||
for key in weechat.config_get_plugin('sort').split(','):
|
||||
if key in funcs:
|
||||
listbuf = sorted(listbuf, key=funcs[key])
|
||||
|
||||
if not strinput:
|
||||
index = [i for i, buf in enumerate(listbuf)
|
||||
if buf['pointer'] == weechat.current_buffer()]
|
||||
if index:
|
||||
buffers_pos = index[0]
|
||||
|
||||
return listbuf
|
||||
|
||||
|
||||
def go_buffers_to_string(listbuf, pos, strinput):
|
||||
"""Return string built with list of buffers found (matching user input)."""
|
||||
string = ''
|
||||
strinput = strinput.lower()
|
||||
for i in range(len(listbuf)):
|
||||
selected = '_selected' if i == pos else ''
|
||||
index = listbuf[i]['name'].lower().find(strinput)
|
||||
if index >= 0:
|
||||
index2 = index + len(strinput)
|
||||
name = '%s%s%s%s%s' % (
|
||||
listbuf[i]['name'][:index],
|
||||
weechat.color(weechat.config_get_plugin(
|
||||
'color_name_highlight' + selected)),
|
||||
listbuf[i]['name'][index:index2],
|
||||
weechat.color(weechat.config_get_plugin(
|
||||
'color_name' + selected)),
|
||||
listbuf[i]['name'][index2:])
|
||||
else:
|
||||
name = listbuf[i]['name']
|
||||
string += ' %s%s%s%s%s' % (
|
||||
weechat.color(weechat.config_get_plugin(
|
||||
'color_number' + selected)),
|
||||
str(listbuf[i]['number']),
|
||||
weechat.color(weechat.config_get_plugin(
|
||||
'color_name' + selected)),
|
||||
name,
|
||||
weechat.color('reset'))
|
||||
return ' ' + string if string else ''
|
||||
|
||||
|
||||
def go_input_modifier(data, modifier, modifier_data, string):
|
||||
"""This modifier is called when input text item is built by WeeChat.
|
||||
|
||||
This is commonly called after changes in input or cursor move: it builds
|
||||
a new input with prefix ("Go to:"), and suffix (list of buffers found).
|
||||
"""
|
||||
global old_input, buffers, buffers_pos
|
||||
if modifier_data != weechat.current_buffer():
|
||||
return ''
|
||||
names = ''
|
||||
new_input = weechat.string_remove_color(string, '')
|
||||
new_input = new_input.lstrip()
|
||||
if old_input is None or new_input != old_input:
|
||||
old_buffers = buffers
|
||||
buffers = go_matching_buffers(new_input)
|
||||
if buffers != old_buffers and len(new_input) > 0:
|
||||
if len(buffers) == 1 and go_option_enabled('auto_jump'):
|
||||
weechat.command(modifier_data, '/wait 1ms /input return')
|
||||
buffers_pos = 0
|
||||
old_input = new_input
|
||||
names = go_buffers_to_string(buffers, buffers_pos, new_input.strip())
|
||||
return weechat.config_get_plugin('message') + string + names
|
||||
|
||||
|
||||
def go_command_run_input(data, buf, command):
|
||||
"""Function called when a command "/input xxx" is run."""
|
||||
global buffers, buffers_pos
|
||||
if command == '/input search_text' or command.find('/input jump') == 0:
|
||||
# search text or jump to another buffer is forbidden now
|
||||
return weechat.WEECHAT_RC_OK_EAT
|
||||
elif command == '/input complete_next':
|
||||
# choose next buffer in list
|
||||
buffers_pos += 1
|
||||
if buffers_pos >= len(buffers):
|
||||
buffers_pos = 0
|
||||
weechat.hook_signal_send('input_text_changed',
|
||||
weechat.WEECHAT_HOOK_SIGNAL_STRING, '')
|
||||
return weechat.WEECHAT_RC_OK_EAT
|
||||
elif command == '/input complete_previous':
|
||||
# choose previous buffer in list
|
||||
buffers_pos -= 1
|
||||
if buffers_pos < 0:
|
||||
buffers_pos = len(buffers) - 1
|
||||
weechat.hook_signal_send('input_text_changed',
|
||||
weechat.WEECHAT_HOOK_SIGNAL_STRING, '')
|
||||
return weechat.WEECHAT_RC_OK_EAT
|
||||
elif command == '/input return':
|
||||
# switch to selected buffer (if any)
|
||||
go_end(buf)
|
||||
if len(buffers) > 0:
|
||||
weechat.command(
|
||||
buf, '/buffer ' + str(buffers[buffers_pos]['full_name']))
|
||||
return weechat.WEECHAT_RC_OK_EAT
|
||||
return weechat.WEECHAT_RC_OK
|
||||
|
||||
|
||||
def go_command_run_buffer(data, buf, command):
|
||||
"""Function called when a command "/buffer xxx" is run."""
|
||||
return weechat.WEECHAT_RC_OK_EAT
|
||||
|
||||
|
||||
def go_command_run_window(data, buf, command):
|
||||
"""Function called when a command "/window xxx" is run."""
|
||||
return weechat.WEECHAT_RC_OK_EAT
|
||||
|
||||
|
||||
def go_unload_script():
|
||||
"""Function called when script is unloaded."""
|
||||
go_unhook_all()
|
||||
return weechat.WEECHAT_RC_OK
|
||||
|
||||
|
||||
def go_main():
|
||||
"""Entry point."""
|
||||
if not weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION,
|
||||
SCRIPT_LICENSE, SCRIPT_DESC,
|
||||
'go_unload_script', ''):
|
||||
return
|
||||
weechat.hook_command(
|
||||
SCRIPT_COMMAND,
|
||||
'Quick jump to buffers', '[name]',
|
||||
'name: directly jump to buffer by name (without argument, list is '
|
||||
'displayed)\n\n'
|
||||
'You can bind command to a key, for example:\n'
|
||||
' /key bind meta-g /go\n\n'
|
||||
'You can use completion key (commonly Tab and shift-Tab) to select '
|
||||
'next/previous buffer in list.',
|
||||
'%(buffers_names)',
|
||||
'go_cmd', '')
|
||||
|
||||
# set default settings
|
||||
version = weechat.info_get('version_number', '') or 0
|
||||
for option, value in SETTINGS.items():
|
||||
if not weechat.config_is_set_plugin(option):
|
||||
weechat.config_set_plugin(option, value[0])
|
||||
if int(version) >= 0x00030500:
|
||||
weechat.config_set_desc_plugin(
|
||||
option, '%s (default: "%s")' % (value[1], value[0]))
|
||||
weechat.hook_info('go_running',
|
||||
'Return "1" if go is running, otherwise "0"',
|
||||
'',
|
||||
'go_info_running', '')
|
||||
|
||||
|
||||
if __name__ == "__main__" and IMPORT_OK:
|
||||
go_main()
|
@ -26,7 +26,7 @@ buffer_search_case_sensitive = off
|
||||
buffer_search_force_default = off
|
||||
buffer_search_regex = off
|
||||
buffer_search_where = message
|
||||
buffer_time_format = " "
|
||||
buffer_time_format = "%H:%M:%S"
|
||||
color_basic_force_bold = off
|
||||
color_inactive_buffer = on
|
||||
color_inactive_message = on
|
||||
@ -123,19 +123,19 @@ chat_buffer = white
|
||||
chat_channel = white
|
||||
chat_day_change = cyan
|
||||
chat_delimiters = green
|
||||
chat_highlight = yellow
|
||||
chat_highlight = default
|
||||
chat_highlight_bg = magenta
|
||||
chat_host = cyan
|
||||
chat_inactive_buffer = default
|
||||
chat_inactive_window = default
|
||||
chat_nick = lightcyan
|
||||
chat_nick_colors = "cyan,magenta,green,brown,lightblue,default,lightcyan,lightmagenta,lightgreen,blue"
|
||||
chat_nick_colors = "cyan,green,brown,lightblue,lightred,lightcyan,lightgreen,yellow"
|
||||
chat_nick_offline = default
|
||||
chat_nick_offline_highlight = default
|
||||
chat_nick_offline_highlight_bg = blue
|
||||
chat_nick_other = cyan
|
||||
chat_nick_prefix = green
|
||||
chat_nick_self = white
|
||||
chat_nick_self = magenta
|
||||
chat_nick_suffix = green
|
||||
chat_prefix_action = white
|
||||
chat_prefix_buffer = brown
|
||||
@ -152,7 +152,7 @@ chat_server = brown
|
||||
chat_tags = red
|
||||
chat_text_found = yellow
|
||||
chat_text_found_bg = lightmagenta
|
||||
chat_time = default
|
||||
chat_time = darkgray
|
||||
chat_time_delimiters = brown
|
||||
chat_value = cyan
|
||||
emphasized = yellow
|
||||
@ -513,6 +513,7 @@ meta-j99 = "/buffer 99"
|
||||
meta-k = "/input grab_key_command"
|
||||
meta-m = "/mute mouse toggle"
|
||||
meta-n = "/window scroll_next_highlight"
|
||||
meta-o = "/go"
|
||||
meta-p = "/window scroll_previous_highlight"
|
||||
meta-r = "/input delete_line"
|
||||
meta-s = "/mute aspell toggle"
|
||||
|
Loading…
Reference in New Issue
Block a user