src/applications/auth/query/PhabricatorAuthMessageQuery.php

   1 <?php
   2
   3 final class PhabricatorAuthMessageQuery
   4   extends PhabricatorCursorPagedPolicyAwareQuery {
   5
   6   private $ids;
   7   private $phids;
   8   private $messageKeys;
   9
  10   public function withIDs(array $ids) {
  11     $this->ids = $ids;
  12     return $this;
  13   }
  14
  15   public function withPHIDs(array $phids) {
  16     $this->phids = $phids;
  17     return $this;
  18   }
  19
  20   public function withMessageKeys(array $keys) {
  21     $this->messageKeys = $keys;
  22     return $this;
  23   }
  24
  25   public function newResultObject() {
  26     return new PhabricatorAuthMessage();
  27   }
  28
  29   protected function loadPage() {
  30     return $this->loadStandardPage($this->newResultObject());
  31   }
  32
  33   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
  34     $where = parent::buildWhereClauseParts($conn);
  35
  36     if ($this->ids !== null) {
  37       $where[] = qsprintf(
  38         $conn,
  39         'id IN (%Ld)',
  40         $this->ids);
  41     }
  42
  43     if ($this->phids !== null) {
  44       $where[] = qsprintf(
  45         $conn,
  46         'phid IN (%Ls)',
  47         $this->phids);
  48     }
  49
  50     if ($this->messageKeys !== null) {
  51       $where[] = qsprintf(
  52         $conn,
  53         'messageKey IN (%Ls)',
  54         $this->messageKeys);
  55     }
  56
  57     return $where;
  58   }
  59
  60   protected function willFilterPage(array $messages) {
  61     $message_types = PhabricatorAuthMessageType::getAllMessageTypes();
  62
  63     foreach ($messages as $key => $message) {
  64       $message_key = $message->getMessageKey();
  65
  66       $message_type = idx($message_types, $message_key);
  67       if (!$message_type) {
  68         unset($messages[$key]);
  69         $this->didRejectResult($message);
  70         continue;
  71       }
  72
  73       $message->attachMessageType($message_type);
  74     }
  75
  76     return $messages;
  77   }
  78
  79   public function getQueryApplicationClass() {
  80     return 'PhabricatorAuthApplication';
  81   }
  82
  83 }