Refactored pagination into search method, Sweet syntax is now in Compat

[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / Table.pm

diff --git a/lib/DBIx/Class/Table.pm b/lib/DBIx/Class/Table.pm
index 9aa00ab..451b0bd 100644 (file)
--- a/lib/DBIx/Class/Table.pm
+++ b/lib/DBIx/Class/Table.pm
@@ -4,6 +4,7 @@ use strict;
 use warnings;
 
 use DBIx::Class::ResultSet;
+use Data::Page;
 
 use base qw/Class::Data::Inheritable/;
 
@@ -15,6 +16,8 @@ __PACKAGE__->mk_classdata('table_alias'); # FIXME: Doesn't actually do anything
 
 __PACKAGE__->mk_classdata('_resultset_class' => 'DBIx::Class::ResultSet');
 
+__PACKAGE__->mk_classdata('_page_object');
+
 sub iterator_class { shift->_resultset_class(@_) }
 
 =head1 NAME 
@@ -112,9 +115,26 @@ sub search {
     $attrs = { %{ pop(@_) } };
   }
   $attrs->{where} = (@_ == 1 || ref $_[0] eq "HASH" ? shift: {@_});
+  
+  # for pagination, we create the resultset with no limit and slice it later
+  my $page = {};
+  if ( $attrs->{page} ) {
+    map { $page->{$_} = $attrs->{$_} } qw/rows page/;
+    delete $attrs->{$_} for qw/rows offset page/;
+  }
 
   my $rs = $class->resultset($attrs);
-
+  
+  if ( $page->{page} ) {
+    my $pager = Data::Page->new( 
+      $rs->count, 
+      $page->{rows} || 10, 
+      $page->{page} || 1 );
+    $class->_page_object( $pager );
+    return $rs->slice( $pager->skipped,
+      $pager->skipped + $pager->entries_per_page - 1 );
+  }
+  
   return (wantarray ? $rs->all : $rs);
 }
 
@@ -175,6 +195,17 @@ sub find_or_create {
 
 sub columns { return keys %{shift->_columns}; }
 
+=item page
+
+  $pager = $class->page;
+  
+Returns a Data::Page object for the most recent search that was performed
+using the page parameter.
+
+=cut
+
+sub page { shift->_page_object }
+
 1;
 
 =back