Swiftgram/SwiftSignalKit/Signal_Timing.swift

import Foundation

public func delay<T, E>(timeout: NSTimeInterval, queue: Queue)(signal: Signal<T, E>) -> Signal<T, E> {
    return Signal<T, E> { subscriber in
        let disposable = MetaDisposable()
        let timer = Timer(timeout: timeout, repeat: false, completion: {
            disposable.set(signal.start(next: { next in
                subscriber.putNext(next)
            }, error: { error in
                subscriber.putError(error)
            }, completed: {
                subscriber.putCompletion()
            }))
        }, queue: queue)
        disposable.set(ActionDisposable {
            timer.invalidate()
        })
        timer.start()
        return disposable
    }
}

public func timeout<T, E>(timeout: NSTimeInterval, queue: Queue, alternate: Signal<T, E>)(signal: Signal<T, E>) -> Signal<T, E> {
    return Signal<T, E> { subscriber in
        let disposable = MetaDisposable()
        let timer = Timer(timeout: timeout, repeat: false, completion: {
            disposable.set(alternate.start(next: { next in
                subscriber.putNext(next)
            }, error: { error in
                subscriber.putError(error)
            }, completed: {
                subscriber.putCompletion()
            }))
        }, queue: queue)
        
        disposable.set(signal.start(next: { next in
            timer.invalidate()
            subscriber.putNext(next)
        }, error: { error in
            timer.invalidate()
            subscriber.putError(error)
        }, completed: {
            timer.invalidate()
            subscriber.putCompletion()
        }))
        timer.start()
        
        let disposableSet = DisposableSet()
        disposableSet.add(ActionDisposable {
            timer.invalidate()
        })
        disposableSet.add(disposable)
        
        return disposableSet
    }
}