Add JQValue interface to support custom values

cli/encoder.go

@@ -9,6 +9,8 @@ import (
 	"sort"
 	"strconv"
 	"unicode/utf8"
+
+	"github.com/wader/gojq"
 )
 
 type encoder struct {
@@ -66,6 +68,10 @@ func (e *encoder) encode(v any) error {
 		if err := e.encodeObject(v); err != nil {
 			return err
 		}
+	case gojq.JQValue:
+		if err := e.encode(v.JQValueToGoJQ()); err != nil {
+			return err
+		}
 	default:
 		panic(fmt.Sprintf("invalid type: %[1]T (%[1]v)", v))
 	}

debug.go

@@ -198,6 +198,8 @@ func debugValue(v any) string {
 		return fmt.Sprintf("gojq.Iter(%#v)", v)
 	case []pathValue:
 		return fmt.Sprintf("[]gojq.pathValue(%v)", v)
+	case JQValue:
+		return fmt.Sprintf("gojq.JQValue(%s)", Preview(v))
 	case [2]int:
 		return fmt.Sprintf("[%d,%d]", v[0], v[1])
 	case [3]int:

encoder.go

@@ -68,6 +68,8 @@ func (e *encoder) encode(v any) {
 		e.encodeArray(v)
 	case map[string]any:
 		e.encodeObject(v)
+	case JQValue:
+		e.encode(v.JQValueToGoJQ())
 	default:
 		panic(fmt.Sprintf("invalid type: %[1]T (%[1]v)", v))
 	}

execute.go

@@ -61,6 +61,9 @@ loop:
 			m := make(map[string]any, n)
 			for i := 0; i < n; i++ {
 				v, k := env.pop(), env.pop()
+				if jv, ok := k.(JQValue); ok {
+					k = jv.JQValueToString()
+				}
 				s, ok := k.(string)
 				if !ok {
 					err = &objectKeyNotStringError{k}
@@ -141,7 +144,9 @@ loop:
 			pc = code.v.(int)
 			goto loop
 		case opjumpifnot:
-			if v := env.pop(); v == nil || v == false {
+			v := env.pop()
+			b, bOk := toBoolean(v)
+			if isNull(v) || (bOk && !b) {
 				pc = code.v.(int)
 				goto loop
 			}
@@ -151,6 +156,9 @@ loop:
 			}
 			p, v := code.v, env.pop()
 			if code.op == opindexarray && v != nil {
+				if jqv, ok := v.(JQValue); ok {
+					v = jqv.JQValueToGoJQ()
+				}
 				if _, ok := v.([]any); !ok {
 					err = &expectedArrayError{v}
 					break loop
@@ -312,6 +320,32 @@ loop:
 					continue
 				}
 				break loop
+			case JQValue:
+				if !env.paths.empty() && env.expdepth == 0 && !env.pathIntact(v) {
+					err = &invalidPathIterError{v}
+					break loop
+				}
+				xsv := v.JQValueEach()
+				if e, ok := xsv.(error); ok {
+					err = e
+					break loop
+				}
+				switch xsv := xsv.(type) {
+				case []PathValue:
+					// convert from external PathValue to internal pathValue to make it easier to follow upstream
+					xs = make([]pathValue, len(xsv))
+					if len(xsv) == 0 {
+						break loop
+					}
+					for i, pv := range xsv {
+						xs[i] = pathValue{path: pv.Path, value: pv.Value}
+					}
+				case nil:
+					break loop
+				default:
+					err = &iteratorError{xsv}
+					break loop
+				}
 			default:
 				err = &iteratorError{v}
 				env.push(emptyIter{})
@@ -423,6 +457,9 @@ func (env *env) pathIntact(v any) bool {
 		if w, ok := w.(float64); ok {
 			return v == w || math.IsNaN(v) && math.IsNaN(w)
 		}
+	case JQValue:
+		// TODO: JQValue: should understand this better
+		return true
 	}
 	return v == w
 }