Open-ILS/src/perlmods/OpenILS/Utils/CStoreEditor.pm

   1 use strict; use warnings;
   2 package OpenILS::Utils::CStoreEditor;
   3 use OpenILS::Application::AppUtils;
   4 use OpenSRF::AppSession;
   5 use OpenSRF::EX qw(:try);
   6 use OpenILS::Utils::Fieldmapper;
   7 use OpenILS::Event;
   8 use Data::Dumper;
   9 use JSON;
  10 use OpenSRF::Utils::Logger qw($logger);
  11 my $U = "OpenILS::Application::AppUtils";
  12 my %PERMS;
  13 my $cache;
  14
  15 #my %PERMS = (
  16 #       'biblio.record_entry'   => { update => 'UPDATE_MARC' },
  17 #       'asset.copy'                            => { update => 'UPDATE_COPY'},
  18 #       'asset.call_number'             => { update => 'UPDATE_VOLUME'},
  19 #       'action.circulation'            => { retrieve => 'VIEW_CIRCULATIONS'},
  20 #);
  21
  22
  23
  24 # -----------------------------------------------------------------------------
  25 # Export some useful functions
  26 # -----------------------------------------------------------------------------
  27 use vars qw(@EXPORT_OK %EXPORT_TAGS);
  28 use Exporter;
  29 use base qw/Exporter/;
  30 push @EXPORT_OK, ( 'new_editor', 'new_rstore_editor' );
  31 %EXPORT_TAGS = ( funcs => [ qw/ new_editor new_rstore_editor / ] );
  32
  33 sub new_editor { return OpenILS::Utils::CStoreEditor->new(@_); }
  34
  35 sub new_rstore_editor {
  36         my $e = OpenILS::Utils::CStoreEditor->new(@_);
  37         $e->app('open-ils.reporter-store');
  38         return $e;
  39 }
  40
  41
  42 # -----------------------------------------------------------------------------
  43 # Log levels
  44 # -----------------------------------------------------------------------------
  45 use constant E => 'error';
  46 use constant W => 'warn';
  47 use constant I => 'info';
  48 use constant D => 'debug';
  49 use constant A => 'activity';
  50
  51
  52
  53 # -----------------------------------------------------------------------------
  54 # Params include:
  55 #       xact=><true> : creates a storage transaction
  56 #       authtoken=>$token : the login session key
  57 # -----------------------------------------------------------------------------
  58 sub new {
  59         my( $class, %params ) = @_;
  60         $class = ref($class) || $class;
  61         my $self = bless( \%params, $class );
  62         $self->{checked_perms} = {};
  63         return $self;
  64 }
  65
  66
  67 sub app {
  68         my( $self, $app ) = @_;
  69         $self->{app} = $app if $app;
  70         $self->{app} = 'open-ils.cstore' unless $self->{app};
  71         return $self->{app};
  72 }
  73
  74
  75 # -----------------------------------------------------------------------------
  76 # Log the editor metadata along with the log string
  77 # -----------------------------------------------------------------------------
  78 sub log {
  79         my( $self, $lev, $str ) = @_;
  80         my $s = "editor[";
  81         $s .= "0|" unless $self->{xact};
  82         $s .= "1|" if $self->{xact};
  83         $s .= "0" unless $self->requestor;
  84         $s .= $self->requestor->id if $self->requestor;
  85         $s .= "]";
  86         $logger->$lev("$s $str");
  87 }
  88
  89 # -----------------------------------------------------------------------------
  90 # Verifies the auth token and fetches the requestor object
  91 # -----------------------------------------------------------------------------
  92 sub checkauth {
  93         my $self = shift;
  94         $self->log(D, "checking auth token ".$self->authtoken);
  95         my ($reqr, $evt) = $U->checkses($self->authtoken);
  96         $self->event($evt) if $evt;
  97         return $self->{requestor} = $reqr;
  98 }
  99
 100
 101 =head test
 102 sub checkauth {
 103         my $self = shift;
 104         $cache = OpenSRF::Utils::Cache->new('global') unless $cache;
 105         $self->log(D, "checking cached auth token ".$self->authtoken);
 106         my $user = $cache->get_cache("oils_auth_".$self->authtoken);
 107         return $self->{requestor} = $user->{userobj} if $user;
 108         $self->event(OpenILS::Event->new('NO_SESSION'));
 109         return undef;
 110 }
 111 =cut
 112
 113
 114 # -----------------------------------------------------------------------------
 115 # Returns the last generated event
 116 # -----------------------------------------------------------------------------
 117 sub event {
 118         my( $self, $evt ) = @_;
 119         $self->{event} = $evt if $evt;
 120         return $self->{event};
 121 }
 122
 123 # -----------------------------------------------------------------------------
 124 # Destroys the transaction and disconnects where necessary,
 125 # then returns the last event that occurred
 126 # -----------------------------------------------------------------------------
 127 sub die_event {
 128         my $self = shift;
 129         $self->rollback;
 130         return $self->event;
 131 }
 132
 133
 134 # -----------------------------------------------------------------------------
 135 # Clears the last caught event
 136 # -----------------------------------------------------------------------------
 137 sub clear_event {
 138         my $self = shift;
 139         $self->{event} = undef;
 140 }
 141
 142 sub authtoken {
 143         my( $self, $auth ) = @_;
 144         $self->{authtoken} = $auth if $auth;
 145         return $self->{authtoken};
 146 }
 147
 148 # -----------------------------------------------------------------------------
 149 # fetches the session, creating if necessary.  If 'xact' is true on this
 150 # object, a db session is created
 151 # -----------------------------------------------------------------------------
 152 sub session {
 153         my( $self, $session ) = @_;
 154         $self->{session} = $session if $session;
 155
 156         if(!$self->{session}) {
 157                 $self->{session} = OpenSRF::AppSession->create($self->app);
 158
 159                 if( ! $self->{session} ) {
 160                         my $str = "Error creating cstore session with OpenSRF::AppSession->create()!";
 161                         $self->log(E, $str);
 162                         throw OpenSRF::EX::ERROR ($str);
 163                 }
 164
 165                 $self->{session}->connect if $self->{xact} or $self->{connect};
 166                 $self->xact_start if $self->{xact};
 167         }
 168         return $self->{session};
 169 }
 170
 171
 172 # -----------------------------------------------------------------------------
 173 # Starts a storage transaction
 174 # -----------------------------------------------------------------------------
 175 sub xact_start {
 176         my $self = shift;
 177         $self->log(D, "starting new db session");
 178         my $stat = $self->request($self->app . '.transaction.begin');
 179         $self->log(E, "error starting database transaction") unless $stat;
 180         return $stat;
 181 }
 182
 183 # -----------------------------------------------------------------------------
 184 # Commits a storage transaction
 185 # -----------------------------------------------------------------------------
 186 sub xact_commit {
 187         my $self = shift;
 188         $self->log(D, "comitting db session");
 189         my $stat = $self->request($self->app.'.transaction.commit');
 190         $self->log(E, "error comitting database transaction") unless $stat;
 191         return $stat;
 192 }
 193
 194 # -----------------------------------------------------------------------------
 195 # Rolls back a storage stransaction
 196 # -----------------------------------------------------------------------------
 197 sub xact_rollback {
 198         my $self = shift;
 199         $self->log(I, "rolling back db session");
 200         return $self->request($self->app.".transaction.rollback");
 201 }
 202
 203
 204
 205 # -----------------------------------------------------------------------------
 206 # Rolls back the transaction and disconnects
 207 # -----------------------------------------------------------------------------
 208 sub rollback {
 209         my $self = shift;
 210         $self->xact_rollback if $self->{xact};
 211         $self->disconnect;
 212 }
 213
 214 sub disconnect {
 215         my $self = shift;
 216         $self->session->disconnect if $self->{session};
 217 }
 218
 219
 220 # -----------------------------------------------------------------------------
 221 # commits the db session and destroys the session
 222 # -----------------------------------------------------------------------------
 223 sub commit {
 224         my $self = shift;
 225         return unless $self->{xact};
 226         $self->xact_commit;
 227         $self->session->disconnect;
 228         $self->{session} = undef;
 229 }
 230
 231 # -----------------------------------------------------------------------------
 232 # clears all object data. Does not commit the db transaction.
 233 # -----------------------------------------------------------------------------
 234 sub reset {
 235         my $self = shift;
 236         $self->disconnect;
 237         $$self{$_} = undef for (keys %$self);
 238 }
 239
 240
 241 # -----------------------------------------------------------------------------
 242 # commits and resets
 243 # -----------------------------------------------------------------------------
 244 sub finish {
 245         my $self = shift;
 246         $self->commit;
 247         $self->reset;
 248 }
 249
 250
 251
 252 # -----------------------------------------------------------------------------
 253 # Does a simple storage request
 254 # -----------------------------------------------------------------------------
 255 sub request {
 256         my( $self, $method, @params ) = @_;
 257
 258         my $val;
 259         my $err;
 260         my $argstr = __arg_to_string( (scalar(@params)) == 1 ? $params[0] : \@params);
 261
 262         $self->log(I, "request $method : $argstr");
 263
 264         if( $self->{xact} and
 265                         $self->session->state != OpenSRF::AppSession::CONNECTED() ) {
 266                 $logger->error("CStoreEditor lost it's connection!!");
 267                 #throw OpenSRF::EX::ERROR ("CStoreEditor lost it's connection - cannot continue");
 268         }
 269
 270         try {
 271                 $val = $self->session->request($method, @params)->gather(1);
 272
 273         } catch Error with {
 274                 $err = shift;
 275                 $self->log(E, "request error $method : $argstr : $err");
 276         };
 277
 278         throw $err if $err;
 279         return $val;
 280 }
 281
 282
 283 # -----------------------------------------------------------------------------
 284 # Sets / Returns the requstor object.  This is set when checkauth succeeds.
 285 # -----------------------------------------------------------------------------
 286 sub requestor {
 287         my($self, $requestor) = @_;
 288         $self->{requestor} = $requestor if $requestor;
 289         return $self->{requestor};
 290 }
 291
 292
 293
 294 # -----------------------------------------------------------------------------
 295 # Holds the last data received from a storage call
 296 # -----------------------------------------------------------------------------
 297 sub data {
 298         my( $self, $data ) = @_;
 299         $self->{data} = $data if defined $data;
 300         return $self->{data};
 301 }
 302
 303
 304 # -----------------------------------------------------------------------------
 305 # True if this perm has already been checked at this org
 306 # -----------------------------------------------------------------------------
 307 sub perm_checked {
 308         my( $self, $perm, $org ) = @_;
 309         $self->{checked_perms}->{$org} = {}
 310                 unless $self->{checked_perms}->{$org};
 311         my $checked = $self->{checked_perms}->{$org}->{$perm};
 312         if(!$checked) {
 313                 $self->{checked_perms}->{$org}->{$perm} = 1;
 314                 return 0;
 315         }
 316         return 1;
 317 }
 318
 319
 320
 321 # -----------------------------------------------------------------------------
 322 # Returns true if the requested perm is allowed.  If the perm check fails,
 323 # $e->event is set and undef is returned
 324 # The perm user is $e->requestor->id and perm org defaults to the requestor's
 325 # ws_ou
 326 # If this perm at the given org has already been verified, true is returned
 327 # and the perm is not re-checked
 328 # -----------------------------------------------------------------------------
 329 sub allowed {
 330         my( $self, $perm, $org ) = @_;
 331         my $uid = $self->requestor->id;
 332         $org ||= $self->requestor->ws_ou;
 333         $self->log(I, "checking perms user=$uid, org=$org, perm=$perm");
 334         return 1 if $self->perm_checked($perm, $org);
 335         return $self->checkperm($uid, $org, $perm);
 336 }
 337
 338 sub checkperm {
 339         my($self, $userid, $org, $perm) = @_;
 340         my $s = $U->storagereq(
 341                 "open-ils.storage.permission.user_has_perm", $userid, $perm, $org );
 342
 343         if(!$s) {
 344                 my $e = OpenILS::Event->new('PERM_FAILURE', ilsperm => $perm, ilspermloc => $org);
 345                 $self->event($e);
 346                 return undef;
 347         }
 348
 349         return 1;
 350 }
 351
 352
 353
 354 # -----------------------------------------------------------------------------
 355 # checks the appropriate perm for the operation
 356 # -----------------------------------------------------------------------------
 357 sub _checkperm {
 358         my( $self, $ptype, $action, $org ) = @_;
 359         $org ||= $self->requestor->ws_ou;
 360         my $perm = $PERMS{$ptype}{$action};
 361         if( $perm ) {
 362                 return undef if $self->perm_checked($perm, $org);
 363                 return $self->event unless $self->allowed($perm, $org);
 364         } else {
 365                 $self->log(I, "no perm provided for $ptype.$action");
 366         }
 367         return undef;
 368 }
 369
 370
 371
 372 # -----------------------------------------------------------------------------
 373 # Logs update actions to the activity log
 374 # -----------------------------------------------------------------------------
 375 sub log_activity {
 376         my( $self, $type, $action, $arg ) = @_;
 377         my $str = "$type.$action";
 378         $str .= _prop_string($arg);
 379         $self->log(A, $str);
 380 }
 381
 382
 383
 384 sub _prop_string {
 385         my $obj = shift;
 386         my @props = $obj->properties;
 387         my $str = "";
 388         for(@props) {
 389                 my $prop = $obj->$_() || "";
 390                 $prop = substr($prop, 0, 128) . "..." if length $prop > 131;
 391                 $str .= " $_=$prop";
 392         }
 393         return $str;
 394 }
 395
 396
 397 sub __arg_to_string {
 398         my $arg = shift;
 399         return "" unless defined $arg;
 400         if( UNIVERSAL::isa($arg, "Fieldmapper") ) {
 401                 return (defined $arg->id) ? $arg->id : '<new object>';
 402         }
 403         return JSON->perl2JSON($arg);
 404         return "";
 405 }
 406
 407
 408 # -----------------------------------------------------------------------------
 409 # This does the actual storage query.
 410 #
 411 # 'search' calls become search_where calls and $arg can be a search hash or
 412 # an array-ref of storage search options.
 413 #
 414 # 'retrieve' expects an id
 415 # 'update' expects an object
 416 # 'create' expects an object
 417 # 'delete' expects an object
 418 #
 419 # All methods return true on success and undef on failure.  On failure,
 420 # $e->event is set to the generated event.
 421 # Note: this method assumes that updating a non-changed object and
 422 # thereby receiving a 0 from storage, is a successful update.
 423 #
 424 # The method will therefore return true so the caller can just do
 425 # $e->update_blah($x) or return $e->event;
 426 # The true value returned from storage for all methods will be stored in
 427 # $e->data, until the next method is called.
 428 #
 429 # not-found events are generated on retrieve and serach methods.
 430 # action=search methods will return [] (==true) if no data is found.  If the
 431 # caller is interested in the not found event, they can do:
 432 # return $e->event unless @$results;
 433 # -----------------------------------------------------------------------------
 434 sub runmethod {
 435         my( $self, $action, $type, $arg, $options ) = @_;
 436
 437         if( $action eq 'retrieve' ) {
 438                 if(! defined($arg) ) {
 439                         $self->log(W,"$action $type called with no ID...");
 440                         $self->event(_mk_not_found($type, $arg));
 441                         return undef;
 442                 } elsif( ref($arg) =~ /Fieldmapper/ ) {
 443                         $self->log(E,"$action $type called with an object.. attempting ID retrieval..");
 444                         $arg = $arg->id;
 445                 }
 446         }
 447
 448         my @arg = ( ref($arg) eq 'ARRAY' ) ? @$arg : ($arg);
 449         my $method = $self->app.".direct.$type.$action";
 450
 451         if( $action eq 'search' ) {
 452                 $method = "$method.atomic";
 453
 454         } elsif( $action eq 'batch_retrieve' ) {
 455                 $action = 'search';
 456                 @arg = ( { id => $arg } );
 457                 $method =~ s/batch_retrieve/search/o;
 458                 $method = "$method.atomic";
 459
 460         } elsif( $action eq 'retrieve_all' ) {
 461                 $action = 'search';
 462                 $method =~ s/retrieve_all/search/o;
 463                 my $tt = $type;
 464                 $tt =~ s/\./::/og;
 465                 my $fmobj = "Fieldmapper::$tt";
 466                 @arg = ( { $fmobj->Identity => { '!=' => undef } } );
 467                 $method = "$method.atomic";
 468         }
 469
 470         $method =~ s/search/id_list/o if $options->{idlist};
 471
 472         # remove any stale events
 473         $self->clear_event;
 474
 475         if( $action eq 'update' or $action eq 'delete' or $action eq 'create' ) {
 476                 if(!$self->{xact}) {
 477                         $logger->error("Attempt to update DB while not in a transaction : $method");
 478                         throw OpenSRF::EX::ERROR ("Attempt to update DB while not in a transaction : $method");
 479                 }
 480                 $self->log_activity($type, $action, $arg);
 481         }
 482
 483         if($$options{checkperm}) {
 484                 my $a = ($action eq 'search') ? 'retrieve' : $action;
 485                 my $e = $self->_checkperm($type, $a, $$options{permorg});
 486                 if($e) {
 487                         $self->event($e);
 488                         return undef;
 489                 }
 490         }
 491
 492         my $obj;
 493         my $err;
 494
 495         try {
 496                 $obj = $self->request($method, @arg);
 497         } catch Error with { $err = shift; };
 498
 499
 500         if(!defined $obj) {
 501                 $self->log(I, "request returned no data : $method");
 502
 503                 if( $action eq 'retrieve' ) {
 504                         $self->event(_mk_not_found($type, $arg));
 505
 506                 } elsif( $action eq 'update' or
 507                                 $action eq 'delete' or $action eq 'create' ) {
 508                         my $evt = OpenILS::Event->new(
 509                                 'DATABASE_UPDATE_FAILED', payload => $arg, debug => "$err" );
 510                         $self->event($evt);
 511                 }
 512
 513                 if( $err ) {
 514                         $self->event(
 515                                 OpenILS::Event->new( 'DATABASE_QUERY_FAILED',
 516                                         payload => $arg, debug => "$err" ));
 517                         return undef;
 518                 }
 519
 520                 return undef;
 521         }
 522
 523         if( $action eq 'create' and $obj == 0 ) {
 524                 my $evt = OpenILS::Event->new(
 525                         'DATABASE_UPDATE_FAILED', payload => $arg, debug => "$err" );
 526                 $self->event($evt);
 527                 return undef;
 528         }
 529
 530         # If we havn't dealt with the error in a nice way, go ahead and throw it
 531         if( $err ) {
 532                 $self->event(
 533                         OpenILS::Event->new( 'DATABASE_QUERY_FAILED',
 534                                 payload => $arg, debug => "$err" ));
 535                 return undef;
 536         }
 537
 538         if( $action eq 'search' or $action eq 'batch_retrieve' or $action eq 'retrieve_all') {
 539                 $self->log(I, "$type.$action : returned ".scalar(@$obj). " result(s)");
 540                 $self->event(_mk_not_found($type, $arg)) unless @$obj;
 541         }
 542
 543         $arg->id($obj->id) if $action eq 'create'; # grabs the id on create
 544         $self->data($obj); # cache the data for convenience
 545
 546         return ($obj) ? $obj : 1;
 547 }
 548
 549
 550 sub _mk_not_found {
 551         my( $type, $arg ) = @_;
 552         (my $t = $type) =~ s/\./_/og;
 553         $t = uc($t);
 554         return OpenILS::Event->new("${t}_NOT_FOUND", payload => $arg);
 555 }
 556
 557
 558
 559 # utility method for loading
 560 sub __fm2meth {
 561         my $str = shift;
 562         my $sep = shift;
 563         $str =~ s/Fieldmapper:://o;
 564         $str =~ s/::/$sep/g;
 565         return $str;
 566 }
 567
 568
 569 # -------------------------------------------------------------
 570 # Load up the methods from the FM classes
 571 # -------------------------------------------------------------
 572 my $map = $Fieldmapper::fieldmap;
 573 for my $object (keys %$map) {
 574         my $obj = __fm2meth($object,'_');
 575         my $type = __fm2meth($object, '.');
 576
 577         my $update = "update_$obj";
 578         my $updatef =
 579                 "sub $update {return shift()->runmethod('update', '$type', \@_);}";
 580         eval $updatef;
 581
 582         my $retrieve = "retrieve_$obj";
 583         my $retrievef =
 584                 "sub $retrieve {return shift()->runmethod('retrieve', '$type', \@_);}";
 585         eval $retrievef;
 586
 587         my $search = "search_$obj";
 588         my $searchf =
 589                 "sub $search {return shift()->runmethod('search', '$type', \@_);}";
 590         eval $searchf;
 591
 592         my $create = "create_$obj";
 593         my $createf =
 594                 "sub $create {return shift()->runmethod('create', '$type', \@_);}";
 595         eval $createf;
 596
 597         my $delete = "delete_$obj";
 598         my $deletef =
 599                 "sub $delete {return shift()->runmethod('delete', '$type', \@_);}";
 600         eval $deletef;
 601
 602         my $bretrieve = "batch_retrieve_$obj";
 603         my $bretrievef =
 604                 "sub $bretrieve {return shift()->runmethod('batch_retrieve', '$type', \@_);}";
 605         eval $bretrievef;
 606
 607         my $retrieveall = "retrieve_all_$obj";
 608         my $retrieveallf =
 609                 "sub $retrieveall {return shift()->runmethod('retrieve_all', '$type', \@_);}";
 610         eval $retrieveallf;
 611 }
 612
 613
 614
 615 1;
 616
 617