[Refactor] Enum을 사용해서 화면 분기처리하기

 

enum ViewType {
    case start
    case change
    
    var menuTitle: String {
        switch self {
        case .start: return "시작하기"
        case .change: return "변경하기"
        }
    }
    
    var navigationTitle: String {
        switch self {
        case .start: return "다마고치 선택하기"
        case .change: return "다마고치 변경하기"
        }
    }
}

 

    var viewType: ViewType = .start

	override func viewDidLoad() {
        super.viewDidLoad()
        setupNavigation()
        setupCollectionView()
    }
    
    // MARK: - Custom Method
    
    private func setupNavigation() {
        navigationItem.title = viewType.navigationTitle
    }

vc.startTitle = viewType.menuTitle

 

 

화면전환 시에는

guard let vc = sb.instantiateViewController(withIdentifier: SelectCollectionViewController.identifier)
                    as? SelectCollectionViewController else { return }
vc.viewType = .change
navigationController?.pushViewController(vc, animated: true)