wip

Author: mbrandonw

Tests/MacroTestingTests/MacroExamples/AddAsyncMacro.swift

@@ -53,7 +53,8 @@ public struct AddAsyncMacro: PeerMacro {
       let completionHandlerParameterAttribute = funcDecl.signature.parameterClause.parameters.last?
         .type.as(AttributedTypeSyntax.self),
       let completionHandlerParameter = completionHandlerParameterAttribute.baseType.as(
-        FunctionTypeSyntax.self)
+        FunctionTypeSyntax.self
+      )
     else {
       throw CustomError.message(
         "@addAsync requires an function that has a completion handler as last parameter"
@@ -70,12 +71,27 @@ public struct AddAsyncMacro: PeerMacro {
     }
 
     let returnType = completionHandlerParameter.parameters.first?.type
-
     let isResultReturn = returnType?.children(viewMode: .all).first?.description == "Result"
-    let successReturnType =
-      isResultReturn
-      ? returnType!.as(IdentifierTypeSyntax.self)!.genericArgumentClause?.arguments.first!.argument
-      : returnType
+
+    let successReturnType: TypeSyntax? = {
+      guard isResultReturn
+      else { return returnType }
+
+      let successType = returnType!.as(IdentifierTypeSyntax.self)!.genericArgumentClause?.arguments.first!
+        .argument
+      #if canImport(SwiftSyntax601)
+        switch successType {
+        case .none:
+          return nil
+        case .type(let syntax):
+          return syntax
+        default:
+          return nil
+        }
+      #else
+        return successType
+      #endif
+    }()
 
     // Remove completionHandler and comma from the previous parameter
     var newParameterList = funcDecl.signature.parameterClause.parameters
@@ -141,7 +157,9 @@ public struct AddAsyncMacro: PeerMacro {
     // add result type
     if let successReturnType {
       funcDecl.signature.returnClause = ReturnClauseSyntax(
-        leadingTrivia: .space, type: successReturnType.with(\.leadingTrivia, .space))
+        leadingTrivia: .space,
+        type: successReturnType.with(\.leadingTrivia, .space)
+      )
     } else {
       funcDecl.signature.returnClause = nil
     }

Tests/MacroTestingTests/MacroExamples/OptionSetMacro.swift

@@ -93,7 +93,9 @@ public struct OptionSetMacro {
       else {
         context.diagnose(
           OptionSetMacroDiagnostic.requiresStringLiteral(optionsEnumNameArgumentLabel).diagnose(
-            at: optionEnumNameArg.expression))
+            at: optionEnumNameArg.expression
+          )
+        )
         return nil
       }
 
@@ -121,7 +123,8 @@ public struct OptionSetMacro {
       }).first
     else {
       context.diagnose(
-        OptionSetMacroDiagnostic.requiresOptionsEnum(optionsEnumName).diagnose(at: decl))
+        OptionSetMacroDiagnostic.requiresOptionsEnum(optionsEnumName).diagnose(at: decl)
+      )
       return nil
     }
 
@@ -135,7 +138,25 @@ public struct OptionSetMacro {
       return nil
     }
 
-    return (structDecl, optionsEnum, rawType)
+    let rawTypeSyntax: TypeSyntax?
+    #if canImport(SwiftSyntax601)
+      switch rawType {
+      case .type(let typeSyntax):
+        rawTypeSyntax = typeSyntax
+      default:
+        rawTypeSyntax = nil
+      }
+    #else
+      rawTypeSyntax = rawType
+    #endif
+
+    guard let rawTypeSyntax
+    else {
+      context.diagnose(OptionSetMacroDiagnostic.requiresOptionsEnumRawType.diagnose(at: attribute))
+      return nil
+    }
+
+    return (structDecl, optionsEnum, rawTypeSyntax)
   }
 }