Expose known correlation associations

christophhagen · Mar 12, 2024 · a04812e · a04812e
1 parent e86c751
commit a04812e
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/Sources/HealthKitExtensions/Correlations/BloodPressure.swift b/Sources/HealthKitExtensions/Correlations/BloodPressure.swift
@@ -38,5 +38,15 @@ extension BloodPressure {
     public var foodType: String? {
         metadata?[.foodType]
     }
+
+    /// The associated systolic blood pressure sample
+    public var systolic: BloodPressureSystolic? {
+        first()
+    }
+
+    /// The associated diastolic blood pressure sample
+    public var diastolic: BloodPressureDiastolic? {
+        first()
+    }
 }
 
diff --git a/Sources/HealthKitExtensions/Correlations/Food.swift b/Sources/HealthKitExtensions/Correlations/Food.swift
@@ -33,5 +33,10 @@ extension Food {
     public var foodType: String? {
         metadata?[.foodType]
     }
+
+    /// The associated sample of consumed dietary energy
+    public var dietaryEnergyConsumed: DietaryEnergyConsumed? {
+        first()
+    }
 }
 
diff --git a/Sources/HealthKitExtensions/Generic/Correlation/HKCorrelationContainer.swift b/Sources/HealthKitExtensions/Generic/Correlation/HKCorrelationContainer.swift
@@ -45,4 +45,9 @@ extension HKCorrelationContainer {
     public func samples<T>(of type: T.Type = T.self) -> [T] where T: HKQuantitySampleContainer {
         correlation.objects(for: T.quantitySampleType).map(T.init)
     }
+
+    func first<T>() -> T? where T: HKQuantitySampleContainer {
+        objects.first { ($0 as? HKQuantitySample)?.quantityType == T.quantitySampleType }
+            .map { T.init(sample: $0) }
+    }
 }