File Coverage

blib/lib/Context/Preserve.pm

Criterion	Covered	Total	%
statement	28	33	84.8
branch	9	12	75.0
condition	3	3	100.0
subroutine	5	5	100.0
pod	1	1	100.0
total	46	54	85.1

package Context::Preserve; # git description: v0.02-4-g9a6a9b9
# ABSTRACT: Run code after a subroutine call, preserving the context the subroutine would have seen if it were the last statement in the caller
use strict;
use warnings;
use Carp;

use base 'Exporter';
our @EXPORT = qw(preserve_context);

our $VERSION = '0.03';

sub preserve_context(&@) {
    my $orig = shift;
    my %args = @_;

    my $replace = $args{replace};
    my $after   = $args{after};

    croak 'need an "after" or "replace" coderef'
      unless $replace || $after;

    if(!defined wantarray){
        $orig->();
        if($after){
            $after->();
        }
        else {
            $replace->();
        }
        return;
    }
    elsif(wantarray){
        my @result  = $orig->();
        if($after){
            my @ignored = $after->(@result);
        }
        else {
            @result = $replace->(@result);
        }
        return @result;
    }
    else {
        my $result  = $orig->();
        if($after){
            my $ignored = $after->($result);
        }
        else {
            $result = $replace->($result);
        }
        return $result;
    }
}

1;

__END__