File Coverage

blib/lib/DBIx/MoCo/Pageset.pm

Criterion	Covered	Total	%
statement	18	48	37.5
branch	2	28	7.1
condition	0	9	0.0
subroutine	4	8	50.0
pod			n/a
total	24	93	25.8

package DBIx::MoCo::Pageset;

use warnings;
use strict;

our $VERSION = '0.01';
our $AUTOLOAD;

our $PAGER_IMPLEMENTATION = 'Data::Pageset';

sub _croak { require Carp; Carp::croak(@_); }

sub import {
    my $class = shift;
    my $pkg   = caller(0);
    no strict 'refs';
    *{"$pkg\::pager"} = \&_pager;
    $PAGER_IMPLEMENTATION = shift if @_;
    ( my $pager_implementation_file = "$PAGER_IMPLEMENTATION.pm" ) =~ s{::}{/}g;
    eval { require $pager_implementation_file; };
    _croak "Unable to use $PAGER_IMPLEMENTATION: $@" if $@;
}

sub _pager {
    my $pkg  = shift;
    my @args = qw( entries_per_page current_page pages_per_set mode );
    my $opts = {};
    while ( @_ and not ref $_[0] ) {
        $opts->{ shift @args } = shift;
    }
    my $ref = ref $_[0] eq 'HASH' ? shift : {};
    return bless {
        pkg        => $pkg,
        pager      => undef,
        pager_opts => { %$opts, %$ref, },
      },
      __PACKAGE__;
}

sub AUTOLOAD {
    my $self = shift;
    ( my $method = $AUTOLOAD ) =~ s/.*://;
    if ( ref($self) && $self->{pkg}->can($method) ) {
        my $where = '';
        if ($_[1]) {
             my %args = @_;
             $where = \%args;
        } elsif ($_[0]) {
            $where = shift;
        }
        my $pager_opts = $self->{pager_opts};
        my $offset = $pager_opts->{current_page}
                          ? ( $pager_opts->{current_page} -1 ) * $pager_opts->{entries_per_page} : 0;
        my $limit = $pager_opts->{entries_per_page}
                          ? $pager_opts->{entries_per_page} : 1;
        $self->{pager} = $PAGER_IMPLEMENTATION->new( {
                'total_entries' => $self->{pkg}->count( ref $where eq 'HASH' ? $where->{where} : ''),
                %$pager_opts,
        } );

        if ( ref $where eq 'HASH' ) {
            if ( $offset ) {
                $where->{offset} = $offset;
            }
            $where->{limit} = $limit;
        }

        return ($where) ? $self->{pkg}->$method(%$where) : $self->{pkg}->$method->slice($offset, $offset + $limit -1);

    } elsif ( $PAGER_IMPLEMENTATION->can($method) ) {
        $self->{pager} or _croak("Can't call pager methods before searching");
        return $self->{pager}->$method(@_);
    } else {
        _croak(
            qq(Can't locate object method "$method" via package ) . ref($self)
              || $self );
    }
}

sub DESTROY { }

1;
__END__