Collection.php

Go to the documentation of this file.

<?php
namespace Magento\CheckoutAgreements\Model\ResourceModel\Agreement;

class Collection extends \Magento\Framework\Model\ResourceModel\Db\Collection\AbstractCollection
{
    protected $_map = [
        'fields' => [
            'agreement_id' => 'main_table.agreement_id'
        ]
    ];

    protected $_isStoreFilterWithAdmin = true;

    protected function _construct()
    {
        $this->_init(
            \Magento\CheckoutAgreements\Model\Agreement::class,
            \Magento\CheckoutAgreements\Model\ResourceModel\Agreement::class
        );
    }

    public function addStoreFilter($store)
    {
        // check and prepare data
        if ($store instanceof \Magento\Store\Model\Store) {
            $store = [$store->getId()];
        } elseif (is_numeric($store)) {
            $store = [$store];
        }

        $alias = 'store_table_' . implode('_', $store);
        if ($this->getFlag($alias)) {
            return $this;
        }

        $storeFilter = [$store];
        if ($this->_isStoreFilterWithAdmin) {
            $storeFilter[] = 0;
        }

        // add filter
        $this->getSelect()->join(
            [$alias => $this->getTable('checkout_agreement_store')],
            'main_table.agreement_id = ' . $alias . '.agreement_id',
            []
        )->where(
            $alias . '.store_id IN (?)',
            $storeFilter
        )->group(
            'main_table.agreement_id'
        );

        $this->setFlag($alias, true);
        return $this;
    }

    public function setIsStoreFilterWithAdmin($value)
    {
        $this->_isStoreFilterWithAdmin = (bool)$value;
        return $this;
    }
}