103 lines
2.4 KiB
PHP
103 lines
2.4 KiB
PHP
<?php
|
|
|
|
namespace SilverStripe\Security;
|
|
|
|
use Psr\Log\InvalidArgumentException;
|
|
use SilverStripe\Core\Flushable;
|
|
use SilverStripe\ORM\DataExtension;
|
|
use SilverStripe\ORM\DataObject;
|
|
use SilverStripe\Core\Cache\MemberCacheFlusher;
|
|
|
|
class InheritedPermissionFlusher extends DataExtension implements Flushable
|
|
{
|
|
/**
|
|
* @var MemberCacheFlusher[]
|
|
*/
|
|
protected $services = [];
|
|
|
|
/**
|
|
* Flush all MemberCacheFlusher services
|
|
*/
|
|
public static function flush()
|
|
{
|
|
singleton(__CLASS__)->flushCache();
|
|
}
|
|
|
|
/**
|
|
* @param DataObject $owner
|
|
*/
|
|
public function setOwner($owner)
|
|
{
|
|
if (!$owner instanceof Member && !$owner instanceof Group) {
|
|
throw new InvalidArgumentException(sprintf(
|
|
'%s can only be applied to %s or %s',
|
|
__CLASS__,
|
|
Member::class,
|
|
Group::class
|
|
));
|
|
}
|
|
|
|
parent::setOwner($owner);
|
|
}
|
|
|
|
/**
|
|
* @param MemberCacheFlusher[] $services
|
|
* @throws InvalidArgumentException
|
|
* @return $this
|
|
*/
|
|
public function setServices($services)
|
|
{
|
|
foreach ($services as $service) {
|
|
if (!$service instanceof MemberCacheFlusher) {
|
|
throw new InvalidArgumentException(sprintf(
|
|
'%s.services must contain only %s instances. %s provided.',
|
|
__CLASS__,
|
|
MemberCacheFlusher::class,
|
|
get_class($service)
|
|
));
|
|
}
|
|
}
|
|
|
|
$this->services = $services;
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* @return MemberCacheFlusher[]
|
|
*/
|
|
public function getServices()
|
|
{
|
|
return $this->services;
|
|
}
|
|
|
|
/**
|
|
* Flushes all registered MemberCacheFlusher services
|
|
*/
|
|
public function flushCache()
|
|
{
|
|
$ids = $this->getMemberIDList();
|
|
foreach ($this->getServices() as $service) {
|
|
$service->flushMemberCache($ids);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get a list of member IDs that need their permissions flushed
|
|
*
|
|
* @return array|null
|
|
*/
|
|
protected function getMemberIDList()
|
|
{
|
|
if (!$this->owner || !$this->owner->exists()) {
|
|
return null;
|
|
}
|
|
|
|
if ($this->owner instanceof Group) {
|
|
return $this->owner->Members()->column('ID');
|
|
}
|
|
|
|
return [$this->owner->ID];
|
|
}
|
|
}
|