SleepKit/SleepKitLib.swift at master · isulbi/SleepKit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
//
//  SleepKitLib.swift
//  SleepKitProto
//
//  Created by 현 on 2020/06/26.
//  Copyright © 2020 현. All rights reserved.
//

import Foundation
import HealthKit

public class SleepKitStore: SleepKitStruct {
    var store: HKHealthStore? = nil

    @Published var avaliable: Bool = false
    @Published var loaded: Bool = false

    var cache: Array<HKCategorySample> = []

    override init () {
        super.init()
        self.avaliable = HKHealthStore.isHealthDataAvailable()
        if !self.avaliable { return }

        // Authorization
        let readDatas = Set([HKObjectType.categoryType(forIdentifier: .sleepAnalysis)!])
        self.store = HKHealthStore()
        self.store?.requestAuthorization(toShare: nil, read: readDatas, completion: {
            (success, err) in
            if (err == nil || !success) { self.avaliable = false }
            self.avaliable = true
            self.refresh()
        })
    }

    func refresh () {
        if !self.avaliable { return }
        guard let sleepType = HKObjectType.categoryType(forIdentifier: .sleepAnalysis) else { return }
        let sortDescriptor = NSSortDescriptor(key:
                                                HKSampleSortIdentifierEndDate, ascending: false)
        let query = HKSampleQuery(sampleType: sleepType, predicate: nil, limit: 30, sortDescriptors: [sortDescriptor], resultsHandler: {
            (query, result, err) -> Void in
            if (err != nil) { return }
            var cache: Array<HKCategorySample> = []
            if let items = result {
                for item in items {
                    if let sample = item as? HKCategorySample {
                        cache.append(sample)
                    }
                }
            }

            self.cache = cache
            self.loaded = true
        })

        self.store?.execute(query)
    }

    func get (_ gap: Int = 0) -> Int {
        var result = 0
        if self.cache.capacity > 0 {
            let today = Calendar.current.dateComponents([.day], from: Date()).day
            let todayDate = gap > today! ? (31 - gap) : (today! - gap)
            for lastest in self.cache {
                print("\(lastest.startDate), \(lastest.endDate), \(lastest.value)")
                let lastDate = Calendar.current.dateComponents([.day], from: lastest.endDate).day
                if (lastDate != todayDate) || lastest.value == 0 {
                    continue
                }
                result += Int(lastest.endDate.timeIntervalSince(lastest.startDate))
            }
            if result == 0 { return get(gap + 1) }
        }
        self.save(result)
        return result
    }

    func format (_ timeGap: Int) -> String {
        let hours = Int(timeGap / 3600)
        let minutes = Int((Int(timeGap) - (hours * 3600)) / 60)
        return "\(hours)시간 \(minutes)분"
    }

    func save (_ time: Int = 0) {
        defaults!.set(String(time), forKey: localKeys.cache)
        defaults!.synchronize()
    }

    func test () {
        self.sleepTime += 3600
        self.save(self.sleepTime)
    }
}