Better formatting and expression handling

test/lsp_recursive_index_test.go

@@ -0,0 +1,88 @@
+package integration
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/marte-community/marte-dev-tools/internal/index"
+	"github.com/marte-community/marte-dev-tools/internal/lsp"
+)
+
+func TestLSPRecursiveIndexing(t *testing.T) {
+	// Setup directory structure
+	rootDir, err := os.MkdirTemp("", "lsp_recursive")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(rootDir)
+
+	// root/main.marte
+	mainContent := `
+#package App
++Main = {
+    Ref = SubComp
+}
+`
+	if err := os.WriteFile(filepath.Join(rootDir, "main.marte"), []byte(mainContent), 0644); err != nil {
+		t.Fatal(err)
+	}
+
+	// root/subdir/sub.marte
+	subDir := filepath.Join(rootDir, "subdir")
+	if err := os.Mkdir(subDir, 0755); err != nil {
+		t.Fatal(err)
+	}
+	subContent := `
+#package App
++SubComp = { Class = Component }
+`
+	if err := os.WriteFile(filepath.Join(subDir, "sub.marte"), []byte(subContent), 0644); err != nil {
+		t.Fatal(err)
+	}
+
+	// Initialize LSP
+	lsp.Tree = index.NewProjectTree()
+	lsp.Documents = make(map[string]string)
+
+	// Simulate ScanDirectory
+	if err := lsp.Tree.ScanDirectory(rootDir); err != nil {
+		t.Fatalf("ScanDirectory failed: %v", err)
+	}
+	lsp.Tree.ResolveReferences()
+
+	// Check if SubComp is in the tree
+	// Root -> App -> SubComp
+	appNode := lsp.Tree.Root.Children["App"]
+	if appNode == nil {
+		t.Fatal("App package not found")
+	}
+
+	subComp := appNode.Children["SubComp"]
+	if subComp == nil {
+		t.Fatal("SubComp not found in tree (recursive scan failed)")
+	}
+
+	mainURI := "file://" + filepath.Join(rootDir, "main.marte")
+
+	// Definition Request
+	params := lsp.DefinitionParams{
+		TextDocument: lsp.TextDocumentIdentifier{URI: mainURI},
+		Position:     lsp.Position{Line: 3, Character: 12},
+	}
+
+	res := lsp.HandleDefinition(params)
+	if res == nil {
+		t.Fatal("Definition not found for SubComp")
+	}
+
+	locs, ok := res.([]lsp.Location)
+	if !ok || len(locs) == 0 {
+		t.Fatal("Expected location list")
+	}
+
+	expectedFile := filepath.Join(subDir, "sub.marte")
+	if locs[0].URI != "file://"+expectedFile {
+		t.Errorf("Expected definition in %s, got %s", expectedFile, locs[0].URI)
+	}
+}