Collection.php

Go to the documentation of this file.

<?php
namespace Magento\Store\Model\ResourceModel\Group;

class Collection extends \Magento\Framework\Model\ResourceModel\Db\Collection\AbstractCollection
{
    protected function _construct()
    {
        $this->setFlag('load_default_store_group', false);
        $this->_init(\Magento\Store\Model\Group::class, \Magento\Store\Model\ResourceModel\Group::class);
    }

    public function setLoadDefault($loadDefault)
    {
        return $this->setFlag('load_default_store_group', (bool)$loadDefault);
    }

    public function getLoadDefault()
    {
        return $this->getFlag('load_default_store_group');
    }

    public function setWithoutDefaultFilter()
    {
        return $this->addFieldToFilter('main_table.group_id', ['gt' => 0]);
    }

    public function setWithoutStoreViewFilter()
    {
        return $this->addFieldToFilter('main_table.default_store_id', ['gt' => 0]);
    }

    public function setWithoutAssignedCategoryFilter()
    {
        return $this->addFieldToFilter('main_table.root_category_id', ['eq' => 0])
            ->addFieldToFilter('main_table.group_id', ['neq' => 0]);
    }

    public function _beforeLoad()
    {
        if (!$this->getLoadDefault()) {
            $this->setWithoutDefaultFilter();
        }
        $this->addOrder('main_table.name', self::SORT_ORDER_ASC);
        return parent::_beforeLoad();
    }

    public function toOptionArray()
    {
        return $this->_toOptionArray('group_id', 'name');
    }

    public function addWebsiteFilter($website)
    {
        return $this->addFieldToFilter('main_table.website_id', ['in' => $website]);
    }
}