]> git.evergreen-ils.org Git - Evergreen.git/blob - Open-ILS/src/support-scripts/generate_circ_notices.pl
specify utf8 output encoding on the email outfile
[Evergreen.git] / Open-ILS / src / support-scripts / generate_circ_notices.pl
1 #!/usr/bin/perl
2 # ---------------------------------------------------------------
3 # Copyright (C) 2008  Georgia Public Library Service
4 # Bill Erickson <erickson@esilibrary.com>
5 #
6 # This program is free software; you can redistribute it and/or
7 # modify it under the terms of the GNU General Public License
8 # as published by the Free Software Foundation; either version 2
9 # of the License, or (at your option) any later version.
10 #
11 # This program is distributed in the hope that it will be useful,
12 # but 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 use strict; use warnings;
17 require 'oils_header.pl';
18 use vars qw/$logger/;
19 use DateTime;
20 use Template;
21 use Data::Dumper;
22 use Email::Send;
23 use Getopt::Long;
24 use Unicode::Normalize;
25 use DateTime::Format::ISO8601;
26 use OpenSRF::Utils qw/:datetime/;
27 use OpenSRF::Utils::JSON;
28 use OpenSRF::Utils::SettingsClient;
29 use OpenSRF::AppSession;
30 use OpenILS::Const qw/:const/;
31 use OpenILS::Application::AppUtils;
32 use OpenILS::Const qw/:const/;
33 my $U = 'OpenILS::Application::AppUtils';
34
35 my $settings = undef;
36 my $e = OpenILS::Utils::CStoreEditor->new;
37
38 my @global_overdue_circs; # all circ collections stored here go into the final global XML file
39
40 # - set up default values
41 my $opt_osrf_config = '/openils/conf/opensrf_core.xml';
42 my $opt_send_email = 0;
43 my $opt_gen_day_intervals = 0;
44 my $opt_days_back = '';
45 my $opt_gen_global_templates = 0;
46 my $opt_show_help = 0;
47 my $opt_append_global_email_fail;
48 my $opt_notice_types = '';
49 my $opt_use_email_outfile;
50 my $opt_use_email_recipient;
51
52 GetOptions(
53     'osrf_config=s' => \$opt_osrf_config,
54     'send-email' => \$opt_send_email,
55     'generate-day-intervals' => \$opt_gen_day_intervals,
56     'generate-global-templates' => \$opt_gen_global_templates,
57     'days-back=s' => \$opt_days_back,
58     'append-global-email-fail' => \$opt_append_global_email_fail,
59     'notice-types=s' => \$opt_notice_types,
60     'use-email-outfile=s' => \$opt_use_email_outfile,
61     'use-email-recipient=s' => \$opt_use_email_recipient,
62     'help' => \$opt_show_help,
63 );
64
65 help() and exit if $opt_show_help;
66
67 sub help {
68     print <<HELP;
69
70 Evergreen Circulation Notice Generator
71
72     ---osrf_config <config_file>
73     
74     --send-email
75         If set, generate email notices
76
77     --use-email-outfile <file>
78         Instead of sending emails, append each email (complete with headers) to this file
79
80     --use-email-recipient <user\@example.com>
81         Send all email notices to this user instead of the patron.  This updates the 
82         email address on the user object attached to the circulations, so no 
83         change to the email templates is required.
84
85     --generate-day-intervals
86         If set, notices which have a notify_interval of >= 1 day will be processed.
87
88     --generate-global-templates
89         Collect all non-emailed notices into a global set and generate templates based on that set.
90
91     --append-global-email-fail
92         If an attempt was made to send an email notice but it failed, the notice is appended
93         to the global notice file set.  This will only have any bearing if --generate-global-templates
94         is enabled.
95
96     --days-back <days_back_comma_separted>  
97         This is used to set the effective run date of the script.
98         This is useful if you don't want to generate notices on certain days.  For example, if you don't 
99         generate notices on the weekend, you would run this script on weekdays and set --days-back to 
100         0,1,2 when it's run on Monday to capture any notices from Saturday and Sunday. 
101
102     --notice-types <overdue,predue,...>
103         Comma-separated list of notice types to generate for this run of the script
104
105     --help 
106         Print this help message
107 HELP
108 }
109
110
111 sub main {
112     osrf_connect($opt_osrf_config);
113     $settings = OpenSRF::Utils::SettingsClient->new;
114
115     die "Please specify at least 1 type of notice to generate with --notice-types\n"
116         unless $opt_notice_types;
117
118     unlink $opt_use_email_outfile if $opt_use_email_outfile;
119
120     my $sender_address = $settings->config_value(notifications => 'sender_address');
121     my $od_sender_addr = $settings->config_value(notifications => overdue => 'sender_address') || $sender_address;
122     my $pd_sender_addr = $settings->config_value(notifications => predue => 'sender_address') || $sender_address;
123     my $overdue_notices = $settings->config_value(notifications => overdue => 'notice');
124     my $predue_notices = $settings->config_value(notifications => predue => 'notice');
125
126     $overdue_notices = [$overdue_notices] unless ref $overdue_notices eq 'ARRAY'; 
127     $predue_notices = [$predue_notices] unless ref $predue_notices eq 'ARRAY'; 
128
129     my @overdues = sort { 
130         OpenSRF::Utils->interval_to_seconds($a->{notify_interval}) <=> 
131         OpenSRF::Utils->interval_to_seconds($b->{notify_interval}) } @$overdue_notices;
132
133     my @predues = sort { 
134         OpenSRF::Utils->interval_to_seconds($a->{notify_interval}) <=> 
135         OpenSRF::Utils->interval_to_seconds($b->{notify_interval}) } @$predue_notices;
136
137     for my $db (($opt_days_back) ? split(',', $opt_days_back) : 0) {
138         if($opt_notice_types =~ /overdue/) {
139             generate_notice_set($_, 'overdue', $db) for @overdues;
140         }
141         if($opt_notice_types =~ /predue/) {
142             generate_notice_set($_, 'predue', $db) for @predues;
143         }
144     }
145
146     generate_global_overdue_file() if $opt_gen_global_templates;
147 }
148
149 sub generate_global_overdue_file {
150     $logger->info("notice: processing ".scalar(@global_overdue_circs)." for global template");
151     return unless @global_overdue_circs;
152
153     my $tt = Template->new({ABSOLUTE => 1});
154
155     $tt->process(
156         $settings->config_value(notifications => overdue => 'combined_template'),
157         {
158             overdues => \@global_overdue_circs,
159             get_bib_attr => \&get_bib_attr,
160             parse_due_date => \&parse_due_date, # let the templates decide date format
161             escape_xml => \&escape_xml,
162         }, 
163         \&global_overdue_output
164     ) or $logger->error('notice: Template error '.$tt->error);
165 }
166
167 sub global_overdue_output {
168     print shift();
169 }
170
171
172 sub generate_notice_set {
173     my($notice, $type, $days_back) = @_;
174
175     my $notify_interval = OpenSRF::Utils->interval_to_seconds($notice->{notify_interval});
176     $notify_interval = -$notify_interval if $type eq 'overdue';
177
178     my ($start_date, $end_date) = make_date_range($notify_interval - $days_back * 86400);
179
180     $logger->info("notice: retrieving circs with due date in range $start_date -> $end_date");
181
182     my $QUERY = {
183         select => {
184             circ => ['id']
185         }, 
186         from => 'circ', 
187         where => {
188             '+circ' => {
189                 checkin_time => undef, 
190                 '-or' => [
191                     {stop_fines => {'not in' => ["LOST","LONGOVERDUE","CLAIMSRETURNED"]}},
192                     {stop_fines => undef}
193                 ],
194                                 due_date => {between => [$start_date, $end_date]},
195             }
196         }
197     };
198
199     # if a circ duration is defined for this type of notice
200     if(my $durs = $notice->{circ_duration_range}) {
201         $QUERY->{where}->{'+circ'}->{duration} = {between => [$durs->{from}, $durs->{to}]};
202     }
203
204     my $circs = $e->json_query($QUERY, {timeout => 18000, substream => 1});
205     process_circs($notice, $type, map {$_->{id}} @$circs);
206 }
207
208
209 sub process_circs {
210     my $notice = shift;
211     my $type = shift;
212     my @circs = @_;
213
214         return unless @circs;
215
216         $logger->info("notice: processing $type notices with notify interval ". 
217         $notice->{notify_interval}."  and ".scalar(@circs)." circs");
218
219         my $org; 
220         my $patron;
221         my @current;
222
223         my $x = 0;
224         for my $circ (@circs) {
225                 $circ = $e->retrieve_action_circulation($circ);
226
227                 if( !defined $org or 
228                                 $circ->circ_lib != $org  or $circ->usr ne $patron ) {
229                         $org = $circ->circ_lib;
230                         $patron = $circ->usr;
231                         generate_notice($notice, $type, @current) if @current;
232                         @current = ();
233                 }
234
235                 push(@current, $circ);
236                 $x++;
237         }
238
239         $logger->info("notice: processed $x circs");
240         generate_notice($notice, $type, @current);
241 }
242
243 my %ORG_CACHE;
244 my %ORG_FROM_CACHE;
245
246 sub get_from_addr {
247     my $type = shift;
248     my $org_id = shift;
249     my $sender;
250
251     if(defined $ORG_FROM_CACHE{$org_id}) {
252         # we have we already loaded the setting for this org unit
253                 $sender = $ORG_FROM_CACHE{$org_id};
254
255     } elsif(my $set = $e->search_actor_org_unit_setting( 
256                         {name => 'org.bounced_emails', org_unit => $org_id} )->[0]) {
257                 my $bemail = OpenSRF::Utils::JSON->JSON2perl($set->value);
258                 $sender = $ORG_FROM_CACHE{$org_id} = $bemail if $bemail;
259         }
260
261     unless($sender) {
262         # there is no setting, use the configured sender
263         $sender = $settings->config_value(
264             notifications => $type => 'sender_address') || 
265             $settings->config_value(notifications => 'sender_address');
266         $ORG_FROM_CACHE{$org_id} = '';
267     }
268
269     return $sender;
270 }
271
272 sub generate_notice {
273     my $notice = shift;
274     my $type = shift;
275     my @circs = @_;
276     return unless @circs;
277     my $circ_list = fetch_circ_data(@circs);
278     my $tt = Template->new({ABSOLUTE => 1});
279
280     # see if there is a configured bounce address for this org unit.
281     # if so, use that as the sender
282     my $org_id = $circ_list->[0]->circ_lib->id;
283     my $sender = get_from_addr($type, $org_id);
284
285     my $context = {   
286         circ_list => $circ_list,
287         get_bib_attr => \&get_bib_attr,
288         parse_due_date => \&parse_due_date, # let the templates decide date format
289         smtp_sender => $sender,
290         notice => $notice,
291     };
292
293     push(@global_overdue_circs, $context) if 
294         $type eq 'overdue' and $notice->{file_append} =~ /always/i;
295
296     if( ($opt_send_email or $opt_use_email_outfile) and 
297             $notice->{email_notify} and 
298             my $email = $circ_list->[0]->usr->email) {
299
300         if(my $tmpl = $notice->{email_template}) {
301             $tt->process($tmpl, $context, 
302                 sub { 
303                     handle_email_template_output($notice, $type, $context, $email, @_); 
304                 }
305             ) or $logger->error('notice: Template error '.$tt->error);
306         } 
307     } else {
308         push(@global_overdue_circs, $context) 
309             if $type eq 'overdue' and $notice->{file_append} =~ /noemail/i;
310     }
311 }
312
313 my $last_mvr;
314 sub get_bib_attr {
315     my $circ = shift;
316     my $attr = shift;
317     my $copy = $circ->target_copy;
318     if($copy->call_number->id == OILS_PRECAT_CALL_NUMBER) {
319         return $copy->dummy_title || '' if $attr eq 'title';
320         return $copy->dummy_author || '' if $attr eq 'author';
321     } else {
322         $last_mvr = $U->record_to_mvr($copy->call_number->record)
323             unless $last_mvr and $last_mvr->doc_id == $copy->call_number->record->id;
324         return $last_mvr->title || '' if $attr eq 'title';
325         return $last_mvr->author || '' if $attr eq 'author';
326     }
327 }
328
329 # provides a date that Template::Plugin::Date can parse
330 sub parse_due_date {
331     my $circ = shift;
332     my $due = DateTime::Format::ISO8601->new->parse_datetime(clense_ISO8601($circ->due_date));
333     return sprintf(
334         "%0.2d:%0.2d:%0.2d %0.2d-%0.2d-%0.4d",
335         $due->hour,
336         $due->minute,
337         $due->second,
338         $due->day,
339         $due->month,
340         $due->year
341     );
342 }
343
344 sub escape_xml {
345     my $str = shift;
346     $str =~ s/&/&amp;/sog;
347     $str =~ s/</&lt;/sog;
348     $str =~ s/>/&gt;/sog;
349     return $str;
350 }
351
352
353 sub handle_email_template_output {
354     my $notice = shift;
355     my $type = shift;
356     my $context = shift;
357     my $email = shift;
358     my $msg = shift;
359
360     if($opt_use_email_outfile) {
361         if(open(F, ">>$opt_use_email_outfile")) {
362             binmode F, ":utf8";
363             $logger->debug("notice: appending emails to outfile $opt_use_email_outfile");
364             print F $msg;
365             close F;
366         } else {
367             $logger->error("notice: unable to open --use-email-outfile $opt_use_email_outfile for writing: $@");
368         }
369     }
370
371
372     if($opt_send_email) {
373             my $sender = Email::Send->new({mailer => 'SMTP'});
374         my $smtp_server = $settings->config_value(notifications => 'smtp_server');
375         $logger->debug("notice: smtp server is $smtp_server");
376             $sender->mailer_args([Host => $smtp_server]);
377             my $stat = $sender->send($msg);
378     
379             if( $stat and $stat->type eq 'success' ) {
380                     $logger->info("notice: successfully sent $type email to $email");
381             } else {
382                     $logger->warn("notice: unable to send $type email to $email: ".Dumper($stat));
383             # if we were unable to send the email, add this notice set to the global notify set
384             push(@global_overdue_circs, $context) 
385                 if $opt_append_global_email_fail and 
386                     $type eq 'overdue' and $notice->{file_append} =~ /noemail/i;
387             }
388     }
389 }
390
391 sub fetch_circ_data {
392     my @circs = @_;
393
394         my $circ_lib_id = $circs[0]->circ_lib;
395         my $usr_id = $circs[0]->usr;
396         $logger->debug("notice: printing user:$usr_id circ_lib:$circ_lib_id");
397
398     my $usr = $e->retrieve_actor_user([
399         $usr_id,
400         {   flesh => 1,
401             flesh_fields => {
402                 au => [qw/card billing_address mailing_address/] 
403             }
404         }
405     ]);
406
407     # if the caller has defined a test email recipient, override 
408     # the user's email if they have one.
409     $usr->email($opt_use_email_recipient) if $usr->email and $opt_use_email_recipient;
410
411     my $circ_lib = $ORG_CACHE{$circ_lib_id} ||
412         $e->retrieve_actor_org_unit([
413             $circ_lib_id,
414             {   flesh => 1,
415                 flesh_fields => {
416                     aou => [qw/billing_address mailing_address/],
417                 }
418             }
419         ]);
420     $ORG_CACHE{$circ_lib_id} = $circ_lib;
421
422     my $circ_objs = $e->search_action_circulation([
423         {id => [map {$_->id} @circs]},
424         {   flesh => 3,
425             flesh_fields => {
426                 circ => [q/target_copy/],
427                 acp => ['call_number'],
428                 acn => ['record'],
429             }
430         }
431     ]);
432
433     $_->circ_lib($circ_lib) for @$circ_objs;
434     $_->usr($usr) for @$circ_objs;
435
436     return $circ_objs
437 }
438
439
440 sub make_date_range {
441         my $offset = shift;
442     #my $is_day_precision = shift; # window?
443
444         my $epoch = CORE::time + $offset;
445         my $date = DateTime->from_epoch(epoch => $epoch, time_zone => 'local');
446
447         $date->set_hour(0);
448         $date->set_minute(0);
449         $date->set_second(0);
450         my $start = "$date";
451         
452         $date->set_hour(23);
453         $date->set_minute(59);
454         $date->set_second(59);
455
456         return ($start, "$date");
457 }
458
459 main();