iOS多主题切换问题：自定义按钮如何自动响应主题变更

核心思路是让控件监听主题变化事件，并通过「关联主题样式的引用路径」而非「静态样式实例」来动态获取最新样式，从而实现一次配置、自动更新的效果。

步骤1：改造ThemeManager，添加通知机制 #

将ThemeManager改为单例模式，增加主题切换通知，同时自动适配系统明暗模式：

import UIKit

protocol ThemeProtocol {
    var basicButtonStyle: ButtonStyle { get }
    // 可扩展其他样式属性
}

struct ButtonStyle {
    let backgroundColor: UIColor
    let titleColor: UIColor
}

struct Theme1Light: ThemeProtocol {
    var basicButtonStyle: ButtonStyle { ButtonStyle(backgroundColor: .red, titleColor: .white) }
}

struct Theme1Dark: ThemeProtocol {
    var basicButtonStyle: ButtonStyle { ButtonStyle(backgroundColor: .blue, titleColor: .white) }
}

class ThemeManager {
    static let shared = ThemeManager()
    private(set) var currentTheme: ThemeProtocol
    
    // 主题变化通知名称
    static let themeChangedNotification = Notification.Name("ThemeChanged")
    
    private init() {
        // 初始化时根据系统外观自动设置主题
        currentTheme = UITraitCollection.current.userInterfaceStyle == .dark ? Theme1Dark() : Theme1Light()
    }
    
    func setTheme(theme: ThemeProtocol) {
        currentTheme = theme
        // 发送通知告知所有控件更新样式
        NotificationCenter.default.post(name: Self.themeChangedNotification, object: nil)
    }
    
    // 处理系统明暗模式切换的快捷方法
    func updateThemeForTraitCollection(_ traitCollection: UITraitCollection) {
        let newTheme: ThemeProtocol = traitCollection.userInterfaceStyle == .dark ? Theme1Dark() : Theme1Light()
        // 避免重复设置相同主题
        if type(of: currentTheme) != type(of: newTheme) {
            setTheme(theme: newTheme)
        }
    }
}

步骤2：改造自定义按钮，实现自动更新逻辑 #

让按钮存储「主题样式的引用路径」而非静态样式实例，监听主题变化通知自动更新：

class MyButton: UIButton {
    // 存储指向ThemeProtocol中样式属性的KeyPath
    private var themeStyleKey: KeyPath<ThemeProtocol, ButtonStyle>?
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        setupNotificationObserver()
    }
    
    required init?(coder: NSCoder) {
        super.init(coder: coder)
        setupNotificationObserver()
    }
    
    private func setupNotificationObserver() {
        // 监听主题变化通知
        NotificationCenter.default.addObserver(
            self,
            selector: #selector(themeDidChange),
            name: ThemeManager.themeChangedNotification,
            object: nil
        )
    }
    
    deinit {
        NotificationCenter.default.removeObserver(self)
    }
    
    // 关联主题中的目标样式，只需调用一次
    func bindThemeStyle(_ keyPath: KeyPath<ThemeProtocol, ButtonStyle>) {
        themeStyleKey = keyPath
        // 立即应用当前主题的样式
        applyCurrentThemeStyle()
    }
    
    @objc private func themeDidChange() {
        applyCurrentThemeStyle()
    }
    
    private func applyCurrentThemeStyle() {
        guard let keyPath = themeStyleKey else { return }
        let style = ThemeManager.shared.currentTheme[keyPath: keyPath]
        backgroundColor = style.backgroundColor
        setTitleColor(style.titleColor, for: .normal)
    }
    
    override func traitCollectionDidChange(_ previousTraitCollection: UITraitCollection?) {
        super.traitCollectionDidChange(previousTraitCollection)
        // 委托ThemeManager处理系统明暗切换
        ThemeManager.shared.updateThemeForTraitCollection(traitCollection)
    }
}

步骤3：初始化按钮，一次绑定永久生效 #

只需绑定一次主题样式的引用路径，后续主题切换时自动更新：

let button: MyButton = {
    let button = MyButton()
    // 绑定主题中的basicButtonStyle，无需手动赋值样式
    button.bindThemeStyle(\.basicButtonStyle)
    return button
}()

方案优势 #

1. 解耦控件与主题实现：按钮无需知道具体的主题类型，只需关注要使用的样式属性
2. 全局统一管理：主题切换逻辑集中在ThemeManager，所有控件自动同步更新
3. 扩展性强：新增主题样式或主题类型时，只需扩展ThemeProtocol，控件无需修改代码

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