Merge pull request #3 from Spameri/filter_aggs

Add support for filter in aggregation collection

Author: Spamercz

src/Aggregation/LeafAggregationCollection.php

@@ -62,8 +62,10 @@ public function toArray() : array
 
 		foreach ($this->aggregations as $aggregation) {
 			if ($aggregation instanceof \Spameri\ElasticQuery\Aggregation\LeafAggregationCollection) {
-				$array[$this->key()]['aggs'] = $aggregation->toArray();
-
+				$array[$this->key()] = [
+					'aggs' => $aggregation->toArray(),
+					'filter' => $this->filter->toArray(),
+				];
 			} else {
 				$array[$this->key()] = $array + $aggregation->toArray();
 			}

src/Response/ResultMapper.php

@@ -182,14 +182,27 @@ private function mapAggregation(
 		, array $aggregationArray
 	) : \Spameri\ElasticQuery\Response\Result\Aggregation
 	{
+		$i = 0;
 		$buckets = [];
 		$aggregations = [];
-		foreach ($aggregationArray['buckets'] as $bucketPosition => $bucket) {
-			$buckets[] = $this->mapBucket($bucketPosition, $bucket);
+
+		if (isset($aggregationArray['buckets'])) {
+			foreach ($aggregationArray['buckets'] as $bucketPosition => $bucket) {
+				$buckets[] = $this->mapBucket($bucketPosition, $bucket);
+			}
+		}
+
+		if (isset($aggregationArray['doc_count']) && $aggregationArray['doc_count'] > 0) {
+			foreach ($aggregationArray as $aggregationName => $aggregation) {
+				if ( ! \is_array($aggregation)) {
+					continue;
+				}
+				$aggregations[] = $this->mapAggregation($aggregationName, $i, $aggregation);
+				$i++;
+			}
 		}
 
 		if (isset($aggregationArray['aggregations'])) {
-			$i = 0;
 			foreach ($aggregationArray['aggregations'] as $aggregationName => $aggregation) {
 				$aggregations[] = $this->mapAggregation($aggregationName, $i, $aggregation);
 				$i++;