Move code to post processing middleware

Author: intitni

Core/Sources/SuggestionService/SuggestionService.swift

@@ -84,16 +84,7 @@ public extension SuggestionService {
                 }
             }
 
-            var result = try await getSuggestion(request, workspaceInfo)
-            if !request.lines.isEmpty {
-                for index in result.indices {
-                    let range = result[index].range
-                    let lowerBound = max(0, range.start.line)
-                    let upperBound = max(lowerBound, min(request.lines.count - 1, range.end.line))
-                    result[index].replacingLines = Array(request.lines[lowerBound...upperBound])
-                }
-            }
-            return result
+            return try await getSuggestion(request, workspaceInfo)
         } catch let error as SuggestionServiceError {
             throw error
         } catch {

Tool/Sources/SuggestionProvider/PostProcessingSuggestionServiceMiddleware.swift

@@ -17,13 +17,25 @@ public struct PostProcessingSuggestionServiceMiddleware: SuggestionServiceMiddle
             Self.removeTrailingWhitespacesAndNewlines(&suggestion)
             Self.removeRedundantClosingParenthesis(&suggestion, lines: request.lines)
             if !Self.checkIfSuggestionHasNoEffect(suggestion, request: request) { return nil }
+            Self.injectReplacingLines(&suggestion, request: request)
             return suggestion
         }
     }
 
     static func removeTrailingWhitespacesAndNewlines(_ suggestion: inout CodeSuggestion) {
         suggestion.text = suggestion.text.removedTrailingWhitespacesAndNewlines()
     }
+    
+    static func injectReplacingLines(
+        _ suggestion: inout CodeSuggestion,
+        request: SuggestionRequest
+    ) {
+        guard !request.lines.isEmpty else { return }
+        let range = suggestion.range
+        let lowerBound = max(0, range.start.line)
+        let upperBound = max(lowerBound, min(request.lines.count - 1, range.end.line))
+        suggestion.replacingLines = Array(request.lines[lowerBound...upperBound])
+    }
 
     /// Remove the parenthesis in the last line of the suggestion if
     /// - It contains only closing parenthesis