E[attr]

[catagits/HTML-Zoom.git] / lib / HTML / Zoom / StreamBase.pm

diff --git a/lib/HTML/Zoom/StreamBase.pm b/lib/HTML/Zoom/StreamBase.pm
index 7584ed6..a76a636 100644 (file)
--- a/lib/HTML/Zoom/StreamBase.pm
+++ b/lib/HTML/Zoom/StreamBase.pm
@@ -2,5 +2,78 @@ package HTML::Zoom::StreamBase;
 
 use strict;
 use warnings FATAL => 'all';
+use HTML::Zoom::MatchWithoutFilter;
+
+sub _zconfig { shift->{_zconfig} }
+
+sub peek {
+  my ($self) = @_;
+  if (exists $self->{_peeked}) {
+    return ($self->{_peeked});
+  }
+  if (my ($peeked) = $self->next) {
+    return ($self->{_peeked} = $peeked);
+  }
+  return;
+}
+
+sub flatten {
+  my $source_stream = shift;
+  require HTML::Zoom::CodeStream;
+  my $cur_stream;
+  HTML::Zoom::CodeStream->new({
+    code => sub {
+      return unless $source_stream;
+      my $next;
+      until (($next) = ($cur_stream ? $cur_stream->next : ())) {
+        unless (($cur_stream) = $source_stream->next) {
+          undef $source_stream; return;
+        }
+      }
+      return $next;
+    }
+  });
+}
+
+sub map {
+  my ($source_stream, $map_func) = @_;
+  require HTML::Zoom::CodeStream;
+  HTML::Zoom::CodeStream->new({
+    code => sub {
+      return unless $source_stream;
+      # If we were aiming for a "true" perl-like map then we should
+      # elegantly handle the case where the map function returns 0 events
+      # and the case where it returns >1 - if you're reading this comment
+      # because you wanted it to do that, now would be the time to fix it :)
+      if (my ($next) = $source_stream->next) {
+        #### XXXX collapsing this into a return doesn't work. what the
+        #### flying fornication ... -- mst
+        my $mapped = do { local $_ = $next; $map_func->($next) };
+        return $mapped;
+      }
+      undef $source_stream; return;
+    }
+  });
+}
+
+sub with_filter {
+  my ($self, $selector, $filter) = @_;
+  my $match = $self->_parse_selector($selector);
+  $self->_zconfig->stream_utils->wrap_with_filter($self, $match, $filter);
+}
+
+sub select {
+  my ($self, $selector) = @_;
+  my $match = $self->_parse_selector($selector);
+  return HTML::Zoom::MatchWithoutFilter->construct(
+    $self, $match, $self->_zconfig->filter_builder,
+  );
+}
+
+sub _parse_selector {
+  my ($self, $selector) = @_;
+  return $selector if ref($selector); # already a match sub
+  $self->_zconfig->selector_parser->parse_selector($selector);
+}
 
 1;