BleMessage.swift 14.8 KB
//
//  BleMessage.swift
//  Twear
//
//  Created by yangbin on 2021/12/8.
//

import UIKit
import SwiftDate
import SwiftUI

class BleMessage: NSObject {
    static let shared = BleMessage()
    
    func getSettingCmd() -> Data {
        return getSendData(cmd: .set, key: .setting, bytes: [])
    }
    
    func getUnitCmd(_ distance: DistanceUnit, _ temperature: TemperatureUnit) -> Data {
        return getSendData(cmd: .set, key: .unit, bytes: [distance.rawValue, temperature.rawValue])
    }
    
    func getCameraCmd(_ open: Bool) -> Data {
        return getSendData(cmd: .device, key: open ? .openCamera : .exitCamera, bytes: [0x01])
    }
    
    func getFindCmd(_ status: Bool) -> Data {
        return getSendData(cmd: .find, key: .find_device, bytes: [])
    }
    
    func getBatteryCmd() -> Data {
        return getSendData(cmd: .device, key: .battery, bytes: [])
    }
    
    
    func getWeatherCmd(_ array: [WeatherModel]) -> Data {
        var weatherBytes: [UInt8] = []
        for weather in array {
            weatherBytes.append(contentsOf: weather.toBytes())
        }
        return getSendData(cmd: .weather, key: .weather, bytes: (weatherBytes))
    }
    
    func getMessagePush(_ push: MessagePushModel) -> Data {
        return getSendData(cmd: .set, key: .push, bytes: [UInt8(push.value & 0xFF), UInt8(push.value >> 8)])
    }
    
    func getAlarmClockCmd(_ array: [AlarmClockModel]) -> Data {
        var alarmClockBytes: [UInt8] = [UInt8](repeating: 0, count: 30)
        for (i, alarm) in array.enumerated() {
            alarmClockBytes[i*5] = UInt8(alarm.date.hour)
            alarmClockBytes[i*5+1] = UInt8(alarm.date.minute)
            alarmClockBytes[i*5+2] = UInt8(alarm.cycle)
            alarmClockBytes[i*5+3] = 0x01
            alarmClockBytes[i*5+4] = alarm.isOn ? 0x01 : 0x00
        }
        return getSendData(cmd: .set, key: .alarmClock, bytes: alarmClockBytes)
    }
    
    func getDrinkWaterCmd(_ remind: RemindModel) -> Data {
        let bytes: [UInt8] = [remind.isOn ? 0x01 : 0x00, UInt8(remind.startDate.hour),  UInt8(remind.startDate.minute), UInt8(remind.endDate.hour), UInt8(remind.endDate.minute), 127, UInt8(remind.interval & 0xFF), UInt8(remind.interval >> 8)]
        return getSendData(cmd: .set, key: .drink, bytes: bytes)
    }
    
    func getSedentaryCmd(_ remind: RemindModel) -> Data {
        let target = remind.interval/60*100
        let bytes: [UInt8] = [remind.isOn ? 0x01 : 0x00, UInt8(remind.startDate.hour), UInt8(remind.endDate.hour), 127, UInt8(remind.interval & 0xFF), UInt8(remind.interval >> 8),  UInt8(target & 0xFF), UInt8(target >> 8)]
        return getSendData(cmd: .set, key: .sedentary, bytes: bytes)
    }
    
    func getAutoMeasureCmd(_ bool: Bool) -> Data {
        let bytes: [UInt8] = bool ? [0x01] : [0x00]
        return getSendData(cmd: .device, key: .autoMeasure, bytes: bytes)
    }
    
    
    func getMenstrualCmd(_ menstrual: MenstrualCalendarModel) -> Data {
        var status: UInt8 = 0x01
        var type: MenstrualDateModel
        if menstrual.newType == nil {
            type = menstrual.type!
        } else {
            type = menstrual.newType!
        }
        switch type {
        case .menstrualPeriod:
            status = 0x04
        case .fertilePeriod:
            status = 0x02
        case .safetyPeriod:
            status = 0x01
        case .ovulationDay:
            status = 0x03
        default:
            status = 0x01
        }
        let bytes: [UInt8] = [status, UInt8(menstrual.days), UInt8(menstrual.whichDay)]
        return getSendData(cmd: .set, key: .menstrual, bytes: bytes)
    }
    func getCardCmd(_ string: String, type: Int) -> Data {
        var cardBytes: [UInt8] = [UInt8(type)]
        for (i, scalar) in string.unicodeScalars.enumerated() {
//            if i < 20 {
                cardBytes.append(UInt8(scalar.value & 0xFF))
                cardBytes.append(UInt8(scalar.value >> 8))
//            }
        }
        return getSendData(cmd: .set, key: .card, bytes: cardBytes)
    }
    
    func getPayCmd(_ string: String, type: Int) -> Data {
        var payBytes: [UInt8] = [UInt8(type)]
        for scalar in string.unicodeScalars {
            payBytes.append(UInt8(scalar.value & 0xFF))
            payBytes.append(UInt8(scalar.value >> 8))
        }
        return getSendData(cmd: .set, key: .pay, bytes: payBytes)
    }
    
    func getSyncContactsCmd(_ contacts: [ContactModel], num: Int) -> Data {
        var contactBytes: [UInt8] = []
        var startIndex: Int = 0
        var key: SetCmd = .contact1
        if num == 2 {
            startIndex = 2
            key = .contact2
        } else if num == 3 {
            startIndex = 4
            key = .contact3
        } else if num == 4 {
            startIndex = 6
            key = .battery
        }
        for (i, contact) in contacts.enumerated() {
            contactBytes.append(contentsOf: contact.toBytes(startIndex+i+1))
            print(contact.toBytes(startIndex+i+1))
        }
//        print(contactBytes)
        return getSendData(cmd: .set, key: key, bytes: contactBytes)
    }
    
    func getSyncDialCmd(bytes: [UInt8]) -> Data {
        return getSendData(cmd: .set, key: .dial, bytes: bytes)
    }
    
    func getCustomDialCmd(bytes: [UInt8], color: Int, position: Int) -> Data {
        let data = getSendData(cmd: .set, key: .custom, bytes: bytes)
        var mdata = Data(repeating: 0x00, count: 120)
        mdata[0..<13] = data[0..<13]
        mdata[13] = UInt8(color)
        mdata[14] = UInt8(position)
        print([UInt8](mdata))
        return mdata
    }
    
    func getFirmwareCmd() -> Data {
        return getSendData(cmd: .firmware, key: .firmware, bytes: [])
    }
    
    
    func getSendData(cmd: BleCmd, key: SetCmd, bytes: [UInt8]) -> Data {
        let ack: UInt8 = 0
        let err: UInt8 = 0
        let version: UInt8 = 0
        let serial: UInt16 = 1

        
        if key == .custom {
            let count = bytes.count/5
            var mBytes: [UInt8] = [cmd.rawValue, 0x00, key.rawValue, UInt8(count >> 8), UInt8(count & 0xFF)]
            mBytes.append(contentsOf: bytes)
            
            let crc8 = crc8(bytes: mBytes)
            let dataLength = count + 1
            var header: [UInt8] = [0xBA, ack << 5 | err << 4 | version, UInt8(serial >> 8), UInt8(serial & 0xFF), UInt8(crc8 >> 8), UInt8(crc8 & 0xFF), UInt8(dataLength >> 8), UInt8(dataLength & 0xFF)]
            header.append(contentsOf: mBytes)
            let data = Data(bytes: header, count: header.count)
            return data
        } else {
            var mBytes: [UInt8] = [cmd.rawValue, 0x00, key.rawValue, UInt8(bytes.count >> 8), UInt8(bytes.count & 0xFF)]
            mBytes.append(contentsOf: bytes)
            
            let crc8 = crc8(bytes: mBytes)
            
            var header: [UInt8] = [0xBA, ack << 5 | err << 4 | version, UInt8(serial >> 8), UInt8(serial & 0xFF), UInt8(crc8 >> 8), UInt8(crc8 & 0xFF), UInt8(mBytes.count >> 8), UInt8(mBytes.count & 0xFF)]
            header.append(contentsOf: mBytes)
            let data = Data(bytes: header, count: header.count)
            return data
        }
 
        
    }
    
    //MARK: - 组包
    func createDataPacket(key: NewCmd, bytes: [UInt8]) -> Data {
        let identifier: [UInt8] = [0xED, 0x7E] // 标识
        let version: [UInt8] = [0x00, 0x01] // 协议版本号
        let messageID: [UInt8] = [0x00, key.rawValue] // 消息ID
        let messageSequence: [UInt8] = [0x00, 0x01] // 消息序号(示例为固定值)
        let contentLength: [UInt8] = [UInt8(bytes.count >> 8), UInt8(bytes.count & 0xFF)] // 内容长度
        let toCrc: [UInt8] = identifier + version + messageID + messageSequence + contentLength + bytes
        let crc16 = crc16(toCrc) // 计算CRC-16校验值
        let crc: [UInt8] = [UInt8(crc16 >> 8), UInt8(crc16 & 0xFF)] // 帧校验值

        // 拼接各个数据段
        let dataSegments = [identifier, version, messageID, messageSequence, contentLength, bytes, crc]

        // 合并数据段为单个数据包
        let packetData = dataSegments.reduce(Data()) { $0 + Data($1) }
        return packetData
    }
    
    let CRC16_TABLE: [UInt16] = [
        0x0000, 0x1021, 0x2042, 0x3063, 0x4084, 0x50a5, 0x60c6, 0x70e7,
        0x8108, 0x9129, 0xa14a, 0xb16b, 0xc18c, 0xd1ad, 0xe1ce, 0xf1ef,
        0x1231, 0x0210, 0x3273, 0x2252, 0x52b5, 0x4294, 0x72f7, 0x62d6,
        0x9339, 0x8318, 0xb37b, 0xa35a, 0xd3bd, 0xc39c, 0xf3ff, 0xe3de,
        0x2462, 0x3443, 0x0420, 0x1401, 0x64e6, 0x74c7, 0x44a4, 0x5485,
        0xa56a, 0xb54b, 0x8528, 0x9509, 0xe5ee, 0xf5cf, 0xc5ac, 0xd58d,
        0x3653, 0x2672, 0x1611, 0x0630, 0x76d7, 0x66f6, 0x5695, 0x46b4,
        0xb75b, 0xa77a, 0x9719, 0x8738, 0xf7df, 0xe7fe, 0xd79d, 0xc7bc,
        0x48c4, 0x58e5, 0x6886, 0x78a7, 0x0840, 0x1861, 0x2802, 0x3823,
        0xc9cc, 0xd9ed, 0xe98e, 0xf9af, 0x8948, 0x9969, 0xa90a, 0xb92b,
        0x5af5, 0x4ad4, 0x7ab7, 0x6a96, 0x1a71, 0x0a50, 0x3a33, 0x2a12,
        0xdbfd, 0xcbdc, 0xfbbf, 0xeb9e, 0x9b79, 0x8b58, 0xbb3b, 0xab1a,
        0x6ca6, 0x7c87, 0x4ce4, 0x5cc5, 0x2c22, 0x3c03, 0x0c60, 0x1c41,
        0xedae, 0xfd8f, 0xcdec, 0xddcd, 0xad2a, 0xbd0b, 0x8d68, 0x9d49,
        0x7e97, 0x6eb6, 0x5ed5, 0x4ef4, 0x3e13, 0x2e32, 0x1e51, 0x0e70,
        0xff9f, 0xefbe, 0xdfdd, 0xcffc, 0xbf1b, 0xaf3a, 0x9f59, 0x8f78,
        0x9188, 0x81a9, 0xb1ca, 0xa1eb, 0xd10c, 0xc12d, 0xf14e, 0xe16f,
        0x1080, 0x00a1, 0x30c2, 0x20e3, 0x5004, 0x4025, 0x7046, 0x6067,
        0x83b9, 0x9398, 0xa3fb, 0xb3da, 0xc33d, 0xd31c, 0xe37f, 0xf35e,
        0x02b1, 0x1290, 0x22f3, 0x32d2, 0x4235, 0x5214, 0x6277, 0x7256,
        0xb5ea, 0xa5cb, 0x95a8, 0x8589, 0xf56e, 0xe54f, 0xd52c, 0xc50d,
        0x34e2, 0x24c3, 0x14a0, 0x0481, 0x7466, 0x6447, 0x5424, 0x4405,
        0xa7db, 0xb7fa, 0x8799, 0x97b8, 0xe75f, 0xf77e, 0xc71d, 0xd73c,
        0x26d3, 0x36f2, 0x0691, 0x16b0, 0x6657, 0x7676, 0x4615, 0x5634,
        0xd94c, 0xc96d, 0xf90e, 0xe92f, 0x99c8, 0x89e9, 0xb98a, 0xa9ab,
        0x5844, 0x4865, 0x7806, 0x6827, 0x18c0, 0x08e1, 0x3882, 0x28a3,
        0xcb7d, 0xdb5c, 0xeb3f, 0xfb1e, 0x8bf9, 0x9bd8, 0xabbb, 0xbb9a,
        0x4a75, 0x5a54, 0x6a37, 0x7a16, 0x0af1, 0x1ad0, 0x2ab3, 0x3a92,
        0xfd2e, 0xed0f, 0xdd6c, 0xcd4d, 0xbdaa, 0xad8b, 0x9de8, 0x8dc9,
        0x7c26, 0x6c07, 0x5c64, 0x4c45, 0x3ca2, 0x2c83, 0x1ce0, 0x0cc1,
        0xef1f, 0xff3e, 0xcf5d, 0xdf7c, 0xaf9b, 0xbfba, 0x8fd9, 0x9ff8,
        0x6e17, 0x7e36, 0x4e55, 0x5e74, 0x2e93, 0x3eb2, 0x0ed1, 0x1ef0
    ]
    
    func crc16(_ bytes: [UInt8]) -> UInt16 {
        var crc: UInt16 = 0x0000
        for b in bytes {
            crc = (crc << 8) ^ UInt16(CRC16_TABLE[Int((crc >> 8) ^ UInt16(b)) & 0xFF])
        }
        return crc & 0xFFFF
    }

    func crc16(_ bytes: [UInt8], start: Int, len: Int) -> UInt16 {
        var crc: UInt16 = 0x0000
        let end = min(start + len, bytes.count)
        for i in start..<end {
            crc = (crc << 8) ^ UInt16(CRC16_TABLE[Int((crc >> 8) ^ UInt16(bytes[i])) & 0xFF])
        }
        return crc & 0xFFFF
    }
    

    // CRC-16校验算法
//    func calculateCRC16(_ data: Data) -> UInt16 {
//        var crc: UInt16 = 0x0000
//
//        for byte in data {
//            crc = crc ^ UInt16(byte)
//
//            for _ in 0..<8 {
//                if (crc & 0x0001) != 0 {
//                    crc = (crc >> 1) ^ 0xA001
//                } else {
//                    crc = crc >> 1
//                }
//            }
//        }
//
//        return crc
//    }
//
//    func CCITT_CRC16_FFFF(_ data: [UInt8]) -> UInt16 {

//
//        var nAccum: UInt16 = 0x0000
//
//        for byte in data {
//            let index = Int((nAccum >> 8) ^ UInt16(byte))
//            nAccum = (nAccum << 8) ^ Table_CRC16_CCITT1021[index]
//        }
//
//        return nAccum
//    }
    
    
//    func getPackData(key: String, contentBytes: [UInt8]) -> [UInt8] {
//    let messageNumber = 0xed7e
//        var messageIndexByte: [UInt8] = [0, 0]
//        messageIndexByte[1] = UInt8(messageNumber & 0x00ff)
//        messageIndexByte[0] = UInt8((messageNumber & 0xff00) >> 8)
//
//        let contentLength = contentBytes.count
//        var contentLengthByte: [UInt8] = [0, 0]
//        contentLengthByte[1] = UInt8(contentLength & 0x00ff)
//        contentLengthByte[0] = UInt8((contentLength & 0xff00) >> 8)
//
//        let messageIdByte = Array<UInt8>(hexString: key)
//
//        var length = headBytes.count + messageIdByte.count + messageIndexByte.count + contentLengthByte.count + contentBytes.count
//
//        var paramsByte = [UInt8](repeating: 0, count: length)
//        var insertLength = 0
//
//        paramsByte[insertLength..<insertLength+headBytes.count] = headBytes
//        insertLength += headBytes.count
//
//        paramsByte[insertLength..<insertLength+messageIdByte.count] = messageIdByte
//        insertLength += messageIdByte.count
//
//        paramsByte[insertLength..<insertLength+messageIndexByte.count] = messageIndexByte
//        insertLength += messageIndexByte.count
//
//        paramsByte[insertLength..<insertLength+contentLengthByte.count] = contentLengthByte
//        insertLength += contentLengthByte.count
//
//        paramsByte[insertLength..<insertLength+contentBytes.count] = contentBytes
//        insertLength += contentBytes.count
//
//        let crcResult = CRCUtils.crc16(paramsByte)
//        var crc: [UInt8] = [0, 0]
//        crc[0] = UInt8((crcResult >> 8) & 0xff)
//        crc[1] = UInt8(crcResult & 0xff)
//
//        length += 2
//
//        var total = [UInt8](repeating: 0, count: length)
//        total[0..<paramsByte.count] = paramsByte
//        total[paramsByte.count..<total.count] = crc
//
//        return total
//    }
    
    func getPackData(key: NewCmd, contentBytes: [UInt8]) -> [UInt8] {
        let headBytes: [UInt8] = [0xED, 0x7E, 0x00, 0x01]
        let messageNumber: UInt16 = 1
        var messageIndexByte: [UInt8] = [0, 0]
        messageIndexByte[1] = UInt8(messageNumber & 0x00ff)
        messageIndexByte[0] = UInt8((messageNumber & 0xff00) >> 8)
        
        let contentLength = contentBytes.count
        var contentLengthByte: [UInt8] = [0, 0]
        contentLengthByte[1] = UInt8(contentLength & 0x00ff)
        contentLengthByte[0] = UInt8((contentLength & 0xff00) >> 8)
        
        let messageIdByte: [UInt8] = [0x00, key.rawValue] // 消息ID
        
        var paramsByte: [UInt8] = []
        paramsByte.append(contentsOf: headBytes)
        paramsByte.append(contentsOf: messageIdByte)
        paramsByte.append(contentsOf: messageIndexByte)
        paramsByte.append(contentsOf: contentLengthByte)
        paramsByte.append(contentsOf: contentBytes)
        
        let crc_result = crc16(paramsByte)
        let crc: [UInt8] = [UInt8((crc_result >> 8) & 0xff), UInt8(crc_result & 0xff)]
        
        var total: [UInt8] = []
        total.append(contentsOf: paramsByte)
        total.append(contentsOf: crc)
        
        return total
    }
}