<?php
/**
 * This file is part of the Tmdb PHP API created by Michael Roterman.
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 *
 * @package Tmdb
 * @author Michael Roterman <michael@wtfz.net>
 * @copyright (c) 2013, Michael Roterman
 * @version 0.0.1
 */
namespace Tmdb\Repository;

use Tmdb\Client;
use Tmdb\Factory\CollectionFactory;
use Tmdb\Factory\ImageFactory;
use Tmdb\Model\Collection as ApiCollection;
use Tmdb\Model\Collection\QueryParameter\AppendToResponse;

/**
 * Class CollectionRepository
 * @package Tmdb\Repository
 * @see http://docs.themoviedb.apiary.io/#collections
 */
class CollectionRepository extends AbstractRepository
{
    private $imageFactory;

    public function __construct(Client $client)
    {
        parent::__construct($client);

        $this->imageFactory = new ImageFactory();
    }

    /**
     * Load a collection with the given identifier
     *
     * If you want to optimize the result set/bandwidth you
     * should define the AppendToResponse parameter
     *
     * @param $id
     * @param $parameters
     * @param $headers
     * @return ApiCollection
     */
    public function load($id, array $parameters = array(), array $headers = array())
    {
        if (empty($parameters)) {
            $parameters = array(
                new AppendToResponse(array(
                    AppendToResponse::IMAGES,
                ))
            );
        }

        $data = $this->getApi()->getCollection($id, $this->parseQueryParameters($parameters), $headers);

        return $this->getFactory()->create($data);
    }

    /**
     * Get all of the images for a particular collection by collection id.
     *
     * @param $id
     * @param $parameters
     * @param $headers
     * @return null|\Tmdb\Model\AbstractModel
     */
    public function getImages($id, array $parameters = array(), array $headers = array())
    {
        $data  = $this->getApi()->getImages($id, $this->parseQueryParameters($parameters), $headers);
        $movie = $this->getFactory()->create(array('images' => $data));

        return $movie->getImages();
    }

    /**
     * Return the Collection API Class
     *
     * @return \Tmdb\Api\Collections
     */
    public function getApi()
    {
        return $this->getClient()->getCollectionsApi();
    }

    /**
     * @return CollectionFactory
     */
    public function getFactory()
    {
        return new CollectionFactory();
    }

    /**
     * @param  mixed $imageFactory
     * @return $this
     */
    public function setImageFactory($imageFactory)
    {
        $this->imageFactory = $imageFactory;

        return $this;
    }

    /**
     * @return mixed
     */
    public function getImageFactory()
    {
        return $this->imageFactory;
    }
}