enhancement of attachment model function "getThumbnailUrl"

Now you can call: ```php $attachment->getThumbnailUrl('thumbnail'); ```
kayue · Jan 4, 2014 · 72ca5bf · 72ca5bf · benglass · Jun 20, 2014
1 parent a765258
commit 72ca5bf
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 58 deletions.
diff --git a/Model/Attachment.php b/Model/Attachment.php
@@ -5,44 +5,34 @@
 class Attachment extends Post implements AttachmentInterface
 {
     protected $post;
-    protected $url;
     protected $metadata;
+    protected $url;
     protected $thumbnailUrl;
-    protected $mimeType;
 
     public function __construct(Post $post)
     {
-        $this->post = $post;
-    }
-
-    public function setMetadata($metadata)
-    {
-        $this->metadata = $metadata;
+        $this->metadata = $post->getMetas()
+            ->filter(function (PostMeta $meta) {
+                return '_wp_attachment_metadata' == $meta->getKey();
+            })
+            ->first()
+        ;
     }
 
     public function getMetadata()
     {
         return $this->metadata;
     }
 
-    public function setMimeType($mimeType)
-    {
-        $this->mimeType = $mimeType;
-    }
-
-    public function getMimeType()
+    public function getThumbnailUrl($size = 'post-thumbnail')
     {
-        return $this->mimeType;
-    }
+        $rawMetadata = $this->metadata->getValue();
 
-    public function setThumbnailUrl($thumbnail)
-    {
-        $this->thumbnailUrl = $thumbnail;
-    }
+        if (isset($rawMetadata['sizes'][$size])) {
+            return dirname($rawMetadata['file']) . '/' . $rawMetadata['sizes'][$size]['file'];
+        }
 
-    public function getThumbnailUrl($size = null)
-    {
-        return $this->thumbnailUrl;
+        return null;
     }
 
     public function setUrl($url)

diff --git a/Model/AttachmentManager.php b/Model/AttachmentManager.php
@@ -49,23 +49,7 @@ public function findAttachmentsByPost(Post $post)
         $result = array();
         /** @var $post Post */
         foreach ($posts as $post) {
-            /** @var $meta PostMeta */
-            $meta = $this->postMetaManager->findOneMetaBy(array(
-                'post' => $post,
-                'key'  => '_wp_attachment_metadata'
-            ));
-
-            if ($meta) {
-                $rawMeta = $meta->getValue();
-                $attachment = new Attachment($post);
-
-                $attachment->setUrl($rawMeta['file']);
-                if (isset($rawMeta['sizes']['thumbnail'])) {
-                    $attachment->setThumbnailUrl(substr($rawMeta['file'], 0, strrpos($rawMeta['file'], '/') + 1) . $rawMeta['sizes']['thumbnail']['file']);
-                }
-
-                $result[] = $attachment;
-            }
+            $result[] = new Attachment($post);
         }
 
         return $result;
@@ -83,24 +67,7 @@ public function findOneAttachmentById($id)
             'type'   => 'attachment',
         ));
 
-        /** @var $meta PostMeta */
-        $meta = $this->postMetaManager->findOneMetaBy(array(
-            'post' => $post,
-            'key'  => '_wp_attachment_metadata'
-        ));
-
-        if ($meta) {
-            $rawMeta = $meta->getValue();
-            $attachment = new Attachment($post);
-
-            $attachment->setUrl($rawMeta['file']);
-            if (isset($rawMeta['sizes']['thumbnail'])) {
-                $attachment->setThumbnailUrl(substr($rawMeta['file'], 0, strrpos($rawMeta['file'], '/') + 1) . $rawMeta['sizes']['thumbnail']['file']);
-            }
-            return $attachment;
-        }
-
-        return null;
+        return new Attachment($post);
     }
 
     public function getAttachmentOfSize(Attachment $attachment, $size = null)