👷 Populate env with contract declarations

analyzer/analyzer.go

@@ -94,11 +94,29 @@ func (a *Analyzer) discoverSymbols(node ast.Node, env *symbols.Environment) {
 		for _, decl := range n.Declarations {
 			a.discoverSymbols(decl, env)
 		}
+	case *ast.ContractDeclaration:
+		a.populateContractDeclaration(n, env)
 	case *ast.FunctionDeclaration:
 		a.populateFunctionDeclaration(n, env)
 	}
 }
 
+func (a *Analyzer) populateContractDeclaration(
+	node *ast.ContractDeclaration, env *symbols.Environment) {
+	baseSymbol := symbols.BaseSymbol{
+		Name:       node.Name.Value,
+		SourceFile: a.currentFile.SourceFile,
+		Offset:     node.Pos,
+		AstNode:    node,
+	}
+
+	contractSymbol := &symbols.Contract{
+		BaseSymbol: baseSymbol,
+	}
+
+	env.Set(node.Name.Value, contractSymbol)
+}
+
 func (a *Analyzer) populateFunctionDeclaration(
 	node *ast.FunctionDeclaration, env *symbols.Environment) {
 	baseSymbol := symbols.BaseSymbol{

analyzer/analyzer_test.go

@@ -4,7 +4,7 @@ import (
 	"testing"
 )
 
-func Test_PopulateSymbols_GetSymbolsByName(t *testing.T) {
+func Test_DiscoverSymbols_GetSymbolsByName(t *testing.T) {
 	testContractPath := "testdata/foundry/src/002_SimpleCounter.sol"
 	analyzer := Analyzer{}
 	analyzer.Init(testContractPath)
@@ -15,6 +15,7 @@ func Test_PopulateSymbols_GetSymbolsByName(t *testing.T) {
 	tests := []struct {
 		expectedSymbolName string
 	}{
+		{"Counter"},
 		{"increment"},
 		{"decrement"},
 		{"reset"},