Bug 14939: Modularize OAI Server existing classes
[koha.git] / Koha / OAI / Server / ListSets.pm
1 # Copyright Tamil s.a.r.l. 2008-2015
2 # Copyright Biblibre 2008-2015
3 #
4 # This file is part of Koha.
5 #
6 # Koha is free software; you can redistribute it and/or modify it
7 # under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 3 of the License, or
9 # (at your option) any later version.
10 #
11 # Koha is distributed in the hope that it will be useful, but
12 # WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 # GNU General Public License for more details.
15 #
16 # You should have received a copy of the GNU General Public License
17 # along with Koha; if not, see <http://www.gnu.org/licenses>.
18
19 package Koha::OAI::Server::ListSets;
20
21 use Modern::Perl;
22 use HTTP::OAI;
23 use Koha::OAI::Server::ResumptionToken;
24 use Koha::OAI::Server::Description;
25 use C4::OAI::Sets;
26
27 use base ("HTTP::OAI::ListSets");
28
29
30 sub new {
31     my ( $class, $repository, %args ) = @_;
32
33     my $self = HTTP::OAI::ListSets->new(%args);
34     my $token = Koha::OAI::Server::ResumptionToken->new(%args);
35     my $sets = GetOAISets;
36     my $pos = 0;
37     foreach my $set (@$sets) {
38         if ($pos < $token->{offset}) {
39             $pos++;
40             next;
41         }
42         my @descriptions = map {
43             Koha::OAI::Server::Description->new( setDescription => $_ );
44         } @{$set->{'descriptions'}};
45         $self->set(
46             HTTP::OAI::Set->new(
47                 setSpec => $set->{'spec'},
48                 setName => $set->{'name'},
49                 setDescription => \@descriptions,
50             )
51         );
52         $pos++;
53         last if ($pos + 1 - $token->{offset}) > $repository->{koha_max_count};
54     }
55
56     $self->resumptionToken(
57         new Koha::OAI::Server::ResumptionToken(
58             metadataPrefix => $token->{metadata_prefix},
59             offset         => $pos
60         )
61     ) if ( $pos > $token->{offset} );
62
63     return $self;
64 }
65
66 1;