Follow changes in rewriteToEdit

Author: pepeiborra

ghcide/src/Development/IDE/Plugin/CodeAction/ExactPrint.hs

@@ -10,7 +10,7 @@ module Development.IDE.Plugin.CodeAction.ExactPrint
     -- * Utilities
     appendConstraint,
     extendImport,
-  )
+  rewriteToWEdit)
 where
 
 import Control.Applicative
@@ -32,6 +32,7 @@ import OccName
 import Outputable (ppr, showSDocUnsafe)
 import Data.Generics (listify)
 import Development.IDE.Spans.Common (showNameWithoutUniques)
+import GHC.Exts (IsList (fromList))
 
 ------------------------------------------------------------------------------
 
@@ -61,6 +62,15 @@ rewriteToEdit dflags anns (Rewrite dst f) = do
                 ]
   pure editMap
 
+rewriteToWEdit :: DynFlags -> Uri -> Anns -> Rewrite -> Either String WorkspaceEdit
+rewriteToWEdit dflags uri anns r = do
+    edits <- rewriteToEdit dflags anns r
+    return $
+        WorkspaceEdit
+            { _changes = Just (fromList [(uri, List edits)])
+            , _documentChanges = Nothing
+            }
+
 srcSpanToRange :: SrcSpan -> Maybe Range
 srcSpanToRange (UnhelpfulSpan _) = Nothing
 srcSpanToRange (RealSrcSpan real) = Just $ realSrcSpanToRange real

ghcide/src/Development/IDE/Plugin/Completions.hs

@@ -178,7 +178,7 @@ extendImportHandler' ideState ExtendImport {..}
       imp <- liftMaybe $ find (isWantedModule wantedModule) imps
       wedit <-
         liftEither $
-          rewriteToEdit df doc (annsA ps) $
+          rewriteToWEdit df doc (annsA ps) $
             extendImport (T.unpack <$> thingParent) (T.unpack newThing) imp
       return (WorkspaceApplyEdit, ApplyWorkspaceEditParams wedit)
   | otherwise =