Bug 14939: Modularize OAI Server existing classes
[koha.git] / Koha / OAI / Server / ListRecords.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::ListRecords;
20
21 use Modern::Perl;
22 use C4::Biblio;
23 use HTTP::OAI;
24 use Koha::OAI::Server::ResumptionToken;
25 use Koha::OAI::Server::Record;
26 use Koha::OAI::Server::DeletedRecord;
27 use C4::OAI::Sets;
28 use MARC::File::XML;
29
30 use base ("HTTP::OAI::ListRecords");
31
32
33 sub new {
34     my ($class, $repository, %args) = @_;
35
36     my $self = HTTP::OAI::ListRecords->new(%args);
37
38     my $token = new Koha::OAI::Server::ResumptionToken( %args );
39     my $dbh = C4::Context->dbh;
40     my $set;
41     if(defined $token->{'set'}) {
42         $set = GetOAISetBySpec($token->{'set'});
43     }
44     my $max = $repository->{koha_max_count};
45     my $sql = "
46         (SELECT biblioitems.biblionumber, biblioitems.timestamp, marcxml
47         FROM biblioitems
48     ";
49     $sql .= " JOIN oai_sets_biblios ON biblioitems.biblionumber = oai_sets_biblios.biblionumber " if defined $set;
50     $sql .= " WHERE timestamp >= ? AND timestamp <= ? ";
51     $sql .= " AND oai_sets_biblios.set_id = ? " if defined $set;
52     $sql .= ") UNION
53         (SELECT deletedbiblio.biblionumber, null as marcxml, timestamp FROM deletedbiblio";
54     $sql .= " JOIN oai_sets_biblios ON deletedbiblio.biblionumber = oai_sets_biblios.biblionumber " if defined $set;
55     $sql .= " WHERE DATE(timestamp) >= ? AND DATE(timestamp) <= ? ";
56     $sql .= " AND oai_sets_biblios.set_id = ? " if defined $set;
57
58     $sql .= ") ORDER BY biblionumber
59         LIMIT " . ($max + 1) . "
60         OFFSET $token->{offset}
61     ";
62     my $sth = $dbh->prepare( $sql );
63     my @bind_params = ($token->{'from_arg'}, $token->{'until_arg'});
64     push @bind_params, $set->{'id'} if defined $set;
65     push @bind_params, ($token->{'from'}, $token->{'until'});
66     push @bind_params, $set->{'id'} if defined $set;
67     $sth->execute( @bind_params );
68
69     my $count = 0;
70     my $format = $args{metadataPrefix} || $token->{metadata_prefix};
71     while ( my ($biblionumber, $timestamp) = $sth->fetchrow ) {
72         $count++;
73         if ( $count > $max ) {
74             $self->resumptionToken(
75                 new Koha::OAI::Server::ResumptionToken(
76                     metadataPrefix  => $token->{metadata_prefix},
77                     from            => $token->{from},
78                     until           => $token->{until},
79                     offset          => $token->{offset} + $max,
80                     set             => $token->{set}
81                 )
82             );
83             last;
84         }
85         my $marcxml = $repository->get_biblio_marcxml($biblionumber, $format);
86         my $oai_sets = GetOAISetsBiblio($biblionumber);
87         my @setSpecs;
88         foreach (@$oai_sets) {
89             push @setSpecs, $_->{spec};
90         }
91         if ($marcxml) {
92           $self->record( Koha::OAI::Server::Record->new(
93               $repository, $marcxml, $timestamp, \@setSpecs,
94               identifier      => $repository->{ koha_identifier } . ':' . $biblionumber,
95               metadataPrefix  => $token->{metadata_prefix}
96           ) );
97         } else {
98           $self->record( Koha::OAI::Server::DeletedRecord->new(
99           $timestamp, \@setSpecs, identifier => $repository->{ koha_identifier } . ':' . $biblionumber ) );
100         }
101     }
102
103     # Return error if no results
104     unless ($count) {
105         return HTTP::OAI::Response->new(
106             requestURL => $repository->self_url(),
107             errors     => [ new HTTP::OAI::Error( code => 'noRecordsMatch' ) ],
108         );
109     }
110
111     return $self;
112 }
113
114 1;