Swift 方法交换: 灵活控制应用屏幕方向 (UIInterfaceOrientationMask)

在 Swift 中，可以通过使用运行时库来交换方法，以实现对系统方法的修改。在这种情况下，我们可以使用method_exchangeImplementations函数来交换两个方法的实现。

首先，我们需要创建一个extension来交换application(_:supportedInterfaceOrientationsFor:)方法的实现：

extension UIApplication {
    @objc func my_supportedInterfaceOrientations(for window: UIWindow?) -> UIInterfaceOrientationMask {
        // 自定义的实现逻辑
        // ...
        // 返回你想要的UIInterfaceOrientationMask值
    }
}

然后，我们可以在AppDelegate中使用load方法来交换方法的实现：

@UIApplicationMain
class AppDelegate: UIResponder, UIApplicationDelegate {
    override init() {
        super.init()
        // 交换方法实现
        let originalSelector = #selector(UIApplicationDelegate.application(_:supportedInterfaceOrientationsFor:))
        let swizzledSelector = #selector(UIApplication.my_supportedInterfaceOrientations(for:))
        
        let originalMethod = class_getInstanceMethod(UIApplicationDelegate.self, originalSelector)
        let swizzledMethod = class_getInstanceMethod(UIApplication.self, swizzledSelector)
        
        method_exchangeImplementations(originalMethod!, swizzledMethod!)
    }
    
    // ...
    
    func application(_ application: UIApplication, supportedInterfaceOrientationsFor window: UIWindow?) -> UIInterfaceOrientationMask {
        // 这里实际上会调用交换后的方法实现
        return application.my_supportedInterfaceOrientations(for: window)
    }
    
    // ...
}

在自定义的my_supportedInterfaceOrientations(for:)方法中，你可以实现你自己的逻辑，并返回你想要的UIInterfaceOrientationMask值。

这样，当你在应用程序中调用supportedInterfaceOrientationsForWindow方法时，实际上会调用my_supportedInterfaceOrientations(for:)方法。