add semantic location

gravataLonga · Jul 17, 2022 · c96264e · c96264e
1 parent 31e570e
commit c96264e
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 0 deletions.
diff --git a/semantic/semantic.go b/semantic/semantic.go
@@ -82,6 +82,7 @@ func (s *Semantic) expectIdentifierDeclare(ident *ast.Identifier) bool {
 		return false
 	}
 
+	s.localVariables[ident] = s.scopeStack.Size()
 	return true
 }
 

diff --git a/semantic/stack.go b/semantic/stack.go
@@ -38,3 +38,7 @@ func (s *Scope) Put(name string, ready bool) {
 func (s *Stack) Get(index int) *Scope {
 	return (*s)[index]
 }
+
+func (s *Stack) Size() int {
+	return len(*s)
+}
diff --git a/semantic/stack_test.go b/semantic/stack_test.go
@@ -104,3 +104,31 @@ func TestStack_Get(t *testing.T) {
 		t.Fatalf("Unable to get stack by index")
 	}
 }
+
+func TestStack_Size(t *testing.T) {
+	tests := []struct {
+		stack Stack
+		size  int
+	}{
+		{
+			Stack{},
+			0,
+		},
+		{
+			Stack{&Scope{}},
+			1,
+		},
+		{
+			Stack{&Scope{}, &Scope{}},
+			2,
+		},
+	}
+
+	for i, tt := range tests {
+		t.Run(fmt.Sprintf("TestStack_Size[%d]", i), func(t *testing.T) {
+			if tt.stack.Size() != tt.size {
+				t.Errorf("Expected %d, got: %d", tt.size, tt.stack.Size())
+			}
+		})
+	}
+}