OneTimeModifyIterator.swift

import Foundation
import PromiseKit

/**
 Каждый раз, когда будет вызван nextUnused будет возвращаться
 объект входного итератора, на котором применен модификатор, но модификатор применяется только единожды.
 Все следующие вызховы будет просто отдаваться объект, который был изменен в первый вызов.
 
 После того как будет вызван useCallback (второй элемент кортежа), курсор перейдет к следующему объекту.
 Следующий запрошенный объект снова будет запрошен из оригинального итератора, пройдет через модификатор и до следующего вызова useCallback будет отдаваться без дополнительных модификаций
 */
fileprivate class OneTimeModifyIterator<TCollection: Collection, U> {
    typealias ModifiedObjectPromise = Promise<U>
    typealias IteratorType = IndexingIterator<TCollection>
    typealias T = TCollection.Element
    
    private var currentObject: T?
    private var currentModifiedObjectPromise: ModifiedObjectPromise?
    
    private var iterator: IteratorType
    private let modifier: (T) -> ModifiedObjectPromise
    
    init(_ plainIterator: IteratorType, modifier: @escaping (T) -> ModifiedObjectPromise) {
        self.iterator = plainIterator
        self.modifier = modifier
    }
    
    // Должен отдавать следующий
    func nextUnused() -> (ModifiedObjectPromise, () -> Void)? {
        if currentObject == nil {
            currentObject = iterator.next()
            guard currentObject != nil else { return nil }
            currentModifiedObjectPromise = modifier(currentObject!)
        }
        guard currentModifiedObjectPromise != nil else { return nil }
        
        return (currentModifiedObjectPromise!, setAsUsed)
    }
    
    private func setAsUsed() {
        currentObject = nil
        currentModifiedObjectPromise = nil
    }
}