source: npl/overig/irssi/hostcolor.pl

use strict;
use Irssi 20020101.0250 ();
use vars qw($VERSION %IRSSI); 
$VERSION = "1";
%IRSSI = (
    authors     => "Timo Sirainen, Ian Peters (Modfied by Edwin Eefting aka Psy)",
    contact     => "tss\@iki.fi (psy\@datux.nl)", 
    name        => "HostColor (based on nickcolor.pl)",
    description => "assign a different color for each hostname",
    license     => "Public Domain",
    url         => "http://www.datux.nl/",
    changed     => "2003-30-07T3:47+0100"
);

# 30-jul 2003, changed by psy@datux.nl (#leip @ EFNET)
# -colors are now assigned on hostmask. (no need for nick tracking anymore!)
#  So people will always have their own color, even if they join with a different nick.
# -session colors are saved as well. (automaticly)
# -much more colors, also using background color. (see table below)
# 
# This should be a default feature of irssi. ;)
# Just try it, colored nicks is the natural way for our brain to 
# follow multiple conversations at once. (especially when your
# drunk or stoned!) 
############## #leip @ EFNET the stonerz chan


# hm.. i should make it possible to use the existing one..
Irssi::theme_register([
  'pubmsg_hilight', '{pubmsghinick $0 $3 $1}$2'
]);

my %saved_colors;
my %session_colors ;

#list of cool colors that will be automatcily assigned
my @colors = qw/18 19 20 21 22 23 24 25 26 27 28 29 30 31 
                32 33 35 36 37 39 40 41 42 43 44 45 47
                48 49 50 53 56 57 59 61 62 63
                80 81 82 84 88 89 90 91 93 95
                96 97 98 100 104 105 106 107 109 111
                112 113 114 117 120 121 123 125 127
                160 161 162 165 166 167 168 169 171 172 174
                240 241 242 243 245 246 247 248 250 251 252 254/;

sub load_colors {
  open COLORS, "$ENV{HOME}/.irssi/saved_colors";
  while (<COLORS>) {
    chomp;
    my $line =$_;
    my($nick, $color) = split ":", $line;
    $saved_colors{$nick} = $color;
  }
  close COLORS;

}

sub save_colors {
  open COLORS, ">$ENV{HOME}/.irssi/saved_colors";
  foreach my $nick (keys %saved_colors) {
    print COLORS "$nick:$saved_colors{$nick}\n";
  }
  close COLORS;

}


# This gave reasonable distribution values when run across
# /usr/share/dict/words

sub simple_hash {
  my ($string) = @_;
  chomp $string;
  my @chars = split //, $string;
  my $counter;

  foreach my $char (@chars) {
    $counter += ord $char;
  }

 # $counter = $colors[$counter % 11];
 $counter =$counter % scalar(@colors);
 $counter = $colors[$counter];
 return $counter;
}

sub colorcode
{
    my $colornr=shift;
    return chr(3).sprintf("%02d,%02d",$colornr%16, $colornr/16);

}

# FIXME: breaks /HILIGHT etc.
sub sig_public {
  my ($server, $msg, $nick, $address, $target) = @_;
  my $chanrec = $server->channel_find($target);
  return if not $chanrec;
  my $nickrec = $chanrec->nick_find($nick);
  return if not $nickrec;
  my $nickmode = $nickrec->{op} ? "@" : $nickrec->{voice} ? "+" : "";
    
  # Has the user assigned this nick a color?
  my $color = $saved_colors{$address};

  # Have -we- already assigned this nick a color?
  if (!$color) {
    $color = $session_colors{$address};
  }

  # Let's assign this nick a color
  if (!$color) {
    $color = simple_hash $address;
    $session_colors{$address} = $color;
    #save automaticly
    save_colors;
  }

  $server->command('/^format pubmsg {pubmsgnick $2 {pubnick '.colorcode($color).'$0}}$1');
}

# FIXME: breaks /HILIGHT etc.
sub sig_publicaction {
  my ($server, $msg, $nick, $address, $target) = @_;
  my $chanrec = $server->channel_find($target);
  return if not $chanrec;
  my $nickrec = $chanrec->nick_find($nick);
  return if not $nickrec;
  my $nickmode = $nickrec->{op} ? "@" : $nickrec->{voice} ? "+" : "";
    
  # Has the user assigned this nick a color?
  my $color = $saved_colors{$address};

  # Have -we- already assigned this nick a color?
  if (!$color) {
    $color = $session_colors{$address};
  }

  # Let's assign this nick a color
  if (!$color) {
    $color = simple_hash $address;
    $session_colors{$address} = $color;
  }

  $server->command('/^format pubmsg {pubmsgnick $2 {pubnick '.colorcode($color).'$0}}$1');
}



sub cmd_color {
  my ($data, $server, $witem) = @_;
  my ($op, $nick, $color) = split " ", $data;

  $op = lc $op;

  if (!$op) {
    Irssi::print ("No operation given");
  } elsif ($op eq "save") {
    save_colors;
  } elsif ($op eq "set") {
    if (!$nick) {
      Irssi::print ("Host not given");
    } elsif (!$color) {
      Irssi::print ("Color not given");
    } elsif ($color < 1 || $color > 256) {
      Irssi::print ("Color must be between 1 and 256 inclusive");
    } else {
      $saved_colors{$nick} = $color;
      save_colors; #i hate manual saving :D
    }
  } elsif ($op eq "clear") {
    if (!$nick) {
      Irssi::print ("Host not given");
    } else {
      delete ($saved_colors{$nick});
    }
  } elsif ($op eq "list") {
    Irssi::print ("\nSaved Colors:");
    foreach my $nick (keys %saved_colors) {
      Irssi::print (colorcode($saved_colors{$nick})."$nick" .
                    chr (3) . "0,1 ($saved_colors{$nick})");
    }
    Irssi::print ("\nAutomaticly assigned colors:");
    foreach my $nick (keys %session_colors) {
      Irssi::print (colorcode($session_colors{$nick})."$nick" .
                    chr (3) . "0,1 ($session_colors{$nick})");
    }
  } elsif ($op eq "preview") {
    Irssi::print ("\nAvailable colors:");
    foreach my $i (0..(16*16)) {
      Irssi::print (colorcode($i). "Color #$i");
    }
  }
}

Irssi::print(chr(3)."03Chill mode activated - nicks are now colored. Psy #datux \@ FREENODE \\|/");

load_colors;

Irssi::command_bind('color', 'cmd_color');

Irssi::signal_add('message public', 'sig_public');
#Irssi::signal_add('ctcp action', 'sig_publicaction');
#Irssi::signal_add('event nick', 'sig_nick');