|
1 | 1 | package vibes |
2 | 2 |
|
3 | 3 | import ( |
4 | | - "encoding/json" |
5 | 4 | "fmt" |
6 | | - "io" |
7 | | - "reflect" |
8 | 5 | "regexp" |
9 | | - "strings" |
10 | 6 | "unicode/utf8" |
11 | 7 | ) |
12 | 8 |
|
13 | | -func builtinJSONParse(exec *Execution, receiver Value, args []Value, kwargs map[string]Value, block Value) (Value, error) { |
14 | | - if len(args) != 1 || args[0].Kind() != KindString { |
15 | | - return NewNil(), fmt.Errorf("JSON.parse expects a single JSON string argument") |
16 | | - } |
17 | | - if len(kwargs) > 0 { |
18 | | - return NewNil(), fmt.Errorf("JSON.parse does not accept keyword arguments") |
19 | | - } |
20 | | - if !block.IsNil() { |
21 | | - return NewNil(), fmt.Errorf("JSON.parse does not accept blocks") |
22 | | - } |
23 | | - |
24 | | - raw := args[0].String() |
25 | | - if len(raw) > maxJSONPayloadBytes { |
26 | | - return NewNil(), fmt.Errorf("JSON.parse input exceeds limit %d bytes", maxJSONPayloadBytes) |
27 | | - } |
28 | | - |
29 | | - decoder := json.NewDecoder(strings.NewReader(raw)) |
30 | | - decoder.UseNumber() |
31 | | - |
32 | | - var decoded any |
33 | | - if err := decoder.Decode(&decoded); err != nil { |
34 | | - return NewNil(), fmt.Errorf("JSON.parse invalid JSON: %v", err) |
35 | | - } |
36 | | - if err := decoder.Decode(&struct{}{}); err != io.EOF { |
37 | | - return NewNil(), fmt.Errorf("JSON.parse invalid JSON: trailing data") |
38 | | - } |
39 | | - |
40 | | - value, err := jsonValueToVibeValue(decoded) |
41 | | - if err != nil { |
42 | | - return NewNil(), err |
43 | | - } |
44 | | - return value, nil |
45 | | -} |
46 | | - |
47 | | -func builtinJSONStringify(exec *Execution, receiver Value, args []Value, kwargs map[string]Value, block Value) (Value, error) { |
48 | | - if len(args) != 1 { |
49 | | - return NewNil(), fmt.Errorf("JSON.stringify expects a single value argument") |
50 | | - } |
51 | | - if len(kwargs) > 0 { |
52 | | - return NewNil(), fmt.Errorf("JSON.stringify does not accept keyword arguments") |
53 | | - } |
54 | | - if !block.IsNil() { |
55 | | - return NewNil(), fmt.Errorf("JSON.stringify does not accept blocks") |
56 | | - } |
57 | | - |
58 | | - state := &jsonStringifyState{ |
59 | | - seenArrays: map[uintptr]struct{}{}, |
60 | | - seenHashes: map[uintptr]struct{}{}, |
61 | | - } |
62 | | - encoded, err := vibeValueToJSONValue(args[0], state) |
63 | | - if err != nil { |
64 | | - return NewNil(), err |
65 | | - } |
66 | | - |
67 | | - payload, err := json.Marshal(encoded) |
68 | | - if err != nil { |
69 | | - return NewNil(), fmt.Errorf("JSON.stringify failed: %v", err) |
70 | | - } |
71 | | - if len(payload) > maxJSONPayloadBytes { |
72 | | - return NewNil(), fmt.Errorf("JSON.stringify output exceeds limit %d bytes", maxJSONPayloadBytes) |
73 | | - } |
74 | | - return NewString(string(payload)), nil |
75 | | -} |
76 | | - |
77 | | -func jsonValueToVibeValue(val any) (Value, error) { |
78 | | - switch v := val.(type) { |
79 | | - case nil: |
80 | | - return NewNil(), nil |
81 | | - case bool: |
82 | | - return NewBool(v), nil |
83 | | - case string: |
84 | | - return NewString(v), nil |
85 | | - case json.Number: |
86 | | - if i, err := v.Int64(); err == nil { |
87 | | - return NewInt(i), nil |
88 | | - } |
89 | | - f, err := v.Float64() |
90 | | - if err != nil { |
91 | | - return NewNil(), fmt.Errorf("JSON.parse invalid number %q", v.String()) |
92 | | - } |
93 | | - return NewFloat(f), nil |
94 | | - case float64: |
95 | | - return NewFloat(v), nil |
96 | | - case []any: |
97 | | - arr := make([]Value, len(v)) |
98 | | - for i, item := range v { |
99 | | - converted, err := jsonValueToVibeValue(item) |
100 | | - if err != nil { |
101 | | - return NewNil(), err |
102 | | - } |
103 | | - arr[i] = converted |
104 | | - } |
105 | | - return NewArray(arr), nil |
106 | | - case map[string]any: |
107 | | - obj := make(map[string]Value, len(v)) |
108 | | - for key, item := range v { |
109 | | - converted, err := jsonValueToVibeValue(item) |
110 | | - if err != nil { |
111 | | - return NewNil(), err |
112 | | - } |
113 | | - obj[key] = converted |
114 | | - } |
115 | | - return NewHash(obj), nil |
116 | | - default: |
117 | | - return NewNil(), fmt.Errorf("JSON.parse unsupported value type %T", val) |
118 | | - } |
119 | | -} |
120 | | - |
121 | | -func vibeValueToJSONValue(val Value, state *jsonStringifyState) (any, error) { |
122 | | - switch val.Kind() { |
123 | | - case KindNil: |
124 | | - return nil, nil |
125 | | - case KindBool: |
126 | | - return val.Bool(), nil |
127 | | - case KindInt: |
128 | | - return val.Int(), nil |
129 | | - case KindFloat: |
130 | | - return val.Float(), nil |
131 | | - case KindString, KindSymbol: |
132 | | - return val.String(), nil |
133 | | - case KindArray: |
134 | | - arr := val.Array() |
135 | | - id := reflect.ValueOf(arr).Pointer() |
136 | | - if id != 0 { |
137 | | - if _, seen := state.seenArrays[id]; seen { |
138 | | - return nil, fmt.Errorf("JSON.stringify does not support cyclic arrays") |
139 | | - } |
140 | | - state.seenArrays[id] = struct{}{} |
141 | | - defer delete(state.seenArrays, id) |
142 | | - } |
143 | | - |
144 | | - out := make([]any, len(arr)) |
145 | | - for i, item := range arr { |
146 | | - converted, err := vibeValueToJSONValue(item, state) |
147 | | - if err != nil { |
148 | | - return nil, err |
149 | | - } |
150 | | - out[i] = converted |
151 | | - } |
152 | | - return out, nil |
153 | | - case KindHash, KindObject: |
154 | | - hash := val.Hash() |
155 | | - id := reflect.ValueOf(hash).Pointer() |
156 | | - if id != 0 { |
157 | | - if _, seen := state.seenHashes[id]; seen { |
158 | | - return nil, fmt.Errorf("JSON.stringify does not support cyclic objects") |
159 | | - } |
160 | | - state.seenHashes[id] = struct{}{} |
161 | | - defer delete(state.seenHashes, id) |
162 | | - } |
163 | | - |
164 | | - out := make(map[string]any, len(hash)) |
165 | | - for key, item := range hash { |
166 | | - converted, err := vibeValueToJSONValue(item, state) |
167 | | - if err != nil { |
168 | | - return nil, err |
169 | | - } |
170 | | - out[key] = converted |
171 | | - } |
172 | | - return out, nil |
173 | | - default: |
174 | | - return nil, fmt.Errorf("JSON.stringify unsupported value type %s", val.Kind()) |
175 | | - } |
176 | | -} |
177 | | - |
178 | 9 | func builtinRegexMatch(exec *Execution, receiver Value, args []Value, kwargs map[string]Value, block Value) (Value, error) { |
179 | 10 | if len(args) != 2 { |
180 | 11 | return NewNil(), fmt.Errorf("Regex.match expects pattern and text") |
|
0 commit comments