WIP

Author: Anton-Latukha

ghcide/src/Development/IDE/Core/Compile.hs

@@ -110,6 +110,8 @@ import           Development.IDE.GHC.Compat.Util   (emptyUDFM, plusUDFM)
 import qualified Language.LSP.Server               as LSP
 import qualified Language.LSP.Types                as LSP
 import           Unsafe.Coerce
+import Data.Tuple.Extra (first)
+import Data.Tuple (swap)
 
 -- | Given a string buffer, return the string (after preprocessing) and the 'ParsedModule'.
 parseModule
@@ -992,26 +994,80 @@ getDocsBatch
   :: HscEnv
   -> Module  -- ^ a moudle where the names are in scope
   -> [Name]
+  -- 2021-11-19: NOTE: Don't forget these 'Map' currently lazy.
   --  2021-11-18: NOTE: Map Int would become IntMap if next GHCs.
   -> IO (Either ErrorMessages (Map.Map Name (Either GetDocsFailure (Maybe HsDocString, Maybe (Map.Map Int HsDocString)))))
   -- ^ Return a 'Map' of 'Name's to 'Either' (no docs messages) (general doc body & arg docs)
 getDocsBatch hsc_env _mod _names = do
-    ((_warns,errs), res) <- initTc hsc_env HsSrcFile False _mod fakeSpan $ Map.fromList <$> traverse findNameInfo _names
+    ((_warns,errs), res) <- initTc hsc_env HsSrcFile False _mod fakeSpan $ Map.fromList <$> traverse undefined undefined
     pure $ maybeToEither errs res
  where
-    findNameInfo :: Name -> IOEnv (Env TcGblEnv TcLclEnv) (Name, Either GetDocsFailure (Maybe HsDocString, Maybe (Map.Map Int HsDocString)))
-    findNameInfo name =
-        case nameModule_maybe name of
-            Nothing -> return (name, Left $ NameHasNoModule name)
-            Just mod -> do
+    -- fun1 :: (Map Name (IOEnv (Env TcGblEnv TcLclEnv) (Maybe HsDocString, Maybe (Map Int HsDocString))))
+    -- fun1 = Map.fromList fun
+
+    fun :: IOEnv (Env TcGblEnv TcLclEnv) (Map Name (Maybe HsDocString, Maybe (Map Int HsDocString)))
+    fun = _ getAskedIfaceDocs loadIfaces
+
+    getAskedIfaceDocs :: ((Map Name HsDocString, Map Name (Map Int HsDocString)), [Name]) -> Map Name (Maybe HsDocString, Maybe (Map Int HsDocString))
+    getAskedIfaceDocs a = Map.fromList $
+        fun2 <$> snd a
+      where
+       fun2 :: Name -> (Name, (Maybe HsDocString, Maybe (Map Int HsDocString)))
+       fun2 n = (n, bimap (Map.lookup n) (Map.lookup n) $ fst a)
+
+    loadIfaces :: IOEnv (Env TcGblEnv TcLclEnv) [((Map Name HsDocString, Map Name (Map Int HsDocString)), [Name])]
+    loadIfaces = fun3 (fmap (first getIfaceGenNArgDocMaps) loadModules)
+      where
+          fun3 :: [(env ms, ns)] -> env [(ms, ns)]
+          fun3 a = (fmap . fmap) swap $ sequenceA $ fmap swap a
+
+    getIfaceGenNArgDocMaps :: TcRn ModIface -> IOEnv (Env TcGblEnv TcLclEnv) (Map Name HsDocString, Map Name (Map Int HsDocString))
+    getIfaceGenNArgDocMaps mi = do
+        ModIface
+            { mi_doc_hdr = mb_doc_hdr
+            , mi_decl_docs = DeclDocMap dmap
+            , mi_arg_docs = ArgDocMap amap
+            }
+            <- mi
+        pure $
+            if isNothing mb_doc_hdr && Map.null dmap && Map.null amap
+            then error "Instead of 'error' here handle 'NoDocsInIface mod $ isCompiled name' case"
+            else (dmap, amap)
+
+    loadModules :: [(TcRn ModIface, [Name])]
+    loadModules = fmap loadAvailableModules namesGroupedByModule
+     where
+        loadAvailableModules :: (Module, [Name]) -> (TcRn ModIface, [Name])
+        loadAvailableModules = first loadModuleInterfaceOnce
+
+
+        loadModuleInterfaceOnce :: Module -> TcRn ModIface
+        loadModuleInterfaceOnce =
+            loadModuleInterface "getModuleInterface"
+
+        namesGroupedByModule :: [(Module, [Name])]
+        namesGroupedByModule =
+            groupSort $ fmap (first (fromMaybe (error "Instead of 'error' handle here 'NameHasNoModule' case") . nameModule_maybe) . dupe) _names
+
+        -- modulesPartitionedOnAvalability :: [(Either (Name -> GetDocsFailure) Module, [Name])]
+        -- modulesPartitionedOnAvalability = fmap partitionOnModuleAvalibility namesGroupedByModule
+
+        -- partitionOnModuleAvalibility :: (Maybe Module, [Name]) -> (Either (Name -> GetDocsFailure) Module, [Name])
+        -- partitionOnModuleAvalibility =
+        --     first (maybeToEither NameHasNoModule)
+
+
+    --  2021-11-18: NOTE: This code initially was taken from: https://hackage.haskell.org/package/ghc-9.2.1/docs/src/GHC.Runtime.Eval.html#getDocs
+    findNameInfo :: Maybe Module -> Name -> IOEnv (Env TcGblEnv TcLclEnv) (Name, Either GetDocsFailure (Maybe HsDocString, Maybe (Map.Map Int HsDocString)))
+    findNameInfo Nothing name = return (name, Left $ NameHasNoModule name)
+    findNameInfo (Just mod) name = do
               ModIface
                 { mi_doc_hdr = mb_doc_hdr
                 , mi_decl_docs = DeclDocMap dmap
                 , mi_arg_docs = ArgDocMap amap
                 }
                 <- loadModuleInterface "getModuleInterface" mod
               pure . (name,) $
-                --  2021-11-17: NOTE: one does not simply check into Mordor (not 1 mode)
                 if isNothing mb_doc_hdr && Map.null dmap && Map.null amap
                 then Left $ NoDocsInIface mod $ isCompiled name
                 else Right (Map.lookup name dmap, Map.lookup name amap)