Get cast from dom

Author: hmerritt

src/Dom.php

@@ -33,7 +33,7 @@ public function fetch(string $url, array $options)
      * @param object $dom
      * @param string $selection
      *
-     * @return array
+     * @return array|object
      */
     public function find(object $dom, string $selection)
     {

src/HtmlPieces.php

@@ -62,14 +62,81 @@ public function get(object $page, string $element)
                 return $this->strClean($poster);
                 break;
 
+            case "trailer":
+                $trailerLink = $page->find('.slate a[data-video]');
+                $trailerId = $this->count($trailerLink) ? $trailerLink->getAttribute("data-video") : "";
+                $trailerLink = $this->count($trailerId) ? "https://www.imdb.com/videoplayer/".$trailerId : "";
+                return [
+                    "id" => $trailerId,
+                    "link" => $trailerLink
+                ];
+                break;
+
+            case "cast":
+                $cast = [];
+                $findAllCast = $dom->find($page, 'table.cast_list tr');
+                foreach ($findAllCast as $castRow)
+                {
+                    if (count($castRow->find('.primary_photo')) === 0) {
+                        continue;
+                    }
+                    $actor = [];
+
+                    $characterLink = $castRow->find('.character a');
+                    $actor["character"] = count($characterLink) ? $characterLink->text : $dom->find($castRow, '.character')->text;
+
+                    $actorRow = $castRow->find('td')[1];
+                    $actorLink = $actorRow->find('a');
+                    if (count($actorLink) > 0)
+                    {
+                        // Set actor name to text within link
+                        $actor["actor"] = $actorLink->text;
+                        $actor["actor_id"] = $this->extractImdbId($actorLink->href);
+                    } else
+                    {
+                        // No link found
+                        // Set actor name to whatever is there
+                        $actor["actor"] = $actorRow->text;
+                    }
+
+                    $actor["character"] = $this->strClean($actor["character"]);
+                    $actor["actor"]     = $this->strClean($actor["actor"]);
+                    $actor["actor_id"]  = $this->strClean($actor["actor_id"]);
+
+                    array_push($cast, $actor);
+                }
+                return $cast;
+                break;
+
             default:
                 return "";
         }
     }
 
+    /**
+     * Extract an imdb-id from a string '/ttxxxxxxx/'
+     * Returns string of id or empty string if none found
+     *
+     * @param string $str
+     * @return string
+     */
+    private function extractImdbId($str)
+    {
+        // Search string for 2 letters followed by numbers
+        // '/yyxxxxxxx'
+        preg_match('/\/[A-Za-z]{2}[0-9]+/', $str, $imdbIds);
+        $id = substr($imdbIds[0], 1);
+        if ($id == NULL)
+        {
+            $id = "";
+        }
+        return $id;
+    }
+
     /**
      * Cleans-up string
      * -> removes white-space and html entitys
+     *    turns null into empty string
      *
      * @param $string
      * @return string
@@ -79,4 +146,14 @@ private function strClean($string)
         return empty($string) ? "" : str_replace(chr(194).chr(160), '', html_entity_decode(trim($string)));
     }
 
+    /**
+     * Count (either array items or string length)
+     *
+     * @param array|string $item
+     * @return string
+     */
+    private function count($item) {
+        return (is_array($item) ? count($item) : strlen($item));
+    }
+
 }

src/imdb.php

@@ -45,11 +45,11 @@ private function populateOptions(array $options = []): array
     /**
      * Gets film data from IMDB. Will first search if the
      * film name is passed instead of film-id
-     * @param string $film
+     * @param string $filmId
      * @param array  $options
      * @return array $response
      */
-    public function film(string $film, array $options = []): array
+    public function film(string $filmId, array $options = []): array
     {
         //  Combine user options with default ones
         $options = $this->populateOptions($options);
@@ -67,10 +67,10 @@ public function film(string $film, array $options = []): array
         $htmlPieces = new HtmlPieces;
 
         //  Load imdb page and parse the dom
-        $page = $dom->fetch($this->baseUrl."title/".$film, $options);
+        $page = $dom->fetch($this->baseUrl."title/".$filmId, $options);
 
         //  Add all film data to response $store
-        $response->add("id", $film);
+        $response->add("id", $filmId);
         $response->add("title", $htmlPieces->get($page, "title"));
         $response->add("year", $htmlPieces->get($page, "year"));
         $response->add("length", $htmlPieces->get($page, "length"));