class ClosureExpressionVisitor

ArrayCollection.php in drupal/core/vendor/doctrine/common/lib/Doctrine/Common/Collections/ArrayCollection.php

ClosureExpressionVisitorTest.php in drupal/core/vendor/doctrine/common/tests/Doctrine/Tests/Common/Collections/ClosureExpressionVisitorTest.php

class ClosureExpressionVisitor extends ExpressionVisitor {

  /**
   * Access the field of a given object. This field has to be public directly
   * or indirectly (through an accessor get* or a magic method, __get, __call).
   *
   * is*() is not supported.
   *
   * @return mixed
   */
  public static function getObjectFieldValue($object, $field) {
    $accessor = "get" . $field;
    if (method_exists($object, $accessor) || method_exists($object, '__call')) {
      return $object
        ->{$accessor}();
    }
    if ($object instanceof \ArrayAccess) {
      return $object[$field];
    }
    return $object->{$field};
  }

  /**
   * Helper for sorting arrays of objects based on multiple fields +
   * orientations.
   *
   * @param string $name
   * @param int $orientation
   * @param Closure $next
   * @return Closure
   */
  public static function sortByField($name, $orientation = 1, \Closure $next = null) {
    if (!$next) {
      $next = function () {
        return 0;
      };
    }
    return function ($a, $b) use ($name, $next, $orientation) {
      $aValue = ClosureExpressionVisitor::getObjectFieldValue($a, $name);
      $bValue = ClosureExpressionVisitor::getObjectFieldValue($b, $name);
      if ($aValue === $bValue) {
        return $next($a, $b);
      }
      return ($aValue > $bValue ? 1 : -1) * $orientation;
    };
  }

  /**
   * {@inheritDoc}
   */
  public function walkComparison(Comparison $comparison) {
    $field = $comparison
      ->getField();
    $value = $comparison
      ->getValue()
      ->getValue();

    // shortcut for walkValue()
    switch ($comparison
      ->getOperator()) {
      case Comparison::EQ:
      case Comparison::IS:
        return function ($object) use ($field, $value) {
          return ClosureExpressionVisitor::getObjectFieldValue($object, $field) === $value;
        };
      case Comparison::NEQ:
        return function ($object) use ($field, $value) {
          return ClosureExpressionVisitor::getObjectFieldValue($object, $field) !== $value;
        };
      case Comparison::LT:
        return function ($object) use ($field, $value) {
          return ClosureExpressionVisitor::getObjectFieldValue($object, $field) < $value;
        };
      case Comparison::LTE:
        return function ($object) use ($field, $value) {
          return ClosureExpressionVisitor::getObjectFieldValue($object, $field) <= $value;
        };
      case Comparison::GT:
        return function ($object) use ($field, $value) {
          return ClosureExpressionVisitor::getObjectFieldValue($object, $field) > $value;
        };
      case Comparison::GTE:
        return function ($object) use ($field, $value) {
          return ClosureExpressionVisitor::getObjectFieldValue($object, $field) >= $value;
        };
      case Comparison::IN:
        return function ($object) use ($field, $value) {
          return in_array(ClosureExpressionVisitor::getObjectFieldValue($object, $field), $value);
        };
      case Comparison::NIN:
        return function ($object) use ($field, $value) {
          return !in_array(ClosureExpressionVisitor::getObjectFieldValue($object, $field), $value);
        };
      default:
        throw new \RuntimeException("Unknown comparison operator: " . $comparison
          ->getOperator());
    }
  }

  /**
   * {@inheritDoc}
   */
  public function walkValue(Value $value) {
    return $value
      ->getValue();
  }

  /**
   * {@inheritDoc}
   */
  public function walkCompositeExpression(CompositeExpression $expr) {
    $expressionList = array();
    foreach ($expr
      ->getExpressionList() as $child) {
      $expressionList[] = $this
        ->dispatch($child);
    }
    switch ($expr
      ->getType()) {
      case CompositeExpression::TYPE_AND:
        return $this
          ->andExpressions($expressionList);
      case CompositeExpression::TYPE_OR:
        return $this
          ->orExpressions($expressionList);
      default:
        throw new \RuntimeException("Unknown composite " . $expr
          ->getType());
    }
  }
  private function andExpressions($expressions) {
    return function ($object) use ($expressions) {
      foreach ($expressions as $expression) {
        if (!$expression($object)) {
          return false;
        }
      }
      return true;
    };
  }
  private function orExpressions($expressions) {
    return function ($object) use ($expressions) {
      foreach ($expressions as $expression) {
        if ($expression($object)) {
          return true;
        }
      }
      return false;
    };
  }

}