#!perl -w

=head1 NAME

connection_time - log the duration of a connection

=head1 DESCRIPTION

The B<connection_time> plugin records the time of a connection between the
first and the last possible hook in qpsmtpd (I<pre-connection> and
I<post-connection>) and writes a C<LOGINFO> (default, see below) line to
the log.

=head1 CONFIG

=head2 loglevel

Adjust the quantity of logging for this plugin. See docs/logging.pod

 connection_time loglevel +1   (less logging)

 connection_time loglevel -1   (more logging)

=cut

use strict;
use warnings;

use Qpsmtpd::Constants;

use Time::HiRes qw(gettimeofday tv_interval);

sub register {
    my ($self, $qp) = (shift, shift);
    if (@_ == 1) {    # backwards compatible
        $self->{_args}{loglevel} = shift;
        if ($self->{_args}{loglevel} =~ /\D/) {
            $self->{_args}{loglevel} =
              Qpsmtpd::Constants::log_level($self->{_args}{loglevel});
        }
        $self->{_args}{loglevel} ||= 6;
    }
    elsif (@_ % 2) {
        $self->log(LOGERROR, "invalid arguments");
    }
    else {
        $self->{_args} = {@_};    # named args, inherits loglevel
    }

    # pre-connection is not available in the tcpserver deployment model.
    # duplicate the handler, so it works both ways with no redudant methods
    $self->register_hook('pre-connection', 'connect_handler');
    $self->register_hook('connect',        'connect_handler');
}

sub connect_handler {
    my $self = shift;
    return DECLINED
      if ($self->hook_name eq 'connect' && defined $self->{_connection_start});
    $self->{_connection_start} = [gettimeofday];
    $self->log(LOGDEBUG, "started at " . scalar gettimeofday);
    return (DECLINED);
}

sub hook_post_connection {
    my $self = shift;

    if (!$self->{_connection_start}) {
        $self->log(LOGERROR, "Start time not set?!");
        return (DECLINED);
    }

    my $elapsed = tv_interval($self->{_connection_start}, [gettimeofday]);

    $self->log(LOGINFO, sprintf "%.3f s.", $elapsed);
    return (DECLINED);
}