OST08

OST08.swift

/// `ViewController`
class ViewController: UIViewController {
    private let bag = DisposeBag()
    private let indicator = PublishRelay<RxMBProgressHUD.State>()
    private let viewModel: ViewModelType

    init(_ viewModel: ViewModelType = ViewModel()) {
        self.viewModel = viewModel
        super.init(nibName: nil, bundle: nil)
        configure()
    }

    required init?(coder aDecoder: NSCoder) { fatalError("Please initialise programmatically") }

    override func viewWillAppear(_ animated: Bool) {
        super.viewWillAppear(animated)

        indicator.accept(RxMBProgressHUD.State.show(view: self.view, animated: true))
        let imageView = configuredImageView()
        view.addSubview(imageView)

        viewModel
            .source
            .drive(imageView.rx.image)
            .disposed(by: bag)
    }

    private func configure() {
        self.view.backgroundColor = .red

        indicator.bind(to: RxMBProgressHUD.shared).disposed(by: bag)
        viewModel.finished
            .subscribe(onCompleted: { self.indicator.accept(RxMBProgressHUD.State.hide(view: self.view, animated: true)) })
            .disposed(by: bag)
    }

    private func configuredImageView() -> UIImageView {
        let imageView = UIImageView(frame: CGRect(x: 0, y: 0, width: 375.0, height: 668.0))
        imageView.contentMode = .scaleAspectFit
        imageView.clipsToBounds = true
        return imageView
    }
}
// Present the view controller in the Live View window
PlaygroundPage.current.liveView = ViewController()