SwiftUI中如何通过View扩展直接访问EnvironmentObject

正确实现方案 #

首先保留你的ColorTheme类不变：

class ColorTheme: ObservableObject {
    @Published var lightColor = Colors.light
}

接下来扩展EnvironmentValues，将ColorTheme与环境绑定，利用EnvironmentObject的底层存储逻辑：

extension EnvironmentValues {
    var colorTheme: ColorTheme {
        get {
            guard let theme = self[ObjectIdentifier(ColorTheme.self)] as? ColorTheme else {
                fatalError("请先通过 .environmentObject(ColorTheme()) 注入 ColorTheme 实例")
            }
            return theme
        }
        set {
            self[ObjectIdentifier(ColorTheme.self)] = newValue
        }
    }
}

然后给View添加扩展，让所有视图都能直接访问colorTheme属性：

extension View {
    var colorTheme: ColorTheme {
        get { Environment(\.colorTheme).wrappedValue }
        set { self.environment(\.colorTheme, newValue) }
    }
}

使用方式 #

在App入口注入实例（两种方式任选其一，效果完全一致）：

@main
struct MyApp: App {
    var body: some Scene {
        WindowGroup {
            // 方式1：沿用原来的environmentObject
            ContentView()
                .environmentObject(ColorTheme())
            
            // 方式2：使用自定义环境键
            // ContentView()
            //     .environment(\.colorTheme, ColorTheme())
        }
    }
}

之后在任意视图中直接访问，无需声明@EnvironmentObject：

struct ContentView: View {
    var body: some View {
        Text("示例文本")
            .foregroundColor(colorTheme.lightColor)
    }
}

为什么你的原有方案不生效 #

你之前用普通EnvironmentKey实现的方式，只是把ColorTheme当成普通值存储在环境中，没有利用EnvironmentObject的响应式机制。SwiftUI只有在识别到对象是通过EnvironmentObject注入的ObservableObject时，才会自动监听其@Published属性的变化并触发视图刷新。

可选安全版本（避免崩溃） #

如果不想因未注入实例导致崩溃，可以将属性改为可选类型：

extension EnvironmentValues {
    var colorTheme: ColorTheme? {
        get { self[ObjectIdentifier(ColorTheme.self)] as? ColorTheme }
        set { self[ObjectIdentifier(ColorTheme.self)] = newValue }
    }
}

extension View {
    var colorTheme: ColorTheme? {
        get { Environment(\.colorTheme).wrappedValue }
        set { self.environment(\.colorTheme, newValue) }
    }
}

使用时处理可选值：

struct ContentView: View {
    var body: some View {
        Text("示例文本")
            .foregroundColor(colorTheme?.lightColor ?? .gray)
    }
}

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