summaryrefslogtreecommitdiffstats
path: root/lib/MooseX/Mangle.pm
blob: c555a6a4920894d89c142750a5d40994b860f802 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package MooseX::Mangle;
use Moose ();
use Moose::Exporter;

sub mangle_args {
    my $caller = shift;
    my ($method_name, $code) = @_;
    my $meta = Class::MOP::class_of($caller);
    $meta->add_around_method_modifier($method_name => sub {
        my $orig = shift;
        my $self = shift;
        my @args = $self->$code(@_);
        return $self->$orig(@args);
    });
}

sub mangle_return {
    my $caller = shift;
    my ($method_name, $code) = @_;
    my $meta = Class::MOP::class_of($caller);
    $meta->add_around_method_modifier($method_name => sub {
        my $orig = shift;
        my $self = shift;
        if (wantarray) {
            my @ret = $self->$orig(@_);
            return $self->$code(@ret);
        }
        else {
            my $ret = $self->$orig(@_);
            return $self->$code($ret);
        }
    });
}

Moose::Exporter->setup_import_methods(
    with_caller => [qw(mangle_args mangle_return)],
);

1;