Open-ILS/src/perlmods/lib/OpenILS/Application/Vandelay.pm

   1 package OpenILS::Application::Vandelay;
   2 use strict; use warnings;
   3 use OpenILS::Application;
   4 use base qw/OpenILS::Application/;
   5 use Unicode::Normalize;
   6 use OpenSRF::EX qw/:try/;
   7 use OpenSRF::AppSession;
   8 use OpenSRF::Utils::SettingsClient;
   9 use OpenSRF::Utils::Cache;
  10 use OpenILS::Utils::Fieldmapper;
  11 use OpenILS::Utils::CStoreEditor qw/:funcs/;
  12 use MARC::Batch;
  13 use MARC::Record;
  14 use MARC::File::XML ( BinaryEncoding => 'UTF-8' );
  15 use OpenILS::Utils::Fieldmapper;
  16 use Time::HiRes qw(time);
  17 use OpenSRF::Utils::Logger qw/$logger/;
  18 use MIME::Base64;
  19 use OpenILS::Const qw/:const/;
  20 use OpenILS::Application::AppUtils;
  21 use OpenILS::Application::Cat::BibCommon;
  22 use OpenILS::Application::Cat::AuthCommon;
  23 use OpenILS::Application::Cat::AssetCommon;
  24 my $U = 'OpenILS::Application::AppUtils';
  25
  26 # A list of LDR/06 values from http://loc.gov/marc
  27 my %record_types = (
  28         a => 'bib',
  29         c => 'bib',
  30         d => 'bib',
  31         e => 'bib',
  32         f => 'bib',
  33         g => 'bib',
  34         i => 'bib',
  35         j => 'bib',
  36         k => 'bib',
  37         m => 'bib',
  38         o => 'bib',
  39         p => 'bib',
  40         r => 'bib',
  41         t => 'bib',
  42         u => 'holdings',
  43         v => 'holdings',
  44         x => 'holdings',
  45         y => 'holdings',
  46         z => 'auth',
  47       ' ' => 'bib',
  48 );
  49
  50 sub initialize {}
  51 sub child_init {}
  52
  53 # --------------------------------------------------------------------------------
  54 # Biblio ingest
  55
  56 sub create_bib_queue {
  57     my $self = shift;
  58     my $client = shift;
  59     my $auth = shift;
  60     my $name = shift;
  61     my $owner = shift;
  62     my $type = shift;
  63     my $match_set = shift;
  64     my $import_def = shift;
  65
  66     my $e = new_editor(authtoken => $auth, xact => 1);
  67
  68     return $e->die_event unless $e->checkauth;
  69     return $e->die_event unless $e->allowed('CREATE_BIB_IMPORT_QUEUE');
  70     $owner ||= $e->requestor->id;
  71
  72     if ($e->search_vandelay_bib_queue( {name => $name, owner => $owner, queue_type => $type})->[0]) {
  73         $e->rollback;
  74         return OpenILS::Event->new('BIB_QUEUE_EXISTS')
  75     }
  76
  77     my $queue = new Fieldmapper::vandelay::bib_queue();
  78     $queue->name( $name );
  79     $queue->owner( $owner );
  80     $queue->queue_type( $type ) if ($type);
  81     $queue->item_attr_def( $import_def ) if ($import_def);
  82     $queue->match_set($match_set) if $match_set;
  83
  84     my $new_q = $e->create_vandelay_bib_queue( $queue );
  85     return $e->die_event unless ($new_q);
  86     $e->commit;
  87
  88     return $new_q;
  89 }
  90 __PACKAGE__->register_method(
  91     api_name   => "open-ils.vandelay.bib_queue.create",
  92     method     => "create_bib_queue",
  93     api_level  => 1,
  94     argc       => 4,
  95 );
  96
  97
  98 sub create_auth_queue {
  99     my $self = shift;
 100     my $client = shift;
 101     my $auth = shift;
 102     my $name = shift;
 103     my $owner = shift;
 104     my $type = shift;
 105     my $match_set = shift;
 106
 107     my $e = new_editor(authtoken => $auth, xact => 1);
 108
 109     return $e->die_event unless $e->checkauth;
 110     return $e->die_event unless $e->allowed('CREATE_AUTHORITY_IMPORT_QUEUE');
 111     $owner ||= $e->requestor->id;
 112
 113     if ($e->search_vandelay_bib_queue({name => $name, owner => $owner, queue_type => $type})->[0]) {
 114         $e->rollback;
 115         return OpenILS::Event->new('AUTH_QUEUE_EXISTS')
 116     }
 117
 118     my $queue = new Fieldmapper::vandelay::authority_queue();
 119     $queue->name( $name );
 120     $queue->owner( $owner );
 121     $queue->queue_type( $type ) if ($type);
 122
 123     my $new_q = $e->create_vandelay_authority_queue( $queue );
 124     $e->die_event unless ($new_q);
 125     $e->commit;
 126
 127     return $new_q;
 128 }
 129 __PACKAGE__->register_method(
 130     api_name   => "open-ils.vandelay.authority_queue.create",
 131     method     => "create_auth_queue",
 132     api_level  => 1,
 133     argc       => 3,
 134 );
 135
 136 sub add_record_to_bib_queue {
 137     my $self = shift;
 138     my $client = shift;
 139     my $auth = shift;
 140     my $queue = shift;
 141     my $marc = shift;
 142     my $purpose = shift;
 143     my $bib_source = shift;
 144
 145     my $e = new_editor(authtoken => $auth, xact => 1);
 146
 147     $queue = $e->retrieve_vandelay_bib_queue($queue);
 148
 149     return $e->die_event unless $e->checkauth;
 150     return $e->die_event unless
 151         ($e->allowed('CREATE_BIB_IMPORT_QUEUE', undef, $queue) ||
 152          $e->allowed('CREATE_BIB_IMPORT_QUEUE'));
 153
 154     my $new_rec = _add_bib_rec($e, $marc, $queue->id, $purpose, $bib_source);
 155
 156     return $e->die_event unless ($new_rec);
 157     $e->commit;
 158     return $new_rec;
 159 }
 160 __PACKAGE__->register_method(
 161     api_name   => "open-ils.vandelay.queued_bib_record.create",
 162     method     => "add_record_to_bib_queue",
 163     api_level  => 1,
 164     argc       => 3,
 165 );
 166
 167 sub _add_bib_rec {
 168     my $e = shift;
 169     my $marc = shift;
 170     my $queue = shift;
 171     my $purpose = shift;
 172     my $bib_source = shift;
 173
 174     my $rec = new Fieldmapper::vandelay::queued_bib_record();
 175     $rec->marc( $marc );
 176     $rec->queue( $queue );
 177     $rec->purpose( $purpose ) if ($purpose);
 178     $rec->bib_source($bib_source);
 179
 180     return $e->create_vandelay_queued_bib_record( $rec );
 181 }
 182
 183 sub add_record_to_authority_queue {
 184     my $self = shift;
 185     my $client = shift;
 186     my $auth = shift;
 187     my $queue = shift;
 188     my $marc = shift;
 189     my $purpose = shift;
 190
 191     my $e = new_editor(authtoken => $auth, xact => 1);
 192
 193     $queue = $e->retrieve_vandelay_authority_queue($queue);
 194
 195     return $e->die_event unless $e->checkauth;
 196     return $e->die_event unless
 197         ($e->allowed('CREATE_AUTHORITY_IMPORT_QUEUE', undef, $queue) ||
 198          $e->allowed('CREATE_AUTHORITY_IMPORT_QUEUE'));
 199
 200     my $new_rec = _add_auth_rec($e, $marc, $queue->id, $purpose);
 201
 202     return $e->die_event unless ($new_rec);
 203     $e->commit;
 204     return $new_rec;
 205 }
 206 __PACKAGE__->register_method(
 207     api_name   => "open-ils.vandelay.queued_authority_record.create",
 208     method     => "add_record_to_authority_queue",
 209     api_level  => 1,
 210     argc       => 3,
 211 );
 212
 213 sub _add_auth_rec {
 214     my $e = shift;
 215     my $marc = shift;
 216     my $queue = shift;
 217     my $purpose = shift;
 218
 219     my $rec = new Fieldmapper::vandelay::queued_authority_record();
 220     $rec->marc( $marc );
 221     $rec->queue( $queue );
 222     $rec->purpose( $purpose ) if ($purpose);
 223
 224     return $e->create_vandelay_queued_authority_record( $rec );
 225 }
 226
 227 sub process_spool {
 228     my $self = shift;
 229     my $client = shift;
 230     my $auth = shift;
 231     my $fingerprint = shift || '';
 232     my $queue_id = shift;
 233     my $purpose = shift;
 234     my $filename = shift;
 235     my $bib_source = shift;
 236
 237     my $e = new_editor(authtoken => $auth, xact => 1);
 238     return $e->die_event unless $e->checkauth;
 239
 240     my $queue;
 241     my $type = $self->{record_type};
 242
 243     if($type eq 'bib') {
 244         $queue = $e->retrieve_vandelay_bib_queue($queue_id) or return $e->die_event;
 245     } else {
 246         $queue = $e->retrieve_vandelay_authority_queue($queue_id) or return $e->die_event;
 247     }
 248
 249     my $evt = check_queue_perms($e, $type, $queue);
 250     return $evt if ($evt);
 251
 252     my $cache = new OpenSRF::Utils::Cache();
 253
 254     if($fingerprint) {
 255         my $data = $cache->get_cache('vandelay_import_spool_' . $fingerprint);
 256         $purpose = $data->{purpose};
 257         $filename = $data->{path};
 258         $bib_source = $data->{bib_source};
 259     }
 260
 261     unless(-r $filename) {
 262         $logger->error("unable to read MARC file $filename");
 263         return -1; # make this an event XXX
 264     }
 265
 266     $logger->info("vandelay spooling $fingerprint purpose=$purpose file=$filename");
 267
 268     my $marctype = 'USMARC';
 269
 270     open F, $filename;
 271     $marctype = 'XML' if (getc(F) =~ /^\D/o);
 272     close F;
 273
 274     my $batch = new MARC::Batch ($marctype, $filename);
 275     $batch->strict_off;
 276
 277     my $response_scale = 10;
 278     my $count = 0;
 279     my $r = -1;
 280     while (try { $r = $batch->next } otherwise { $r = -1 }) {
 281         if ($r == -1) {
 282             $logger->warn("Processing of record $count in set $filename failed.  Skipping this record");
 283             $count++;
 284         }
 285
 286         $logger->info("processing record $count");
 287
 288         try {
 289             (my $xml = $r->as_xml_record()) =~ s/\n//sog;
 290             $xml =~ s/^<\?xml.+\?\s*>//go;
 291             $xml =~ s/>\s+</></go;
 292             $xml =~ s/\p{Cc}//go;
 293             $xml = $U->entityize($xml);
 294             $xml =~ s/[\x00-\x1f]//go;
 295
 296             my $qrec;
 297             # Check the leader to ensure we've got something resembling the expected
 298             # Allow spaces to give records the benefit of the doubt
 299             my $ldr_type = substr($r->leader(), 6, 1);
 300             if ($type eq 'bib' && ($record_types{$ldr_type}) eq 'bib' || $ldr_type eq ' ') {
 301                 $qrec = _add_bib_rec( $e, $xml, $queue_id, $purpose, $bib_source ) or return $e->die_event;
 302             } elsif ($type eq 'auth' && ($record_types{$ldr_type}) eq 'auth' || $ldr_type eq ' ') {
 303                 $qrec = _add_auth_rec( $e, $xml, $queue_id, $purpose ) or return $e->die_event;
 304             } else {
 305                 # I don't know how to handle this type; rock on
 306                 $logger->error("In process_spool(), type was $type and leader type was $ldr_type ; not currently supported");
 307                 next;
 308             }
 309
 310             if($self->api_name =~ /stream_results/ and $qrec) {
 311                 $client->respond($qrec->id)
 312             } else {
 313                 $client->respond($count) if (++$count % $response_scale) == 0;
 314                 $response_scale *= 10 if ($count == ($response_scale * 10));
 315             }
 316         } catch Error with {
 317             my $error = shift;
 318             $logger->warn("Encountered a bad record at Vandelay ingest: ".$error);
 319         }
 320     }
 321
 322     $e->commit;
 323     unlink($filename);
 324     $cache->delete_cache('vandelay_import_spool_' . $fingerprint) if $fingerprint;
 325     return $count;
 326 }
 327
 328 __PACKAGE__->register_method(
 329     api_name    => "open-ils.vandelay.bib.process_spool",
 330     method      => "process_spool",
 331     api_level   => 1,
 332     argc        => 3,
 333     max_chunk_size => 0,
 334     record_type => 'bib'
 335 );
 336 __PACKAGE__->register_method(
 337     api_name    => "open-ils.vandelay.auth.process_spool",
 338     method      => "process_spool",
 339     api_level   => 1,
 340     argc        => 3,
 341     max_chunk_size => 0,
 342     record_type => 'auth'
 343 );
 344
 345 __PACKAGE__->register_method(
 346     api_name    => "open-ils.vandelay.bib.process_spool.stream_results",
 347     method      => "process_spool",
 348     api_level   => 1,
 349     argc        => 3,
 350     stream      => 1,
 351     max_chunk_size => 0,
 352     record_type => 'bib'
 353 );
 354 __PACKAGE__->register_method(
 355     api_name    => "open-ils.vandelay.auth.process_spool.stream_results",
 356     method      => "process_spool",
 357     api_level   => 1,
 358     argc        => 3,
 359     stream      => 1,
 360     max_chunk_size => 0,
 361     record_type => 'auth'
 362 );
 363
 364 __PACKAGE__->register_method(
 365     api_name    => "open-ils.vandelay.bib_queue.records.retrieve",
 366     method      => 'retrieve_queued_records',
 367     api_level   => 1,
 368     argc        => 2,
 369     stream      => 1,
 370     record_type => 'bib'
 371 );
 372 __PACKAGE__->register_method(
 373     api_name    => "open-ils.vandelay.auth_queue.records.retrieve",
 374     method      => 'retrieve_queued_records',
 375     api_level   => 1,
 376     argc        => 2,
 377     stream      => 1,
 378     record_type => 'auth'
 379 );
 380
 381 __PACKAGE__->register_method(
 382     api_name    => "open-ils.vandelay.bib_queue.records.matches.retrieve",
 383     method      => 'retrieve_queued_records',
 384     api_level   => 1,
 385     argc        => 2,
 386     stream      => 1,
 387     record_type => 'bib',
 388     signature   => {
 389         desc => q/Only retrieve queued bib records that have matches against existing records/
 390     }
 391 );
 392 __PACKAGE__->register_method(
 393     api_name    => "open-ils.vandelay.auth_queue.records.matches.retrieve",
 394     method      => 'retrieve_queued_records',
 395     api_level   => 1,
 396     argc        => 2,
 397     stream      => 1,
 398     record_type => 'auth',
 399     signature   => {
 400         desc => q/Only retrieve queued authority records that have matches against existing records/
 401     }
 402 );
 403
 404 sub retrieve_queued_records {
 405     my($self, $conn, $auth, $queue_id, $options) = @_;
 406
 407     $options ||= {};
 408     my $limit = $$options{limit} || 20;
 409     my $offset = $$options{offset} || 0;
 410     my $type = $self->{record_type};
 411
 412     my $e = new_editor(authtoken => $auth, xact => 1);
 413     return $e->die_event unless $e->checkauth;
 414
 415     my $queue;
 416     if($type eq 'bib') {
 417         $queue = $e->retrieve_vandelay_bib_queue($queue_id) or return $e->die_event;
 418     } else {
 419         $queue = $e->retrieve_vandelay_authority_queue($queue_id) or return $e->die_event;
 420     }
 421     my $evt = check_queue_perms($e, $type, $queue);
 422     return $evt if ($evt);
 423
 424     my $class = ($type eq 'bib') ? 'vqbr' : 'vqar';
 425     my $query = {
 426         select => {$class => ['id']},
 427         from => $class,
 428         where => {queue => $queue_id},
 429         distinct => 1,
 430         order_by => {$class => ['id']},
 431         limit => $limit,
 432         offset => $offset,
 433     };
 434
 435     $query->{where}->{import_time} = undef if $$options{non_imported};
 436
 437     if($$options{with_import_error}) {
 438
 439         $query->{from} = {$class => {vii => {type => 'right'}}};
 440         $query->{where}->{'-or'} = [
 441             {'+vqbr' => {import_error => {'!=' => undef}}},
 442             {'+vii' => {import_error => {'!=' => undef}}}
 443         ];
 444
 445     } else {
 446
 447         if($$options{with_rec_import_error}) {
 448             $query->{where}->{import_error} = {'!=' => undef};
 449
 450         } elsif( $$options{with_item_import_error} and $type eq 'bib') {
 451
 452             $query->{from} = {$class => 'vii'};
 453             $query->{where}->{'+vii'} = {import_error => {'!=' => undef}};
 454         }
 455     }
 456
 457     if($self->api_name =~ /matches/) {
 458         # find only records that have matches
 459         my $mclass = $type eq 'bib' ? 'vbm' : 'vam';
 460         $query->{from} = {$class => {$mclass => {type => 'right'}}};
 461     }
 462
 463     my $record_ids = $e->json_query($query);
 464
 465     my $retrieve = ($type eq 'bib') ?
 466         'retrieve_vandelay_queued_bib_record' : 'retrieve_vandelay_queued_authority_record';
 467
 468     for my $rec_id (@$record_ids) {
 469         my $flesh = ['attributes', 'matches'];
 470         push(@$flesh, 'import_items') if $$options{flesh_import_items};
 471         my $params = {flesh => 1, flesh_fields => {$class => $flesh}};
 472         my $rec = $e->$retrieve([$rec_id->{id}, $params]);
 473         $rec->clear_marc if $$options{clear_marc};
 474         $conn->respond($rec);
 475     }
 476
 477     $e->rollback;
 478     return undef;
 479 }
 480
 481 __PACKAGE__->register_method(
 482     api_name    => 'open-ils.vandelay.import_item.queue.retrieve',
 483     method      => 'retrieve_queue_import_items',
 484     api_level   => 1,
 485     argc        => 2,
 486     stream      => 1,
 487     authoritative => 1,
 488     signature => q/
 489         Returns Import Item (vii) objects for the selected queue.
 490         Filter options:
 491             with_import_error : only return items that failed to import
 492     /
 493 );
 494
 495 sub retrieve_queue_import_items {
 496     my($self, $conn, $auth, $q_id, $options) = @_;
 497
 498     $options ||= {};
 499     my $limit = $$options{limit} || 20;
 500     my $offset = $$options{offset} || 0;
 501
 502     my $e = new_editor(authtoken => $auth);
 503     return $e->event unless $e->checkauth;
 504
 505     my $queue = $e->retrieve_vandelay_bib_queue($q_id) or return $e->event;
 506     my $evt = check_queue_perms($e, 'bib', $queue);
 507     return $evt if $evt;
 508
 509     my $query = {
 510         select => {vii => ['id']},
 511         from => {
 512             vii => {
 513                 vqbr => {
 514                     join => {
 515                         'vbq' => {
 516                             field => 'id',
 517                             fkey => 'queue',
 518                             filter => {id => $q_id}
 519                         }
 520                     }
 521                 }
 522             }
 523         },
 524         order_by => {'vii' => ['record','id']},
 525         limit => $limit,
 526         offset => $offset
 527     };
 528
 529     $query->{where} = {'+vii' => {import_error => {'!=' => undef}}}
 530         if $$options{with_import_error};
 531
 532     my $items = $e->json_query($query);
 533     for my $item (@$items) {
 534         $conn->respond($e->retrieve_vandelay_import_item($item->{id}));
 535     }
 536
 537     return undef;
 538 }
 539
 540 sub check_queue_perms {
 541     my($e, $type, $queue) = @_;
 542     if ($type eq 'bib') {
 543         return $e->die_event unless
 544             ($e->allowed('CREATE_BIB_IMPORT_QUEUE', undef, $queue) ||
 545              $e->allowed('CREATE_BIB_IMPORT_QUEUE'));
 546     } else {
 547         return $e->die_event unless
 548             ($e->allowed('CREATE_AUTHORITY_IMPORT_QUEUE', undef, $queue) ||
 549              $e->allowed('CREATE_AUTHORITY_IMPORT_QUEUE'));
 550     }
 551
 552     return undef;
 553 }
 554
 555 __PACKAGE__->register_method(
 556     api_name    => "open-ils.vandelay.bib_record.list.import",
 557     method      => 'import_record_list',
 558     api_level   => 1,
 559     argc        => 2,
 560     stream      => 1,
 561     record_type => 'bib'
 562 );
 563
 564 __PACKAGE__->register_method(
 565     api_name    => "open-ils.vandelay.auth_record.list.import",
 566     method      => 'import_record_list',
 567     api_level   => 1,
 568     argc        => 2,
 569     stream      => 1,
 570     record_type => 'auth'
 571 );
 572
 573 sub import_record_list {
 574     my($self, $conn, $auth, $rec_ids, $args) = @_;
 575     my $e = new_editor(authtoken => $auth, xact => 1);
 576     return $e->die_event unless $e->checkauth;
 577     $args ||= {};
 578     my $err = import_record_list_impl($self, $conn, $rec_ids, $e->requestor, $args);
 579     try {$e->rollback} otherwise {};
 580     return $err if $err;
 581     return {complete => 1};
 582 }
 583
 584
 585 __PACKAGE__->register_method(
 586     api_name    => "open-ils.vandelay.bib_queue.import",
 587     method      => 'import_queue',
 588     api_level   => 1,
 589     argc        => 2,
 590     stream      => 1,
 591     max_chunk_size => 0,
 592     record_type => 'bib'
 593 );
 594
 595 __PACKAGE__->register_method(
 596     api_name    => "open-ils.vandelay.auth_queue.import",
 597     method      => 'import_queue',
 598     api_level   => 1,
 599     argc        => 2,
 600     stream      => 1,
 601     max_chunk_size => 0,
 602     record_type => 'auth'
 603 );
 604 __PACKAGE__->register_method(
 605     api_name    => "open-ils.vandelay.bib_queue.nomatch.import",
 606     method      => 'import_queue',
 607     api_level   => 1,
 608     argc        => 2,
 609     stream      => 1,
 610     signature   => {
 611         desc => q/Only import records that have no collisions/
 612     },
 613     max_chunk_size => 0,
 614     record_type => 'bib'
 615 );
 616
 617 __PACKAGE__->register_method(
 618     api_name    => "open-ils.vandelay.auth_queue.nomatch.import",
 619     method      => 'import_queue',
 620     api_level   => 1,
 621     argc        => 2,
 622     stream      => 1,
 623     signature   => {
 624         desc => q/Only import records that have no collisions/
 625     },
 626     max_chunk_size => 0,
 627     record_type => 'auth'
 628 );
 629 sub import_queue {
 630     my($self, $conn, $auth, $q_id, $options) = @_;
 631     my $e = new_editor(authtoken => $auth, xact => 1);
 632     return $e->die_event unless $e->checkauth;
 633     $options ||= {};
 634     my $type = $self->{record_type};
 635     my $class = ($type eq 'bib') ? 'vqbr' : 'vqar';
 636
 637     my $query = {queue => $q_id, import_time => undef};
 638
 639     if($self->api_name =~ /nomatch/) {
 640         my $matched_recs = queued_records_with_matches($e, $type, $q_id, undef, undef, {import_time => undef});
 641         $query->{id} = {'not in' => $matched_recs} if @$matched_recs;
 642     }
 643
 644     my $search = ($type eq 'bib') ?
 645         'search_vandelay_queued_bib_record' : 'search_vandelay_queued_authority_record';
 646     my $rec_ids = $e->$search($query, {idlist => 1});
 647     my $err = import_record_list_impl($self, $conn, $rec_ids, $e->requestor, $options);
 648     try {$e->rollback} otherwise {}; # only using this to make the read authoritative -- don't die from it
 649     return $err if $err;
 650     return {complete => 1};
 651 }
 652
 653 # returns a list of queued record IDs for a given queue that
 654 # have at least one entry in the match table
 655 sub queued_records_with_matches {
 656     my($e, $type, $q_id, $limit, $offset, $filter) = @_;
 657
 658     my $match_class = 'vbm';
 659     my $rec_class = 'vqbr';
 660     if($type eq 'auth') {
 661         $match_class = 'vam';
 662          $rec_class = 'vqar';
 663     }
 664
 665     $filter ||= {};
 666     $filter->{queue} = $q_id;
 667
 668     my $query = {
 669         distinct => 1,
 670         select => {$match_class => ['queued_record']},
 671         from => {
 672             $match_class => {
 673                 $rec_class => {
 674                     field => 'id',
 675                     fkey => 'queued_record',
 676                     filter => $filter,
 677                 }
 678             }
 679         }
 680     };
 681
 682     if($limit or defined $offset) {
 683         $limit ||= 20;
 684         $offset ||= 0;
 685         $query->{limit} = $limit;
 686         $query->{offset} = $offset;
 687     }
 688
 689     my $data = $e->json_query($query);
 690     return [ map {$_->{queued_record}} @$data ];
 691 }
 692
 693
 694 sub import_record_list_impl {
 695     my($self, $conn, $rec_ids, $requestor, $args) = @_;
 696
 697     my $overlay_map = $args->{overlay_map} || {};
 698     my $type = $self->{record_type};
 699     my %queues;
 700
 701     my $report_args = {
 702         progress => 1,
 703         step => 1,
 704         conn => $conn,
 705         total => scalar(@$rec_ids),
 706         report_all => $$args{report_all}
 707     };
 708
 709     my $auto_overlay_exact = $$args{auto_overlay_exact};
 710     my $auto_overlay_1match = $$args{auto_overlay_1match};
 711     my $auto_overlay_best = $$args{auto_overlay_best_match};
 712     my $match_quality_ratio = $$args{match_quality_ratio};
 713     my $merge_profile = $$args{merge_profile};
 714     my $bib_source = $$args{bib_source};
 715     my $import_no_match = $$args{import_no_match};
 716
 717     my $overlay_func = 'vandelay.overlay_bib_record';
 718     my $auto_overlay_func = 'vandelay.auto_overlay_bib_record';
 719     my $auto_overlay_best_func = 'vandelay.auto_overlay_bib_record_with_best'; # XXX bib-only
 720     my $retrieve_func = 'retrieve_vandelay_queued_bib_record';
 721     my $update_func = 'update_vandelay_queued_bib_record';
 722     my $search_func = 'search_vandelay_queued_bib_record';
 723     my $retrieve_queue_func = 'retrieve_vandelay_bib_queue';
 724     my $update_queue_func = 'update_vandelay_bib_queue';
 725     my $rec_class = 'vqbr';
 726
 727     my $editor = new_editor();
 728
 729     my %bib_sources;
 730     my $sources = $editor->search_config_bib_source({id => {'!=' => undef}});
 731     $bib_sources{$_->id} = $_->source for @$sources;
 732
 733     if($type eq 'auth') {
 734         $overlay_func =~ s/bib/auth/o;
 735         $auto_overlay_func = s/bib/auth/o;
 736         $retrieve_func =~ s/bib/authority/o;
 737         $retrieve_queue_func =~ s/bib/authority/o;
 738         $update_queue_func =~ s/bib/authority/o;
 739         $update_func =~ s/bib/authority/o;
 740         $search_func =~ s/bib/authority/o;
 741         $rec_class = 'vqar';
 742     }
 743
 744     my @success_rec_ids;
 745     for my $rec_id (@$rec_ids) {
 746
 747         my $error = 0;
 748         my $overlay_target = $overlay_map->{$rec_id};
 749
 750         my $e = new_editor(xact => 1);
 751         $e->requestor($requestor);
 752
 753         $$report_args{e} = $e;
 754         $$report_args{import_error} = undef;
 755         $$report_args{evt} = undef;
 756
 757         my $rec = $e->$retrieve_func([
 758             $rec_id,
 759             {   flesh => 1,
 760                 flesh_fields => { $rec_class => ['matches']},
 761             }
 762         ]);
 763
 764         unless($rec) {
 765             $$report_args{evt} = $e->event;
 766             finish_rec_import_attempt($report_args);
 767             next;
 768         }
 769
 770         $$report_args{rec} = $rec;
 771
 772         if($rec->import_time) {
 773             $e->rollback;
 774             next;
 775         }
 776
 777         $queues{$rec->queue} = 1;
 778
 779         my $record;
 780         my $imported = 0;
 781
 782         if(defined $overlay_target) {
 783             # Caller chose an explicit overlay target
 784
 785             my $res = $e->json_query(
 786                 {
 787                     from => [
 788                         $overlay_func,
 789                         $rec->id,
 790                         $overlay_target,
 791                         $merge_profile
 792                     ]
 793                 }
 794             );
 795
 796             if($res and ($res = $res->[0])) {
 797
 798                 if($res->{$overlay_func} eq 't') {
 799                     $logger->info("vl: $type direct overlay succeeded for queued rec " .
 800                         $rec->id . " and overlay target $overlay_target");
 801                     $imported = 1;
 802                 }
 803
 804             } else {
 805                 $error = 1;
 806                 $logger->error("vl: Error attempting overlay with func=$overlay_func, profile=$merge_profile, record=$rec_id");
 807             }
 808
 809         } else {
 810
 811             if($auto_overlay_1match) {
 812                 # caller says to overlay if there is exactly 1 match
 813
 814                 my %match_recs = map { $_->eg_record => 1 } @{$rec->matches};
 815
 816                 if( scalar(keys %match_recs) == 1) { # all matches point to the same record
 817
 818                     my $res = $e->json_query(
 819                         {
 820                             from => [
 821                                 $overlay_func,
 822                                 $rec->id,
 823                                 $rec->matches->[0]->eg_record,
 824                                 $merge_profile
 825                             ]
 826                         }
 827                     );
 828
 829                     if($res and ($res = $res->[0])) {
 830
 831                         if($res->{$overlay_func} eq 't') {
 832                             $logger->info("vl: $type overlay-1match succeeded for queued rec " . $rec->id);
 833                             $imported = 1;
 834                         }
 835
 836                     } else {
 837                         $error = 1;
 838                         $logger->error("vl: Error attempting overlay with func=$overlay_func, profile=$merge_profile, record=$rec_id");
 839                     }
 840                 }
 841             }
 842
 843             if(!$imported and !$error and $auto_overlay_exact and scalar(@{$rec->matches}) == 1 ) {
 844
 845                 # caller says to overlay if there is an /exact/ match
 846
 847                 my $res = $e->json_query(
 848                     {
 849                         from => [
 850                             $auto_overlay_func,
 851                             $rec->id,
 852                             $merge_profile
 853                         ]
 854                     }
 855                 );
 856
 857                 if($res and ($res = $res->[0])) {
 858
 859                     if($res->{$auto_overlay_func} eq 't') {
 860                         $logger->info("vl: $type auto-overlay succeeded for queued rec " . $rec->id);
 861                         $imported = 1;
 862                     } else {
 863                         $logger->info("vl: $type auto-overlay failed for queued rec " . $rec->id);
 864                     }
 865
 866                 } else {
 867                     $error = 1;
 868                     $logger->error("vl: Error attempting overlay with func=$auto_overlay_func, profile=$merge_profile, record=$rec_id");
 869                 }
 870             }
 871
 872             if(!$imported and !$error and $auto_overlay_best and scalar(@{$rec->matches}) > 0 ) {
 873
 874                 # caller says to overlay the best match
 875
 876                 my $res = $e->json_query(
 877                     {
 878                         from => [
 879                             $auto_overlay_best_func,
 880                             $rec->id,
 881                             $merge_profile,
 882                             $match_quality_ratio
 883                         ]
 884                     }
 885                 );
 886
 887                 if($res and ($res = $res->[0])) {
 888
 889                     if($res->{$auto_overlay_best_func} eq 't') {
 890                         $logger->info("vl: $type auto-overlay-best succeeded for queued rec " . $rec->id);
 891                         $imported = 1;
 892                     } else {
 893                         $$report_args{import_error} = 'overlay.record.quality' if $match_quality_ratio > 0;
 894                         $logger->info("vl: $type auto-overlay-best failed for queued rec " . $rec->id);
 895                     }
 896
 897                 } else {
 898                     $error = 1;
 899                     $logger->error("vl: Error attempting overlay with func=$auto_overlay_best_func, ".
 900                         "quality_ratio=$match_quality_ratio, profile=$merge_profile, record=$rec_id");
 901                 }
 902             }
 903
 904             if(!$imported and !$error and $import_no_match and scalar(@{$rec->matches}) == 0) {
 905
 906                 # No overlay / merge occurred.  Do a traditional record import by creating a new record
 907
 908                 $logger->info("vl: creating new $type record for queued record $rec_id");
 909                 if($type eq 'bib') {
 910                     $record = OpenILS::Application::Cat::BibCommon->biblio_record_xml_import($e, $rec->marc, $bib_sources{$rec->bib_source});
 911                 } else {
 912
 913                     $record = OpenILS::Application::Cat::AuthCommon->import_authority_record($e, $rec->marc); #$source);
 914                 }
 915
 916                 if($U->event_code($record)) {
 917                     $$report_args{import_error} = 'import.duplicate.tcn' if $record->{textcode} eq 'TCN_EXISTS';
 918                     $$report_args{evt} = $record;
 919
 920                 } else {
 921
 922                     $logger->info("vl: successfully imported new $type record");
 923                     $rec->imported_as($record->id);
 924                     $rec->import_time('now');
 925                     $rec->clear_import_error;
 926                     $rec->clear_error_detail;
 927                     $imported = 1 if $e->$update_func($rec);
 928                 }
 929             }
 930         }
 931
 932         if($imported) {
 933             push @success_rec_ids, $rec_id;
 934             finish_rec_import_attempt($report_args);
 935
 936         } else {
 937             # Send an update whenever there's an error
 938             $$report_args{evt} = $e->event unless $$report_args{evt};
 939             finish_rec_import_attempt($report_args);
 940         }
 941     }
 942
 943     # see if we need to mark any queues as complete
 944     for my $q_id (keys %queues) {
 945
 946         my $e = new_editor(xact => 1);
 947         my $remaining = $e->$search_func(
 948             [{queue => $q_id, import_time => undef}, {limit =>1}], {idlist => 1});
 949
 950         unless(@$remaining) {
 951             my $queue = $e->$retrieve_queue_func($q_id);
 952
 953             unless($U->is_true($queue->complete)) {
 954                 $queue->complete('t');
 955                 $e->$update_queue_func($queue) or return $e->die_event;
 956                 $e->commit;
 957                 next;
 958             }
 959         }
 960         $e->rollback;
 961     }
 962
 963     # import the copies
 964     import_record_asset_list_impl($conn, \@success_rec_ids, $requestor);
 965
 966     $conn->respond({total => $$report_args{total}, progress => $$report_args{progress}});
 967     return undef;
 968 }
 969
 970 # tracks any import errors, commits the current xact, responds to the client
 971 sub finish_rec_import_attempt {
 972     my $args = shift;
 973     my $evt = $$args{evt};
 974     my $rec = $$args{rec};
 975     my $e = $$args{e};
 976
 977     my $error = $$args{import_error};
 978     $error = 'general.unknown' if $evt and not $error;
 979
 980     # error tracking
 981     if($rec) {
 982
 983         if($error or $evt) {
 984             # failed import
 985             # since an error occurred, there's no guarantee the transaction wasn't
 986             # rolled back.  force a rollback and create a new editor.
 987             $e->rollback;
 988             $e = new_editor(xact => 1);
 989             $rec->import_error($error);
 990
 991             if($evt) {
 992                 my $detail = sprintf("%s : %s", $evt->{textcode}, substr($evt->{desc}, 0, 140));
 993                 $rec->error_detail($detail);
 994             }
 995
 996             my $method = 'update_vandelay_queued_bib_record';
 997             $method =~ s/bib/authority/ if $$args{type} eq 'auth';
 998             $e->$method($rec) and $e->commit or $e->rollback;
 999
1000         } else {
1001             # commit the successful import
1002             $e->commit;
1003         }
1004
1005     } else {
1006         # requested queued record was not found
1007         $e->rollback;
1008     }
1009
1010     # respond to client
1011     if($$args{report_all} or ($$args{progress} % $$args{step}) == 0) {
1012         $$args{conn}->respond({
1013             total => $$args{total},
1014             progress => $$args{progress},
1015             imported => ($rec) ? $rec->id : undef,
1016             err_event => $evt
1017         });
1018         $$args{step} *= 2 unless $$args{step} == 256;
1019     }
1020
1021     $$args{progress}++;
1022 }
1023
1024
1025
1026
1027
1028 __PACKAGE__->register_method(
1029     api_name    => "open-ils.vandelay.bib_queue.owner.retrieve",
1030     method      => 'owner_queue_retrieve',
1031     api_level   => 1,
1032     argc        => 2,
1033     stream      => 1,
1034     record_type => 'bib'
1035 );
1036 __PACKAGE__->register_method(
1037     api_name    => "open-ils.vandelay.authority_queue.owner.retrieve",
1038     method      => 'owner_queue_retrieve',
1039     api_level   => 1,
1040     argc        => 2,
1041     stream      => 1,
1042     record_type => 'auth'
1043 );
1044
1045 sub owner_queue_retrieve {
1046     my($self, $conn, $auth, $owner_id, $filters) = @_;
1047     my $e = new_editor(authtoken => $auth, xact => 1);
1048     return $e->die_event unless $e->checkauth;
1049     $owner_id = $e->requestor->id; # XXX add support for viewing other's queues?
1050     my $queues;
1051     $filters ||= {};
1052     my $search = {owner => $owner_id};
1053     $search->{$_} = $filters->{$_} for keys %$filters;
1054
1055     if($self->{record_type} eq 'bib') {
1056         $queues = $e->search_vandelay_bib_queue(
1057             [$search, {order_by => {vbq => 'evergreen.lowercase(name)'}}]);
1058     } else {
1059         $queues = $e->search_vandelay_authority_queue(
1060             [$search, {order_by => {vaq => 'evergreen.lowercase(name)'}}]);
1061     }
1062     $conn->respond($_) for @$queues;
1063     $e->rollback;
1064     return undef;
1065 }
1066
1067 __PACKAGE__->register_method(
1068     api_name    => "open-ils.vandelay.bib_queue.delete",
1069     method      => "delete_queue",
1070     api_level   => 1,
1071     argc        => 2,
1072     record_type => 'bib'
1073 );
1074 __PACKAGE__->register_method(
1075     api_name    => "open-ils.vandelay.auth_queue.delete",
1076     method      => "delete_queue",
1077     api_level   => 1,
1078     argc        => 2,
1079     record_type => 'auth'
1080 );
1081
1082 sub delete_queue {
1083     my($self, $conn, $auth, $q_id) = @_;
1084     my $e = new_editor(xact => 1, authtoken => $auth);
1085     return $e->die_event unless $e->checkauth;
1086     if($self->{record_type} eq 'bib') {
1087         return $e->die_event unless $e->allowed('CREATE_BIB_IMPORT_QUEUE');
1088         my $queue = $e->retrieve_vandelay_bib_queue($q_id)
1089             or return $e->die_event;
1090         $e->delete_vandelay_bib_queue($queue)
1091             or return $e->die_event;
1092     } else {
1093            return $e->die_event unless $e->allowed('CREATE_AUTHORITY_IMPORT_QUEUE');
1094         my $queue = $e->retrieve_vandelay_authority_queue($q_id)
1095             or return $e->die_event;
1096         $e->delete_vandelay_authority_queue($queue)
1097             or return $e->die_event;
1098     }
1099     $e->commit;
1100     return 1;
1101 }
1102
1103
1104 __PACKAGE__->register_method(
1105     api_name    => "open-ils.vandelay.queued_bib_record.html",
1106     method      => 'queued_record_html',
1107     api_level   => 1,
1108     argc        => 2,
1109     stream      => 1,
1110     record_type => 'bib'
1111 );
1112 __PACKAGE__->register_method(
1113     api_name    => "open-ils.vandelay.queued_authority_record.html",
1114     method      => 'queued_record_html',
1115     api_level   => 1,
1116     argc        => 2,
1117     stream      => 1,
1118     record_type => 'auth'
1119 );
1120
1121 sub queued_record_html {
1122     my($self, $conn, $auth, $rec_id) = @_;
1123     my $e = new_editor(xact=>1,authtoken => $auth);
1124     return $e->die_event unless $e->checkauth;
1125     my $rec;
1126     if($self->{record_type} eq 'bib') {
1127         $rec = $e->retrieve_vandelay_queued_bib_record($rec_id)
1128             or return $e->die_event;
1129     } else {
1130         $rec = $e->retrieve_vandelay_queued_authority_record($rec_id)
1131             or return $e->die_event;
1132     }
1133
1134     $e->rollback;
1135     return $U->simplereq(
1136         'open-ils.search',
1137         'open-ils.search.biblio.record.html', undef, 1, $rec->marc);
1138 }
1139
1140
1141 __PACKAGE__->register_method(
1142     api_name    => "open-ils.vandelay.bib_queue.summary.retrieve",
1143     method      => 'retrieve_queue_summary',
1144     api_level   => 1,
1145     argc        => 2,
1146     stream      => 1,
1147     record_type => 'bib'
1148 );
1149 __PACKAGE__->register_method(
1150     api_name    => "open-ils.vandelay.auth_queue.summary.retrieve",
1151     method      => 'retrieve_queue_summary',
1152     api_level   => 1,
1153     argc        => 2,
1154     stream      => 1,
1155     record_type => 'auth'
1156 );
1157
1158 sub retrieve_queue_summary {
1159     my($self, $conn, $auth, $queue_id) = @_;
1160     my $e = new_editor(xact=>1, authtoken => $auth);
1161     return $e->die_event unless $e->checkauth;
1162
1163     my $queue;
1164     my $type = $self->{record_type};
1165     if($type eq 'bib') {
1166         $queue = $e->retrieve_vandelay_bib_queue($queue_id)
1167             or return $e->die_event;
1168     } else {
1169         $queue = $e->retrieve_vandelay_authority_queue($queue_id)
1170             or return $e->die_event;
1171     }
1172
1173     my $evt = check_queue_perms($e, $type, $queue);
1174     return $evt if $evt;
1175
1176     my $search = 'search_vandelay_queued_bib_record';
1177     $search =~ s/bib/authority/ if $type ne 'bib';
1178
1179     my $summary = {
1180         queue => $queue,
1181         total => scalar(@{$e->$search({queue => $queue_id}, {idlist=>1})}),
1182         imported => scalar(@{$e->$search({queue => $queue_id, import_time => {'!=' => undef}}, {idlist=>1})}),
1183     };
1184
1185     my $class = ($type eq 'bib') ? 'vqbr' : 'vqar';
1186     $summary->{rec_import_errors} = $e->json_query({
1187         select => {$class => [{alias => 'count', column => 'id', transform => 'count', aggregate => 1}]},
1188         from => $class,
1189         where => {queue => $queue_id, import_error => {'!=' => undef}}
1190     })->[0]->{count};
1191
1192     if($type eq 'bib') {
1193
1194         my $query = {
1195             select => {vii => [{alias => 'count', column => 'id', transform => 'count', aggregate => 1}]},
1196             from => 'vii',
1197             where => {
1198                 record => {
1199                     in => {
1200                         select => {vqbr => ['id']},
1201                         from => 'vqbr',
1202                         where => {queue => $queue_id}
1203                     }
1204                 }
1205             }
1206         };
1207
1208         $summary->{total_items} = $e->json_query($query)->[0]->{count};
1209         $query->{where}->{import_error} = {'!=' => undef};
1210         $summary->{item_import_errors} = $e->json_query($query)->[0]->{count};
1211     }
1212
1213     return $summary;
1214 }
1215
1216 # --------------------------------------------------------------------------------
1217 # Given a list of queued record IDs, imports all items attached to those records
1218 # --------------------------------------------------------------------------------
1219 sub import_record_asset_list_impl {
1220     my($conn, $rec_ids, $requestor) = @_;
1221
1222     my $roe = new_editor(xact=> 1, requestor => $requestor);
1223
1224     # for speed, filter out any records have not been
1225     # imported or have no import items to load
1226     $rec_ids = $roe->json_query({
1227         select => {vqbr => ['id']},
1228         from => {vqbr => 'vii'},
1229         where => {'+vqbr' => {import_time => {'!=' => undef}}},
1230         distinct => 1
1231     });
1232     $rec_ids = [map {$_->{id}} @$rec_ids];
1233
1234     my $report_args = {
1235         conn => $conn,
1236         total => scalar(@$rec_ids),
1237         step => 1, # how often to respond
1238         progress => 1,
1239         in_count => 0,
1240     };
1241
1242     for my $rec_id (@$rec_ids) {
1243         my $rec = $roe->retrieve_vandelay_queued_bib_record($rec_id);
1244         my $item_ids = $roe->search_vandelay_import_item({record => $rec->id}, {idlist=>1});
1245
1246         for my $item_id (@$item_ids) {
1247             my $e = new_editor(requestor => $requestor, xact => 1);
1248             my $item = $e->retrieve_vandelay_import_item($item_id);
1249             $$report_args{import_item} = $item;
1250             $$report_args{e} = $e;
1251             $$report_args{import_error} = undef;
1252             $$report_args{evt} = undef;
1253
1254             # --------------------------------------------------------------------------------
1255             # Find or create the volume
1256             # --------------------------------------------------------------------------------
1257             my ($vol, $evt) =
1258                 OpenILS::Application::Cat::AssetCommon->find_or_create_volume(
1259                     $e, $item->call_number, $rec->imported_as, $item->owning_lib);
1260
1261             if($evt) {
1262
1263                 $$report_args{evt} = $evt;
1264                 respond_with_status($report_args);
1265                 next;
1266             }
1267
1268             # --------------------------------------------------------------------------------
1269             # Create the new copy
1270             # --------------------------------------------------------------------------------
1271             my $copy = Fieldmapper::asset::copy->new;
1272             $copy->loan_duration(2);
1273             $copy->fine_level(2);
1274             $copy->barcode($item->barcode);
1275             $copy->location($item->location);
1276             $copy->circ_lib($item->circ_lib || $item->owning_lib);
1277             $copy->status( defined($item->status) ? $item->status : OILS_COPY_STATUS_IN_PROCESS );
1278             $copy->circulate($item->circulate);
1279             $copy->deposit($item->deposit);
1280             $copy->deposit_amount($item->deposit_amount);
1281             $copy->ref($item->ref);
1282             $copy->holdable($item->holdable);
1283             $copy->price($item->price);
1284             $copy->circ_as_type($item->circ_as_type);
1285             $copy->alert_message($item->alert_message);
1286             $copy->opac_visible($item->opac_visible);
1287             $copy->circ_modifier($item->circ_modifier);
1288
1289             # --------------------------------------------------------------------------------
1290             # see if a valid circ_modifier was provided
1291             # --------------------------------------------------------------------------------
1292             if($copy->circ_modifier and not $e->search_config_circ_modifier({code=>$item->circ_modifier})->[0]) {
1293                 $$report_args{evt} = $e->die_event;
1294                 $$report_args{import_error} = 'import.item.invalid.circ_modifier';
1295                 respond_with_status($report_args);
1296                 next;
1297             }
1298
1299             if($copy->location and not $e->retrieve_asset_copy_location($copy->location)) {
1300                 $$report_args{evt} = $e->die_event;
1301                 $$report_args{import_error} = 'import.item.invalid.location';
1302                 respond_with_status($report_args);
1303                 next;
1304             }
1305
1306             if($evt = OpenILS::Application::Cat::AssetCommon->create_copy($e, $vol, $copy)) {
1307                 $$report_args{evt} = $evt;
1308                 $$report_args{import_error} = 'import.item.duplicate.barcode'
1309                     if $evt->{textcode} eq 'ITEM_BARCODE_EXISTS';
1310                 respond_with_status($report_args);
1311                 next;
1312             }
1313
1314             # --------------------------------------------------------------------------------
1315             # create copy notes
1316             # --------------------------------------------------------------------------------
1317             $evt = OpenILS::Application::Cat::AssetCommon->create_copy_note(
1318                 $e, $copy, '', $item->pub_note, 1) if $item->pub_note;
1319
1320             if($evt) {
1321                 $$report_args{evt} = $evt;
1322                 respond_with_status($report_args);
1323                 next;
1324             }
1325
1326             $evt = OpenILS::Application::Cat::AssetCommon->create_copy_note(
1327                 $e, $copy, '', $item->priv_note, 1) if $item->priv_note;
1328
1329             if($evt) {
1330                 $$report_args{evt} = $evt;
1331                 respond_with_status($report_args);
1332                 next;
1333             }
1334
1335             # --------------------------------------------------------------------------------
1336             # Item import succeeded
1337             # --------------------------------------------------------------------------------
1338             $e->commit;
1339             $$report_args{in_count}++;
1340             respond_with_status($report_args);
1341             $logger->info("vl: successfully imported item " . $item->barcode);
1342         }
1343
1344     }
1345
1346     $roe->rollback;
1347     return undef;
1348 }
1349
1350
1351 sub respond_with_status {
1352     my $args = shift;
1353     my $e = $$args{e};
1354
1355     #  If the import failed, track the failure reason
1356
1357     my $error = $$args{import_error};
1358     my $evt = $$args{evt};
1359
1360     if($error or $evt) {
1361
1362         my $item = $$args{import_item};
1363         $logger->info("vl: unable to import item " . $item->barcode);
1364
1365         $error ||= 'general.unknown';
1366         $item->import_error($error);
1367
1368         if($evt) {
1369             my $detail = sprintf("%s : %s", $evt->{textcode}, substr($evt->{desc}, 0, 140));
1370             $item->error_detail($detail);
1371         }
1372
1373         # state of the editor is unknown at this point.  Force a rollback and start over.
1374         $e->rollback;
1375         $e = new_editor(xact => 1);
1376         $e->update_vandelay_import_item($item);
1377         $e->commit;
1378     }
1379
1380     if($$args{report_all} or ($$args{progress} % $$args{step}) == 0) {
1381         $$args{conn}->respond({
1382             map { $_ => $args->{$_} } qw/total progress success_count/,
1383             err_event => $evt
1384         });
1385         $$args{step} *= 2 unless $$args{step} == 256;
1386     }
1387
1388     $$args{progress}++;
1389 }
1390
1391 __PACKAGE__->register_method(
1392     api_name    => "open-ils.vandelay.match_set.get_tree",
1393     method      => "match_set_get_tree",
1394     api_level   => 1,
1395     argc        => 2,
1396     signature   => {
1397         desc    => q/For a given vms object, return a tree of match set points
1398                     represented by a vmsp object with recursively fleshed
1399                     children./
1400     }
1401 );
1402
1403 sub match_set_get_tree {
1404     my ($self, $conn, $authtoken, $match_set_id) = @_;
1405
1406     $match_set_id = int($match_set_id) or return;
1407
1408     my $e = new_editor("authtoken" => $authtoken);
1409     $e->checkauth or return $e->die_event;
1410
1411     my $set = $e->retrieve_vandelay_match_set($match_set_id) or
1412         return $e->die_event;
1413
1414     $e->allowed("ADMIN_IMPORT_MATCH_SET", $set->owner) or
1415         return $e->die_event;
1416
1417     my $tree = $e->search_vandelay_match_set_point([
1418         {"match_set" => $match_set_id, "parent" => undef},
1419         {"flesh" => -1, "flesh_fields" => {"vmsp" => ["children"]}}
1420     ]) or return $e->die_event;
1421
1422     return pop @$tree;
1423 }
1424
1425
1426 __PACKAGE__->register_method(
1427     api_name    => "open-ils.vandelay.match_set.update",
1428     method      => "match_set_update_tree",
1429     api_level   => 1,
1430     argc        => 3,
1431     signature   => {
1432         desc => q/Replace any vmsp objects associated with a given (by ID) vms
1433                 with the given objects (recursively fleshed vmsp tree)./
1434     }
1435 );
1436
1437 sub _walk_new_vmsp {
1438     my ($e, $match_set_id, $node, $parent_id) = @_;
1439
1440     my $point = new Fieldmapper::vandelay::match_set_point;
1441     $point->parent($parent_id);
1442     $point->match_set($match_set_id);
1443     $point->$_($node->$_) for (qw/bool_op svf tag subfield negate quality/);
1444
1445     $e->create_vandelay_match_set_point($point) or return $e->die_event;
1446
1447     $parent_id = $e->data->id;
1448     if ($node->children && @{$node->children}) {
1449         for (@{$node->children}) {
1450             return $e->die_event if
1451                 _walk_new_vmsp($e, $match_set_id, $_, $parent_id);
1452         }
1453     }
1454
1455     return;
1456 }
1457
1458 sub match_set_update_tree {
1459     my ($self, $conn, $authtoken, $match_set_id, $tree) = @_;
1460
1461     my $e = new_editor("xact" => 1, "authtoken" => $authtoken);
1462     $e->checkauth or return $e->die_event;
1463
1464     my $set = $e->retrieve_vandelay_match_set($match_set_id) or
1465         return $e->die_event;
1466
1467     $e->allowed("ADMIN_IMPORT_MATCH_SET", $set->owner) or
1468         return $e->die_event;
1469
1470     my $existing = $e->search_vandelay_match_set_point([
1471         {"match_set" => $match_set_id},
1472         {"order_by" => {"vmsp" => "id DESC"}}
1473     ]) or return $e->die_event;
1474
1475     # delete points, working up from leaf points to the root
1476     while(@$existing) {
1477         for my $point (shift @$existing) {
1478             if( grep {$_->parent eq $point->id} @$existing) {
1479                 push(@$existing, $point);
1480             } else {
1481                 $e->delete_vandelay_match_set_point($point) or return $e->die_event;
1482             }
1483         }
1484     }
1485
1486     _walk_new_vmsp($e, $match_set_id, $tree);
1487
1488     $e->commit or return $e->die_event;
1489 }
1490
1491 1;