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::Biblio qw( GetMarcBiblio );
  26 use C4::Output qw( output_html_with_http_headers );
  27 use C4::Record;
  28 use C4::Ris qw( marc2ris );
  29
  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 $record = GetMarcBiblio({
  72                         biblionumber => $biblionumber,
  73                         embed_items  => 1 });
  74                     if ($format eq 'iso2709') {
  75                         $output .= $record->as_usmarc();
  76                     }
  77                     elsif ($format eq 'ris') {
  78                         $output .= marc2ris($record);
  79                     }
  80                     elsif ($format eq 'bibtex') {
  81                         $output .= marc2bibtex($record, $biblionumber);
  82                     }
  83                 }
  84             }
  85
  86             # If it was a CSV export we change the format after the export so the file extension is fine
  87             $format = "csv" if ($format =~ m/^\d+$/);
  88
  89             print $query->header(
  90             -type => 'application/octet-stream',
  91             -'Content-Transfer-Encoding' => 'binary',
  92             -attachment=>"shelf.$format");
  93             print $output;
  94             exit;
  95         } else {
  96             push @messages, { type => 'error', code => 'unauthorized' };
  97         }
  98     } else {
  99         push @messages, { type => 'error', code => 'does_not_exist' };
 100     }
 101 }
 102 else {
 103     $template->param(csv_profiles => [ Koha::CsvProfiles->search({ type => 'marc', used_for => 'export_records' })->as_list ]);
 104     $template->param(shelfid => $shelfid);
 105 }
 106 $template->param( messages => \@messages );
 107 output_html_with_http_headers $query, $cookie, $template->output;