Bundle.php

Go to the documentation of this file.

<?php
namespace Magento\Bundle\Model\ResourceModel;

class Bundle extends \Magento\Framework\Model\ResourceModel\Db\AbstractDb
{
    protected $_productRelation;

    protected $quoteResource;

    public function __construct(
        \Magento\Framework\Model\ResourceModel\Db\Context $context,
        \Magento\Catalog\Model\ResourceModel\Product\Relation $productRelation,
        \Magento\Quote\Model\ResourceModel\Quote $quoteResource,
        $connectionName = null
    ) {
        parent::__construct($context, $connectionName);
        $this->_productRelation = $productRelation;
        $this->quoteResource = $quoteResource;
    }

    protected function _construct()
    {
        $this->_init('catalog_product_entity', 'entity_id');
    }

    protected function _getSelect($productId, $columns = [])
    {
        return $this->getConnection()->select()->from(
            ["bo" => $this->getTable('catalog_product_bundle_option')],
            ['type', 'option_id']
        )->where(
            "bo.parent_id = ?",
            $productId
        )->where(
            "bo.required = 1"
        )->joinLeft(
            ["bs" => $this->getTable('catalog_product_bundle_selection')],
            "bs.option_id = bo.option_id AND bs.parent_product_id = bo.parent_id",
            $columns
        );
    }

    public function getSelectionsData($productId)
    {
        return $this->getConnection()->fetchAll($this->_getSelect($productId, ["*"]));
    }

    public function dropAllQuoteChildItems($productId)
    {
        $connection = $this->quoteResource->getConnection();
        $select = $connection->select();
        $quoteItemIds = $connection->fetchCol(
            $select->from(
                $this->getTable('quote_item'),
                ['item_id']
            )->where(
                'product_id = :product_id'
            ),
            ['product_id' => $productId]
        );

        if ($quoteItemIds) {
            $connection->delete(
                $this->getTable('quote_item'),
                ['parent_item_id IN(?)' => $quoteItemIds]
            );
        }
    }

    public function dropAllUnneededSelections($productId, $ids)
    {
        $where = ['parent_product_id = ?' => $productId];
        if (!empty($ids)) {
            $where['selection_id NOT IN (?) '] = $ids;
        }
        $this->getConnection()->delete($this->getTable('catalog_product_bundle_selection'), $where);
    }

    public function saveProductRelations($parentId, $childIds)
    {
        $this->_productRelation->processRelations($parentId, $childIds);

        return $this;
    }

    public function addProductRelation($parentId, $childId)
    {
        $this->_productRelation->addRelation($parentId, $childId);
        return $this;
    }

    public function addProductRelations($parentId, $childIds)
    {
        $this->_productRelation->addRelations($parentId, $childIds);
        return $this;
    }

    public function removeProductRelations($parentId, $childIds)
    {
        $this->_productRelation->removeRelations($parentId, $childIds);
        return $this;
    }
}