go-jsonnet/builtins.go
Stanisław Barzowski 2824579c7d Update cpp-jsonnet
2020-05-22 14:54:20 +02:00

1477 lines
44 KiB
Go

/*
Copyright 2017 Google Inc. All rights reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package jsonnet
import (
"bytes"
"crypto/md5"
"encoding/base64"
"encoding/hex"
"encoding/json"
"fmt"
"math"
"os"
"sort"
"strings"
"github.com/google/go-jsonnet/ast"
)
func builtinPlus(i *interpreter, trace traceElement, x, y value) (value, error) {
// TODO(sbarzowski) perhaps a more elegant way to dispatch
switch right := y.(type) {
case valueString:
left, err := builtinToString(i, trace, x)
if err != nil {
return nil, err
}
return concatStrings(left.(valueString), right), nil
}
switch left := x.(type) {
case *valueNumber:
right, err := i.getNumber(y, trace)
if err != nil {
return nil, err
}
return makeDoubleCheck(i, trace, left.value+right.value)
case valueString:
right, err := builtinToString(i, trace, y)
if err != nil {
return nil, err
}
return concatStrings(left, right.(valueString)), nil
case *valueObject:
switch right := y.(type) {
case *valueObject:
return makeValueExtendedObject(left, right), nil
default:
return nil, i.typeErrorSpecific(y, &valueObject{}, trace)
}
case *valueArray:
right, err := i.getArray(y, trace)
if err != nil {
return nil, err
}
return concatArrays(left, right), nil
default:
return nil, i.typeErrorGeneral(x, trace)
}
}
func builtinMinus(i *interpreter, trace traceElement, xv, yv value) (value, error) {
x, err := i.getNumber(xv, trace)
if err != nil {
return nil, err
}
y, err := i.getNumber(yv, trace)
if err != nil {
return nil, err
}
return makeDoubleCheck(i, trace, x.value-y.value)
}
func builtinMult(i *interpreter, trace traceElement, xv, yv value) (value, error) {
x, err := i.getNumber(xv, trace)
if err != nil {
return nil, err
}
y, err := i.getNumber(yv, trace)
if err != nil {
return nil, err
}
return makeDoubleCheck(i, trace, x.value*y.value)
}
func builtinDiv(i *interpreter, trace traceElement, xv, yv value) (value, error) {
x, err := i.getNumber(xv, trace)
if err != nil {
return nil, err
}
y, err := i.getNumber(yv, trace)
if err != nil {
return nil, err
}
if y.value == 0 {
return nil, i.Error("Division by zero.", trace)
}
return makeDoubleCheck(i, trace, x.value/y.value)
}
func builtinModulo(i *interpreter, trace traceElement, xv, yv value) (value, error) {
x, err := i.getNumber(xv, trace)
if err != nil {
return nil, err
}
y, err := i.getNumber(yv, trace)
if err != nil {
return nil, err
}
if y.value == 0 {
return nil, i.Error("Division by zero.", trace)
}
return makeDoubleCheck(i, trace, math.Mod(x.value, y.value))
}
func valueLess(i *interpreter, trace traceElement, x, yv value) (bool, error) {
switch left := x.(type) {
case *valueNumber:
right, err := i.getNumber(yv, trace)
if err != nil {
return false, err
}
return left.value < right.value, nil
case valueString:
right, err := i.getString(yv, trace)
if err != nil {
return false, err
}
return stringLessThan(left, right), nil
default:
return false, i.typeErrorGeneral(x, trace)
}
}
func builtinLess(i *interpreter, trace traceElement, x, yv value) (value, error) {
b, err := valueLess(i, trace, x, yv)
return makeValueBoolean(b), err
}
func builtinGreater(i *interpreter, trace traceElement, x, y value) (value, error) {
return builtinLess(i, trace, y, x)
}
func builtinGreaterEq(i *interpreter, trace traceElement, x, y value) (value, error) {
res, err := builtinLess(i, trace, x, y)
if err != nil {
return nil, err
}
return res.(*valueBoolean).not(), nil
}
func builtinLessEq(i *interpreter, trace traceElement, x, y value) (value, error) {
res, err := builtinGreater(i, trace, x, y)
if err != nil {
return nil, err
}
return res.(*valueBoolean).not(), nil
}
func builtinLength(i *interpreter, trace traceElement, x value) (value, error) {
var num int
switch x := x.(type) {
case *valueObject:
num = len(objectFields(x, withoutHidden))
case *valueArray:
num = len(x.elements)
case valueString:
num = x.length()
case *valueFunction:
for _, param := range x.parameters() {
if param.defaultArg == nil {
num++
}
}
default:
return nil, i.typeErrorGeneral(x, trace)
}
return makeValueNumber(float64(num)), nil
}
func builtinToString(i *interpreter, trace traceElement, x value) (value, error) {
switch x := x.(type) {
case valueString:
return x, nil
}
var buf bytes.Buffer
err := i.manifestAndSerializeJSON(&buf, trace, x, false, "")
if err != nil {
return nil, err
}
return makeValueString(buf.String()), nil
}
func builtinTrace(i *interpreter, trace traceElement, x value, y value) (value, error) {
xStr, err := i.getString(x, trace)
if err != nil {
return nil, err
}
filename := trace.loc.FileName
line := trace.loc.Begin.Line
fmt.Fprintf(
os.Stderr, "TRACE: %s:%d %s\n", filename, line, xStr.getGoString())
return y, nil
}
// astMakeArrayElement wraps the function argument of std.makeArray so that
// it can be embedded in cachedThunk without needing to execute it ahead of
// time. It is equivalent to `local i = 42; func(i)`. It therefore has no
// free variables and needs only an empty environment to execute.
type astMakeArrayElement struct {
ast.NodeBase
function *valueFunction
index int
}
func builtinMakeArray(i *interpreter, trace traceElement, szv, funcv value) (value, error) {
sz, err := i.getInt(szv, trace)
if err != nil {
return nil, err
}
fun, err := i.getFunction(funcv, trace)
if err != nil {
return nil, err
}
var elems []*cachedThunk
for i := 0; i < sz; i++ {
elem := &cachedThunk{
env: &environment{},
body: &astMakeArrayElement{
NodeBase: ast.NodeBase{},
function: fun,
index: i,
},
}
elems = append(elems, elem)
}
return makeValueArray(elems), nil
}
func builtinFlatMap(i *interpreter, trace traceElement, funcv, arrv value) (value, error) {
arr, err := i.getArray(arrv, trace)
if err != nil {
return nil, err
}
fun, err := i.getFunction(funcv, trace)
if err != nil {
return nil, err
}
num := arr.length()
// Start with capacity of the original array.
// This may spare us a few reallocations.
// TODO(sbarzowski) verify that it actually helps
elems := make([]*cachedThunk, 0, num)
for counter := 0; counter < num; counter++ {
returnedValue, err := fun.call(i, trace, args(arr.elements[counter]))
if err != nil {
return nil, err
}
returned, err := i.getArray(returnedValue, trace)
if err != nil {
return nil, err
}
elems = append(elems, returned.elements...)
}
return makeValueArray(elems), nil
}
func joinArrays(i *interpreter, trace traceElement, sep *valueArray, arr *valueArray) (value, error) {
result := make([]*cachedThunk, 0, arr.length())
first := true
for _, elem := range arr.elements {
elemValue, err := i.evaluatePV(elem, trace)
if err != nil {
return nil, err
}
switch v := elemValue.(type) {
case *valueNull:
continue
case *valueArray:
if !first {
result = append(result, sep.elements...)
}
result = append(result, v.elements...)
default:
return nil, i.typeErrorSpecific(elemValue, &valueArray{}, trace)
}
first = false
}
return makeValueArray(result), nil
}
func joinStrings(i *interpreter, trace traceElement, sep valueString, arr *valueArray) (value, error) {
result := make([]rune, 0, arr.length())
first := true
for _, elem := range arr.elements {
elemValue, err := i.evaluatePV(elem, trace)
if err != nil {
return nil, err
}
switch v := elemValue.(type) {
case *valueNull:
continue
case valueString:
if !first {
result = append(result, sep.getRunes()...)
}
result = append(result, v.getRunes()...)
default:
return nil, i.typeErrorSpecific(elemValue, emptyString(), trace)
}
first = false
}
return makeStringFromRunes(result), nil
}
func builtinJoin(i *interpreter, trace traceElement, sep, arrv value) (value, error) {
arr, err := i.getArray(arrv, trace)
if err != nil {
return nil, err
}
switch sep := sep.(type) {
case valueString:
return joinStrings(i, trace, sep, arr)
case *valueArray:
return joinArrays(i, trace, sep, arr)
default:
return nil, i.Error("join first parameter should be string or array, got "+sep.getType().name, trace)
}
}
func builtinReverse(i *interpreter, trace traceElement, arrv value) (value, error) {
arr, err := i.getArray(arrv, trace)
if err != nil {
return nil, err
}
lenArr := len(arr.elements) // lenx holds the original array length
reversedArray := make([]*cachedThunk, lenArr) // creates a slice that refer to a new array of length lenx
for i := 0; i < lenArr; i++ {
j := lenArr - (i + 1) // j initially holds (lenx - 1) and decreases to 0 while i initially holds 0 and increase to (lenx - 1)
reversedArray[i] = arr.elements[j]
}
return makeValueArray(reversedArray), nil
}
func builtinFilter(i *interpreter, trace traceElement, funcv, arrv value) (value, error) {
arr, err := i.getArray(arrv, trace)
if err != nil {
return nil, err
}
fun, err := i.getFunction(funcv, trace)
if err != nil {
return nil, err
}
num := arr.length()
// Start with capacity of the original array.
// This may spare us a few reallocations.
// TODO(sbarzowski) verify that it actually helps
elems := make([]*cachedThunk, 0, num)
for counter := 0; counter < num; counter++ {
includedValue, err := fun.call(i, trace, args(arr.elements[counter]))
if err != nil {
return nil, err
}
included, err := i.getBoolean(includedValue, trace)
if err != nil {
return nil, err
}
if included.value {
elems = append(elems, arr.elements[counter])
}
}
return makeValueArray(elems), nil
}
type sortData struct {
i *interpreter
trace traceElement
thunks []*cachedThunk
keys []value
err error
}
func (d *sortData) Len() int {
return len(d.thunks)
}
func (d *sortData) Less(i, j int) bool {
b, err := valueLess(d.i, d.trace, d.keys[i], d.keys[j])
if err != nil {
d.err = err
panic("Error while comparing elements")
}
return b
}
func (d *sortData) Swap(i, j int) {
d.thunks[i], d.thunks[j] = d.thunks[j], d.thunks[i]
d.keys[i], d.keys[j] = d.keys[j], d.keys[i]
}
func (d *sortData) Sort() (err error) {
defer func() {
if d.err != nil {
if r := recover(); r != nil {
err = d.err
}
}
}()
sort.Stable(d)
return
}
func builtinSort(i *interpreter, trace traceElement, arguments []value) (value, error) {
arrv := arguments[0]
keyFv := arguments[1]
arr, err := i.getArray(arrv, trace)
if err != nil {
return nil, err
}
keyF, err := i.getFunction(keyFv, trace)
if err != nil {
return nil, err
}
num := arr.length()
data := sortData{i: i, trace: trace, thunks: make([]*cachedThunk, num), keys: make([]value, num)}
for counter := 0; counter < num; counter++ {
var err error
data.thunks[counter] = arr.elements[counter]
data.keys[counter], err = keyF.call(i, trace, args(arr.elements[counter]))
if err != nil {
return nil, err
}
}
err = data.Sort()
if err != nil {
return nil, err
}
return makeValueArray(data.thunks), nil
}
func builtinRange(i *interpreter, trace traceElement, fromv, tov value) (value, error) {
from, err := i.getInt(fromv, trace)
if err != nil {
return nil, err
}
to, err := i.getInt(tov, trace)
if err != nil {
return nil, err
}
elems := make([]*cachedThunk, to-from+1)
for i := from; i <= to; i++ {
elems[i-from] = readyThunk(intToValue(i))
}
return makeValueArray(elems), nil
}
func builtinNegation(i *interpreter, trace traceElement, x value) (value, error) {
b, err := i.getBoolean(x, trace)
if err != nil {
return nil, err
}
return makeValueBoolean(!b.value), nil
}
func builtinBitNeg(i *interpreter, trace traceElement, x value) (value, error) {
n, err := i.getNumber(x, trace)
if err != nil {
return nil, err
}
intValue := int64(n.value)
return int64ToValue(^intValue), nil
}
func builtinIdentity(i *interpreter, trace traceElement, x value) (value, error) {
return x, nil
}
func builtinUnaryPlus(i *interpreter, trace traceElement, x value) (value, error) {
n, err := i.getNumber(x, trace)
if err != nil {
return nil, err
}
return makeValueNumber(n.value), nil
}
func builtinUnaryMinus(i *interpreter, trace traceElement, x value) (value, error) {
n, err := i.getNumber(x, trace)
if err != nil {
return nil, err
}
return makeValueNumber(-n.value), nil
}
// TODO(sbarzowski) since we have a builtin implementation of equals it's no longer really
// needed and we should deprecate it eventually
func primitiveEquals(i *interpreter, trace traceElement, x, y value) (value, error) {
if x.getType() != y.getType() {
return makeValueBoolean(false), nil
}
switch left := x.(type) {
case *valueBoolean:
right, err := i.getBoolean(y, trace)
if err != nil {
return nil, err
}
return makeValueBoolean(left.value == right.value), nil
case *valueNumber:
right, err := i.getNumber(y, trace)
if err != nil {
return nil, err
}
return makeValueBoolean(left.value == right.value), nil
case valueString:
right, err := i.getString(y, trace)
if err != nil {
return nil, err
}
return makeValueBoolean(stringEqual(left, right)), nil
case *valueNull:
return makeValueBoolean(true), nil
case *valueFunction:
return nil, i.Error("Cannot test equality of functions", trace)
default:
return nil, i.Error(
"primitiveEquals operates on primitive types, got "+x.getType().name,
trace,
)
}
}
func rawEquals(i *interpreter, trace traceElement, x, y value) (bool, error) {
if x.getType() != y.getType() {
return false, nil
}
switch left := x.(type) {
case *valueBoolean:
right, err := i.getBoolean(y, trace)
if err != nil {
return false, err
}
return left.value == right.value, nil
case *valueNumber:
right, err := i.getNumber(y, trace)
if err != nil {
return false, err
}
return left.value == right.value, nil
case valueString:
right, err := i.getString(y, trace)
if err != nil {
return false, err
}
return stringEqual(left, right), nil
case *valueNull:
return true, nil
case *valueArray:
right, err := i.getArray(y, trace)
if err != nil {
return false, err
}
if left.length() != right.length() {
return false, nil
}
for j := range left.elements {
leftElem, err := i.evaluatePV(left.elements[j], trace)
if err != nil {
return false, err
}
rightElem, err := i.evaluatePV(right.elements[j], trace)
if err != nil {
return false, err
}
eq, err := rawEquals(i, trace, leftElem, rightElem)
if err != nil {
return false, err
}
if !eq {
return false, nil
}
}
return true, nil
case *valueObject:
right, err := i.getObject(y, trace)
if err != nil {
return false, err
}
leftFields := objectFields(left, withoutHidden)
rightFields := objectFields(right, withoutHidden)
sort.Strings(leftFields)
sort.Strings(rightFields)
if len(leftFields) != len(rightFields) {
return false, nil
}
for i := range leftFields {
if leftFields[i] != rightFields[i] {
return false, nil
}
}
for j := range leftFields {
fieldName := leftFields[j]
leftField, err := left.index(i, trace, fieldName)
if err != nil {
return false, err
}
rightField, err := right.index(i, trace, fieldName)
if err != nil {
return false, err
}
eq, err := rawEquals(i, trace, leftField, rightField)
if err != nil {
return false, err
}
if !eq {
return false, nil
}
}
return true, nil
case *valueFunction:
return false, i.Error("Cannot test equality of functions", trace)
}
panic(fmt.Sprintf("Unhandled case in equals %#+v %#+v", x, y))
}
func builtinEquals(i *interpreter, trace traceElement, x, y value) (value, error) {
eq, err := rawEquals(i, trace, x, y)
if err != nil {
return nil, err
}
return makeValueBoolean(eq), nil
}
func builtinNotEquals(i *interpreter, trace traceElement, x, y value) (value, error) {
eq, err := rawEquals(i, trace, x, y)
if err != nil {
return nil, err
}
return makeValueBoolean(!eq), nil
}
func builtinType(i *interpreter, trace traceElement, x value) (value, error) {
return makeValueString(x.getType().name), nil
}
func builtinMd5(i *interpreter, trace traceElement, x value) (value, error) {
str, err := i.getString(x, trace)
if err != nil {
return nil, err
}
hash := md5.Sum([]byte(str.getGoString()))
return makeValueString(hex.EncodeToString(hash[:])), nil
}
func builtinBase64(i *interpreter, trace traceElement, input value) (value, error) {
var byteArr []byte
var sanityCheck = func(v int) (string, bool) {
if v < 0 || 255 < v {
msg := fmt.Sprintf("base64 encountered invalid codepoint value in the array (must be 0 <= X <= 255), got %d", v)
return msg, false
}
return "", true
}
switch input.(type) {
case valueString:
vStr, err := i.getString(input, trace)
if err != nil {
return nil, err
}
str := vStr.getGoString()
for _, r := range str {
n := int(r)
msg, ok := sanityCheck(n)
if !ok {
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
}
byteArr = []byte(str)
case *valueArray:
vArr, err := i.getArray(input, trace)
if err != nil {
return nil, err
}
for _, cThunk := range vArr.elements {
cTv, err := cThunk.getValue(i, trace)
if err != nil {
return nil, err
}
vInt, err := i.getInt(cTv, trace)
if err != nil {
msg := fmt.Sprintf("base64 encountered a non-integer value in the array, got %s", cTv.getType().name)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
msg, ok := sanityCheck(vInt)
if !ok {
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
byteArr = append(byteArr, byte(vInt))
}
default:
msg := fmt.Sprintf("base64 can only base64 encode strings / arrays of single bytes, got %s", input.getType().name)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
sEnc := base64.StdEncoding.EncodeToString(byteArr)
return makeValueString(sEnc), nil
}
func builtinEncodeUTF8(i *interpreter, trace traceElement, x value) (value, error) {
str, err := i.getString(x, trace)
if err != nil {
return nil, err
}
s := str.getGoString()
elems := make([]*cachedThunk, 0, len(s)) // it will be longer if characters fall outside of ASCII
for _, c := range []byte(s) {
elems = append(elems, readyThunk(makeValueNumber(float64(c))))
}
return makeValueArray(elems), nil
}
func builtinDecodeUTF8(i *interpreter, trace traceElement, x value) (value, error) {
arr, err := i.getArray(x, trace)
if err != nil {
return nil, err
}
bs := make([]byte, len(arr.elements)) // it will be longer if characters fall outside of ASCII
for pos := range arr.elements {
v, err := i.evaluateInt(arr.elements[pos], trace)
if err != nil {
return nil, err
}
if v < 0 || v > 255 {
return nil, i.Error(fmt.Sprintf("Bytes must be integers in range [0, 255], got %d", v), trace)
}
bs[pos] = byte(v)
}
return makeValueString(string(bs)), nil
}
// Maximum allowed unicode codepoint
// https://en.wikipedia.org/wiki/Unicode#Architecture_and_terminology
const codepointMax = 0x10FFFF
func builtinChar(i *interpreter, trace traceElement, x value) (value, error) {
n, err := i.getNumber(x, trace)
if err != nil {
return nil, err
}
if n.value > codepointMax {
return nil, i.Error(fmt.Sprintf("Invalid unicode codepoint, got %v", n.value), trace)
} else if n.value < 0 {
return nil, i.Error(fmt.Sprintf("Codepoints must be >= 0, got %v", n.value), trace)
}
return makeValueString(string(rune(n.value))), nil
}
func builtinCodepoint(i *interpreter, trace traceElement, x value) (value, error) {
str, err := i.getString(x, trace)
if err != nil {
return nil, err
}
if str.length() != 1 {
return nil, i.Error(fmt.Sprintf("codepoint takes a string of length 1, got length %v", str.length()), trace)
}
return makeValueNumber(float64(str.getRunes()[0])), nil
}
func makeDoubleCheck(i *interpreter, trace traceElement, x float64) (value, error) {
if math.IsNaN(x) {
return nil, i.Error("Not a number", trace)
}
if math.IsInf(x, 0) {
return nil, i.Error("Overflow", trace)
}
return makeValueNumber(x), nil
}
func liftNumeric(f func(float64) float64) func(*interpreter, traceElement, value) (value, error) {
return func(i *interpreter, trace traceElement, x value) (value, error) {
n, err := i.getNumber(x, trace)
if err != nil {
return nil, err
}
return makeDoubleCheck(i, trace, f(n.value))
}
}
var builtinSqrt = liftNumeric(math.Sqrt)
var builtinCeil = liftNumeric(math.Ceil)
var builtinFloor = liftNumeric(math.Floor)
var builtinSin = liftNumeric(math.Sin)
var builtinCos = liftNumeric(math.Cos)
var builtinTan = liftNumeric(math.Tan)
var builtinAsin = liftNumeric(math.Asin)
var builtinAcos = liftNumeric(math.Acos)
var builtinAtan = liftNumeric(math.Atan)
var builtinLog = liftNumeric(math.Log)
var builtinExp = liftNumeric(func(f float64) float64 {
res := math.Exp(f)
if res == 0 && f > 0 {
return math.Inf(1)
}
return res
})
var builtinMantissa = liftNumeric(func(f float64) float64 {
mantissa, _ := math.Frexp(f)
return mantissa
})
var builtinExponent = liftNumeric(func(f float64) float64 {
_, exponent := math.Frexp(f)
return float64(exponent)
})
func liftBitwise(f func(int64, int64) int64) func(*interpreter, traceElement, value, value) (value, error) {
return func(i *interpreter, trace traceElement, xv, yv value) (value, error) {
x, err := i.getNumber(xv, trace)
if err != nil {
return nil, err
}
y, err := i.getNumber(yv, trace)
if err != nil {
return nil, err
}
if x.value < math.MinInt64 || x.value > math.MaxInt64 {
msg := fmt.Sprintf("Bitwise operator argument %v outside of range [%v, %v]", x.value, int64(math.MinInt64), int64(math.MaxInt64))
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
if y.value < math.MinInt64 || y.value > math.MaxInt64 {
msg := fmt.Sprintf("Bitwise operator argument %v outside of range [%v, %v]", y.value, int64(math.MinInt64), int64(math.MaxInt64))
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
return makeDoubleCheck(i, trace, float64(f(int64(x.value), int64(y.value))))
}
}
// TODO(sbarzowski) negative shifts
var builtinShiftL = liftBitwise(func(x, y int64) int64 { return x << uint(y%64) })
var builtinShiftR = liftBitwise(func(x, y int64) int64 { return x >> uint(y%64) })
var builtinBitwiseAnd = liftBitwise(func(x, y int64) int64 { return x & y })
var builtinBitwiseOr = liftBitwise(func(x, y int64) int64 { return x | y })
var builtinBitwiseXor = liftBitwise(func(x, y int64) int64 { return x ^ y })
func builtinObjectFieldsEx(i *interpreter, trace traceElement, objv, includeHiddenV value) (value, error) {
obj, err := i.getObject(objv, trace)
if err != nil {
return nil, err
}
includeHidden, err := i.getBoolean(includeHiddenV, trace)
if err != nil {
return nil, err
}
fields := objectFields(obj, withHiddenFromBool(includeHidden.value))
sort.Strings(fields)
elems := []*cachedThunk{}
for _, fieldname := range fields {
elems = append(elems, readyThunk(makeValueString(fieldname)))
}
return makeValueArray(elems), nil
}
func builtinObjectHasEx(i *interpreter, trace traceElement, objv value, fnamev value, includeHiddenV value) (value, error) {
obj, err := i.getObject(objv, trace)
if err != nil {
return nil, err
}
fname, err := i.getString(fnamev, trace)
if err != nil {
return nil, err
}
includeHidden, err := i.getBoolean(includeHiddenV, trace)
if err != nil {
return nil, err
}
h := withHiddenFromBool(includeHidden.value)
hasField := objectHasField(objectBinding(obj), string(fname.getRunes()), h)
return makeValueBoolean(hasField), nil
}
func builtinPow(i *interpreter, trace traceElement, basev value, expv value) (value, error) {
base, err := i.getNumber(basev, trace)
if err != nil {
return nil, err
}
exp, err := i.getNumber(expv, trace)
if err != nil {
return nil, err
}
return makeDoubleCheck(i, trace, math.Pow(base.value, exp.value))
}
func builtinSubstr(i *interpreter, trace traceElement, inputStr, inputFrom, inputLen value) (value, error) {
strV, err := i.getString(inputStr, trace)
if err != nil {
msg := fmt.Sprintf("substr first parameter should be a string, got %s", inputStr.getType().name)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
fromV, err := i.getNumber(inputFrom, trace)
if err != nil {
msg := fmt.Sprintf("substr second parameter should be a number, got %s", inputFrom.getType().name)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
if math.Mod(fromV.value, 1) != 0 {
msg := fmt.Sprintf("substr second parameter should be an integer, got %f", fromV.value)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
lenV, err := i.getNumber(inputLen, trace)
if err != nil {
msg := fmt.Sprintf("substr third parameter should be a number, got %s", inputLen.getType().name)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
lenInt, err := i.getInt(lenV, trace)
if err != nil {
msg := fmt.Sprintf("substr third parameter should be an integer, got %f", lenV.value)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
if lenInt < 0 {
msg := fmt.Sprintf("substr third parameter should be greater than zero, got %d", lenInt)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
fromInt := int(fromV.value)
strStr := strV.getGoString()
endIndex := fromInt + lenInt
if endIndex > len(strStr) {
endIndex = len(strStr)
}
if fromInt > len(strStr) {
return makeValueString(""), nil
}
runes := []rune(strStr)
return makeValueString(string(runes[fromInt:endIndex])), nil
}
func builtinSplitLimit(i *interpreter, trace traceElement, strv, cv, maxSplitsV value) (value, error) {
str, err := i.getString(strv, trace)
if err != nil {
return nil, err
}
c, err := i.getString(cv, trace)
if err != nil {
return nil, err
}
maxSplits, err := i.getInt(maxSplitsV, trace)
if err != nil {
return nil, err
}
if maxSplits < -1 {
return nil, i.Error(fmt.Sprintf("std.splitLimit third parameter should be -1 or non-negative, got %v", maxSplits), trace)
}
sStr := str.getGoString()
sC := c.getGoString()
if len(sC) != 1 {
return nil, i.Error(fmt.Sprintf("std.splitLimit second parameter should have length 1, got %v", len(sC)), trace)
}
// the convention is slightly different from strings.splitN in Go (the meaning of non-negative values is shifted by one)
var strs []string
if maxSplits == -1 {
strs = strings.SplitN(sStr, sC, -1)
} else {
strs = strings.SplitN(sStr, sC, maxSplits+1)
}
res := make([]*cachedThunk, len(strs))
for i := range strs {
res[i] = readyThunk(makeValueString(strs[i]))
}
return makeValueArray(res), nil
}
func builtinStrReplace(i *interpreter, trace traceElement, strv, fromv, tov value) (value, error) {
str, err := i.getString(strv, trace)
if err != nil {
return nil, err
}
from, err := i.getString(fromv, trace)
if err != nil {
return nil, err
}
to, err := i.getString(tov, trace)
if err != nil {
return nil, err
}
sStr := str.getGoString()
sFrom := from.getGoString()
sTo := to.getGoString()
if len(sFrom) == 0 {
return nil, i.Error("'from' string must not be zero length.", trace)
}
return makeValueString(strings.Replace(sStr, sFrom, sTo, -1)), nil
}
func base64DecodeGoBytes(i *interpreter, trace traceElement, str string) ([]byte, error) {
strLen := len(str)
if strLen%4 != 0 {
msg := fmt.Sprintf("input string appears not to be a base64 encoded string. Wrong length found (%d)", strLen)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
decodedBytes, err := base64.StdEncoding.DecodeString(str)
if err != nil {
return nil, i.Error(fmt.Sprintf("failed to decode: %s", err), trace)
}
return decodedBytes, nil
}
func builtinBase64DecodeBytes(i *interpreter, trace traceElement, input value) (value, error) {
vStr, err := i.getString(input, trace)
if err != nil {
msg := fmt.Sprintf("base64DecodeBytes requires a string, got %s", input.getType().name)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
decodedBytes, err := base64DecodeGoBytes(i, trace, vStr.getGoString())
if err != nil {
return nil, err
}
res := make([]*cachedThunk, len(decodedBytes))
for i := range decodedBytes {
res[i] = readyThunk(makeValueNumber(float64(int(decodedBytes[i]))))
}
return makeValueArray(res), nil
}
func builtinBase64Decode(i *interpreter, trace traceElement, input value) (value, error) {
vStr, err := i.getString(input, trace)
if err != nil {
msg := fmt.Sprintf("base64DecodeBytes requires a string, got %s", input.getType().name)
return nil, makeRuntimeError(msg, i.getCurrentStackTrace(trace))
}
decodedBytes, err := base64DecodeGoBytes(i, trace, vStr.getGoString())
if err != nil {
return nil, err
}
return makeValueString(string(decodedBytes)), nil
}
func builtinUglyObjectFlatMerge(i *interpreter, trace traceElement, x value) (value, error) {
// TODO(sbarzowski) consider keeping comprehensions in AST
// It will probably be way less hacky, with better error messages and better performance
objarr, err := i.getArray(x, trace)
if err != nil {
return nil, err
}
newFields := make(simpleObjectFieldMap)
var anyObj *simpleObject
for _, elem := range objarr.elements {
obj, err := i.evaluateObject(elem, trace)
if err != nil {
return nil, err
}
// starts getting ugly - we mess with object internals
simpleObj := obj.uncached.(*simpleObject)
// there is only one field, really
for fieldName, fieldVal := range simpleObj.fields {
if _, alreadyExists := newFields[fieldName]; alreadyExists {
return nil, i.Error(duplicateFieldNameErrMsg(fieldName), trace)
}
// Here is the tricky part. Each field in a comprehension has different
// upValues, because for example in {[v]: v for v in ["x", "y", "z"] },
// the v is different for each field.
// Yet, even though upValues are field-specific, they are shadowed by object locals,
// so we need to make holes to let them pass through
upValues := simpleObj.upValues
for _, l := range simpleObj.locals {
delete(upValues, l.name)
}
newFields[fieldName] = simpleObjectField{
hide: fieldVal.hide,
field: &bindingsUnboundField{
inner: fieldVal.field,
bindings: simpleObj.upValues,
},
}
}
anyObj = simpleObj
}
var locals []objectLocal
var localUpValues bindingFrame
if len(objarr.elements) > 0 {
// another ugliness - we just take the locals of our last object,
// we assume that the locals are the same for each of merged objects
locals = anyObj.locals
// note that there are already holes for object locals
localUpValues = anyObj.upValues
}
return makeValueSimpleObject(
localUpValues,
newFields,
[]unboundField{}, // No asserts allowed
locals,
), nil
}
func builtinParseJSON(i *interpreter, trace traceElement, str value) (value, error) {
sval, err := i.getString(str, trace)
if err != nil {
return nil, err
}
s := sval.getGoString()
var parsedJSON interface{}
err = json.Unmarshal([]byte(s), &parsedJSON)
if err != nil {
return nil, i.Error(fmt.Sprintf("failed to parse JSON: %v", err.Error()), trace)
}
return jsonToValue(i, trace, parsedJSON)
}
func builtinExtVar(i *interpreter, trace traceElement, name value) (value, error) {
str, err := i.getString(name, trace)
if err != nil {
return nil, err
}
index := str.getGoString()
if pv, ok := i.extVars[index]; ok {
return i.evaluatePV(pv, trace)
}
return nil, i.Error("Undefined external variable: "+string(index), trace)
}
func builtinNative(i *interpreter, trace traceElement, name value) (value, error) {
str, err := i.getString(name, trace)
if err != nil {
return nil, err
}
index := str.getGoString()
if f, exists := i.nativeFuncs[index]; exists {
return &valueFunction{ec: f}, nil
}
return &valueNull{}, nil
}
// Utils for builtins - TODO(sbarzowski) move to a separate file in another commit
type builtin interface {
evalCallable
Name() ast.Identifier
}
func flattenArgs(args callArguments, params []namedParameter, defaults []value) []*cachedThunk {
positions := make(map[ast.Identifier]int)
for i, param := range params {
positions[param.name] = i
}
flatArgs := make([]*cachedThunk, len(params))
// Bind positional arguments
copy(flatArgs, args.positional)
// Bind named arguments
for _, arg := range args.named {
flatArgs[positions[arg.name]] = arg.pv
}
// Bind defaults for unsatisfied named parameters
for i := range params {
if flatArgs[i] == nil {
flatArgs[i] = readyThunk(defaults[i])
}
}
return flatArgs
}
type unaryBuiltinFunc func(*interpreter, traceElement, value) (value, error)
type unaryBuiltin struct {
name ast.Identifier
function unaryBuiltinFunc
params ast.Identifiers
}
func getBuiltinTrace(trace traceElement, name ast.Identifier) traceElement {
context := "builtin function <" + string(name) + ">"
return traceElement{loc: trace.loc, context: &context}
}
func (b *unaryBuiltin) evalCall(args callArguments, i *interpreter, trace traceElement) (value, error) {
flatArgs := flattenArgs(args, b.parameters(), []value{})
builtinTrace := getBuiltinTrace(trace, b.name)
x, err := flatArgs[0].getValue(i, trace)
if err != nil {
return nil, err
}
return b.function(i, builtinTrace, x)
}
func (b *unaryBuiltin) parameters() []namedParameter {
ret := make([]namedParameter, len(b.params))
for i := range ret {
ret[i].name = b.params[i]
}
return ret
}
func (b *unaryBuiltin) Name() ast.Identifier {
return b.name
}
type binaryBuiltinFunc func(*interpreter, traceElement, value, value) (value, error)
type binaryBuiltin struct {
name ast.Identifier
function binaryBuiltinFunc
params ast.Identifiers
}
func (b *binaryBuiltin) evalCall(args callArguments, i *interpreter, trace traceElement) (value, error) {
flatArgs := flattenArgs(args, b.parameters(), []value{})
builtinTrace := getBuiltinTrace(trace, b.name)
x, err := flatArgs[0].getValue(i, trace)
if err != nil {
return nil, err
}
y, err := flatArgs[1].getValue(i, trace)
if err != nil {
return nil, err
}
return b.function(i, builtinTrace, x, y)
}
func (b *binaryBuiltin) parameters() []namedParameter {
ret := make([]namedParameter, len(b.params))
for i := range ret {
ret[i].name = b.params[i]
}
return ret
}
func (b *binaryBuiltin) Name() ast.Identifier {
return b.name
}
type ternaryBuiltinFunc func(*interpreter, traceElement, value, value, value) (value, error)
type ternaryBuiltin struct {
name ast.Identifier
function ternaryBuiltinFunc
params ast.Identifiers
}
func (b *ternaryBuiltin) evalCall(args callArguments, i *interpreter, trace traceElement) (value, error) {
flatArgs := flattenArgs(args, b.parameters(), []value{})
builtinTrace := getBuiltinTrace(trace, b.name)
x, err := flatArgs[0].getValue(i, trace)
if err != nil {
return nil, err
}
y, err := flatArgs[1].getValue(i, trace)
if err != nil {
return nil, err
}
z, err := flatArgs[2].getValue(i, trace)
if err != nil {
return nil, err
}
return b.function(i, builtinTrace, x, y, z)
}
func (b *ternaryBuiltin) parameters() []namedParameter {
ret := make([]namedParameter, len(b.params))
for i := range ret {
ret[i].name = b.params[i]
}
return ret
}
func (b *ternaryBuiltin) Name() ast.Identifier {
return b.name
}
type generalBuiltinFunc func(*interpreter, traceElement, []value) (value, error)
type generalBuiltinParameter struct {
name ast.Identifier
// Note that the defaults are passed as values rather than AST nodes like in Parameters.
// This spares us unnecessary evaluation.
defaultValue value
}
// generalBuiltin covers cases that other builtin structures do not,
// in particular it can have any number of parameters. It can also
// have optional parameters. The optional ones have non-nil defaultValues
// at the same index.
type generalBuiltin struct {
name ast.Identifier
params []generalBuiltinParameter
function generalBuiltinFunc
}
func (b *generalBuiltin) parameters() []namedParameter {
ret := make([]namedParameter, len(b.params))
for i := range ret {
ret[i].name = b.params[i].name
if b.params[i].defaultValue != nil {
// This is not actually used because the defaultValue is used instead.
// The only reason we don't leave it nil is because the checkArguments
// function uses the non-nil status to indicate that the parameter
// is optional.
ret[i].defaultArg = &ast.LiteralNull{}
}
}
return ret
}
func (b *generalBuiltin) defaultValues() []value {
ret := make([]value, len(b.params))
for i := range ret {
ret[i] = b.params[i].defaultValue
}
return ret
}
func (b *generalBuiltin) Name() ast.Identifier {
return b.name
}
func (b *generalBuiltin) evalCall(args callArguments, i *interpreter, trace traceElement) (value, error) {
flatArgs := flattenArgs(args, b.parameters(), b.defaultValues())
builtinTrace := getBuiltinTrace(trace, b.name)
values := make([]value, len(flatArgs))
for j := 0; j < len(values); j++ {
var err error
values[j], err = flatArgs[j].getValue(i, trace)
if err != nil {
return nil, err
}
}
return b.function(i, builtinTrace, values)
}
// End of builtin utils
var builtinID = &unaryBuiltin{name: "id", function: builtinIdentity, params: ast.Identifiers{"x"}}
var functionID = &valueFunction{ec: builtinID}
var bopBuiltins = []*binaryBuiltin{
// Note that % and `in` are desugared instead of being handled here
ast.BopMult: &binaryBuiltin{name: "operator*", function: builtinMult, params: ast.Identifiers{"x", "y"}},
ast.BopDiv: &binaryBuiltin{name: "operator/", function: builtinDiv, params: ast.Identifiers{"x", "y"}},
ast.BopPlus: &binaryBuiltin{name: "operator+", function: builtinPlus, params: ast.Identifiers{"x", "y"}},
ast.BopMinus: &binaryBuiltin{name: "operator-", function: builtinMinus, params: ast.Identifiers{"x", "y"}},
ast.BopShiftL: &binaryBuiltin{name: "operator<<", function: builtinShiftL, params: ast.Identifiers{"x", "y"}},
ast.BopShiftR: &binaryBuiltin{name: "operator>>", function: builtinShiftR, params: ast.Identifiers{"x", "y"}},
ast.BopGreater: &binaryBuiltin{name: "operator>", function: builtinGreater, params: ast.Identifiers{"x", "y"}},
ast.BopGreaterEq: &binaryBuiltin{name: "operator>=", function: builtinGreaterEq, params: ast.Identifiers{"x", "y"}},
ast.BopLess: &binaryBuiltin{name: "operator<,", function: builtinLess, params: ast.Identifiers{"x", "y"}},
ast.BopLessEq: &binaryBuiltin{name: "operator<=", function: builtinLessEq, params: ast.Identifiers{"x", "y"}},
ast.BopManifestEqual: &binaryBuiltin{name: "operator==", function: builtinEquals, params: ast.Identifiers{"x", "y"}},
ast.BopManifestUnequal: &binaryBuiltin{name: "operator!=", function: builtinNotEquals, params: ast.Identifiers{"x", "y"}}, // Special case
ast.BopBitwiseAnd: &binaryBuiltin{name: "operator&", function: builtinBitwiseAnd, params: ast.Identifiers{"x", "y"}},
ast.BopBitwiseXor: &binaryBuiltin{name: "operator^", function: builtinBitwiseXor, params: ast.Identifiers{"x", "y"}},
ast.BopBitwiseOr: &binaryBuiltin{name: "operator|", function: builtinBitwiseOr, params: ast.Identifiers{"x", "y"}},
}
var uopBuiltins = []*unaryBuiltin{
ast.UopNot: &unaryBuiltin{name: "operator!", function: builtinNegation, params: ast.Identifiers{"x"}},
ast.UopBitwiseNot: &unaryBuiltin{name: "operator~", function: builtinBitNeg, params: ast.Identifiers{"x"}},
ast.UopPlus: &unaryBuiltin{name: "operator+ (unary)", function: builtinUnaryPlus, params: ast.Identifiers{"x"}},
ast.UopMinus: &unaryBuiltin{name: "operator- (unary)", function: builtinUnaryMinus, params: ast.Identifiers{"x"}},
}
func buildBuiltinMap(builtins []builtin) map[string]evalCallable {
result := make(map[string]evalCallable)
for _, b := range builtins {
result[string(b.Name())] = b
}
return result
}
var funcBuiltins = buildBuiltinMap([]builtin{
builtinID,
&unaryBuiltin{name: "extVar", function: builtinExtVar, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "length", function: builtinLength, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "toString", function: builtinToString, params: ast.Identifiers{"a"}},
&binaryBuiltin{name: "trace", function: builtinTrace, params: ast.Identifiers{"str", "rest"}},
&binaryBuiltin{name: "makeArray", function: builtinMakeArray, params: ast.Identifiers{"sz", "func"}},
&binaryBuiltin{name: "flatMap", function: builtinFlatMap, params: ast.Identifiers{"func", "arr"}},
&binaryBuiltin{name: "join", function: builtinJoin, params: ast.Identifiers{"sep", "arr"}},
&unaryBuiltin{name: "reverse", function: builtinReverse, params: ast.Identifiers{"arr"}},
&binaryBuiltin{name: "filter", function: builtinFilter, params: ast.Identifiers{"func", "arr"}},
&binaryBuiltin{name: "range", function: builtinRange, params: ast.Identifiers{"from", "to"}},
&binaryBuiltin{name: "primitiveEquals", function: primitiveEquals, params: ast.Identifiers{"x", "y"}},
&binaryBuiltin{name: "equals", function: builtinEquals, params: ast.Identifiers{"x", "y"}},
&binaryBuiltin{name: "objectFieldsEx", function: builtinObjectFieldsEx, params: ast.Identifiers{"obj", "hidden"}},
&ternaryBuiltin{name: "objectHasEx", function: builtinObjectHasEx, params: ast.Identifiers{"obj", "fname", "hidden"}},
&unaryBuiltin{name: "type", function: builtinType, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "char", function: builtinChar, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "codepoint", function: builtinCodepoint, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "ceil", function: builtinCeil, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "floor", function: builtinFloor, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "sqrt", function: builtinSqrt, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "sin", function: builtinSin, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "cos", function: builtinCos, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "tan", function: builtinTan, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "asin", function: builtinAsin, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "acos", function: builtinAcos, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "atan", function: builtinAtan, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "log", function: builtinLog, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "exp", function: builtinExp, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "mantissa", function: builtinMantissa, params: ast.Identifiers{"x"}},
&unaryBuiltin{name: "exponent", function: builtinExponent, params: ast.Identifiers{"x"}},
&binaryBuiltin{name: "pow", function: builtinPow, params: ast.Identifiers{"base", "exp"}},
&binaryBuiltin{name: "modulo", function: builtinModulo, params: ast.Identifiers{"x", "y"}},
&unaryBuiltin{name: "md5", function: builtinMd5, params: ast.Identifiers{"x"}},
&ternaryBuiltin{name: "substr", function: builtinSubstr, params: ast.Identifiers{"str", "from", "len"}},
&ternaryBuiltin{name: "splitLimit", function: builtinSplitLimit, params: ast.Identifiers{"str", "c", "maxsplits"}},
&ternaryBuiltin{name: "strReplace", function: builtinStrReplace, params: ast.Identifiers{"str", "from", "to"}},
&unaryBuiltin{name: "base64Decode", function: builtinBase64Decode, params: ast.Identifiers{"str"}},
&unaryBuiltin{name: "base64DecodeBytes", function: builtinBase64DecodeBytes, params: ast.Identifiers{"str"}},
&unaryBuiltin{name: "parseJson", function: builtinParseJSON, params: ast.Identifiers{"str"}},
&unaryBuiltin{name: "base64", function: builtinBase64, params: ast.Identifiers{"input"}},
&unaryBuiltin{name: "encodeUTF8", function: builtinEncodeUTF8, params: ast.Identifiers{"str"}},
&unaryBuiltin{name: "decodeUTF8", function: builtinDecodeUTF8, params: ast.Identifiers{"arr"}},
&generalBuiltin{name: "sort", function: builtinSort, params: []generalBuiltinParameter{{name: "arr"}, {name: "keyF", defaultValue: functionID}}},
&unaryBuiltin{name: "native", function: builtinNative, params: ast.Identifiers{"x"}},
// internal
&unaryBuiltin{name: "$objectFlatMerge", function: builtinUglyObjectFlatMerge, params: ast.Identifiers{"x"}},
})