8c7d5851d5c1560f7ed7916c8206fef6d7460572
[Evergreen.git] / Open-ILS / src / perlmods / OpenILS / WWW / Vandelay.pm
1 package OpenILS::WWW::Vandelay;
2 use strict;
3 use warnings;
4 use bytes;
5
6 use Apache2::Log;
7 use Apache2::Const -compile => qw(OK REDIRECT DECLINED NOT_FOUND FORBIDDEN :log);
8 use APR::Const    -compile => qw(:error SUCCESS);
9 use APR::Table;
10
11 use Apache2::RequestRec ();
12 use Apache2::RequestIO ();
13 use Apache2::RequestUtil;
14 use CGI;
15 use Data::Dumper;
16 use Text::CSV;
17
18 use OpenSRF::EX qw(:try);
19 use OpenSRF::Utils::Cache;
20 use OpenSRF::System;
21 use OpenSRF::AppSession;
22 use XML::LibXML;
23
24 use OpenILS::Utils::Fieldmapper;
25 use OpenSRF::Utils::Logger qw/$logger/;
26
27 use MARC::Record;
28 use MARC::File::XML;
29
30 use MIME::Base64;
31 use Digest::MD5 qw/md5_hex/;
32 use OpenSRF::Utils::SettingsClient;
33
34 use UNIVERSAL::require;
35
36 our @formats = qw/USMARC UNIMARC XML BRE/;
37 my $MAX_FILE_SIZE = 10737418240; #10G
38 my $FILE_READ_SIZE = 4096;
39
40 # set the bootstrap config and template include directory when
41 # this module is loaded
42 my $bootstrap;
43
44 sub import {
45         my $self = shift;
46         $bootstrap = shift;
47 }
48
49
50 sub child_init {
51         OpenSRF::System->bootstrap_client( config_file => $bootstrap );
52 }
53
54 sub spool_marc {
55         my $r = shift;
56         my $cgi = new CGI;
57
58         my $auth = $cgi->param('ses') || $cgi->cookie('ses');
59
60         unless(verify_login($auth)) {
61         $logger->error("authentication failed on vandelay record import");
62             return Apache2::Const::FORBIDDEN;
63     }
64
65     my $data_fingerprint = '';
66         my $purpose = $cgi->param('purpose');
67         my $infile = $cgi->param('marc_upload');
68
69         my $conf = OpenSRF::Utils::SettingsClient->new;
70         my $dir = $conf->config_value(
71         apps => 'open-ils.vandelay' => app_settings => databases => 'importer');
72
73     unless(-w $dir) {
74         $logger->error("We need some place to store our MARC files");
75             return Apache2::Const::FORBIDDEN;
76     }
77
78     if($infile and -e $infile) {
79         my ($total_bytes, $buf, $bytes) = (0);
80             $data_fingerprint = md5_hex(time."$$".rand());
81         my $outfile = "$dir/$data_fingerprint.mrc";
82
83         unless(open(OUTFILE, ">$outfile")) {
84             $logger->error("unable to open MARC file for writing: $@");
85                 return Apache2::Const::FORBIDDEN;
86         }
87
88         while($bytes = sysread($infile, $buf, $FILE_READ_SIZE)) {
89             $total_bytes += $bytes;
90             if($total_bytes >= $MAX_FILE_SIZE) {
91                 close(OUTFILE);
92                 unlink $outfile;
93                 $logger->error("import exceeded upload size: $MAX_FILE_SIZE");
94                     return Apache2::Const::FORBIDDEN;
95             }
96             print OUTFILE $buf;
97         }
98
99         close(OUTFILE);
100
101             OpenSRF::Utils::Cache->new->put_cache(
102                     'vandelay_import_spool_' . $data_fingerprint,
103                     { purpose => $purpose, path => $outfile }
104             );
105     }
106
107     $logger->info("uploaded MARC batch with key $data_fingerprint");
108     $r->content_type('text/plain; charset=utf-8');
109         print "$data_fingerprint";
110         return Apache2::Const::OK;
111 }
112
113 sub verify_login {
114         my $auth_token = shift;
115         return undef unless $auth_token;
116
117         my $user = OpenSRF::AppSession
118                 ->create("open-ils.auth")
119                 ->request( "open-ils.auth.session.retrieve", $auth_token )
120                 ->gather(1);
121
122         if (ref($user) eq 'HASH' && $user->{ilsevent} == 1001) {
123                 return undef;
124         }
125
126         return $user if ref($user);
127         return undef;
128 }
129
130 1;