LSP: Collect program type keyword (FuelLabs#3737)

closes FuelLabs#3712
8siceVobpu · Jan 10, 2023 · 9eac174 · 9eac174
1 parent e24eb0c
commit 9eac174
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sway-lsp/src/traverse/lexed_tree.rs b/sway-lsp/src/traverse/lexed_tree.rs
@@ -23,11 +23,13 @@ impl<'a> LexedTree<'a> {
     }
 
     pub fn parse(&self, lexed_program: &LexedProgram) {
+        insert_keyword(self.tokens, lexed_program.root.tree.kind.span());
         for item in &lexed_program.root.tree.items {
             item.value.parse(self.tokens);
         }
 
         for (.., dep) in &lexed_program.root.submodules {
+            insert_keyword(self.tokens, dep.module.tree.kind.span());
             for item in &dep.module.tree.items {
                 item.value.parse(self.tokens);
             }