如何实现UIViewController扩展隐式返回子类ViewController类型？

To get implicit type casting when assigning the returned view controller to your specific subclass (like MyViewController), you’ll need to refactor your extension to use generics instead of returning a base UIViewController type. Here’s how to make that work:

Step 1: Updated UIViewController Extension #

import UIKit

extension UIViewController {
    @nonobjc func initiateAndAddToContainerReturnVc<T: UIViewController>(identifier: String, storyboard: String, container: UIView) -> T {
        // Initialize the specific subclass using generics
        let storyboardInstance = UIStoryboard(name: storyboard, bundle: nil)
        guard let viewController = storyboardInstance.instantiateViewController(withIdentifier: identifier) as? T else {
            fatalError("Failed to instantiate \(T.self) with identifier \(identifier) from storyboard \(storyboard)")
        }
        
        // Handle child VC lifecycle and container setup
        addChild(viewController)
        viewController.addToContainer(container: container)
        viewController.didMove(toParent: self) // Critical for proper lifecycle events
        
        return viewController
    }
    
    // Helper for adding VC view to container (assuming you had this already)
    func addToContainer(container: UIView) {
        container.addSubview(view)
        view.translatesAutoresizingMaskIntoConstraints = false
        NSLayoutConstraint.activate([
            view.topAnchor.constraint(equalTo: container.topAnchor),
            view.leadingAnchor.constraint(equalTo: container.leadingAnchor),
            view.trailingAnchor.constraint(equalTo: container.trailingAnchor),
            view.bottomAnchor.constraint(equalTo: container.bottomAnchor)
        ])
    }
}

Step 2: Usage (No Explicit Casting Required!) #

Swift will infer the generic type from the variable you’re assigning to, so you can write exactly what you wanted:

// Implicitly typed - no casting needed!
var myViewController: MyViewController = initiateAndAddToContainerReturnVc(
    identifier: "MyViewControllerIdentifier",
    storyboard: "Main",
    container: myContainerView
)

Key Notes #

• Safety: The fatalError gives you clear debug feedback if the storyboard identifier doesn’t map to the expected subclass. For a non-crashing alternative, change the return type to T? and return nil instead.
• Lifecycle: I added viewController.didMove(toParent: self) because it’s an easy-to-miss step that ensures the child VC receives all proper lifecycle events.
• Compatibility: This pattern works on iOS 13+. For older versions, you can replace the guard with instantiateViewController(withIdentifier:) as! T—but the guard still helps catch mismatches early.

内容的提问来源于stack exchange，提问作者LateNate