File Coverage

blib/lib/Mojolicious/Plugin/Narada.pm

Criterion	Covered	Total	%
statement	15	66	22.7
branch	0	22	0.0
condition	0	6	0.0
subroutine	5	11	45.4
pod	1	1	100.0
total	21	106	19.8

package Mojolicious::Plugin::Narada;

use Mojo::Base 'Mojolicious::Plugin';

our $VERSION = 'v1.0.1';

use MojoX::Log::Fast;
use Narada::Config qw( get_config get_config_line );
use Narada::Lock qw( unlock );
use Scalar::Util qw( weaken );

my ($Log, $Ident);
our $IN_CB = 0;


sub register {
    my ($self, $app, $conf) = @_;

    $Log = MojoX::Log::Fast->new($conf->{log});
    $Ident = $Log->ident();

    # Replace default logger with Log::Fast.
    $app->log($Log);

    # Load Mojo-specific config files.
    if ($app->can('secrets')) {
        $app->secrets([split /\n/ms, get_config('cookie.secret')]);
    } else {
        $app->secret(get_config_line('cookie.secret'));
    }
    $app->config(hypnotoad => {
        listen      => [split /\n/ms, get_config('hypnotoad/listen')],
        proxy       => get_config_line('hypnotoad/proxy'),
        accepts     => get_config_line('hypnotoad/accepts'),
        workers     => get_config_line('hypnotoad/workers'),
        pid_file    => 'var/hypnotoad.pid',
    });

    # * Fix url->path and url->base->path.
    # * Set correct ident while handler runs.
    # * unlock() if handler died.
    my $realbase = Mojo::Path->new( get_config_line('basepath') )
        ->trailing_slash(0)
        ->leading_slash(1)
        ->to_string;
    $app->hook(around_dispatch => sub {
        my ($next, $c) = @_;
        my $url = $c->req->url;
        my $base = $url->base->path;
        my $path = $url->path;
        if ($base eq q{} && $path =~ m{\A\Q$realbase\E(.*)\z}mso) {
            $path->parse($1);
        }
        $base->parse($realbase);
        $path->leading_slash(1);
        $Log->ident($url->path);
        my $err = eval { $next->(); 1 } ? undef : $@;
        unlock();
        die $err if defined $err;   ## no critic(RequireCarping)
    });

    $app->helper(proxy      => sub { return _proxy(0, @_) });
    $app->helper(weak_proxy => sub { return _proxy(1, @_) });

    return;
}

sub _proxy {
    my ($is_weak, $this, $cb, @p) = @_;
    if ($is_weak) {
        weaken($this);
    }
    my $is_global_cb= ref $this eq 'Mojolicious::Controller';
    my $ident       = $is_global_cb ? $Ident : $Log->ident;
    my $__warn__    = $SIG{__WARN__};
    return sub {
        return if !$this;
        my $cur_ident = $Log->ident($ident);
        local $SIG{__WARN__} = $__warn__;
        my $err = eval { local $IN_CB=1; $cb->($this, @p, @_); 1 } ? undef : $@;
        if (defined $err) {
            $Log->ident($ident);
            if (!$IN_CB) {
                unlock()
            }
            if ($is_global_cb || $is_weak) {
                die $err;   ## no critic(RequireCarping)
            }
            else {
                $this->reply->exception($err);
            }
        }
        $Log->ident($cur_ident);
    };
}


1; # Magic true value required at end of module
__END__