LizardMedia\ProductAttachment\Model\Attachment\UpdateHandler

Complexity

Total Complexity

7

Size/Duplication

Total Lines	45
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
B	execute()	0	24	6
A	__construct()	0	4	1

<?php

declare(strict_types = 1);

/**
 * File: UpdateHandler.php
 *
 * @author Bartosz Kubicki bartosz.kubicki@lizardmedia.pl>
 * @copyright Copyright (C) 2018 Lizard Media (http://lizardmedia.pl)
 */

namespace LizardMedia\ProductAttachment\Model\Attachment;

use LizardMedia\ProductAttachment\Api\AttachmentRepositoryInterface;
use LizardMedia\ProductAttachment\Api\Data\AttachmentInterface;
use Magento\Catalog\Api\Data\ProductInterface;
use Magento\Framework\EntityManager\Operation\ExtensionInterface;

/**
 * Class UpdateHandler
 * @package LizardMedia\ProductAttachment\Model\Attachment
 */
class UpdateHandler implements ExtensionInterface
{
    /**
     * @var AttachmentRepositoryInterface
     */
    private $attachmentRepository;

    /**
     * @param AttachmentRepositoryInterface $attachmentRepository
     */
    public function __construct(AttachmentRepositoryInterface $attachmentRepository)
    {
        $this->attachmentRepository = $attachmentRepository;
    }

    /**
     * @param object $entity
     * @param array $arguments
     * @return ProductInterface|object $entity
     */
    public function execute($entity, $arguments = [])
    {
        $attachments = $entity->getExtensionAttributes()->getProductAttachments() ?: [];
        /** @var $attachments AttachmentInterface[] */

        $updatedAttachments = [];
        $oldAttachments = $this->attachmentRepository->getAttachmentsByProduct($entity);

        foreach ($attachments as $attachment) {
            if ($attachment->getId()) {
                $updatedAttachments[(int) $attachment->getId()] = true;
            }
            $this->attachmentRepository->save($entity->getSku(), $attachment, !(bool) $entity->getStoreId());
        }

        /** @var ProductInterface $entity */
        foreach ($oldAttachments as $attachment) {
            if (!isset($updatedAttachments[(int) $attachment->getId()])) {
                $this->attachmentRepository->delete((int) $attachment->getId());
            }
        }

        return $entity;
    }
}