add node types to GenericDocumentUseCase

mrjosh · Apr 12, 2024 · 75bed53 · 75bed53
1 parent 52b9ae9
commit 75bed53
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 8 deletions.
diff --git a/internal/handler/generic_document_usecase.go b/internal/handler/generic_document_usecase.go
@@ -22,11 +22,19 @@ func (h *langHandler) NewGenericDocumentUseCase(params lsp.TextDocumentPositionP
 	if node == nil {
 		return languagefeatures.GenericDocumentUseCase{}, errors.New("Could not get node for: " + params.TextDocument.URI.Filename())
 	}
+	parentNode := node.Parent()
+	var parentNodeType string
+	if parentNode != nil {
+		parentNodeType = parentNode.Type()
+	}
 	return languagefeatures.GenericDocumentUseCase{
-		Document:      doc,
-		DocumentStore: h.documents,
-		Chart:         chart,
-		Node:          node,
+		Document:       doc,
+		DocumentStore:  h.documents,
+		Chart:          chart,
+		Node:           node,
+		ParentNode:     parentNode,
+		ParentNodeType: parentNodeType,
+		NodeType:       node.Type(),
 	}, nil
 }
 

diff --git a/internal/language_features/generic_document_usecase.go b/internal/language_features/generic_document_usecase.go
@@ -7,10 +7,13 @@ import (
 )
 
 type GenericDocumentUseCase struct {
-	Document      *lsplocal.Document
-	DocumentStore *lsplocal.DocumentStore
-	Chart         *charts.Chart
-	Node          *sitter.Node
+	Document       *lsplocal.Document
+	DocumentStore  *lsplocal.DocumentStore
+	Chart          *charts.Chart
+	Node           *sitter.Node
+	NodeType       string
+	ParentNode     *sitter.Node
+	ParentNodeType string
 }
 
 func (u *GenericDocumentUseCase) NodeContent() string {