This repository has been archived by the owner on Jan 16, 2019. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 37
/
jsx.ex
195 lines (163 loc) · 5.78 KB
/
jsx.ex
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
defmodule JSX do
def encode!(term, opts \\ []) do
parser_opts = :jsx_config.extract_config(opts ++ [:escaped_strings])
parser(:jsx_to_json, opts, parser_opts).(JSX.Encoder.json(term) ++ [:end_json])
end
def encode(term, opts \\ []) do
{ :ok, encode!(term, opts) }
rescue
ArgumentError -> { :error, :badarg }
end
def decode!(json, opts \\ []) do
decoder_opts = :jsx_config.extract_config(opts)
case decoder(JSX.Decoder, opts, decoder_opts).(json) do
{ :incomplete, _ } -> raise ArgumentError
result -> result
end
end
def decode(term, opts \\ []) do
{ :ok, decode!(term, opts) }
rescue
ArgumentError -> { :error, :badarg }
end
def format!(json, opts \\ []) do
case :jsx.format(json, opts) do
{ :incomplete, _ } -> raise ArgumentError
result -> result
end
end
def format(json, opts \\ []) do
{ :ok, format!(json, opts) }
rescue
ArgumentError -> { :error, :badarg }
end
def minify!(json), do: format!(json, [space: 0, indent: 0])
def minify(json) do
{ :ok, minify!(json) }
rescue
ArgumentError -> { :error, :badarg }
end
def prettify!(json), do: format!(json, [space: 1, indent: 2])
def prettify(json) do
{ :ok, prettify!(json) }
rescue
ArgumentError -> { :error, :badarg }
end
def is_json?(json, opts \\ []) do
case :jsx.is_json(json, opts) do
{ :incomplete, _ } -> false
result -> result
end
rescue
_ -> false
end
def is_term?(term, opts \\ []) do
parser_opts = :jsx_config.extract_config(opts)
parser(:jsx_verify, opts, parser_opts).(JSX.Encoder.json(term) ++ [:end_json])
rescue
_ -> false
end
def encoder(handler, initialstate, opts) do
:jsx.encoder(handler, initialstate, opts)
end
def decoder(handler, initialstate, opts) do
:jsx.decoder(handler, initialstate, opts)
end
def parser(handler, initialstate, opts) do
:jsx.parser(handler, initialstate, opts)
end
end
defmodule JSX.Decoder do
def init(opts) do
:jsx_to_term.init(opts)
end
def handle_event(:end_json, state), do: get_value(state)
def handle_event(:start_object, state), do: start_object(state)
def handle_event(:end_object, state), do: finish(state)
def handle_event(:start_array, state), do: start_array(state)
def handle_event(:end_array, state), do: finish(state)
def handle_event({ :key, key }, { _, config } = state), do: insert(format_key(key, config), state)
def handle_event({ :literal, :null }, state), do: insert(:nil, state)
def handle_event({ _, event }, state), do: insert(event, state)
defp format_key(key, { _, :binary, _ }), do: key
defp format_key(key, { _, :atom, _ }), do: :erlang.binary_to_atom(key, :utf8)
defp format_key(key, { _, :existing_atom, _ }), do: :erlang.binary_to_existing_atom(key, :utf8)
defp format_key(key, { _, :attempt_atom, _ }) do
:erlang.binary_to_existing_atom(key, :utf8)
rescue
ArgumentError -> key
end
defp start_object({ stack, config }), do: { [{ :object, %{} }] ++ stack, config }
defp start_array({ stack, config }), do: { [{ :array, [] }] ++ stack, config }
defp finish({ [{ :object, emptyMap }], config }) when is_map(emptyMap) and map_size(emptyMap) < 1 do
{ %{}, config }
end
defp finish({ [{ :object, emptyMap }|rest], config }) when is_map(emptyMap) and map_size(emptyMap) < 1 do
insert(%{}, { rest, config })
end
defp finish({ [{ :object, pairs }], config }), do: { pairs, config }
defp finish({ [{ :object, pairs}|rest], config }), do: insert(pairs, { rest, config })
defp finish({ [{ :array, values }], config }), do: { Enum.reverse(values), config }
defp finish({ [{ :array, values}|rest], config}), do: insert(Enum.reverse(values), { rest, config })
defp finish(_), do: raise ArgumentError
defp insert(value, { [], config }), do: { value, config }
defp insert(key, { [{ :object, pairs }|rest], config }) do
{ [{ :object, key, pairs }] ++ rest, config }
end
defp insert(value, { [{ :object, key, pairs }|rest], config }) do
{ [{ :object, Map.put(pairs, key, value) }] ++ rest, config }
end
defp insert(value, { [{ :array, values}|rest], config }) do
{ [{ :array, [value] ++ values}] ++ rest, config }
end
defp insert(_, _), do: raise ArgumentError
defp get_value({ value, _config }), do: value
defp get_value(_), do: raise ArgumentError
end
defprotocol JSX.Encoder do
@fallback_to_any true
def json(term)
end
defimpl JSX.Encoder, for: Map do
def json(map) do
[:start_object] ++ unpack(map, Map.keys(map))
end
defp unpack(map, [k|rest]) when is_integer(k) or is_binary(k) or is_atom(k) do
[k] ++ JSX.Encoder.json(Map.get(map, k)) ++ unpack(map, rest)
end
defp unpack(_, []), do: [:end_object]
end
defimpl JSX.Encoder, for: List do
def json([]), do: [:start_array, :end_array]
def json([{}]), do: [:start_object, :end_object]
def json([{ _, _ }|_] = list) do
[:start_object] ++ unzip(list)
end
def json(list) do
[:start_array] ++ unhitch(list)
end
defp unzip([{k, v}|rest]) when is_integer(k) or is_binary(k) or is_atom(k) do
[k] ++ JSX.Encoder.json(v) ++ unzip(rest)
end
defp unzip([]), do: [:end_object]
defp unhitch([v|rest]) do
JSX.Encoder.json(v) ++ unhitch(rest)
end
defp unhitch([]), do: [:end_array]
end
defimpl JSX.Encoder, for: HashDict do
def json(dict), do: JSX.Encoder.json(HashDict.to_list(dict))
end
defimpl JSX.Encoder, for: Atom do
def json(nil), do: [:null]
def json(true), do: [true]
def json(false), do: [false]
def json(atom), do: [atom]
end
defimpl JSX.Encoder, for: [Integer, Float, BitString] do
def json(value), do: [value]
end
defimpl JSX.Encoder, for: [Tuple, PID, Port, Reference, Function, Any] do
def json(map) when is_map(map), do: JSX.Encoder.Map.json(Map.delete(map, :__struct__))
def json(_), do: raise ArgumentError
end