src/applications/badges/query/PhabricatorBadgesAwardQuery.php

   1 <?php
   2
   3 final class PhabricatorBadgesAwardQuery
   4   extends PhabricatorCursorPagedPolicyAwareQuery {
   5
   6   private $badgePHIDs;
   7   private $recipientPHIDs;
   8   private $awarderPHIDs;
   9   private $badgeStatuses = null;
  10
  11   protected function willFilterPage(array $awards) {
  12     $badge_phids = array();
  13     foreach ($awards as $key => $award) {
  14       $badge_phids[] = $award->getBadgePHID();
  15     }
  16
  17     $badges = id(new PhabricatorBadgesQuery())
  18       ->setViewer($this->getViewer())
  19       ->withPHIDs($badge_phids)
  20       ->execute();
  21
  22     $badges = mpull($badges, null, 'getPHID');
  23     foreach ($awards as $key => $award) {
  24       $award_badge = idx($badges, $award->getBadgePHID());
  25       if (!$award_badge) {
  26         unset($awards[$key]);
  27         $this->didRejectResult($award);
  28         continue;
  29       }
  30       $award->attachBadge($award_badge);
  31     }
  32
  33     return $awards;
  34   }
  35
  36   public function withBadgePHIDs(array $phids) {
  37     $this->badgePHIDs = $phids;
  38     return $this;
  39   }
  40
  41   public function withRecipientPHIDs(array $phids) {
  42     $this->recipientPHIDs = $phids;
  43     return $this;
  44   }
  45
  46   public function withAwarderPHIDs(array $phids) {
  47     $this->awarderPHIDs = $phids;
  48     return $this;
  49   }
  50
  51   public function withBadgeStatuses(array $statuses) {
  52     $this->badgeStatuses = $statuses;
  53     return $this;
  54   }
  55
  56   private function shouldJoinBadge() {
  57     return (bool)$this->badgeStatuses;
  58   }
  59
  60   protected function loadPage() {
  61     return $this->loadStandardPage($this->newResultObject());
  62   }
  63
  64   public function newResultObject() {
  65     return new PhabricatorBadgesAward();
  66   }
  67
  68   protected function getPrimaryTableAlias() {
  69     return 'badges_award';
  70   }
  71
  72   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
  73     $where = parent::buildWhereClauseParts($conn);
  74
  75     if ($this->badgePHIDs !== null) {
  76       $where[] = qsprintf(
  77         $conn,
  78         'badges_award.badgePHID IN (%Ls)',
  79         $this->badgePHIDs);
  80     }
  81
  82     if ($this->recipientPHIDs !== null) {
  83       $where[] = qsprintf(
  84         $conn,
  85         'badges_award.recipientPHID IN (%Ls)',
  86         $this->recipientPHIDs);
  87     }
  88
  89     if ($this->awarderPHIDs !== null) {
  90       $where[] = qsprintf(
  91         $conn,
  92         'badges_award.awarderPHID IN (%Ls)',
  93         $this->awarderPHIDs);
  94     }
  95
  96     if ($this->badgeStatuses !== null) {
  97       $where[] = qsprintf(
  98         $conn,
  99         'badges_badge.status IN (%Ls)',
 100         $this->badgeStatuses);
 101     }
 102
 103
 104     return $where;
 105   }
 106
 107   protected function buildJoinClauseParts(AphrontDatabaseConnection $conn) {
 108     $join = parent::buildJoinClauseParts($conn);
 109     $badges = new PhabricatorBadgesBadge();
 110
 111     if ($this->shouldJoinBadge()) {
 112       $join[] = qsprintf(
 113         $conn,
 114         'JOIN %T badges_badge ON badges_award.badgePHID = badges_badge.phid',
 115         $badges->getTableName());
 116     }
 117
 118     return $join;
 119   }
 120
 121   public function getQueryApplicationClass() {
 122     return 'PhabricatorBadgesApplication';
 123   }
 124
 125 }