NewSleepModel.swift 4 KB
import UIKit
import HandyJSON
import SwiftDate

enum NewSleepType: UInt8 {
    case nightSleep = 1
    case shortNap = 2
    case nightSleepAndNap = 3
}

enum NewSleepStatus: UInt8 {
    case awake = 7
    case rapidEyeMovement = 8
    case lightSleep = 9
    case deepSleep = 10
}

class NewSleepModel: NSObject {
    required override init() { }
    
    struct SleepFragment {
        var sleepStatus: NewSleepStatus
        var startTime: Date?
        var duration: UInt32
    }
    
    struct NapFragment {
        var startTime: Date?
        var endTime: Date?
        var duration: UInt32
    }
    
    var sleepType: NewSleepType = .nightSleep
    var sleepFragmentCount: Int = 0
    var sleepFragments: [SleepFragment] = []
    var napFragments: [NapFragment] = []
    
    init(sleepType: NewSleepType, sleepFragmentCount: Int, sleepFragments: [SleepFragment], napFragments: [NapFragment]) {
        self.sleepType = sleepType
        self.sleepFragmentCount = sleepFragmentCount
        self.sleepFragments = sleepFragments
        self.napFragments = napFragments
    }
    
    class func toSleepModel(_ data: [UInt8]) -> NewSleepModel {
        let sleepModel = NewSleepModel()
        guard data.count >= 1 else {
            return sleepModel // 数据长度不够
        }
        sleepModel.sleepType = NewSleepType(rawValue:data[0]) ?? .nightSleep
        switch sleepModel.sleepType {
        case .nightSleep:
            sleepModel.sleepFragmentCount = Int(data[1])
            guard data.count >= 2 + 9 * sleepModel.sleepFragmentCount else {
                return sleepModel
            }
            sleepModel.sleepFragments = sleepModel.parseNightSleepData(data: Array(data[2..<2 + 9 * sleepModel.sleepFragmentCount]))
        case .shortNap:
            sleepModel.napFragments = sleepModel.parseNapData(data: Array(data[2..<data.count]))
        case .nightSleepAndNap:
            sleepModel.sleepFragmentCount = Int(data[1])
            guard data.count >= 2 + 9 * sleepModel.sleepFragmentCount else {
                return sleepModel
            }
            sleepModel.sleepFragments = sleepModel.parseNightSleepData(data: Array(data[2..<2 + 9 * sleepModel.sleepFragmentCount]))
            sleepModel.napFragments = sleepModel.parseNapData(data: Array(data[2 + 9 * sleepModel.sleepFragmentCount..<data.count]))
        default:
            break
        }
        return sleepModel
    }
    
    func parseNightSleepData(data: [UInt8]) -> [SleepFragment] {
        // 解析 SleepFragment 数组
        var sleepFragments = [SleepFragment]()
        // 将数组分成9个一组
        for index in stride(from: 0, to: data.count, by: 9) {
            let sleepStatus = NewSleepStatus(rawValue: data[index]) ?? .awake
            let combinedUInt32 = data[1..<5].reduce(0) { ($0 << 8) + UInt32($1) }
            let startTime = Date(timeIntervalSince1970: TimeInterval(combinedUInt32))
            let duration = UInt32(data[5..<9].reduce(0) { ($0 << 8) + UInt32($1) })
            
            let fragment = SleepFragment(sleepStatus: sleepStatus, startTime: startTime, duration: duration)
            sleepFragments.append(fragment)
        }
        return sleepFragments
    }
    
    func parseNapData(data: [UInt8]) -> [NapFragment] {
        // 解析 NapFragment 数组
        var napFragments = [NapFragment]()
        // 将数组分成12个一组
        for index in stride(from: 0, to: data.count, by: 12) {
            let startCombinedUInt32 = data[0..<4].reduce(0) { ($0 << 8) + UInt32($1) }
            let startTime = Date(timeIntervalSince1970: TimeInterval(startCombinedUInt32))
            let endCombinedUInt32 = data[4..<8].reduce(0) { ($0 << 8) + UInt32($1) }
            let endTime = Date(timeIntervalSince1970: TimeInterval(endCombinedUInt32))
            let duration = data[8..<12].reduce(0) { ($0 << 8) + UInt32($1) }
            
            let napFragment = NapFragment(startTime: startTime, endTime: endTime, duration: duration)
            napFragments.append(napFragment)
        }
        return napFragments
    }
}