44 "context"
55 "strings"
66
7+ "github.com/google/go-jsonnet"
78 "github.com/google/go-jsonnet/ast"
9+ "github.com/google/go-jsonnet/toolutils"
810 "github.com/grafana/jsonnet-language-server/pkg/ast/processing"
911 "github.com/grafana/jsonnet-language-server/pkg/nodestack"
1012 position "github.com/grafana/jsonnet-language-server/pkg/position_conversion"
@@ -38,7 +40,9 @@ func (s *Server) Completion(ctx context.Context, params *protocol.CompletionPara
3840 return nil , nil
3941 }
4042
41- items := s .completionFromStack (line , searchStack )
43+ vm := s .getVM (doc .item .URI .SpanURI ().Filename ())
44+
45+ items := s .completionFromStack (line , searchStack , vm )
4246 return & protocol.CompletionList {IsIncomplete : false , Items : items }, nil
4347}
4448
@@ -52,43 +56,15 @@ func getCompletionLine(fileContent string, position protocol.Position) string {
5256 return line
5357}
5458
55- func (s * Server ) completionFromStack (line string , stack * nodestack.NodeStack ) []protocol.CompletionItem {
56- var items []protocol.CompletionItem
57-
59+ func (s * Server ) completionFromStack (line string , stack * nodestack.NodeStack , vm * jsonnet.VM ) []protocol.CompletionItem {
5860 lineWords := strings .Split (line , " " )
5961 lastWord := lineWords [len (lineWords )- 1 ]
6062
6163 indexes := strings .Split (lastWord , "." )
6264 firstIndex , indexes := indexes [0 ], indexes [1 :]
6365
64- if firstIndex == "self" && len (indexes ) > 0 {
65- fieldPrefix := indexes [0 ]
66-
67- for ! stack .IsEmpty () {
68- curr := stack .Pop ()
69-
70- switch curr := curr .(type ) {
71- case * ast.Binary :
72- stack .Push (curr .Left )
73- stack .Push (curr .Right )
74- case * ast.DesugaredObject :
75- for _ , field := range curr .Fields {
76- label := processing .FieldNameToString (field .Name )
77- // Ignore fields that don't match the prefix
78- if ! strings .HasPrefix (label , fieldPrefix ) {
79- continue
80- }
81-
82- // Ignore the current field
83- if strings .Contains (line , label + ":" ) {
84- continue
85- }
86-
87- items = append (items , createCompletionItem (label , "self." + label , protocol .FieldCompletion , field .Body ))
88- }
89- }
90- }
91- } else if len (indexes ) == 0 {
66+ if len (indexes ) == 0 {
67+ var items []protocol.CompletionItem
9268 // firstIndex is a variable (local) completion
9369 for ! stack .IsEmpty () {
9470 if curr , ok := stack .Pop ().(* ast.Local ); ok {
@@ -103,13 +79,51 @@ func (s *Server) completionFromStack(line string, stack *nodestack.NodeStack) []
10379 }
10480 }
10581 }
82+ return items
10683 }
10784
108- return items
85+ if len (indexes ) > 1 {
86+ // TODO: Support multiple indexes, the objects to search through will be the reference in the last index
87+ return nil
88+ }
89+
90+ var (
91+ objectsToSearch []* ast.DesugaredObject
92+ )
93+
94+ if firstIndex == "self" {
95+ // Search through the current stack
96+ objectsToSearch = processing .FindTopLevelObjects (stack , vm )
97+ } else {
98+ // If the index is something other than 'self', find what it refers to (Var reference) and find objects in that
99+ for ! stack .IsEmpty () {
100+ curr := stack .Pop ()
101+
102+ if targetVar , ok := curr .(* ast.Var ); ok && string (targetVar .Id ) == firstIndex {
103+ ref , _ := processing .FindVarReference (targetVar , vm )
104+
105+ switch ref := ref .(type ) {
106+ case * ast.DesugaredObject :
107+ objectsToSearch = []* ast.DesugaredObject {ref }
108+ case * ast.Import :
109+ filename := ref .File .Value
110+ objectsToSearch = processing .FindTopLevelObjectsInFile (vm , filename , string (curr .Loc ().File .DiagnosticFileName ))
111+ }
112+ break
113+ }
114+
115+ for _ , node := range toolutils .Children (curr ) {
116+ stack .Push (node )
117+ }
118+ }
119+ }
120+
121+ fieldPrefix := indexes [0 ]
122+ return createCompletionItemsFromObjects (objectsToSearch , firstIndex , fieldPrefix , line )
109123}
110124
111125func (s * Server ) completionStdLib (line string ) []protocol.CompletionItem {
112- items := []protocol.CompletionItem {}
126+ var items []protocol.CompletionItem
113127
114128 stdIndex := strings .LastIndex (line , "std." )
115129 if stdIndex != - 1 {
@@ -147,6 +161,36 @@ func (s *Server) completionStdLib(line string) []protocol.CompletionItem {
147161 return items
148162}
149163
164+ func createCompletionItemsFromObjects (objects []* ast.DesugaredObject , firstIndex , fieldPrefix , currentLine string ) []protocol.CompletionItem {
165+ var items []protocol.CompletionItem
166+ labels := make (map [string ]bool )
167+
168+ for _ , obj := range objects {
169+ for _ , field := range obj .Fields {
170+ label := processing .FieldNameToString (field .Name )
171+
172+ if labels [label ] {
173+ continue
174+ }
175+
176+ // Ignore fields that don't match the prefix
177+ if ! strings .HasPrefix (label , fieldPrefix ) {
178+ continue
179+ }
180+
181+ // Ignore the current field
182+ if strings .Contains (currentLine , label + ":" ) {
183+ continue
184+ }
185+
186+ items = append (items , createCompletionItem (label , firstIndex + "." + label , protocol .FieldCompletion , field .Body ))
187+ labels [label ] = true
188+ }
189+ }
190+
191+ return items
192+ }
193+
150194func createCompletionItem (label , detail string , kind protocol.CompletionItemKind , body ast.Node ) protocol.CompletionItem {
151195 insertText := label
152196 if asFunc , ok := body .(* ast.Function ); ok {
0 commit comments