apply review

Author: mfbmina

taskfile/ast/prompt.go

@@ -1,21 +1,29 @@
 package ast
 
+import (
+	"gopkg.in/yaml.v3"
+
+	"github.com/go-task/task/v3/errors"
+)
+
 type Prompt []string
 
-func (p *Prompt) UnmarshalYAML(unmarshal func(any) error) error {
-	var m []string
-	err := unmarshal(&m)
-	if err == nil {
-		*p = m
+func (p *Prompt) UnmarshalYAML(node *yaml.Node) error {
+	switch node.Kind {
+	case yaml.ScalarNode:
+		var str string
+		if err := node.Decode(&str); err != nil {
+			return errors.NewTaskfileDecodeError(err, node)
+		}
+		*p = []string{str}
+		return nil
+	case yaml.SequenceNode:
+		var list []string
+		if err := node.Decode(&list); err != nil {
+			return errors.NewTaskfileDecodeError(err, node)
+		}
+		*p = list
 		return nil
 	}
-
-	var s string
-	err = unmarshal(&s)
-	if err != nil {
-		return err
-	}
-
-	*p = []string{s}
-	return nil
+	return errors.NewTaskfileDecodeError(nil, node).WithTypeMessage("prompt")
 }

website/static/schema.json

@@ -60,10 +60,17 @@
         },
         "prompt": {
           "description": "One or more prompts that will be presented before a task is run. Declining will cancel running the current and any subsequent tasks.",
-          "type": "array",
-          "items": {
-            "type": "string"
-          }
+          "oneOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "array",
+              "items": {
+                "type": "string"
+              }
+            }
+          ]
         },
         "summary": {
           "description": "A longer description of the task. This is displayed when calling `task --summary [task]`.",