Merge pull request #11 from SwingDev/task/improve-needsInjector-macro

Append Injector even if class doesn't have Impl in name.
SwingDev · Jan 3, 2024 · dc73e46 · dc73e46
2 parents 072f3c6 + 70508ff
commit dc73e46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/InjectGrailMacros/Sources/InjectGrailMacrosMacros/InjectGrailMacrosMacro.swift b/InjectGrailMacros/Sources/InjectGrailMacrosMacros/InjectGrailMacrosMacro.swift
@@ -42,7 +42,7 @@ public struct NeedsInjectorMacro: MemberMacro {
         }
 
         let className = classDeclaration.name.text
-        let injectorProtocolType = className.replacingOccurrences(of: "Impl", with: "Injector")
+        let injectorProtocolType = className.replacingOccurrences(of: "Impl", with: "") + "Injector"
 
         let noLet = declaration.memberBlock.members.first(where: { $0.decl.as(VariableDeclSyntax.self)?.bindings.first(where: { "\($0.pattern)"  == "injector" }) != nil}) != nil
         let noConstructor = declaration.memberBlock.members.first(where: { $0.decl.as(InitializerDeclSyntax.self) != nil})?.decl.as(InitializerDeclSyntax.self)!.signature.parameterClause.parameters.first?.firstName.text == "injector"