src/applications/tokens/query/PhabricatorTokenGivenQuery.php

   1 <?php
   2
   3 final class PhabricatorTokenGivenQuery
   4   extends PhabricatorCursorPagedPolicyAwareQuery {
   5
   6   private $authorPHIDs;
   7   private $objectPHIDs;
   8   private $tokenPHIDs;
   9
  10   public function withTokenPHIDs(array $token_phids) {
  11     $this->tokenPHIDs = $token_phids;
  12     return $this;
  13   }
  14
  15   public function withObjectPHIDs(array $object_phids) {
  16     $this->objectPHIDs = $object_phids;
  17     return $this;
  18   }
  19
  20   public function withAuthorPHIDs(array $author_phids) {
  21     $this->authorPHIDs = $author_phids;
  22     return $this;
  23   }
  24
  25   public function newResultObject() {
  26     return new PhabricatorTokenGiven();
  27   }
  28
  29   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
  30     $where = parent::buildWhereClauseParts($conn);
  31
  32     if ($this->authorPHIDs !== null) {
  33       $where[] = qsprintf(
  34         $conn,
  35         'authorPHID IN (%Ls)',
  36         $this->authorPHIDs);
  37     }
  38
  39     if ($this->objectPHIDs !== null) {
  40       $where[] = qsprintf(
  41         $conn,
  42         'objectPHID IN (%Ls)',
  43         $this->objectPHIDs);
  44     }
  45
  46     if ($this->tokenPHIDs !== null) {
  47       $where[] = qsprintf(
  48         $conn,
  49         'tokenPHID IN (%Ls)',
  50         $this->tokenPHIDs);
  51     }
  52
  53     return $where;
  54   }
  55
  56   protected function willFilterPage(array $results) {
  57     $viewer = $this->getViewer();
  58
  59     $object_phids = mpull($results, 'getObjectPHID');
  60
  61     $objects = id(new PhabricatorObjectQuery())
  62       ->setViewer($viewer)
  63       ->withPHIDs($object_phids)
  64       ->execute();
  65     $objects = mpull($objects, null, 'getPHID');
  66
  67     foreach ($results as $key => $result) {
  68       $object = idx($objects, $result->getObjectPHID());
  69
  70       if ($object) {
  71         if ($object instanceof PhabricatorTokenReceiverInterface) {
  72           $result->attachObject($object);
  73           continue;
  74         }
  75       }
  76
  77       $this->didRejectResult($result);
  78       unset($results[$key]);
  79     }
  80
  81     if (!$results) {
  82       return $results;
  83     }
  84
  85     $token_phids = mpull($results, 'getTokenPHID');
  86
  87     $tokens = id(new PhabricatorTokenQuery())
  88       ->setViewer($viewer)
  89       ->withPHIDs($token_phids)
  90       ->execute();
  91     $tokens = mpull($tokens, null, 'getPHID');
  92
  93     foreach ($results as $key => $result) {
  94       $token_phid = $result->getTokenPHID();
  95
  96       $token = idx($tokens, $token_phid);
  97       if (!$token) {
  98         $this->didRejectResult($result);
  99         unset($results[$key]);
 100         continue;
 101       }
 102
 103       $result->attachToken($token);
 104     }
 105
 106     return $results;
 107   }
 108
 109   public function getQueryApplicationClass() {
 110     return 'PhabricatorTokensApplication';
 111   }
 112
 113 }