LSP: Collect supertraits from the ParseProgram (FuelLabs#3739)

partially addresses FuelLabs#3723
Kristal1523 · Jan 10, 2023 · 565662b · 565662b
1 parent 50c8f1e
commit 565662b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/sway-core/src/language/parsed/declaration/trait.rs b/sway-core/src/language/parsed/declaration/trait.rs
@@ -10,7 +10,7 @@ pub struct TraitDeclaration {
     pub attributes: transform::AttributesMap,
     pub interface_surface: Vec<TraitFn>,
     pub methods: Vec<FunctionDeclaration>,
-    pub(crate) supertraits: Vec<Supertrait>,
+    pub supertraits: Vec<Supertrait>,
     pub visibility: Visibility,
     pub span: Span,
 }

diff --git a/sway-lsp/src/traverse/parsed_tree.rs b/sway-lsp/src/traverse/parsed_tree.rs
@@ -136,6 +136,16 @@ impl<'a> ParsedTree<'a> {
                 for func_dec in &trait_decl.methods {
                     self.handle_function_declation(func_dec);
                 }
+
+                for supertrait in &trait_decl.supertraits {
+                    self.tokens.insert(
+                        to_ident_key(&supertrait.name.suffix),
+                        Token::from_parsed(
+                            AstToken::Declaration(declaration.clone()),
+                            SymbolKind::Trait,
+                        ),
+                    );
+                }
             }
             Declaration::StructDeclaration(struct_dec) => {
                 self.tokens.insert(