<?php

namespace FOQ\ElasticaBundle\Paginator;

use Elastica_Searchable;
use Elastica_Query;
use Elastica_ResultSet;
use FOQ\ElasticaBundle\Paginator\PaginatorAdapterInterface;
use FOQ\ElasticaBundle\Paginator\RawPartialResults;
use FOQ\ElasticaBundle\Paginator\PartialResultsInterface;

/**
 * Allows pagination of Elastica_Query. Does not map results
 */
class RawPaginatorAdapter implements PaginatorAdapterInterface
{
    /**
     * @var Elastica_Searchable the object to search in
     */
    private $searchable = null;

    /**
     * @var Elastica_Query the query to search
     */
    private $query = null;

    /**
     * @see PaginatorAdapterInterface::__construct
     *
     * @param Elastica_Searchable $searchable the object to search in
     * @param Elastica_Query $query the query to search
     */
    public function __construct(Elastica_Searchable $searchable, Elastica_Query $query)
    {
        $this->searchable = $searchable;
        $this->query      = $query;
    }

    /**
     * Returns the paginated results.
     *
     * @param $offset
     * @param $itemCountPerPage
     * @return Elastica_ResultSet
     */
    protected function getElasticaResults($offset, $itemCountPerPage)
    {
        $query = clone $this->query;
        $query->setFrom($offset);
        $query->setLimit($itemCountPerPage);

        return $this->searchable->search($query);
    }

    /**
     * Returns the paginated results.
     *
     * @param int $offset
     * @param int $itemCountPerPage
     * @return PartialResultsInterface
     */
    public function getResults($offset, $itemCountPerPage)
    {
        return new RawPartialResults($this->getElasticaResults($offset, $itemCountPerPage));
    }

    /**
     * Returns the number of results.
     *
     * @return integer The number of results.
     */
    public function getTotalHits()
    {
        return $this->searchable->search($this->query)->getTotalHits();
    }
}