Jh\DataImportMagento\Writer\ShipmentWriter

Complexity

Total Complexity

10

Size/Duplication

Total Lines	104
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	8

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	11	1
A	setOptions()	0	4	1
C	writeItem()	0	45	8

<?php
namespace Jh\DataImportMagento\Writer;

use Ddeboer\DataImport\Exception\WriterException;
use Ddeboer\DataImport\Writer\AbstractWriter;
use Jh\DataImportMagento\Exception\MagentoSaveException;
use Jh\DataImportMagento\Options\OptionsParseTrait;

/**
 * Class ShipmentWriter
 * @package Jh\DataImportMagento\Writer
 * @author Aydin Hassan <aydin@hotmail.co.uk>
 */
class ShipmentWriter extends AbstractWriter
{
    use OptionsParseTrait;

    /**
     * @var \Mage_Core_Model_Resource_Transaction
     */
    protected $transactionResourceModel;

    /**
     * @var \Mage_Sales_Model_Order
     */
    protected $orderModel;

    /**
     * @var \Mage_Sales_Model_Order_Shipment_Track
     */
    protected $trackingModel;

    /**
     * @var array
     */
    protected $options = [
        'send_shipment_email' => false
    ];

    /**
     * @param \Mage_Sales_Model_Order $order
     * @param \Mage_Core_Model_Resource_Transaction $transactionResourceModel
     * @param \Mage_Sales_Model_Order_Shipment_Track $trackingModel
     * @param array $options
     */
    public function __construct(
        \Mage_Sales_Model_Order $order,
        \Mage_Core_Model_Resource_Transaction $transactionResourceModel,
        \Mage_Sales_Model_Order_Shipment_Track $trackingModel,
        array $options
    ) {
        $this->orderModel               = $order;
        $this->transactionResourceModel = $transactionResourceModel;
        $this->trackingModel            = $trackingModel;
        $this->setOptions($options);
    }

    /**
     * @param array $options
     */
    public function setOptions(array $options)
    {
        $this->options = $this->parseOptions($this->options, $options);
    }

    /**
     * @param array $item
     * @return \Ddeboer\DataImport\Writer\WriterInterface|void
     * @throws \Ddeboer\DataImport\Exception\WriterException
     * @throws \Jh\DataImportMagento\Exception\MagentoSaveException
     */
    public function writeItem(array $item)
    {
        if (!isset($item['order_id'])) {
            throw new WriterException('order_id must be set');
        }
        $order = clone $this->orderModel;
        $order->loadByIncrementId($item['order_id']);
        if (!$order->getId()) {
            throw new WriterException(sprintf('Order with ID: "%s" cannot be found', $item['order_id']));
        }

        try {
            $shipment = $order->prepareShipment();
            $shipment->register();
            $shipment->getOrder()->setData('is_in_process', true);

            $transactionSave = clone $this->transactionResourceModel;
            $transactionSave
                ->addObject($shipment)
                ->addObject($shipment->getOrder())
                ->save();

            if (array_key_exists('tracks', $item) && is_array($item['tracks'])) {
                foreach ($item['tracks'] as $currentTrack) {
                    $tracking = clone $this->trackingModel;
                    $tracking->setShipment($shipment);
                    $tracking->setData(
                        [
                            'title'         => $currentTrack['carrier'],
                            'number'        => $currentTrack['tracking_number'],
                            'carrier_code'  => 'custom',
                            'order_id'      => $order->getId()
                        ]
                    );
                    $tracking->save();
                }
            }
        } catch (\Exception $e) {
            throw new MagentoSaveException($e->getMessage());
        }

        if ($this->options['send_shipment_email']) {
            $shipment->sendEmail(true);
        }
    }
}