CustomerGroupRetriever.php 1.16 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
namespace Magento\Sales\Model;

use Magento\Backend\Model\Session\Quote;
use Magento\Customer\Api\GroupManagementInterface;

/**
 * Class for getting customer group from quote session for adminhtml area.
 */
class CustomerGroupRetriever implements \Magento\Customer\Model\Group\RetrieverInterface
{
    /**
     * @var Quote
     */
    private $quoteSession;

    /**
     * @var GroupManagementInterface
     */
    private $groupManagement;

    /**
     * @param Quote $quoteSession
     * @param GroupManagementInterface $groupManagement
     */
    public function __construct(Quote $quoteSession, GroupManagementInterface $groupManagement)
    {
        $this->quoteSession = $quoteSession;
        $this->groupManagement = $groupManagement;
    }

    /**
     * @inheritdoc
     */
    public function getCustomerGroupId()
    {
        if ($this->quoteSession->getQuoteId() && $this->quoteSession->getQuote()) {
            return $this->quoteSession->getQuote()->getCustomerGroupId();
        }
        return $this->groupManagement->getNotLoggedInGroup()->getId();
    }
}