Bug 32351: Fix CourseItems.t

[koha-ffzg.git] / Koha / Suggestions.pm

diff --git a/Koha/Suggestions.pm b/Koha/Suggestions.pm
index 03262bb..9b1d266 100644 (file)
--- a/Koha/Suggestions.pm
+++ b/Koha/Suggestions.pm
@@ -21,7 +21,7 @@ use Modern::Perl;
 
 
 use Koha::Database;
-
+use Koha::DateUtils qw(dt_from_string);
 use Koha::Suggestion;
 
 use base qw(Koha::Objects);
@@ -32,11 +32,71 @@ Koha::Suggestions - Koha Suggestion object set class
 
 =head1 API
 
-=head2 Class Methods
+=head2 Class methods
+
+=head3 search_limited
+
+    my $suggestions = Koha::Suggestions->search_limited( $params, $attributes );
+
+Returns all the suggestions the logged in user is allowed to see.
+
+=cut
+
+sub search_limited {
+    my ( $self, $params, $attributes ) = @_;
+
+    my $resultset = $self;
+
+    # filter on user branch
+    if (   C4::Context->preference('IndependentBranches')
+        && !C4::Context->IsSuperLibrarian() )
+    {
+        # If IndependentBranches is set and the logged in user is not superlibrarian
+        # Then we want to filter by the user's library (i.e. cannot see suggestions
+        # from other libraries)
+        my $userenv = C4::Context->userenv;
+
+        $resultset = $self->search({ branchcode => $userenv->{branch} })
+            if $userenv && $userenv->{branch};
+    }
+
+    return $resultset->search( $params, $attributes);
+}
+
+=head3 filter_by_pending
+
+    my $open = $suggestions->filter_by_pending;
+
+Filters the resultset on those that are considered pending (i.e. STATUS = ASKED).
+
+=cut
+
+sub filter_by_pending {
+    my ($self) = @_;
+
+    return $self->search( { STATUS => 'ASKED' } );
+}
+
+=head3 filter_by_suggested_days_range
+
+    my $suggestions = $suggestions->filter_by_suggested_days_range( $days );
+
+Filters the resultset on those placed within some I<$days> range.
 
 =cut
 
-=head3 type
+sub filter_by_suggested_days_range {
+    my ( $self, $days ) = @_;
+
+    my $dtf = Koha::Database->new->schema->storage->datetime_parser;
+
+    return $self->search(
+        { suggesteddate => { '>=' => $dtf->format_date( dt_from_string->subtract( days => $days ) ) } } );
+}
+
+=head2 Internal methods
+
+=head3 _type
 
 =cut