<?php

namespace FOQ\ElasticaBundle\Paginator;

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

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

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

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

    /**
     * Returns the paginated results.
     *
     * @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.
     *
     * @return FOQ\ElasticaBundle\Paginator\PartialResultInterface
     */
    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->count($this->query);
    }
}