src/applications/people/query/PhabricatorPeopleUserEmailQuery.php

   1 <?php
   2
   3 final class PhabricatorPeopleUserEmailQuery
   4   extends PhabricatorCursorPagedPolicyAwareQuery {
   5
   6   private $ids;
   7   private $phids;
   8
   9   public function withIDs(array $ids) {
  10     $this->ids = $ids;
  11     return $this;
  12   }
  13
  14   public function withPHIDs(array $phids) {
  15     $this->phids = $phids;
  16     return $this;
  17   }
  18
  19   public function newResultObject() {
  20     return new PhabricatorUserEmail();
  21   }
  22
  23   protected function loadPage() {
  24     return $this->loadStandardPage($this->newResultObject());
  25   }
  26
  27   protected function getPrimaryTableAlias() {
  28     return 'email';
  29   }
  30
  31   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
  32     $where = parent::buildWhereClauseParts($conn);
  33
  34     if ($this->ids !== null) {
  35       $where[] = qsprintf(
  36         $conn,
  37         'email.id IN (%Ld)',
  38         $this->ids);
  39     }
  40
  41     if ($this->phids !== null) {
  42       $where[] = qsprintf(
  43         $conn,
  44         'email.phid IN (%Ls)',
  45         $this->phids);
  46     }
  47
  48     return $where;
  49   }
  50
  51   protected function willLoadPage(array $page) {
  52
  53     $user_phids = mpull($page, 'getUserPHID');
  54
  55     $users = id(new PhabricatorPeopleQuery())
  56       ->setViewer($this->getViewer())
  57       ->setParentQuery($this)
  58       ->withPHIDs($user_phids)
  59       ->execute();
  60     $users = mpull($users, null, 'getPHID');
  61
  62     foreach ($page as $key => $address) {
  63       $user = idx($users, $address->getUserPHID());
  64
  65       if (!$user) {
  66         unset($page[$key]);
  67         $this->didRejectResult($address);
  68         continue;
  69       }
  70
  71       $address->attachUser($user);
  72     }
  73
  74     return $page;
  75   }
  76
  77   public function getQueryApplicationClass() {
  78     return 'PhabricatorPeopleApplication';
  79   }
  80
  81 }