<?php

/**
 * @extends PhabricatorCursorPagedPolicyAwareQuery<PhabricatorConduitMethodCallLog>
 */
final class PhabricatorConduitLogQuery
  extends PhabricatorCursorPagedPolicyAwareQuery {

  private $ids;
  private $callerPHIDs;
  private $methods;
  private $methodStatuses;
  private $isSystemAgent;
  private $epochMin;
  private $epochMax;

  public function withIDs(array $ids) {
    $this->ids = $ids;
    return $this;
  }

  public function withCallerPHIDs(array $phids) {
    $this->callerPHIDs = $phids;
    return $this;
  }

  public function withMethods(array $methods) {
    $this->methods = $methods;
    return $this;
  }

  public function withMethodStatuses(array $statuses) {
    $this->methodStatuses = $statuses;
    return $this;
  }

  public function withIsSystemAgent($system_agent) {
    $this->isSystemAgent = $system_agent;
    return $this;
  }

  public function withEpochBetween($epoch_min, $epoch_max) {
    $this->epochMin = $epoch_min;
    $this->epochMax = $epoch_max;
    return $this;
  }

  public function newResultObject() {
    return new PhabricatorConduitMethodCallLog();
  }

  protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
    $where = parent::buildWhereClauseParts($conn);

    if ($this->ids !== null) {
      $where[] = qsprintf(
        $conn,
        'conduitlog.id IN (%Ld)',
        $this->ids);
    }

    if ($this->callerPHIDs !== null) {
      $where[] = qsprintf(
        $conn,
        'conduitlog.callerPHID IN (%Ls)',
        $this->callerPHIDs);
    }

    if ($this->methods !== null) {
      $where[] = qsprintf(
        $conn,
        'conduitlog.method IN (%Ls)',
        $this->methods);
    }

    if ($this->methodStatuses !== null) {
      $statuses = array_fuse($this->methodStatuses);

      $methods = id(new PhabricatorConduitMethodQuery())
        ->setViewer($this->getViewer())
        ->execute();

      $method_names = array();
      foreach ($methods as $method) {
        $status = $method->getMethodStatus();
        if (isset($statuses[$status])) {
          $method_names[] = $method->getAPIMethodName();
        }
      }

      if (!$method_names) {
        throw new PhabricatorEmptyQueryException();
      }

      $where[] = qsprintf(
        $conn,
        'conduitlog.method IN (%Ls)',
        $method_names);
    }

    if ($this->isSystemAgent !== null) {
      $where[] = qsprintf(
        $conn,
        'u.isSystemAgent = %d',
        (int)$this->isSystemAgent);
    }

    if ($this->epochMin !== null) {
      $where[] = qsprintf(
        $conn,
        'conduitlog.dateCreated >= %d',
        $this->epochMin);
    }

    if ($this->epochMax !== null) {
      $where[] = qsprintf(
        $conn,
        'conduitlog.dateCreated <= %d',
        $this->epochMax);
    }

    return $where;
  }

  protected function buildJoinClauseParts(AphrontDatabaseConnection $conn) {
    $joins = parent::buildJoinClauseParts($conn);

    if ($this->isSystemAgent !== null) {
      $user_table = new PhabricatorUser();
      $joins[] = qsprintf(
        $conn,
        'JOIN %R u ON u.phid = conduitlog.callerPHID',
        $user_table);
    }

    return $joins;
  }

  protected function getPrimaryTableAlias() {
    return 'conduitlog';
  }

  public function getQueryApplicationClass() {
    return PhabricatorConduitApplication::class;
  }

}