--- trunk/bin/irc-logger.pl	2008/03/07 10:30:57	92
+++ trunk/bin/irc-logger.pl	2008/03/09 19:13:16	105
@@ -2,6 +2,23 @@
 use strict;
 $|++;
 
+use POE qw(Component::IRC Component::Server::HTTP);
+use HTTP::Status;
+use DBI;
+use Regexp::Common qw /URI/;
+use CGI::Simple;
+use HTML::TagCloud;
+use POSIX qw/strftime/;
+use HTML::CalendarMonthSimple;
+use Getopt::Long;
+use DateTime;
+use URI::Escape;
+use Data::Dump qw/dump/;
+use DateTime::Format::ISO8601;
+use Carp qw/confess/;
+use XML::Feed;
+use DateTime::Format::Flexible;
+
 =head1 NAME
 
 irc-logger.pl
@@ -35,15 +52,22 @@
 
 my $NICK = 'irc-logger';
 $NICK .= '-dev' if ($HOSTNAME =~ m/llin/);
-my $CONNECT =
-  {Server => 'irc.freenode.net',
-   Nick => $NICK,
-   Ircname => "try /msg $NICK help",
-  };
+my $CONNECT = {
+	Server => 'irc.freenode.net',
+	Nick => $NICK,
+	Ircname => "try /msg $NICK help",
+};
 my $CHANNEL = '#razmjenavjestina';
 $CHANNEL = '#irc-logger' if ($HOSTNAME =~ m/llin/);
 my $IRC_ALIAS = "log";
 
+if ( $HOSTNAME =~ m/lugarin/ ) {
+	$CONNECT->{Server} = 'irc.carnet.hr';
+	$CHANNEL = '#riss';
+}
+
+warn dump( $HOSTNAME, $CONNECT );
+
 my $DSN = 'DBI:Pg:dbname=' . $NICK;
 
 my $TIMESTAMP = '%Y-%m-%d %H:%M:%S';
@@ -55,7 +79,6 @@
 
 # don't pull rss feeds more often than this
 my $rss_min_delay = 60;
-$rss_min_delay = 15;
 
 my $http_port = $NICK =~ m/-dev/ ? 8001 : 8000;
 
@@ -63,23 +86,6 @@
 
 ## END CONFIG
 
-use POE qw(Component::IRC Component::Server::HTTP);
-use HTTP::Status;
-use DBI;
-use Regexp::Common qw /URI/;
-use CGI::Simple;
-use HTML::TagCloud;
-use POSIX qw/strftime/;
-use HTML::CalendarMonthSimple;
-use Getopt::Long;
-use DateTime;
-use URI::Escape;
-use Data::Dump qw/dump/;
-use DateTime::Format::ISO8601;
-use Carp qw/confess/;
-use XML::Feed;
-use DateTime::Format::Flexible;
-
 my $use_twitter = 1;
 eval { require Net::Twitter; };
 $use_twitter = 0 if ($@);
@@ -118,11 +124,11 @@
 		# protect HTML from wiki modifications
 		sub e {
 			my $t = shift;
-			return 'uri_unescape{' . uri_escape($t) . '}';
+			return 'uri_unescape{' . uri_escape($t, '^a-zA-Z0-9') . '}';
 		}
 
 		$m =~ s/($escape_re)/$escape{$1}/gs;
-		$m =~ s#($RE{URI}{HTTP})#e(qq{<a href="$1">$1</a>})#egs ||
+		$m =~ s#($RE{URI}{HTTP})#e(qq{<a href="$1">$1</a>})#egs;
 		$m =~ s#\/(\w+)\/#<i>$1</i>#gs;
 		$m =~ s#$tag_regex#e(qq{<a href="$url?tag=$1" class="tag">$1</a>})#egs;
 		$m =~ s#\*(\w+)\*#<b>$1</b>#gs;
@@ -179,12 +185,15 @@
 	name text,
 	delay interval not null default '5 min',
 	active boolean default true,
+	channel text not null,
+	nick text not null,
+	private boolean default false,
 	last_update timestamp default 'now()',
 	polls int default 0,
 	updates int default 0
 );
 create unique index feeds_url on feeds(url);
-insert into feeds (url,name) values ('http://wiki.razmjenavjestina.org/feed/workspace/razmjenavjestina?category=Recent%20Changes','wiki');
+insert into feeds (url,name,channel,nick) values ('http://wiki.razmjenavjestina.org/feed/workspace/razmjenavjestina?category=Recent%20Changes','wiki','$CHANNEL','dpavlin');
 	},
 };
 
@@ -228,9 +237,9 @@
 		if ( $@ || ! $sth->rows ) {
 			$sth = $dbh->prepare(qq{ insert into meta (value,nick,channel,name,changed) values (?,?,?,?,now()) });
 			$sth->execute( $value, $nick, $channel, $name );
-			_log "created $nick/$channel/$name = $value";
+			warn "## created $nick/$channel/$name = $value\n";
 		} else {
-			_log "updated $nick/$channel/$name = $value ";
+			warn "## updated $nick/$channel/$name = $value\n";
 		}
 
 		return $value;
@@ -240,7 +249,7 @@
 		my $sth = $dbh->prepare(qq{ select value,changed from meta where nick = ? and channel = ? and name = ? });
 		$sth->execute( $nick, $channel, $name );
 		my ($v,$c) = $sth->fetchrow_array;
-		_log "fetched $nick/$channel/$name = $v [$c]";
+		warn "## fetched $nick/$channel/$name = $v [$c]\n";
 		return ($v,$c) if wantarray;
 		return $v;
 
@@ -337,25 +346,26 @@
 
 	my @where;
 	my @args;
+	my $msg;
 
 	if (my $search = $args->{search}) {
 		$search =~ s/^\s+//;
 		$search =~ s/\s+$//;
 		push @where, 'message ilike ? or nick ilike ?';
 		push @args, ( ( '%' . $search . '%' ) x 2 );
-		_log "search for '$search'";
+		$msg = "Search for '$search'";
 	}
 
 	if ($args->{tag} && $tags->{ $args->{tag} }) {
 		push @where, 'id in (' . join(',', @{ $tags->{ $args->{tag} } }) . ')';
-		_log "search for tags $args->{tag}";
+		$msg = "Search for tags $args->{tag}";
 	}
 
 	if (my $date = $args->{date} ) {
 		$date = check_date( $date );
 		push @where, 'date(time) = ?';
 		push @args, $date;
-		_log "search for date $date";
+		$msg = "search for date $date";
 	}
 
 	$sql .= " where " . join(" and ", @where) if @where;
@@ -369,6 +379,8 @@
 	eval { $sth->execute( @args ) };
 	return if $@;
 
+	my $nr_results = $sth->rows;
+
 	my $last_row = {
 		date => '',
 		time => '',
@@ -389,10 +401,14 @@
 
 	return @rows if ($args->{full_rows});
 
-	my @msgs = (
-		"Showing " . ($#rows + 1) . " messages..."
+	$msg .= ' produced ' . (
+		$nr_results == 0 ? 'no results' :
+		$nr_results == 0 ? 'one result' :
+			$nr_results . ' results'
 	);
 
+	my @msgs = ( $msg );
+
 	if ($context) {
 		my @ids = @rows;
 		@rows = ();
@@ -449,6 +465,8 @@
 #			$row->{nick} = $nick;
 #		}
 
+		$append = 0 if $row->{me};
+
 		if ($last_row->{nick} ne $nick) {
 			# obfu way to find format for me_nick if needed or fallback to default
 			my $fmt = $row->{me} ? ( $args->{fmt}->{me_nick} || $args->{fmt}->{nick} ) : $args->{fmt}->{nick};
@@ -634,26 +652,42 @@
 		$total++;
 
 		# seen allready?
-		next if $_rss->{$feed->link}->{seen}->{$entry->id}++ > 0;
+		next if $_rss->{$args->{channel}}->{$feed->link}->{$entry->id}++ > 0;
 
 		sub prefix {
 			my ($txt,$var) = @_;
+			$var =~ s/\s+/ /gs;
 			$var =~ s/^\s+//g;
+			$var =~ s/\s+$//g;
 			return $txt . $var if $var;
 		}
 
+		# fix absolute and relative links to feed entries
+		my $link = $entry->link;
+		if ( $link =~ m!^/! ) {
+			my $host = $args->{url};
+			$host =~ s!^(http://[^/]+).*$!$1!;	#!vim
+			$link = "$host/$link";
+		} elsif ( $link !~ m!^http! ) {
+			$link = $args->{url} . $link;
+		}
+
 		my $msg;
 		$msg .= prefix( 'From: ' , $args->{name} || $feed->title );
 		$msg .= prefix( ' by ' , $entry->author );
 		$msg .= prefix( ' | ' , $entry->title );
-		$msg .= prefix( ' | ' , $entry->link );
+		$msg .= prefix( ' | ' , $link );
 #		$msg .= prefix( ' id ' , $entry->id );
 
 		if ( $args->{kernel} && $send_rss_msgs ) {
 			$send_rss_msgs--;
-			_log('>>', $msg);
-			$sth_insert_log->execute( $CHANNEL, 1, $NICK, $msg, 'now()' );
-			$args->{kernel}->post( $IRC_ALIAS => notice => $CHANNEL, $msg );
+			# FIXME bug! should be save_message
+#			save_message( channel => $args->{channel}, me => 1, nick => $NICK, message => $msg );
+			$sth_insert_log->execute( $args->{channel}, 1, $NICK, $msg, 'now()' );
+			my ( $type, $to ) = ( 'notice', $args->{channel} );
+			( $type, $to ) = ( 'privmsg', $args->{nick} ) if $args->{private};
+			_log(">> $type $to |", $msg);
+			$args->{kernel}->post( $IRC_ALIAS => $type => $to, $msg );
 			$updates++;
 		}
 	}
@@ -671,7 +705,7 @@
 sub rss_fetch_all {
 	my $kernel = shift;
 	my $sql = qq{
-		select id, url, name
+		select id, url, name, channel, nick, private
 		from feeds
 		where active is true
 	};
@@ -691,10 +725,11 @@
 
 sub rss_check_updates {
 	my $kernel = shift;
-	my $last_t = $_rss->{last_poll} || time();
-	my $t = time();
-	if ( $t - $last_t > $rss_min_delay ) {
-		$_rss->{last_poll} = $t;
+	$_rss->{last_poll} ||= time();
+	my $dt = time() - $_rss->{last_poll};
+	warn "## rss_check_updates $dt > $rss_min_delay\n";
+	if ( $dt > $rss_min_delay ) {
+		$_rss->{last_poll} = time();
 		_log rss_fetch_all( $kernel );
 	}
 }
@@ -727,6 +762,7 @@
 
 		save_message( channel => $channel, me => 0, nick => $nick, message => $msg);
 		meta( $nick, $channel, 'last-msg', $msg );
+		rss_check_updates( $kernel );
     },
     irc_ctcp_action => sub {
 		my $kernel = $_[KERNEL];
@@ -900,29 +936,43 @@
 			$dbh->do( qq{ update feeds set last_update = now() - delay } );
 			$res = "OK, cleaned RSS cache";
 		} elsif ($msg =~ m/^rss-list/) {
-			my $sth = $dbh->prepare(qq{ select url,name,last_update,active from feeds });
+			my $sth = $dbh->prepare(qq{ select url,name,last_update,active,channel,nick,private from feeds });
 			$sth->execute;
 			while (my @row = $sth->fetchrow_array) {
 				$_[KERNEL]->post( $IRC_ALIAS => privmsg => $nick, join(' | ',@row) );
 			}
 			$res = '';
-		} elsif ($msg =~ m!^rss-(add|remove|stop|start)\s+(http://\S+)\s*(.*)!) {
+		} elsif ($msg =~ m!^rss-(add|remove|stop|start)(?:-(private))?\s+(http://\S+)\s*(.*)!) {
+			my ( $command, $sub, $url, $arg ) = ( $1,$2,$3,$4 );
+
+			my $channel = $1 if ( $arg =~ s/\s*(#\S+)\s*// );
+			$channel = $nick if $sub eq 'private';
+
 			my $sql = {
-				add		=> qq{ insert into feeds (url,name) values (?,?) },
+				add	=> qq{ insert into feeds (url,name,channel,nick,private) values (?,?,?,?,?) },
 #				remove	=> qq{ delete from feeds				where url = ? and name = ? },
 				start	=> qq{ update feeds set active = true	where url = ? },
 				stop	=> qq{ update feeds set active = false	where url = ? },
-				
 			};
-			if (my $q = $sql->{$1} ) {
+
+			if ( $command eq 'add' && ! $channel ) {
+				$res = "ERROR: got '$msg' which doesn't have #channel in it, ignoring!";
+			} elsif (my $q = $sql->{$command} ) {
 				my $sth = $dbh->prepare( $q );
-				my @data = ( $2 );
-				push @data, $3 if ( $q =~ s/\?//g == 2 );
-				warn "## $1 SQL $q with ",dump( @data ),"\n";
+				my @data = ( $url );
+				if ( $command eq 'add' ) {
+					push @data, ( $arg, $channel, $nick, $sub eq 'private' ? 1 : 0 );
+				}
+				warn "## $command SQL $q with ",dump( @data ),"\n";
 				eval { $sth->execute( @data ) };
+				if ($@) {
+					$res = "ERROR: $@";
+				} else {
+					$res = "OK, RSS [$command|$sub|$url|$arg]";
+				}
+			} else {
+				$res = "ERROR: don't know what to do with: $msg";
 			}
-
-			$res = "OK, RSS $1 : $2 - $3";
 		}
 
 		if ($res) {
@@ -933,33 +983,42 @@
 		rss_check_updates( $_[KERNEL] );
 	},
 	irc_477 => sub {
-		_log "# irc_477: ",$_[ARG1];
+		_log "<< irc_477: ",$_[ARG1];
 		$_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "register $NICK" );
 	},
 	irc_505 => sub {
-		_log "# irc_505: ",$_[ARG1];
+		_log "<< irc_505: ",$_[ARG1];
 		$_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "register $NICK" );
 #		$_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "set hide email on" );
 #		$_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "set email dpavlin\@rot13.org" );
 	},
 	irc_registered => sub {
-		_log "## registrated $NICK";
+		_log "## registrated $NICK, /msg nickserv IDENTIFY $NICK";
 		$_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "IDENTIFY $NICK" );
 	},
 	irc_disconnected => sub {
-		_log "## disconnected, reconnecting again";
-		$_[KERNEL]->post($IRC_ALIAS => connect => $CONNECT);
+		_log "## disconnected.. sleeping for $sleep_on_error seconds and reconnecting again";
+		sleep($sleep_on_error);
+		$_[KERNEL]->post( $IRC_ALIAS => connect => $CONNECT);
 	},
 	irc_socketerr => sub {
 		_log "## socket error... sleeping for $sleep_on_error seconds and retry";
 		sleep($sleep_on_error);
-		$_[KERNEL]->post($IRC_ALIAS => connect => $CONNECT);
+		$_[KERNEL]->post( $IRC_ALIAS => connect => $CONNECT);
 	},
 #	irc_433 => sub {
 #		print "# irc_433: ",$_[ARG1], "\n";
 #		warn "## indetify $NICK\n";
 #		$_[KERNEL]->post( $IRC_ALIAS => privmsg => 'nickserv', "IDENTIFY $NICK" );
 #	},
+#	irc_451	# please register
+	irc_snotice => sub {
+		_log "<< snotice",$_[ARG0];
+		if ( $_[ARG0] =~ m!/(QUOTE)\s+(PASS\s+\d+)!i ) {
+			warn ">> $1 | $2\n";
+			$_[KERNEL]->post( $IRC_ALIAS => lc($1) => $2);
+		}
+	},
     _child => sub {},
     _default => sub {
 		_log sprintf "sID:%s %s %s",
@@ -1126,6 +1185,13 @@
 				$feed->add_entry( $feed_entry );
 			}
 
+			my $feed_entry = XML::Feed::Entry->new($type);
+			$feed_entry->title( "Internal stats" );
+			$feed_entry->content(
+				'<![CDATA[<pre>' . dump( $_rss ) . '</pre>]]>'
+			);
+			$feed->add_entry( $feed_entry );
+
 		} else {
 			_log "unknown rss request ",$request->url;
 			return RC_DENY;