package DBIx::Custom::Result;

use Object::Simple -base;

use Carp 'croak';

use DBIx::Custom::Util qw/_array_to_hash _subname/;

has [qw/dbi sth/],

  stash => sub { {} };

*all = \&fetch_hash_all;

sub column {

  my $self = shift;

  my $column = [];

  my $rows = $self->fetch_all;

  push @$column, $_->[0] for @$rows;

  return $column;

}

sub filter {

  my $self = shift;

  # Set

  if (@_) {

    # Convert filter name to subroutine

    my $filter = @_ == 1 ? $_[0] : [@_];

    $filter = _array_to_hash($filter);

    for my $column (keys %$filter) {

      my $fname = $filter->{$column};

      if  (exists $filter->{$column}

        && defined $fname

        && ref $fname ne 'CODE') 

      {

        croak qq{Filter "$fname" is not registered" } . _subname

          unless exists $self->dbi->filters->{$fname};

        $filter->{$column} = $self->dbi->filters->{$fname};

      }

    }

    # Merge

    $self->{filter} = {%{$self->filter}, %$filter};

    return $self;

  }

  return $self->{filter} ||= {};

}

sub fetch {

  my $self = shift;

  # Info

  $self->_cache unless $self->{_cache};

  # Fetch

  my @row = $self->{sth}->fetchrow_array;

  return unless @row;

  # Type rule

  if ($self->{type_rule}->{from1} && !$self->{type_rule_off} && !$self->{type_rule1_off}) {

    my $from = $self->{type_rule}->{from1};

    for my $type (keys %$from) {

      for my $column (@{$self->{_type_map}->{$type}}) {

        $row[$_] = $from->{$type}->($row[$_])

          for @{$self->{_pos}{$column} || []};

      }

    }

  }

  if ($self->{type_rule}->{from2} && !$self->{type_rule_off} && !$self->{type_rule2_off}) {

    my $from = $self->{type_rule}->{from2};

    for my $type (keys %$from) {

      for my $column (@{$self->{_type_map}->{$type}}) {

        $row[$_] = $from->{$type}->($row[$_])

          for @{$self->{_pos}{$column} || []};

      }

    }

  }

  # Filter

  if (($self->{filter} || $self->{default_filter}) && !$self->{filter_off}) {

     my @columns = $self->{default_filter} ? keys %{$self->{_columns}}

       : keys %{$self->{filter}};

     for my $column (@columns) {

       my $filter = exists $self->{filter}->{$column} ? $self->{filter}->{$column}

         : $self->{default_filter};

       next unless $filter;

       $row[$_] = $filter->($row[$_])

         for @{$self->{_pos}{$column} || []};

     }

  }

  if ($self->{end_filter} && !$self->{filter_off}) {

     for my $column (keys %{$self->{end_filter}}) {

       next unless $self->{end_filter}->{$column};

       $row[$_] = $self->{end_filter}->{$column}->($row[$_])

         for @{$self->{_pos}{$column} || []};

     }

  }

  return \@row;

}

sub fetch_hash {

  my $self = shift;

  # Info

  $self->_cache unless $self->{_cache};

  # Fetch

  return unless my $row = $self->{sth}->fetchrow_hashref;

  # Type rule

  if ($self->{type_rule}->{from1} &&

    !$self->{type_rule_off} && !$self->{type_rule1_off})

  {

    my $from = $self->{type_rule}->{from1};

    for my $type (keys %$from) {

      $from->{$type} and $row->{$_} = $from->{$type}->($row->{$_})

        for @{$self->{_type_map}->{$type}};

    }

  }

  if ($self->{type_rule}->{from2} &&

    !$self->{type_rule_off} && !$self->{type_rule2_off})

  {

    my $from = $self->{type_rule}->{from2};

    for my $type (keys %{$self->{type_rule}->{from2}}) {

      $from->{$type} and $row->{$_} = $from->{$type}->($row->{$_})

        for @{$self->{_type_map}->{$type}};

    }

  }        

  # Filter

  if (($self->{filter} || $self->{default_filter}) &&

    !$self->{filter_off})

  {

     my @columns = $self->{default_filter} ? keys %{$self->{_columns}}

       : keys %{$self->{filter}};

     for my $column (@columns) {

       next unless exists $row->{$column};

       my $filter = exists $self->{filter}->{$column} ? $self->{filter}->{$column}

         : $self->{default_filter};

       $row->{$column} = $filter->($row->{$column}) if $filter;

     }

  }

  if ($self->{end_filter} && !$self->{filter_off}) {

     exists $self->{_columns}{$_} && $self->{end_filter}->{$_} and

         $row->{$_} = $self->{end_filter}->{$_}->($row->{$_})

       for keys %{$self->{end_filter}};

  }

  $row;

}

sub fetch_all {

  my $self = shift;

  # Fetch all rows

  my $rows = [];

  while(my $row = $self->fetch) { push @$rows, $row}

  return $rows;

}

sub fetch_hash_all {

  my $self = shift;

  # Fetch all rows as hash

  my $rows = [];

  while(my $row = $self->fetch_hash) { push @$rows, $row }

  return $rows;

}

sub fetch_hash_one {

  my $self = shift;

  # Fetch hash

  my $row = $self->fetch_hash;

  return unless $row;

  # Finish statement handle

  $self->sth->finish;

  return $row;

}

sub fetch_hash_multi {

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

  # Fetch multiple rows

  croak 'Row count must be specified ' . _subname

    unless $count;

  return if $self->{_finished};

  my $rows = [];

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

    my $row = $self->fetch_hash;

    unless ($row) {

      $self->{_finished} = 1;

      last;

    }

    push @$rows, $row;

  }

  return unless @$rows;

  return $rows;

}

sub fetch_multi {

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

  # Row count not specifed

  croak 'Row count must be specified ' . _subname

    unless $count;

  return if $self->{_finished};

  # Fetch multi rows

  my $rows = [];

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

    my $row = $self->fetch;

    unless ($row) {

      $self->{_finished} = 1;

      last;

    }

    push @$rows, $row;

  }

  return unless @$rows;

  return $rows;

}

sub fetch_one {

  my $self = shift;

  # Fetch

  my $row = $self->fetch;

  return unless $row;

  # Finish statement handle

  $self->sth->finish;

  return $row;

}

sub header { shift->sth->{NAME} }

*one = \&fetch_hash_one;

sub type_rule {

  my $self = shift;

  if (@_) {

    my $type_rule = ref $_[0] eq 'HASH' ? $_[0] : {@_};

    # From

    for my $i (1 .. 2) {

      $type_rule->{"from$i"} = _array_to_hash($type_rule->{"from$i"});

      for my $data_type (keys %{$type_rule->{"from$i"} || {}}) {

        croak qq{data type of from$i section must be lower case or number}

          if $data_type =~ /[A-Z]/;

        my $fname = $type_rule->{"from$i"}{$data_type};

        if (defined $fname && ref $fname ne 'CODE') {

          croak qq{Filter "$fname" is not registered" } . _subname

            unless exists $self->dbi->filters->{$fname};

          $type_rule->{"from$i"}{$data_type} = $self->dbi->filters->{$fname};

        }

      }

    }

    $self->{type_rule} = $type_rule;

    return $self;

  }

  return $self->{type_rule} || {};

}

sub type_rule_off {

  my $self = shift;

  $self->{type_rule_off} = 1;

  return $self;

}

sub type_rule_on {

  my $self = shift;

  $self->{type_rule_off} = 0;

  return $self;

}

sub type_rule1_off {

  my $self = shift;

  $self->{type_rule1_off} = 1;

  return $self;

}

sub type_rule1_on {

  my $self = shift;

  $self->{type_rule1_off} = 0;

  return $self;

}

sub type_rule2_off {

  my $self = shift;

  $self->{type_rule2_off} = 1;

  return $self;

}

sub type_rule2_on {

  my $self = shift;

  $self->{type_rule2_off} = 0;

  return $self;

}

sub value {

  my $self = shift;

  my $row = $self->fetch_one;

  my $value = $row ? $row->[0] : undef;

  return $value;

}

sub _cache {

  my $self = shift;

  $self->{_type_map} = {};

  $self->{_pos} = {};

  $self->{_columns} = {};

  for (my $i = 0; $i < @{$self->{sth}->{NAME} || []}; $i++) {

    my $type = lc $self->{sth}{TYPE}[$i];

    my $name = $self->{sth}{NAME}[$i];

    $self->{_type_map}{$type} ||= [];

    push @{$self->{_type_map}{$type}}, $name;

    $self->{_pos}{$name} ||= [];

    push @{$self->{_pos}{$name}}, $i;

    $self->{_columns}{$name} = 1;

  }

  $self->{_cache} = 1;

}

# DEPRECATED!

sub fetch_hash_first {

  my $self = shift;

  warn "DBIx::Custom::Result::fetch_hash_first is DEPRECATED! use fetch_hash_one instead";

  return $self->fetch_hash_one(@_);

}

# DEPRECATED!

sub fetch_first {

  my $self = shift;

  warn "DBIx::Custom::Result::fetch_first is DEPRECATED! use fetch_one instead";

  return $self->fetch_one(@_);

}

# DEPRECATED!

sub filter_off {

  warn "filter_off method is DEPRECATED!";

  my $self = shift;

  $self->{filter_off} = 1;

  return $self;

}

# DEPRECATED!

sub filter_on {

  warn "filter_on method is DEPRECATED!";

  my $self = shift;

  $self->{filter_off} = 0;

  return $self;

}

# DEPRECATED!

sub end_filter {

  warn "end_filter method is DEPRECATED!";

  my $self = shift;

  if (@_) {

    my $end_filter = {};

    if (ref $_[0] eq 'HASH') { $end_filter = $_[0] }

    else { 

      $end_filter = _array_to_hash(

          @_ > 1 ? [@_] : $_[0]

      );

    }

    for my $column (keys %$end_filter) {

      my $fname = $end_filter->{$column};

      if (exists $end_filter->{$column}

        && defined $fname

        && ref $fname ne 'CODE') 

      {

        croak qq{Filter "$fname" is not registered" } . _subname

          unless exists $self->dbi->filters->{$fname};

        $end_filter->{$column} = $self->dbi->filters->{$fname};

      }

    }

    $self->{end_filter} = {%{$self->end_filter}, %$end_filter};

    return $self;

  }

  return $self->{end_filter} ||= {};

}

# DEPRECATED!

sub remove_end_filter {

  warn "remove_end_filter is DEPRECATED!";

  my $self = shift;

  $self->{end_filter} = {};

  return $self;

}

# DEPRECATED!

sub remove_filter {

  warn "remove_filter is DEPRECATED!";

  my $self = shift;

  $self->{filter} = {};

  return $self;

}

# DEPRECATED!

sub default_filter {

  warn "default_filter is DEPRECATED!";

  my $self = shift;

  if (@_) {

    my $fname = $_[0];

    if (@_ && !$fname) {

      $self->{default_filter} = undef;

    }

    else {

      croak qq{Filter "$fname" is not registered}

        unless exists $self->dbi->filters->{$fname};

      $self->{default_filter} = $self->dbi->filters->{$fname};

    }

    return $self;

  }

  return $self->{default_filter};

}

# DEPRECATED!

has 'filter_check'; 

1;

=head1 NAME

DBIx::Custom::Result - Result of select statement

=head1 SYNOPSIS

  # Result

  my $result = $dbi->select(table => 'book');

  # Fetch a row and put it into array reference

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

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

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

  }

  # Fetch only a first row and put it into array reference

  my $row = $result->fetch_one;

  # Fetch all rows and put them into array of array reference

  my $rows = $result->fetch_all;

  # Fetch a row and put it into hash reference

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

    my $title  = $row->{title};

    my $author = $row->{author};

  }

  # Fetch only a first row and put it into hash reference

  my $row = $result->fetch_hash_one;

  my $row = $result->one; # Alias for "fetch_hash_one"

  # Fetch all rows and put them into array of hash reference

  my $rows = $result->fetch_hash_all;

  my $rows = $result->all; # Alias for "fetch_hash_all"

=head1 ATTRIBUTES

=head2 C<dbi>

  my $dbi = $result->dbi;

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

L<DBIx::Custom> object.

=head2 C<sth>

  my $sth = $reuslt->sth

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

Statement handle of L<DBI>.

=head1 METHODS

L<DBIx::Custom::Result> inherits all methods from L<Object::Simple>

and implements the following new ones.

=head2 C<all>

  my $rows = $result->all;

Same as C<fetch_hash_all>.

=head2 C<column>

  my $column = $result->column;

Get first column's all values.

  my $names = $dbi->select('name', table => 'book')->column;

=head2 C<fetch>

  my $row = $result->fetch;

Fetch a row and put it into array reference.

=head2 C<fetch_all>

  my $rows = $result->fetch_all;

Fetch all rows and put them into array of array reference.

=head2 C<fetch_one>

  my $row = $result->fetch_one;

Fetch only a first row and put it into array reference,

and finish statment handle.

=head2 C<fetch_hash>

  my $row = $result->fetch_hash;

Fetch a row and put it into hash reference.

=head2 C<fetch_hash_all>

  my $rows = $result->fetch_hash_all;

Fetch all rows and put them into array of hash reference.

=head2 C<fetch_hash_one>

  my $row = $result->fetch_hash_one;

Fetch only a first row and put it into hash reference,

and finish statment handle.

=head2 C<fetch_hash_multi>

  my $rows = $result->fetch_hash_multi(5);

Fetch multiple rows and put them into array of hash reference.

=head2 C<fetch_multi>

  my $rows = $result->fetch_multi(5);

Fetch multiple rows and put them into array of array reference.

=head2 C<filter>

  $result->filter(title  => sub { uc $_[0] }, author => 'to_upper');

  $result->filter([qw/title author/] => 'to_upper');

Set filter for column.

You can use subroutine or filter name as filter.

This filter is executed after C<type_rule> filter.

=head2 C<header>

  my $header = $result->header;

Get header column names.

=head2 C<one>

  my $row = $result->one;

Alias for C<fetch_hash_one>.

=head2 C<stash>

  my $stash = $result->stash;

  my $foo = $result->stash->{foo};

  $result->stash->{foo} = $foo;

Stash is hash reference to save some data.

=head2 C<type_rule>

  # Merge type rule

  $result->type_rule(

    # DATE

    9 => sub { ... },

    # DATETIME or TIMESTAMP

    11 => sub { ... }

  );

  # Replace type rule(by reference)

  $result->type_rule([

    # DATE

    9 => sub { ... },

    # DATETIME or TIMESTAMP

    11 => sub { ... }

  ]);

This is same as L<DBIx::Custom>'s C<type_rule>'s <from>.

=head2 C<type_rule_off>

  $result = $result->type_rule_off;

Turn C<from1> and C<from2> type rule off.

By default, type rule is on.

=head2 C<type_rule_on>

  $result = $result->type_rule_on;

Turn C<from1> and C<from2> type rule on.

By default, type rule is on.

=head2 C<type_rule1_off>

  $result = $result->type_rule1_off;

Turn C<from1> type rule off.

By default, type rule is on.

=head2 C<type_rule1_on>

  $result = $result->type_rule1_on;

Turn C<from1> type rule on.

By default, type rule is on.

=head2 C<type_rule2_off>

  $result = $result->type_rule2_off;

Turn C<from2> type rule off.

By default, type rule is on.

=head2 C<type_rule2_on>

  $result = $result->type_rule2_on;

Turn C<from2> type rule on.

By default, type rule is on.

=head2 C<value>

  my $value = $result->value;

Get first column's first value.

  my $count = $dbi->select('count(*)')->value;

=cut