/
transcoder.ex
90 lines (74 loc) · 2.45 KB
/
transcoder.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
defmodule JSONC.Transcoder do
import JSONC.Parser
@doc """
transcodes a jsonc document into a json document and raises in case of error
"""
def transcode!(content) when is_binary(content) do
case transcode(content) do
{:ok, result} -> result
{:error, reason} -> raise reason
end
end
@doc """
transcodes a jsonc document into a json document
"""
def transcode(content) when is_binary(content) do
case parse(content) do
{:ok, %{type: :root, value: value}} ->
{:ok, encode_value(value)}
{:error, reason} ->
{:error, reason}
end
end
defp encode_value(value, level \\ 1) when is_map(value) do
case value do
%{type: :object, value: object_node} ->
"{\n#{String.duplicate(" ", level * 4)}#{encode_object(object_node, level)}\n#{String.duplicate(" ", (level - 1) * 4)}}"
%{type: :array, value: array_node} ->
"[\n#{String.duplicate(" ", level * 4)}#{encode_array(array_node, level)}\n#{String.duplicate(" ", (level - 1) * 4)}]"
%{type: :number, subtype: :integer, value: integer} ->
Integer.to_string(integer)
%{type: :number, subtype: :float, value: float} ->
Float.to_string(float)
%{type: :string, subtype: _, value: string} ->
string =
string
|> String.replace("\n", "\\n")
|> String.replace("\t", "\\t")
|> String.replace("\r", "\\r")
|> String.replace("\v", "\\v")
|> String.replace("\b", "\\b")
|> String.replace("\f", "\\f")
"\"#{string}\""
%{type: :boolean, value: boolean} ->
boolean |> to_string()
%{type: nil} ->
"null"
end
end
defp encode_object(node, level) when is_map(node) do
node
|> Enum.sort(fn {_, v1}, {_, v2} -> sort_values(v1, v2) end)
|> Enum.map(fn {k, v} -> "\"#{k}\": #{encode_value(v, level + 1)}" end)
|> Enum.join(",\n#{String.duplicate(" ", level * 4)}")
end
defp encode_array(node, level) when is_list(node) do
node
|> Enum.sort(&sort_values/2)
|> Enum.map(fn v -> "#{encode_value(v, level + 1)}" end)
|> Enum.join(",\n#{String.duplicate(" ", level * 4)}")
end
defp sort_values(%{place: {v1_line, v1_column}}, %{place: {v2_line, v2_column}}) do
cond do
v1_line > v2_line ->
false
v1_line == v2_line ->
cond do
v1_column > v2_column -> false
true -> true
end
true ->
true
end
end
end