bluetooth_low_energy/bluetooth_low_energy_darwin/darwin/Classes/MyPeripheralManager.swift

//
//  MyPeripheralManager.swift
//  bluetooth_low_energy_darwin
//
//  Created by 闫守旺 on 2023/10/7.
//

import Foundation
import CoreBluetooth

#if os(iOS)
import Flutter
#elseif os(macOS)
import FlutterMacOS
#else
#error("Unsupported platform.")
#endif

class MyPeripheralManager: MyPeripheralManagerHostApi {
    private let _api: MyPeripheralManagerFlutterApi
    private let _peripheralManager: CBPeripheralManager
    
    private lazy var _peripheralManagerDelegate = MyPeripheralManagerDelegate(peripheralManager: self)
    
    private var _servicesArgs: [Int: MyGattServiceArgs]
    private var _characteristicsArgs: [Int: MyGattCharacteristicArgs]
    private var _descriptorsArgs: [Int: MyGattDescriptorArgs]
    
    private var _centrals: [String: CBCentral]
    private var _services: [Int64: CBMutableService]
    private var _characteristics: [Int64: CBMutableCharacteristic]
    private var _descriptors: [Int64: CBMutableDescriptor]
    private var _requests: [Int64: CBATTRequest]
    
    private var _addServiceCompletion: ((Result<Void, Error>) -> Void)?
    private var _startAdvertisingCompletion: ((Result<Void, Error>) -> Void)?
    private var _isReadyToUpdateSubscribersCallbacks: [() -> Void]
    
    init(messenger: FlutterBinaryMessenger) {
        _api = MyPeripheralManagerFlutterApi(binaryMessenger: messenger)
        _peripheralManager = CBPeripheralManager()
        
        _servicesArgs = [:]
        _characteristicsArgs = [:]
        _descriptorsArgs = [:]
        
        _centrals = [:]
        _services = [:]
        _characteristics = [:]
        _descriptors = [:]
        _requests = [:]
        
        _addServiceCompletion = nil
        _startAdvertisingCompletion = nil
        _isReadyToUpdateSubscribersCallbacks = []
    }
    
    func setUp() throws {
        _clearState()
        if _peripheralManager.delegate == nil {
            _peripheralManager.delegate = _peripheralManagerDelegate
        }
        _onStateChanged()
    }
    
    func addService(serviceArgs: MyGattServiceArgs, completion: @escaping (Result<Void, Error>) -> Void) {
        let service = serviceArgs.toService()
        var characteristics = [CBMutableCharacteristic]()
        let characteristicsArgs = serviceArgs.characteristicsArgs
        for args in characteristicsArgs {
            guard let characteristicArgs = args else {
                continue
            }
            let characteristic = characteristicArgs.toCharacteristic()
            characteristics.append(characteristic)
            var descriptors = [CBMutableDescriptor]()
            let descriptorsArgs = characteristicArgs.descriptorsArgs
            for args in descriptorsArgs {
                guard let descriptorArgs = args else {
                    continue
                }
                let descriptor = descriptorArgs.toDescriptor()
                descriptors.append(descriptor)
                let descriptorHashCodeArgs = descriptorArgs.hashCodeArgs
                let descriptorHashCode = descriptor.hash
                _descriptorsArgs[descriptorHashCode] = descriptorArgs
                _descriptors[descriptorHashCodeArgs] = descriptor
            }
            characteristic.descriptors = descriptors
            let characteristicHashCodeArgs = characteristicArgs.hashCodeArgs
            let characteristicHashCode = characteristic.hash
            _characteristicsArgs[characteristicHashCode] = characteristicArgs
            _characteristics[characteristicHashCodeArgs] = characteristic
        }
        service.characteristics = characteristics
        let serviceHashCodeArgs = serviceArgs.hashCodeArgs
        let serviceHashCode = service.hash
        _servicesArgs[serviceHashCode] = serviceArgs
        _services[serviceHashCodeArgs] = service
        _peripheralManager.add(service)
        _addServiceCompletion = completion
    }
    
    func removeService(hashCodeArgs: Int64) throws {
        guard let service = _services[hashCodeArgs] else {
            throw MyError.illegalArgument
        }
        let hashCode = service.hash
        guard let serviceArgs = _servicesArgs[hashCode] else {
            throw MyError.illegalArgument
        }
        _peripheralManager.remove(service)
        _clearService(serviceArgs: serviceArgs)
    }
    
    func clearServices() throws {
        _peripheralManager.removeAllServices()
        let servicesArgs = _servicesArgs.values
        for serviceArgs in servicesArgs {
            _clearService(serviceArgs: serviceArgs)
        }
    }
    
    func startAdvertising(advertisementArgs: MyAdvertisementArgs, completion: @escaping (Result<Void, Error>) -> Void) {
        let advertisement = advertisementArgs.toAdvertisement()
        _peripheralManager.startAdvertising(advertisement)
        _startAdvertisingCompletion = completion
    }
    
    func stopAdvertising() throws {
        _peripheralManager.stopAdvertising()
    }
    
    func getMaximumUpdateValueLength(uuidArgs: String) throws -> Int64 {
        guard let central = _centrals[uuidArgs] else {
            throw MyError.illegalArgument
        }
        let maximumUpdateValueLength = central.maximumUpdateValueLength
        let maximumUpdateValueLengthArgs = maximumUpdateValueLength.toInt64()
        return maximumUpdateValueLengthArgs
    }
    
    func respond(idArgs: Int64, errorNumberArgs: Int64, valueArgs: FlutterStandardTypedData?) throws {
        guard let request = _requests.removeValue(forKey: idArgs) else {
            throw MyError.illegalArgument
        }
        if valueArgs != nil {
            request.value = valueArgs?.data
        }
        let errorArgsRawValue = errorNumberArgs.toInt()
        guard let errorArgs = MyGattErrorArgs(rawValue: errorArgsRawValue) else {
            throw MyError.illegalArgument
        }
        let result = errorArgs.toError()
        _peripheralManager.respond(to: request, withResult: result)
    }
    
    func updateCharacteristic(hashCodeArgs: Int64, valueArgs: FlutterStandardTypedData, uuidsArgs: [String]?, completion: @escaping (Result<Void, Error>) -> Void) {
        do {
            let centrals = try uuidsArgs?.map { uuidArgs in
                guard let central = _centrals[uuidArgs] else {
                    throw MyError.illegalArgument
                }
                return central
            }
            guard let characteristic = _characteristics[hashCodeArgs] else {
                throw MyError.illegalArgument
            }
            let value = valueArgs.data
            let updated = _peripheralManager.updateValue(value, for: characteristic, onSubscribedCentrals: centrals)
            if updated {
                completion(.success(()))
            } else {
                _isReadyToUpdateSubscribersCallbacks.append {
                    self.updateCharacteristic(hashCodeArgs: hashCodeArgs, valueArgs: valueArgs, uuidsArgs: uuidsArgs, completion: completion)
                }
            }
        } catch {
            completion(.failure(error))
        }
    }
    
    func didUpdateState(peripheral: CBPeripheralManager) {
        _onStateChanged()
    }
    
    func didAdd(peripheral: CBPeripheralManager, service: CBService, error: Error?) {
        guard let completion = _addServiceCompletion else {
            return
        }
        _addServiceCompletion = nil
        if error == nil {
            completion(.success(()))
        } else {
            completion(.failure(error!))
            let hashCode = service.hash
            guard let serviceArgs = _servicesArgs[hashCode] else {
                return
            }
            _clearService(serviceArgs: serviceArgs)
        }
    }
    
    func didStartAdvertising(peripheral: CBPeripheralManager, error: Error?) {
        guard let completion = _startAdvertisingCompletion else {
            return
        }
        _startAdvertisingCompletion = nil
        if error == nil {
            completion(.success(()))
        } else {
            completion(.failure(error!))
        }
    }
    
    func didReceiveRead(peripheral: CBPeripheralManager, request: CBATTRequest) {
        let central = request.central
        let centralArgs = central.toArgs()
        _centrals[centralArgs.uuidArgs] = central
        let characteristic = request.characteristic
        let hashCode = characteristic.hash
        guard let characteristicArgs = _characteristicsArgs[hashCode] else {
            _peripheralManager.respond(to: request, withResult: .attributeNotFound)
            return
        }
        let hashCodeArgs = characteristicArgs.hashCodeArgs
        let idArgs = request.hash.toInt64()
        let offsetArgs = request.offset.toInt64()
        _requests[idArgs] = request
        _api.onCharacteristicReadRequest(centralArgs: centralArgs, hashCodeArgs: hashCodeArgs, idArgs: idArgs, offsetArgs: offsetArgs) {_ in }
    }
    
    func didReceiveWrite(peripheral: CBPeripheralManager, requests: [CBATTRequest]) {
        // When you respond to a write request, note that the first parameter of the respond(to:with
        // Result:) method expects a single CBATTRequest object, even though you received an array of
        // them from the peripheralManager(_:didReceiveWrite:) method. To respond properly,
        // pass in the first request of the requests array.
        // see: https://developer.apple.com/documentation/corebluetooth/cbperipheralmanagerdelegate/1393315-peripheralmanager#discussion
        guard let request = requests.first else {
            return
        }
        if requests.count > 1 {
            // TODO: 支持多写入请求，暂时不清楚此处应如何处理
            let result = CBATTError.requestNotSupported
            _peripheralManager.respond(to: request, withResult: result)
            return
        }
        let central = request.central
        let centralArgs = central.toArgs()
        _centrals[centralArgs.uuidArgs] = central
        let characteristic = request.characteristic
        let hashCode = characteristic.hash
        guard let characteristicArgs = _characteristicsArgs[hashCode] else {
            _peripheralManager.respond(to: request, withResult: .attributeNotFound)
            return
        }
        let hashCodeArgs = characteristicArgs.hashCodeArgs
        let idArgs = request.hash.toInt64()
        let offsetArgs = request.offset.toInt64()
        guard let value = request.value else {
            _peripheralManager.respond(to: request, withResult: .requestNotSupported)
            return
        }
        let valueArgs = FlutterStandardTypedData(bytes: value)
        _requests[idArgs] = request
        _api.onCharacteristicWriteRequest(centralArgs: centralArgs, hashCodeArgs: hashCodeArgs, idArgs: idArgs, offsetArgs: offsetArgs, valueArgs: valueArgs) {_ in }
    }
    
    func didSubscribeTo(peripheral: CBPeripheralManager, central: CBCentral, characteristic: CBCharacteristic) {
        let centralArgs = central.toArgs()
        _centrals[centralArgs.uuidArgs] = central
        let hashCode = characteristic.hash
        guard let characteristicArgs = _characteristicsArgs[hashCode] else {
            return
        }
        let hashCodeArgs = characteristicArgs.hashCodeArgs
        let stateArgs = true
        _api.onCharacteristicNotifyStateChanged(centralArgs: centralArgs, hashCodeArgs: hashCodeArgs, stateArgs: stateArgs) {_ in }
    }
    
    func didUnsubscribeFrom(peripheral: CBPeripheralManager, central: CBCentral, characteristic: CBCharacteristic) {
        let centralArgs = central.toArgs()
        _centrals[centralArgs.uuidArgs] = central
        let hashCode = characteristic.hash
        guard let characteristicArgs = _characteristicsArgs[hashCode] else {
            return
        }
        let hashCodeArgs = characteristicArgs.hashCodeArgs
        let stateArgs = false
        _api.onCharacteristicNotifyStateChanged(centralArgs: centralArgs, hashCodeArgs: hashCodeArgs, stateArgs: stateArgs) {_ in }
    }
    
    func isReadyToUpdateSubscribers(peripheral: CBPeripheralManager) {
        let callbacks = _isReadyToUpdateSubscribersCallbacks
        _isReadyToUpdateSubscribersCallbacks.removeAll()
        for callback in callbacks {
            callback()
        }
    }
    
    private func _clearState() {
        if(_peripheralManager.isAdvertising) {
            _peripheralManager.stopAdvertising()
        }
        
        _servicesArgs.removeAll()
        _characteristicsArgs.removeAll()
        _descriptorsArgs.removeAll()
        
        _centrals.removeAll()
        _services.removeAll()
        _characteristics.removeAll()
        _descriptors.removeAll()
        _requests.removeAll()
        
        _addServiceCompletion = nil
        _startAdvertisingCompletion = nil
        _isReadyToUpdateSubscribersCallbacks.removeAll()
    }
    
    private func _clearService(serviceArgs: MyGattServiceArgs) {
        let characteristicsArgs = serviceArgs.characteristicsArgs
        for args in characteristicsArgs {
            guard let characteristicArgs = args else {
                continue
            }
            let descriptorsArgs = characteristicArgs.descriptorsArgs
            for args in descriptorsArgs {
                guard let descriptorArgs = args else {
                    continue
                }
                let descriptorHashCodeArgs = descriptorArgs.hashCodeArgs
                guard let descriptor = _descriptors.removeValue(forKey: descriptorHashCodeArgs) else {
                    continue
                }
                let descriptorHashCode = descriptor.hash
                _descriptorsArgs.removeValue(forKey: descriptorHashCode)
            }
            let characteristicHashCodeArgs = characteristicArgs.hashCodeArgs
            guard let characteristic = _characteristics.removeValue(forKey: characteristicHashCodeArgs) else {
                continue
            }
            let characteristicHashCode = characteristic.hash
            _characteristicsArgs.removeValue(forKey: characteristicHashCode)
        }
        let serviceHashCodeArgs = serviceArgs.hashCodeArgs
        guard let service = _services.removeValue(forKey: serviceHashCodeArgs) else {
            return
        }
        let serviceHashCode = service.hash
        _servicesArgs.removeValue(forKey: serviceHashCode)
    }
    
    private func _onStateChanged() {
        let state = _peripheralManager.state
        let stateArgs = state.toArgs()
        let stateNumberArgs = stateArgs.rawValue.toInt64()
        _api.onStateChanged(stateNumberArgs: stateNumberArgs) {_ in }
    }
}