AST::set_source affects internal module id

2021-01-03 09:03:41 +07:00
parent ef48f47b74
commit acbf5d8723
1 changed files with 5 additions and 1 deletions
--- a/src/ast.rs
+++ b/src/ast.rs
@@ -224,7 +224,11 @@ impl AST {
    /// Set the source.
    #[inline(always)]
    pub fn set_source<S: Into<ImmutableString>>(&mut self, source: Option<S>) {
-        self.source = source.map(|s| s.into())
+        self.source = source.map(|s| s.into());
+
+        if let Some(module) = Shared::get_mut(&mut self.functions) {
+            module.set_id(self.source.clone());
+        }
    }
    /// Get the statements.
    #[cfg(not(feature = "internals"))]