src/applications/phame/query/PhameBlogQuery.php

   1 <?php
   2
   3 final class PhameBlogQuery extends PhabricatorCursorPagedPolicyAwareQuery {
   4
   5   private $ids;
   6   private $phids;
   7   private $domain;
   8   private $statuses;
   9
  10   private $needBloggers;
  11   private $needProfileImage;
  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 withDomain($domain) {
  25     $this->domain = $domain;
  26     return $this;
  27   }
  28
  29   public function withStatuses(array $status) {
  30     $this->statuses = $status;
  31     return $this;
  32   }
  33
  34   public function needProfileImage($need) {
  35     $this->needProfileImage = $need;
  36     return $this;
  37   }
  38
  39   public function needHeaderImage($need) {
  40     $this->needHeaderImage = $need;
  41     return $this;
  42   }
  43
  44   public function newResultObject() {
  45     return new PhameBlog();
  46   }
  47
  48   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
  49     $where = parent::buildWhereClauseParts($conn);
  50
  51     if ($this->statuses !== null) {
  52       $where[] = qsprintf(
  53         $conn,
  54         'b.status IN (%Ls)',
  55         $this->statuses);
  56     }
  57
  58     if ($this->ids !== null) {
  59       $where[] = qsprintf(
  60         $conn,
  61         'b.id IN (%Ls)',
  62         $this->ids);
  63     }
  64
  65     if ($this->phids !== null) {
  66       $where[] = qsprintf(
  67         $conn,
  68         'b.phid IN (%Ls)',
  69         $this->phids);
  70     }
  71
  72     if ($this->domain !== null) {
  73       $where[] = qsprintf(
  74         $conn,
  75         'b.domain = %s',
  76         $this->domain);
  77     }
  78
  79     return $where;
  80   }
  81
  82   protected function didFilterPage(array $blogs) {
  83     if ($this->needProfileImage) {
  84       $default = null;
  85
  86       $file_phids = mpull($blogs, 'getProfileImagePHID');
  87       $file_phids = array_filter($file_phids);
  88       if ($file_phids) {
  89         $files = id(new PhabricatorFileQuery())
  90           ->setParentQuery($this)
  91           ->setViewer($this->getViewer())
  92           ->withPHIDs($file_phids)
  93           ->execute();
  94         $files = mpull($files, null, 'getPHID');
  95       } else {
  96         $files = array();
  97       }
  98
  99       foreach ($blogs as $blog) {
 100         $file = idx($files, $blog->getProfileImagePHID());
 101         if (!$file) {
 102           if (!$default) {
 103             $default = PhabricatorFile::loadBuiltin(
 104               $this->getViewer(),
 105               'blog.png');
 106           }
 107           $file = $default;
 108         }
 109         $blog->attachProfileImageFile($file);
 110       }
 111     }
 112
 113     if ($this->needHeaderImage) {
 114       $file_phids = mpull($blogs, 'getHeaderImagePHID');
 115       $file_phids = array_filter($file_phids);
 116       if ($file_phids) {
 117         $files = id(new PhabricatorFileQuery())
 118           ->setParentQuery($this)
 119           ->setViewer($this->getViewer())
 120           ->withPHIDs($file_phids)
 121           ->execute();
 122         $files = mpull($files, null, 'getPHID');
 123       } else {
 124         $files = array();
 125       }
 126
 127       foreach ($blogs as $blog) {
 128         $file = idx($files, $blog->getHeaderImagePHID());
 129         if ($file) {
 130           $blog->attachHeaderImageFile($file);
 131         }
 132       }
 133     }
 134     return $blogs;
 135   }
 136
 137   public function getQueryApplicationClass() {
 138     // TODO: Can we set this without breaking public blogs?
 139     return null;
 140   }
 141
 142   protected function getPrimaryTableAlias() {
 143     return 'b';
 144   }
 145
 146 }