adamc@25
|
1 (* Copyright (c) 2008, Adam Chlipala
|
adamc@25
|
2 * All rights reserved.
|
adamc@25
|
3 *
|
adamc@25
|
4 * Redistribution and use in source and binary forms, with or without
|
adamc@25
|
5 * modification, are permitted provided that the following conditions are met:
|
adamc@25
|
6 *
|
adamc@25
|
7 * - Redistributions of source code must retain the above copyright notice,
|
adamc@25
|
8 * this list of conditions and the following disclaimer.
|
adamc@25
|
9 * - Redistributions in binary form must reproduce the above copyright notice,
|
adamc@25
|
10 * this list of conditions and the following disclaimer in the documentation
|
adamc@25
|
11 * and/or other materials provided with the distribution.
|
adamc@25
|
12 * - The names of contributors may not be used to endorse or promote products
|
adamc@25
|
13 * derived from this software without specific prior written permission.
|
adamc@25
|
14 *
|
adamc@25
|
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
adamc@25
|
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
adamc@25
|
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
adamc@25
|
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
adamc@25
|
19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
adamc@25
|
20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
adamc@25
|
21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
adamc@25
|
22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
adamc@25
|
23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
adamc@25
|
24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
adamc@25
|
25 * POSSIBILITY OF SUCH DAMAGE.
|
adamc@25
|
26 *)
|
adamc@25
|
27
|
adamc@244
|
28 (* Pretty-printing monomorphic Ur/Web *)
|
adamc@25
|
29
|
adamc@25
|
30 structure MonoPrint :> MONO_PRINT = struct
|
adamc@25
|
31
|
adamc@25
|
32 open Print.PD
|
adamc@25
|
33 open Print
|
adamc@25
|
34
|
adamc@25
|
35 open Mono
|
adamc@25
|
36
|
adamc@25
|
37 structure E = MonoEnv
|
adamc@25
|
38
|
adamc@25
|
39 val debug = ref false
|
adamc@25
|
40
|
adamc@252
|
41 val dummyt = (TRecord [], ErrorMsg.dummySpan)
|
adamc@252
|
42
|
adamc@25
|
43 fun p_typ' par env (t, _) =
|
adamc@25
|
44 case t of
|
adamc@25
|
45 TFun (t1, t2) => parenIf par (box [p_typ' true env t1,
|
adamc@25
|
46 space,
|
adamc@25
|
47 string "->",
|
adamc@25
|
48 space,
|
adamc@25
|
49 p_typ env t2])
|
adamc@25
|
50 | TRecord xcs => box [string "{",
|
adamc@25
|
51 p_list (fn (x, t) =>
|
adamc@25
|
52 box [string x,
|
adamc@25
|
53 space,
|
adamc@25
|
54 string ":",
|
adamc@25
|
55 space,
|
adamc@25
|
56 p_typ env t]) xcs,
|
adamc@25
|
57 string "}"]
|
adamc@196
|
58 | TDatatype (n, _) =>
|
adamc@178
|
59 ((if !debug then
|
adamc@178
|
60 string (#1 (E.lookupDatatype env n) ^ "__" ^ Int.toString n)
|
adamc@178
|
61 else
|
adamc@178
|
62 string (#1 (E.lookupDatatype env n)))
|
adamc@178
|
63 handle E.UnboundNamed _ => string ("UNBOUND_DATATYPE_" ^ Int.toString n))
|
adamc@51
|
64 | TFfi (m, x) => box [string "FFI(", string m, string ".", string x, string ")"]
|
adamc@292
|
65 | TOption t => box [string "option(",
|
adamc@292
|
66 p_typ env t,
|
adamc@292
|
67 string ")"]
|
adamc@577
|
68 | TSource => string "source"
|
adamc@568
|
69 | TSignal t => box [string "signal(",
|
adamc@568
|
70 p_typ env t,
|
adamc@568
|
71 string ")"]
|
adamc@25
|
72
|
adamc@25
|
73 and p_typ env = p_typ' false env
|
adamc@25
|
74
|
adamc@109
|
75 fun p_enamed env n =
|
adamc@178
|
76 (if !debug then
|
adamc@178
|
77 string (#1 (E.lookupENamed env n) ^ "__" ^ Int.toString n)
|
adamc@178
|
78 else
|
adamc@178
|
79 string (#1 (E.lookupENamed env n)))
|
adamc@178
|
80 handle E.UnboundNamed _ => string ("UNBOUNDN_" ^ Int.toString n)
|
adamc@178
|
81
|
adamc@178
|
82 fun p_con_named env n =
|
adamc@178
|
83 (if !debug then
|
adamc@178
|
84 string (#1 (E.lookupConstructor env n) ^ "__" ^ Int.toString n)
|
adamc@178
|
85 else
|
adamc@178
|
86 string (#1 (E.lookupConstructor env n)))
|
adamc@178
|
87 handle E.UnboundNamed _ => string ("CONSTRUCTOR_" ^ Int.toString n)
|
adamc@178
|
88
|
adamc@178
|
89 fun p_patCon env pc =
|
adamc@178
|
90 case pc of
|
adamc@178
|
91 PConVar n => p_con_named env n
|
adamc@186
|
92 | PConFfi {mod = m, con, ...} => box [string "FFIC(",
|
adamc@185
|
93 string m,
|
adamc@185
|
94 string ".",
|
adamc@185
|
95 string con,
|
adamc@185
|
96 string ")"]
|
adamc@178
|
97
|
adamc@178
|
98 fun p_pat' par env (p, _) =
|
adamc@178
|
99 case p of
|
adamc@178
|
100 PWild => string "_"
|
adamc@182
|
101 | PVar (s, _) => string s
|
adamc@178
|
102 | PPrim p => Prim.p_t p
|
adamc@188
|
103 | PCon (_, n, NONE) => p_patCon env n
|
adamc@188
|
104 | PCon (_, n, SOME p) => parenIf par (box [p_patCon env n,
|
adamc@288
|
105 space,
|
adamc@288
|
106 p_pat' true env p])
|
adamc@178
|
107 | PRecord xps =>
|
adamc@178
|
108 box [string "{",
|
adamc@182
|
109 p_list_sep (box [string ",", space]) (fn (x, p, _) =>
|
adamc@178
|
110 box [string x,
|
adamc@178
|
111 space,
|
adamc@178
|
112 string "=",
|
adamc@178
|
113 space,
|
adamc@178
|
114 p_pat env p]) xps,
|
adamc@178
|
115 string "}"]
|
adamc@288
|
116 | PNone _ => string "None"
|
adamc@288
|
117 | PSome (_, p) => box [string "Some",
|
adamc@288
|
118 space,
|
adamc@288
|
119 p_pat' true env p]
|
adamc@178
|
120
|
adamc@178
|
121 and p_pat x = p_pat' false x
|
adamc@109
|
122
|
adamc@591
|
123 fun p_mode env m =
|
adamc@591
|
124 case m of
|
adamc@591
|
125 Attribute => string "Attribute"
|
adamc@591
|
126 | Script => string "Script"
|
adamc@591
|
127 | Source t => box [string "Source", space, p_typ env t]
|
adamc@591
|
128
|
adamc@25
|
129 fun p_exp' par env (e, _) =
|
adamc@25
|
130 case e of
|
adamc@25
|
131 EPrim p => Prim.p_t p
|
adamc@25
|
132 | ERel n =>
|
adamc@178
|
133 ((if !debug then
|
adamc@178
|
134 string (#1 (E.lookupERel env n) ^ "_" ^ Int.toString n)
|
adamc@178
|
135 else
|
adamc@178
|
136 string (#1 (E.lookupERel env n)))
|
adamc@178
|
137 handle E.UnboundRel _ => string ("UNBOUND_" ^ Int.toString n))
|
adamc@109
|
138 | ENamed n => p_enamed env n
|
adamc@188
|
139 | ECon (_, pc, NONE) => p_patCon env pc
|
adamc@188
|
140 | ECon (_, pc, SOME e) => parenIf par (box [p_patCon env pc,
|
adamc@188
|
141 space,
|
adamc@188
|
142 p_exp' true env e])
|
adamc@297
|
143 | ENone _ => string "None"
|
adamc@290
|
144 | ESome (_, e) => parenIf par (box [string "Some",
|
adamc@290
|
145 space,
|
adamc@290
|
146 p_exp' true env e])
|
adamc@109
|
147
|
adamc@51
|
148 | EFfi (m, x) => box [string "FFI(", string m, string ".", string x, string ")"]
|
adamc@51
|
149 | EFfiApp (m, x, es) => box [string "FFI(",
|
adamc@51
|
150 string m,
|
adamc@51
|
151 string ".",
|
adamc@51
|
152 string x,
|
adamc@51
|
153 string "(",
|
adamc@51
|
154 p_list (p_exp env) es,
|
adamc@51
|
155 string "))"]
|
adamc@25
|
156 | EApp (e1, e2) => parenIf par (box [p_exp env e1,
|
adamc@25
|
157 space,
|
adamc@25
|
158 p_exp' true env e2])
|
adamc@252
|
159 | EAbs (x, t, _, e) => parenIf true (box [string "fn",
|
adamc@252
|
160 space,
|
adamc@252
|
161 string x,
|
adamc@252
|
162 space,
|
adamc@252
|
163 string ":",
|
adamc@252
|
164 space,
|
adamc@252
|
165 p_typ env t,
|
adamc@252
|
166 space,
|
adamc@252
|
167 string "=>",
|
adamc@252
|
168 space,
|
adamc@252
|
169 p_exp (E.pushERel env x t NONE) e])
|
adamc@25
|
170
|
adamc@387
|
171 | EUnop (s, e) => parenIf true (box [string s,
|
adamc@387
|
172 space,
|
adamc@387
|
173 p_exp' true env e])
|
adamc@387
|
174 | EBinop (s, e1, e2) => parenIf true (box [p_exp' true env e1,
|
adamc@387
|
175 space,
|
adamc@387
|
176 string s,
|
adamc@387
|
177 space,
|
adamc@387
|
178 p_exp' true env e2])
|
adamc@387
|
179
|
adamc@25
|
180 | ERecord xes => box [string "{",
|
adamc@29
|
181 p_list (fn (x, e, _) =>
|
adamc@25
|
182 box [string x,
|
adamc@25
|
183 space,
|
adamc@25
|
184 string "=",
|
adamc@25
|
185 space,
|
adamc@25
|
186 p_exp env e]) xes,
|
adamc@25
|
187 string "}"]
|
adamc@25
|
188 | EField (e, x) =>
|
adamc@25
|
189 box [p_exp' true env e,
|
adamc@25
|
190 string ".",
|
adamc@25
|
191 string x]
|
adamc@25
|
192
|
adamc@252
|
193 | ECase (e, pes, _) => parenIf true (box [string "case",
|
adamc@252
|
194 space,
|
adamc@252
|
195 p_exp env e,
|
adamc@252
|
196 space,
|
adamc@252
|
197 string "of",
|
adamc@252
|
198 space,
|
adamc@252
|
199 p_list_sep (box [space, string "|", space])
|
adamc@252
|
200 (fn (p, e) => box [p_pat env p,
|
adamc@252
|
201 space,
|
adamc@252
|
202 string "=>",
|
adamc@252
|
203 space,
|
adamc@252
|
204 p_exp (E.patBinds env p) e]) pes])
|
adamc@94
|
205
|
adamc@283
|
206 | EError (e, t) => box [string "(error",
|
adamc@283
|
207 space,
|
adamc@283
|
208 p_exp env e,
|
adamc@283
|
209 space,
|
adamc@283
|
210 string ":",
|
adamc@283
|
211 space,
|
adamc@283
|
212 p_typ env t,
|
adamc@283
|
213 string ")"]
|
adamc@283
|
214
|
adamc@331
|
215 | EStrcat (e1, e2) => parenIf par (box [p_exp' true env e1,
|
adamc@331
|
216 space,
|
adamc@331
|
217 string "^",
|
adamc@331
|
218 space,
|
adamc@331
|
219 p_exp env e2])
|
adamc@94
|
220
|
adamc@102
|
221 | EWrite e => box [string "write(",
|
adamc@102
|
222 p_exp env e,
|
adamc@102
|
223 string ")"]
|
adamc@102
|
224
|
adamc@578
|
225 | ESeq (e1, e2) => box [string "(",
|
adamc@578
|
226 p_exp env e1,
|
adamc@106
|
227 string ";",
|
adamc@106
|
228 space,
|
adamc@578
|
229 p_exp env e2,
|
adamc@578
|
230 string ")"]
|
adamc@252
|
231 | ELet (x, t, e1, e2) => box [string "(let",
|
adamc@251
|
232 space,
|
adamc@251
|
233 string x,
|
adamc@251
|
234 space,
|
adamc@251
|
235 string ":",
|
adamc@251
|
236 space,
|
adamc@251
|
237 p_typ env t,
|
adamc@251
|
238 space,
|
adamc@251
|
239 string "=",
|
adamc@251
|
240 space,
|
adamc@252
|
241 string "(",
|
adamc@251
|
242 p_exp env e1,
|
adamc@252
|
243 string ")",
|
adamc@251
|
244 space,
|
adamc@251
|
245 string "in",
|
adamc@251
|
246 space,
|
adamc@252
|
247 string "(",
|
adamc@252
|
248 p_exp (E.pushERel env x t NONE) e2,
|
adamc@252
|
249 string "))"]
|
adamc@106
|
250
|
adamc@111
|
251 | EClosure (n, es) => box [string "CLOSURE(",
|
adamc@111
|
252 p_enamed env n,
|
adamc@111
|
253 p_list_sep (string "") (fn e => box [string ", ",
|
adamc@111
|
254 p_exp env e]) es,
|
adamc@111
|
255 string ")"]
|
adamc@111
|
256
|
adamc@252
|
257 | EQuery {exps, tables, state, query, body, initial} =>
|
adamc@252
|
258 box [string "query[",
|
adamc@252
|
259 p_list (fn (x, t) => box [string x, space, string ":", space, p_typ env t]) exps,
|
adamc@252
|
260 string "] [",
|
adamc@252
|
261 p_list (fn (x, xts) => box [string x,
|
adamc@252
|
262 space,
|
adamc@252
|
263 string ":",
|
adamc@252
|
264 space,
|
adamc@252
|
265 string "{",
|
adamc@252
|
266 p_list (fn (x, t) => box [string x, space, string ":", space, p_typ env t]) xts,
|
adamc@252
|
267 string "}"]) tables,
|
adamc@252
|
268 string "] [",
|
adamc@252
|
269 p_typ env state,
|
adamc@252
|
270 string "]",
|
adamc@252
|
271 space,
|
adamc@252
|
272 p_exp env query,
|
adamc@252
|
273 space,
|
adamc@252
|
274 string "initial",
|
adamc@252
|
275 space,
|
adamc@252
|
276 p_exp env initial,
|
adamc@252
|
277 space,
|
adamc@252
|
278 string "in",
|
adamc@252
|
279 space,
|
adamc@252
|
280 p_exp (E.pushERel (E.pushERel env "r" dummyt NONE) "acc" dummyt NONE) body]
|
adamc@307
|
281 | EDml e => box [string "dml(",
|
adamc@307
|
282 p_exp env e,
|
adamc@307
|
283 string ")"]
|
adamc@338
|
284 | ENextval e => box [string "nextval(",
|
adamc@338
|
285 p_exp env e,
|
adamc@338
|
286 string ")"]
|
adamc@463
|
287 | EUnurlify (e, _) => box [string "unurlify(",
|
adamc@463
|
288 p_exp env e,
|
adamc@463
|
289 string ")"]
|
adamc@591
|
290 | EJavaScript (m, e, NONE) => box [string "JavaScript(",
|
adamc@591
|
291 p_mode env m,
|
adamc@591
|
292 string ",",
|
adamc@591
|
293 space,
|
adamc@578
|
294 p_exp env e,
|
adamc@578
|
295 string ")"]
|
adamc@578
|
296 | EJavaScript (_, _, SOME e) => p_exp env e
|
adamc@568
|
297
|
adamc@568
|
298 | ESignalReturn e => box [string "Return(",
|
adamc@568
|
299 p_exp env e,
|
adamc@568
|
300 string ")"]
|
adamc@574
|
301 | ESignalBind (e1, e2) => box [string "Bind(",
|
adamc@572
|
302 p_exp env e1,
|
adamc@572
|
303 string ",",
|
adamc@572
|
304 space,
|
adamc@572
|
305 p_exp env e2,
|
adamc@572
|
306 string ")"]
|
adamc@574
|
307 | ESignalSource e => box [string "Source(",
|
adamc@574
|
308 p_exp env e,
|
adamc@574
|
309 string ")"]
|
adamc@252
|
310
|
adamc@609
|
311 | EServerCall (n, es, e, _) => box [string "Server(",
|
adamc@609
|
312 string n,
|
adamc@609
|
313 string ",",
|
adamc@609
|
314 space,
|
adamc@609
|
315 p_list (p_exp env) es,
|
adamc@609
|
316 string ")[",
|
adamc@609
|
317 p_exp env e,
|
adamc@609
|
318 string "]"]
|
adamc@608
|
319
|
adamc@25
|
320 and p_exp env = p_exp' false env
|
adamc@25
|
321
|
adamc@126
|
322 fun p_vali env (x, n, t, e, s) =
|
adamc@126
|
323 let
|
adamc@126
|
324 val xp = if !debug then
|
adamc@126
|
325 box [string x,
|
adamc@126
|
326 string "__",
|
adamc@126
|
327 string (Int.toString n)]
|
adamc@126
|
328 else
|
adamc@126
|
329 string x
|
adamc@126
|
330 in
|
adamc@126
|
331 box [xp,
|
adamc@126
|
332 space,
|
adamc@126
|
333 string "as",
|
adamc@126
|
334 space,
|
adamc@126
|
335 string s,
|
adamc@126
|
336 space,
|
adamc@126
|
337 string ":",
|
adamc@126
|
338 space,
|
adamc@126
|
339 p_typ env t,
|
adamc@126
|
340 space,
|
adamc@126
|
341 string "=",
|
adamc@126
|
342 space,
|
adamc@126
|
343 p_exp env e]
|
adamc@126
|
344 end
|
adamc@126
|
345
|
adamc@164
|
346 fun p_datatype env (x, n, cons) =
|
adamc@164
|
347 let
|
adamc@168
|
348 val env = E.pushDatatype env x n cons
|
adamc@164
|
349 in
|
adamc@164
|
350 box [string "datatype",
|
adamc@164
|
351 space,
|
adamc@164
|
352 string x,
|
adamc@164
|
353 space,
|
adamc@164
|
354 string "=",
|
adamc@164
|
355 space,
|
adamc@164
|
356 p_list_sep (box [space, string "|", space])
|
adamc@164
|
357 (fn (x, n, NONE) => if !debug then (string (x ^ "__" ^ Int.toString n))
|
adamc@164
|
358 else string x
|
adamc@164
|
359 | (x, _, SOME t) => box [if !debug then (string (x ^ "__" ^ Int.toString n))
|
adamc@164
|
360 else string x, space, string "of", space, p_typ env t])
|
adamc@164
|
361 cons]
|
adamc@164
|
362 end
|
adamc@164
|
363
|
adamc@126
|
364 fun p_decl env (dAll as (d, _) : decl) =
|
adamc@25
|
365 case d of
|
adamc@164
|
366 DDatatype x => p_datatype env x
|
adamc@164
|
367 | DVal vi => box [string "val",
|
adamc@126
|
368 space,
|
adamc@126
|
369 p_vali env vi]
|
adamc@126
|
370 | DValRec vis =>
|
adamc@25
|
371 let
|
adamc@126
|
372 val env = E.declBinds env dAll
|
adamc@25
|
373 in
|
adamc@25
|
374 box [string "val",
|
adamc@25
|
375 space,
|
adamc@126
|
376 string "rec",
|
adamc@25
|
377 space,
|
adamc@126
|
378 p_list_sep (box [newline, string "and", space]) (p_vali env) vis]
|
adamc@25
|
379 end
|
adamc@109
|
380
|
adamc@609
|
381 | DExport (ek, s, n, ts, t) => box [string "export",
|
adamc@609
|
382 space,
|
adamc@609
|
383 CorePrint.p_export_kind ek,
|
adamc@609
|
384 space,
|
adamc@609
|
385 p_enamed env n,
|
adamc@609
|
386 space,
|
adamc@609
|
387 string "as",
|
adamc@609
|
388 space,
|
adamc@609
|
389 string s,
|
adamc@609
|
390 p_list_sep (string "") (fn t => box [space,
|
adamc@609
|
391 string "(",
|
adamc@609
|
392 p_typ env t,
|
adamc@609
|
393 string ")"]) ts,
|
adamc@609
|
394 space,
|
adamc@609
|
395 string "->",
|
adamc@609
|
396 space,
|
adamc@609
|
397 p_typ env t]
|
adamc@271
|
398
|
adamc@273
|
399 | DTable (s, xts) => box [string "(* SQL table ",
|
adamc@273
|
400 string s,
|
adamc@273
|
401 space,
|
adamc@273
|
402 string ":",
|
adamc@273
|
403 space,
|
adamc@273
|
404 p_list (fn (x, t) => box [string x,
|
adamc@273
|
405 space,
|
adamc@273
|
406 string ":",
|
adamc@273
|
407 space,
|
adamc@273
|
408 p_typ env t]) xts,
|
adamc@273
|
409 space,
|
adamc@273
|
410 string "*)"]
|
adamc@338
|
411 | DSequence s => box [string "(* SQL sequence ",
|
adamc@338
|
412 string s,
|
adamc@338
|
413 string "*)"]
|
adamc@271
|
414 | DDatabase s => box [string "database",
|
adamc@271
|
415 space,
|
adamc@271
|
416 string s]
|
adamc@569
|
417 | DJavaScript s => box [string "JavaScript(",
|
adamc@569
|
418 string s,
|
adamc@569
|
419 string ")"]
|
adamc@569
|
420
|
adamc@100
|
421
|
adamc@25
|
422 fun p_file env file =
|
adamc@25
|
423 let
|
adamc@31
|
424 val (pds, _) = ListUtil.foldlMap (fn (d, env) =>
|
adamc@31
|
425 (p_decl env d,
|
adamc@31
|
426 E.declBinds env d))
|
adamc@25
|
427 env file
|
adamc@25
|
428 in
|
adamc@25
|
429 p_list_sep newline (fn x => x) pds
|
adamc@25
|
430 end
|
adamc@25
|
431
|
adamc@25
|
432 end
|