iOS17修改自定义Trait后，如何让所有ViewController同步感知变化？

你当前通过修改单个视图控制器的traitOverrides来切换主题，只能影响该控制器及其子视图的Trait环境，导航栈中其他控制器属于独立的Trait环境，自然无法同步感知变化。要实现所有现存视图控制器同步响应主题变更，需要将主题变更的作用域提升到应用的根Trait环境（UIWindow），因为根环境的Trait重写会自动传递给所有子环境（即所有现存视图控制器）。

具体实现步骤 #

1. 保留现有自定义Trait定义与扩展

你已实现的自定义Trait和扩展无需修改，直接沿用：

enum ThemeTraitType: Int {
    case light
    case dark
    case monochrome
}

struct ThemeTrait: UITraitDefinition {
    typealias Value = ThemeTraitType
    
    static let defaultValue: ThemeTraitType = .light
    static var name: String = "Theme"
    static var affectsColorAppearance: Bool = true
}

extension UITraitCollection {
    var currentTheme: ThemeTraitType { self[ThemeTrait.self] }
}

extension UIMutableTraits {
    var currentTheme: ThemeTraitType {
        get { self[ThemeTrait.self] }
        set { self[ThemeTrait.self] = newValue }
    }
}

2. 创建全局主题管理器

实现一个单例管理器，统一管理主题状态，并在主题变更时更新所有窗口的根Trait环境：

class ThemeManager {
    static let shared = ThemeManager()
    
    private(set) var currentTheme: ThemeTraitType = .light {
        didSet {
            guard currentTheme != oldValue else { return }
            // 遍历所有已连接的窗口场景，更新根窗口的traitOverrides
            UIApplication.shared.connectedScenes.forEach { scene in
                guard let windowScene = scene as? UIWindowScene,
                      let keyWindow = windowScene.windows.first(where: { $0.isKeyWindow }) else {
                    return
                }
                keyWindow.traitOverrides.currentTheme = currentTheme
            }
        }
    }
    
    // 外部调用的主题切换入口
    func switchToTheme(_ theme: ThemeTraitType) {
        currentTheme = theme
    }
}

3. 修改主题切换逻辑

替换原来直接修改当前控制器traitOverrides的代码，改用全局管理器触发主题变更：

// 替换 traitOverrides.currentTheme = .dark
ThemeManager.shared.switchToTheme(.dark)

4. 保留视图控制器的Trait监听逻辑

视图控制器的注册监听代码保持不变，注册后会自动接收全局主题变更的回调：

final class ViewController: UIViewController {
    override func viewDidLoad() {
        super.viewDidLoad()
        
        registerForTraitChanges([ThemeTrait.self], handler: { (self: Self, previousTraitCollection: UITraitCollection) in
            // 处理主题变更后的UI更新，比如调整颜色、字体等
            print("主题从 \(previousTraitCollection.currentTheme) 切换到 \(self.traitCollection.currentTheme)")
        })
    }
}

额外注意事项 #

• 支持多窗口场景：管理器会自动遍历所有已连接的窗口，确保每个窗口下的控制器都能同步主题。
• 栈内控制器同步：当从导航栈返回之前的控制器时，其traitCollection已自动更新为最新主题，若注册了监听会触发回调。
• 初始UI设置：可在控制器的viewDidLoad或viewWillAppear中直接读取traitCollection.currentTheme，设置初始UI状态。

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