wpsy/Entity/PostCollection.php line 65

Open in your IDE?
  1. <?php
  3. namespace Metabolism\WordpressBundle\Entity;
  5. use ArrayIterator;
  6. use Metabolism\WordpressBundle\Factory\PostFactory;
  7. use Metabolism\WordpressBundle\Service\PaginationService;
  9. /**
  10.  * Class Metabolism\WordpressBundle Framework
  11.  */
  12. class PostCollection implements \IteratorAggregate, \Countable {
  14.     public $query;
  15.     protected $items;
  17.     protected $pagination;
  19.     /**
  20.      * @param $query
  21.      */
  22.     public function __construct($query)
  23.     {
  24.         $this->query = new \WP_Query( $query );
  26.         $items = [];
  28.         foreach ($this->query->posts as $post)
  29.             $items[] = PostFactory::create( $post );
  31.         $this->items = array_filter($items);
  32.     }
  35.     /**
  36.      * @param $args
  37.      * @return array
  38.      */
  39.     public function getPagination($args=[]){
  41.         if( is_null($this->pagination) ){
  43.             $paginationService = new PaginationService();
  44.             $this->pagination = $paginationService->build($args);
  45.         }
  47.         return $this->pagination;
  48.     }
  51.     /**
  52.      * @return ArrayIterator
  53.      */
  54.     public function getIterator() {
  56.         return new ArrayIterator($this->items);
  57.     }
  60.     /**
  61.      * Get total post count
  62.      *
  63.      * @return int
  64.      */
  65.     public function count()
  66.     {
  67.         return $this->query->found_posts;
  68.     }
  69. }