libnixf/Sema: reduce range of with def (#397)

nix-community · Apr 9, 2024 · 9de8eba · 9de8eba
1 parent 0c2cba7
commit 9de8eba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/libnixf/src/Sema/VariableLookup.cpp b/libnixf/src/Sema/VariableLookup.cpp
@@ -206,7 +206,7 @@ void VariableLookupAnalysis::dfs(const ExprWith &With,
                                  const std::shared_ptr<EnvNode> &Env) {
   auto NewEnv = std::make_shared<EnvNode>(Env, EnvNode::DefMap{}, &With);
   if (!WithDefs.contains(&With))
-    WithDefs.insert({&With, std::make_shared<Definition>(&With)});
+    WithDefs.insert({&With, std::make_shared<Definition>(&With.kwWith())});
 
   if (With.with())
     dfs(*With.with(), Env);

diff --git a/libnixf/test/Sema/VariableLookup.cpp b/libnixf/test/Sema/VariableLookup.cpp
@@ -120,6 +120,8 @@ TEST_F(VLATest, LookupWith) {
   VLAResult Result = VLA.query(Var);
 
   ASSERT_EQ(Result.Kind, VLAResultKind::FromWith);
+  ASSERT_EQ(Result.Def->syntax()->range().lCur().column(), 0);
+  ASSERT_EQ(Result.Def->syntax()->range().rCur().column(), 4);
 }
 
 TEST_F(VLATest, LivenessRec) {