Parameterize Macro names

joshuawright11 · Nov 27, 2023 · cf0a4a1 · cf0a4a1
1 parent a93ff51
commit cf0a4a1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/PapyrusPlugin/Sources/APIMacro.swift b/PapyrusPlugin/Sources/APIMacro.swift
@@ -10,7 +10,7 @@ public struct APIMacro: PeerMacro {
                 throw PapyrusPluginError("@API can only be applied to protocols.")
             }
 
-            let name = node.firstArgument ?? "\(type.typeName)API"
+            let name = node.firstArgument ?? "\(type.typeName)\(node.attributeName)"
             return try type.createAPI(named: name)
         }
     }

diff --git a/PapyrusPlugin/Sources/MockMacro.swift b/PapyrusPlugin/Sources/MockMacro.swift
@@ -10,7 +10,7 @@ public struct MockMacro: PeerMacro {
                 throw PapyrusPluginError("@Mock can only be applied to protocols.")
             }
 
-            let name = node.firstArgument ?? "\(type.typeName)Mock"
+            let name = node.firstArgument ?? "\(type.typeName)\(node.attributeName)"
             return try type.createMock(named: name)
         }
     }