Onis::Plugins::Conversations has mostly been rewritten.

diff --git a/lib/Onis/Plugins/Conversations.pm b/lib/Onis/Plugins/Conversations.pm
new file mode 100644 (file)
index 0000000..5af0717
--- /dev/null
+++ b/lib/Onis/Plugins/Conversations.pm
@@ -0,0 +1,243 @@
+package Onis::Plugins::Conversations;
+
+use strict;
+use warnings;
+
+use Onis::Config qw(get_config);
+use Onis::Html qw(get_filehandle);
+use Onis::Language qw(translate);
+use Onis::Data::Core qw(register_plugin get_main_nick nick_to_ident);
+use Onis::Users qw(ident_to_name);
+use Onis::Data::Persistent;
+
+our $ConversationCache = Onis::Data::Persistent->new ('ConversationCache', 'partners', qw(time0 time1 time2 time3));
+our $ConversationData = {};
+
+our @H_IMAGES = qw#dark-theme/h-red.png dark-theme/h-blue.png dark-theme/h-yellow.png dark-theme/h-green.png#;
+our $BAR_WIDTH  = 100;
+
+if (get_config ('horizontal_images'))
+{
+       my @tmp = get_config ('horizontal_images');
+       my $i;
+       
+       if (scalar (@tmp) != 4)
+       {
+               print STDERR $/, __FILE__, ": The number of horizontal images is not four. The output might look weird.", $/;
+       }
+
+       for ($i = 0; $i < 4; $i++)
+       {
+               next unless (defined ($tmp[$i]));
+               $H_IMAGES[$i] = $tmp[$i];
+       }
+}
+if (get_config ('bar_width'))
+{
+       my $tmp = get_config ('bar_width');
+       $tmp =~ s/\D//g;
+       $BAR_WIDTH = 2 * $tmp if ($tmp >= 10);
+}
+
+register_plugin ('TEXT', \&add);
+register_plugin ('OUTPUT', \&output);
+
+my $VERSION = '$Id: Conversations.pm,v 1.7 2004/09/15 19:42:04 octo Exp $';
+print STDERR $/, __FILE__, ": $VERSION" if ($::DEBUG);
+
+return (1);
+
+sub add
+{
+       my $data = shift;
+       my $text = $data->{'text'};
+       my $nick = $data->{'nick'};
+       my $ident = $data->{'ident'};
+
+       my $time = int ($data->{'hour'} / 6);
+
+       # <taken from lib/Onis/Plugins/Nicks.pm>
+       my @potential_nicks = split (/[^\w\`\~\^\-\|\[\]]+/, $text);
+       my $talk_to = '';
+       
+       for (@potential_nicks)
+       {
+               my $other_nick = $_;
+               my $other_ident = nick_to_ident ($other_nick);
+               
+               if ($other_ident)
+               {
+                       $talk_to = $other_nick;
+                       last;
+               }
+       }
+       # </taken>
+       
+       if ($talk_to)
+       {
+               my $key = "$nick:$talk_to";
+               my @data = $ConversationCache->get ($key);
+               @data = (0, 0, 0, 0) unless (@data);
+
+               my $chars = length ($text);
+
+               $data[$time] += $chars;
+               
+               $ConversationCache->put ($key, @data);
+       }
+}
+
+sub calculate
+{
+       for ($ConversationCache->keys ())
+       {
+               my $key = $_;
+               my ($nick_from, $nick_to) = split (m/:/, $key);
+               my @data = $ConversationCache->get ($key);
+
+               $nick_from = get_main_nick ($nick_from);
+               $nick_to   = get_main_nick ($nick_to);
+
+               next if (!$nick_from or !$nick_to or ($nick_from eq $nick_to));
+
+               if ($ConversationData->{$nick_from}{$nick_to})
+               {
+                       $ConversationData->{$nick_from}{$nick_to} =
+                       {
+                               total => 0,
+                               nicks =>
+                               {
+                                       $nick_from => [0, 0, 0, 0],
+                                       $nick_to   => [0, 0, 0, 0]
+                               }
+                       };
+                       $ConversationData->{$nick_to}{$nick_from} = $ConversationData->{$nick_from}{$nick_to};
+               }
+
+               for (my $i = 0; $i < 4; $i++)
+               {
+                       $ConversationData->{$nick_from}{$nick_to}{'nicks'}{$nick_from}[$i] += $data[$i];
+                       $ConversationData->{$nick_from}{$nick_to}{'total'} += $data[$i];
+               }
+       }
+}
+
+sub get_top
+{
+       my $num = shift;
+       my @data = ();
+
+       for (keys %$ConversationData)
+       {
+               my $nick0 = $_;
+
+               for (keys %{$ConversationData->{$nick0}})
+               {
+                       my $nick1 = $_;
+                       next unless ($nick0 lt $nick1);
+
+                       push (@data, [$ConversationData->{$nick0}{$nick1}{'total'}, $nick0, $nick1]);
+               }
+       }
+
+       @data = sort { $b->[0] <=> $a->[0] } (@data);
+       splice (@data, $num) if (scalar (@data) > $num);
+
+       return (@data);
+}
+
+sub output
+{
+       calculate ();
+
+       my $fh = get_filehandle ();
+       my $title = translate ('Conversation partners');
+
+       my $max_num = 0;
+       my $factor = 0;
+
+       my @img = get_config ('horizontal_images');
+
+       my @data = get_top (10);
+       return (undef) unless (@data);
+
+       for (@data)
+       {
+               my $nick0 = $_->[1];
+               my $nick1 = $_->[2];
+               my $rec = $ConversationData->{$nick0}{$nick1};
+
+               my $sum0 = 0;
+               my $sum1 = 0;
+
+               for (my $i = 0; $i < 4; $i++)
+               {
+                       $sum0 += $rec->{'users'}{$nick0}[$i];
+                       $sum1 += $rec->{'users'}{$nick1}[$i];
+               }
+
+               $max_num = $sum0 if ($max_num < $sum0);
+               $max_num = $sum1 if ($max_num < $sum1);
+       }
+       
+       $factor = $BAR_WIDTH / $max_num;
+
+       print $fh <<EOF;
+<table class="plugin conversations">
+  <tr>
+    <th colspan="2">$title</th>
+  </tr>
+EOF
+       foreach (@data)
+       {
+               my $nick0 = $_->[1];
+               my $nick1 = $_->[2];
+               my $name0 = nick_to_name ($nick0) || $nick0;
+               my $name1 = nick_to_name ($nick1) || $nick1;
+               my $rec = $ConversationData->{$nick0}{$nick1};
+
+               print $fh <<EOF;
+  <tr>
+    <td class="nick left">$name0</td>
+    <td class="nick right">$name1</td>
+  </tr>
+  <tr>
+EOF
+
+               print $fh '    <td class="bar left">';
+               for (3, 2, 1, 0)
+               {
+                       my $i = $img[$_];
+                       my $w = int (0.5 + ($rec->{'users'}{$nick0}[$_] * $factor));
+                       my $c = '';
+                       $w ||= 1;
+
+                       $w = $w . 'px';
+
+                       if    ($_ == 3) { $c = qq# class="first"#; }
+                       elsif ($_ == 0) { $c = qq# class="last"#;  }
+
+                       print $fh qq#<img src="$i" style="width: $w;"$c alt="" />#;
+               }
+
+               print $fh qq#</td>\n    <td class="bar right">#;
+
+               for (0, 1, 2, 3)
+               {
+                       my $i = $img[$_];
+                       my $w = int (0.5 + ($rec->{'users'}{$nick1}[$_] * $factor));
+                       my $c = '';
+                       $w ||= 1;
+
+                       $w = $w . 'px';
+
+                       if    ($_ == 0) { $c = qq# class="first"#; }
+                       elsif ($_ == 3) { $c = qq# class="last"#;  }
+
+                       print $fh qq#<img src="$i" style="width: $w;"$c alt=""/>#;
+               }
+               print $fh "</td>\n  </tr>\n";
+       }
+
+       print $fh "</table>\n\n";
+}