SwiftUI中如何让@ViewBuilder返回some Shape而非some View？

问题场景 #

假设你有多个自定义Shape，都遵循继承自Shape的HasNormalizedAABB协议：

protocol HasNormalizedAABB: Shape {
    func normalizedAABB() -> CGRect
}

struct LoggedInShape: HasNormalizedAABB {
    func normalizedAABB() -> CGRect {
        CGRect(x: 0, y: 0, width: 100, height: 100)
    }
    
    func path(in rect: CGRect) -> Path {
        // 自定义路径实现
        Path()
    }
}

struct NoAuthShape: HasNormalizedAABB {
    func normalizedAABB() -> CGRect {
        CGRect(x: 0, y: 0, width: 80, height: 80)
    }
    
    func path(in rect: CGRect) -> Path {
        Path()
    }
}

struct AdminShape: HasNormalizedAABB {
    func normalizedAABB() -> CGRect {
        CGRect(x: 0, y: 0, width: 120, height: 120)
    }
    
    func path(in rect: CGRect) -> Path {
        Path()
    }
}

同时有一个认证状态枚举：

enum AuthStatus {
    case loggedIn
    case noAuth
    case admin
}

当你尝试根据状态返回对应Shape并统一加样式时，会遇到两个问题：

1. 用@ViewBuilder返回some View时，调用.fill()会报错，因为View没有这个方法
2. 把返回类型改成some Shape，会因为Swift的条件视图（_ConditionalView）不遵循Shape协议而失败

临时的分支内单独加样式方案，存在重复代码多、维护成本高的问题——新增枚举成员或修改样式时都要逐个修改分支。

解决方案 #

方案1：Swift 5.7+ 直接返回any Shape

Swift 5.7引入的existential类型允许我们直接返回any Shape，不需要@ViewBuilder：

func shapeFor(status: AuthStatus) -> any Shape {
    switch status {
    case .loggedIn:
        LoggedInShape()
    case .noAuth:
        NoAuthShape()
    case .admin:
        AdminShape()
    }
}

调用时就能正常使用所有Shape的方法：

shapeFor(status: currentAuthStatus)
    .fill(.red)
    .frame(width: 200, height: 200)

方案2：自定义类型擦除的AnyShape（兼容旧Swift版本）

如果项目还在用Swift 5.7以下版本，可以写一个类型擦除的包装器来统一Shape类型：

struct AnyShape: Shape {
    private let pathClosure: (CGRect) -> Path
    
    init<S: Shape>(_ shape: S) {
        pathClosure = shape.path(in:)
    }
    
    func path(in rect: CGRect) -> Path {
        pathClosure(rect)
    }
}

然后修改返回函数，用AnyShape包装每个自定义Shape：

func shapeFor(status: AuthStatus) -> some Shape {
    switch status {
    case .loggedIn:
        AnyShape(LoggedInShape())
    case .noAuth:
        AnyShape(NoAuthShape())
    case .admin:
        AnyShape(AdminShape())
    }
}

这样返回的是具体的some Shape类型，能正常调用.fill()等方法，同时兼容旧版本Swift。

方案3：用ViewModifier封装样式逻辑

如果需要复用多种样式组合，把样式封装成ViewModifier能进一步降低维护成本：

struct DefaultShapeStyle: ViewModifier {
    func body(content: Content) -> some View {
        content
            .fill(.red)
            .stroke(.black, lineWidth: 2)
            .frame(width: 150, height: 150)
    }
}

// 给View加扩展，方便调用
extension View {
    func applyDefaultShapeStyle() -> some View {
        self.modifier(DefaultShapeStyle())
    }
}

调用时直接用扩展方法：

shapeFor(status: currentAuthStatus)
    .applyDefaultShapeStyle()

后续修改样式只需要调整DefaultShapeStyle的实现，新增样式也只需要加新的Modifier，完全不需要改动返回Shape的函数。

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