Merge pull request #13 from SwingDev/task/fix-extracting-from-Injects…

…-macro Fix extracting from Injects macro
SwingDev · Apr 15, 2024 · 6519743 · 6519743
2 parents 5241d2c + 4a82c98
commit 6519743
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Templates/Inject.swift b/Templates/Inject.swift
@@ -242,9 +242,10 @@ func extractNeededName(_ type: Type) -> String? {
 }
 
 func extractInjects(_ type: Type, _ injectablesToInjectors: [String: String], _ protocolsToInjectables: [String: [String]]) -> [String] {
-    guard 
-       let attribute = type.attributes["Injects"]?.first
-       else { return [] }
+    guard
+        let injectsKey = type.attributes.keys.first(where: { $0.hasPrefix("Injects") }),
+        let attribute = type.attributes[injectsKey]?.first
+    else { return [] }
 
     let name = String(describing: attribute)