Open-ILS/src/perlmods/OpenILS/Application/Search/Z3950.pm

   1 #!/usr/bin/perl
   2 package OpenILS::Application::Search::Z3950;
   3 use strict; use warnings;
   4 use base qw/OpenSRF::Application/;
   5
   6
   7 use Net::Z3950;
   8 use MARC::Record;
   9 use MARC::File::XML;
  10 use OpenSRF::Utils::SettingsClient;
  11
  12 use OpenILS::Utils::FlatXML;
  13 use OpenILS::Application::Cat::Utils;
  14 my $utils = "OpenILS::Application::Cat::Utils";
  15
  16 use OpenILS::Utils::ModsParser;
  17
  18 my $output = "USMARC"; # only support output for now
  19 my $host;
  20 my $port;
  21 my $database;
  22 my $attr;
  23 my $username;
  24 my $password;
  25
  26 my $settings_client;
  27
  28 sub initialize {
  29         $settings_client = OpenSRF::Utils::SettingsClient->new();
  30         $host                   = $settings_client->config_value("z3950", "oclc", "host");
  31         $port                   = $settings_client->config_value("z3950", "oclc", "port");
  32         $database       = $settings_client->config_value("z3950", "oclc", "db");
  33         $attr                   = $settings_client->config_value("z3950", "oclc", "attr");
  34         $username       = $settings_client->config_value("z3950", "oclc", "username");
  35         $password       = $settings_client->config_value("z3950", "oclc", "password");
  36 }
  37
  38
  39 __PACKAGE__->register_method(
  40         method  => "z39_search_by_string",
  41         api_name        => "open-ils.search.z3950.raw_string",
  42 );
  43
  44 sub z39_search_by_string {
  45
  46         my( $self, $client, $server,
  47                         $port, $db, $search, $user, $pw ) = @_;
  48
  49         throw OpenSRF::EX::InvalidArg unless(
  50                         $server and $port and $db and $search);
  51
  52
  53         warn "Z39.50 search for $search\n";
  54
  55         $user ||= "";
  56         $pw     ||= "";
  57
  58         my $conn = new Net::Z3950::Connection(
  59                 $server, $port,
  60                 databaseName                            => $db,
  61                 user                                                    => $user,
  62                 password                                                => $pw,
  63                 preferredRecordSyntax   => $output,
  64         );
  65
  66
  67         my $rs = $conn->search( $search );
  68         if(!$rs) {
  69                 throw OpenSRF::EX::ERROR ("z39 search failed");
  70         }
  71
  72         my $records = [];
  73         my $hash = {};
  74
  75         $hash->{count} =  $rs->size();
  76         warn "Z3950 Search recovered " . $hash->{count} . " records\n";
  77
  78         # until there is a more graceful way to handle this
  79         if($hash->{count} > 20) { return $hash; }
  80
  81
  82         for( my $x = 0; $x != $hash->{count}; $x++ ) {
  83                 my $rec = $rs->record($x+1);
  84                 my $marc = MARC::Record->new_from_usmarc($rec->rawdata());
  85
  86                 my $nodes = OpenILS::Utils::FlatXML->new()->xml_to_nodeset( $marc->as_xml() );
  87                 warn "turning nodeset into tree\n";
  88                 my $tree = $utils->nodeset2tree( $nodes->nodeset );
  89
  90                 push @$records, $tree;
  91         }
  92
  93         $hash->{records} = $records;
  94         return $hash;
  95
  96 }
  97
  98
  99 __PACKAGE__->register_method(
 100         method  => "import_search",
 101         api_name        => "open-ils.search.z3950.import",
 102 );
 103
 104 sub import_search {
 105         my($self, $client, $string) = @_;
 106
 107         return $self->z39_search_by_string(
 108                 $client, $host, $port, $database,
 109                         "\@attr 1=$attr \"$string\"", $username, $password );
 110 }
 111
 112
 113
 114
 115 1;