Update CustomESResponseSerializer.jsx

src/components/Searchkit/CustomESResponseSerializer.jsx

@@ -1,20 +1,27 @@
 // TODO add data of .title to serialized response
 // aggregations.organisationunit_agg.organisationunit_token.buckets[1].somemoredatafromelasticsearch.hits.hits[0]._source.title
 
+import { forEach, has } from 'lodash';
+
 function _pimpedAggregations(aggregations) {
   let result = Object.assign({}, aggregations);
   let buckets = [];
-  Object.keys(aggregations).forEach((element) => {
-    Object.keys(aggregations[element]).forEach((token) => {
-      if (token !== 'doc_count') {
-        buckets = aggregations[element][token].buckets;
-        buckets &&
-          buckets.forEach((bucket) => {
-            bucket.label =
-              bucket.somemoredatafromelasticsearch.hits.hits[0]._source.title;
-          });
-      }
-    });
+  Object.keys(result).forEach((element) => {
+    if (result[element].buckets) {
+      buckets = result[element].buckets;
+    } else {
+      forEach(result[element].inner, function (value, key) {
+        if (has(value, 'buckets')) {
+          buckets = result[element].inner[key].buckets;
+        }
+      });
+    }
+    buckets &&
+      buckets.forEach((bucket) => {
+        bucket.label =
+          bucket.somemoredatafromelasticsearch?.hits.hits[0]._source.title ??
+          bucket.key; 
+      });
   });
 
   return result;