add realSrcSpanToCodePointRange, realSrcLocToCodePointPosition to Development.IDE.GHC.Error, to put them beside realSrcSpanToRange, realSrcLocToPosition

Author: soulomoon

plugins/hls-semantic-tokens-plugin/src/Ide/Plugin/SemanticTokens/Internal.hs

@@ -94,7 +94,7 @@ semanticTokensFull state _ param = do
 -- Visible names from 'tmrRenamed'
 --
 -- It then combines this information to compute the semantic tokens for the file.
-getSemanticTokensRule :: Recorder (WithPriority SemanticLog) ->  Rules ()
+getSemanticTokensRule :: Recorder (WithPriority SemanticLog) -> Rules ()
 getSemanticTokensRule recorder =
   define (cmapWithPrio LogShake recorder) $ \GetSemanticTokens nfp -> handleError recorder $ do
     (HAR {..}) <- lift $ use_ GetHieAst nfp

plugins/hls-semantic-tokens-plugin/src/Ide/Plugin/SemanticTokens/Mappings.hs

@@ -20,7 +20,7 @@ import qualified Data.Set                        as Set
 import           Data.Text                       (Text, unpack)
 import           Development.IDE                 (HieKind (HieFresh, HieFromDisk))
 import           Development.IDE.GHC.Compat
-import           Ide.Plugin.SemanticTokens.Types hiding (tokens)
+import           Ide.Plugin.SemanticTokens.Types
 import           Ide.Plugin.SemanticTokens.Utils (mkRange, recoverFunMaskArray)
 import           Language.LSP.Protocol.Types     (LspEnum (knownValues),
                                                   SemanticTokenAbsolute (SemanticTokenAbsolute),

plugins/hls-semantic-tokens-plugin/src/Ide/Plugin/SemanticTokens/Query.hs

@@ -19,11 +19,11 @@ import           Data.Text                            (Text)
 import           Development.IDE.Core.PositionMapping (PositionMapping,
                                                        toCurrentRange)
 import           Development.IDE.GHC.Compat
+import           Development.IDE.GHC.Error            (realSrcSpanToCodePointRange)
 import           Ide.Plugin.SemanticTokens.Mappings
 import           Ide.Plugin.SemanticTokens.Types      (HieFunMaskKind,
                                                        HsSemanticTokenType,
                                                        NameSemanticMap)
-import           Ide.Plugin.SemanticTokens.Utils      (realSrcSpanToCodePointRange)
 import           Language.LSP.Protocol.Types
 import           Language.LSP.VFS                     (VirtualFile,
                                                        codePointRangeToRange)
@@ -50,6 +50,8 @@ nameNameSemanticFromHie hieKind rm ns = do
     nameSemanticFromRefMap rm' name' = do
       spanInfos <- -- traceShow ("getting spans:", nameString) $
         Map.lookup (Right name') rm'
+    --   let combinedFunction x = (identType . snd) x <|> (identInfo . snd) x
+    --   let result = foldMap (typeSemantic hieKind) $ listToMaybe $ mapMaybe combinedFunction spanInfos
       let typeTokenType = foldMap (typeSemantic hieKind) $ listToMaybe $ mapMaybe (identType . snd) spanInfos
       contextInfoTokenType <- foldMap (contextInfosMaybeTokenType . identInfo . snd) spanInfos
       fold [typeTokenType, Just contextInfoTokenType]

plugins/hls-semantic-tokens-plugin/src/Ide/Plugin/SemanticTokens/Utils.hs

@@ -123,18 +123,7 @@ recoverFunMaskArray flattened = unflattened
     go (HTyConApp _ _)             = False
 
 
-realSrcSpanToCodePointRange :: RealSrcSpan -> CodePointRange
-realSrcSpanToCodePointRange real =
-  CodePointRange (realSrcLocToCodePointPosition $ Compat.realSrcSpanStart real)
-        (realSrcLocToCodePointPosition $ Compat.realSrcSpanEnd   real)
-
-
-realSrcLocToCodePointPosition :: RealSrcLoc -> CodePointPosition
-realSrcLocToCodePointPosition real =
-  CodePointPosition (fromIntegral $ srcLocLine real - 1) (fromIntegral $ srcLocCol real - 1)
-
 -- rangeToCodePointRange
--- mkRange :: Int -> Int -> Int -> Range
 mkRange :: (Integral a1, Integral a2) => a1 -> a2 -> a2 -> Range
 mkRange startLine startCol len =
     Range (Position (fromIntegral startLine) (fromIntegral startCol)) (Position (fromIntegral startLine) (fromIntegral $ startCol + len))