JSJP_ManagerNew/JiaPeiManage/Sources/Extensions/Array+SectionModel.swift

//
//  Array+SectionModel.swift
//  SwiftBilibili
//
//  Created by 罗文 on 2021/1/13.
//  Copyright © 2021年 罗文. All rights reserved.
//

import RxDataSources

extension Array where Element: SectionModelType {
    subscript(indexPath: IndexPath) -> Element.Item {
        get {
            return self[indexPath.section].items[indexPath.item]
        }
        mutating set {
            self.update(section: indexPath.section) { items in
                items[indexPath.item] = newValue
            }
        }
    }
    
    mutating func insert(newElement: Element.Item, at indexPath: IndexPath) {
        self.update(section: indexPath.section) { items in
            items.insert(newElement, at: indexPath.item)
        }
    }
    
    @discardableResult
    mutating func remove(at indexPath: IndexPath) -> Element.Item {
        return self.update(section: indexPath.section) { items in
            return items.remove(at: indexPath.item)
        }
    }
    
    mutating func replace(section: Int, items: [Element.Item]) {
        self[section] = Element.init(original: self[section], items: items)
    }
    
    private mutating func update<T>(section: Int, mutate: (inout [Element.Item]) -> T) -> T {
        var items = self[section].items
        let value = mutate(&items)
        self[section] = Element.init(original: self[section], items: items)
        return value
    }
}