Add autowrap function.

gisle · Jan 2, 2009 · 50dc94a · 50dc94a
1 parent f63a7d6
commit 50dc94a
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/lib/Data/Dump/Trace.pm b/lib/Data/Dump/Trace.pm
@@ -3,14 +3,16 @@ package Data::Dump::Trace;
 use strict;
 
 use base 'Exporter';
-our @EXPORT_OK = qw(call mcall wrap);
+our @EXPORT_OK = qw(call mcall wrap autowrap);
 
 use Data::Dump qw(dump);
 use Term::ANSIColor qw(YELLOW CYAN RESET);
 use Carp qw(croak);
 use overload ();
 
 my %obj_name;
+my %autowrap_class;
+my %name_count;
 
 sub dumpav {
     return "(" . dump(@_) . ")" if @_ == 1;
@@ -25,6 +27,19 @@ sub dumpkv {
     return $str;
 }
 
+sub autowrap {
+    while (@_) {
+        my $class = shift;
+        my $name = shift;
+        unless ($name) {
+            $name = lc($class);
+            $name =~ s/.*:://;
+        }
+        $name = '$' . $name unless $name =~ /^\$/;
+        $autowrap_class{$class} = $name;
+    }
+}
+
 sub wrap {
     my %arg = @_;
     my $name = $arg{name} || "func";
@@ -81,7 +96,14 @@ sub mcall {
     }
     else {
         my $s = $o->$method(@_);
-        print " ==> ", CYAN, dump($s), RESET, "\n";
+        if (my $name = $autowrap_class{ref($s)}) {
+            $name .= $name_count{$name} if $name_count{$name}++;
+            print " ==> ", CYAN, $name, RESET, "\n";
+            $s = wrap(name => $name, obj => $s);
+        }
+        else {
+            print " ==> ", CYAN, dump($s), RESET, "\n";
+        }
         return $s;
     }
 }