Open-ILS/src/perlmods/lib/OpenILS/WWW/EGCatLoader/Search.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 OpenSRF::Utils::Logger qw/$logger/;
   5 use OpenILS::Utils::CStoreEditor qw/:funcs/;
   6 use OpenILS::Utils::Fieldmapper;
   7 use OpenILS::Application::AppUtils;
   8 use OpenSRF::Utils::JSON;
   9 use Data::Dumper;
  10 $Data::Dumper::Indent = 0;
  11 my $U = 'OpenILS::Application::AppUtils';
  12
  13 # when fetching "all" search results for staff client
  14 # start/end paging, fetch this many IDs at most
  15 my $all_recs_limit = 10000;
  16
  17
  18 sub _prepare_biblio_search_basics {
  19     my ($cgi) = @_;
  20
  21     return $cgi->param('query') unless $cgi->param('qtype');
  22
  23     my %parts;
  24     my @part_names = qw/qtype contains query bool/;
  25     $parts{$_} = [ $cgi->param($_) ] for (@part_names);
  26
  27     my $full_query = '';
  28     for (my $i = 0; $i < scalar @{$parts{'qtype'}}; $i++) {
  29         my ($qtype, $contains, $query, $bool) = map { $parts{$_}->[$i] } @part_names;
  30
  31         next unless $query =~ /\S/;
  32
  33         # This stuff probably will need refined or rethought to better handle
  34         # the weird things Real Users will surely type in.
  35         $contains = "" unless defined $contains; # silence warning
  36         if ($contains eq 'nocontains') {
  37             $query =~ s/"//g;
  38             $query = ('"' . $query . '"') if index $query, ' ';
  39             $query = '-' . $query;
  40         } elsif ($contains eq 'phrase') {
  41             $query =~ s/"//g;
  42             $query = ('"' . $query . '"') if index $query, ' ';
  43         } elsif ($contains eq 'exact') {
  44             $query =~ s/[\^\$]//g;
  45             $query = '^' . $query . '$';
  46         }
  47         $query = "$qtype:$query" unless $qtype eq 'keyword' and $i == 0;
  48
  49         $bool = ($bool and $bool eq 'or') ? '||' : '&&';
  50         $full_query = $full_query ? "($full_query $bool $query)" : $query;
  51     }
  52
  53     return $full_query;
  54 }
  55
  56 sub _prepare_biblio_search {
  57     my ($cgi, $ctx) = @_;
  58
  59     my $query = _prepare_biblio_search_basics($cgi) || '';
  60
  61     foreach ($cgi->param('modifier')) {
  62         # The unless bit is to avoid stacking modifiers.
  63         $query = ('#' . $_ . ' ' . $query) unless $query =~ qr/\#\Q$_/;
  64     }
  65
  66     # filters
  67     foreach (grep /^fi:/, $cgi->param) {
  68         /:(-?\w+)$/ or next;
  69         my $term = join(",", $cgi->param($_));
  70         $query .= " $1($term)" if length $term;
  71     }
  72
  73     # sort is treated specially, even though it's actually a filter
  74     if ($cgi->param('sort')) {
  75         $query =~ s/sort\([^\)]*\)//g;  # override existing sort(). no stacking.
  76         my ($axis, $desc) = split /\./, $cgi->param('sort');
  77         $query .= " sort($axis)";
  78         if ($desc and not $query =~ /\#descending/) {
  79             $query .= '#descending';
  80         } elsif (not $desc) {
  81             $query =~ s/\#descending//;
  82         }
  83     }
  84
  85     if ($cgi->param('pubdate') && $cgi->param('date1')) {
  86         if ($cgi->param('pubdate') eq 'between') {
  87             $query .= ' between(' . $cgi->param('date1');
  88             $query .= ',' .  $cgi->param('date2') if $cgi->param('date2');
  89             $query .= ')';
  90         } elsif ($cgi->param('pubdate') eq 'is') {
  91             $query .= ' between(' . $cgi->param('date1') .
  92                 ',' .  $cgi->param('date1') . ')';  # sic, date1 twice
  93         } else {
  94             $query .= ' ' . $cgi->param('pubdate') .
  95                 '(' . $cgi->param('date1') . ')';
  96         }
  97     }
  98
  99     my $site;
 100     my $org = $cgi->param('loc');
 101     if (defined($org) and $org ne '' and ($org ne $ctx->{aou_tree}->()->id) and not $query =~ /site\(\S+\)/) {
 102         $site = $ctx->{get_aou}->($org)->shortname;
 103         $query .= " site($site)";
 104     }
 105
 106     if(!$site) {
 107         ($site) = ($query =~ /site\(([^\)]+)\)/);
 108         $site ||= $ctx->{aou_tree}->()->shortname;
 109     }
 110
 111
 112     my $depth;
 113     if (defined($cgi->param('depth')) and not $query =~ /depth\(\d+\)/) {
 114         $depth = defined $cgi->param('depth') ?
 115             $cgi->param('depth') : $ctx->{get_aou}->($site)->ou_type->depth;
 116         $query .= " depth($depth)";
 117     }
 118
 119     return ($query, $site, $depth);
 120 }
 121
 122 sub _get_search_limit {
 123     my $self = shift;
 124
 125     # param takes precedence
 126     my $limit = $self->cgi->param('limit');
 127     return $limit if $limit;
 128
 129     if($self->editor->requestor) {
 130         # See if the user has a hit count preference
 131         my $lset = $self->editor->search_actor_user_setting({
 132             usr => $self->editor->requestor->id,
 133             name => 'opac.hits_per_page'
 134         })->[0];
 135         return OpenSRF::Utils::JSON->JSON2perl($lset->value) if $lset;
 136     }
 137
 138     return 10; # default
 139 }
 140
 141 # context additions:
 142 #   page_size
 143 #   hit_count
 144 #   records : list of bre's and copy-count objects
 145 sub load_rresults {
 146     my $self = shift;
 147     my %args = @_;
 148     my $internal = $args{internal};
 149     my $cgi = $self->cgi;
 150     my $ctx = $self->ctx;
 151     my $e = $self->editor;
 152
 153     $ctx->{page} = 'rresult' unless $internal;
 154     $ctx->{ids} = [];
 155     $ctx->{records} = [];
 156     $ctx->{search_facets} = {};
 157     $ctx->{hit_count} = 0;
 158
 159     # Special alternative searches here.  This could all stand to be cleaner.
 160     if ($cgi->param("_special")) {
 161         return $self->marc_expert_search(%args) if scalar($cgi->param("tag"));
 162         return $self->item_barcode_shortcut if (
 163             $cgi->param("qtype") and ($cgi->param("qtype") eq "item_barcode")
 164         );
 165         return $self->call_number_browse_standalone if (
 166             $cgi->param("qtype") and ($cgi->param("qtype") eq "cnbrowse")
 167         );
 168     }
 169
 170     my $page = $cgi->param('page') || 0;
 171     my $facet = $cgi->param('facet');
 172     my $limit = $self->_get_search_limit;
 173     my $loc = $cgi->param('loc') || $ctx->{aou_tree}->()->id;
 174     my $offset = $page * $limit;
 175     my $metarecord = $cgi->param('metarecord');
 176     my $results;
 177
 178     $ctx->{page_size} = $limit;
 179     $ctx->{search_page} = $page;
 180
 181     # fetch the first hit from the next page
 182     if ($internal) {
 183         $limit = $all_recs_limit;
 184         $offset = 0;
 185     }
 186
 187     my ($query, $site, $depth) = _prepare_biblio_search($cgi, $ctx);
 188
 189     $self->get_staff_search_settings;
 190
 191     if ($ctx->{staff_saved_search_size}) {
 192         my ($key, $list) = $self->staff_save_search($query);
 193         if ($key) {
 194             $self->apache->headers_out->add(
 195                 "Set-Cookie" => $self->cgi->cookie(
 196                     -name => (ref $self)->COOKIE_ANON_CACHE,
 197                     -path => "/",
 198                     -value => ($key || ''),
 199                     -expires => ($key ? undef : "-1h")
 200                 )
 201             );
 202             $ctx->{saved_searches} = $list;
 203         }
 204     }
 205
 206     if ($metarecord and !$internal) {
 207
 208         # TODO: other limits, like SVF/format, etc.
 209         $results = $U->simplereq(
 210             'open-ils.search',
 211             'open-ils.search.biblio.metarecord_to_records',
 212             $metarecord, {org => $loc, depth => $depth}
 213         );
 214
 215         # force the metarecord result blob to match the format of regular search results
 216         $results->{ids} = [map { [$_] } @{$results->{ids}}];
 217
 218     } else {
 219
 220         if (!$query) {
 221             return Apache2::Const::OK if $internal;
 222             return $self->generic_redirect;
 223         }
 224
 225         # Limit and offset will stay here. Everything else should be part of
 226         # the query string, not special args.
 227         my $args = {'limit' => $limit, 'offset' => $offset};
 228
 229         # Stuff these into the TT context so that templates can use them in redrawing forms
 230         $ctx->{processed_search_query} = $query;
 231
 232         $query = "$query $facet" if $facet; # TODO
 233
 234         $logger->activity("EGWeb: [search] $query");
 235
 236         try {
 237
 238             my $method = 'open-ils.search.biblio.multiclass.query';
 239             $method .= '.staff' if $ctx->{is_staff};
 240             $results = $U->simplereq('open-ils.search', $method, $args, $query, 1);
 241
 242         } catch Error with {
 243             my $err = shift;
 244             $logger->error("multiclass search error: $err");
 245             $results = {count => 0, ids => []};
 246         };
 247     }
 248
 249     my $rec_ids = [map { $_->[0] } @{$results->{ids}}];
 250
 251     $ctx->{ids} = $rec_ids;
 252     $ctx->{hit_count} = $results->{count};
 253     $ctx->{parsed_query} = $results->{parsed_query};
 254
 255     return Apache2::Const::OK if @$rec_ids == 0 or $internal;
 256
 257     my ($facets, @data) = $self->get_records_and_facets(
 258         $rec_ids, $results->{facet_key},
 259         {
 260             flesh => '{holdings_xml,mra,acp}',
 261             site => $site,
 262             depth => $depth
 263         }
 264     );
 265
 266     # shove recs into context in search results order
 267     for my $rec_id (@$rec_ids) {
 268         push(
 269             @{$ctx->{records}},
 270             grep { $_->{id} == $rec_id } @data
 271         );
 272     }
 273
 274     $ctx->{search_facets} = $facets;
 275
 276     return Apache2::Const::OK;
 277 }
 278
 279 # Searching by barcode is a special search that does /not/ respect any other
 280 # of the usual search parameters, not even the ones for sorting and paging!
 281 sub item_barcode_shortcut {
 282     my ($self) = @_;
 283
 284     my $method = "open-ils.search.multi_home.bib_ids.by_barcode";
 285     if (my $search = create OpenSRF::AppSession("open-ils.search")) {
 286         my $rec_ids = $search->request(
 287             $method, $self->cgi->param("query")
 288         )->gather(1);
 289         $search->kill_me;
 290
 291         if (ref $rec_ids ne 'ARRAY') {
 292
 293             if($U->event_equals($rec_ids, 'ASSET_COPY_NOT_FOUND')) {
 294                 $rec_ids = [];
 295
 296             } else {
 297                 if (defined $U->event_code($rec_ids)) {
 298                     $self->apache->log->warn(
 299                         "$method returned event: " . $U->event_code($rec_ids)
 300                     );
 301                 } else {
 302                     $self->apache->log->warn(
 303                         "$method returned something unexpected: $rec_ids"
 304                     );
 305                 }
 306                 return Apache2::Const::HTTP_INTERNAL_SERVER_ERROR;
 307             }
 308         }
 309
 310         my ($facets, @data) = $self->get_records_and_facets(
 311             $rec_ids, undef, {flesh => "{holdings_xml,mra}"}
 312         );
 313
 314         $self->ctx->{records} = [@data];
 315         $self->ctx->{search_facets} = {};
 316         $self->ctx->{hit_count} = scalar @data;
 317         $self->ctx->{page_size} = $self->ctx->{hit_count};
 318
 319         return Apache2::Const::OK;
 320     } {
 321         $self->apache->log->warn("couldn't connect to open-ils.search");
 322         return Apache2::Const::HTTP_INTERNAL_SERVER_ERROR;
 323     }
 324 }
 325
 326 # like item_barcode_search, this can't take all the usual search params, but
 327 # this one will at least do site, limit and page
 328 sub marc_expert_search {
 329     my ($self, %args) = @_;
 330
 331     my @tags = $self->cgi->param("tag");
 332     my @subfields = $self->cgi->param("subfield");
 333     my @terms = $self->cgi->param("term");
 334
 335     my $query = [];
 336     for (my $i = 0; $i < scalar @tags; $i++) {
 337         next if ($tags[$i] eq "" || $terms[$i] eq "");
 338         $subfields[$i] = '_' unless $subfields[$i];
 339         push @$query, {
 340             "term" => $terms[$i],
 341             "restrict" => [{"tag" => $tags[$i], "subfield" => $subfields[$i]}]
 342         };
 343     }
 344
 345     $logger->info("query for expert search: " . Dumper($query));
 346
 347     # loc, limit and offset
 348     my $page = $self->cgi->param("page") || 0;
 349     my $limit = $self->_get_search_limit;
 350     my $org_unit = $self->cgi->param("loc") || $self->ctx->{aou_tree}->()->id;
 351     my $offset = $page * $limit;
 352
 353     $self->ctx->{records} = [];
 354     $self->ctx->{search_facets} = {};
 355     $self->ctx->{page_size} = $limit;
 356     $self->ctx->{hit_count} = 0;
 357     $self->ctx->{ids} = [];
 358     $self->ctx->{search_page} = $page;
 359
 360     # nothing to do
 361     return Apache2::Const::OK if @$query == 0;
 362
 363     if ($args{internal}) {
 364         $limit = $all_recs_limit;
 365         $offset = 0;
 366     }
 367
 368     my $timeout = 120;
 369     my $ses = OpenSRF::AppSession->create('open-ils.search');
 370     my $req = $ses->request(
 371         'open-ils.search.biblio.marc',
 372         {searches => $query, org_unit => $org_unit},
 373         $limit, $offset, $timeout);
 374
 375     my $resp = $req->recv($timeout);
 376     my $results = $resp ? $resp->content : undef;
 377     $ses->kill_me;
 378
 379     if (defined $U->event_code($results)) {
 380         $self->apache->log->warn(
 381             "open-ils.search.biblio.marc returned event: " .
 382             $U->event_code($results)
 383         );
 384         return Apache2::Const::HTTP_INTERNAL_SERVER_ERROR;
 385     }
 386
 387     $self->ctx->{ids} = [ grep { $_ } @{$results->{ids}} ];
 388
 389     my ($facets, @data) = $self->get_records_and_facets(
 390         $self->ctx->{ids}, undef, {flesh => "{holdings_xml,mra}"}
 391     );
 392
 393     $self->ctx->{records} = [@data];
 394     $self->ctx->{hit_count} = $results->{count};
 395
 396     return Apache2::Const::OK;
 397 }
 398
 399 sub call_number_browse_standalone {
 400     my ($self) = @_;
 401
 402     if (my $cnfrag = $self->cgi->param("query")) {
 403         my $url = sprintf(
 404             'http%s://%s%s/cnbrowse?cn=%s',
 405             $self->cgi->https ? "s" : "",
 406             $self->apache->hostname,
 407             $self->ctx->{opac_root},
 408             $cnfrag # XXX some kind of escaping needed here?
 409         );
 410         return $self->generic_redirect($url);
 411     } else {
 412         return $self->generic_redirect; # return to search page
 413     }
 414 }
 415
 416 sub load_cnbrowse {
 417     my ($self) = @_;
 418
 419     $self->prepare_browse_call_numbers();
 420
 421     return Apache2::Const::OK;
 422 }
 423
 424 sub get_staff_search_settings {
 425     my ($self) = @_;
 426
 427     unless ($self->ctx->{is_staff}) {
 428         $self->ctx->{staff_saved_search_size} = 0;
 429         return;
 430     }
 431
 432     my $sss_size = $self->ctx->{get_org_setting}->(
 433         $self->ctx->{orig_loc} || $self->ctx->{aou_tree}->()->id,
 434         "opac.staff_saved_search.size",
 435     );
 436
 437     # Sic: 0 is 0 (off), but undefined is 10.
 438     $sss_size = 10 unless defined $sss_size;
 439
 440     $self->ctx->{staff_saved_search_size} = $sss_size;
 441 }
 442
 443 sub staff_load_searches {
 444     my ($self) = @_;
 445
 446     my $cache_key = $self->cgi->cookie((ref $self)->COOKIE_ANON_CACHE);
 447
 448     my $list = [];
 449     if ($cache_key) {
 450         $list = $U->simplereq(
 451             "open-ils.actor",
 452             "open-ils.actor.anon_cache.get_value",
 453             $cache_key, (ref $self)->ANON_CACHE_STAFF_SEARCH
 454         );
 455
 456         unless ($list) {
 457             undef $cache_key;
 458             $list = [];
 459         }
 460     }
 461
 462     return ($cache_key, $list);
 463 }
 464
 465 sub staff_save_search {
 466     my ($self, $query) = @_;
 467
 468     my $sss_size = $self->ctx->{staff_saved_search_size};
 469     return unless $sss_size > 0;
 470
 471     my ($cache_key, $list) = $self->staff_load_searches;
 472     my %already = ( map { $_ => 1 } @$list );
 473
 474     unshift @$list, $query unless $already{$query};
 475
 476     splice @$list, $sss_size;
 477
 478     $cache_key = $U->simplereq(
 479         "open-ils.actor",
 480         "open-ils.actor.anon_cache.set_value",
 481         $cache_key, (ref $self)->ANON_CACHE_STAFF_SEARCH, $list
 482     );
 483
 484     return ($cache_key, $list);
 485 }
 486
 487 1;