eval-plugin for GHC 9.2: find all comments

Author: guibou

plugins/hls-eval-plugin/src/Ide/Plugin/Eval/Rules.hs

@@ -58,12 +58,20 @@ queueForEvaluation ide nfp = do
 
 #if MIN_VERSION_ghc(9,2,0)
 getAnnotations :: Development.IDE.GHC.Compat.Located HsModule -> [LEpaComment]
-getAnnotations (L _ (HsModule { hsmodAnn = anns'})) = priorComments $ epAnnComments anns'
+getAnnotations (L _ m@(HsModule { hsmodAnn = anns'})) = let annComments = epAnnComments anns'
+  in priorComments annComments
+     <> concatMap getCommentsForDecl (hsmodDecls m)
+     <> concatMap getCommentsForDecl (hsmodImports m)
+
+getCommentsForDecl :: GenLocated (SrcSpanAnn' (EpAnn ann)) e
+                            -> [LEpaComment]
+getCommentsForDecl (L (SrcSpanAnn (EpAnn _ _ cs) _) _) = priorComments cs <> getFollowingComments cs
+getCommentsForDecl (L (SrcSpanAnn (EpAnnNotUsed) _) _) = []
 
 apiAnnComments' :: ParsedModule -> [SrcLoc.RealLocated EpaCommentTok]
 apiAnnComments' pm = do
-  L span' (EpaComment c span) <- getAnnotations $ pm_parsed_source pm
-  pure (L (anchor span') c)
+  L span (EpaComment c _) <- getAnnotations $ pm_parsed_source pm
+  pure (L (anchor span) c)
 
 pattern RealSrcSpanAlready :: SrcLoc.RealSrcSpan -> SrcLoc.RealSrcSpan
 pattern RealSrcSpanAlready x = x