File Coverage

blib/lib/Nephia/Plugin/Teng.pm

Criterion	Covered	Total	%
statement	58	60	96.6
branch	2	4	50.0
condition	6	15	40.0
subroutine	16	16	100.0
pod	3	4	75.0
total	85	99	85.8

package Nephia::Plugin::Teng;
use 5.008005;
use strict;
use warnings;
use Teng::Schema::Loader;
use DBI;
use parent 'Nephia::Plugin';

our $VERSION = "0.04";

sub new {
    my ($class, %opts) = @_;
    my $self = $class->SUPER::new(%opts);
    $self->{RUN_SQL} = [];
    return $self;
}

sub exports { qw/database_do teng/ }

sub database_do {
    my $self = shift;
    my $context = shift;
    $self->_load_context_config($context);
    return sub ($) {
        my $sql = shift;
        push @{$self->{RUN_SQL}}, $sql;
    };
}

sub _on_connect_do {
    my $self = shift;
    my $dbh = $self->_create_dbh();
    my @RUN_SQL = @{$self->{RUN_SQL}};
    for my $sql (@RUN_SQL) {
        $dbh->do($sql);
    }
}

sub teng {
    my $self = shift;
    my $context = shift;
    $self->_load_context_config($context);
    return sub (@) {
        $self->{teng} ||= $self->_create_teng();
    };
}

sub _create_teng {
    my $self = shift;
    my $config = $self->{opts};
    my $pkg = $config->{teng_namespace} // $self->app->{caller}.'::DB';

    $self->_on_connect_do();

    my $teng =
        Teng::Schema::Loader->load(
            dbh => $self->_create_dbh(),
            namespace => $pkg
        );

    for my $plugin (@{$config->{teng_plugins}}) {
        $pkg->load_plugin($plugin);
    }

    return $teng;
};

sub _create_dbh {
    my $self = shift;
    my @connect_info;

    @connect_info = @{$self->{opts}->{connect_info}}
        if exists $self->{opts}->{connect_info}
            && ref $self->{opts}->{connect_info} eq "ARRAY";
    @connect_info = @{$self->{config}->{DBI}->{connect_info}}
        if !@connect_info
            && exists $self->{config}->{DBI}->{connect_info}
            && ref $self->{config}->{DBI}->{connect_info} eq "ARRAY";

    return DBI->connect(@connect_info);
}

sub _load_context_config {
    my $self = shift;
    my $context = shift;
    $self->{config} = $context->get('config');
}

1;
__END__