]> git.evergreen-ils.org Git - working/Evergreen.git/blob - Open-ILS/src/perlmods/lib/OpenILS/WWW/EGCatLoader/Util.pm
c78876103085e13cada96fce3e27f48f507cd349
[working/Evergreen.git] / Open-ILS / src / perlmods / lib / OpenILS / WWW / EGCatLoader / Util.pm
1 package OpenILS::WWW::EGCatLoader;
2 use strict; use warnings;
3 use Apache2::Const -compile => qw(OK DECLINED FORBIDDEN HTTP_INTERNAL_SERVER_ERROR REDIRECT HTTP_BAD_REQUEST);
4 use File::Spec;
5 use Time::HiRes qw/time sleep/;
6 use OpenSRF::Utils::Cache;
7 use OpenSRF::Utils::Logger qw/$logger/;
8 use OpenILS::Utils::CStoreEditor qw/:funcs/;
9 use OpenILS::Utils::Fieldmapper;
10 use OpenILS::Application::AppUtils;
11 use OpenSRF::MultiSession;
12
13 my $U = 'OpenILS::Application::AppUtils';
14
15 my $ro_object_subs; # cached subs
16 our %cache = ( # cached data
17     map => {en_us => {}},
18     list => {en_us => {}},
19     search => {en_us => {}},
20     org_settings => {en_us => {}},
21     search_filter_groups => {en_us => {}},
22     aou_tree => {en_us => undef},
23     aouct_tree => {},
24     eg_cache_hash => undef,
25     authority_fields => {en_us => {}}
26 );
27
28 sub child_init {
29     my $class = shift;
30     my %locales = @_;
31
32     # create a stub object with just enough in place
33     # to call init_ro_object_cache()
34     my $stub = bless({}, ref($class) || $class);
35     my $ctx = {};
36     $stub->ctx($ctx);
37
38     foreach my $locale (sort keys %locales) {
39         OpenSRF::AppSession->default_locale($locales{$locale});
40         $ctx->{locale} = $locale;
41         $stub->init_ro_object_cache();
42
43         # pre-cache various sets of objects
44         # known to be time-consuming to retrieve
45         # the first go around
46         $ro_object_subs->{$locale}->{aou_tree}();
47         $ro_object_subs->{$locale}->{aouct_tree}();
48         $ro_object_subs->{$locale}->{ccvm_list}();
49         $ro_object_subs->{$locale}->{crad_list}();
50         $ro_object_subs->{$locale}->{get_authority_fields}(1);
51     }
52 }
53
54 sub init_ro_object_cache {
55     my $self = shift;
56     my $ctx = $self->ctx;
57     my $memcache ||= OpenSRF::Utils::Cache->new('global');
58
59     # reset org unit setting cache on each page load to avoid the
60     # requirement of reloading apache with each org-setting change
61     $cache{org_settings} = {};
62
63     if($ro_object_subs->{$ctx->{locale}}) {
64         # subs have been built.  insert into the context then move along.
65         $ctx->{$_} = $ro_object_subs->{$ctx->{locale}}->{$_} for keys %{ $ro_object_subs->{$ctx->{locale}} };
66         return;
67     }
68
69     my $locale_subs = {};
70     my $locale = $ctx->{locale};
71
72     # make all "field_safe" classes accesible by default in the template context
73     my @classes = grep {
74         ($Fieldmapper::fieldmap->{$_}->{field_safe} || '') =~ /true/i
75     } keys %{ $Fieldmapper::fieldmap };
76
77     for my $class (@classes) {
78
79         my $hint = $Fieldmapper::fieldmap->{$class}->{hint};
80         next if $hint eq 'aou'; # handled separately
81
82         my $ident_field =  $Fieldmapper::fieldmap->{$class}->{identity};
83         (my $eclass = $class) =~ s/Fieldmapper:://o;
84         $eclass =~ s/::/_/g;
85
86         my $list_key = "${hint}_list";
87         my $get_key = "get_$hint";
88         my $search_key = "search_$hint";
89
90         my $memcache_key = join('.', 'EGWeb',$locale,$hint) . '.';
91
92         # Retrieve the full set of objects with class $hint
93         $locale_subs->{$list_key} = sub {
94             my $from_memcache = 0;
95             my $list = $memcache->get_cache($memcache_key.'list');
96             if ($list) {
97                 $cache{list}{$locale}{$hint} = $list;
98                 $from_memcache = 1;
99             }
100             my $method = "retrieve_all_$eclass";
101             my $e = new_editor();
102             $cache{list}{$locale}{$hint} = $e->$method() unless $cache{list}{$locale}{$hint};
103             undef $e;
104             $memcache->put_cache($memcache_key.'list',$cache{list}{$locale}{$hint}) unless $from_memcache;
105             return $cache{list}{$locale}{$hint};
106         };
107
108         # locate object of class $hint with Ident field $id
109         $cache{map}{$hint} = {};
110         $locale_subs->{$get_key} = sub {
111             my $id = shift;
112             return $cache{map}{$locale}{$hint}{$id} if $cache{map}{$locale}{$hint}{$id};
113             ($cache{map}{$locale}{$hint}{$id}) = grep { $_->$ident_field eq $id } @{$locale_subs->{$list_key}->()};
114             return $cache{map}{$locale}{$hint}{$id};
115         };
116
117         # search for objects of class $hint where field=value
118         $cache{search}{$hint} = {};
119         $locale_subs->{$search_key} = sub {
120             my ($field, $val, $filterfield, $filterval) = @_;
121             my $method = "search_$eclass";
122             my $cacheval = $val;
123             my $scalar_cacheval = 1;
124
125             if (ref $val) {
126                 $scalar_cacheval = 0;
127                 $val = [sort(@$val)] if ref $val eq 'ARRAY';
128                 $cacheval = OpenSRF::Utils::JSON->perl2JSON($val);
129                 #$self->apache->log->info("cacheval : $cacheval");
130             }
131
132             my $search_obj = {$field => $val};
133             if($filterfield) {
134                 $search_obj->{$filterfield} = $filterval;
135                 $cacheval .= ':' . $filterfield . ':' . $filterval;
136             } elsif (
137                 $scalar_cacheval
138                 and $cache{list}{$locale}{$hint}
139                 and !$cache{search}{$locale}{$hint}{$field}{$cacheval}
140             ) {
141                 return $cache{search}{$locale}{$hint}{$field}{$cacheval} =
142                     [ grep { $_->$field() eq $val } @{$cache{list}{$locale}{$hint}} ];
143             }
144
145             my $e = new_editor();
146             $cache{search}{$locale}{$hint}{$field}{$cacheval} = $e->$method($search_obj)
147                 unless $cache{search}{$locale}{$hint}{$field}{$cacheval};
148             undef $e;
149             return $cache{search}{$locale}{$hint}{$field}{$cacheval};
150         };
151     }
152
153     $locale_subs->{aou_tree} = sub {
154
155         # fetch the org unit tree
156         unless($cache{aou_tree}{$locale}) {
157             my $e = new_editor();
158             my $tree = $e->search_actor_org_unit([
159                 {   parent_ou => undef},
160                 {   flesh            => -1,
161                     flesh_fields    => {aou =>  ['children']},
162                     order_by        => {aou => 'name'}
163                 }
164             ])->[0];
165
166             # flesh the org unit type for each org unit
167             # and simultaneously set the id => aou map cache
168             sub flesh_aout {
169                 my $node = shift;
170                 my $locale_subs = shift;
171                 my $locale = shift;
172                 $node->ou_type( $locale_subs->{get_aout}->($node->ou_type) );
173                 $cache{map}{$locale}{aou}{$node->id} = $node;
174                 flesh_aout($_, $locale_subs, $locale) foreach @{$node->children};
175             };
176             flesh_aout($tree, $locale_subs, $locale);
177             undef $e;
178             $cache{aou_tree}{$locale} = $tree;
179         }
180
181         return $cache{aou_tree}{$locale};
182     };
183
184     # Add a special handler for the tree-shaped org unit cache
185     $locale_subs->{get_aou} = sub {
186         my $org_id = shift;
187         return undef unless defined $org_id;
188         $locale_subs->{aou_tree}->(); # force the org tree to load
189         return $cache{map}{$locale}{aou}{$org_id};
190     };
191
192     # Returns a flat list of aou objects.  often easier to manage than a tree.
193     $locale_subs->{aou_list} = sub {
194         $locale_subs->{aou_tree}->(); # force the org tree to load
195         return [ values %{$cache{map}{$locale}{aou}} ];
196     };
197
198     # returns the org unit object by shortname
199     $locale_subs->{get_aou_by_shortname} = sub {
200         my $sn = shift or return undef;
201         my $list = $locale_subs->{aou_list}->();
202         return (grep {$_->shortname eq $sn} @$list)[0];
203     };
204
205     $locale_subs->{aouct_tree} = sub {
206
207         # fetch the org unit tree
208         unless(exists $cache{aouct_tree}{$locale}) {
209             $cache{aouct_tree}{$locale} = undef;
210
211             my $e = new_editor();
212             my $tree_id = $e->search_actor_org_unit_custom_tree(
213                 {purpose => 'opac', active => 't'},
214                 {idlist => 1}
215             )->[0];
216
217             if ($tree_id) {
218                 my $node_tree = $e->search_actor_org_unit_custom_tree_node([
219                 {parent_node => undef, tree => $tree_id},
220                 {   flesh        => -1,
221                     flesh_fields => {aouctn => ['children', 'org_unit']},
222                     order_by     => {aouctn => 'sibling_order'}
223                 }
224                 ])->[0];
225
226                 # tree-ify the org units.  note that since the orgs are fleshed
227                 # upon retrieval, this org tree will not clobber ctx->{aou_tree}.
228                 my @nodes = ($node_tree);
229                 while (my $node = shift(@nodes)) {
230                     my $aou = $node->org_unit;
231                     $aou->children([]);
232                     for my $cnode (@{$node->children}) {
233                         my $child_org = $cnode->org_unit;
234                         $child_org->parent_ou($aou->id);
235                         $child_org->ou_type( $locale_subs->{get_aout}->($child_org->ou_type) );
236                         push(@{$aou->children}, $child_org);
237                         push(@nodes, $cnode);
238                     }
239                 }
240
241                 $cache{aouct_tree}{$locale} = 
242                     $node_tree->org_unit if $node_tree;
243             }
244             undef $e;
245         }
246
247         return $cache{aouct_tree}{$locale};
248     };
249
250     # turns an ISO date into something TT can understand
251     $locale_subs->{parse_datetime} = sub {
252         my $date = shift;
253         my $context_org = shift; # optional, for setting timezone via YAOUS
254
255         # Calling parse_datetime() with empty $date will lead to Internal Server Error
256         return '' if (!defined($date) or $date eq '');
257
258         # Probably an accidental entry like '0212' instead of '2012',
259         # but 1) the leading 0 may get stripped in cstore and
260         # 2) DateTime::Format::ISO8601 returns an error as years
261         # must be 2 or 4 digits
262         if ($date =~ m/^\d{3}-/) {
263             $logger->warn("Invalid date had a 3-digit year: $date");
264             $date = '0' . $date;
265         } elsif ($date =~ m/^\d{1}-/) {
266             $logger->warn("Invalid date had a 1-digit year: $date");
267             $date = '000' . $date;
268         }
269
270         my $cleansed_date = cleanse_ISO8601($date);
271
272         $date = DateTime::Format::ISO8601->new->parse_datetime($cleansed_date);
273         if ($context_org) {
274             $context_org = $context_org->id if ref($context_org);
275             my $tz = $locale_subs->{get_org_setting}->($context_org,'lib.timezone');
276             $date->set_time_zone($tz) if ($tz);
277         }
278         return sprintf(
279             "%0.2d:%0.2d:%0.2d %0.2d-%0.2d-%0.4d",
280             $date->hour,
281             $date->minute,
282             $date->second,
283             $date->day,
284             $date->month,
285             $date->year
286         );
287     };
288
289     # retrieve and cache org unit setting values
290     $locale_subs->{get_org_setting} = sub {
291         my($org_id, $setting) = @_;
292
293         $cache{org_settings}{$locale}{$org_id}{$setting} =
294             $U->ou_ancestor_setting_value($org_id, $setting)
295                 unless exists $cache{org_settings}{$locale}{$org_id}{$setting};
296
297         return $cache{org_settings}{$locale}{$org_id}{$setting};
298     };
299
300     # retrieve and cache acsaf values
301     $locale_subs->{get_authority_fields} = sub {
302         my ($control_set) = @_;
303
304         if (not exists $cache{authority_fields}{$locale}{$control_set}) {
305             my $e = new_editor();
306             if (my $acs = $e->search_authority_control_set_authority_field(
307                                     {control_set => $control_set}
308                                 )
309             ) {
310                 $cache{authority_fields}{$locale}{$control_set} =
311                  +{ map { $_->id => $_ } @$acs };
312                 undef $e;
313             } else {
314                 undef $e;
315                 return;
316             }
317         }
318
319         return $cache{authority_fields}{$locale}{$control_set};
320     };
321
322     $ctx->{$_} = $locale_subs->{$_} for keys %$locale_subs;
323     $ro_object_subs->{$locale} = $locale_subs;
324 }
325
326 sub generic_redirect {
327     my $self = shift;
328     my $url = shift;
329     my $cookie = shift; # can be an array of cgi.cookie's
330
331     $self->apache->print(
332         $self->cgi->redirect(
333             -url => $url || 
334                 $self->cgi->param('redirect_to') || 
335                 $self->ctx->{referer} || 
336                 $self->ctx->{home_page},
337             -cookie => $cookie
338         )
339     );
340
341     return Apache2::Const::REDIRECT;
342 }
343
344 my $unapi_cache;
345 sub get_records_and_facets {
346     my ($self, $rec_ids, $facet_key, $unapi_args) = @_;
347
348     # collect the facet data
349     my $search = OpenSRF::AppSession->create('open-ils.search');
350     my $facet_req;
351     if ($facet_key) {
352         $facet_req = $search->request(
353             'open-ils.search.facet_cache.retrieve', $facet_key
354         );
355     }
356
357     $unapi_args ||= {};
358     $unapi_args->{site} ||= $self->ctx->{aou_tree}->()->shortname;
359     $unapi_args->{depth} ||= $self->ctx->{aou_tree}->()->ou_type->depth;
360     $unapi_args->{flesh_depth} ||= 5;
361
362     my $is_meta = delete $unapi_args->{metarecord};
363     #my $unapi_type = $is_meta ? 'unapi.mmr' : 'unapi.bre';
364     my $unapi_type = $is_meta ? 'unapi.metabib_virtual_record_feed' : 'unapi.biblio_record_entry_feed';
365
366     $unapi_cache ||= OpenSRF::Utils::Cache->new('global');
367     my $unapi_cache_key_suffix = join(
368         '_',
369         $is_meta || 0,
370         $unapi_args->{site},
371         $unapi_args->{depth},
372         $unapi_args->{flesh_depth},
373         ($unapi_args->{pref_lib} || '')
374     );
375
376     my %tmp_data;
377     my $outer_self = $self;
378
379     my $sdepth = $unapi_args->{flesh_depth};
380     my $slimit = "acn=>$sdepth,acp=>$sdepth";
381     $slimit .= ",bre=>$sdepth" if $is_meta;
382     my $flesh = $unapi_args->{flesh} || '';
383
384     # tag the record with the MR id
385     $flesh =~ s/}$/,mmr.unapi}/g if $is_meta;
386
387     my $ses = OpenSRF::AppSession->create('open-ils.cstore');
388
389     my @loop_recs;
390     for my $bid (@$rec_ids) {
391         my $unapi_cache_key = 'TPAC_unapi_cache_'.$bid.'_'.$unapi_cache_key_suffix;
392         my $unapi_data = $unapi_cache->get_cache($unapi_cache_key);
393
394         if (!$unapi_data || $unapi_data->{running}) { #cache entry not done yet, get our own copy
395             push(@loop_recs, $bid);
396         } else {
397             $unapi_data->{marc_xml} = XML::LibXML->new->parse_string($unapi_data->{marc_xml})->documentElement;
398             $tmp_data{$unapi_data->{id}} = $unapi_data;
399         }
400     }
401
402     my $unapi_req = $ses->request(
403         'open-ils.cstore.json_query',
404          {from => [
405             $unapi_type, '{'.join(',',@loop_recs).'}', 'marcxml', $flesh,
406             $unapi_args->{site}, 
407             $unapi_args->{depth}, 
408             $slimit,
409             undef, undef, $unapi_args->{pref_lib}
410         ]}
411     );
412
413     my $facets = {};
414     if ($facet_req) {
415         $self->timelog("get_records_and_facets():almost ready to fetch facets");
416
417         my $tmp_facets = $facet_req->gather(1);
418         $self->timelog("get_records_and_facets(): gathered facet data");
419         for my $cmf_id (keys %$tmp_facets) {
420
421             # sort highest to lowest match count
422             my @entries;
423             my $entries = $tmp_facets->{$cmf_id};
424             for my $ent (keys %$entries) {
425                 push(@entries, {value => $ent, count => $$entries{$ent}});
426             };
427
428             # Sort facet entries by 1) count descending, 2) text ascending
429             @entries = sort {
430                 $b->{count} <=> $a->{count} ||
431                 $a->{value} cmp $b->{value}
432             } @entries;
433
434             $facets->{$cmf_id} = {
435                 cmf => $self->ctx->{get_cmf}->($cmf_id),
436                 data => \@entries
437             }
438         }
439         $self->timelog("get_records_and_facets(): gathered/sorted facet data");
440     } else {
441         $facets = undef;
442     }
443     $search->kill_me;
444
445     my $data = $unapi_req->gather(1);
446
447     $outer_self->timelog("get_records_and_facets(): got response content");
448
449     # Protect against requests for non-existent records
450     return unless $data->{$unapi_type};
451
452     my $doc = XML::LibXML->new->parse_string($data->{$unapi_type})->documentElement;
453
454     $outer_self->timelog("get_records_and_facets(): parsed xml");
455     for my $xml ($doc->getElementsByTagName('record')) {
456         $xml = XML::LibXML->new->parse_string($xml->toString)->documentElement;
457
458         # Protect against legacy invalid MARCXML that might not have a 901c
459         my $bre_id;
460         my $mmr_id;
461         my $bre_id_nodes =  $xml->find('*[@tag="901"]/*[@code="c"]');
462         if ($bre_id_nodes) {
463             $bre_id =  $bre_id_nodes->[0]->textContent;
464         } else {
465             $logger->warn("Missing 901 subfield 'c' in " . $xml->toString());
466         }
467     
468         if ($is_meta) {
469             # extract metarecord ID from mmr.unapi tag
470             for my $node ($xml->getElementsByTagName('abbr')) {
471                 my $title = $node->getAttribute('title');
472                 ($mmr_id = $title) =~ 
473                     s/tag:open-ils.org:U2\@mmr\/(\d+)\/.*/$1/g;
474                 last if $mmr_id;
475             }
476         }
477     
478         my $rec_id = $mmr_id ? $mmr_id : $bre_id;
479         $tmp_data{$rec_id} = {
480             id => $rec_id, 
481             bre_id => $bre_id, 
482             mmr_id => $mmr_id,
483             marc_xml => $xml
484         };
485     
486         if ($rec_id) {
487             # Let other backends grab our data now that we're done.
488             my $key = 'TPAC_unapi_cache_'.$rec_id.'_'.$unapi_cache_key_suffix;
489             my $cache_data = $unapi_cache->get_cache($key);
490             if ($$cache_data{running}) {
491                 $unapi_cache->put_cache($key, {
492                     bre_id => $bre_id,
493                     mmr_id => $mmr_id,
494                     id => $rec_id, 
495                     marc_xml => $xml->toString
496                 }, 10);
497             }
498         }
499     }
500
501     return ($facets, map { $tmp_data{$_} } @$rec_ids);
502 }
503
504 sub _resolve_org_id_or_shortname {
505     my ($self, $str) = @_;
506
507     if (length $str) {
508         # Match on shortname case insensitively, but only if there's exactly
509         # one match.  We wouldn't want the system to arbitrarily interpret
510         # 'foo' as either the org unit with shortname 'FOO' or 'Foo' and fail
511         # to make it clear to the user which one was chosen and why.
512         my $res = $self->editor->search_actor_org_unit({
513             shortname => {
514                 '=' => {
515                     transform => 'evergreen.lowercase',
516                     value => lc($str)
517                 }
518             }
519         });
520         return $res->[0]->id if $res and @$res == 1;
521     }
522
523     # Note that we don't validate IDs; we only try a shortname lookup and then
524     # assume anything else must be an ID.
525     return int($str); # Wrapping in int() prevents 500 on unmatched string.
526 }
527
528 sub _get_search_lib {
529     my $self = shift;
530     my $ctx = $self->ctx;
531
532     # avoid duplicate lookups
533     return $ctx->{search_ou} if $ctx->{search_ou};
534
535     my $loc = $ctx->{copy_location_group_org};
536     return $loc if $loc;
537
538     # loc param takes precedence
539     # XXX ^-- over what exactly? We could use clarification here. To me it looks
540     # like locg takes precedence over loc which in turn takes precedence over
541     # request headers which take precedence over pref_lib (which can be
542     # specified a lot of different ways and eventually falls back to
543     # physical_loc) and it all finally defaults to top of the org tree.
544     # To say nothing of all the code that doesn't look to this function at all
545     # but rather accesses some subset of these inputs directly.
546
547     $loc = $self->cgi->param('loc');
548     return $loc if $loc;
549
550     if ($self->apache->headers_in->get('OILS-Search-Lib')) {
551         return $self->apache->headers_in->get('OILS-Search-Lib');
552     }
553     if ($self->cgi->cookie('eg_search_lib')) {
554         return $self->cgi->cookie('eg_search_lib');
555     }
556
557     my $pref_lib = $self->_get_pref_lib();
558     return $pref_lib if $pref_lib;
559
560     return $ctx->{aou_tree}->()->id;
561 }
562
563 sub _get_pref_lib {
564     my $self = shift;
565     my $ctx = $self->ctx;
566
567     # plib param takes precedence
568     my $plib = $self->cgi->param('plib');
569     return $plib if $plib;
570
571     if ($self->apache->headers_in->get('OILS-Pref-Lib')) {
572         return $self->apache->headers_in->get('OILS-Pref-Lib');
573     }
574     if ($self->cgi->cookie('eg_pref_lib')) {
575         return $self->cgi->cookie('eg_pref_lib');
576     }
577
578     if ($ctx->{user}) {
579         # See if the user has a search library preference
580         my $lset = $self->editor->search_actor_user_setting({
581             usr => $ctx->{user}->id, 
582             name => 'opac.default_search_location'
583         })->[0];
584         return OpenSRF::Utils::JSON->JSON2perl($lset->value) if $lset;
585
586         # Otherwise return the user's home library
587         my $ou = $ctx->{user}->home_ou;
588         return ref($ou) ? $ou->id : $ou;
589     }
590
591     if ($ctx->{physical_loc}) {
592         return $ctx->{physical_loc};
593     }
594
595 }
596
597 # This is defensively coded since we don't do much manual reading from the
598 # file system in this module.
599 sub load_eg_cache_hash {
600     my ($self) = @_;
601
602     # just a context helper
603     $self->ctx->{eg_cache_hash} = sub { return $cache{eg_cache_hash}; };
604
605     # Need to actually load the value? If already done, move on.
606     return if defined $cache{eg_cache_hash};
607
608     # In this way even if we fail, we won't slow things down by ever trying
609     # again within this Apache process' lifetime.
610     $cache{eg_cache_hash} = 0;
611
612     my $path = File::Spec->catfile(
613         $self->apache->document_root, "eg_cache_hash"
614     );
615
616     if (not open FH, "<$path") {
617         $self->apache->log->warn("error opening $path : $!");
618         return;
619     } else {
620         my $buf;
621         my $rv = read FH, $buf, 64;  # defensive
622         close FH;
623
624         if (not defined $rv) {  # error
625             $self->apache->log->warn("error reading $path : $!");
626         } elsif ($rv > 0) {     # no error, something read
627             chomp $buf;
628             $cache{eg_cache_hash} = $buf;
629         }
630     }
631 }
632
633 # Extracts the copy location org unit and group from the 
634 # "logc" param, which takes the form org_id:grp_id.
635 sub extract_copy_location_group_info {
636     my $self = shift;
637     my $ctx = $self->ctx;
638     if (my $clump = $self->cgi->param('locg')) {
639         my ($org, $grp) = split(/:/, $clump);
640         $ctx->{copy_location_group_org} =
641             $self->_resolve_org_id_or_shortname($org);
642         $ctx->{copy_location_group} = $grp if $grp;
643     }
644 }
645
646 sub load_copy_location_groups {
647     my $self = shift;
648     my $ctx = $self->ctx;
649
650     # User can access to the search location groups at the current 
651     # search lib, the physical location lib, and the patron's home ou.
652     my @ctx_orgs = $ctx->{search_ou};
653     push(@ctx_orgs, $ctx->{physical_loc}) if $ctx->{physical_loc};
654     push(@ctx_orgs, $ctx->{user}->home_ou) if $ctx->{user};
655
656     my $grps = $self->editor->search_asset_copy_location_group([
657         {
658             opac_visible => 't',
659             owner => {
660                 in => {
661                     select => {aou => [{
662                         column => 'id', 
663                         transform => 'actor.org_unit_full_path',
664                         result_field => 'id',
665                     }]},
666                     from => 'aou',
667                     where => {id => \@ctx_orgs}
668                 }
669             }
670         },
671         {order_by => {acplg => 'pos'}}
672     ]);
673
674     my %buckets;
675     push(@{$buckets{$_->owner}}, $_) for @$grps;
676     $ctx->{copy_location_groups} = \%buckets;
677 }
678
679 sub set_file_download_headers {
680     my $self = shift;
681     my $filename = shift;
682     my $ctype = shift || "text/plain; encoding=utf8";
683
684     $self->apache->content_type($ctype);
685
686     $self->apache->headers_out->add(
687         "Content-Disposition",
688         "attachment;filename=$filename"
689     );
690
691     return Apache2::Const::OK;
692 }
693
694 sub apache_log_if_event {
695     my ($self, $event, $prefix_text, $success_ok, $level) = @_;
696
697     $prefix_text ||= "Evergreen returned event";
698     $success_ok ||= 0;
699     $level ||= "warn";
700
701     chomp $prefix_text;
702     $prefix_text .= ": ";
703
704     my $code = $U->event_code($event);
705     if (defined $code and ($code or not $success_ok)) {
706         $self->apache->log->$level(
707             $prefix_text .
708             ($event->{textcode} || "") . " ($code)" .
709             ($event->{note} ? (": " . $event->{note}) : "")
710         );
711         return 1;
712     }
713
714     return;
715 }
716
717 sub load_search_filter_groups {
718     my $self = shift;
719     my $ctx_org = shift;
720     my $org_list = $U->get_org_ancestors($ctx_org, 1);
721
722     my %seen;
723     for my $org_id (@$org_list) {
724
725         my $grps;
726         if (! ($grps = $cache{search_filter_groups}{$org_id}) ) {
727             $grps = $self->editor->search_actor_search_filter_group([
728                 {owner => $org_id},
729                 {   flesh => 2, 
730                     flesh_fields => {
731                         asfg => ['entries'],
732                         asfge => ['query']
733                     },
734                     order_by => {asfge => 'pos'}
735                 }
736             ]);
737             $cache{search_filter_groups}{$org_id} = $grps;
738         }
739
740         # for the current context, if a descendant org has a group 
741         # with a matching code replace the group from the parent.
742         $seen{$_->code} = $_ for @$grps;
743     }
744
745     return $self->ctx->{search_filter_groups} = \%seen;
746 }
747
748
749 sub check_for_temp_list_warning {
750     my $self = shift;
751     my $ctx = $self->ctx;
752     my $cgi = $self->cgi;
753
754     my $lib = $self->_get_search_lib;
755     my $warn = ($ctx->{get_org_setting}->($lib || 1, 'opac.patron.temporary_list_warn')) ? 1 : 0;
756
757     if ($warn && $ctx->{user}) {
758         $self->_load_user_with_prefs;
759         my $map = $ctx->{user_setting_map};
760         $warn = 0 if ($$map{'opac.temporary_list_no_warn'});
761     }
762
763     # Check for a cookie disabling the warning.
764     $warn = 0 if ($warn && $cgi->cookie('no_temp_list_warn'));
765
766     return $warn;
767 }
768
769 sub load_org_util_funcs {
770     my $self = shift;
771     my $ctx = $self->ctx;
772
773     # evaluates to true if test_ou is within the same depth-
774     # scoped tree as ctx_ou. both ou's are org unit objects.
775     $ctx->{org_within_scope} = sub {
776         my ($ctx_ou, $test_ou, $depth) = @_;
777
778         return 1 if $ctx_ou->id == $test_ou->id;
779
780         if ($depth) {
781
782             # find the top-most ctx-org ancestor at the provided depth
783             while ($depth < $ctx_ou->ou_type->depth 
784                     and $ctx_ou->id != $test_ou->id) {
785                 $ctx_ou = $ctx->{get_aou}->($ctx_ou->parent_ou);
786             }
787
788             # the preceeding loop may have landed on our org
789             return 1 if $ctx_ou->id == $test_ou->id;
790
791         } else {
792
793             return 1 if defined $depth; # $depth == 0;
794         }
795
796         for my $child (@{$ctx_ou->children}) {
797             return 1 if $ctx->{org_within_scope}->($child, $test_ou);
798         }
799
800         return 0;
801     };
802
803     # Returns true if the provided org unit is within the same 
804     # org unit hiding depth-scoped tree as the physical location.
805     # Org unit hiding is based on the immutable physical_loc
806     # and is not meant to change as search/pref/etc libs change
807     $ctx->{org_within_hiding_scope} = sub {
808         my $org_id = shift;
809         my $ploc = $ctx->{physical_loc} or return 1;
810
811         my $depth = $ctx->{get_org_setting}->(
812             $ploc, 'opac.org_unit_hiding.depth');
813
814         return 1 unless $depth; # 0 or undef
815
816         return $ctx->{org_within_scope}->( 
817             $ctx->{get_aou}->($ploc), 
818             $ctx->{get_aou}->($org_id), $depth);
819  
820     };
821
822     # Evaluates to true if the context org (defaults to get_library) 
823     # is not within the hiding scope.  Also evaluates to true if the 
824     # user's pref_ou is set and it's out of hiding scope.
825     # Always evaluates to true when ctx.is_staff
826     $ctx->{org_hiding_disabled} = sub {
827         my $ctx_org = shift || $ctx->{search_ou};
828
829         return 1 if $ctx->{is_staff};
830
831         # beware locg values formatted as org:loc
832         $ctx_org =~ s/:.*//g;
833
834         return 1 if !$ctx->{org_within_hiding_scope}->($ctx_org);
835
836         return 1 if $ctx->{pref_ou} and $ctx->{pref_ou} != $ctx_org 
837             and !$ctx->{org_within_hiding_scope}->($ctx->{pref_ou});
838
839         return 0;
840     };
841
842 }
843
844 # returns the list of org unit IDs for which the 
845 # selected org unit setting returned a true value
846 sub setting_is_true_for_orgs {
847     my ($self, $setting) = @_;
848     my $ctx = $self->ctx;
849     my @valid_orgs;
850
851     my $test_org;
852     $test_org = sub {
853         my $org = shift;
854         push (@valid_orgs, $org->id) if
855             $ctx->{get_org_setting}->($org->id, $setting);
856         $test_org->($_) for @{$org->children};
857     };
858
859     $test_org->($ctx->{aou_tree}->());
860     return \@valid_orgs;
861 }
862
863 # Builds and links a perm checking function, testing permissions against
864 # the currently logged in user.  
865 # ctx->{has_perm}->(perm_code, org_id) => 1/undef
866 # For security, perm checks are cached per page, not per process.
867 sub load_perm_funcs {
868     my $self = shift;
869     my %perm_cache;
870     $self->ctx->{has_perm} = sub {
871         my ($perm_code, $org_id) = @_;
872         return 0 unless $self->editor->requestor;
873
874         if ($perm_cache{$org_id}) {
875             return $perm_cache{$org_id}{$perm_code} 
876                 if exists $perm_cache{$org_id}{$perm_code};
877         } else {
878             $perm_cache{$org_id} = {};
879         }
880         return $perm_cache{$org_id}{$perm_code} =
881             $self->editor->allowed($perm_code, $org_id);
882     }
883 }
884     
885
886
887 1;