lekoala / silverstripe-email-templates

src/Extensions/EmailTemplateSubsiteExtension.php

<?php

namespace LeKoala\EmailTemplates\Extensions;

use SilverStripe\ORM\DataQuery;
use SilverStripe\ORM\DataExtension;
use SilverStripe\Security\Permission;
use LeKoala\Base\Subsite\SubsiteHelper;

The type LeKoala\Base\Subsite\SubsiteHelper was not found. Maybe you did not declare it correctly or list all dependencies?

use SilverStripe\ORM\Queries\SQLSelect;

/**
 * Add subsites support
 *
 * @author lekoala
 */
class EmailSubsiteExtension extends DataExtension
{

    private static $has_one = array(

The private property $has_one is not used, and could be removed.

        'Subsite' => 'Subsite',
    );

    public function isMainDataObject()
    {
        if ($this->owner->SubsiteID == 0) {

It seems like you are loosely comparing $this->owner->SubsiteID of type mixed|null to 0; this is ambiguous as not only 0 == 0 is true, but null == 0 is true, too. Consider using a strict comparison ===.

            return true;
        }
        return false;
    }

    /**
     * Update any requests to limit the results to the current site
     */
    public function augmentSQL(SQLSelect $query, DataQuery $dataQuery = null)
    {
        if (SubsiteHelper::subsiteFilterDisabled()) {
            return;
        }
        if ($dataQuery && $dataQuery->getQueryParam('Subsite.filter') === false) {

The condition $dataQuery->getQueryParam('Subsite.filter') === false is always false.

            return;
        }

        // If you're querying by ID, don't filter
        if ($query->filtersOnID()) {
            return;
        }

        // Don't run on delete queries, since they are always tied to a specific ID.
        // if ($query->getDelete()) {
        //     return;
        // }

        // If we match on a subsite, don't filter twice
        $regexp = '/^(.*\.)?("|`)?SubsiteID("|`)?\s?=/';
        foreach ($query->getWhereParameterised($parameters) as $predicate) {
            if (preg_match($regexp, $predicate)) {
                return;
            }
        }

        $subsiteID = (int) SubsiteHelper::currentSubsiteID();

        $froms = $query->getFrom();
        $froms = array_keys($froms);
        $tableName = array_shift($froms);

        if ($subsiteID) {
            $query->addWhere("\"$tableName\".\"SubsiteID\" IN ($subsiteID)");
        }
    }

    public function onBeforeWrite()
    {
        parent::onBeforeWrite();

        // Assign to current subsite when created
        if (!$this->owner->ID && !$this->owner->SubsiteID) {
            $this->owner->SubsiteID = SubsiteHelper::currentSubsiteID();
        }
    }

    public function canView($member = null)
    {
        return $this->canEdit($member);
    }

    /**
     * @param Member
     * @return boolean|null
     */
    public function canEdit($member = null)
    {
        return Permission::check('CMS_ACCESS', 'any', $member);
    }

    /**
     * @param Member
     * @return boolean|null
     */
    public function canCreate($member = null)
    {
        return Permission::check('CMS_ACCESS', 'any', $member);
    }

    /**
     * @param Member
     * @return boolean|null
     */
    public function canDelete($member = null)
    {
        return Permission::check('CMS_ACCESS', 'any', $member);
    }

    /**
     * Return a piece of text to keep DataObject cache keys appropriately specific
     */
    public function cacheKeyComponent()
    {
        return 'subsite-' . SubsiteHelper::currentSubsiteID();
    }
}