Fix action removes ticks from TemplateHaskellQuotes (#628)

Author: bendo

plugins/hls-refactor-plugin/src/Development/IDE/Plugin/CodeAction.hs

@@ -1771,15 +1771,17 @@ extractDoesNotExportModuleName x
 
 
 mkRenameEdit :: Maybe T.Text -> Range -> T.Text -> TextEdit
-mkRenameEdit contents range name =
-    if maybeIsInfixFunction == Just True
-      then TextEdit range ("`" <> name <> "`")
-      else TextEdit range name
+mkRenameEdit contents range name
+    | maybeIsInfixFunction == Just True = TextEdit range ("`" <> name <> "`")
+    | maybeIsTemplateFunction == Just True = TextEdit range ("'" <> name)
+    | otherwise = TextEdit range name
   where
     maybeIsInfixFunction = do
       curr <- textInRange range <$> contents
       pure $ "`" `T.isPrefixOf` curr && "`" `T.isSuffixOf` curr
-
+    maybeIsTemplateFunction = do
+      curr <- textInRange range <$> contents
+      pure $ "'" `T.isPrefixOf` curr
 
 -- | Extract the type and surround it in parentheses except in obviously safe cases.
 --

plugins/hls-refactor-plugin/test/Main.hs

@@ -609,6 +609,28 @@ renameActionTests = testGroup "rename actions"
             , "foo x y = x `monus` y"
             ]
       liftIO $ expectedContentAfterAction @=? contentAfterAction
+  , testSession "change template function" $ do
+      let content = T.unlines
+            [ "{-# LANGUAGE TemplateHaskellQuotes #-}"
+            , "module Testing where"
+            , "import Language.Haskell.TH (Name)"
+            , "foo :: Name"
+            , "foo = 'bread"
+            ]
+      doc <- createDoc "Testing.hs" "haskell" content
+      diags <- waitForDiagnostics
+      actionsOrCommands <- getCodeActions doc (Range (Position 4 6) (Position 4 12))
+      [fixTypo] <- pure [action | InR action@CodeAction{ _title = actionTitle } <- actionsOrCommands, "break" `T.isInfixOf` actionTitle ]
+      executeCodeAction fixTypo
+      contentAfterAction <- documentContents doc
+      let expectedContentAfterAction = T.unlines
+            [ "{-# LANGUAGE TemplateHaskellQuotes #-}"
+            , "module Testing where"
+            , "import Language.Haskell.TH (Name)"
+            , "foo :: Name"
+            , "foo = 'break"
+            ]
+      liftIO $ expectedContentAfterAction @=? contentAfterAction
   ]
 
 typeWildCardActionTests :: TestTree