File Coverage

blib/lib/HTML/Trim.pm

Criterion	Covered	Total	%
statement	72	72	100.0
branch	14	14	100.0
condition	3	6	50.0
subroutine	18	18	100.0
pod	2	3	66.6
total	109	113	96.4

package HTML::Trim;
use strict;
use warnings;
use utf8;
our $VERSION = '0.02';

use Exporter::Lite;
use HTML::Parser;
use Unicode::EastAsianWidth;

our @EXPORT_OK = qw(htrim hvtrim);

sub trim {
	my ($str, $max, $delim) = @_;
	if (ref $str eq __PACKAGE__) {
		return _trim(@_);
	}
	__PACKAGE__->new(
		length => sub {
			length $_[0];
		},
		substr => sub {
			substr $_[0], $_[1], $_[2];
		}
	)->trim($str, $max, $delim);
}
*htrim = \&trim;

sub vtrim {
	my ($str, $max, $delim) = @_;
	__PACKAGE__->new(
		length => sub {
			my ($str) = @_;
			my $ret = 0;

			local $_ = $str;
			while (/(?:(\p{InFullwidth}+)|(\p{InHalfwidth}+))/g) {
				$ret += $1 ? length($1) * 2 : length($2);
			}
			$ret;
		},
		substr => sub {
			my ($str, $offset, $limit) = @_; # ignoring offset
			my $ret   = "";
			my $count = 0;

			local $_ = $str;
			while (/(?:(\p{InFullwidth})|(\p{InHalfwidth}))/g) {
				$count += $1 ? length($1) * 2 : length($2);
				last if $count > $limit;
				$ret .= $1 || $2;
			}
			$ret;
		}
	)->trim($str, $max, $delim);
}
*hvtrim = \&vtrim;

sub new {
	my ($class, %opts) = @_;
	bless {
		length => $opts{length},
		substr => $opts{substr}
	}, $class;
}

sub _trim {
	my ($self, $str, $max, $delim) = @_;

	my $ret   = "";
	my $count = 0;
	my $opened = [];

	my $p; $p = HTML::Parser->new(
		api_version => 3,
		handlers => {
			start     => [ sub {
				my ($text, $tagname) = @_;
				push @$opened, $tagname unless $tagname =~ /^(input|img|br)$/;;
				$ret .= $text;
			}, "text, tagname"],
			end       => [ sub {
				my ($text, $tagname) = @_;
				until (!@$opened || pop @$opened eq $tagname) { }
				$ret .= $text;
			}, "text, tagname"],
			text      => [ sub {
				my ($text) = @_;
				my $length += $self->{length}->($text);
				if ($count + $length > $max) {
					$ret .= $self->{substr}->($text, 0, $max - $count - 1);
					$p->eof; # end parse immediately
				} else {
					$ret .= $text;
				}
				$count += $length
			}, 'text'],
		}
	);
	$p->parse($str);
	$p->eof;

	while (my $tagname = pop @$opened) {
		$ret .= sprintf('', $tagname);
	}

	if ($count > $max) {
		$ret .= $delim;
	}

	$ret;
}

1;
__END__