File Coverage

blib/lib/Dancer2/Plugin/Role/Shutdown.pm

Criterion	Covered	Total	%
statement	55	61	90.1
branch	13	22	59.0
condition	7	14	50.0
subroutine	11	11	100.0
pod	3	3	100.0
total	89	111	80.1

use strictures 2;

package # for internal use only
    Dancer2::Plugin::Role::Shutdown;

# ABSTRACT: Role for L

use Carp qw(croak);
use Class::Load qw(load_class);
use Scalar::Util qw(blessed);
use Moo::Role 2;

use constant NORMAL => 0;
use constant GRACEFUL => 1;

our $VERSION = '0.002'; # VERSION


has shared => (
    is => 'rwp',
    default => sub { {} },
);


has validator => (
    is => 'rw',
    default => sub {
        sub {
            my ($app, $rest, $sessid) = @_;
            return unless $sessid;
            my $sx = $app->session->expires // 0;
            $app->session->expires($rest) if $rest > $sx;
            $app->response->header(Warning => "199 Application shuts down in $rest seconds");
            return 1;
        }
    }
);


sub has_valid_session {
    my $app = shift;

    my $engine = $app->session_engine // return;

    return if $app->has_destroyed_session;

    my $session_cookie = $app->cookie( $engine->cookie_name ) // return;
    my $session_id = $session_cookie->value // return;

    eval  { $engine->retrieve( id => $session_id ) };
    return if $@;

    return $session_id;
}


sub session_status {
    my $app = shift;

    my $engine = $app->session_engine // return "unsupported";

    return "destroyed" if $app->has_destroyed_session;

    my $session_cookie = $app->cookie( $engine->cookie_name ) // return "missing";
    my $session_id = $session_cookie->value // return "empty";

    eval  { $engine->retrieve( id => $session_id ) };
    return "invalid" if $@;

    return "ok";
}


sub before_hook {
    my $self = shift;
    return unless $self->shared->{state};
    my $app  = $self->app;
    my $time = $self->shared->{final};
    my $rest = $time - time;
    if ($rest < 0) {
        $self->status(503);
        $self->halt;
    } elsif ($self->shared->{state} == GRACEFUL) {
        if (my $validator = $self->validator) {
            my $sessid = has_valid_session($app);
            unless ($validator->($app, $rest, $sessid)) {
                $self->status(503);
                $self->halt;
            }
        }
    } else {
      croak "bad state: ".$self->shared->{state};
    }
}

sub _shutdown_at {
    my $self = shift;
    croak "a validator isn't installed yet" unless ref $self->validator eq 'CODE';
    my $time = shift // 0;
    if ($time < time) {
        $time += time;
    }
    $self->shared->{final} = $time;
    $self->shared->{state} = GRACEFUL;
    return $time;
}

1;

__END__