Limit CodeActions within passed range

Return only actions within range specified by client.

Author: aufarg

hls-plugin-api/src/Ide/PluginUtils.hs

@@ -19,7 +19,7 @@ module Ide.PluginUtils
     mkLspCommand,
     mkLspCmdId,
     getPid,
-  allLspCmdIds,allLspCmdIds',installSigUsr1Handler, subRange)
+  allLspCmdIds,allLspCmdIds',installSigUsr1Handler, subRange, positionInRange)
 where
 
 

plugins/hls-hlint-plugin/src/Ide/Plugin/Hlint.hs

@@ -268,20 +268,20 @@ getHlintSettingsRule usage =
 -- ---------------------------------------------------------------------
 
 codeActionProvider :: PluginMethodHandler IdeState TextDocumentCodeAction
-codeActionProvider ideState plId (CodeActionParams _ _ docId _ context) = Right . LSP.List . map InR <$> liftIO getCodeActions
+codeActionProvider ideState plId (CodeActionParams _ _ docId caRange context) = Right . LSP.List . map InR <$> liftIO getCodeActions
   where
-
     getCodeActions = do
         diags <- getDiagnostics ideState
         let docNfp = toNormalizedFilePath' <$> uriToFilePath' (docId ^. LSP.uri)
-            numHintsInDoc = length
-              [d | (nfp, _, d) <- diags
+            numHintsInRange = length
+              [d | (nfp, _, d@LSP.Diagnostic{_range = LSP.Range{_start = start}}) <- diags
                  , validCommand d
                  , Just nfp == docNfp
+                 , positionInRange start caRange
               ]
         -- We only want to show the applyAll code action if there is more than 1
         -- hint in the current document
-        if numHintsInDoc > 1 then do
+        if numHintsInRange > 1 then do
           pure $ applyAllAction:applyOneActions
         else
           pure applyOneActions

test/functional/FunctionalCodeAction.hs

@@ -170,6 +170,22 @@ hlintTests = testGroup "hlint suggestions" [
 
     , testCase "apply-refact preserve regular comments" $ runHlintSession "" $ do
         testRefactor "ApplyRefact6.hs" "Redundant bracket" expectedComments
+
+    , testCase "applyAll is shown only when there is at least one diagnostic in range" $  runHlintSession "" $ do
+        doc <- openDoc "ApplyRefact7.hs" "haskell"
+        _ <- waitForDiagnosticsFromSource doc "hlint"
+
+        firstLine <- map fromAction <$> getCodeActions doc (mkRange 0 0 0 0)
+        secondLine <- map fromAction <$> getCodeActions doc (mkRange 1 0 1 0)
+        thirdLine <- map fromAction <$> getCodeActions doc (mkRange 2 0 2 0)
+        multiLine <- map fromAction <$> getCodeActions doc (mkRange 0 0 2 0)
+
+        let hasApplyAll = isJust . find (\ca -> "Apply all hints" `T.isSuffixOf` (ca ^. L.title))
+
+        liftIO $ not (hasApplyAll firstLine) @? "There is no apply all code action"
+        liftIO $ not (hasApplyAll secondLine) @? "There is no apply all code action"
+        liftIO $ not (hasApplyAll thirdLine) @? "There is an apply all code action"
+        liftIO $ hasApplyAll multiLine @? "There is no apply all code action"
     ]
     where
         runHlintSession :: FilePath -> Session a -> IO a

test/testdata/hlint/ApplyRefact7.hs

@@ -0,0 +1,3 @@
+f = (1)
+g = (1)
+