SyncClientConnection.pm

# ---- class SyncClientConnection ----
# ddgrey client connection

package DDgrey::SyncClientConnection;

use strict;
use integer;

use DDgrey::Perl6::Parameters;
use Data::Dumper; # DEBUG
use Net::hostent;
use Socket;

use DDgrey::Config;
use DDgrey::Report;
use DDgrey::RemoteServer;

use parent qw(DDgrey::ClientConnection);

# ---- constructor ----

sub new($class,$server,$fh,$permission){
    # return: new client connection of class to server, fh and permission
    # effect: registers with server and select
    #         may raise exception

    my $self=$class->SUPER::new($server,$fh);
    $self->{permission}=$permission;
    return $self;
};

# ---- methods ----

sub handle_command($self,$item){
    # return/effect: execution of command described in item
    
    my $command=shift(@{$item->{arg}});
    if(!defined($command)){
	return "500 no command received\r\n";
    };

    # -- helo --
    if($command eq 'helo'){
	my $arg=$item->{arg}->[0];
	defined($arg) or return "500 argument missing\r\n";
	$arg=~/^[\w\.\-]+$/ or return "500 bad argument\r\n";
	$self->{peername}=$arg;
	return "200 welcome. I am ".$main::hostname."\r\n";
    };

    # -- quit --
    if($command eq 'quit'){
	$self->{closing}=1;
	return "250 closing connection\r\n";
    };

    # -- list --
    if($command eq 'list'){
	# check permission
	$self->{permission}->{read} or return "500 permission denied\r\n";

	my $arg=($item->{arg}->[0] or 0);
	$arg=~/^\d+$/ or return "500 bad argument\r\n";
	my @list=DDgrey::Report->list($arg);
	# yield (and catch possible closed connection)
	$main::select->run_once(0);
	$self->send("300 document following\r\n");
	$self->{pending_send}++;
	for my $l (@list){
	    # quit if closing
	    $self->{closing} and last;
	    # send
	    $self->send($l->{id}."\t".$l->{origin}."\t".$l->{origin_id}."\t".$l->{stored}."\r\n");
	    # yield and write
	    $main::select->run_once(0);
	};
	$self->{pending_send}--;
	return ".\r\n";
    };

    # -- get --
    if($command eq 'get'){
	# check permission
	$self->{permission}->{read} or return "500 permission denied\r\n";

	my $arg=$item->{arg}->[0];
	defined($arg) or return "500 argument missing\r\n";
	$arg=~/^\d+$/ or return "500 bad argument\r\n";
	
	my $report=DDgrey::Report->get($arg);
	defined($report) or return "500 report not found\r\n";
	    
	$self->send("300 document following\r\n");
	$self->send($report->as_text());
	$self->send("\r\n");
	return;
    };

    # -- subscribe --
    if($command eq 'subscribe'){ 
    	# check permission
	$self->{permission}->{read} or return "500 permission denied\r\n";

	my $subscription=$main::dispatcher->register_subscriber(sub{$self->send_report(@_)});
	defined($subscription) or return "502 subscription failed\r\n";
	$self->{subscription}=$subscription;
	$self->{data_handler}=sub{$self->handle_subscribe()};
	
	# if argument, start with reports from earlier than now
	my $arg=($item->{arg}->[0]);
	my @list=();
	if(defined($arg)){
	    $arg=~/^\d+$/ or return "500 bad argument\r\n";
	    @list=DDgrey::Report->list($arg);
	};

	# send start, possibly starting with existing reports
	$self->send("302 document following (interrupt with single dot)\r\n");
	$self->{pending_send}++;
	foreach my $l (@list){
	    # quit if closing
	    $self->{closing} and last;
	    my $report=DDgrey::Report->get($l->{id});
	    $self->send_report($report);
	    # yield and write
	    $main::select->run_once(0);
	};
	$self->{pending_send}--;
	$self->close_if_done();
	return;
    };
    
    # -- report --
    if($command eq 'report'){
	# check permission
	$self->{permission}->{write} or return "500 permission denied\n";
	$self->{data_handler}=sub{$self->handle_report()};
	return "301 go ahead\n";
    };

    # ---- other commands ----
    return "500 unknown command\r\n";
};

sub close($self){
    # effect: deregisters where registered and closes

    if($self->{subscription}){
	$main::dispatcher->unregister_subscriber($self->{subscription});
    };
    return $self->SUPER::close();
};

sub handle_subscribe($self){
    # return: status line
    # effect: handles end of subscription command
    # pre   : subscription is active

    $main::dispatcher->unregister_subscriber($self->{subscription});
    $self->{data} and return "500 no text accepted during subscribe\r\n";
    return "200 subscription finished\r\n";
};

sub send_report($self,$report){
    # effect: sends report to client

    $main::debug > 1 and main::lm("sending report ".$report->unicode(),$self->service());
    $self->send($report->as_text());
    # extra blank row indicates end of report
    $self->send("\r\n");
};

sub handle_report($self){
    # return: status line
    # effect: handles report stored in self->data
    
    my $report=eval{DDgrey::Report->from_text($self->{data})};
    if($@ or !defined($report)){
	my $e=$@;
	chomp($e);
	main::lm("bad report from client ($e)",$self->service(),"warning");
	return "500 bad report ($e)\r\n";
    }; 
    $main::debug > 1 and main::lm("got report ".$report->unicode(),$self->service());
    $main::dispatcher->report($report);
    return "200 accepted\r\n";
};

# ---- package init ----
return 1;