package HTML::Zoom::FilterBuilder;
-use Devel::Dwarn;
-
use strict;
use warnings FATAL => 'all';
use HTML::Zoom::CodeStream;
HTML::Zoom::CodeStream->from_array(@_)
}
+sub _stream_from_proto {
+ my ($self, $proto) = @_;
+ my $ref = ref $proto;
+ if (not $ref) {
+ require HTML::Zoom::Parser::BuiltIn;
+ return $self->_stream_from_array({
+ type => 'TEXT',
+ raw => HTML::Zoom::Parser::BuiltIn->html_escape($proto)
+ });
+ } elsif ($ref eq 'ARRAY') {
+ return $self->_stream_from_array(@$proto);
+ } elsif ($ref eq 'CODE') {
+ return $proto->();
+ } elsif ($ref eq 'SCALAR') {
+ require HTML::Zoom::Parser::BuiltIn;
+ return HTML::Zoom::Parser::BuiltIn->html_to_stream($$proto);
+ }
+ die "Don't know how to turn $proto (ref $ref) into a stream";
+}
+
sub _stream_concat {
- shift; # lose $self
- my @streams = @_;
- my $cur_stream = shift(@streams) or die "No streams passed";
- HTML::Zoom::CodeStream->new({
- code => sub {
- return unless $cur_stream;
- my $evt;
- until (($evt) = $cur_stream->next) {
- return unless $cur_stream = shift(@streams);
- }
- return $evt;
- }
- });
+ shift->_stream_from_array(@_)->flatten;
}
sub set_attribute {
sub collect {
my ($self, $options) = @_;
- my ($into, $passthrough, $inside) = @{$options}{qw(into passthrough inside)};
+ my ($into, $passthrough, $content, $filter) =
+ @{$options}{qw(into passthrough content filter)};
sub {
my ($evt, $stream) = @_;
- push(@$into, $evt) if $into && !$inside;
+ # We wipe the contents of @$into here so that other actions depending
+ # on this (such as a repeater) can be invoked multiple times easily.
+ # I -suspect- it's better for that state reset to be managed here; if it
+ # ever becomes painful the decision should be revisited
+ if ($into) {
+ @$into = $content ? () : ($evt);
+ }
if ($evt->{is_in_place_close}) {
- return $evt if $passthrough || $inside;
+ return $evt if $passthrough || $content;
return;
}
my $name = $evt->{name};
my $depth = 1;
- my $_next = $inside ? 'peek' : 'next';
+ my $_next = $content ? 'peek' : 'next';
+ $stream = $filter->($stream) if $filter;
my $collector = $self->_stream_from_code(sub {
return unless $stream;
while (my ($evt) = $stream->$_next) {
$depth-- if ($evt->{type} eq 'CLOSE');
unless ($depth) {
undef $stream;
- return if $inside;
+ return if $content;
push(@$into, $evt) if $into;
return $evt if $passthrough;
return;
}
push(@$into, $evt) if $into;
- $stream->next if $inside;
+ $stream->next if $content;
return $evt if $passthrough;
}
die "Never saw closing </${name}> before end of source";
});
- return ($passthrough||$inside) ? [ $evt, $collector ] : $collector;
+ return ($passthrough||$content) ? [ $evt, $collector ] : $collector;
};
}
+sub collect_content {
+ my ($self, $options) = @_;
+ $self->collect({ %{$options||{}}, content => 1 })
+}
+
sub add_before {
my ($self, $events) = @_;
sub { return $self->_stream_from_array(@$events, $_[0]) };
sub add_after {
my ($self, $events) = @_;
+ my $coll_proto = $self->collect({ passthrough => 1 });
sub {
my ($evt) = @_;
my $emit = $self->_stream_from_array(@$events);
- my $coll = $self->collect({ passthrough => 1 })->(@_);
+ my $coll = &$coll_proto;
return ref($coll) eq 'HASH' # single event, no collect
? [ $coll, $emit ]
: [ $coll->[0], $self->_stream_concat($coll->[1], $emit) ];
};
}
-sub prepend_inside {
+sub prepend_content {
my ($self, $events) = @_;
sub {
my ($evt) = @_;
};
}
-sub append_inside {
+sub append_content {
my ($self, $events) = @_;
+ my $coll_proto = $self->collect({ passthrough => 1, content => 1 });
sub {
my ($evt) = @_;
if ($evt->{is_in_place_close}) {
@$events, { type => 'CLOSE', name => $evt->{name} }
) ];
}
- my $coll = $self->collect({ passthrough => 1, inside => 1 })->(@_);
+ my $coll = &$coll_proto;
my $emit = $self->_stream_from_array(@$events);
return [ $coll->[0], $self->_stream_concat($coll->[1], $emit) ];
};
}
sub replace {
- my ($self, $events, $options) = @_;
+ my ($self, $replace_with, $options) = @_;
+ my $coll_proto = $self->collect($options);
sub {
my ($evt, $stream) = @_;
- my $emit = $self->_stream_from_array(@$events);
- my $coll = $self->collect($options)->(@_);
+ my $emit = $self->_stream_from_proto($replace_with);
+ my $coll = &$coll_proto;
+ # For a straightforward replace operation we can, in fact, do the emit
+ # -before- the collect, and my first cut did so. However in order to
+ # use the captured content in generating the new content, we need
+ # the collect stage to happen first - and it seems highly unlikely
+ # that in normal operation the collect phase will take long enough
+ # for the difference to be noticeable
return
($coll
? (ref $coll eq 'ARRAY'
- ? [ $coll->[0], $self->_stream_concat($emit, $coll->[1]) ]
- : $self->_stream_concat($emit, $coll)
+ ? [ $coll->[0], $self->_stream_concat($coll->[1], $emit) ]
+ : $self->_stream_concat($coll, $emit)
)
: $emit
);
};
}
+sub replace_content {
+ my ($self, $replace_with, $options) = @_;
+ $self->replace($replace_with, { %{$options||{}}, content => 1 })
+}
+
+sub repeat {
+ my ($self, $repeat_for, $options) = @_;
+ $options->{into} = \my @into;
+ my @between;
+ my $repeat_between = delete $options->{repeat_between};
+ if ($repeat_between) {
+ require HTML::Zoom::SelectorParser;
+ require HTML::Zoom::FilterStream;
+ my $sp = HTML::Zoom::SelectorParser->new;
+ my $filter = $self->collect({ into => \@between });
+ $options->{filter} = sub {
+ HTML::Zoom::FilterStream->new({
+ stream => $_[0],
+ match => $sp->parse_selector($repeat_between),
+ filter => $filter
+ })
+ };
+ }
+ my $repeater = sub {
+ my $s = $self->_stream_from_proto($repeat_for);
+ # We have to test $repeat_between not @between here because
+ # at the point we're constructing our return stream @between
+ # hasn't been populated yet - but we can test @between in the
+ # map routine because it has been by then and that saves us doing
+ # the extra stream construction if we don't need it.
+ if ($repeat_between) {
+ $s->map(sub {
+ local $_ = $self->_stream_from_array(@into);
+ (@between && $s->peek)
+ ? $self->_stream_concat(
+ $_[0]->($_), $self->_stream_from_array(@between)
+ )
+ : $_[0]->($_)
+ })
+ ->flatten;
+ } else {
+ $s->map(sub {
+ local $_ = $self->_stream_from_array(@into);
+ $_[0]->($_)
+ })
+ ->flatten;
+ }
+ };
+ $self->replace($repeater, $options);
+}
+
+sub repeat_content {
+ my ($self, $repeat_for, $options) = @_;
+ $self->repeat($repeat_for, { %{$options||{}}, content => 1 })
+}
+
1;