//
//  DelegateProxyTest+UIKit.swift
//  RxTests
//
//  Created by Krunoslav Zaher on 12/5/15.
//  Copyright © 2015 Krunoslav Zaher. All rights reserved.
//

import Foundation
import UIKit
@testable import RxCocoa
@testable import RxSwift
import XCTest

// MARK: Protocols

@objc protocol UITableViewDelegateSubclass
    : UITableViewDelegate
    , TestDelegateProtocol {
    @objc optional func testEventHappened(_ value: Int)
}

@objc protocol UITableViewDataSourceSubclass
    : UITableViewDataSource
    , TestDelegateProtocol {
    @objc optional func testEventHappened(_ value: Int)
}

@objc protocol UICollectionViewDelegateSubclass
    : UICollectionViewDelegate
    , TestDelegateProtocol {
    @objc optional func testEventHappened(_ value: Int)
}

@objc protocol UICollectionViewDataSourceSubclass
    : UICollectionViewDataSource
    , TestDelegateProtocol {
    @objc optional func testEventHappened(_ value: Int)
}

@objc protocol UIScrollViewDelegateSubclass
    : UIScrollViewDelegate
    , TestDelegateProtocol {
    @objc optional func testEventHappened(_ value: Int)
}

#if os(iOS)
@objc protocol UISearchBarDelegateSubclass
    : UISearchBarDelegate
    , TestDelegateProtocol {
    @objc optional func testEventHappened(_ value: Int)
}
#endif

@objc protocol UITextViewDelegateSubclass
    : UITextViewDelegate
    , TestDelegateProtocol {
    @objc optional func testEventHappened(_ value: Int)
}
#if os(iOS)
extension RxSearchControllerDelegateProxy: TestDelegateProtocol {
}
extension RxPickerViewDelegateProxy: TestDelegateProtocol {
}
#endif

// MARK: Tests

// MARK: UITableView

extension DelegateProxyTest {
    func test_UITableViewDelegateExtension() {
        performDelegateTest(UITableViewSubclass1(frame: CGRect.zero))
    }

    func test_UITableViewDataSourceExtension() {
        performDelegateTest(UITableViewSubclass2(frame: CGRect.zero))
    }
}

// MARK: UICollectionView

extension DelegateProxyTest {

    func test_UICollectionViewDelegateExtension() {
        let layout = UICollectionViewFlowLayout()
        performDelegateTest(UICollectionViewSubclass1(frame: CGRect.zero, collectionViewLayout: layout))
    }

    func test_UICollectionViewDataSourceExtension() {
        let layout = UICollectionViewFlowLayout()
        performDelegateTest(UICollectionViewSubclass2(frame: CGRect.zero, collectionViewLayout: layout))
    }
}

// MARK: UIScrollView

extension DelegateProxyTest {
    func test_UIScrollViewDelegateExtension() {
        performDelegateTest(UIScrollViewSubclass(frame: CGRect.zero))
    }
}

// MARK: UISearchBar

#if os(iOS)
extension DelegateProxyTest {
    func test_UISearchBarDelegateExtension() {
        performDelegateTest(UISearchBarSubclass(frame: CGRect.zero))
    }
}
#endif

// MARK: UITextView

extension DelegateProxyTest {
    func test_UITextViewDelegateExtension() {
        performDelegateTest(UITextViewSubclass(frame: CGRect.zero))
    }
}

// MARK UISearchController
#if os(iOS)
extension DelegateProxyTest {
    func test_UISearchController() {
        performDelegateTest(UISearchControllerSubclass())
    }
}
    
extension DelegateProxyTest {
    func test_UIPickerViewExtension() {
        performDelegateTest(UIPickerViewSubclass(frame: CGRect.zero))
    }
}
#endif
// MARK: Mocks

class ExtendTableViewDelegateProxy
    : RxTableViewDelegateProxy
    , UITableViewDelegateSubclass {
    weak fileprivate(set) var control: UITableViewSubclass1?

    required init(parentObject: AnyObject) {
        self.control = (parentObject as! UITableViewSubclass1)
        super.init(parentObject: parentObject)
    }
}

class UITableViewSubclass1
    : UITableView
    , TestDelegateControl {
    override func createRxDelegateProxy() -> RxScrollViewDelegateProxy {
        return ExtendTableViewDelegateProxy(parentObject: self)
    }

    func doThatTest(_ value: Int) {
        (delegate as! TestDelegateProtocol).testEventHappened?(value)
    }

    var test: Observable<Int> {
        return rx.delegate
            .observe(#selector(TestDelegateProtocol.testEventHappened(_:)))
            .map { a in (a[0] as! NSNumber).intValue }
    }

    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
        return RxScrollViewDelegateProxy.installForwardDelegate(testDelegate, retainDelegate: false, onProxyForObject: self)
    }
}

class ExtendTableViewDataSourceProxy
    : RxTableViewDataSourceProxy
    , UITableViewDelegateSubclass {
    weak fileprivate(set) var control: UITableViewSubclass2?

    required init(parentObject: AnyObject) {
        self.control = (parentObject as! UITableViewSubclass2)
        super.init(parentObject: parentObject)
    }
}

class UITableViewSubclass2
    : UITableView
    , TestDelegateControl {
    override func createRxDataSourceProxy() -> RxTableViewDataSourceProxy {
        return ExtendTableViewDataSourceProxy(parentObject: self)
    }

    func doThatTest(_ value: Int) {
        (dataSource as! TestDelegateProtocol).testEventHappened?(value)
    }

    var test: Observable<Int> {
        return rx.dataSource
            .observe(#selector(TestDelegateProtocol.testEventHappened(_:)))
            .map { a in (a[0] as! NSNumber).intValue }
    }

    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
        return RxTableViewDataSourceProxy.installForwardDelegate(testDelegate, retainDelegate: false, onProxyForObject: self)
    }
}

class ExtendCollectionViewDelegateProxy
    : RxCollectionViewDelegateProxy
    , UITableViewDelegateSubclass {
    weak fileprivate(set) var control: UICollectionViewSubclass1?

    required init(parentObject: AnyObject) {
        self.control = (parentObject as! UICollectionViewSubclass1)
        super.init(parentObject: parentObject)
    }
}

class UICollectionViewSubclass1
    : UICollectionView
    , TestDelegateControl {
    override func createRxDelegateProxy() -> RxScrollViewDelegateProxy {
        return ExtendCollectionViewDelegateProxy(parentObject: self)
    }

    func doThatTest(_ value: Int) {
        (delegate as! TestDelegateProtocol).testEventHappened?(value)
    }

    var test: Observable<Int> {
        return rx.delegate
            .observe(#selector(TestDelegateProtocol.testEventHappened(_:)))
            .map { a in (a[0] as! NSNumber).intValue }
    }

    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
        return RxScrollViewDelegateProxy.installForwardDelegate(testDelegate, retainDelegate: false, onProxyForObject: self)
    }
}

class ExtendCollectionViewDataSourceProxy
    : RxCollectionViewDataSourceProxy
    , UICollectionViewDelegateSubclass {
    weak fileprivate(set) var control: UICollectionViewSubclass2?

    required init(parentObject: AnyObject) {
        self.control = (parentObject as! UICollectionViewSubclass2)
        super.init(parentObject: parentObject)
    }
}

class UICollectionViewSubclass2
    : UICollectionView
    , TestDelegateControl {
    override func createRxDataSourceProxy() -> RxCollectionViewDataSourceProxy {
        return ExtendCollectionViewDataSourceProxy(parentObject: self)
    }

    func doThatTest(_ value: Int) {
        (dataSource as! TestDelegateProtocol).testEventHappened?(value)
    }

    var test: Observable<Int> {
        return rx.dataSource
            .observe(#selector(TestDelegateProtocol.testEventHappened(_:)))
            .map { a in (a[0] as! NSNumber).intValue }
    }

    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
        return RxCollectionViewDataSourceProxy.installForwardDelegate(testDelegate, retainDelegate: false, onProxyForObject: self)
    }
}

class ExtendScrollViewDelegateProxy
    : RxScrollViewDelegateProxy
    , UIScrollViewDelegateSubclass {
    weak fileprivate(set) var control: UIScrollViewSubclass?

    required init(parentObject: AnyObject) {
        self.control = (parentObject as! UIScrollViewSubclass)
        super.init(parentObject: parentObject)
    }
}

class UIScrollViewSubclass
    : UIScrollView
    , TestDelegateControl {
    override func createRxDelegateProxy() -> RxScrollViewDelegateProxy {
        return ExtendScrollViewDelegateProxy(parentObject: self)
    }

    func doThatTest(_ value: Int) {
        (delegate as! TestDelegateProtocol).testEventHappened?(value)
    }

    var test: Observable<Int> {
        return rx.delegate
            .observe(#selector(TestDelegateProtocol.testEventHappened(_:)))
            .map { a in (a[0] as! NSNumber).intValue }
    }

    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
        return RxScrollViewDelegateProxy.installForwardDelegate(testDelegate, retainDelegate: false, onProxyForObject: self)
    }
}

#if os(iOS)
class ExtendSearchBarDelegateProxy
    : RxSearchBarDelegateProxy
    , UISearchBarDelegateSubclass {
    weak fileprivate(set) var control: UISearchBarSubclass?
    
    required init(parentObject: AnyObject) {
        self.control = (parentObject as! UISearchBarSubclass)
        super.init(parentObject: parentObject)
    }
}

class UISearchBarSubclass
    : UISearchBar
    , TestDelegateControl {
    
    override func createRxDelegateProxy() -> RxSearchBarDelegateProxy {
        return ExtendSearchBarDelegateProxy(parentObject: self)
    }
    
    func doThatTest(_ value: Int) {
        (delegate as! TestDelegateProtocol).testEventHappened?(value)
    }
    
    var test: Observable<Int> {
        return rx.delegate
            .observe(#selector(TestDelegateProtocol.testEventHappened(_:)))
            .map { a in (a[0] as! NSNumber).intValue }
    }

    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
        return RxSearchBarDelegateProxy.installForwardDelegate(testDelegate, retainDelegate: false, onProxyForObject: self)
    }
}
#endif

class ExtendTextViewDelegateProxy
    : RxTextViewDelegateProxy
    , UITextViewDelegateSubclass {
    weak fileprivate(set) var control: UITextViewSubclass?

    required init(parentObject: AnyObject) {
        self.control = (parentObject as! UITextViewSubclass)
        super.init(parentObject: parentObject)
    }
}

class UITextViewSubclass
    : UITextView
    , TestDelegateControl {
    override func createRxDelegateProxy() -> RxScrollViewDelegateProxy {
        return ExtendTextViewDelegateProxy(parentObject: self)
    }

    func doThatTest(_ value: Int) {
        (delegate as! TestDelegateProtocol).testEventHappened?(value)
    }

    var test: Observable<Int> {
        return rx.delegate
            .observe(#selector(TestDelegateProtocol.testEventHappened(_:)))
            .map { a in (a[0] as! NSNumber).intValue }
    }

    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
        return RxScrollViewDelegateProxy.installForwardDelegate(testDelegate, retainDelegate: false, onProxyForObject: self)
    }
}
#if os(iOS)
class UISearchControllerSubclass
    : UISearchController
    , TestDelegateControl {

    func doThatTest(_ value: Int) {
        (delegate as! TestDelegateProtocol).testEventHappened?(value)
    }
    
    var test: Observable<Int> {
        return rx.delegate
            .observe(#selector(TestDelegateProtocol.testEventHappened(_:)))
            .map { a in (a[0] as! NSNumber).intValue }
    }

    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
        return RxSearchControllerDelegateProxy.installForwardDelegate(testDelegate, retainDelegate: false, onProxyForObject: self)
    }
}
    
class UIPickerViewSubclass
    : UIPickerView
    , TestDelegateControl {
    
    func doThatTest(_ value: Int) {
        (delegate as! TestDelegateProtocol).testEventHappened?(value)
    }
    
    var test: Observable<Int> {
        return rx.delegate
            .observe(#selector(TestDelegateProtocol.testEventHappened(_:)))
            .map { a in (a[0] as! NSNumber).intValue }
    }
    
    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
        return RxPickerViewDelegateProxy.installForwardDelegate(testDelegate,
                                                                retainDelegate: false,
                                                                onProxyForObject: self)
    }
}
    
#endif