File Coverage

blib/lib/MIME/Disclaimer.pm

Criterion	Covered	Total	%
statement	29	36	80.5
branch	0	2	0.0
condition			n/a
subroutine	10	12	83.3
pod	1	1	100.0
total	40	51	78.4

package MIME::Disclaimer;

use 5.014;
use strict;
use warnings;
use parent -norequire, 'MIME::Signature';

use MIME::Signature qw(_decoded_body _replace_body);
use Class::Method::Modifiers;

our $VERSION = '0.17';

use constant EMPTY           => '';
use constant NEWLINE         => "\n";
use constant LINEBREAK       => '';
use constant HORIZONTAL_RULE => '';

around 'new' => sub {
  my $orig  = shift;
  my $class = shift;
  my %args  = @_;
  my $self  = $class->$orig(%args);

  for (qw(plain_delimiter html_delimiter enriched_delimiter)) {
    unless (exists $args{$_}) {
      my $delimiter = ($_ eq 'html_delimiter') ? LINEBREAK . HORIZONTAL_RULE : NEWLINE;
      $self->$_($delimiter);
    }
  }

  return $self;
};

around 'handler_text_enriched' => sub {
  my $orig   = shift;
  my $self   = shift;
  my $entity = shift;

  $orig->($self, $entity);

  _replace_body($entity, $self->_disclaimer('enriched') . _decoded_body($entity));
};

around 'handler_text_html' => sub {
  my $orig   = shift;
  my $self   = shift;
  my $entity = shift;

  $orig->($self, $entity);

  my $body = _decoded_body($entity);
  require HTML::Parser;
  my $new_body;
  my $parser = HTML::Parser->new(
    start_h => [
      sub {
        my ($text, $tagname) = @_;
        $new_body .= $text;
        $new_body .= $self->_disclaimer('html') if lc $tagname eq 'body';
      },
      'text,tagname'
    ],
    default_h => [sub {$new_body .= shift}, 'text'],
  );
  $parser->parse($body);
  _replace_body($entity, $new_body);
};

around 'handler_text_plain' => sub {
  my $orig   = shift;
  my $self   = shift;
  my $entity = shift;

  $orig->($self, $entity);

  _replace_body($entity, $self->_disclaimer('plain') . _decoded_body($entity));
};

sub _disclaimer {
  my ($self, $type) = @_;

  defined(my $signature = $self->$type) or return;
  my $delimiter_method = $type . '_delimiter';
  my $delimiter        = $self->$delimiter_method;

  return join(EMPTY, ($delimiter, $signature, $delimiter));
}

sub add {
  my $self = shift;
  return $self->append(@_);
}

1;

__END__