add parser and autocomplete support for "plan_name" and "key"

Author: parkuman

pkg/ast/job.go

@@ -52,6 +52,12 @@ type Job struct {
 
 	Type      string
 	TypeRange protocol.Range
+
+	PlanName      string
+	PlanNameRange protocol.Range
+
+	Key      string
+	KeyRange protocol.Range
 }
 
 func (job *Job) AddCompletionItem(label string, commitCharacters []string) {

pkg/parser/jobs.go

@@ -125,6 +125,15 @@ func (doc *YamlDocument) parseSingleJob(jobNode *sitter.Node) ast.Job {
 			case "type":
 				res.Type = doc.GetNodeText(valueNode)
 				res.TypeRange = doc.NodeToRange(child)
+
+			case "plan_name":
+				res.PlanName = doc.GetNodeText(valueNode)
+				res.PlanNameRange = doc.NodeToRange(child)
+
+			case "key":
+				res.Key = doc.GetNodeText(valueNode)
+				res.KeyRange = doc.NodeToRange(child)
+
 			}
 		}
 	})
@@ -165,4 +174,10 @@ func (doc *YamlDocument) jobCompletionItem(job ast.Job) {
 	if job.Type == "" {
 		job.AddCompletionItem("type", []string{":", " "})
 	}
+	if job.PlanName == "" {
+		job.AddCompletionItem("plan_name", []string{":", " "})
+	}
+	if job.Key == "" {
+		job.AddCompletionItem("key", []string{":", " "})
+	}
 }

pkg/services/complete_test.go

@@ -127,6 +127,16 @@ func TestComplete(t *testing.T) {
 					InsertText: "type: ",
 					Kind:       protocol.CompletionItemKindProperty,
 				},
+				{
+					Label:      "plan_name",
+					InsertText: "plan_name: ",
+					Kind:       protocol.CompletionItemKindProperty,
+				},
+				{
+					Label:      "key",
+					InsertText: "key: ",
+					Kind:       protocol.CompletionItemKindProperty,
+				},
 			},
 		},
 		{