SwiftUI如何限制扩展仅适用于自定义视图并返回对应类型？

需求目标 #

想要实现仅CustomView可调用专属修饰符，其他视图调用会编译失败，效果如下：

CustomView()
    .doSomething() // ← 仅CustomView可调用
    .doSomethingElse() // ← 仅CustomView可调用

AnyOtherView()
    .doSomething() // ← 编译不通过

实现逻辑参考SwiftUI中Text组件的专属修饰符设计。

已尝试方案 #

最初尝试给CustomView扩展返回不透明的some CustomView：

struct CustomView: View {
    ...
}

extension CustomView {
    func doSomething() -> some CustomView {
        self.environment(\.someKey, someValue)
    }

    func doSomethingElse() -> some CustomView {
        self.environment(\.someOtherKey, someOtherValue)
    }
}

报错信息："不透明类型只能指定Any、AnyObject、协议和/或基类"

另一种尝试 #

改为直接返回CustomView类型：

extension CustomView {
    func doSomething() -> CustomView {
        self.environment(\.someKey, someValue)
    }

    func doSomethingElse() -> CustomView {
        self.environment(\.someOtherKey, someOtherValue)
    }
}

报错信息：无法将类型为'some View'的返回表达式转换为返回类型'CustomView'

Xcode给出的修复方案是强制类型转换，但这种方式不安全，并非理想方案：

extension CustomView {
    func doSomething() -> CustomView {
        self.environment(\.someKey, someValue) as! CustomView
    }
}

核心诉求：仅扩展CustomView提供专属修饰符，不想扩展View协议（会让所有视图都能调用）；若扩展CustomView返回some View，则无法实现链式调用多个专属修饰符。

补充需求背景 #

正在开发Swift Package，为多个项目提供CustomView。为提升易用性，希望用链式视图修饰符替代初始化器配置视图：

• 原本的使用方式：

  CustomView(value1: someValue, value2: someOtherValue)
  

• 期望的SwiftUI风格用法（可选修饰符）：

  CustomView()
      .value1(someValue)
      .value2(someOtherValue)
  

这样在添加tint(...)或fixedSize()等通用视图修饰符时更美观，和Text组件通过可选修饰符自定义配置的逻辑一致。

解决方案 #

问题根源在于environment(_:_:)方法会返回一个ModifiedContent类型的包装视图，而非原CustomView实例，导致类型不匹配。正确的思路是将配置参数存储在CustomView自身的属性中，在body内根据属性应用环境修饰符，同时让专属修饰符返回Self以支持链式调用：

import SwiftUI

// 定义自定义环境键（根据实际需求调整）
private struct SomeKey: EnvironmentKey {
    static let defaultValue: String = ""
}

private struct SomeOtherKey: EnvironmentKey {
    static let defaultValue: Int = 0
}

extension EnvironmentValues {
    var someValue: String {
        get { self[SomeKey.self] }
        set { self[SomeKey.self] = newValue }
    }
    
    var someOtherValue: Int {
        get { self[SomeOtherKey.self] }
        set { self[SomeOtherKey.self] = newValue }
    }
}

struct CustomView: View {
    // 存储配置参数的私有属性
    private var storedSomeValue: String?
    private var storedSomeOtherValue: Int?
    
    var body: some View {
        // 核心视图内容
        Text("Custom View")
            // 根据存储的参数应用环境修饰符
            .applyIf(storedSomeValue != nil) { view in
                view.environment(\.someValue, storedSomeValue!)
            }
            .applyIf(storedSomeOtherValue != nil) { view in
                view.environment(\.someOtherValue, storedSomeOtherValue!)
            }
    }
}

extension CustomView {
    // 专属修饰符：返回Self以支持链式调用
    func value1(_ value: String) -> Self {
        var copy = self
        copy.storedSomeValue = value
        return copy
    }
    
    func value2(_ value: Int) -> Self {
        var copy = self
        copy.storedSomeOtherValue = value
        return copy
    }
}

// 辅助工具：条件应用视图修饰符
extension View {
    func applyIf<Content: View>(_ condition: Bool, transform: (Self) -> Content) -> some View {
        condition ? transform(self) : self
    }
}

方案说明 #

1. 存储配置参数：在CustomView中添加私有属性存储配置值，避免直接调用environment返回包装视图。
2. 返回Self实现链式调用：专属修饰符通过创建自身实例的副本并修改属性，返回Self类型，确保链式调用的兼容性。
3. 条件应用环境修饰符：在body中根据存储的参数决定是否应用环境修饰符，既保留环境传递的能力，又保证视图类型的一致性。

这样实现后，value1(_:)和value2(_:)仅CustomView可调用，其他视图调用会编译报错，同时支持链式调用，完全符合需求。

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