Add seasons and episodes

Author: steph

src/HtmlPieces.php

@@ -19,7 +19,7 @@ class HtmlPieces
      * @param string $element
      * @return string
      */
-    public function get(object $page, string $element)
+    public function get(object $page, string $element, string $url='')
     {
         //  Initiate dom object
         //  -> handles page scraping
@@ -32,7 +32,7 @@ public function get(object $page, string $element)
 
                 return $this->strClean($title);
                 break;
-                
+
             case "genre":
                 $allGenres = $dom->find($page, "div[data-testid=genres] a");
                 $genres = [];
@@ -193,14 +193,14 @@ public function get(object $page, string $element)
                         if ($this->count($castRow->find('img')) === 0) {
                             continue;
                         }
-    
+
                         $actor = [];
                         $actor["actor"] = "";
                         $actor["avatar"] = "";
                         $actor["avatar_hq"] = "";
                         $actor["actor_id"] = "";
                         $actor["character"] = "";
-    
+
                         // Actor
                         $actorLink = $castRow->find('a[data-testid=title-cast-item__actor]');
                         if ($this->count($actorLink)) {
@@ -217,7 +217,7 @@ public function get(object $page, string $element)
                                 $actor["avatar_hq"] = preg_match('/\.\_/', $actor["avatar_hq"]) ? preg_split('/\.\_.*/', $actor["avatar_hq"])[0] . ".jpg" : $actor["avatar_hq"];
                             }
                         }
-    
+
                         // Actor ID
                         $link = $castRow->find('a');
                         if ($this->count($link)) {
@@ -227,24 +227,61 @@ public function get(object $page, string $element)
                                 $actor["actor_id"] = $matches[0];
                             }
                         }
-    
+
                         // Character
                         $characterLink = $castRow->find('[data-testid=cast-item-characters-link] span');
                         if ($this->count($characterLink)) {
                             $actor["character"] = $characterLink->text;
                         }
-    
+
                         $actor["character"] = $this->strClean($actor["character"]);
                         $actor["actor"]     = $this->strClean($actor["actor"]);
                         $actor["avatar"]    = $this->strClean($actor["avatar"]);
                         $actor["actor_id"]  = $this->strClean($actor["actor_id"]);
-    
+
                         array_push($cast, $actor);
                     }
                 }
                 return $cast;
                 break;
 
+            case "seasons":
+                $seasons = [];
+                $findAllSeasons = $dom->find($page, "#bySeason > option");
+                $dom = new \PHPHtmlParser\Dom();
+                foreach ($findAllSeasons as $seasonRow){
+                    $season = [];
+                    $seasonValue = $seasonRow->getAttribute('value');
+                    $season['season'] = $seasonValue;
+                    // Using imdb ajax api to get episodes
+                    $season['episodes'] = $this->get($dom->loadFromUrl($url."/_ajax?season=".$seasonValue), "episodes");
+                    array_push($seasons, $season);
+                }
+                return $seasons;
+                break;
+
+            case "episodes":
+                $episodes = [];
+                $findAllEpisodes = $dom->find($page, ".eplist > .list_item");
+                foreach ($findAllEpisodes as $episodeRow){
+                    $episode = [];
+                    $hyperlink = $episodeRow->find("a[itemprop=url]");
+                    $episode["id"]          = $this->extractImdbId($hyperlink->getAttribute("href"));
+                    $episode['title']       = $episodeRow->find('a[itemprop=name]')->text;
+                    $episode['description'] = $episodeRow->find(".item_description")->text;
+                    $rating                 = $episodeRow->find(".ipl-rating-star__rating");
+                    if($this->count($rating)) {
+                        $episode['rating'] = $rating->text;
+                    }
+                    $image = $hyperlink->find("img");
+                    if($this->count($image)) {
+                        $episode["poster"] = $image->getAttribute("src");
+                    }
+                    array_push($episodes, $episode);
+                }
+                return $episodes;
+                break;
+
             case "technical_specs":
                 $technical_specs = [];
                 $table = $dom->find($page, '.dataTable tr');
@@ -315,7 +352,7 @@ public function get(object $page, string $element)
      *
      * @param  object $page
      * @param  array  $patterns
-     * @return string 
+     * @return string
      */
     public function findMatchInPatterns(object $dom, object $page, array $patterns, string $type = "text")
     {

src/Imdb.php

@@ -25,6 +25,7 @@ private function populateOptions(array $options = []): array
             'category'     => 'all',
             'curlHeaders'  => ['Accept-Language: en-US,en;q=0.5'],
             'techSpecs'    => true,
+            'seasons'      => true,
         ];
 
         //  Merge any user options with the default ones
@@ -115,6 +116,16 @@ public function film(string $filmId, array $options = []): array
             $response->add("technical_specs",  []);
         }
 
+        // If seasons is enabled
+        if ($options['seasons']) {
+            $url = "https://www.imdb.com/title/$filmId/episodes";
+            $page_seasons = $dom->fetch($url, $options);
+            // If film has episodes or seasons
+            if (count($page_seasons->find(".error_code_404")) == 0) {
+                $response->add("seasons", $htmlPieces->get($page_seasons, "seasons", $url));
+            }
+        }
+
         //  If caching is enabled
         if ($options["cache"]) {
             //  Add result to the cache

src/Response.php

@@ -80,6 +80,7 @@ public function default(string $endpoint): array
                         "id" => "",
                         "link" => ""
                     ],
+                    "seasons" => [],
                     "cast" => [],
                     "technical_specs" => []
                 ];