diff options
Diffstat (limited to 'scripts/colorize_nicks.pl')
-rw-r--r-- | scripts/colorize_nicks.pl | 217 |
1 files changed, 217 insertions, 0 deletions
diff --git a/scripts/colorize_nicks.pl b/scripts/colorize_nicks.pl new file mode 100644 index 0000000..afeb707 --- /dev/null +++ b/scripts/colorize_nicks.pl @@ -0,0 +1,217 @@ +use strict; +use warnings; + +our $VERSION = '0.4.1'; # ed9cb119fc4b3d1 +our %IRSSI = ( + authors => 'Nei', + contact => 'Nei @ anti@conference.jabber.teamidiot.de', + url => "http://anti.teamidiot.de/", + name => 'colorize_nicks', + description => 'Colourise mention of nicks in the message body.', + license => 'GNU GPLv2 or later', + ); + +# inspired by mrwright's nickcolor.pl and xt's colorize_nicks.pl +# +# you need nickcolor_expando or another nickcolor script providing the +# get_nick_color2 function + +# Usage +# ===== +# should start working once loaded + +# Options +# ======= +# /set colorize_nicks_skip_formats <num> +# * how many forms (blocks of irssi format codes or non-letters) to +# skip at the beginning of line before starting to colourise nicks +# (you usually want to skip the speaker's nick itself and the +# timestamp) +# +# /set colorize_nicks_ignore_list <words to ignore> +# * list of nicks (words) that should never be coloured +# +# /set colorize_nicks_repeat_formats <ON|OFF> +# * repeat the format stack from the beginning of line, enable when +# using per-line colours and colorize_nicks breaks it + +# Commands +# ======== +# you can use this alias: +# +# /alias nocolorize set colorize_nicks_ignore_list $colorize_nicks_ignore_list +# +# /nocolorize <nick> +# * quickly add nick to the bad word list of nicks that should not be +# colourised + +no warnings 'redefine'; +use Irssi; + +my $irssi_mumbo = qr/\cD[`-i]|\cD[&-@\xff]./; + +my $nickchar = qr/[\]\[[:alnum:]\\|`^{}_-]/; +my $nick_pat = qr/($nickchar+)/; + +my @ignore_list; + +my $colourer_script; + +sub _find_colourer { + my $colourer; + unless ($colourer_script + && ($colourer = "Irssi::Script::$colourer_script"->can('get_nick_color2'))) { + for my $script (sort map { s/::$//r } keys %Irssi::Script::) { + if ($colourer = "Irssi::Script::$script"->can('get_nick_color2')) { + $colourer_script = $script; + last; + } + } + } + $colourer +} + +sub _get_chanref { + my ($dest) = @_; + return unless $dest->{level} & MSGLEVEL_PUBLIC; + return unless defined $dest->{target}; + return unless ref $dest->{server}; + $dest->{server}->channel_find($dest->{target}) +} + +sub _colourise_nicks { + my ($dest, $chanref, $colourer, @nicks) = @_; + + my %nicks = map { $_->[0] => $colourer->($dest->{server}{tag}, $chanref->{name}, $_->[1], 1) } + grep { defined } + map { if (my $nr = $chanref->nick_find($_)) { + [ $_ => $nr->{nick} ] + } } + keys %{ +{ map { $_ => undef } @nicks } }; + delete @nicks{ @ignore_list }; + + my $nick_re = join '|', map { quotemeta } sort { length $b <=> length $a } grep { length $nicks{$_} } keys %nicks; + + (\%nicks, $nick_re) +} + +sub _colourise_form { + my ( $text, + $skip, + $nicks, + $nick_re ) = @_; + return if $skip < 0; + + my $repeat = Irssi::settings_get_bool('colorize_nicks_repeat_formats'); + + my @forms = split /((?:$irssi_mumbo|\s|[.,*@%+&!#$()=~'";:?\/><]+(?=$irssi_mumbo|\s))+)/, $text, -1; + my $ret = ''; + my $fmtstack = ''; + while (@forms) { + my ($t, $form) = splice @forms, 0, 2; + if ($skip > 0) { + --$skip; + $ret .= $t; + $ret .= $form if defined $form; + if ($repeat) { + $fmtstack .= join '', $form =~ /$irssi_mumbo/g if defined $form; + $fmtstack =~ s/\cDe//g; + } + } + elsif (length $nick_re + && $t =~ s/((?:^|\s)\W{0,3}?)(?<!$nickchar|')($nick_re)(?!$nickchar)/$1$nicks->{$2}$2\cDg$fmtstack/g) { + $ret .= "$t\cDg$fmtstack"; + $ret .= $form if defined $form; + $fmtstack .= join '', $form =~ /$irssi_mumbo/g if defined $form; + $fmtstack =~ s/\cDe//g; + } + else { + $ret .= $t; + $ret .= $form if defined $form; + } + } + + $ret +} + +# TXT_OWN_MSG, server->nick, msg, nickmode +# TXT_OWN_MSG_CHANNEL, server->nick, target, msg, nickmode +# TXT_PUBMSG_HILIGHT, color, printnick, msg, nickmode +# TXT_PUBMSG_HILIGHT_CHANNEL, color, printnick, target, msg, nickmode +# for_me ? TXT_PUBMSG_ME : TXT_PUBMSG, printnick, msg, nickmode +# for_me ? TXT_PUBMSG_ME_CHANNEL : TXT_PUBMSG_CHANNEL, printnick, target, msg, nickmode +sub prt_format_issue { + my ( $theme, + $module, + $dest, + $format, + @args + ) = @_; + my $chanref = _get_chanref($dest); + return unless $chanref; + my $colourer = _find_colourer(); + return unless $colourer; + + my $arg = 1; + $arg++ if $format =~ /_channel/; + $arg++ if $format =~ /_hilight/; + return unless @args > $arg; + + utf8::decode($args[$arg]); + my $text = $args[$arg]; + my $stripped = Irssi::strip_codes($text); + + utf8::decode($stripped); + my ($nicks, $nick_re) = _colourise_nicks($dest, $chanref, $colourer, $stripped =~ /$nick_pat/g); + return unless $nicks; + + $args[$arg] = _colourise_form($text, 0, $nicks, $nick_re); + Irssi::signal_continue($theme, $module, $dest, $format, @args) + if defined $args[$arg] && $args[$arg] ne $text; +} + +sub prt_text_issue { + my ( $dest, + $text, + $stripped + ) = @_; + my $chanref = _get_chanref($dest); + return unless $chanref; + my $colourer = _find_colourer(); + return unless $colourer; + + utf8::decode($text); + utf8::decode($stripped); + my ($nicks, $nick_re) = _colourise_nicks($dest, $chanref, $colourer, $stripped =~ /$nick_pat/g); + return unless $nicks; + + my $skip = Irssi::settings_get_int('colorize_nicks_skip_formats'); + my $ret = _colourise_form($text, $skip, $nicks, $nick_re); + Irssi::signal_continue($dest, $ret, $stripped) + if defined $ret && $ret ne $text; +} + +sub setup_changed { + @ignore_list = split /\s+|,/, Irssi::settings_get_str('colorize_nicks_ignore_list'); +} + +sub init { + setup_changed(); +} + +if ((Irssi::parse_special('$abiversion')||0) >= 28) { + Irssi::signal_add( + 'print format' => 'prt_format_issue' + ); +} else { + Irssi::signal_add( + 'print text' => 'prt_text_issue' + ); + Irssi::settings_add_int('colorize_nicks', 'colorize_nicks_skip_formats' => 2); +} +Irssi::signal_add_last('setup changed' => 'setup_changed'); + +Irssi::settings_add_str('colorize_nicks', 'colorize_nicks_ignore_list' => ''); +Irssi::settings_add_bool('colorize_nicks', 'colorize_nicks_repeat_formats' => 0); + +init(); |