/[irc-logger]/trunk/bin/irc-logger.pl
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /trunk/bin/irc-logger.pl

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 10 by dpavlin, Thu Mar 2 00:19:12 2006 UTC revision 15 by dpavlin, Mon Mar 13 12:56:26 2006 UTC
# Line 35  my %FOLLOWS = Line 35  my %FOLLOWS =
35    
36  my $DSN = 'DBI:Pg:dbname=irc-logger';  my $DSN = 'DBI:Pg:dbname=irc-logger';
37    
38    my $ENCODING = 'ISO-8859-2';
39    
40  ## END CONFIG  ## END CONFIG
41    
42    
43    
44  use POE qw(Component::IRC Wheel::FollowTail);  use POE qw(Component::IRC Wheel::FollowTail Component::Server::HTTP);
45    use HTTP::Status;
46  use DBI;  use DBI;
47  use Encode qw/from_to/;  use Encode qw/from_to/;
48    use Regexp::Common qw /URI/;
49    
50    
51  my $dbh = DBI->connect($DSN,"","", { RaiseError => 1, AutoCommit => 1 }) || die $DBI::errstr;  my $dbh = DBI->connect($DSN,"","", { RaiseError => 1, AutoCommit => 1 }) || die $DBI::errstr;
# Line 72  insert into log Line 76  insert into log
76  values (?,?,?)  values (?,?,?)
77  });  });
78    
79    =head2 get_from_log
80    
81     my @messages = get_from_log(
82            limit => 42,
83            search => '%what to stuff in ilike%',
84            fmt => {
85                    time => '{%s} ',
86                    time_channel => '{%s %s} ',
87                    nick => '%s: ',
88                    message => '%s',
89            },
90            message_filter => sub {
91                    # modify message content
92                    return shift;
93            }
94     );
95    
96    =cut
97    
98    sub get_from_log {
99            my $args = {@_};
100    
101            $args->{limit} ||= 10;
102    
103            $args->{fmt} ||= {
104                    time => '{%s} ',
105                    time_channel => '{%s %s} ',
106                    nick => '%s: ',
107                    message => '%s',
108            };
109    
110            my $sql = qq{
111                    select
112                            time::date as date,
113                            time::time as time,
114                            channel,
115                            nick,
116                            message
117                    from log
118            };
119            $sql .= " where message ilike ? " if ($args->{search});
120            $sql .= " order by log.time desc";
121            $sql .= " limit " . $args->{limit};
122    
123            my $sth = $dbh->prepare( $sql );
124            if ($args->{search}) {
125                    $sth->execute( $args->{search} );
126            } else {
127                    $sth->execute();
128            }
129            my $last_row = {
130                    date => '',
131                    time => '',
132                    channel => '',
133                    nick => '',
134            };
135    
136            my @rows;
137    
138            while (my $row = $sth->fetchrow_hashref) {
139                    unshift @rows, $row;
140            }
141    
142            my @msgs;
143    
144            foreach my $row (@rows) {
145    
146                    $row->{time} =~ s#\.\d+##;
147    
148                    my $t;
149                    $t = $row->{date} . ' ' if ($last_row->{date} ne $row->{date});
150                    $t .= $row->{time};
151    
152                    my $msg = '';
153    
154                    if ($last_row->{channel} ne $row->{channel}) {
155                            $msg .= sprintf($args->{fmt}->{time_channel}, $t, $row->{channel});
156                    } else {
157                            $msg .= sprintf($args->{fmt}->{time}, $t);
158                    }
159    
160                    my $append = 1;
161    
162                    if ($last_row->{nick} ne $row->{nick}) {
163                            $msg .= sprintf($args->{fmt}->{nick}, $row->{nick});
164                            $append = 0;
165                    }
166    
167                    if (ref($args->{message_filter}) eq 'CODE') {
168                            $msg .= sprintf($args->{fmt}->{message},
169                                    $args->{message_filter}->(
170                                            $row->{message}
171                                    )
172                            );
173                    } else {
174                            $msg .= sprintf($args->{fmt}->{message}, $row->{message});
175                    }
176    
177                    if ($append && @msgs) {
178                            $msgs[$#msgs] .= " " . $msg;
179                    } else {
180                            push @msgs, $msg;
181                    }
182    
183                    $last_row = $row;
184            }
185    
186            return @msgs;
187    }
188    
189    
190  my $SKIPPING = 0;               # if skipping, how many we've done  my $SKIPPING = 0;               # if skipping, how many we've done
191  my $SEND_QUEUE;                 # cache  my $SEND_QUEUE;                 # cache
# Line 89  POE::Session->create Line 203  POE::Session->create
203                  $_[KERNEL]->post($IRC_ALIAS => join => '#logger');                  $_[KERNEL]->post($IRC_ALIAS => join => '#logger');
204                  $_[KERNEL]->yield("heartbeat"); # start heartbeat                  $_[KERNEL]->yield("heartbeat"); # start heartbeat
205  #               $_[KERNEL]->yield("my_add", $_) for keys %FOLLOWS;  #               $_[KERNEL]->yield("my_add", $_) for keys %FOLLOWS;
206                    $_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "IDENTIFY $NICK" );
207      },      },
208      irc_public => sub {      irc_public => sub {
209                  my $kernel = $_[KERNEL];                  my $kernel = $_[KERNEL];
# Line 96  POE::Session->create Line 211  POE::Session->create
211                  my $channel = $_[ARG1]->[0];                  my $channel = $_[ARG1]->[0];
212                  my $msg = $_[ARG2];                  my $msg = $_[ARG2];
213    
214                  from_to($msg, 'UTF-8', 'ISO-8859-2');                  from_to($msg, 'UTF-8', $ENCODING);
215    
216                  print "$channel: <$nick> $msg\n";                  print "$channel: <$nick> $msg\n";
217                  $sth->execute($channel, $nick, $msg);                  $sth->execute($channel, $nick, $msg);
# Line 105  POE::Session->create Line 220  POE::Session->create
220                  my $kernel = $_[KERNEL];                  my $kernel = $_[KERNEL];
221                  my $nick = (split /!/, $_[ARG0])[0];                  my $nick = (split /!/, $_[ARG0])[0];
222                  my $msg = $_[ARG2];                  my $msg = $_[ARG2];
223                  from_to($msg, 'UTF-8', 'ISO-8859-2');                  from_to($msg, 'UTF-8', $ENCODING);
224    
225                  my $res = "unknown command '$msg', try /msg $NICK help!";                  my $res = "unknown command '$msg', try /msg $NICK help!";
226                    my @out;
227    
228                  print "<< $msg\n";                  print "<< $msg\n";
229    
230                  if ($msg =~ m/^help/i) {                  if ($msg =~ m/^help/i) {
231    
232                          $res = "usage: /msg $NICK stat - shows user statistics | /msg $NICK last - show backtrace of conversations";                          $res = "usage: /msg $NICK comand | commands: stat - user/message stat | last - show backtrace | grep foobar - find foobar";
233    
234                  } elsif ($msg =~ m/^msg\s+(\S+)\s+(.*)$/i) {                  } elsif ($msg =~ m/^msg\s+(\S+)\s+(.*)$/i) {
235    
# Line 137  POE::Session->create Line 253  POE::Session->create
253                          $res .= join(" | ", @users);                          $res .= join(" | ", @users);
254                  } elsif ($msg =~ m/^last.*?\s*(\d*)/i) {                  } elsif ($msg =~ m/^last.*?\s*(\d*)/i) {
255    
256                          my $nr = $1 || 10;                          foreach my $res (get_from_log( limit => $1 )) {
257                                    print "last: $res\n";
258                          my $sth = $dbh->prepare(qq{                                  from_to($res, $ENCODING, 'UTF-8');
259                                  select                                  $_[KERNEL]->post( $IRC_ALIAS => privmsg => $nick, $res );
                                         time::date as date,  
                                         time::time as time,  
                                         channel,  
                                         nick,  
                                         message  
                                 from log order by log.time desc limit $nr  
                         });  
                         $sth->execute();  
                         $res = "Last $nr messages: ";  
                         my $last_row = {  
                                 date => '',  
                                 time => '',  
                                 channel => '',  
                                 nick => '',  
                         };  
   
                         my @rows;  
   
                         while (my $row = $sth->fetchrow_hashref) {  
                                 unshift @rows, $row;  
260                          }                          }
261    
262                          my @msgs;                          $res = '';
   
                         foreach my $row (@rows) {  
   
                                 $row->{time} =~ s#\.\d+##;  
   
                                 my $t;  
                                 $t = $row->{date} . ' ' if ($last_row->{date} ne $row->{date});  
                                 $t .= $row->{time};  
   
                                 my $msg = '';  
   
                                 $msg .= "($t";  
                                 $msg .= ' ' . $row->{channel} if ($last_row->{channel} ne $row->{channel});  
                                 $msg .= ") ";  
   
                                 $msg .= $row->{nick} . ': '  if ($last_row->{nick} ne $row->{nick});  
   
                                 $msg .= $row->{message};  
263    
264                                  push @msgs, $msg;                  } elsif ($msg =~ m/^(search|grep)\s+(.*)$/i) {
265    
266                                  $last_row = $row;                          my $what = $2;
                         }  
267    
268                          foreach my $res (@msgs) {                          foreach my $res (get_from_log( limit => 20, search => "%${what}%" )) {
269                                  print "last: $res\n";                                  print "search [$what]: $res\n";
270                                  from_to($res, 'ISO-8859-2', 'UTF-8');                                  from_to($res, $ENCODING, 'UTF-8');
271                                  $_[KERNEL]->post( $IRC_ALIAS => privmsg => $nick, $res );                                  $_[KERNEL]->post( $IRC_ALIAS => privmsg => $nick, $res );
272                          }                          }
273    
274                          $res = '';                          $res = '';
275    
276                  }                  }
277    
278                  if ($res) {                  if ($res) {
279                          print ">> [$nick] $res\n";                          print ">> [$nick] $res\n";
280                          from_to($res, 'ISO-8859-2', 'UTF-8');                          from_to($res, $ENCODING, 'UTF-8');
281                          $_[KERNEL]->post( $IRC_ALIAS => privmsg => $nick, $res );                          $_[KERNEL]->post( $IRC_ALIAS => privmsg => $nick, $res );
282                  }                  }
283    
# Line 218  POE::Session->create Line 296  POE::Session->create
296                  warn "## indetify $NICK\n";                  warn "## indetify $NICK\n";
297                  $_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "IDENTIFY $NICK" );                  $_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "IDENTIFY $NICK" );
298          },          },
299          irc_433 => sub {  #       irc_433 => sub {
300                  print "# irc_433: ",$_[ARG1], "\n";  #               print "# irc_433: ",$_[ARG1], "\n";
301                  warn "## indetify $NICK\n";  #               warn "## indetify $NICK\n";
302                  $_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "IDENTIFY $NICK" );  #               $_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "IDENTIFY $NICK" );
303          },  #       },
304          irc_372 => sub {          irc_372 => sub {
305                  print "MOTD: ", $_[ARG1], "\n";                  print "MOTD: ", $_[ARG1], "\n";
306          },          },
# Line 312  POE::Session->create Line 390  POE::Session->create
390     },     },
391    );    );
392    
393    # http server
394    
395    my $httpd = POE::Component::Server::HTTP->new(
396            Port => $NICK =~ m/-dev/ ? 8001 : 8000,
397            ContentHandler => { '/' => \&root_handler },
398            Headers        => { Server => 'irc-logger' },
399    );
400    
401    my $style = <<'_END_OF_STYLE_';
402    .time, .channel { color: #808080; font-size: 60%; }
403    .nick { color: #0000ff; font-size: 80%; }
404    .message { color: #000000; font-size: 100%; }
405    _END_OF_STYLE_
406    
407    sub root_handler {
408            my ($request, $response) = @_;
409            $response->code(RC_OK);
410            $response->content_type("text/html; charset=$ENCODING");
411            $response->content(
412                    qq{<html><head><title>$NICK</title><style type="text/css">$style</style></head><body>} .
413                    "irc-logger url: " . $request->uri . '<br/>' .
414                    join("<br/>",
415                            get_from_log(
416                                    limit => 100,
417                                    fmt => {
418                                            time => '<span class="time">%s</span> ',
419                                            time_channel => '<span class="channel">%s %s</span> ',
420                                            nick => '<span class="nick">%s:</span> ',
421                                            message => '<span class="message">%s</span>',
422                                    },
423                                    message_filter => sub {
424                                            my $m = shift || return;
425                                            $m =~ s#($RE{URI}{HTTP})#<a href="$1">$1</a>#gs;
426                                            return $m;
427                                    },
428                            )
429                    ) .
430                    qq{</body></html>}
431            );
432            return RC_OK;
433    }
434    
435  POE::Kernel->run;  POE::Kernel->run;
Legend:
Removed from v.10  
changed lines
  Added in v.15
  ViewVC Help
Powered by ViewVC 1.1.26