Bug 32694: (QA follow-up) Fix missing TT filters
[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
24 use C4::Auth qw( get_template_and_user );
25 use C4::Output qw( output_html_with_http_headers );
26 use C4::Record;
27 use C4::Ris qw( marc2ris );
28
29 use Koha::Biblios;
30 use Koha::CsvProfiles;
31 use Koha::Virtualshelves;
32
33 use utf8;
34 my $query = CGI->new;
35
36 my ( $template, $loggedinuser, $cookie ) = get_template_and_user (
37     {
38         template_name   => "virtualshelves/downloadshelf.tt",
39         query           => $query,
40         type            => "intranet",
41         flagsrequired   => { catalogue => 1 },
42     }
43 );
44
45 my $shelfid = $query->param('shelfid');
46 my $format  = $query->param('format');
47 my $dbh     = C4::Context->dbh;
48 my @messages;
49
50 if ($shelfid && $format) {
51
52     my $marcflavour         = C4::Context->preference('marcflavour');
53     my $output='';
54
55     my $shelf = Koha::Virtualshelves->find($shelfid);
56     if ( $shelf ) {
57         if ( $shelf->can_be_viewed( $loggedinuser ) ) {
58
59             my $contents = $shelf->get_contents;
60             # CSV
61             if ($format =~ /^\d+$/) {
62                 my @biblios;
63                 while ( my $content = $contents->next ) {
64                     push @biblios, $content->biblionumber;
65                 }
66                 $output = marc2csv(\@biblios, $format);
67             }
68             else { #Other formats
69                 while ( my $content = $contents->next ) {
70                     my $biblionumber = $content->biblionumber;
71                     my $biblio = Koha::Biblios->find($biblionumber);
72                     my $record = $biblio->metadata->record({ embed_items => 1 });
73                     if ($format eq 'iso2709') {
74                         $output .= $record->as_usmarc();
75                     }
76                     elsif ($format eq 'ris') {
77                         $output .= marc2ris($record);
78                     }
79                     elsif ($format eq 'bibtex') {
80                         $output .= marc2bibtex($record, $biblionumber);
81                     }
82                 }
83             }
84
85             # If it was a CSV export we change the format after the export so the file extension is fine
86             $format = "csv" if ($format =~ m/^\d+$/);
87
88             print $query->header(
89             -type => 'application/octet-stream',
90             -'Content-Transfer-Encoding' => 'binary',
91             -attachment=>"shelf.$format");
92             print $output;
93             exit;
94         } else {
95             push @messages, { type => 'error', code => 'unauthorized' };
96         }
97     } else {
98         push @messages, { type => 'error', code => 'does_not_exist' };
99     }
100 }
101 else {
102     $template->param(csv_profiles => [ Koha::CsvProfiles->search({ type => 'marc', used_for => 'export_records' })->as_list ]);
103     $template->param(shelfid => $shelfid); 
104 }
105 $template->param( messages => \@messages );
106 output_html_with_http_headers $query, $cookie, $template->output;