BehaviorRelay+Collection.swift

BehaviorRelay+Collection.swift

extension BehaviorRelay where Element: RangeReplaceableCollection & MutableCollection, Element.Index == Int {

    func append(element: Element.Element) {
        var array = self.value
        array.append(element)
        self.accept(array)
    }

    func append<S: Sequence>(contentsOf collection: S) where S.Element == Element.Element {
        var array = self.value
        array.append(contentsOf: collection)
        self.accept(array)
    }

    func changeElement(at index: Element.Index, _ clb: (inout Element.Element) -> Void) {
        var array = self.value
        clb(&array[index])
        self.accept(array)
    }

    func changeElement(where finder: (Element.Element) -> Bool,
                       _ clb: (inout Element.Element) -> Void) {
        var i: Element.Index?
        for (index, element) in self.value.enumerated() {
            if finder(element) {
                i = index
                break
            }
        }
        guard let index = i else { return }
        self.changeElement(at: index, clb)
    }
}