Bug 23354: (follow-up) Update for bugs 23049 and 23805

[koha-ffzg.git] / Koha / Virtualshelf.pm

diff --git a/Koha/Virtualshelf.pm b/Koha/Virtualshelf.pm
index 7edacaa..3992830 100644 (file)
--- a/Koha/Virtualshelf.pm
+++ b/Koha/Virtualshelf.pm
@@ -19,12 +19,16 @@ use Modern::Perl;
 
 use Carp;
 
+use C4::Auth;
+
+use Koha::Patrons;
 use Koha::Database;
 use Koha::DateUtils qw( dt_from_string );
 use Koha::Exceptions;
 use Koha::Virtualshelfshare;
 use Koha::Virtualshelfshares;
 use Koha::Virtualshelfcontent;
+use Koha::Virtualshelfcontents;
 
 use base qw(Koha::Object);
 
@@ -48,22 +52,35 @@ our $PUBLIC = 2;
 sub store {
     my ( $self ) = @_;
 
+    unless ( $self->owner ) {
+        Koha::Exceptions::Virtualshelves::UseDbAdminAccount->throw;
+    }
+
     unless ( $self->is_shelfname_valid ) {
         Koha::Exceptions::Virtualshelves::DuplicateObject->throw;
     }
 
-    $self->allow_add( 0 )
-        unless defined $self->allow_add;
-    $self->allow_delete_own( 1 )
-        unless defined $self->allow_delete_own;
-    $self->allow_delete_other( 0 )
-        unless defined $self->allow_delete_other;
+    $self->allow_change_from_owner( 1 )
+        unless defined $self->allow_change_from_owner;
+    $self->allow_change_from_others( 0 )
+        unless defined $self->allow_change_from_others;
 
-    $self->created_on( dt_from_string );
+    $self->created_on( dt_from_string )
+        unless defined $self->created_on;
 
     return $self->SUPER::store( $self );
 }
 
+sub is_public {
+    my ( $self ) = @_;
+    return $self->category == $PUBLIC;
+}
+
+sub is_private {
+    my ( $self ) = @_;
+    return $self->category == $PRIVATE;
+}
+
 sub is_shelfname_valid {
     my ( $self ) = @_;
 
@@ -72,14 +89,14 @@ sub is_shelfname_valid {
         ( $self->shelfnumber ? ( "me.shelfnumber" => { '!=', $self->shelfnumber } ) : () ),
     };
 
-    if ( $self->category == $PRIVATE and defined $self->owner ) {
+    if ( $self->is_private and defined $self->owner ) {
         $conditions->{-or} = {
             "virtualshelfshares.borrowernumber" => $self->owner,
             "me.owner" => $self->owner,
         };
         $conditions->{category} = $PRIVATE;
     }
-    elsif ( $self->category == $PRIVATE and not defined $self->owner ) {
+    elsif ( $self->is_private and not defined $self->owner ) {
         $conditions->{owner} = undef;
         $conditions->{category} = $PRIVATE;
     }
@@ -98,12 +115,16 @@ sub is_shelfname_valid {
 
 sub get_shares {
     my ( $self ) = @_;
-    return $self->{_result}->virtualshelfshares;
+    my $rs = $self->{_result}->virtualshelfshares;
+    my $shares = Koha::Virtualshelfshares->_new_from_dbic( $rs );
+    return $shares;
 }
 
 sub get_contents {
     my ( $self ) = @_;
-    return $self->{_result}->virtualshelfcontents;
+    my $rs = $self->{_result}->virtualshelfcontents;
+    my $contents = Koha::Virtualshelfcontents->_new_from_dbic( $rs );
+    return $contents;
 }
 
 sub share {
@@ -129,6 +150,16 @@ sub is_shared {
     )->count;
 }
 
+sub is_shared_with {
+    my ( $self, $borrowernumber ) = @_;
+    return unless $borrowernumber;
+    return  $self->get_shares->search(
+        {
+            borrowernumber => $borrowernumber,
+        }
+    )->count;
+}
+
 sub remove_share {
     my ( $self, $borrowernumber ) = @_;
     my $shelves = Koha::Virtualshelfshares->search(
@@ -152,6 +183,10 @@ sub add_biblio {
         }
     )->count;
     return if $already_exists;
+
+    # Check permissions
+    return unless ( $self->owner == $borrowernumber && $self->allow_change_from_owner ) || $self->allow_change_from_others;
+
     my $content = Koha::Virtualshelfcontent->new(
         {
             shelfnumber => $self->shelfnumber,
@@ -172,36 +207,63 @@ sub remove_biblios {
     return unless @$biblionumbers;
 
     my $number_removed = 0;
-    for my $biblionumber ( @$biblionumbers ) {
-        if ( $self->owner == $borrowernumber or $self->allow_delete_own ) {
-            $number_removed += $self->get_contents->search(
-                {
-                    biblionumber => $biblionumber,
-                    borrowernumber => $borrowernumber,
-                }
-            )->delete;
-        }
-        if ( $self->allow_delete_other ) {
-            $number_removed += $self->get_contents->search(
-                {
-                    biblionumber => $biblionumber,
-                    # FIXME
-                    # This does not make sense, but it's has been backported from DelFromShelf.
-                    # Why shouldn't we allow to remove his own contribution if allow_delete_other is on?
-                    borrowernumber => {
-                        -or => {
-                            '!=' => $borrowernumber,
-                            '=' => undef
-                        }
-                    },
-                }
-            )->delete;
-        }
+    if( ( $self->owner == $borrowernumber && $self->allow_change_from_owner )
+      || $self->allow_change_from_others ) {
+        $number_removed += $self->get_contents->search({
+            biblionumber => $biblionumbers,
+        })->delete;
     }
     return $number_removed;
 }
 
-sub type {
+sub can_be_viewed {
+    my ( $self, $borrowernumber ) = @_;
+    return 1 if $self->is_public;
+    return 0 unless $borrowernumber;
+    return 1 if $self->owner == $borrowernumber;
+    return $self->get_shares->search(
+        {
+            borrowernumber => $borrowernumber,
+        }
+    )->count;
+}
+
+sub can_be_deleted {
+    my ( $self, $borrowernumber ) = @_;
+
+    return 0 unless $borrowernumber;
+    return 1 if $self->owner == $borrowernumber;
+
+    my $patron = Koha::Patrons->find( $borrowernumber );
+
+    return 1 if $self->is_public and C4::Auth::haspermission( $patron->userid, { lists => 'delete_public_lists' } );
+
+    return 0;
+}
+
+sub can_be_managed {
+    my ( $self, $borrowernumber ) = @_;
+    return 1
+      if $borrowernumber and $self->owner == $borrowernumber;
+    return 0;
+}
+
+sub can_biblios_be_added {
+    my ( $self, $borrowernumber ) = @_;
+
+    return 1
+      if $borrowernumber
+      and ( ( $self->owner == $borrowernumber && $self->allow_change_from_owner ) or $self->allow_change_from_others );
+    return 0;
+}
+
+sub can_biblios_be_removed {
+    my ( $self, $borrowernumber ) = @_;
+    return $self->can_biblios_be_added( $borrowernumber );
+    # Same answer since bug 18228
+}
+
+sub _type {
     return 'Virtualshelve';
 }