Bug 19889: (follow-up) update DB adjustments
[koha.git] / virtualshelves / downloadshelf.pl
1 #!/usr/bin/perl
2
3 # Copyright 2009 BibLibre
4 #
5 # This file is part of Koha.
6 #
7 # Koha is free software; you can redistribute it and/or modify it
8 # under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 3 of the License, or
10 # (at your option) any later version.
11 #
12 # Koha is distributed in the hope that it will be useful, but
13 # WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
16 #
17 # You should have received a copy of the GNU General Public License
18 # along with Koha; if not, see <http://www.gnu.org/licenses>.
19
20 use Modern::Perl;
21
22 use CGI qw ( -utf8 );
23 use Encode qw(encode);
24
25 use C4::Auth;
26 use C4::Biblio;
27 use C4::Items;
28 use C4::Output;
29 use C4::Record;
30 use C4::Ris;
31
32 use Koha::CsvProfiles;
33 use Koha::Virtualshelves;
34
35 use utf8;
36 my $query = new CGI;
37
38 my ( $template, $loggedinuser, $cookie ) = get_template_and_user (
39     {
40         template_name   => "virtualshelves/downloadshelf.tt",
41         query           => $query,
42         type            => "intranet",
43         flagsrequired   => { catalogue => 1 },
44     }
45 );
46
47 my $shelfid = $query->param('shelfid');
48 my $format  = $query->param('format');
49 my $dbh     = C4::Context->dbh;
50 my @messages;
51
52 if ($shelfid && $format) {
53
54     my $marcflavour         = C4::Context->preference('marcflavour');
55     my $output='';
56
57     my $shelf = Koha::Virtualshelves->find($shelfid);
58     if ( $shelf ) {
59         if ( $shelf->can_be_viewed( $loggedinuser ) ) {
60
61             my $contents = $shelf->get_contents;
62             # CSV
63             if ($format =~ /^\d+$/) {
64                 my @biblios;
65                 while ( my $content = $contents->next ) {
66                     push @biblios, $content->biblionumber;
67                 }
68                 $output = marc2csv(\@biblios, $format);
69             }
70             else { #Other formats
71                 while ( my $content = $contents->next ) {
72                     my $biblionumber = $content->biblionumber;
73                     my $record = GetMarcBiblio({
74                         biblionumber => $biblionumber,
75                         embed_items  => 1 });
76                     if ($format eq 'iso2709') {
77                         $output .= $record->as_usmarc();
78                     }
79                     elsif ($format eq 'ris') {
80                         $output .= marc2ris($record);
81                     }
82                     elsif ($format eq 'bibtex') {
83                         $output .= marc2bibtex($record, $biblionumber);
84                     }
85                 }
86             }
87
88             # If it was a CSV export we change the format after the export so the file extension is fine
89             $format = "csv" if ($format =~ m/^\d+$/);
90
91             print $query->header(
92             -type => 'application/octet-stream',
93             -'Content-Transfer-Encoding' => 'binary',
94             -attachment=>"shelf.$format");
95             print $output;
96             exit;
97         } else {
98             push @messages, { type => 'error', code => 'unauthorized' };
99         }
100     } else {
101         push @messages, { type => 'error', code => 'does_not_exist' };
102     }
103 }
104 else {
105     $template->param(csv_profiles => [ Koha::CsvProfiles->search({ type => 'marc', used_for => 'export_records' }) ]);
106     $template->param(shelfid => $shelfid); 
107 }
108 $template->param( messages => \@messages );
109 output_html_with_http_headers $query, $cookie, $template->output;