@@ -42,7 +42,7 @@ import Development.IDE.Core.PluginUtils (runActionE,
4242 useWithStaleE )
4343import Development.IDE.Core.PositionMapping (idDelta , toCurrentRange )
4444import Development.IDE.Core.Rules (toIdeResult )
45- import Development.IDE.Core.RuleTypes (DocAndKindMap (.. ))
45+ import Development.IDE.Core.RuleTypes (DocAndTyThingMap (.. ))
4646import Development.IDE.Core.Shake (addPersistentRule ,
4747 getVirtualFile ,
4848 useWithStale_ )
@@ -99,7 +99,7 @@ getSemanticTokensRule :: Recorder (WithPriority SemanticLog) -> Rules ()
9999getSemanticTokensRule recorder =
100100 define (cmapWithPrio LogShake recorder) $ \ GetSemanticTokens nfp -> handleError recorder $ do
101101 (HAR {.. }) <- lift $ use_ GetHieAst nfp
102- (DKMap {getKindMap }, _) <- lift $ useWithStale_ GetDocMap nfp
102+ (DKMap {getTyThingMap }, _) <- lift $ useWithStale_ GetDocMap nfp
103103 (_, ast) <- handleMaybe LogNoAST $ listToMaybe $ Map. toList $ getAsts hieAst
104104 virtualFile <- handleMaybeM LogNoVF $ getVirtualFile nfp
105105 -- get current location from the old ones
@@ -108,7 +108,7 @@ getSemanticTokensRule recorder =
108108 let localSemanticMap = mkLocalNameSemanticFromAst names (hieKindFunMasksKind hieKind) refMap
109109 -- get imported name semantic map
110110 -- liftIO $ putStrLn $ unlines $ fmap showClearName $ nameSetElemsStable nameSet
111- let importedNameSemanticMap = foldr (getTypeExclude localSemanticMap getKindMap ) emptyNameEnv names
111+ let importedNameSemanticMap = foldr (getTypeExclude localSemanticMap getTyThingMap ) emptyNameEnv names
112112 -- let importedNameSemanticMap = computeImportedNameSemanticMap $ nameSetElemsStable nameSet
113113 let sMap = plusNameEnv_C (<>) importedNameSemanticMap localSemanticMap
114114 let rangeTokenType = extractSemanticTokensFromNames sMap spanNamesMap
@@ -120,10 +120,10 @@ getSemanticTokensRule recorder =
120120 -> Name
121121 -> NameEnv HsSemanticTokenType
122122 -> NameEnv HsSemanticTokenType
123- getTypeExclude localEnv kindMap n nameMap
123+ getTypeExclude localEnv tyThingMap n nameMap
124124 | n `elemNameEnv` localEnv = nameMap
125125 | otherwise =
126- let tyThing = lookupNameEnv kindMap n in
126+ let tyThing = lookupNameEnv tyThingMap n in
127127 maybe nameMap (extendNameEnv nameMap n) (tyThing >>= tyThingSemantic)
128128
129129-- | Persistent rule to ensure that semantic tokens doesn't block on startup
0 commit comments