[SwiftLanguageService] CodeActions: Use asWorkspaceEdit in more places

Author: xedin

Sources/SwiftLanguageService/CodeActions/PackageManifestEdits.swift

@@ -50,16 +50,20 @@ struct PackageManifestEdits: SyntaxCodeActionProvider {
           type: type
         )
 
-        let edits = try AddPackageTarget.textRefactor(
-          syntax: scope.file,
-          in: .init(target: target)
-        )
+        guard
+          let edit = try AddPackageTarget.textRefactor(
+            syntax: scope.file,
+            in: .init(target: target)
+          ).asWorkspaceEdit(snapshot: scope.snapshot)
+        else {
+          continue
+        }
 
         actions.append(
           CodeAction(
             title: "Add \(name) target",
             kind: .refactor,
-            edit: edits.asWorkspaceEdit(snapshot: scope.snapshot)
+            edit: edit
           )
         )
       }
@@ -98,16 +102,20 @@ struct PackageManifestEdits: SyntaxCodeActionProvider {
           dependencies: [.byName(name: targetName)],
         )
 
-        let edits = try AddPackageTarget.textRefactor(
-          syntax: scope.file,
-          in: .init(target: target, testHarness: testingLibrary)
-        )
+        guard
+          let edit = try AddPackageTarget.textRefactor(
+            syntax: scope.file,
+            in: .init(target: target, testHarness: testingLibrary)
+          ).asWorkspaceEdit(snapshot: scope.snapshot)
+        else {
+          continue
+        }
 
         actions.append(
           CodeAction(
             title: "Add test target (\(libraryName))",
             kind: .refactor,
-            edit: edits.asWorkspaceEdit(snapshot: scope.snapshot)
+            edit: edit
           )
         )
       }
@@ -151,16 +159,20 @@ struct PackageManifestEdits: SyntaxCodeActionProvider {
         targets: [targetName]
       )
 
-      let edits = try AddProduct.textRefactor(
-        syntax: scope.file,
-        in: .init(product: product)
-      )
+      guard
+        let edit = try AddProduct.textRefactor(
+          syntax: scope.file,
+          in: .init(product: product)
+        ).asWorkspaceEdit(snapshot: scope.snapshot)
+      else {
+        return []
+      }
 
       return [
         CodeAction(
           title: "Add product to export this target",
           kind: .refactor,
-          edit: edits.asWorkspaceEdit(snapshot: scope.snapshot)
+          edit: edit
         )
       ]
     } catch {
@@ -175,24 +187,6 @@ struct PackageManifestEdits: SyntaxCodeActionProvider {
   ]
 }
 
-fileprivate extension [SourceEdit] {
-  /// Translate package manifest edits into a workspace edit.
-  /// `snapshot` is the latest snapshot of the `Package.swift` file.
-  func asWorkspaceEdit(snapshot: DocumentSnapshot) -> WorkspaceEdit {
-    // The edits to perform on the manifest itself.
-    let manifestTextEdits = map { edit in
-      TextEdit(
-        range: snapshot.absolutePositionRange(of: edit.range),
-        newText: edit.replacement
-      )
-    }
-
-    return WorkspaceEdit(
-      changes: [snapshot.uri: manifestTextEdits]
-    )
-  }
-}
-
 fileprivate extension SyntaxProtocol {
   // Find an enclosing call syntax expression.
   func findEnclosingCall() -> FunctionCallExprSyntax? {

Sources/SwiftLanguageService/CodeActions/SyntaxRefactoringCodeActionProvider.swift

@@ -37,26 +37,15 @@ extension SyntaxRefactoringCodeActionProvider where Self.Context == Void {
       return []
     }
 
-    let textEdits = sourceEdits.compactMap { (edit) -> TextEdit? in
-      let edit = TextEdit(
-        range: scope.snapshot.absolutePositionRange(of: edit.range),
-        newText: edit.replacement
-      )
-      if edit.isNoOp(in: scope.snapshot) {
-        return nil
-      }
-      return edit
-    }
-
-    if textEdits.isEmpty {
+    guard let workspaceEdit = sourceEdits.asWorkspaceEdit(snapshot: scope.snapshot) else {
       return []
     }
 
     return [
       CodeAction(
         title: Self.title,
         kind: .refactorInline,
-        edit: WorkspaceEdit(changes: [scope.snapshot.uri: textEdits])
+        edit: workspaceEdit
       )
     ]
   }
@@ -140,3 +129,30 @@ extension SyntaxProtocol {
     return nil
   }
 }
+
+extension [SourceEdit] {
+  /// Translate source edits into a workspace edit.
+  /// `snapshot` is the latest snapshot of the document to which these edits belong.
+  func asWorkspaceEdit(snapshot: DocumentSnapshot) -> WorkspaceEdit? {
+    let textEdits = compactMap { edit -> TextEdit? in
+      let edit = TextEdit(
+        range: snapshot.absolutePositionRange(of: edit.range),
+        newText: edit.replacement
+      )
+
+      if edit.isNoOp(in: snapshot) {
+        return nil
+      }
+
+      return edit
+    }
+
+    if textEdits.isEmpty {
+      return nil
+    }
+
+    return WorkspaceEdit(
+      changes: [snapshot.uri: textEdits]
+    )
+  }
+}