forked from LoopKit/Loop
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Incorporate sleep data into complication user info transfer calculati…
…ons (LoopKit#1217) * Actually resolve them :-) * Add what I have * Add sleep permission * Refine complication math * Improvements to complication-refresh code * Update to match dev * Make cartfile accurate * Add newline * TimeInterval -> Date * Ensure last update time is updated in case of failure * Remove print statement * Changes based on review * More changes in response to review * Avoid crash on HKSampleQuery error * Fix crash due to incorrect error type * Fix for authorization error * Remove delay to mirror LoopKit * Update ExponentialInsulinModelPreset.swift Co-authored-by: Pete Schwamb <[email protected]>
- Loading branch information
Showing
6 changed files
with
250 additions
and
17 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,120 @@ | ||
| // | ||
| // SleepStore.swift | ||
| // Loop | ||
| // | ||
| // Created by Anna Quinlan on 12/28/19. | ||
| // Copyright © 2019 LoopKit Authors. All rights reserved. | ||
| // | ||
|
|
||
| import Foundation | ||
| import HealthKit | ||
| import os.log | ||
|
|
||
| enum SleepStoreResult<T> { | ||
| case success(T) | ||
| case failure(SleepStoreError) | ||
| } | ||
|
|
||
| enum SleepStoreError: Error { | ||
| case noMatchingBedtime | ||
| case unknownReturnConfiguration | ||
| case noSleepDataAvailable | ||
| case queryError(String) // String is description of error | ||
| } | ||
|
|
||
| class SleepStore { | ||
| var healthStore: HKHealthStore | ||
|
|
||
| private let log = OSLog(category: "SleepStore") | ||
|
|
||
| public init( | ||
| healthStore: HKHealthStore | ||
| ) { | ||
| self.healthStore = healthStore | ||
| } | ||
|
|
||
| func getAverageSleepStartTime(sampleLimit: Int = 30, _ completion: @escaping (_ result: SleepStoreResult<Date>) -> Void) { | ||
| let inBedPredicate = HKQuery.predicateForCategorySamples( | ||
| with: .equalTo, | ||
| value: HKCategoryValueSleepAnalysis.inBed.rawValue | ||
| ) | ||
|
|
||
| let asleepPredicate = HKQuery.predicateForCategorySamples( | ||
| with: .equalTo, | ||
| value: HKCategoryValueSleepAnalysis.asleep.rawValue | ||
| ) | ||
|
|
||
| getAverageSleepStartTime(matching: inBedPredicate, sampleLimit: sampleLimit) { | ||
| (result) in | ||
| switch result { | ||
| case .success(_): | ||
| completion(result) | ||
| case .failure(let error): | ||
| switch error { | ||
| case SleepStoreError.noSleepDataAvailable: | ||
| // if there were no .inBed samples, check if there are any .asleep samples that could be used to estimate bedtime | ||
| self.getAverageSleepStartTime(matching: asleepPredicate, sampleLimit: sampleLimit, completion) | ||
| default: | ||
| // otherwise, call completion | ||
| completion(result) | ||
| } | ||
| } | ||
|
|
||
| } | ||
| } | ||
|
|
||
| fileprivate func getAverageSleepStartTime(matching predicate: NSPredicate, sampleLimit: Int, _ completion: @escaping (_ result: SleepStoreResult<Date>) -> Void) { | ||
| let sleepType = HKObjectType.categoryType(forIdentifier: HKCategoryTypeIdentifier.sleepAnalysis)! | ||
|
|
||
| // get more-recent values first | ||
| let sortByDate = NSSortDescriptor(key: HKSampleSortIdentifierStartDate, ascending: false) | ||
|
|
||
| let query = HKSampleQuery(sampleType: sleepType, predicate: predicate, limit: sampleLimit, sortDescriptors: [sortByDate]) { (query, samples, error) in | ||
|
|
||
| if let error = error { | ||
| self.log.error("Error fetching sleep data: %{public}@", String(describing: error)) | ||
| completion(.failure(SleepStoreError.queryError(error.localizedDescription))) | ||
| } else if let samples = samples as? [HKCategorySample] { | ||
| guard !samples.isEmpty else { | ||
| completion(.failure(SleepStoreError.noSleepDataAvailable)) | ||
| return | ||
| } | ||
|
|
||
| // find the average hour and minute components from the sleep start times | ||
| let average = samples.reduce(0, { | ||
| if let metadata = $1.metadata, let timezone = metadata[HKMetadataKeyTimeZone] { | ||
| return $0 + $1.startDate.timeOfDayInSeconds(sampleTimeZone: NSTimeZone(name: timezone as! String)! as TimeZone) | ||
| } else { | ||
| // default to the current timezone if the sample does not contain one in its metadata | ||
| return $0 + $1.startDate.timeOfDayInSeconds(sampleTimeZone: Calendar.current.timeZone) | ||
| } | ||
| }) / samples.count | ||
|
|
||
| let averageHour = average / 3600 | ||
| let averageMinute = average % 3600 / 60 | ||
|
|
||
| // find the next time that the user will go to bed, based on the averages we've computed | ||
| if let bedtime = Calendar.current.nextDate(after: Date(), matching: DateComponents(hour: averageHour, minute: averageMinute), matchingPolicy: .nextTime), bedtime.timeIntervalSinceNow <= .hours(24) { | ||
| completion(.success(bedtime)) | ||
| } else { | ||
| completion(.failure(SleepStoreError.noMatchingBedtime)) | ||
| } | ||
| } else { | ||
| completion(.failure(SleepStoreError.unknownReturnConfiguration)) | ||
| } | ||
| } | ||
| healthStore.execute(query) | ||
| } | ||
| } | ||
|
|
||
| extension Date { | ||
| fileprivate func timeOfDayInSeconds(sampleTimeZone: TimeZone) -> Int { | ||
| var calendar = Calendar.current | ||
| calendar.timeZone = sampleTimeZone | ||
|
|
||
| let dateComponents = calendar.dateComponents([.hour, .minute, .second], from: self) | ||
| let dateSeconds = dateComponents.hour! * 3600 + dateComponents.minute! * 60 + dateComponents.second! | ||
|
|
||
| return dateSeconds | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.