Add option to set thumbnail settings and also get the thumbnail

Author: SebKay

src/PostTypes/Post.php

@@ -10,52 +10,22 @@ class Post
 
     protected $WP_Post;
 
-    /**
-     * @var string
-     */
-    protected $title = '';
+    protected $options = [
+        'thumbnail_size'    => 'post-thumbnail',
+        'thumbnail_classes' => '',
+    ];
 
-    /**
-     * @var string
-     */
+    protected $title = '';
     protected $url = '';
-
-    /**
-     * @var string
-     */
     protected $slug = '';
-
-    /**
-     * @var string
-     */
     protected $content = '';
-
-    /**
-     * @var string
-     */
     protected $status = '';
-
-    /**
-     * @var string
-     */
     protected $format = '';
-
-    /**
-     * @var string
-     */
     protected $excerpt = '';
-
-    /**
-     * @var string
-     */
     protected $date = '';
-
-    /**
-     * @var string
-     */
     protected $dateModified = '';
-
     protected $parent;
+    protected $thumbnail = '';
 
     /**
      * Set up
@@ -71,24 +41,44 @@ public function __construct(int $id)
             return;
         }
 
-        $this->title        = \get_the_title($this->WP_Post);
-        $this->url          = \get_permalink($this->WP_Post);
-        $this->slug         = $this->WP_Post->post_name;
-        $this->content      = $this->WP_Post->post_content;
-        $this->status       = $this->WP_Post->post_status;
-        $this->format       = \get_post_format($this->WP_Post) ?: 'standard';
-        $this->excerpt      = \get_the_excerpt($this->WP_Post);
-        $this->date         = $this->WP_Post->post_date;
-        $this->dateModified = $this->WP_Post->post_modified;
-        $this->parent       = new self($this->WP_Post->post_parent);
+        $this->title        = \get_the_title($this->WPPost());
+        $this->url          = \get_permalink($this->WPPost());
+        $this->slug         = $this->WPPost()->post_name;
+        $this->content      = $this->WPPost()->post_content;
+        $this->status       = $this->WPPost()->post_status;
+        $this->format       = \get_post_format($this->WPPost()) ?: 'standard';
+        $this->excerpt      = \get_the_excerpt($this->WPPost());
+        $this->date         = $this->WPPost()->post_date;
+        $this->dateModified = $this->WPPost()->post_modified;
+        $this->parent       = new self($this->WPPost()->post_parent);
+    }
+
+    public function setOption(string $key, $value)
+    {
+        if (isset($this->options[$key])) {
+            $this->options[$key] = $value;
+        }
+
+        return $this;
+    }
+
+    public function withThumbnail()
+    {
+        $this->thumbnail = \get_the_post_thumbnail(
+            $this->WPPost(),
+            $this->options['thumbnail_size'],
+            $this->options['thumbnail_classes']
+        );
+
+        return $this;
     }
 
     public function id()
     {
         return $this->id;
     }
 
-    public function WP_Post()
+    public function WPPost()
     {
         return $this->WP_Post;
     }