util.go

package jqx

import (
	"encoding/json"
	"fmt"
	"io/fs"
	"testing/fstest"
)

type failError string

func (f failError) Error() string { return string(f) }

func must[T any](val T, err error) T {
	if err != nil {
		panic(err)
	}
	return val
}
func failif(err error, format string, args ...any) {
	if err == nil {
		return
	}

	s := fmt.Sprintf(format, args...)
	s = fmt.Sprintf("error while %s: %v", s, err)
	panic(failError(s))
}
func catch[T error](rt *error) {
	err := recover()
	switch err := err.(type) {
	case nil:
	case T:
		*rt = err
	default:
		panic(err)
	}
}

func getMarshaler(tab bool, str bool) func(v any) ([]byte, error) {
	f := json.Marshal
	if tab {
		f = func(v any) ([]byte, error) {
			return json.MarshalIndent(v, "", "\t")
		}
	}
	if str {
		oldf := f
		f = func(v any) ([]byte, error) {
			if v, ok := v.(string); ok {
				return []byte(v), nil
			}
			return oldf(v)
		}
	}
	return f
}
func toFS(m map[string]any, marshaler func(any) ([]byte, error)) fs.FS {
	if marshaler == nil {
		marshaler = getMarshaler(false, true)
	}

	rt := fstest.MapFS{}

	for k, v := range m {
		rt[k] = &fstest.MapFile{Data: must(marshaler(v))}
	}

	return rt
}