Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\media;
Chris@0: 
Chris@0: use Drupal\Core\Access\AccessResult;
Chris@0: use Drupal\Core\Entity\EntityAccessControlHandler;
Chris@0: use Drupal\Core\Entity\EntityInterface;
Chris@0: use Drupal\Core\Session\AccountInterface;
Chris@0: 
Chris@0: /**
Chris@0:  * Defines an access control handler for the media entity.
Chris@0:  */
Chris@0: class MediaAccessControlHandler extends EntityAccessControlHandler {
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function checkAccess(EntityInterface $entity, $operation, AccountInterface $account) {
Chris@0:     if ($account->hasPermission('administer media')) {
Chris@0:       return AccessResult::allowed()->cachePerPermissions();
Chris@0:     }
Chris@0: 
Chris@0:     $is_owner = ($account->id() && $account->id() === $entity->getOwnerId());
Chris@0:     switch ($operation) {
Chris@0:       case 'view':
Chris@0:         $access_result = AccessResult::allowedIf($account->hasPermission('view media') && $entity->isPublished())
Chris@0:           ->cachePerPermissions()
Chris@0:           ->addCacheableDependency($entity);
Chris@0:         if (!$access_result->isAllowed()) {
Chris@0:           $access_result->setReason("The 'view media' permission is required and the media item must be published.");
Chris@0:         }
Chris@0:         return $access_result;
Chris@0: 
Chris@0:       case 'update':
Chris@0:         if ($account->hasPermission('update any media')) {
Chris@0:           return AccessResult::allowed()->cachePerPermissions();
Chris@0:         }
Chris@0:         return AccessResult::allowedIf($account->hasPermission('update media') && $is_owner)
Chris@0:           ->cachePerPermissions()
Chris@0:           ->cachePerUser()
Chris@0:           ->addCacheableDependency($entity);
Chris@0: 
Chris@0:       case 'delete':
Chris@0:         if ($account->hasPermission('delete any media')) {
Chris@0:           return AccessResult::allowed()->cachePerPermissions();
Chris@0:         }
Chris@0:         return AccessResult::allowedIf($account->hasPermission('delete media') && $is_owner)
Chris@0:           ->cachePerPermissions()
Chris@0:           ->cachePerUser()
Chris@0:           ->addCacheableDependency($entity);
Chris@0: 
Chris@0:       default:
Chris@0:         return AccessResult::neutral()->cachePerPermissions();
Chris@0:     }
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function checkCreateAccess(AccountInterface $account, array $context, $entity_bundle = NULL) {
Chris@0:     return AccessResult::allowedIfHasPermissions($account, ['administer media', 'create media'], 'OR');
Chris@0:   }
Chris@0: 
Chris@0: }