swift-composable-architecture-mirror/Examples/TicTacToe/Sources/Views-UIKit/TwoFactorViewController.swift

import Combine
import ComposableArchitecture
import TicTacToeCommon
import TwoFactorCore
import UIKit

public final class TwoFactorViewController: UIViewController {
  struct ViewState: Equatable {
    let alertData: AlertData?
    let code: String?
    let isActivityIndicatorHidden: Bool
    let isLoginButtonEnabled: Bool
  }

  enum ViewAction {
    case alertDismissed
    case codeChanged(String?)
    case loginButtonTapped
  }

  let store: Store<TwoFactorState, TwoFactorAction>
  let viewStore: ViewStore<ViewState, ViewAction>

  private var cancellables: Set<AnyCancellable> = []

  public init(store: Store<TwoFactorState, TwoFactorAction>) {
    self.store = store
    self.viewStore = ViewStore(store.scope(state: { $0.view }, action: TwoFactorAction.view))
    super.init(nibName: nil, bundle: nil)
  }

  required init?(coder: NSCoder) {
    fatalError("init(coder:) has not been implemented")
  }

  public override func viewDidLoad() {
    super.viewDidLoad()

    self.view.backgroundColor = .white

    let titleLabel = UILabel()
    titleLabel.text = "Enter the one time code to continue"
    titleLabel.textAlignment = .center

    let codeTextField = UITextField()
    codeTextField.placeholder = "1234"
    codeTextField.borderStyle = .roundedRect
    codeTextField.addTarget(
      self, action: #selector(codeTextFieldChanged(sender:)), for: .editingChanged)

    let loginButton = UIButton(type: .system)
    loginButton.setTitle("Login", for: .normal)
    loginButton.addTarget(self, action: #selector(loginButtonTapped), for: .touchUpInside)

    let activityIndicator = UIActivityIndicatorView(style: .large)
    activityIndicator.startAnimating()

    let rootStackView = UIStackView(arrangedSubviews: [
      titleLabel,
      codeTextField,
      loginButton,
      activityIndicator,
    ])
    rootStackView.isLayoutMarginsRelativeArrangement = true
    rootStackView.layoutMargins = .init(top: 0, left: 32, bottom: 0, right: 32)
    rootStackView.translatesAutoresizingMaskIntoConstraints = false
    rootStackView.axis = .vertical
    rootStackView.spacing = 24

    self.view.addSubview(rootStackView)

    NSLayoutConstraint.activate([
      rootStackView.leadingAnchor.constraint(equalTo: self.view.leadingAnchor),
      rootStackView.trailingAnchor.constraint(equalTo: self.view.trailingAnchor),
      rootStackView.centerYAnchor.constraint(equalTo: self.view.centerYAnchor),
    ])

    self.viewStore.publisher.isActivityIndicatorHidden
      .assign(to: \.isHidden, on: activityIndicator)
      .store(in: &self.cancellables)

    self.viewStore.publisher.code
      .assign(to: \.text, on: codeTextField)
      .store(in: &self.cancellables)

    self.viewStore.publisher.isLoginButtonEnabled
      .assign(to: \.isEnabled, on: loginButton)
      .store(in: &self.cancellables)

    self.viewStore.publisher.alertData
      .sink { [weak self] alertData in
        guard let self = self else { return }
        guard let alertData = alertData else { return }

        let alertController = UIAlertController(
          title: alertData.title, message: nil, preferredStyle: .alert)
        alertController.addAction(
          UIAlertAction(
            title: "Ok", style: .default,
            handler: { _ in
              self.viewStore.send(.alertDismissed)
            }))
        self.present(alertController, animated: true, completion: nil)
      }
      .store(in: &self.cancellables)
  }

  @objc private func codeTextFieldChanged(sender: UITextField) {
    self.viewStore.send(.codeChanged(sender.text))
  }

  @objc private func loginButtonTapped() {
    self.viewStore.send(.loginButtonTapped)
  }
}

extension TwoFactorState {
  var view: TwoFactorViewController.ViewState {
    .init(
      alertData: self.alertData,
      code: self.code,
      isActivityIndicatorHidden: !self.isTwoFactorRequestInFlight,
      isLoginButtonEnabled: self.isFormValid && !self.isTwoFactorRequestInFlight
    )
  }
}

extension TwoFactorAction {
  static func view(_ localAction: TwoFactorViewController.ViewAction) -> Self {
    switch localAction {
    case .alertDismissed:
      return .alertDismissed
    case let .codeChanged(code):
      return .codeChanged(code ?? "")
    case .loginButtonTapped:
      return .submitButtonTapped
    }
  }
}