File Coverage

lib/Haineko/SMTPD/Relay/File.pm

Criterion	Covered	Total	%
statement	75	88	85.2
branch	7	12	58.3
condition	5	9	55.5
subroutine	12	13	92.3
pod	2	2	100.0
total	101	124	81.4

package Haineko::SMTPD::Relay::File;
use parent 'Haineko::SMTPD::Relay';
use strict;
use warnings;
use Haineko::SMTPD::Response;
use Email::MIME;
use Time::Piece;
use Try::Tiny;
use IO::File;
use Encode;

sub new {
    my $class = shift;
    my $argvs = { @_ };

    $argvs->{'host'}    ||= '/tmp';
    $argvs->{'port'}      = undef,
    $argvs->{'time'}    ||= Time::Piece->new;
    $argvs->{'sleep'}     = 0;
    $argvs->{'retry'}     = 0;
    $argvs->{'timeout'}   = 0;
    $argvs->{'startls'}   = 0;
    return bless $argvs, __PACKAGE__;
}

sub sendmail {
    my $self = shift;

    my $headerlist = [];
    my $emencoding = uc( $self->{'attr'}->{'charset'} || 'UTF-8' );
    my $methodargv = {
        'body' => Encode::encode( $emencoding, ${ $self->{'body'} } ),
        'attributes' => $self->{'attr'},
    };
    utf8::decode $methodargv->{'body'} unless utf8::is_utf8 $methodargv->{'body'} ;

    for my $e ( @{ $self->{'head'}->{'Received'} } ) {
        # Convert email headers
        push @$headerlist, 'Received' => $e;
    }
    push @$headerlist, 'To' => $self->{'rcpt'};

    for my $e ( keys %{ $self->{'head'} } ) {
        # Make email headers except ``Received'' and ``MIME-Version''
        next if $e eq 'Received';
        next if $e eq 'MIME-Version';

        if( ref $self->{'head'}->{ $e } eq 'ARRAY' ) {

            for my $f ( @{ $self->{'head'}->{ $e } } ) {
                push @$headerlist, $e => $f;
            }
        }
        else {
            push @$headerlist, $e => $self->{'head'}->{ $e };
        }
    }
    $methodargv->{'header'} = $headerlist;

    my $mimeobject = Email::MIME->create( %$methodargv );
    my $mailstring = $mimeobject->as_string;
    my $mailfolder = $self->{'host'};
    my $messageid0 = undef;

    if( exists $self->{'head'}->{'Message-Id'} ) {
        # Use the local part of the Message-Id header as a file name.
        $messageid0 = [ split( '@', $self->{'head'}->{'Message-Id'} ) ]->[0];

    } else {
        # Message-Id header is not defined or does not exist
        require Haineko::SMTPD::Session;
        $messageid0 = sprintf( "%s.%d.%d.%03d",
                        Haineko::SMTPD::Session->make_queueid, $$,
                        $self->time->epoch, int(rand(100)) );
    }

    my $timestring = sprintf( "%s-%s", $self->time->ymd('-'), $self->time->hms );
    my $outputfile = sprintf( "%s/haineko.%s.%s.eml", $mailfolder, $timestring, $messageid0 );
    my $filehandle = undef;
    my $smtpparams = undef;
    my $smtpstatus = 0;

    try {
        $outputfile =~ y{/}{}s;
        $smtpparams = {
            'dsn'     => undef,
            'code'    => 200,
            'host'    => undef,
            'port'    => undef,
            'rcpt'    => $self->{'rcpt'},
            'error'   => 0,
            'mailer'  => 'File',
            'message' => [ $outputfile ],
            'command' => 'DATA',
        };
        $filehandle =  IO::File->new( $outputfile, 'w' ) || die $!;
        utf8::encode $mailstring if utf8::is_utf8 $mailstring;

        $filehandle->print( $mailstring );
        $filehandle->close;
        push @{ $smtpparams->{'message'} }, 'Successfully saved';
        $smtpstatus = 1;

    } catch {
        require Haineko::E;
        my $E = Haineko::E->new( $_ );
        $smtpparams->{'code'} = 400;
        $smtpparams->{'error'} = 1;
        push @{ $smtpparams->{'message'} }, 'Failed to save: '.join( ' ', @{ $E->mesg } );
    };

    $self->response( Haineko::SMTPD::Response->new( %$smtpparams ) );
    return $smtpstatus;
}

1;
__END__