array.go

package tpl

import (
	"context"
	"encoding/json"
	"fmt"
	"log"
	"net/url"
	"strconv"

	"github.com/KarpelesLab/pjson"
)

func ResolveValueIndex(ctx context.Context, v any, s string) (any, error) {
	switch o := v.(type) {
	case ArrayAccessGet:
		return o.OffsetGet(ctx, s)
	case ArrayAccessGetAny:
		return o.OffsetGet(ctx, s)
	case map[string]any:
		return o[s], nil
	case map[string]Value:
		return o[s], nil
	case map[string]json.RawMessage:
		return o[s], nil
	case map[string]pjson.RawMessage:
		return o[s], nil
	case url.Values:
		return o[s], nil
	case Values:
		n, err := strconv.ParseInt(s, 0, 64)
		if err != nil {
			log.Printf("[tpl] failed to access array element #%s", s)
			return nil, nil
		}
		if n < 0 || int(n) >= len(o) {
			return nil, nil
		}
		return o[n], nil
	case []any:
		n, err := strconv.ParseInt(s, 0, 64)
		if err != nil {
			log.Printf("[tpl] failed to access array element #%s", s)
			return nil, nil
		}
		if n < 0 || int(n) >= len(o) {
			return nil, nil
		}
		return o[n], nil
	case []string:
		n, err := strconv.ParseInt(s, 0, 64)
		if err != nil {
			log.Printf("[tpl] failed to access array element #%s", s)
			return nil, nil
		}
		if n < 0 || int(n) >= len(o) {
			return nil, nil
		}
		return o[n], nil
	case json.RawMessage:
		// parse at json object
		var sub interface{}
		err := json.Unmarshal(o, &sub)
		if err != nil {
			return nil, fmt.Errorf("failed to parse json: %s", err)
		}
		return ResolveValueIndex(ctx, sub, s)
	case pjson.RawMessage:
		// parse at json object
		var sub interface{}
		err := json.Unmarshal(o, &sub)
		if err != nil {
			return nil, fmt.Errorf("failed to parse json: %s", err)
		}
		return ResolveValueIndex(ctx, sub, s)
	case interface{ RawJSONBytes() []byte }:
		// parse at json object
		var sub any
		err := json.Unmarshal(o.RawJSONBytes(), &sub)
		if err != nil {
			return nil, fmt.Errorf("failed to parse json: %s", err)
		}
		return ResolveValueIndex(ctx, sub, s)
	case ValueReader:
		val, err := o.ReadValue(ctx)
		if err != nil {
			return nil, err
		}
		return ResolveValueIndex(ctx, val, s)
	case nil:
		return nil, nil
	default:
		//log.Printf("unhandled type: %T", val)
		return nil, fmt.Errorf("unhandled type: %T", v)
	}
}