SprykerEco\Yves\Braintree\Model\QuoteExpander\QuoteExpander

Complexity

Total Complexity

2

Size/Duplication

Total Lines	50
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	expandQuoteWithShipmentMethod()	0	12	1
A	__construct()	0	8	1

<?php

/**
 * MIT License
 * For full license information, please view the LICENSE file that was distributed with this source code.
 */

namespace SprykerEco\Yves\Braintree\Model\QuoteExpander;

use Generated\Shared\Transfer\QuoteTransfer;

The type Generated\Shared\Transfer\QuoteTransfer was not found. Maybe you did not declare it correctly or list all dependencies?

use Spryker\Yves\StepEngine\Dependency\Plugin\Handler\StepHandlerPluginInterface;
use SprykerEco\Yves\Braintree\Dependency\Client\BraintreeToCalculationClientInterface;
use SprykerEco\Yves\Braintree\Dependency\Client\BraintreeToPriceClientInterface;
use SprykerEco\Yves\Braintree\Dependency\Client\BraintreeToQuoteClientInterface;
use Symfony\Component\HttpFoundation\Request;

class QuoteExpander implements QuoteExpanderInterface
{
    /**
     * @var \SprykerEco\Yves\Braintree\Dependency\Client\BraintreeToQuoteClientInterface
     */
    public $quoteClient;

    /**
     * @var \SprykerEco\Yves\Braintree\Dependency\Client\BraintreeToCalculationClientInterface
     */
    public $calculationClient;

    /**
     * @var StepHandlerPluginInterface
     */
    public $shipmentHandlerPlugin;

    /**
     * @param \SprykerEco\Yves\Braintree\Dependency\Client\BraintreeToQuoteClientInterface $quoteClient
     * @param \SprykerEco\Yves\Braintree\Dependency\Client\BraintreeToCalculationClientInterface $calculationClient
     * @param StepHandlerPluginInterface $shipmentHandlerPlugin
     */
    public function __construct(
        BraintreeToQuoteClientInterface $quoteClient,
        BraintreeToCalculationClientInterface $calculationClient,
        StepHandlerPluginInterface $shipmentHandlerPlugin
    ) {
        $this->quoteClient = $quoteClient;
        $this->calculationClient = $calculationClient;
        $this->shipmentHandlerPlugin = $shipmentHandlerPlugin;
    }

    /**
     * @param Request $request
     * @param int $idShipmentMethod
     *
     * @return \Generated\Shared\Transfer\QuoteTransfer
     */
    public function expandQuoteWithShipmentMethod(Request $request, int $idShipmentMethod): QuoteTransfer
    {
        $quoteClient = $this->quoteClient;

        $quoteTransfer = $quoteClient->getQuote();
        $quoteTransfer->getShipment()->setShipmentSelection($idShipmentMethod);
        $quoteTransfer = $this->shipmentHandlerPlugin->addToDataClass($request, $quoteTransfer);
        $quoteTransfer = $this->calculationClient->recalculate($quoteTransfer);

        $quoteClient->setQuote($quoteTransfer);

        return $quoteTransfer;
    }
}