1 package OpenILS::Application::Search::Biblio;
2 use base qw/OpenSRF::Application/;
3 use strict; use warnings;
8 use OpenILS::Utils::Fieldmapper;
9 use OpenILS::Utils::ModsParser;
10 use OpenSRF::Utils::SettingsClient;
12 use OpenILS::Application::AppUtils;
16 use Time::HiRes qw(time);
17 use OpenSRF::EX qw(:try);
18 use Digest::MD5 qw(md5_hex);
23 my $apputils = "OpenILS::Application::AppUtils";
25 # Houses biblio search utilites
27 __PACKAGE__->register_method(
28 method => "biblio_search_marc",
29 api_name => "open-ils.search.biblio.marc",
31 note => "Searches biblio information by marc tag",
34 sub biblio_search_marc {
36 my( $self, $client, $search_hash, $string ) = @_;
38 warn "Building biblio marc session\n";
39 my $session = OpenSRF::AppSession->create("open-ils.storage");
42 warn "Sending biblio marc request. String $string\nSearch hash: " . Dumper($search_hash);
43 my $request = $session->request(
44 "open-ils.storage.direct.metabib.full_rec.search_fts.index_vector.atomic",
45 restrict => $search_hash,
47 my $data = $request->gather(1);
52 $session->disconnect();
60 # ---------------------------------------------------------------------------
61 # takes a list of record id's and turns the docs into friendly
62 # mods structures. Creates one MODS structure for each doc id.
63 # ---------------------------------------------------------------------------
64 sub _records_to_mods {
70 my $session = OpenSRF::AppSession->create("open-ils.storage");
71 my $request = $session->request(
72 "open-ils.storage.direct.biblio.record_entry.batch.retrieve", @ids );
74 my $last_content = undef;
76 while( my $response = $request->recv() ) {
79 my $u = OpenILS::Utils::ModsParser->new();
80 $u->start_mods_batch( $last_content->marc );
81 my $mods = $u->finish_mods_batch();
82 $mods->doc_id($last_content->id());
83 warn "Turning doc " . $mods->doc_id() . " into MODS\n";
84 $last_content = undef;
88 next unless $response;
90 if($response->isa("OpenSRF::EX")) {
91 throw $response ($response->stringify);
94 $last_content = $response->content;
99 my $u = OpenILS::Utils::ModsParser->new();
100 $u->start_mods_batch( $last_content->marc );
101 my $mods = $u->finish_mods_batch();
102 $mods->doc_id($last_content->id());
103 push @results, $mods;
108 $session->disconnect();
114 __PACKAGE__->register_method(
115 method => "record_id_to_mods",
116 api_name => "open-ils.search.biblio.record.mods.retrieve",
118 note => "Provide ID, we provide the mods"
121 # converts a record into a mods object with copy counts attached
122 sub record_id_to_mods {
124 my( $self, $client, $org_id, $id ) = @_;
126 my $mods_list = _records_to_mods( $id );
127 my $mods_obj = $mods_list->[0];
128 my $cmethod = $self->method_lookup(
129 "open-ils.search.biblio.record.copy_count");
130 my ($count) = $cmethod->run($org_id, $id);
131 $mods_obj->copy_count($count);
137 __PACKAGE__->register_method(
138 method => "record_id_to_mods_slim",
139 api_name => "open-ils.search.biblio.record.mods_slim.retrieve",
141 note => "Provide ID, we provide the mods"
144 # converts a record into a mods object with NO copy counts attached
145 sub record_id_to_mods_slim {
147 my( $self, $client, $id ) = @_;
148 warn "Retrieving MODS object for record $id\n";
149 return undef unless(defined $id);
151 my $mods_list = _records_to_mods( $id );
152 my $mods_obj = $mods_list->[0];
157 # Returns the number of copies attached to a record based on org location
158 __PACKAGE__->register_method(
159 method => "record_id_to_copy_count",
160 api_name => "open-ils.search.biblio.record.copy_count",
163 __PACKAGE__->register_method(
164 method => "record_id_to_copy_count",
165 api_name => "open-ils.search.biblio.metarecord.copy_count",
168 __PACKAGE__->register_method(
169 method => "record_id_to_copy_count",
170 api_name => "open-ils.search.biblio.metarecord.copy_count.staff",
172 sub record_id_to_copy_count {
173 my( $self, $client, $org_id, $record_id ) = @_;
175 my $method = "open-ils.storage.biblio.record_entry.copy_count.atomic";
177 if($self->api_name =~ /metarecord/) {
178 $method = "open-ils.storage.metabib.metarecord.copy_count.atomic";
182 if($self->api_name =~ /staff/ ) {
183 $method =~ s/atomic/staff\.atomic/og;
184 warn "Doing staff search $method\n";
188 my $session = OpenSRF::AppSession->create("open-ils.storage");
189 warn "copy_count retrieve $record_id\n";
190 return undef unless(defined $record_id);
192 my $request = $session->request(
193 $method, org_unit => $org_id => $key => $record_id );
196 my $count = $request->gather(1);
197 $session->disconnect();
198 return [ sort { $a->{depth} <=> $b->{depth} } @$count ];
203 # used for cat search classes
204 my $cat_search_hash = {
207 { tag => "100", subfield => "a"} ,
208 { tag => "700", subfield => "a"},
212 { tag => "245", subfield => "a"},
213 { tag => "242", subfield => "a"},
214 { tag => "240", subfield => "a"},
215 { tag => "210", subfield => "a"},
219 { tag => "650", subfield => "_" },
223 { tag => "035", subfield => "_" },
227 { tag => "020", subfield => "a" },
233 __PACKAGE__->register_method(
234 method => "biblio_search_tcn",
235 api_name => "open-ils.search.biblio.tcn",
237 note => "Retrieve a record by TCN",
240 sub biblio_search_tcn {
242 my( $self, $client, $tcn ) = @_;
244 $tcn =~ s/.*?(\w+)\s*$/$1/o;
245 warn "Searching TCN $tcn\n";
247 my $session = OpenSRF::AppSession->create( "open-ils.storage" );
248 my $request = $session->request(
249 "open-ils.storage.direct.biblio.record_entry.search.tcn_value", $tcn );
250 my $record_entry = $request->gather(1);
253 for my $record (@$record_entry) {
254 push @ids, $record->id;
257 $session->disconnect();
259 warn "received ID's for tcn search @ids\n";
262 return { count => $size, ids => \@ids };
267 # --------------------------------------------------------------------------------
270 __PACKAGE__->register_method(
271 method => "biblio_search_isbn",
272 api_name => "open-ils.search.biblio.isbn",
275 sub biblio_search_isbn {
276 my( $self, $client, $isbn ) = @_;
277 throw OpenSRF::EX::InvalidArg
279 ("biblio_search_isbn needs an ISBN to search")
280 unless defined $isbn;
282 warn "biblio search for ISBN $isbn\n";
283 my $method = $self->method_lookup("open-ils.search.biblio.marc");
284 my ($records) = $method->run( $cat_search_hash->{isbn}, $isbn );
286 return { count => 0 } unless($records and @$records);
287 my $size = @$records;
288 return { count => $size, ids => $records };
293 # --------------------------------------------------------------------------------
295 __PACKAGE__->register_method(
296 method => "biblio_barcode_to_copy",
297 api_name => "open-ils.search.asset.copy.find_by_barcode",
300 # turns a barcode into a copy object
301 sub biblio_barcode_to_copy {
302 my( $self, $client, $barcode ) = @_;
304 throw OpenSRF::EX::InvalidArg
305 ("search.biblio.barcode needs a barcode to search")
306 unless defined $barcode;
308 warn "copy search for barcode $barcode\n";
309 my $record = OpenILS::Application::AppUtils->simple_scalar_request(
311 "open-ils.storage.direct.asset.copy.search.barcode",
314 return undef unless($record);
319 __PACKAGE__->register_method(
320 method => "biblio_id_to_copy",
321 api_name => "open-ils.search.asset.copy.batch.retrieve",
324 # turns a barcode into a copy object
325 sub biblio_id_to_copy {
326 my( $self, $client, $ids ) = @_;
328 throw OpenSRF::EX::InvalidArg
329 ("search.biblio.batch.retrieve needs a id to search")
332 warn "copy search for ids @$ids\n";
333 my $record = OpenILS::Application::AppUtils->simple_scalar_request(
335 "open-ils.storage.direct.asset.copy.batch.retrieve.atomic",
343 __PACKAGE__->register_method(
344 method => "fleshed_copy_retrieve",
345 api_name => "open-ils.search.asset.copy.fleshed.batch.retrieve",
348 # turns a barcode into a copy object
349 sub fleshed_copy_retrieve {
350 my( $self, $client, $ids ) = @_;
352 throw OpenSRF::EX::InvalidArg
353 ("search.biblio.batch.retrieve needs a id to search")
356 warn "fleshed copy search for id @$ids\n";
357 my $copy = OpenILS::Application::AppUtils->simple_scalar_request(
359 "open-ils.storage.fleshed.asset.copy.batch.retrieve.atomic",
367 __PACKAGE__->register_method(
368 method => "biblio_barcode_to_title",
369 api_name => "open-ils.search.biblio.find_by_barcode",
372 sub biblio_barcode_to_title {
373 my( $self, $client, $barcode ) = @_;
376 throw OpenSRF::EX::ERROR
377 ("Not enough args to find_by_barcode");
380 my $title = $apputils->simple_scalar_request(
382 "open-ils.storage.biblio.record_entry.retrieve_by_barcode",
386 return { ids => [ $title->id ], count => 1 };
388 return { count => 0 };
394 __PACKAGE__->register_method(
395 method => "biblio_copy_to_mods",
396 api_name => "open-ils.search.biblio.copy.mods.retrieve",
399 # takes a copy object and returns it fleshed mods object
400 sub biblio_copy_to_mods {
401 my( $self, $client, $copy ) = @_;
403 throw OpenSRF::EX::InvalidArgs
404 ("copy.mods.retrieve needs a copy") unless( $copy );
406 new Fieldmapper::asset::copy($copy);
408 my $volume = OpenILS::Application::AppUtils->simple_scalar_request(
410 "open-ils.storage.direct.asset.call_number.retrieve",
411 $copy->call_number() );
413 my $mods = _records_to_mods($volume->record());
414 $mods = shift @$mods;
415 $volume->copies([$copy]);
416 push @{$mods->call_numbers()}, $volume;
422 sub barcode_to_mods {
427 # --------------------------------------------------------------------------------
431 __PACKAGE__->register_method(
432 method => "cat_biblio_search_class",
433 api_name => "open-ils.search.cat.biblio.class",
437 sub cat_biblio_search_class {
439 my( $self, $client, $org_id, $class, $sort, $string ) = @_;
441 throw OpenSRF::EX::InvalidArg
442 ("Not enough args to open-ils.search.cat.biblio.class")
443 unless( defined($org_id) and $class and $sort and $string );
448 my $method = $self->method_lookup("open-ils.search.biblio.marc");
450 throw OpenSRF::EX::PANIC
451 ("Can't lookup method 'open-ils.search.biblio.marc'");
454 my ($records) = $method->run( $cat_search_hash->{$class}, $string );
457 for my $i (@$records) { push @ids, $i->[0]; }
459 my $mods_list = _records_to_mods( @ids );
460 return undef unless (ref($mods_list) eq "ARRAY");
462 # ---------------------------------------------------------------
463 # append copy count information to the mods objects
464 my $session = OpenSRF::AppSession->create("open-ils.storage");
466 my $request = $session->request(
467 "open-ils.storage.direct.biblio.record_copy_count.batch", $org_id, @ids );
471 warn "receiving copy counts for doc $id\n";
473 my $response = $request->recv();
474 next unless $response;
476 if( $response and UNIVERSAL::isa($response, "Error")) {
477 throw $response ($response->stringify);
480 my $count = $response->content;
481 my $mods_obj = undef;
482 for my $m (@$mods_list) {
483 $mods_obj = $m if ($m->doc_id() == $id)
486 $mods_obj->copy_count($count);
489 $client->respond( $mods_obj );
495 $session->disconnect();
497 # ---------------------------------------------------------------
505 __PACKAGE__->register_method(
506 method => "cat_biblio_search_class_id",
507 api_name => "open-ils.search.cat.biblio.class.id",
509 note => "Searches biblio information by search class and returns the IDs",
512 sub cat_biblio_search_class_id {
514 my( $self, $client, $org_id, $class, $string, $limit, $offset ) = @_;
521 my $bool = ($class eq "subject" || $class eq "keyword");
522 $string = OpenILS::Application::Search->filter_search($string, $bool);
525 return OpenILS::EX->new("SEARCH_TOO_LARGE")->ex();
528 warn "Searching cat.biblio.class.id string: $string offset: $offset limit: $limit\n";
530 throw OpenSRF::EX::InvalidArg
531 ("Not enough args to open-ils.search.cat.biblio.class")
532 unless( defined($org_id) and $class and $string );
537 my $cache_key = md5_hex( $org_id . $class . $string );
538 my $id_array = OpenILS::Application::SearchCache->get_cache($cache_key);
541 warn "Returning class search from cache\n";
542 my $size = @$id_array;
543 my @ids = @$id_array[ $offset..($offset+$limit) ];
544 return { count => $size, ids => \@ids };
547 my $method = $self->method_lookup("open-ils.search.biblio.marc");
549 throw OpenSRF::EX::PANIC
550 ("Can't lookup method 'open-ils.search.biblio.marc'");
553 my ($records) = $method->run( $cat_search_hash->{$class}, $string );
557 for my $i (@$records) {
558 if(defined($i->[0])) {
559 push @cache_ids, $i->[0];
563 my @ids = @cache_ids[ $offset..($offset+$limit) ];
564 my $size = @$records;
566 OpenILS::Application::SearchCache->put_cache(
567 $cache_key, \@cache_ids, $size );
569 return { count =>$size, ids => \@ids };
574 __PACKAGE__->register_method(
575 method => "biblio_search_class_count",
576 api_name => "open-ils.search.biblio.class.count",
579 __PACKAGE__->register_method(
580 method => "biblio_search_class_count",
581 api_name => "open-ils.search.biblio.class.count.staff",
584 sub biblio_search_class_count {
586 my( $self, $client, $class, $string, $org_id, $org_type ) = @_;
588 warn "org: $org_id : depth: $org_type\n";
590 $org_id = "1" unless defined($org_id); # xxx
591 $org_type = 0 unless defined($org_type);
593 warn "Searching biblio.class.id\n" .
595 "org_id: $org_id\n" .
596 "depth: $org_type\n" ;
598 my $bool = ($class eq "subject" || $class eq "keyword");
599 $string = OpenILS::Application::Search->filter_search($string, $bool);
602 return OpenILS::EX->new("SEARCH_TOO_LARGE")->ex;
606 if( !defined($org_id) or !$class or !$string ) {
607 warn "not enbough args to metarecord search\n";
608 throw OpenSRF::EX::InvalidArg
609 ("Not enough args to open-ils.search.cat.biblio.class")
614 if( ($class ne "title") and ($class ne "author") and
615 ($class ne "subject") and ($class ne "keyword")
616 and ($class ne "series" )) {
617 warn "Invalid search class: $class\n";
618 throw OpenSRF::EX::InvalidArg ("Not a valid search class: $class")
621 # grab the mr id's from storage
623 my $method = "open-ils.storage.cachable.metabib.$class.search_fts.metarecord_count";
624 if($self->api_name =~ /staff/) {
625 $method = "$method.staff";
626 $method =~ s/\.cachable//o;
628 warn "Performing count method $method\n";
629 warn "API name " . $self->api_name() . "\n";
631 my $session = OpenSRF::AppSession->create('open-ils.storage');
633 my $request = $session->request( $method,
636 cache_page_size => 1,
639 my $count = $request->gather(1);
640 warn "Received count $count\n";
646 __PACKAGE__->register_method(
647 method => "biblio_search_class",
648 api_name => "open-ils.search.biblio.class",
651 __PACKAGE__->register_method(
652 method => "biblio_search_class",
653 api_name => "open-ils.search.biblio.class.unordered",
656 __PACKAGE__->register_method(
657 method => "biblio_search_class",
658 api_name => "open-ils.search.biblio.class.staff",
661 __PACKAGE__->register_method(
662 method => "biblio_search_class",
663 api_name => "open-ils.search.biblio.class.unordered.staff",
666 sub biblio_search_class {
668 my( $self, $client, $class, $string,
669 $org_id, $org_type, $limit, $offset ) = @_;
671 warn "org: $org_id : depth: $org_type : limit: $limit : offset: $offset\n";
673 $offset = 0 unless (defined($offset) and $offset > 0);
674 $limit = 100 unless (defined($limit) and $limit > 0);
675 $org_id = "1" unless (defined($org_id)); # xxx
676 $org_type = 0 unless (defined($org_type));
678 warn "Searching biblio.class.id\n" .
680 "\noffset: $offset\n" .
682 "org_id: $org_id\n" .
683 "depth: $org_type\n" ;
685 warn "Search filtering string " . time() . "\n";
686 $string = OpenILS::Application::Search->filter_search($string);
687 if(!$string) { return undef; }
689 if( !defined($org_id) or !$class or !$string ) {
690 warn "not enbough args to metarecord search\n";
691 throw OpenSRF::EX::InvalidArg
692 ("Not enough args to open-ils.search.cat.biblio.class")
697 if( ($class ne "title") and ($class ne "author") and
698 ($class ne "subject") and ($class ne "keyword")
699 and ($class ne "series") ) {
700 warn "Invalid search class: $class\n";
701 throw OpenSRF::EX::InvalidArg ("Not a valid search class: $class")
704 #my $method = "open-ils.storage.metabib.$class.search_fts.metarecord.atomic";
705 my $method = "open-ils.storage.cachable.metabib.$class.search_fts.metarecord.atomic";
707 if($self->api_name =~ /order/) {
708 $method = "open-ils.storage.cachable.metabib.$class.search_fts.metarecord.unordered.atomic",
709 #$method = "open-ils.storage.metabib.$class.search_fts.metarecord.unordered.atomic";
712 if($self->api_name =~ /staff/) {
713 $method =~ s/atomic/staff\.atomic/og;
714 $method =~ s/\.cachable//o;
717 warn "Performing search method $method\n";
718 warn "MR search method is $method\n";
720 my $session = OpenSRF::AppSession->create('open-ils.storage');
722 warn "Search making request " . time() . "\n";
723 my $request = $session->request(
730 cache_page_size => 200,
733 my $records = $request->gather(1);
734 warn "Search request complete " . time() . "\n";
739 warn "Received " . scalar(@$records) . " id's from class search\n";
741 # if we just get one, it won't be wrapped in an array
742 if(!ref($records->[0])) {
743 $records = [$records]; }
744 for my $i (@$records) {
751 @ids = grep { defined($_->[0]) } @ids;
754 $session->disconnect();
756 return { ids => \@ids };
763 __PACKAGE__->register_method(
764 method => "biblio_mrid_to_modsbatch",
765 api_name => "open-ils.search.biblio.metarecord.mods_slim.retrieve",
768 sub biblio_mrid_to_modsbatch {
769 my( $self, $client, $mrid ) = @_;
771 throw OpenSRF::EX::InvalidArg
772 ("search.biblio.metarecord_to_mods requires mr id")
773 unless defined( $mrid );
776 my $metarecord = OpenILS::Application::AppUtils->simple_scalar_request( "open-ils.storage",
777 "open-ils.storage.direct.metabib.metarecord.retrieve", $mrid );
780 throw OpenSRF::EX::ERROR ("No metarecord exists with the given id: $mrid");
783 my $master_id = $metarecord->master_record();
786 # check for existing mods
787 if($metarecord->mods()){
788 warn "We already have mods for " . $metarecord->id . "\n";
789 my $perl = JSON->JSON2perl($metarecord->mods());
790 return Fieldmapper::metabib::virtual_record->new($perl);
795 warn "Creating mods batch for metarecord $mrid\n";
796 my $id_hash = biblio_mrid_to_record_ids( undef, undef, $mrid );
797 my @ids = @{$id_hash->{ids}};
799 if(@ids < 1) { return undef; }
801 warn "Master ID is $master_id\n";
802 # grab the master record to start the mods batch
804 my $record = OpenILS::Application::AppUtils->simple_scalar_request( "open-ils.storage",
805 "open-ils.storage.direct.biblio.record_entry.retrieve", $master_id );
808 throw OpenSRF::EX::ERROR
809 ("No record returned with id $master_id");
812 my $u = OpenILS::Utils::ModsParser->new();
814 $u->start_mods_batch( $record->marc );
815 my $main_doc_id = $record->id();
817 @ids = grep { $_ ne $master_id } @ids;
819 # now we have to collect all of the marc objects and push them into a mods batch
820 my $session = OpenSRF::AppSession->create("open-ils.storage");
821 my $request = $session->request(
822 "open-ils.storage.direct.biblio.record_entry.batch.retrieve", @ids );
824 while( my $response = $request->recv() ) {
826 next unless $response;
827 if(UNIVERSAL::isa( $response,"OpenSRF::EX")) {
828 throw $response ($response->stringify);
831 my $content = $response->content;
834 $u->push_mods_batch( $content->marc );
838 my $mods = $u->finish_mods_batch();
839 $mods->doc_id($mrid);
842 $client->respond_complete($mods);
844 my $mods_string = JSON->perl2JSON($mods->decast);
846 $metarecord->mods($mods_string);
848 my $req = $session->request(
849 "open-ils.storage.direct.metabib.metarecord.update",
856 $session->disconnect();
864 # converts a mr id into a list of record ids
866 __PACKAGE__->register_method(
867 method => "biblio_mrid_to_record_ids",
868 api_name => "open-ils.search.biblio.metarecord_to_records",
871 sub biblio_mrid_to_record_ids {
872 my( $self, $client, $mrid ) = @_;
874 throw OpenSRF::EX::InvalidArg
875 ("search.biblio.metarecord_to_record_ids requires mr id")
876 unless defined( $mrid );
878 warn "Searching for record for MR $mrid\n";
880 my $mrmaps = OpenILS::Application::AppUtils->simple_scalar_request( "open-ils.storage",
881 "open-ils.storage.direct.metabib.metarecord_source_map.search.metarecord", $mrid );
884 for my $map (@$mrmaps) { push @ids, $map->source(); }
886 warn "Recovered id's [@ids] for mr $mrid\n";
890 return { count => $size, ids => \@ids };
896 __PACKAGE__->register_method(
897 method => "biblio_record_to_marc_html",
898 api_name => "open-ils.search.biblio.record.html" );
900 my $parser = XML::LibXML->new();
901 my $xslt = XML::LibXSLT->new();
904 my $settings_client = OpenSRF::Utils::SettingsClient->new();
905 sub biblio_record_to_marc_html {
906 my( $self, $client, $recordid ) = @_;
909 my $dir = $settings_client->config_value( "dirs", "xsl" );
910 my $xsl = $settings_client->config_value(
911 "apps", "open-ils.search", "app_settings", "marc_html_xsl" );
913 $xsl = $parser->parse_file("$dir/$xsl");
914 $marc_sheet = $xslt->parse_stylesheet( $xsl );
918 my $record = $apputils->simple_scalar_request(
920 "open-ils.storage.direct.biblio.record_entry.retrieve",
923 my $xmldoc = $parser->parse_string($record->marc);
924 my $html = $marc_sheet->transform($xmldoc);
925 $html = $html->toString();
932 __PACKAGE__->register_method(
933 method => "retrieve_all_copy_locations",
934 api_name => "open-ils.search.config.copy_location.retrieve.all" );
936 my $shelving_locations;
937 sub retrieve_all_copy_locations {
938 my( $self, $client ) = @_;
939 if(!$shelving_locations) {
940 $shelving_locations = $apputils->simple_scalar_request(
942 "open-ils.storage.direct.asset.copy_location.retrieve.all.atomic");
944 return $shelving_locations;
949 __PACKAGE__->register_method(
950 method => "retrieve_all_copy_statuses",
951 api_name => "open-ils.search.config.copy_status.retrieve.all" );
954 sub retrieve_all_copy_statuses {
955 my( $self, $client ) = @_;
956 if(!$copy_statuses) {
957 $copy_statuses = $apputils->simple_scalar_request(
959 "open-ils.storage.direct.config.copy_status.retrieve.all.atomic" );
961 return $copy_statuses;