swift 概念证明类型安全的RAC包装器

Posted 2021-05-19

protocol ISubscriber {
    typealias Element

    func disposable() -> RACCompoundDisposable

    func sendNext(e: Element)
    func sendError(e: NSError)
    func sendCompleted()
}

/// Wraps RACSubscriber
class Subscriber<T: AnyObject> : ISubscriber {
    let subscriber: RACSubscriber

    init(wrapped: RACSubscriber) 
    {
        subscriber = wrapped
    }

    func disposable() -> RACCompoundDisposable
    {
        return subscriber.disposable
    }

    func sendNext(e: T)
    {
        subscriber.sendNext(e)
    }

    func sendError(e: NSError)
    {
        subscriber.sendError(e)
    }

    func sendCompleted()
    {
        subscriber.sendCompleted()
    }
}

/// Wraps RACSignal
class Signal<T: AnyObject> {
    let signal: RACSignal

    init(wrapped: RACSignal)
    {
        signal = wrapped
    }

    convenience init(didSubscribe: Subscriber<T> -> ())
    {
        return self.init(wrapped: RACSignal.create { racSubscriber in
            let subscriber = Subscriber<T>(wrapped: racSubscriber)
            didSubscribe(subscriber)
        })
    }

    //TODO -[RACSignal return:] could not be called here directly because Swift hates keywords as selectors, hence this suboptimal implementation
    convenience init(value: T)
    {
        return self.init(didSubscribe: { s in
            s.sendNext(value)
            s.sendCompleted()
        })
    }

    // operations

    func flatMap<U: AnyObject>(f: T -> Signal<U>) -> Signal<U>
    {
        return Signal<U>(wrapped: signal.flattenMap { value in
            return f(value as T).signal
        })
    }

    func map<U: AnyObject>(f: T -> U) -> Signal<U>
    {
        return Signal<U>(wrapped: signal.map { value in
            return f(value as T)
        })
    }

    func filter(p: T -> Bool) -> Signal<T>
    {
        return Signal(wrapped: signal.filter { value in
            return p(value as T)
        })
    }

    // subscription

    func subscribe(next: T -> ()) -> RACDisposable
    {
        return signal.subscribeNext { value in next(value as T) }
    }

    func subscribe(next: T -> (), completed: () -> ()) -> RACDisposable
    {
        return signal.subscribeNext(
            { value in next(value as T) },
            completed: { completed() }
        )
    }
}