feat: add support to elevation gained aggregation (#129)

matinzd · Aug 4, 2024 · f513500 · f513500
1 parent 6965097
commit f513500
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/android/src/main/java/dev/matinzd/healthconnect/records/ReactElevationGainedRecord.kt b/android/src/main/java/dev/matinzd/healthconnect/records/ReactElevationGainedRecord.kt
@@ -32,10 +32,19 @@ class ReactElevationGainedRecord : ReactHealthRecordImpl<ElevationGainedRecord>
   }
 
   override fun getAggregateRequest(record: ReadableMap): AggregateRequest {
-    throw AggregationNotSupported()
+    return AggregateRequest(
+      metrics = setOf(
+        ElevationGainedRecord.ELEVATION_GAINED_TOTAL
+      ),
+      timeRangeFilter = record.getTimeRangeFilter("timeRangeFilter"),
+      dataOriginFilter = convertJsToDataOriginSet(record.getArray("dataOriginFilter"))
+    )
   }
 
   override fun parseAggregationResult(record: AggregationResult): WritableNativeMap {
-    throw AggregationNotSupported()
+    return WritableNativeMap().apply {
+      putMap("ELEVATION_GAINED_TOTAL", lengthToJsMap(record[ElevationGainedRecord.ELEVATION_GAINED_TOTAL]))
+      putArray("dataOrigins", convertDataOriginsToJsArray(record.dataOrigins))
+    }
   }
 }
diff --git a/src/types/aggregate.types.ts b/src/types/aggregate.types.ts
@@ -175,6 +175,11 @@ interface PowerAggregateResult extends BaseAggregate {
   POWER_MAX: PowerResult;
 }
 
+interface ElevationGainedAggregateResult extends BaseAggregate {
+  recordType: 'ElevationGained';
+  ELEVATION_GAINED_TOTAL: LengthResult;
+}
+
 export type AggregateRecordResult =
   | ActiveCaloriesBurnedAggregateResult
   | BasalMetabolicRateAggregateResult
@@ -195,7 +200,8 @@ export type AggregateRecordResult =
   | WheelchairPushesAggregateResult
   | StepsCadenceAggregateResult
   | TotalCaloriesBurnedAggregateResult
-  | PowerAggregateResult;
+  | PowerAggregateResult
+  | ElevationGainedAggregateResult;
 
 export type AggregateResultRecordType = AggregateRecordResult['recordType'];