mirror of
https://github.com/wader/fq.git
synced 2024-12-23 13:22:58 +03:00
548a065388
Update msgpack, bson and cbor to use it.
175 lines
3.2 KiB
Go
175 lines
3.2 KiB
Go
// Some of these functions are based on gojq func.go functions
|
|
// TODO: maybe should be exported from gojq fq branch instead?
|
|
// The MIT License (MIT)
|
|
// Copyright (c) 2019-2021 itchyny
|
|
|
|
package gojqextra
|
|
|
|
import (
|
|
"math"
|
|
"math/big"
|
|
"strconv"
|
|
|
|
"github.com/wader/gojq"
|
|
)
|
|
|
|
func ToString(x interface{}) (string, bool) {
|
|
switch x := x.(type) {
|
|
case string:
|
|
return x, true
|
|
case gojq.JQValue:
|
|
return ToString(x.JQValueToGoJQ())
|
|
default:
|
|
return "", false
|
|
}
|
|
}
|
|
|
|
func ToObject(x interface{}) (map[string]interface{}, bool) {
|
|
switch x := x.(type) {
|
|
case map[string]interface{}:
|
|
return x, true
|
|
case gojq.JQValue:
|
|
return ToObject(x.JQValueToGoJQ())
|
|
default:
|
|
return nil, false
|
|
}
|
|
}
|
|
|
|
func ToArray(x interface{}) ([]interface{}, bool) {
|
|
switch x := x.(type) {
|
|
case []interface{}:
|
|
return x, true
|
|
case gojq.JQValue:
|
|
return ToArray(x.JQValueToGoJQ())
|
|
default:
|
|
return nil, false
|
|
}
|
|
}
|
|
|
|
func ToBoolean(x interface{}) (bool, bool) {
|
|
switch x := x.(type) {
|
|
case bool:
|
|
return x, true
|
|
case gojq.JQValue:
|
|
return ToBoolean(x.JQValueToGoJQ())
|
|
default:
|
|
return false, false
|
|
}
|
|
}
|
|
|
|
func IsNull(x interface{}) bool {
|
|
switch x := x.(type) {
|
|
case nil:
|
|
return true
|
|
case gojq.JQValue:
|
|
return IsNull(x.JQValueToGoJQ())
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func ToInt(x interface{}) (int, bool) {
|
|
switch x := x.(type) {
|
|
case int:
|
|
return x, true
|
|
case float64:
|
|
return floatToInt(x), true
|
|
case *big.Int:
|
|
if x.IsInt64() {
|
|
if i := x.Int64(); minInt <= i && i <= maxInt {
|
|
return int(i), true
|
|
}
|
|
}
|
|
if x.Sign() > 0 {
|
|
return maxInt, true
|
|
}
|
|
return minInt, true
|
|
case gojq.JQValue:
|
|
return ToInt(x.JQValueToGoJQ())
|
|
default:
|
|
// nil and other should fail, "null | tonumber" in jq is an error
|
|
return 0, false
|
|
}
|
|
}
|
|
|
|
func floatToInt(x float64) int {
|
|
if minInt <= x && x <= maxInt {
|
|
return int(x)
|
|
}
|
|
if x > 0 {
|
|
return maxInt
|
|
}
|
|
return minInt
|
|
}
|
|
|
|
func ToFloat(x interface{}) (float64, bool) {
|
|
switch x := x.(type) {
|
|
case int:
|
|
return float64(x), true
|
|
case float64:
|
|
return x, true
|
|
case *big.Int:
|
|
return bigToFloat(x), true
|
|
case gojq.JQValue:
|
|
return ToFloat(x.JQValueToGoJQ())
|
|
default:
|
|
// nil and other should fail, "null | tonumber" in jq is an error
|
|
return 0.0, false
|
|
}
|
|
}
|
|
|
|
func bigToFloat(x *big.Int) float64 {
|
|
if x.IsInt64() {
|
|
return float64(x.Int64())
|
|
}
|
|
if f, err := strconv.ParseFloat(x.String(), 64); err == nil {
|
|
return f
|
|
}
|
|
return math.Inf(x.Sign())
|
|
}
|
|
|
|
func ToGoJQValue(v interface{}) (interface{}, bool) {
|
|
switch vv := v.(type) {
|
|
case nil:
|
|
return vv, true
|
|
case bool:
|
|
return vv, true
|
|
case int:
|
|
return vv, true
|
|
case int64:
|
|
return big.NewInt(vv), true
|
|
case uint64:
|
|
return new(big.Int).SetUint64(vv), true
|
|
case float64:
|
|
return vv, true
|
|
case *big.Int:
|
|
return vv, true
|
|
case string:
|
|
return vv, true
|
|
case gojq.JQValue:
|
|
return ToGoJQValue(vv.JQValueToGoJQ())
|
|
case []interface{}:
|
|
vvs := make([]interface{}, len(vv))
|
|
for i, v := range vv {
|
|
v, ok := ToGoJQValue(v)
|
|
if !ok {
|
|
return nil, false
|
|
}
|
|
vvs[i] = v
|
|
}
|
|
return vvs, true
|
|
case map[string]interface{}:
|
|
vvs := make(map[string]interface{}, len(vv))
|
|
for k, v := range vv {
|
|
v, ok := ToGoJQValue(v)
|
|
if !ok {
|
|
return nil, false
|
|
}
|
|
vvs[k] = v
|
|
}
|
|
return vvs, true
|
|
default:
|
|
return nil, false
|
|
}
|
|
}
|