src/applications/phame/query/PhamePostQuery.php

   1 <?php
   2
   3 final class PhamePostQuery extends PhabricatorCursorPagedPolicyAwareQuery {
   4
   5   private $ids;
   6   private $blogPHIDs;
   7   private $bloggerPHIDs;
   8   private $visibility;
   9   private $publishedAfter;
  10   private $phids;
  11
  12   private $needHeaderImage;
  13
  14   public function withIDs(array $ids) {
  15     $this->ids = $ids;
  16     return $this;
  17   }
  18
  19   public function withPHIDs(array $phids) {
  20     $this->phids = $phids;
  21     return $this;
  22   }
  23
  24   public function withBloggerPHIDs(array $blogger_phids) {
  25     $this->bloggerPHIDs = $blogger_phids;
  26     return $this;
  27   }
  28
  29   public function withBlogPHIDs(array $blog_phids) {
  30     $this->blogPHIDs = $blog_phids;
  31     return $this;
  32   }
  33
  34   public function withVisibility(array $visibility) {
  35     $this->visibility = $visibility;
  36     return $this;
  37   }
  38
  39   public function withPublishedAfter($time) {
  40     $this->publishedAfter = $time;
  41     return $this;
  42   }
  43
  44   public function needHeaderImage($need) {
  45     $this->needHeaderImage = $need;
  46     return $this;
  47   }
  48
  49   public function newResultObject() {
  50     return new PhamePost();
  51   }
  52
  53   protected function loadPage() {
  54     return $this->loadStandardPage($this->newResultObject());
  55   }
  56
  57   protected function willFilterPage(array $posts) {
  58     // We require blogs to do visibility checks, so load them unconditionally.
  59     $blog_phids = mpull($posts, 'getBlogPHID');
  60
  61     $blogs = id(new PhameBlogQuery())
  62       ->setViewer($this->getViewer())
  63       ->needProfileImage(true)
  64       ->withPHIDs($blog_phids)
  65       ->execute();
  66
  67     $blogs = mpull($blogs, null, 'getPHID');
  68     foreach ($posts as $key => $post) {
  69       $blog_phid = $post->getBlogPHID();
  70
  71       $blog = idx($blogs, $blog_phid);
  72       if (!$blog) {
  73         $this->didRejectResult($post);
  74         unset($posts[$key]);
  75         continue;
  76       }
  77
  78       $post->attachBlog($blog);
  79     }
  80
  81     if ($this->needHeaderImage) {
  82       $file_phids = mpull($posts, 'getHeaderImagePHID');
  83       $file_phids = array_filter($file_phids);
  84       if ($file_phids) {
  85         $files = id(new PhabricatorFileQuery())
  86           ->setParentQuery($this)
  87           ->setViewer($this->getViewer())
  88           ->withPHIDs($file_phids)
  89           ->execute();
  90         $files = mpull($files, null, 'getPHID');
  91       } else {
  92         $files = array();
  93       }
  94
  95       foreach ($posts as $post) {
  96         $file = idx($files, $post->getHeaderImagePHID());
  97         if ($file) {
  98           $post->attachHeaderImageFile($file);
  99         }
 100       }
 101     }
 102
 103     return $posts;
 104   }
 105
 106   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
 107     $where = parent::buildWhereClauseParts($conn);
 108
 109     if ($this->ids !== null) {
 110       $where[] = qsprintf(
 111         $conn,
 112         'p.id IN (%Ld)',
 113         $this->ids);
 114     }
 115
 116     if ($this->phids !== null) {
 117       $where[] = qsprintf(
 118         $conn,
 119         'p.phid IN (%Ls)',
 120         $this->phids);
 121     }
 122
 123     if ($this->bloggerPHIDs !== null) {
 124       $where[] = qsprintf(
 125         $conn,
 126         'p.bloggerPHID IN (%Ls)',
 127         $this->bloggerPHIDs);
 128     }
 129
 130     if ($this->visibility !== null) {
 131       $where[] = qsprintf(
 132         $conn,
 133         'p.visibility IN (%Ld)',
 134         $this->visibility);
 135     }
 136
 137     if ($this->publishedAfter !== null) {
 138       $where[] = qsprintf(
 139         $conn,
 140         'p.datePublished > %d',
 141         $this->publishedAfter);
 142     }
 143
 144     if ($this->blogPHIDs !== null) {
 145       $where[] = qsprintf(
 146         $conn,
 147         'p.blogPHID in (%Ls)',
 148         $this->blogPHIDs);
 149     }
 150
 151     return $where;
 152   }
 153
 154   public function getBuiltinOrders() {
 155     return array(
 156       'datePublished' => array(
 157         'vector' => array('datePublished', 'id'),
 158         'name' => pht('Publish Date'),
 159       ),
 160     ) + parent::getBuiltinOrders();
 161   }
 162
 163   public function getOrderableColumns() {
 164     return parent::getOrderableColumns() + array(
 165       'datePublished' => array(
 166         'table' => $this->getPrimaryTableAlias(),
 167         'column' => 'datePublished',
 168         'type' => 'int',
 169         'reverse' => false,
 170       ),
 171     );
 172   }
 173
 174   protected function newPagingMapFromPartialObject($object) {
 175     return array(
 176       'id' => (int)$object->getID(),
 177       'datePublished' => (int)$object->getDatePublished(),
 178     );
 179   }
 180
 181   public function getQueryApplicationClass() {
 182     // TODO: Does setting this break public blogs?
 183     return null;
 184   }
 185
 186   protected function getPrimaryTableAlias() {
 187     return 'p';
 188   }
 189
 190 }