/[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 48 by dpavlin, Sat Feb 3 12:50:45 2007 UTC revision 63 by dpavlin, Fri Jun 8 12:07:45 2007 UTC
# Line 73  use POSIX qw/strftime/; Line 73  use POSIX qw/strftime/;
73  use HTML::CalendarMonthSimple;  use HTML::CalendarMonthSimple;
74  use Getopt::Long;  use Getopt::Long;
75  use DateTime;  use DateTime;
76    use URI::Escape;
77  use Data::Dump qw/dump/;  use Data::Dump qw/dump/;
78    use DateTime::Format::ISO8601;
79    
80    my $use_twitter = 1;
81    eval { require Net::Twitter; };
82    $use_twitter = 0 if ($@);
83    
84  my $import_dircproxy;  my $import_dircproxy;
85  my $log_path;  my $log_path;
# Line 90  sub _log { Line 96  sub _log {
96    
97  my $dbh = DBI->connect($DSN,"","", { RaiseError => 1, AutoCommit => 1 }) || die $DBI::errstr;  my $dbh = DBI->connect($DSN,"","", { RaiseError => 1, AutoCommit => 1 }) || die $DBI::errstr;
98    
99  eval {  my $sql_schema = {
100          $dbh->do(qq{ select count(*) from log });          log => '
 };  
   
 if ($@) {  
         warn "creating database table in $DSN\n";  
         $dbh->do(<<'_SQL_SCHEMA_');  
   
101  create table log (  create table log (
102          id serial,          id serial,
103          time timestamp default now(),          time timestamp default now(),
# Line 111  create table log ( Line 111  create table log (
111  create index log_time on log(time);  create index log_time on log(time);
112  create index log_channel on log(channel);  create index log_channel on log(channel);
113  create index log_nick on log(nick);  create index log_nick on log(nick);
114            ',
115            meta => '
116    create table meta (
117            nick text not null,
118            channel text not null,
119            name text not null,
120            value text,
121            changed timestamp default now(),
122            primary key(nick,channel,name)
123    );
124            ',
125    };
126    
127  _SQL_SCHEMA_  foreach my $table ( keys %$sql_schema ) {
128    
129            eval {
130                    $dbh->do(qq{ select count(*) from $table });
131            };
132    
133            if ($@) {
134                    warn "creating database table $table in $DSN\n";
135                    $dbh->do( $sql_schema->{ $table } );
136            }
137  }  }
138    
139    
140    =head2 meta
141    
142    Set or get some meta data into database
143    
144            meta('nick','channel','var_name', $var_value );
145    
146            $var_value = meta('nick','channel','var_name');
147            ( $var_value, $changed ) = meta('nick','channel','var_name');
148    
149    =cut
150    
151    sub meta {
152            my ($nick,$channel,$name,$value) = @_;
153    
154            # normalize channel name
155            $channel =~ s/^#//;
156    
157            if (defined($value)) {
158    
159                    my $sth = $dbh->prepare(qq{ update meta set value = ?, changed = now() where nick = ? and channel = ? and name = ? });
160    
161                    eval { $sth->execute( $value, $nick, $channel, $name ) };
162    
163                    # error or no result
164                    if ( $@ || ! $sth->rows ) {
165                            $sth = $dbh->prepare(qq{ insert into meta (value,nick,channel,name,changed) values (?,?,?,?,now()) });
166                            $sth->execute( $value, $nick, $channel, $name );
167                            _log "created $nick/$channel/$name = $value";
168                    } else {
169                            _log "updated $nick/$channel/$name = $value ";
170                    }
171    
172                    return $value;
173    
174            } else {
175    
176                    my $sth = $dbh->prepare(qq{ select value,changed from meta where nick = ? and channel = ? and name = ? });
177                    $sth->execute( $nick, $channel, $name );
178                    my ($v,$c) = $sth->fetchrow_array;
179                    _log "fetched $nick/$channel/$name = $v [$c]";
180                    return ($v,$c) if wantarray;
181                    return $v;
182    
183            }
184    }
185    
186    
187    
188  my $sth = $dbh->prepare(qq{  my $sth = $dbh->prepare(qq{
189  insert into log  insert into log
190          (channel, me, nick, message, time)          (channel, me, nick, message, time)
191  values (?,?,?,?,?)  values (?,?,?,?,?)
192  });  });
193    
194    
195  my $tags;  my $tags;
196  my $tag_regex = '\b([\w-_]+)//';  my $tag_regex = '\b([\w-_]+)//';
197    
# Line 191  sub get_from_log { Line 262  sub get_from_log {
262    
263          $sql .= " where message ilike ? or nick ilike ? " if ($args->{search});          $sql .= " where message ilike ? or nick ilike ? " if ($args->{search});
264          $sql .= " where id in (" . join(",", @{ $tags->{ $args->{tag} } }) . ") " if ($args->{tag} && $tags->{ $args->{tag} });          $sql .= " where id in (" . join(",", @{ $tags->{ $args->{tag} } }) . ") " if ($args->{tag} && $tags->{ $args->{tag} });
265          $sql .= " where date(time) = ? " if ($args->{date});          if ($args->{date}) {
266                    $args->{date} = DateTime::Format::ISO8601->parse_datetime( $args->{date} )->ymd;
267                    $sql .= " where date(time) = ? ";
268            }
269          $sql .= " order by log.time desc";          $sql .= " order by log.time desc";
270          $sql .= " limit " . $args->{limit} if ($args->{limit});          $sql .= " limit " . $args->{limit} if ($args->{limit});
271    
# Line 405  sub save_message { Line 479  sub save_message {
479                  message => $a->{msg});                  message => $a->{msg});
480  }  }
481    
482    
483  if ($import_dircproxy) {  if ($import_dircproxy) {
484          open(my $l, $import_dircproxy) || die "can't open $import_dircproxy: $!";          open(my $l, $import_dircproxy) || die "can't open $import_dircproxy: $!";
485          warn "importing $import_dircproxy...\n";          warn "importing $import_dircproxy...\n";
# Line 469  POE::Session->create( inline_states => Line 544  POE::Session->create( inline_states =>
544                  my $msg = $_[ARG2];                  my $msg = $_[ARG2];
545    
546                  save_message( channel => $channel, me => 0, nick => $nick, msg => $msg);                  save_message( channel => $channel, me => 0, nick => $nick, msg => $msg);
547                    meta( $nick, $channel, 'last-msg', $msg );
548      },      },
549      irc_ctcp_action => sub {      irc_ctcp_action => sub {
550                  my $kernel = $_[KERNEL];                  my $kernel = $_[KERNEL];
# Line 477  POE::Session->create( inline_states => Line 553  POE::Session->create( inline_states =>
553                  my $msg = $_[ARG2];                  my $msg = $_[ARG2];
554    
555                  save_message( channel => $channel, me => 1, nick => $nick, msg => $msg);                  save_message( channel => $channel, me => 1, nick => $nick, msg => $msg);
556    
557                    if ( $use_twitter ) {
558                            if ( my $twitter = meta( $nick, $NICK, 'twitter' ) ) {
559                                    my ($login,$passwd) = split(/\s+/,$twitter,2);
560                                    _log("sending twitter for $nick/$login on $channel ");
561                                    my $bot = Net::Twitter->new( username=>$login, password=>$passwd );
562                                    $bot->update("<${channel}> $msg");
563                            }
564                    }
565    
566      },      },
567          irc_ping => sub {          irc_ping => sub {
568                  warn "pong ", $_[ARG0], $/;                  warn "pong ", $_[ARG0], $/;
# Line 486  POE::Session->create( inline_states => Line 572  POE::Session->create( inline_states =>
572                  my $kernel = $_[KERNEL];                  my $kernel = $_[KERNEL];
573                  my $nick = (split /!/, $_[ARG0])[0];                  my $nick = (split /!/, $_[ARG0])[0];
574                  my $channel = $_[ARG1];                  my $channel = $_[ARG1];
                   
575    
576                  warn "invited to $channel by $nick";                  warn "invited to $channel by $nick";
577    
# Line 498  POE::Session->create( inline_states => Line 583  POE::Session->create( inline_states =>
583                  my $kernel = $_[KERNEL];                  my $kernel = $_[KERNEL];
584                  my $nick = (split /!/, $_[ARG0])[0];                  my $nick = (split /!/, $_[ARG0])[0];
585                  my $msg = $_[ARG2];                  my $msg = $_[ARG2];
586                    my $channel = $_[ARG1]->[0];
587                  from_to($msg, 'UTF-8', $ENCODING);                  from_to($msg, 'UTF-8', $ENCODING);
588    
589                  my $res = "unknown command '$msg', try /msg $NICK help!";                  my $res = "unknown command '$msg', try /msg $NICK help!";
# Line 521  POE::Session->create( inline_states => Line 607  POE::Session->create( inline_states =>
607    
608                          my $sth = $dbh->prepare(qq{                          my $sth = $dbh->prepare(qq{
609                                  select                                  select
610                                          nick,                                          trim(both '_' from nick) as nick,
611                                          count(*) as count,                                          count(*) as count,
612                                          sum(length(message)) as len                                          sum(length(message)) as len
613                                  from log                                  from log
614                                  group by nick                                  group by trim(both '_' from nick)
615                                  order by len desc,count desc                                  order by len desc,count desc
616                                  limit $nr                                  limit $nr
617                          });                          });
# Line 538  POE::Session->create( inline_states => Line 624  POE::Session->create( inline_states =>
624                          $res .= join(" | ", @users);                          $res .= join(" | ", @users);
625                  } elsif ($msg =~ m/^last.*?\s*(\d*)/i) {                  } elsif ($msg =~ m/^last.*?\s*(\d*)/i) {
626    
627                          foreach my $res (get_from_log( limit => ($1 || 100) )) {                          my $limit = $1 || meta( $nick, $channel, 'last-size' ) || 10;
628    
629                            foreach my $res (get_from_log( limit => $limit )) {
630                                  _log "last: $res";                                  _log "last: $res";
631                                  from_to($res, $ENCODING, 'UTF-8');                                  from_to($res, $ENCODING, 'UTF-8');
632                                  $_[KERNEL]->post( $IRC_ALIAS => privmsg => $nick, $res );                                  $_[KERNEL]->post( $IRC_ALIAS => privmsg => $nick, $res );
# Line 595  POE::Session->create( inline_states => Line 683  POE::Session->create( inline_states =>
683    
684                  } elsif ($msg =~ m/^ping/) {                  } elsif ($msg =~ m/^ping/) {
685                          $res = "ping = " . dump( $ping );                          $res = "ping = " . dump( $ping );
686                    } elsif ($msg =~ m/^conf(?:ig)*\s*(last-size|twitter)*\s*(.*)/) {
687                            if ( ! defined( $1 ) ) {
688                                    my $sth = $dbh->prepare(qq{ select name,value,changed from meta where nick = ? and channel = ? });
689                                    $sth->execute( $nick, $channel );
690                                    $res = "config for $nick on $channel";
691                                    while ( my ($n,$v) = $sth->fetchrow_array ) {
692                                            $res .= " | $n = $v";
693                                    }
694                            } elsif ( ! $2 ) {
695                                    my $val = meta( $nick, $channel, $1 );
696                                    $res = "current $1 = " . ( $val ? $val : 'undefined' );
697                            } else {
698                                    my $validate = {
699                                            'last-size' => qr/^\d+/,
700                                            'twitter' => qr/^\w+\s+\w+/,
701                                    };
702    
703                                    my ( $op, $val ) = ( $1, $2 );
704    
705                                    if ( my $regex = $validate->{$op} ) {
706                                            if ( $val =~ $regex ) {
707                                                    meta( $nick, $channel, $op, $val );
708                                                    $res = "saved $op = $val";
709                                            } else {
710                                                    $res = "config option $op = $val doesn't validate against $regex";
711                                            }
712                                    } else {
713                                            $res = "config option $op doesn't exist";
714                                    }
715                            }
716                  }                  }
717    
718                  if ($res) {                  if ($res) {
# Line 722  p { margin: 0; padding: 0.1em; } Line 840  p { margin: 0; padding: 0.1em; }
840  .nick { color: #000000; font-size: 80%; padding: 2px; font-family: courier, courier new, monospace ; }  .nick { color: #000000; font-size: 80%; padding: 2px; font-family: courier, courier new, monospace ; }
841  .message { color: #000000; font-size: 100%; }  .message { color: #000000; font-size: 100%; }
842  .search { float: right; }  .search { float: right; }
843    a:link.tag, a:visited.tag { border: 1px dashed #ccc; backgound: #ccc; text-decoration: none }
844    a:hover.tag { border: 1px solid #eee }
845    hr { border: 1px dashed #ccc; height: 1px; clear: both; }
846    /*
847  .col-0 { background: #ffff66 }  .col-0 { background: #ffff66 }
848  .col-1 { background: #a0ffff }  .col-1 { background: #a0ffff }
849  .col-2 { background: #99ff99 }  .col-2 { background: #99ff99 }
850  .col-3 { background: #ff9999 }  .col-3 { background: #ff9999 }
851  .col-4 { background: #ff66ff }  .col-4 { background: #ff66ff }
852  a:link.tag, a:visited.tag { border: 1px dashed #ccc; backgound: #ccc; text-decoration: none }  */
 a:hover.tag { border: 1px solid #eee }  
 hr { border: 1px dashed #ccc; height: 1px; clear: both; }  
853  _END_OF_STYLE_  _END_OF_STYLE_
854    
855  my $max_color = 4;  my $max_color = 4;
856    
857    my @cols = qw(
858            #ffcccc #ccffe6 #ccccff #e6ccff #ffccff #ffcce6 #ff9999 #ffcc99 #ffff99
859            #ccff99 #99ff99 #99ffcc #99ccff #9999ff #cc99ff #ff6666 #ffb366 #ffff66
860            #66ff66 #66ffb3 #66b3ff #6666ff #ff3333 #33ff33 #3399ff #3333ff #ff3399
861            #a0a0a0 #ff0000 #ffff00 #80ff00 #0000ff #8000ff #ff00ff #ff0080 #994d00
862            #999900 #009900 #cc0066 #c0c0c0 #ccff99 #99ff33 #808080 #660033 #ffffff
863    );
864    
865    $max_color = 0;
866    foreach my $c (@cols) {
867            $style .= ".col-${max_color} { background: $c }\n";
868            $max_color++;
869    }
870    warn "defined $max_color colors for users...\n";
871    
872  my %nick_enumerator;  my %nick_enumerator;
873    
874  sub root_handler {  sub root_handler {
# Line 811  sub root_handler { Line 946  sub root_handler {
946                                  filter => {                                  filter => {
947                                          message => sub {                                          message => sub {
948                                                  my $m = shift || return;                                                  my $m = shift || return;
949    
950                                                    # protect HTML from wiki modifications
951                                                    sub e {
952                                                            my $t = shift;
953                                                            return 'uri_unescape{' . uri_escape($t) . '}';
954                                                    }
955    
956                                                  $m =~ s/($escape_re)/$escape{$1}/gs;                                                  $m =~ s/($escape_re)/$escape{$1}/gs;
957                                                  $m =~ s#($RE{URI}{HTTP})#<a href="$1">$1</a>#gs;                                                  $m =~ s#($RE{URI}{HTTP})#e(qq{<a href="$1">$1</a>})#egs;
958                                                  $m =~ s#$tag_regex#<a href="?tag=$1" class="tag">$1</a>#g;                                                  $m =~ s#$tag_regex#e(qq{<a href="?tag=$1" class="tag">$1</a>})#egs;
959                                                    $m =~ s#\*(\w+)\*#<b>$1</b>#gs;
960                                                    $m =~ s#_(\w+)_#<u>$1</u>#gs;
961                                                    $m =~ s#\/(\w+)\/#<i>$1</i>#gs;
962    
963                                                    $m =~ s#uri_unescape{([^}]+)}#uri_unescape($1)#egs;
964                                                  return $m;                                                  return $m;
965                                          },                                          },
966                                          nick => sub {                                          nick => sub {
Legend:
Removed from v.48  
changed lines
  Added in v.63
  ViewVC Help
Powered by ViewVC 1.1.26