使用Swift宏时如何获取变量声明的类型？

你现在需要实现@MemberwiseReimplementation宏来自动生成结构体的成员初始化器，但卡在了没法从var someProperty = "John Doe"这类未显式指定类型的变量中拿到String类型——不用自己维护字面量到类型的映射表，Swift宏的API本身就支持直接复用编译器的类型推断结果。

直接利用编译器的类型推断能力 #

Swift宏的扩展逻辑中，可以通过语法节点的type(in:)方法，结合宏的MacroExpansionContext上下文，直接获取编译器已经推断完成的类型信息，完全不需要手动处理各种字面量场景。

具体实现示例 #

以下是一个完整的扩展宏实现，核心逻辑就是获取推断类型并生成初始化器：

import SwiftSyntax
import SwiftSyntaxMacros

struct MemberwiseReimplementationMacro: ExtensionMacro {
    static func expansion(
        of node: AttributeSyntax,
        attachedTo declaration: some DeclGroupSyntax,
        providingExtensionsOf type: some TypeSyntaxProtocol,
        conformingTo protocols: [TypeSyntax],
        in context: some MacroExpansionContext
    ) throws -> [ExtensionDeclSyntax] {
        // 提取结构体中的存储属性
        let storedProperties = declaration.memberBlock.members.compactMap {
            $0.decl.as(VariableDeclSyntax.self)?.filter { $0.isStoredProperty }
        }

        // 生成初始化器的参数列表
        let initParams = TupleTypeElementListSyntax(storedProperties.map { prop in
            let propName = prop.bindings.first?.pattern.as(IdentifierPatternSyntax.self)?.identifier.text ?? ""
            // 优先使用显式类型，无显式类型则获取编译器推断结果
            let propType: TypeSyntax
            if let explicitType = prop.typeAnnotation?.type {
                propType = explicitType
            } else {
                guard let binding = prop.bindings.first,
                      let initExpr = binding.initializer?.value,
                      let inferredType = try? initExpr.type(in: context) else {
                    throw MacroError.cannotInferType(propName: propName)
                }
                propType = inferredType.asTypeSyntax
            }
            return TupleTypeElementSyntax(name: .identifier(propName), type: propType)
        })

        // 生成初始化器内的赋值代码
        let initBody = CodeBlockItemListSyntax(storedProperties.map { prop in
            let propName = prop.bindings.first?.pattern.as(IdentifierPatternSyntax.self)?.identifier.text ?? ""
            return CodeBlockItemSyntax(item: .expression(ExprSyntax("self.\(raw: propName) = \(raw: propName)")))
        })

        // 组装初始化器
        let initializer = InitializerDeclSyntax(
            signature: FunctionSignatureSyntax(input: FunctionParameterClauseSyntax(parameters: initParams)),
            body: CodeBlockSyntax(stmts: initBody)
        )

        // 生成扩展并返回
        let extensionDecl = ExtensionDeclSyntax(
            extendedType: type,
            memberBlock: MemberBlockSyntax(members: [MemberBlockItemSyntax(decl: initializer)])
        )
        return [extensionDecl]
    }
}

enum MacroError: Error {
    case cannotInferType(propName: String)
    var localizedDescription: String {
        switch self {
        case .cannotInferType(let name):
            return "无法推断属性\(name)的类型，请显式指定类型或提供初始化值"
        }
    }
}

关键细节说明 #

• type(in:)核心方法：这个方法能直接获取语法节点经过编译器类型推断后的类型，不管是字符串、数字还是自定义字面量类型，都能准确返回对应的TypeSyntax，完全替代手动映射的方案。
• 显式类型优先：如果变量已经显式声明类型（比如var age: Int = 25），直接使用显式类型可以避免特殊场景下的推断误差。
• 错误兜底：当变量既无显式类型也无初始化值时，抛出明确的错误提示，避免生成无效代码。

避坑提醒 #

• 不要尝试用type(of:)：这是运行时API，而宏是编译期执行的逻辑，根本无法调用。
• 无需手动维护类型映射表：编译器的类型推断已经覆盖了所有合法场景，自己写映射表容易遗漏自定义字面量、类型别名等特殊情况。

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