Adding tv episode and tv season to find

Author: wtfzdotnet

lib/Tmdb/Factory/FindFactory.php

@@ -43,9 +43,11 @@ class FindFactory extends AbstractFactory
      */
     public function __construct(HttpClient $httpClient)
     {
-        $this->movieFactory  = new MovieFactory($httpClient);
-        $this->peopleFactory = new PeopleFactory($httpClient);
-        $this->tvFactory     = new TvFactory($httpClient);
+        $this->movieFactory     = new MovieFactory($httpClient);
+        $this->peopleFactory    = new PeopleFactory($httpClient);
+        $this->tvFactory        = new TvFactory($httpClient);
+        $this->tvSeasonFactory  = new TvFactory($httpClient);
+        $this->tvEpisodeFactory = new TvEpisodeFactory($httpClient);
 
         parent::__construct($httpClient);
     }
@@ -69,6 +71,14 @@ public function create(array $data = [])
             $find->setTvResults($this->getTvFactory()->createCollection($data['tv_results']));
         }
 
+        if (array_key_exists('tv_season_results', $data)) {
+            $find->setTvSeasonResults($this->getTvSeasonFactory()->createCollection($data['tv_season_results']));
+        }
+
+        if (array_key_exists('tv_episode_results', $data)) {
+            $find->setTvEpisodeResults($this->getTvSeasonFactory()->createCollection($data['tv_episode_results']));
+        }
+
         return $find;
     }
 
@@ -140,4 +150,42 @@ public function getTvFactory()
     {
         return $this->tvFactory;
     }
+
+    /**
+     * @return TvEpisodeFactory
+     */
+    public function getTvEpisodeFactory()
+    {
+        return $this->tvEpisodeFactory;
+    }
+
+    /**
+     * @param  TvEpisodeFactory $tvEpisodeFactory
+     * @return $this
+     */
+    public function setTvEpisodeFactory($tvEpisodeFactory)
+    {
+        $this->tvEpisodeFactory = $tvEpisodeFactory;
+
+        return $this;
+    }
+
+    /**
+     * @return TvFactory
+     */
+    public function getTvSeasonFactory()
+    {
+        return $this->tvSeasonFactory;
+    }
+
+    /**
+     * @param  TvFactory $tvSeasonFactory
+     * @return $this
+     */
+    public function setTvSeasonFactory($tvSeasonFactory)
+    {
+        $this->tvSeasonFactory = $tvSeasonFactory;
+
+        return $this;
+    }
 }

lib/Tmdb/Model/Find.php

@@ -36,6 +36,16 @@ class Find extends AbstractModel
      */
     private $tvResults;
 
+    /**
+     * @var GenericCollection
+     */
+    private $tvSeasonResults;
+
+    /**
+     * @var GenericCollection
+     */
+    private $tvEpisodeResults;
+
     /**
      * @param  \Tmdb\Model\Common\GenericCollection $movieResults
      * @return $this
@@ -92,4 +102,42 @@ public function getTvResults()
     {
         return $this->tvResults;
     }
+
+    /**
+     * @return GenericCollection
+     */
+    public function getTvSeasonResults()
+    {
+        return $this->tvSeasonResults;
+    }
+
+    /**
+     * @param  GenericCollection $tvSeasonResults
+     * @return $this
+     */
+    public function setTvSeasonResults($tvSeasonResults)
+    {
+        $this->tvSeasonResults = $tvSeasonResults;
+
+        return $this;
+    }
+
+    /**
+     * @return GenericCollection
+     */
+    public function getTvEpisodeResults()
+    {
+        return $this->tvEpisodeResults;
+    }
+
+    /**
+     * @param  GenericCollection $tvEpisodeResults
+     * @return $this
+     */
+    public function setTvEpisodeResults($tvEpisodeResults)
+    {
+        $this->tvEpisodeResults = $tvEpisodeResults;
+
+        return $this;
+    }
 }