Add naming rule to apply suffix to certain generic types

cashapp · Jan 23, 2025 · abc7ff7 · abc7ff7
1 parent 8eff563
commit abc7ff7
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 3 deletions.
diff --git a/Sources/KnitCodeGen/TypeNamer.swift b/Sources/KnitCodeGen/TypeNamer.swift
@@ -32,12 +32,13 @@ enum TypeNamer {
         }
         // Drop any annotation
         var type = type.replacingOccurrences(of: "any ", with: "")
-        let removedCharacters = CharacterSet(charactersIn: "?[]():&, ")
+        let removedCharacters = CharacterSet(charactersIn: "?[]():& ")
         type = type.components(separatedBy: removedCharacters).joined(separator: "")
         let regex = try! NSRegularExpression(pattern: "<.*>")
         let nsString = type as NSString
         if let match = regex.firstMatch(in: type, range: .init(location: 0, length: type.count)) {
             let range = match.range
+            let mainType = nsString.replacingCharacters(in: match.range, with: "")
             if keepGenerics {
                 var genericName = nsString.substring(
                     with: .init(location: range.location + 1, length: range.length - 2)
@@ -47,9 +48,20 @@ enum TypeNamer {
                     .replacingOccurrences(of: ",", with: "_")
                     .replacingOccurrences(of: " ", with: "")
                 type = nsString.replacingCharacters(in: match.range, with: "_\(genericName)")
+            } else if let suffix = Self.suffixedGenericTypes.first(where: { mainType.hasSuffix($0)} ) {
+                let genericName = nsString.substring(
+                    with: .init(location: range.location + 1, length: range.length - 2)
+                )
+                if let mainGeneric = genericName.components(separatedBy: .init(charactersIn: ",")).first {
+                    type = mainGeneric + suffix
+                } else {
+                    type = mainType
+                }
             } else {
-                type = nsString.replacingCharacters(in: match.range, with: "")
+                type = mainType
             }
+        } else {
+            type = type.components(separatedBy: .init(charactersIn: ",")).joined(separator: "")
         }
         if let dotIndex = type.lastIndex(of: ".") {
             let nameStart = type.index(after: dotIndex)
@@ -66,6 +78,8 @@ enum TypeNamer {
         return type
     }
 
+    private static let suffixedGenericTypes = ["Publisher", "Subject", "Provider"]
+
     static func isClosure(type: String) -> Bool {
         return type.contains("->")
     }

diff --git a/Tests/KnitCodeGenTests/TypeNamerTests.swift b/Tests/KnitCodeGenTests/TypeNamerTests.swift
@@ -78,7 +78,7 @@ final class TypeNamerTests: XCTestCase {
 
         XCTAssertEqual(
             TypeNamer.sanitizeType(type: "Result<String, Error>", keepGenerics: true),
-            "Result_StringError"
+            "Result_String_Error"
         )
 
         XCTAssertEqual(
@@ -127,6 +127,28 @@ final class TypeNamerTests: XCTestCase {
         )
     }
 
+    func testSuffixRule() {
+        assertComputedIdentifier(
+            type: "AnyPublisher<String>",
+            expectedIdentifier: "stringPublisher"
+        )
+
+        assertComputedIdentifier(
+            type: "AnyPublisher<MyType?, Never>",
+            expectedIdentifier: "myTypePublisher"
+        )
+
+        assertComputedIdentifier(
+            type: "CurrentValueSubject<String>",
+            expectedIdentifier: "stringSubject"
+        )
+
+        assertComputedIdentifier(
+            type: "ValueProvider<Int>",
+            expectedIdentifier: "intProvider"
+        )
+    }
+
 }
 
 private func assertComputedIdentifier(