introduce superclass for streams

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

diff --git a/lib/HTML/Zoom/CodeStream.pm b/lib/HTML/Zoom/CodeStream.pm
index 7c8fefb..585e05c 100644 (file)
--- a/lib/HTML/Zoom/CodeStream.pm
+++ b/lib/HTML/Zoom/CodeStream.pm
@@ -2,6 +2,7 @@ package HTML::Zoom::CodeStream;
 
 use strict;
 use warnings FATAL => 'all';
+use base qw(HTML::Zoom::StreamBase);
 
 sub from_array {
   my ($class, @array) = @_;
@@ -16,8 +17,27 @@ sub new {
   bless({ _code => $args->{code} }, $class);
 }
 
+sub peek {
+  my ($self) = @_;
+  if (exists $self->{_peeked}) {
+    return ($self->{_peeked});
+  }
+  if (my ($peeked) = $self->next) {
+    return ($self->{_peeked} = $peeked);
+  }
+  return;
+}
+
 sub next {
-  $_[0]->{_code}->()
+  my ($self) = @_;
+
+  # peeked entry so return that
+
+  if (exists $self->{_peeked}) {
+    return (delete $self->{_peeked});
+  }
+
+  $self->{_code}->();
 }
 
 1;