Phraseanet/lib/Alchemy/Phrasea/Databox/Subdef/DbalMediaSubdefDataRepository.php

<?php
/*
 * This file is part of Phraseanet
 *
 * (c) 2005-2016 Alchemy
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Alchemy\Phrasea\Databox\Subdef;

use Doctrine\DBAL\Connection;
use Doctrine\DBAL\DBALException;
use Doctrine\DBAL\Exception\DriverException;

class DbalMediaSubdefDataRepository implements MediaSubdefDataRepository
{
    /**
     * @var Connection
     */
    private $connection;

    public function __construct(Connection $connection)
    {
        $this->connection = $connection;
    }

    /**
     * @param int[] $recordIds
     * @param string[]|null $names
     * @return array[]
     */
    public function findByRecordIdsAndNames(array $recordIds, array $names = null)
    {
        if (!$recordIds || (null !== $names && !$names)) {
            return [];
        }

        $sql = $this->getSelectSQL() . ' WHERE record_id IN (:recordIds)';
        $params = ['recordIds' => $recordIds];
        $types = ['recordIds' => Connection::PARAM_INT_ARRAY];

        if ($names) {
            $sql .= ' AND name IN (:names)';
            $params['names'] = $names;
            $types['names'] = Connection::PARAM_STR_ARRAY;
        }

        return array_map([$this, 'sqlToPhp'], $this->connection->fetchAll($sql, $params, $types));
    }

    /**
     * @param array[] $subdefIds
     * @return int The number of affected rows
     * @throws \Exception
     */
    public function delete(array $subdefIds)
    {
        if (!$subdefIds) {
            return 0;
        }

        $statement = $this->connection->prepare('DELETE FROM subdef WHERE record_id = :record_id AND name = :name');

        $this->connection->beginTransaction();

        try {
            $deleted = array_reduce($subdefIds, function ($carry, $data) use ($statement) {
                $carry += $statement->execute([
                    'record_id' => $data['record_id'],
                    'name' => $data['name'],
                ]);

                return $carry;
            }, 0);

            $this->connection->commit();
        } catch (\Exception $exception) {
            $this->connection->rollBack();

            throw $exception;
        }

        return $deleted;
    }

    /**
     * @param array $data
     * @throws \Exception
     */
    public function save(array $data)
    {
        $this->connection->transactional(function () use ($data) {
            $partitions = $this->partitionInsertAndUpdate($data);

            $updateNeeded = $this->createMissing($partitions['insert']);

            $this->updatePresent($partitions['update'] + $updateNeeded);
        });
    }

    /**
     * @param array $data
     * @return array
     */
    private function partitionInsertAndUpdate(array $data)
    {
        $partitions = [
            'insert' => [],
            'update' => [],
        ];

        foreach ($data as $index => $item) {
            $partitions[isset($item['subdef_id']) ? 'update' : 'insert'][$index] = $item;
        }

        return $partitions;
    }

    /**
     * @param array $toInsert
     * @return array
     * @throws DBALException
     */
    private function createMissing(array $toInsert)
    {
        if (!$toInsert) {
            return [];
        }

        $statement = $this->connection->prepare($this->getInsertSql());

        $updateNeeded = [];

        foreach ($toInsert as $index => $data) {
            try {
                $statement->execute($this->phpToSql($data));
            } catch (DriverException $exception) {
                $updateNeeded[$index] = $data;
            }
        }

        return $updateNeeded;
    }

    /**
     * @param array $toUpdate
     * @throws DBALException
     */
    private function updatePresent(array $toUpdate)
    {
        if (!$toUpdate) {
            return;
        }

        $statement = $this->connection->prepare($this->getUpdateSql());

        foreach ($toUpdate as $data) {
            $statement->execute($this->phpToSql($data));
        }
    }

    private function getSelectSQL()
    {
        return <<<'SQL'
SELECT subdef_id, record_id, name, path, file, width, height, mime, size, substit, etag, created_on, updated_on
FROM subdef
SQL;
    }

    /**
     * @param array $data
     * @return array
     */
    private function phpToSql(array $data)
    {
        return [
            'record_id' => $data['record_id'],
            'name' => $data['name'],
            'path' => $data['path'],
            'file' => $data['file'],
            'width' => $data['width'],
            'height' => $data['height'],
            'mime' => $data['mime'],
            'size' => $data['size'],
            'substit' => $data['is_substituted'],
            'etag' => $data['etag'],
        ];
    }

    private function sqlToPhp(array $data)
    {
        return [
            'subdef_id' => (int)$data['subdef_id'],
            'record_id' => (int)$data['record_id'],
            'name' => $data['name'],
            'path' => $data['path'],
            'file' => $data['file'],
            'width' => (int)$data['width'],
            'height' => (int)$data['height'],
            'mime' => $data['mime'],
            'size' => (int)$data['size'],
            'is_substituted' => (bool)$data['substit'],
            'etag' => $data['etag'],
            'created_on' => $data['created_on'],
            'updated_on' => $data['updated_on'],
            'physically_present' => true,
        ];
    }

    /**
     * @return string
     */
    private function getInsertSql()
    {
        static $sql;

        if (null !== $sql) {
            return $sql;
        }

        $values = [
            'record_id' => ':record_id',
            'name' => ':name',
            'path' => ':path',
            'file' => ':file',
            'width' => ':width',
            'height' => ':height',
            'mime' => ':mime',
            'size' => ':size',
            'substit' => ':substit',
            'etag' => ':etag',
            'created_on' => 'NOW()',
            'updated_on' => 'NOW()',
            'dispatched' => '1',
        ];

        $sql = sprintf(
            'INSERT INTO subdef (%s) VALUES (%s)',
            implode(', ', array_keys($values)),
            implode(', ', array_values($values))
        );

        return $sql;
    }

    /**
     * @return string
     */
    private function getUpdateSql()
    {
        static $sql;

        if (null !== $sql) {
            return $sql;
        }

        $values = [
            'path = :path',
            'file = :file',
            'width = :width',
            'height = :height',
            'mime = :mime',
            'size = :size',
            'substit = :substit',
            'etag = :etag',
            'updated_on = NOW()',
        ];

        $where = [
            'record_id = :record_id',
            'name = :name',
        ];

        $sql = sprintf('UPDATE subdef SET %s WHERE %s', implode(', ', $values), implode(' AND ', $where));

        return $sql;
    }
}