package DBIx::Custom::Result;

use base 'Object::Simple::Base';

use strict;

use warnings;

use Carp 'croak';

use Object::Simple::Util;

my $p = __PACKAGE__;

$p->attr([qw/_dbi sth fetch_filter/])

  ->attr(_no_fetch_filters_map => sub { {} });

$p->attr(no_fetch_filters => (type => 'array', trigger => sub {

    my $self = shift;

    my $no_fetch_filters = $self->no_fetch_filters || [];

    my %no_fetch_filters_map = map {$_ => 1} @{$no_fetch_filters};

    $self->_no_fetch_filters_map(\%no_fetch_filters_map);

}));

sub new {

    my $self = shift->SUPER::new(@_);

    Object::Simple::Util->init_attrs($self, 'no_fetch_filters');

    return $self;

}

# Fetch (array)

sub fetch {

    my ($self, $type) = @_;

    my $sth = $self->sth;

    my $fetch_filter = $self->fetch_filter;

    # Fetch

    my $row = $sth->fetchrow_arrayref;

    # Cannot fetch

    return unless $row;

    # Filter

    if ($fetch_filter) {

        my $keys  = $sth->{NAME_lc};

        my $types = $sth->{TYPE};

        for (my $i = 0; $i < @$keys; $i++) {

            next if $self->_no_fetch_filters_map->{$keys->[$i]};

            $row->[$i]= $fetch_filter->($row->[$i], $keys->[$i], $self->_dbi,

                                        {type => $types->[$i], sth => $sth, index => $i});

        }

    }

    return wantarray ? @$row : $row;

}

# Fetch (hash)

sub fetch_hash {

    my $self = shift;

    my $sth = $self->sth;

    my $fetch_filter = $self->fetch_filter;

    # Fetch

    my $row = $sth->fetchrow_arrayref;

    # Cannot fetch

    return unless $row;

    # Keys

    my $keys  = $sth->{NAME_lc};

    # Filter

    my $row_hash = {};

    if ($fetch_filter) {

        my $types = $sth->{TYPE};

        for (my $i = 0; $i < @$keys; $i++) {

            if ($self->_no_fetch_filters_map->{$keys->[$i]}) {

                $row_hash->{$keys->[$i]} = $row->[$i];

            }

            else {

                $row_hash->{$keys->[$i]}

                  = $fetch_filter->($row->[$i], $keys->[$i], $self->_dbi,

                                    {type => $types->[$i], sth => $sth, index => $i});

            }

        }

    }

    # No filter

    else {

        for (my $i = 0; $i < @$keys; $i++) {

            $row_hash->{$keys->[$i]} = $row->[$i];

        }

    }

    return wantarray ? %$row_hash : $row_hash;

}

# Fetch only first (array)

sub fetch_first {

    my $self = shift;

    # Fetch

    my $row = $self->fetch;

    # Not exist

    return unless $row;

    # Finish statement handle

    $self->finish;

    return wantarray ? @$row : $row;

}

# Fetch only first (hash)

sub fetch_hash_first {

    my $self = shift;

    # Fetch hash

    my $row = $self->fetch_hash;

    # Not exist

    return unless $row;

    # Finish statement handle

    $self->finish;

    return wantarray ? %$row : $row;

}

# Fetch multi rows (array)

sub fetch_rows {

    my ($self, $count) = @_;

    # Not specified Row count

    croak("Row count must be specified")

      unless $count;

    # Fetch multi rows

    my $rows = [];

    for (my $i = 0; $i < $count; $i++) {

        my @row = $self->fetch;

        last unless @row;

        push @$rows, \@row;

    }

    return unless @$rows;

    return wantarray ? @$rows : $rows;

}

# Fetch multi rows (hash)

sub fetch_hash_rows {

    my ($self, $count) = @_;

    # Not specified Row count

    croak("Row count must be specified")

      unless $count;

    # Fetch multi rows

    my $rows = [];

    for (my $i = 0; $i < $count; $i++) {

        my %row = $self->fetch_hash;

        last unless %row;

        push @$rows, \%row;

    }

    return unless @$rows;

    return wantarray ? @$rows : $rows;

}

# Fetch all (array)

sub fetch_all {

    my $self = shift;

    my $rows = [];

    while(my @row = $self->fetch) {

        push @$rows, [@row];

    }

    return wantarray ? @$rows : $rows;

}

# Fetch all (hash)

sub fetch_hash_all {

    my $self = shift;

    my $rows = [];

    while(my %row = $self->fetch_hash) {

        push @$rows, {%row};

    }

    return wantarray ? @$rows : $rows;

}

# Finish

sub finish { shift->sth->finish }

# Error

sub error { 

    my $self = shift;

    my $sth  = $self->sth;

    return wantarray ? ($sth->errstr, $sth->err, $sth->state) : $sth->errstr;

}

=head1 NAME

DBIx::Custom::Result - DBIx::Custom Resultset

=head1 Synopsis

    my $result = $dbi->query($query);

    # Fetch

    while (my @row = $result->fetch) {

        # Do something

    }

    # Fetch hash

    while (my %row = $result->fetch_hash) {

        # Do something

    }

=head1 Accessors

=head2 sth

Set and Get statement handle

    $result = $result->sth($sth);

    $sth    = $reuslt->sth

=head2 fetch_filter

Set and Get fetch filter

    $result         = $result->fetch_filter($sth);

    $fetch_filter   = $result->fech_filter;

=head2 no_fetch_filters

Set and Get no filter keys when fetching

    $result           = $result->no_fetch_filters($no_fetch_filters);

    $no_fetch_filters = $result->no_fetch_filters;

=head1 Methods

=head2 new

    my $result = DBIx::Custom::Result->new;

=head2 fetch

Fetch a row

    $row = $result->fetch; # array reference

    @row = $result->fecth; # array

The following is fetch sample

    while (my $row = $result->fetch) {

        # do something

        my $val1 = $row->[0];

        my $val2 = $row->[1];

    }

=head2 fetch_hash

Fetch row as hash

    $row = $result->fetch_hash; # hash reference

    %row = $result->fetch_hash; # hash

The following is fetch_hash sample

    while (my $row = $result->fetch_hash) {

        # do something

        my $val1 = $row->{key1};

        my $val2 = $row->{key2};

    }

=head2 fetch_first

Fetch only first row(Scalar context)

    $row = $result->fetch_first; # array reference

    @row = $result->fetch_first; # array

The following is fetch_first sample

    $row = $result->fetch_first;

This method fetch only first row and finish statement handle

=head2 fetch_hash_first

Fetch only first row as hash

    $row = $result->fetch_hash_first; # hash reference

    %row = $result->fetch_hash_first; # hash

The following is fetch_hash_first sample

    $row = $result->fetch_hash_first;

This method fetch only first row and finish statement handle

=head2 fetch_rows

Fetch rows

    $rows = $result->fetch_rows($row_count); # array ref of array ref

    @rows = $result->fetch_rows($row_count); # array of array ref

The following is fetch_rows sample

    while(my $rows = $result->fetch_rows(10)) {

        # do someting

    }

=head2 fetch_hash_rows

Fetch rows as hash

    $rows = $result->fetch_hash_rows($row_count); # array ref of hash ref

    @rows = $result->fetch_hash_rows($row_count); # array of hash ref

The following is fetch_hash_rows sample

    while(my $rows = $result->fetch_hash_rows(10)) {

        # do someting

    }

=head2 fetch_all

Fetch all rows

    $rows = $result->fetch_all; # array ref of array ref

    @rows = $result->fecth_all; # array of array ref

The following is fetch_all sample

    my $rows = $result->fetch_all;

=head2 fetch_hash_all

Fetch all row as array ref of hash ref (Scalar context)

    $rows = $result->fetch_hash_all; # array ref of hash ref

    @rows = $result->fecth_all_hash; # array of hash ref

The following is fetch_hash_all sample

    my $rows = $result->fetch_hash_all;

=head2 error

Get error infomation

    $error_messege = $result->error;

    ($error_message, $error_number, $error_state) = $result->error;

You can get get information. This is same as the following.

    $error_message : $result->sth->errstr

    $error_number  : $result->sth->err

    $error_state   : $result->sth->state

=head2 finish

Finish statement handle

    $result->finish

This is equel to

    $result->sth->finish;

=head1 See also

L<DBIx::Custom>

=head1 Author

Yuki Kimoto, C<< <kimoto.yuki at gmail.com> >>

Github L<http://github.com/yuki-kimoto>

=head1 Copyright & licence

Copyright 2009 Yuki Kimoto, all rights reserved.

This program is free software; you can redistribute it and/or modify it

under the same terms as Perl itself.

=cut