mirror of
https://github.com/Open-Argon/argon-v3.git
synced 2025-12-06 08:56:07 +00:00
fix infinity in json stringify
This commit is contained in:
105
src/json.go
Normal file
105
src/json.go
Normal file
@@ -0,0 +1,105 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"math"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func convertToArgon(obj any) any {
|
||||
switch x := obj.(type) {
|
||||
case map[string]interface{}:
|
||||
newmap := ArMap{}
|
||||
for key, value := range x {
|
||||
newmap[key] = convertToArgon(value)
|
||||
}
|
||||
return newmap
|
||||
case ArArray:
|
||||
newarray := ArArray{}
|
||||
for _, value := range x {
|
||||
newarray = append(newarray, convertToArgon(value))
|
||||
}
|
||||
return newarray
|
||||
case string:
|
||||
return x
|
||||
case float64:
|
||||
return newNumber().SetFloat64(x)
|
||||
case bool:
|
||||
return x
|
||||
case nil:
|
||||
return nil
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func jsonparse(str string) any {
|
||||
var jsonMap any
|
||||
json.Unmarshal([]byte(str), &jsonMap)
|
||||
return convertToArgon(jsonMap)
|
||||
}
|
||||
|
||||
func jsonstringify(obj any, level int) (string, error) {
|
||||
if level > 100 {
|
||||
return "", errors.New("json stringify error: too many levels")
|
||||
}
|
||||
output := []string{}
|
||||
obj = classVal(obj)
|
||||
switch x := obj.(type) {
|
||||
case ArMap:
|
||||
for key, value := range x {
|
||||
str, err := jsonstringify(value, level+1)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
output = append(output, ""+strconv.Quote(anyToArgon(key, false, true, 3, 0, false, 0))+": "+str)
|
||||
}
|
||||
return "{" + strings.Join(output, ", ") + "}", nil
|
||||
case ArArray:
|
||||
for _, value := range x {
|
||||
str, err := jsonstringify(value, level+1)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
output = append(output, str)
|
||||
}
|
||||
return "[" + strings.Join(output, ", ") + "]", nil
|
||||
case string:
|
||||
return strconv.Quote(x), nil
|
||||
case number:
|
||||
num, _ := x.Float64()
|
||||
if math.IsNaN(num) || math.IsInf(num, 0) {
|
||||
return "null", nil
|
||||
}
|
||||
return numberToString(x, false), nil
|
||||
case bool:
|
||||
return strconv.FormatBool(x), nil
|
||||
case nil:
|
||||
return "null", nil
|
||||
}
|
||||
err := errors.New("Cannot stringify '" + typeof(obj) + "'")
|
||||
return "", err
|
||||
}
|
||||
|
||||
var ArJSON = ArMap{
|
||||
"parse": builtinFunc{"parse", func(args ...any) (any, ArErr) {
|
||||
if len(args) == 0 {
|
||||
return ArMap{}, ArErr{TYPE: "Runtime Error", message: "parse takes 1 argument", EXISTS: true}
|
||||
}
|
||||
if typeof(args[0]) != "string" {
|
||||
return ArMap{}, ArErr{TYPE: "Runtime Error", message: "parse takes a string not a '" + typeof(args[0]) + "'", EXISTS: true}
|
||||
}
|
||||
return jsonparse(args[0].(string)), ArErr{}
|
||||
}},
|
||||
"stringify": builtinFunc{"stringify", func(args ...any) (any, ArErr) {
|
||||
if len(args) == 0 {
|
||||
return ArMap{}, ArErr{TYPE: "Runtime Error", message: "stringify takes 1 argument", EXISTS: true}
|
||||
}
|
||||
str, err := jsonstringify(args[0], 0)
|
||||
if err != nil {
|
||||
return ArMap{}, ArErr{TYPE: "Runtime Error", message: err.Error(), EXISTS: true}
|
||||
}
|
||||
return str, ArErr{}
|
||||
}},
|
||||
}
|
||||
Reference in New Issue
Block a user