better-testing

PhabricatorElasticsearchQueryBuilder.php
View Options

			<?php

			class PhabricatorElasticsearchQueryBuilder {
			protected $name;
			protected $clauses = array();


			public function getClauses($termkey = null) {
			$clauses = $this->clauses;
			if ($termkey == null) {
			return $clauses;
			}
			if (isset($clauses[$termkey])) {
			return $clauses[$termkey];
			}
			return array();
			}

			public function getClauseCount($clausekey) {
			if (isset($this->clauses[$clausekey])) {
			return count($this->clauses[$clausekey]);
			} else {
			return 0;
			}
			}

			public function addExistsClause($field) {
			return $this->addClause('filter', array(
			'exists' => array(
			'field' => $field,
			),
			));
			}

			public function addTermsClause($field, $values) {
			return $this->addClause('filter', array(
			'terms' => array(
			$field => array_values($values),
			),
			));
			}

			public function addMustClause($clause) {
			return $this->addClause('must', $clause);
			}

			public function addFilterClause($clause) {
			return $this->addClause('filter', $clause);
			}

			public function addShouldClause($clause) {
			return $this->addClause('should', $clause);
			}

			public function addMustNotClause($clause) {
			return $this->addClause('must_not', $clause);
			}

			public function addClause($clause, $terms) {
			$this->clauses[$clause][] = $terms;
			return $this;
			}

			public function toArray() {
			$clauses = $this->getClauses();
			return $clauses;
			$cleaned = array();
			foreach ($clauses as $clause => $subclauses) {
			if (is_array($subclauses) && count($subclauses) == 1) {
			$cleaned[$clause] = array_shift($subclauses);
			} else {
			$cleaned[$clause] = $subclauses;
			}
			}
			return $cleaned;
			}

			}