Add trait attributes to TyProgram (FuelLabs#3036)

Closes FuelLabs#3035
8siceVobpu · Oct 14, 2022 · ce32f69 · ce32f69
1 parent d8485d6
commit ce32f69
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/sway-core/src/language/ty/declaration/trait.rs b/sway-core/src/language/ty/declaration/trait.rs
@@ -4,6 +4,7 @@ use sway_types::Ident;
 use crate::{
     language::{parsed, ty::*, Visibility},
     type_system::*,
+    AttributesMap,
 };
 
 #[derive(Clone, Debug, Derivative)]
@@ -19,6 +20,7 @@ pub struct TyTraitDeclaration {
     pub(crate) methods: Vec<parsed::FunctionDeclaration>,
     pub(crate) supertraits: Vec<parsed::Supertrait>,
     pub visibility: Visibility,
+    pub attributes: AttributesMap,
 }
 
 impl CopyTypes for TyTraitDeclaration {

diff --git a/sway-core/src/semantic_analysis/ast_node/declaration/trait.rs b/sway-core/src/semantic_analysis/ast_node/declaration/trait.rs
@@ -80,6 +80,7 @@ impl ty::TyTraitDeclaration {
             methods: trait_decl.methods.to_vec(),
             supertraits: trait_decl.supertraits.to_vec(),
             visibility: trait_decl.visibility,
+            attributes: trait_decl.attributes,
         };
         ok(typed_trait_decl, warnings, errors)
     }