uidf refactor

1 files changed, 3 insertions, 0 deletions

diff --git a/rust/src/variable.rs b/rust/src/variable.rs
index 3fdc1177..f6b0f522 100644
--- a/rust/src/variable.rs
+++ b/rust/src/variable.rs
@@ -219,6 +219,7 @@ unsafe impl CoreArrayProviderInner for NamedVariableWithType {
 pub struct UserVariableValue {
     pub variable: Variable,
     pub def_site: Location,
+    pub after: bool,
     pub value: PossibleValueSet,
 }
 
@@ -227,6 +228,7 @@ impl UserVariableValue {
         Self {
             variable: value.var.into(),
             def_site: value.defSite.into(),
+            after: value.after,
             value: PossibleValueSet::from_raw(&value.value),
         }
     }
@@ -235,6 +237,7 @@ impl UserVariableValue {
         BNUserVariableValue {
             var: value.variable.into(),
             defSite: value.def_site.into(),
+            after: value.after,
             value: PossibleValueSet::into_raw(value.value),
         }
     }