自定义Trait在视图层级外传播异常：子组件无法接收更新

你的问题出在两个核心点：

1. registerForTraitChanges仅在Trait值发生变更时触发回调，组件初始化时的初始Trait状态不会触发该方法，导致初始危险样式未被应用。
2. 若自定义Trait未配置传播规则，视图控制器的traitOverrides无法自动传递给子视图的traitCollection。

解决方案1：初始化时手动同步样式 #

在CustomButton的初始化方法中，除了注册Trait变更监听，还要手动调用一次样式更新逻辑，基于当前的traitCollection值：

class CustomButton: UIButton {
    override init(frame: CGRect) {
        super.init(frame: frame)
        setupTraitListener()
        // 初始化时手动应用样式
        updateDangerStyle()
    }
    
    required init?(coder: NSCoder) {
        super.init(coder: coder)
        setupTraitListener()
        updateDangerStyle()
    }
    
    private func setupTraitListener() {
        registerForTraitChanges([DangerStyleTrait.self]) { (self: Self, _) in
            self.updateDangerStyle()
        }
    }
    
    private func updateDangerStyle() {
        let isDangerEnabled = traitCollection.value(for: DangerStyleTrait.self)
        // 替换为你的实际样式逻辑
        backgroundColor = isDangerEnabled ? .systemRed : .systemBlue
        setTitleColor(isDangerEnabled ? .white : .black, for: .normal)
    }
}

解决方案2：确保自定义Trait支持向下传播 #

检查你的DangerStyleTrait实现，必须设置affectsSubviews = true，确保Trait能从父视图控制器自动传递给所有子视图：

enum DangerStyleTrait: UITraitDefinition {
    static let defaultValue: Bool = false
    static let name = "com.yourdomain.DangerStyle"
    // 关键配置：允许Trait向下传播给子视图
    static let affectsSubviews = true
}

可选方案：延迟Trait设置到视图加载完成（不推荐） #

如果可以接受样式在弹窗展示后更新，也可以在present的完成回调中设置traitOverrides，此时视图已加载，会触发Trait变更回调：

func showWarningAlert() {
    let viewController = InfoViewController()
    present(viewController, animated: true) {
        viewController.traitOverrides.dangerStyleEnabled = true
    }
}

该方案会导致视觉跳动，仅在特殊场景下使用。

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