feat(lsp): Add hover for submodules (#2028)

grain-lang · Feb 22, 2024 · 85d8768 · 85d8768
1 parent 53a33e6
commit 85d8768
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/compiler/src/language_server/hover.re b/compiler/src/language_server/hover.re
@@ -182,7 +182,7 @@ let process =
         ~range=Utils.loc_to_range(loc),
         exception_declaration_lens(ident, ext),
       )
-    | [Module({path, decl, loc}), ..._] =>
+    | [Module({decl, loc}), ..._] =>
       send_hover(~id, ~range=Utils.loc_to_range(loc), module_lens(decl))
     | [Include({env, path, loc}), ..._] =>
       send_hover(

diff --git a/compiler/src/language_server/sourcetree.re b/compiler/src/language_server/sourcetree.re
@@ -509,6 +509,26 @@ module Sourcetree: Sourcetree = {
         };
         let enter_toplevel_stmt = stmt => {
           switch (stmt.ttop_desc) {
+          | TTopModule(decl) =>
+            let path = Path.PIdent(decl.tmod_id);
+            try({
+              let mod_decl = Env.find_module(path, None, stmt.ttop_env);
+              segments :=
+                [
+                  (
+                    loc_to_interval(stmt.ttop_loc),
+                    Module({
+                      path,
+                      decl: mod_decl,
+                      loc: stmt.ttop_loc,
+                      definition: None,
+                    }),
+                  ),
+                  ...segments^,
+                ];
+            }) {
+            | Not_found => ()
+            };
           | TTopInclude(inc) =>
             segments :=
               [