SwiftUI：如何通过自定义View Modifier访问自定义视图子视图

我定义了如下自定义视图：

public struct CustomView: View {
    var text1: String
    var text2: String

    public var body: some View {
        VStack {
            Text(text1).tag(0)
            Text(text2).tag(1)
        }
    }
}

希望创建一个可传入自定义Style参数的View Modifier，实现根据不同Style修改两个Text的前景色：

struct Watermark: ViewModifier {
    var style: Style

    func body(content: Content) -> some View {
          if style == .one {
              // 访问tag为0的Text并设置前景色为红色
              // 访问tag为1的Text并设置前景色为蓝色
          } else if style == .two {
              // 访问tag为0的Text并设置前景色为黑色
              // 访问tag为1的Text并设置前景色为粉色
          }
}

要求必须使用View Modifier，而非将Style作为参数传入视图初始化（因为自定义视图会有7个以上参数，用Modifier更简洁），实现类似Apple SF Image的样式切换效果。

SwiftUI中无法直接在View Modifier里访问content的子视图结构（Content是不透明类型），因此需要通过环境变量+自定义Style协议的方式实现，这也是SF Symbols样式系统的核心思路：

1. 定义自定义样式枚举与环境键 #

// 定义样式枚举
enum CustomViewStyle {
    case one
    case two
}

// 定义环境键，用于传递样式
private struct CustomViewStyleKey: EnvironmentKey {
    static let defaultValue: CustomViewStyle = .one // 设置默认样式
}

// 扩展环境值，添加自定义样式的访问接口
extension EnvironmentValues {
    var customViewStyle: CustomViewStyle {
        get { self[CustomViewStyleKey.self] }
        set { self[CustomViewStyleKey.self] = newValue }
    }
}

2. 修改CustomView，读取环境样式并应用 #

让自定义视图主动读取环境中的样式，自行调整子视图的外观：

public struct CustomView: View {
    var text1: String
    var text2: String
    @Environment(\.customViewStyle) private var style // 读取环境样式

    public var body: some View {
        VStack {
            Text(text1)
                .foregroundColor(style == .one ? .red : .black)
            Text(text2)
                .foregroundColor(style == .one ? .blue : .pink)
        }
    }
}

3. 实现View Modifier与便捷调用扩展 #

创建Modifier来设置环境中的样式，并给View添加扩展方便链式调用：

struct CustomViewStyleModifier: ViewModifier {
    let style: CustomViewStyle

    func body(content: Content) -> some View {
        content
            .environment(\.customViewStyle, style)
    }
}

// 扩展View，提供简洁的调用方式
extension View {
    func customViewStyle(_ style: CustomViewStyle) -> some View {
        self.modifier(CustomViewStyleModifier(style: style))
    }
}

4. 使用示例 #

// 应用样式的方式
CustomView(text1: "第一行文本", text2: "第二行文本")
    .customViewStyle(.two) // 链式调用Modifier切换样式

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