number parsing and nested array
[scpubgit/TenDotTcl.git] / json.tcl
1 namespace eval ten::json {
2
3   namespace eval deparse {
4
5     variable quotes [
6       list "\"" "\\\"" / \\/ \\ \\\\ \b \\b \f \\f \n \\n \r \\r \t \\t
7     ]
8
9     variable indent
10     variable indentBy 0
11     variable indentIncr
12     variable nl
13
14     proc indent_one {} {
15       variable indentIncr
16       variable indentBy [ expr $indentBy + $indentIncr ]
17       variable indent [ string repeat " " $indentBy ]
18     }
19
20     proc outdent_one {} {
21       variable indentIncr
22       variable indentBy [ expr $indentBy - $indentIncr ]
23       variable indent [ string repeat " " $indentBy ]
24     }
25
26     proc str {str} {
27       variable quotes
28       return \"[ string map $quotes $str ]\"
29     }
30
31     proc num {num} {
32       return $num
33     }
34
35     proc list {args} {
36       variable indent
37       variable nl
38       set out \[
39       indent_one
40       foreach el [lrange $args 0 end] {
41         append out $nl$indent
42         append out [ deparse $el ],
43       }
44       outdent_one
45       append out $nl$indent\]
46       return $out
47     }
48
49     proc obj {args} {
50       variable indent
51       variable nl
52       set out \{
53       indent_one
54       dict for {k v} $args {
55         append out $nl$indent[ str $k ]:\ [ deparse $v ],
56       }
57       outdent_one
58       append out $nl$indent\}
59     }
60
61     proc deparse {data} {
62       switch -regexp [lindex $data 0] {
63         ^true|false|null$  { lindex $data 0 }
64         ^num|str|obj|list$ { eval $data }
65         default { error [ concat "Invalid JSON type " [lindex $data 0 0] ] }
66       }
67     }
68   }
69
70   proc deparse_json {data {indentIncr 2}} {
71     set deparse::indentBy 0
72     set deparse::indentIncr $indentIncr
73     if [expr $indentIncr == 0] {
74       set deparse::nl ""
75     } else {
76       set deparse::nl "\n"
77     }
78     deparse::deparse $data
79   }
80
81   namespace eval tclify {
82
83     proc str {str} { return $str }
84
85     proc num {num} { return $num }
86
87     proc obj {args} {
88       return $args
89     }
90
91     proc tclify {data} {
92       switch -regexp [lindex $data 0] {
93         ^true|false|null$ { uplevel 1 return [lindex $data 0] }
94         ^num|str|obj|list$ {}
95         default { error [ concat "Invalid JSON type " [lindex $data 0 0] ] }
96       }
97       eval $data
98     }
99   }
100
101   proc tclify_json {data} {
102     tclify::tclify $data
103   }
104
105   namespace eval parse {
106
107     variable json
108
109     proc eat_spaces {} {
110       variable json
111       set json [string trimleft $json]
112     }
113
114     proc eat_char {char} {
115       variable json
116       eat_spaces
117       if {[string index $json 0] eq "$char"} {
118         eat_any
119       }
120     }
121
122     proc eat_any {} {
123       variable json
124       set json [ string range $json 1 end ]
125     }
126
127     proc parse_list {} {
128       variable json
129       eat_any
130       set tcl {list}
131       while {"$json" ne ""} {
132         if {[string index $json 0] eq "]"} {
133           eat_any
134           return $tcl
135         }
136         lappend tcl [ parse ]
137         eat_char ,
138       }
139       error "Ran out of JSON. Confused now."
140     }
141
142     proc parse_obj {} {
143       variable json
144       eat_char
145       set tcl {obj}
146       while {"$json" ne ""} {
147         if {[string index $json 0] eq "]"} {
148           eat_char
149           return $tcl
150         }
151         lappend tcl [ 
152         lappend tcl [ parse ]
153         eat_char ,
154       }
155       error "Ran out of JSON. Confused now."
156     }
157
158     proc parse_str {} {
159       variable json
160       # like Text::Balanced except ugly (borrowed from tcvJSON's code)
161       set reStr {(?:(?:\")(?:[^\\\"]*(?:\\.[^\\\"]*)*)(?:\"))}
162       if {![regexp $reStr $json string]} {
163         error [ concat "Invalid string: " [string range $json 0 32]
164       }
165       set json [string range $json [string length $string] end]
166       # chop off outer ""s and substitute backslashes
167       # This does more than the RFC-specified backslash sequences,
168       # but it does cover them all
169       list str [subst -nocommand -novariable [string range $string 1 end-1]]
170     }
171
172     proc parse_num {} {
173       variable json
174       string is double -failindex last $json
175       if {$last == 0} {
176         error "Saw number but wasn't - $json"
177       }
178       set num [string range $json 0 [expr {$last - 1}]]
179       set json [string range $json $last end]
180       list num $num
181     }
182
183     proc parse {} {
184       variable json
185       eat_spaces
186       if {$json eq ""} {
187         return
188       }
189       switch -regexp [string index $json 0] {
190         {\{} { parse_obj }
191         {\[} { parse_list }
192         {\"} { parse_str }
193
194         {[-0-9]} { parse_num }
195
196         default { error "argh" }
197       }
198     }
199   }
200
201   proc parse_json {json} {
202     set parse::json [ string trim $json ]
203     set result [ parse::parse ]
204     parse::eat_spaces
205     if {$parse::json ne ""} {
206       error "Had JSON left over: $parse::json"
207     }
208     return $result
209   }
210 }
211
212 set ex_json { list {str foo} {num 0} {obj __remote_object__ {str 512}} {null} }
213
214 set jtext {
215   [
216     "foo",
217     0,
218     {
219       "__remote_object__": "512",
220     },
221     null,
222   ]
223 }
224
225 puts [ ten::json::deparse_json $ex_json 2 ]
226
227 dict for {k v} [ ten::json::tclify_json [
228   lindex [ ten::json::tclify_json $ex_json ] 2
229 ] ] { puts "$k: $v" }
230
231 #puts [ ten::json::parse_json $jtext ]
232
233 puts [ ten::json::parse_json {["foo",2345,["bar"]]} ]