[Swift] 보기 좋게? UIViewController 만들어 보기

 

 

읽기 쉬운 코드 만들기를 목적으로 공부한 내용 중 일부를 정리해 보려고 합니다.

 

코딩을 하다보면 복잡하고 유지보수 하기 힘들도록 레거시를 만들어 내는 경우가 많습니다... 뭔가 중구난방...

그래서 기본 틀은 항상 유지할 수 있도록 공통 컨테이너를 만들어 사용하면 좋을 것 같았습니다.

보기 좋은 떡이 먹기도 좋다더라...

 

BaseViewController라는 기본 컨트롤러를 서브클래싱하여 좀 더 깔끔한 코드를 만들어 본 내용을 간단히 정리해 보려 합니다.

지극히 주관적인 내용이라 별 내용이 아닐 수 있지만... 피드백은 언제나 환영합니당~ 👍

 

 

class BaseViewController: UIViewController {
    
    // MARK:- Rx
    
    var disposeBag = DisposeBag()
    

    // MARK:- View Life Cycle
    
    override func viewDidLoad() {
        setupLayout()
        setupConstraints()
        setupAttributes()
        setupLocalization()
        setData()
        setupBinding()
    }
    
    func setupLayout() {
        // Override Layout
    }
    
    func setupConstraints() {
        // Override Constraints
    }
    
    func setupAttributes() {
        // Override Attributes
    }
    
    func setupLocalization() {
        // Override Localization
    }
    
    func setData() {
        // Override Set Data
    }

    func setupBinding() {
        // Override Binding
    }
    
}

위 코드는 viewDidLoad 에서 공통으로 실행될 만한 메소드가 실행되도록 구현했습니다.

BaseViewController를 서브클래싱한 컨트롤러에서는 필요한 메소드를 override 하여 구현만 해주면 됩니다.

 

아래 코드와 같이 서브클래싱하여 사용할 수 있습니다.

class ViewController: BaseViewController {
    
    // MARK:- UI
    
    let titleLabel = UILabel()
    
    let button = UIButton()
    
    
    // MARK:- View Life Cycle
    
    override func viewDidLoad() {
        super.viewDidLoad()
    }
    
    override func setupLayout() {
        view.addSubview(titleLabel)
        view.addSubview(button)
    }
    
    override func setupConstraints() {
        titleLabel.snp.makeConstraints {
            $0.center.equalToSuperview()
        }
        
        button.snp.makeConstraints {
            $0.centerX.equalToSuperview()
            $0.top.equalTo(titleLabel.snp.bottom).offset(20)
        }
    }
    
    override func setData() {
        titleLabel.text = "Hello~"
    }
    
    override func setupBinding() {
        button.rx.tap
            .subscribe(onNext: {
                // Do Something
            })
            .disposed(by: disposeBag)
    }
}

이러면 특정 컨트롤러에 필요한 데이터를 제외하고 항상 같은 구조로 코딩할 수 있을 것 입니다.

 

또한 UIAlertController를 띄워준다던지 로그인 여부를 체크하여 이동하는 등의 필요한 공통기능들을 구현하여 간단히 사용하고, 유지보수도 용이하도록 할 수 있습니다.

extension BaseViewController {
    func showAlert(
        title: String?,
        message: String?,
        confirmHandler: (() -> Void)? = nil,
        completion: (() -> Void)? = nil)
    {
        let alertView = UIAlertController(title: title, message: message, preferredStyle: .alert)
        let confirmAction = UIAlertAction(title: "Confirm", style: .default) { _ in
            if let handler = confirmHandler {
                handler()
            }
        }
        alertView.addAction(confirmAction)
        present(alertView, animated: true, completion: completion)
    }
    
    func goToLoginView() {
        // 로그인 여부 체크 및 이동
    }
}

 

 

 

 

- Reference

https://github.com/devxoul/Drrrible/blob/master/Drrrible/Sources/ViewControllers/BaseViewController.swift

https://tono18.tistory.com/22