bitchat/bitchat/Services/BLE/BLEOutboundNotificationBuffer.swift

import Foundation

struct BLEPendingNotification<Target> {
    let data: Data
    let targets: [Target]?
}

struct BLEOutboundNotificationBuffer<Target> {
    enum EnqueueResult {
        case enqueued(count: Int)
        case full(count: Int)
    }

    private var notifications: [BLEPendingNotification<Target>] = []

    var count: Int {
        notifications.count
    }

    var isEmpty: Bool {
        notifications.isEmpty
    }

    mutating func removeAll() {
        notifications.removeAll()
    }

    mutating func enqueue(data: Data, targets: [Target]?, capCount: Int) -> EnqueueResult {
        guard notifications.count < capCount else {
            return .full(count: notifications.count)
        }

        notifications.append(BLEPendingNotification(data: data, targets: targets))
        return .enqueued(count: notifications.count)
    }

    mutating func takeAll() -> [BLEPendingNotification<Target>] {
        let pending = notifications
        notifications.removeAll()
        return pending
    }

    mutating func prepend(_ pending: [BLEPendingNotification<Target>]) {
        guard !pending.isEmpty else { return }
        notifications.insert(contentsOf: pending, at: 0)
    }
}