add a 'ref' callback

nothingmuch · Dec 19, 2008 · e6da8da · e6da8da
1 parent b6c61e0
commit e6da8da
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/lib/Data/Visitor/Callback.pm b/lib/Data/Visitor/Callback.pm
@@ -103,6 +103,18 @@ sub visit {
 	return ( @_ == 1 ? $ret[0] : @ret );
 }
 
+sub visit_ref {
+	my ( $self, $data ) = @_;
+
+	my $mapped = $self->callback( ref => $data );
+
+	if ( ref $mapped ) {
+		return $self->SUPER::visit_ref($mapped);
+	} else {
+		return $self->visit($mapped);
+	}
+}
+
 sub visit_seen {
 	my ( $self, $data, $result ) = @_;