From e6da8dae50585765e326cbc74bf5e243738414d6 Mon Sep 17 00:00:00 2001
From: Yuval Kogman <nothingmuch@woobling.org>
Date: Fri, 19 Dec 2008 01:52:41 +0000
Subject: [PATCH] add a 'ref' callback

---
 lib/Data/Visitor/Callback.pm | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/lib/Data/Visitor/Callback.pm b/lib/Data/Visitor/Callback.pm
index 168a3d3..ddd4aaa 100644
--- 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 ) = @_;