DisposeBagProvider.swift

//  DisposeBagProvider.swift

import RxSwift
import UIKit

/*
// Just make a class conform to it like this: 
class MyClass : DisposeBagProvider {
}
// And then just use that "disposeBag" variable 
// when you need a disposeBag. 
*/

protocol DisposeBagProvider {
    var disposeBag: DisposeBag {get}
}

extension DisposeBagProvider where Self: AnyObject {
    
    var disposeBag: DisposeBag {
        get {
            if let bag = objc_getAssociatedObject(self, &AssociatedKeys.DisposeBagKey) as? DisposeBag {
                return bag
            }
            
            let bag = DisposeBag()
            objc_setAssociatedObject(self, &AssociatedKeys.DisposeBagKey, bag, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
            
            return bag
        }
    }
}

private struct AssociatedKeys {
    static var DisposeBagKey = "dispose_bag_key"
}