Add Spy generating

raptorxcz · Sep 13, 2023 · fe19c36 · fe19c36
1 parent 9601526
commit fe19c36
Show file tree

Hide file tree

Showing 12 changed files with 587 additions and 1,087 deletions.
diff --git a/Rubicon/Sources/Rubicon/Domain/StructDeclaration.swift b/Rubicon/Sources/Rubicon/Domain/StructDeclaration.swift
@@ -0,0 +1,4 @@
+struct StructDeclaration: Equatable {
+    let name: String
+    let variables: [VarDeclaration]
+}
diff --git a/Rubicon/Sources/Rubicon/Generator/CreateSpyInteractor.swift b/Rubicon/Sources/Rubicon/Generator/CreateSpyInteractor.swift
diff --git a/Rubicon/Sources/Rubicon/Generator/FunctionNameGenerator.swift b/Rubicon/Sources/Rubicon/Generator/FunctionNameGenerator.swift
@@ -1,5 +1,6 @@
 protocol FunctionNameGenerator {
     func makeUniqueName(for function: FunctionDeclaration, in functions: [FunctionDeclaration]) -> String
+    func makeStructUniqueName(for function: FunctionDeclaration, in functions: [FunctionDeclaration]) -> String
 }
 
 final class FunctionNameGeneratorImpl: FunctionNameGenerator {
@@ -36,4 +37,9 @@ final class FunctionNameGeneratorImpl: FunctionNameGenerator {
     private func isFunctionNameUnique(_ function: FunctionDeclaration, in functions: [FunctionDeclaration]) -> Bool {
         return !functions.contains(where:  { $0.name == function.name && $0 != function })
     }
+
+    func makeStructUniqueName(for function: FunctionDeclaration, in functions: [FunctionDeclaration]) -> String {
+        return makeFirstLetterCapitalized(in: makeUniqueName(for: function, in: functions))
+    }
+
 }