<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\MediaGallery\Model\ResourceModel\Keyword;

use Magento\Framework\App\ResourceConnection;
use Magento\Framework\DB\Adapter\AdapterInterface;
use Magento\Framework\DB\Adapter\Pdo\Mysql;
use Magento\Framework\Exception\CouldNotSaveException;
use Magento\MediaGalleryApi\Api\Data\KeywordInterface;
use Magento\MediaGalleryApi\Api\SaveAssetsKeywordsInterface;
use Psr\Log\LoggerInterface;

/**
 * Save keywords of media assets
 */
class SaveAssetsKeywords implements SaveAssetsKeywordsInterface
{
    private const TABLE_KEYWORD = 'media_gallery_keyword';
    private const ID = 'id';
    private const KEYWORD = 'keyword';

    /**
     * @var ResourceConnection
     */
    private $resourceConnection;

    /**
     * @var SaveAssetLinks
     */
    private $saveAssetLinks;

    /**
     * @var LoggerInterface
     */
    private $logger;

    /**
     * SaveAssetKeywords constructor.
     *
     * @param ResourceConnection $resourceConnection
     * @param SaveAssetLinks $saveAssetLinks
     * @param LoggerInterface $logger
     */
    public function __construct(
        ResourceConnection $resourceConnection,
        SaveAssetLinks $saveAssetLinks,
        LoggerInterface $logger
    ) {
        $this->resourceConnection = $resourceConnection;
        $this->saveAssetLinks = $saveAssetLinks;
        $this->logger = $logger;
    }

    /**
     * @inheritdoc
     */
    public function execute(array $assetKeywords): void
    {
        $failedAssetIds = [];
        foreach ($assetKeywords as $assetKeyword) {
            try {
                $this->saveAssetKeywords($assetKeyword->getKeywords(), $assetKeyword->getAssetId());
            } catch (\Exception $exception) {
                $this->logger->critical($exception);
                $failedAssetIds[] = $assetKeyword->getAssetId();
            }
        }

        if (!empty($failedAssetIds)) {
            throw new CouldNotSaveException(
                __('Could not save keywords for asset ids: %ids', ['ids' => implode(' ,', $failedAssetIds)])
            );
        }
    }

    /**
     * Save asset keywords.
     *
     * @param KeywordInterface[] $keywords
     * @param int $assetId
     * @throws CouldNotSaveException
     * @throws \Zend_Db_Exception
     */
    private function saveAssetKeywords(array $keywords, int $assetId): void
    {
        $data = [];
        foreach ($keywords as $keyword) {
            $data[] = $keyword->getKeyword();
        }

        if (!empty($data)) {
            /** @var Mysql $connection */
            $connection = $this->resourceConnection->getConnection();
            $connection->insertArray(
                $this->resourceConnection->getTableName(self::TABLE_KEYWORD),
                [self::KEYWORD],
                $data,
                AdapterInterface::INSERT_IGNORE
            );
        }

        $this->saveAssetLinks->execute($assetId, $this->getKeywordIds($data));
    }

    /**
     * Select keywords by names
     *
     * @param string[] $keywords
     * @return int[]
     */
    private function getKeywordIds(array $keywords): array
    {
        $connection = $this->resourceConnection->getConnection();
        $select = $connection->select()
            ->from(['k' => $this->resourceConnection->getTableName(self::TABLE_KEYWORD)])
            ->columns(self::ID)
            ->where('k.' . self::KEYWORD . ' in (?)', $keywords);

        return $connection->fetchCol($select);
    }
}