<?php

namespace FOS\ElasticaBundle\Persister;

use Elastica\Exception\NotFoundException;
use FOS\ElasticaBundle\Transformer\ModelToElasticaTransformerInterface;
use Elastica\Type;
use Elastica\Document;

/**
 * Inserts, replaces and deletes single documents in an elastica type
 * Accepts domain model objects and converts them to elastica documents
 *
 * @author Thibault Duplessis <thibault.duplessis@gmail.com>
 */
class ObjectPersister implements ObjectPersisterInterface
{
    protected $type;
    protected $transformer;
    protected $objectClass;
    protected $fields;

    public function __construct(Type $type, ModelToElasticaTransformerInterface $transformer, $objectClass, array $fields)
    {
        $this->type            = $type;
        $this->transformer     = $transformer;
        $this->objectClass     = $objectClass;
        $this->fields          = $fields;
    }

    /**
     * Insert one object into the type
     * The object will be transformed to an elastica document
     *
     * @param object $object
     */
    public function insertOne($object)
    {
        $document = $this->transformToElasticaDocument($object);
        $this->type->addDocument($document);
    }

    /**
     * Replaces one object in the type
     *
     * @param object $object
     * @return null
     **/
    public function replaceOne($object)
    {
        $document = $this->transformToElasticaDocument($object);
        try {
            $this->type->deleteById($document->getId());
        } catch (NotFoundException $e) {}
        $this->type->addDocument($document);
    }

    /**
     * Deletes one object in the type
     *
     * @param object $object
     * @return null
     **/
    public function deleteOne($object)
    {
        $document = $this->transformToElasticaDocument($object);
        try {
            $this->type->deleteById($document->getId());
        } catch (NotFoundException $e) {}
    }

    /**
     * Deletes one object in the type by id
     *
     * @param mixed $id
     *
     * @return null
     **/
    public function deleteById($id)
    {
        try {
            $this->type->deleteById($id);
        } catch (NotFoundException $e) {}
    }

    /**
     * Bulk insert an array of objects in the type for the given method
     *
     * @param array $objects array of domain model objects
     * @param string Method to call
     */
    public function insertMany(array $objects)
    {
        $documents = array();
        foreach ($objects as $object) {
            $documents[] = $this->transformToElasticaDocument($object);
        }
        $this->type->addDocuments($documents);
    }

    /**
     * Bulk updates an array of objects in the type
     *
     * @param array $objects array of domain model objects
     */
    public function replaceMany(array $objects)
    {
        $documents = array();
        foreach ($objects as $object) {
            $document = $this->transformToElasticaDocument($object);

            try {
                $this->type->getDocument($document->getId());
            } catch (NotFoundException $e) {
                $this->type->addDocument($document);
            }

            $documents[] = $document;
        }
        $this->type->updateDocuments($documents);
    }

    /**
     * Bulk deletes an array of objects in the type
     *
     * @param array $objects array of domain model objects
     */
    public function deleteMany(array $objects)
    {
        $documents = array();
        foreach ($objects as $object) {
            $documents[] = $this->transformToElasticaDocument($object);
        }
        $this->type->deleteDocuments($documents);
    }

    /**
     * Bulk deletes records from an array of identifiers
     *
     * @param array $identifiers array of domain model object identifiers
     */
    public function deleteManyByIdentifiers(array $identifiers)
    {
        $this->type->getIndex()->getClient()->deleteIds($identifiers, $this->type->getIndex(), $this->type);
    }

    /**
     * Transforms an object to an elastica document
     *
     * @param object $object
     * @return Document the elastica document
     */
    public function transformToElasticaDocument($object)
    {
        return $this->transformer->transform($object, $this->fields);
    }
}