Fix issue when using count query on location search results

Fixes #391

Author: Tam

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 4.0.7 - 2024-06-06
+### Fixed
+- Fix issue when using count query on location search results (Fixes #391)
+
 ## 4.0.6 - 2024-05-08
 ### Added
 - French translations (via @pascalminator)

src/services/MapService.php

@@ -373,19 +373,24 @@ private function _searchLocation (ElementQuery $query, mixed $value, string $tab
 		];
 
 		// Filter the query
-		$query
-			->subQuery
-			->addSelect($search . ' as [[mapsCalculatedDistance]]')
+		$subQuery = $query->subQuery;
+		$isCount = count($query->select) == 1 && array_values($query->select)[0] == 'COUNT(*)';
+
+		if (!$isCount)
+			$subQuery
+				->addSelect($search . ' as [[mapsCalculatedDistance]]');
+
+		$subQuery
 			->andWhere($restrict)
 			->andWhere([
 				'not',
 				['[[' . $table . '.lat]]' => null],
 			]);
 
-		if (Craft::$app->getDb()->driverName === 'pgsql')
-			$query->subQuery->andWhere($search . ' <= ' . $radius);
+		if (Craft::$app->getDb()->driverName === 'pgsql' || $isCount)
+			$subQuery->andWhere($search . ' <= ' . $radius);
 		else
-			$query->subQuery->andHaving('[[mapsCalculatedDistance]] <= ' . $radius);
+			$subQuery->andHaving('[[mapsCalculatedDistance]] <= ' . $radius);
 
 		return '[[mapsCalculatedDistance]]';
 	}