File Coverage

blib/lib/DBIx/Simple/Inject.pm

Criterion	Covered	Total	%
statement	33	54	61.1
branch	0	14	0.0
condition	0	3	0.0
subroutine	11	15	73.3
pod			n/a
total	44	86	51.1

package DBIx::Simple::Inject;
use 5.008001;
use strict;
use warnings;
our $VERSION = '0.04';
use parent 'DBI';

package DBIx::Simple::Inject::db;
use strict;
our @ISA = qw(DBI::db);

use Class::Load;
use DBIx::Simple;
use Scalar::Util qw/weaken/;

sub simple {
    my ($dbh) = @_;
    $dbh->{private_dbixsimple_object} ||= do {
        my $dbis = DBIx::Simple->connect($dbh);
        weaken($dbis->{dbh});

        for my $k (keys %{ $dbh->{private_dbixsimple} || {} }) {
            my $v = $dbh->{private_dbixsimple}{$k};
            # lvalue method
            $dbis->$k = ref $v eq 'CODE' ? $v->($dbh)
                      : $k eq 'abstract' ? _abstract($dbis->{dbh}, $v) : $v;
        }

        $dbis;
    };
}

sub _abstract {
    my ($dbh, $class) = @_;
    Class::Load::load_class($class);
    if ($class eq 'SQL::Abstract') {
        $class->new();
    } elsif ($class eq 'SQL::Abstract::Limit') {
        $class->new(limit_dialect => $dbh);
    } elsif ($class eq 'SQL::Maker') {
        $class->new(driver => $dbh->{Driver}{Name});
    } else {
        $class->new($dbh); # fallback
    }
}

{
    no strict 'refs';
    for my $method (
        qw(
            error
            query
            begin
            disconnect
            select insert update delete
            iquery
        ),
        # unnecessary begin_work(), commit(), rollback(), func() and last_insert_id()
        # there are just alias for DBI::db::*
    ) {
        *$method = sub {
            use strict 'refs';
            shift->simple->$method(@_);
        };
    }

    for my $property (
        qw(
            keep_statements
            lc_columns
            result_class
            abstract
        ),
    ) {
        *$property = sub {
            my ($self, $val) = @_;
            use strict 'refs';
            if ($val) {
                $self->simple->$property = $val;
            } else {
                $self->simple->$property;
            }
        };
    }
}

package DBIx::Simple::Inject::st;
our @ISA = qw(DBI::st);

1;

__END__