adamc@29
|
1 (* Copyright (c) 2008, Adam Chlipala
|
adamc@29
|
2 * All rights reserved.
|
adamc@29
|
3 *
|
adamc@29
|
4 * Redistribution and use in source and binary forms, with or without
|
adamc@29
|
5 * modification, are permitted provided that the following conditions are met:
|
adamc@29
|
6 *
|
adamc@29
|
7 * - Redistributions of source code must retain the above copyright notice,
|
adamc@29
|
8 * this list of conditions and the following disclaimer.
|
adamc@29
|
9 * - Redistributions in binary form must reproduce the above copyright notice,
|
adamc@29
|
10 * this list of conditions and the following disclaimer in the documentation
|
adamc@29
|
11 * and/or other materials provided with the distribution.
|
adamc@29
|
12 * - The names of contributors may not be used to endorse or promote products
|
adamc@29
|
13 * derived from this software without specific prior written permission.
|
adamc@29
|
14 *
|
adamc@29
|
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
adamc@29
|
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
adamc@29
|
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
adamc@29
|
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
adamc@29
|
19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
adamc@29
|
20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
adamc@29
|
21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
adamc@29
|
22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
adamc@29
|
23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
adamc@29
|
24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
adamc@29
|
25 * POSSIBILITY OF SUCH DAMAGE.
|
adamc@29
|
26 *)
|
adamc@29
|
27
|
adamc@29
|
28 (* Pretty-printing C jr. *)
|
adamc@29
|
29
|
adamc@29
|
30 structure CjrPrint :> CJR_PRINT = struct
|
adamc@29
|
31
|
adamc@29
|
32 open Print.PD
|
adamc@29
|
33 open Print
|
adamc@29
|
34
|
adamc@29
|
35 open Cjr
|
adamc@29
|
36
|
adamc@269
|
37 val dummyt = (TRecord 0, ErrorMsg.dummySpan)
|
adamc@269
|
38
|
adamc@29
|
39 structure E = CjrEnv
|
adamc@29
|
40 structure EM = ErrorMsg
|
adamc@29
|
41
|
adamc@144
|
42 structure SK = struct
|
adamc@144
|
43 type ord_key = string
|
adamc@144
|
44 val compare = String.compare
|
adamc@144
|
45 end
|
adamc@144
|
46
|
adamc@144
|
47 structure SS = BinarySetFn(SK)
|
adamc@144
|
48 structure SM = BinaryMapFn(SK)
|
adamc@144
|
49 structure IS = IntBinarySet
|
adamc@144
|
50
|
adamc@144
|
51 structure CM = BinaryMapFn(struct
|
adamc@144
|
52 type ord_key = char
|
adamc@144
|
53 val compare = Char.compare
|
adamc@144
|
54 end)
|
adamc@144
|
55
|
adamc@29
|
56 val debug = ref false
|
adamc@29
|
57
|
adamc@196
|
58 val dummyTyp = (TDatatype (Enum, 0, ref []), ErrorMsg.dummySpan)
|
adamc@29
|
59
|
adamc@29
|
60 fun p_typ' par env (t, loc) =
|
adamc@29
|
61 case t of
|
adamc@269
|
62 TFun (t1, t2) => parenIf par (box [p_typ' true env t2,
|
adamc@109
|
63 space,
|
adamc@109
|
64 string "(*)",
|
adamc@109
|
65 space,
|
adamc@109
|
66 string "(",
|
adamc@109
|
67 p_typ env t1,
|
adamc@109
|
68 string ")"])
|
adamc@29
|
69 | TRecord i => box [string "struct",
|
adamc@29
|
70 space,
|
adamc@29
|
71 string "__lws_",
|
adamc@29
|
72 string (Int.toString i)]
|
adamc@188
|
73 | TDatatype (Enum, n, _) =>
|
adamc@188
|
74 (box [string "enum",
|
adamc@188
|
75 space,
|
adamc@188
|
76 string ("__lwe_" ^ #1 (E.lookupDatatype env n) ^ "_" ^ Int.toString n)]
|
adamc@188
|
77 handle CjrEnv.UnboundNamed _ => string ("__lwd_UNBOUND__" ^ Int.toString n))
|
adamc@198
|
78 | TDatatype (Option, n, xncs) =>
|
adamc@198
|
79 (case ListUtil.search #3 (!xncs) of
|
adamc@198
|
80 NONE => raise Fail "CjrPrint: TDatatype marked Option has no constructor with an argument"
|
adamc@198
|
81 | SOME t =>
|
adamc@198
|
82 case #1 t of
|
adamc@198
|
83 TDatatype _ => p_typ' par env t
|
adamc@199
|
84 | TFfi ("Basis", "string") => p_typ' par env t
|
adamc@198
|
85 | _ => box [p_typ' par env t,
|
adamc@198
|
86 string "*"])
|
adamc@188
|
87 | TDatatype (Default, n, _) =>
|
adamc@165
|
88 (box [string "struct",
|
adamc@165
|
89 space,
|
adamc@166
|
90 string ("__lwd_" ^ #1 (E.lookupDatatype env n) ^ "_" ^ Int.toString n ^ "*")]
|
adamc@166
|
91 handle CjrEnv.UnboundNamed _ => string ("__lwd_UNBOUND__" ^ Int.toString n))
|
adamc@53
|
92 | TFfi (m, x) => box [string "lw_", string m, string "_", string x]
|
adamc@29
|
93
|
adamc@29
|
94 and p_typ env = p_typ' false env
|
adamc@29
|
95
|
adamc@29
|
96 fun p_rel env n = string ("__lwr_" ^ #1 (E.lookupERel env n) ^ "_" ^ Int.toString (E.countERels env - n - 1))
|
adamc@29
|
97 handle CjrEnv.UnboundRel _ => string ("__lwr_UNBOUND_" ^ Int.toString (E.countERels env - n - 1))
|
adamc@29
|
98
|
adamc@109
|
99 fun p_enamed env n =
|
adamc@109
|
100 string ("__lwn_" ^ #1 (E.lookupENamed env n) ^ "_" ^ Int.toString n)
|
adamc@109
|
101 handle CjrEnv.UnboundNamed _ => string ("__lwn_UNBOUND_" ^ Int.toString n)
|
adamc@109
|
102
|
adamc@182
|
103 fun p_con_named env n =
|
adamc@182
|
104 string ("__lwc_" ^ #1 (E.lookupConstructor env n) ^ "_" ^ Int.toString n)
|
adamc@182
|
105 handle CjrEnv.UnboundNamed _ => string ("__lwc_UNBOUND_" ^ Int.toString n)
|
adamc@182
|
106
|
adamc@182
|
107 fun p_pat_preamble env (p, _) =
|
adamc@182
|
108 case p of
|
adamc@182
|
109 PWild => (box [],
|
adamc@182
|
110 env)
|
adamc@182
|
111 | PVar (x, t) => (box [p_typ env t,
|
adamc@182
|
112 space,
|
adamc@182
|
113 string "__lwr_",
|
adamc@182
|
114 string x,
|
adamc@182
|
115 string "_",
|
adamc@182
|
116 string (Int.toString (E.countERels env)),
|
adamc@182
|
117 string ";",
|
adamc@182
|
118 newline],
|
adamc@196
|
119 E.pushERel env x t)
|
adamc@182
|
120 | PPrim _ => (box [], env)
|
adamc@188
|
121 | PCon (_, _, NONE) => (box [], env)
|
adamc@188
|
122 | PCon (_, _, SOME p) => p_pat_preamble env p
|
adamc@182
|
123 | PRecord xps =>
|
adamc@182
|
124 foldl (fn ((_, p, _), (pp, env)) =>
|
adamc@182
|
125 let
|
adamc@182
|
126 val (pp', env) = p_pat_preamble env p
|
adamc@182
|
127 in
|
adamc@182
|
128 (box [pp', pp], env)
|
adamc@182
|
129 end) (box [], env) xps
|
adamc@182
|
130
|
adamc@182
|
131 fun p_patCon env pc =
|
adamc@182
|
132 case pc of
|
adamc@182
|
133 PConVar n => p_con_named env n
|
adamc@186
|
134 | PConFfi {mod = m, con, ...} => string ("lw_" ^ m ^ "_" ^ con)
|
adamc@182
|
135
|
adamc@182
|
136 fun p_pat (env, exit, depth) (p, _) =
|
adamc@182
|
137 case p of
|
adamc@182
|
138 PWild =>
|
adamc@182
|
139 (box [], env)
|
adamc@182
|
140 | PVar (x, t) =>
|
adamc@182
|
141 (box [string "__lwr_",
|
adamc@182
|
142 string x,
|
adamc@182
|
143 string "_",
|
adamc@182
|
144 string (Int.toString (E.countERels env)),
|
adamc@182
|
145 space,
|
adamc@182
|
146 string "=",
|
adamc@182
|
147 space,
|
adamc@182
|
148 string "disc",
|
adamc@182
|
149 string (Int.toString depth),
|
adamc@182
|
150 string ";"],
|
adamc@182
|
151 E.pushERel env x t)
|
adamc@182
|
152 | PPrim (Prim.Int n) =>
|
adamc@182
|
153 (box [string "if",
|
adamc@182
|
154 space,
|
adamc@182
|
155 string "(disc",
|
adamc@182
|
156 string (Int.toString depth),
|
adamc@182
|
157 space,
|
adamc@182
|
158 string "!=",
|
adamc@182
|
159 space,
|
adamc@182
|
160 Prim.p_t (Prim.Int n),
|
adamc@182
|
161 string ")",
|
adamc@182
|
162 space,
|
adamc@182
|
163 exit],
|
adamc@182
|
164 env)
|
adamc@182
|
165 | PPrim (Prim.String s) =>
|
adamc@182
|
166 (box [string "if",
|
adamc@182
|
167 space,
|
adamc@182
|
168 string "(strcmp(disc",
|
adamc@182
|
169 string (Int.toString depth),
|
adamc@182
|
170 string ",",
|
adamc@182
|
171 space,
|
adamc@182
|
172 Prim.p_t (Prim.String s),
|
adamc@182
|
173 string "))",
|
adamc@182
|
174 space,
|
adamc@182
|
175 exit],
|
adamc@182
|
176 env)
|
adamc@182
|
177 | PPrim _ => raise Fail "CjrPrint: Disallowed PPrim primitive"
|
adamc@182
|
178
|
adamc@188
|
179 | PCon (dk, pc, po) =>
|
adamc@182
|
180 let
|
adamc@182
|
181 val (p, env) =
|
adamc@182
|
182 case po of
|
adamc@182
|
183 NONE => (box [], env)
|
adamc@182
|
184 | SOME p =>
|
adamc@182
|
185 let
|
adamc@182
|
186 val (p, env) = p_pat (env, exit, depth + 1) p
|
adamc@182
|
187
|
adamc@182
|
188 val (x, to) = case pc of
|
adamc@182
|
189 PConVar n =>
|
adamc@182
|
190 let
|
adamc@182
|
191 val (x, to, _) = E.lookupConstructor env n
|
adamc@182
|
192 in
|
adamc@196
|
193 ("lw_" ^ x, to)
|
adamc@182
|
194 end
|
adamc@188
|
195 | PConFfi {mod = m, con, arg, ...} =>
|
adamc@188
|
196 ("lw_" ^ m ^ "_" ^ con, arg)
|
adamc@182
|
197
|
adamc@182
|
198 val t = case to of
|
adamc@182
|
199 NONE => raise Fail "CjrPrint: Constructor mismatch"
|
adamc@182
|
200 | SOME t => t
|
adamc@182
|
201 in
|
adamc@182
|
202 (box [string "{",
|
adamc@182
|
203 newline,
|
adamc@182
|
204 p_typ env t,
|
adamc@182
|
205 space,
|
adamc@182
|
206 string "disc",
|
adamc@182
|
207 string (Int.toString (depth + 1)),
|
adamc@182
|
208 space,
|
adamc@182
|
209 string "=",
|
adamc@182
|
210 space,
|
adamc@198
|
211 case dk of
|
adamc@198
|
212 Enum => raise Fail "CjrPrint: Looking at argument of no-argument constructor"
|
adamc@198
|
213 | Default => box [string "disc",
|
adamc@198
|
214 string (Int.toString depth),
|
adamc@198
|
215 string "->data.",
|
adamc@198
|
216 string x]
|
adamc@198
|
217 | Option =>
|
adamc@198
|
218 case #1 t of
|
adamc@198
|
219 TDatatype _ => box [string "disc",
|
adamc@198
|
220 string (Int.toString depth)]
|
adamc@199
|
221 | TFfi ("Basis", "string") => box [string "disc",
|
adamc@199
|
222 string (Int.toString depth)]
|
adamc@198
|
223 | _ => box [string "*disc",
|
adamc@198
|
224 string (Int.toString depth)],
|
adamc@182
|
225 string ";",
|
adamc@182
|
226 newline,
|
adamc@182
|
227 p,
|
adamc@182
|
228 newline,
|
adamc@182
|
229 string "}"],
|
adamc@182
|
230 env)
|
adamc@182
|
231 end
|
adamc@182
|
232 in
|
adamc@182
|
233 (box [string "if",
|
adamc@182
|
234 space,
|
adamc@182
|
235 string "(disc",
|
adamc@182
|
236 string (Int.toString depth),
|
adamc@198
|
237 case (dk, po) of
|
adamc@198
|
238 (Enum, _) => box [space,
|
adamc@198
|
239 string "!=",
|
adamc@198
|
240 space,
|
adamc@198
|
241 p_patCon env pc]
|
adamc@198
|
242 | (Default, _) => box [string "->tag",
|
adamc@198
|
243 space,
|
adamc@198
|
244 string "!=",
|
adamc@198
|
245 space,
|
adamc@198
|
246 p_patCon env pc]
|
adamc@198
|
247 | (Option, NONE) => box [space,
|
adamc@198
|
248 string "!=",
|
adamc@198
|
249 space,
|
adamc@198
|
250 string "NULL"]
|
adamc@198
|
251 | (Option, SOME _) => box [space,
|
adamc@198
|
252 string "==",
|
adamc@198
|
253 space,
|
adamc@198
|
254 string "NULL"],
|
adamc@182
|
255 string ")",
|
adamc@182
|
256 space,
|
adamc@182
|
257 exit,
|
adamc@182
|
258 newline,
|
adamc@182
|
259 p],
|
adamc@182
|
260 env)
|
adamc@182
|
261 end
|
adamc@182
|
262
|
adamc@182
|
263 | PRecord xps =>
|
adamc@182
|
264 let
|
adamc@182
|
265 val (xps, env) =
|
adamc@182
|
266 ListUtil.foldlMap (fn ((x, p, t), env) =>
|
adamc@182
|
267 let
|
adamc@182
|
268 val (p, env) = p_pat (env, exit, depth + 1) p
|
adamc@182
|
269
|
adamc@182
|
270 val p = box [string "{",
|
adamc@182
|
271 newline,
|
adamc@182
|
272 p_typ env t,
|
adamc@182
|
273 space,
|
adamc@182
|
274 string "disc",
|
adamc@182
|
275 string (Int.toString (depth + 1)),
|
adamc@182
|
276 space,
|
adamc@182
|
277 string "=",
|
adamc@182
|
278 space,
|
adamc@182
|
279 string "disc",
|
adamc@182
|
280 string (Int.toString depth),
|
adamc@196
|
281 string ".__lwf_",
|
adamc@182
|
282 string x,
|
adamc@182
|
283 string ";",
|
adamc@182
|
284 newline,
|
adamc@182
|
285 p,
|
adamc@182
|
286 newline,
|
adamc@182
|
287 string "}"]
|
adamc@182
|
288 in
|
adamc@182
|
289 (p, env)
|
adamc@182
|
290 end) env xps
|
adamc@182
|
291 in
|
adamc@182
|
292 (p_list_sep newline (fn x => x) xps,
|
adamc@182
|
293 env)
|
adamc@182
|
294 end
|
adamc@182
|
295
|
adamc@182
|
296 local
|
adamc@182
|
297 val count = ref 0
|
adamc@182
|
298 in
|
adamc@182
|
299 fun newGoto () =
|
adamc@182
|
300 let
|
adamc@182
|
301 val r = !count
|
adamc@182
|
302 in
|
adamc@182
|
303 count := r + 1;
|
adamc@182
|
304 string ("L" ^ Int.toString r)
|
adamc@182
|
305 end
|
adamc@182
|
306 end
|
adamc@182
|
307
|
adamc@185
|
308 fun patConInfo env pc =
|
adamc@185
|
309 case pc of
|
adamc@185
|
310 PConVar n =>
|
adamc@185
|
311 let
|
adamc@185
|
312 val (x, _, dn) = E.lookupConstructor env n
|
adamc@185
|
313 val (dx, _) = E.lookupDatatype env dn
|
adamc@185
|
314 in
|
adamc@185
|
315 ("__lwd_" ^ dx ^ "_" ^ Int.toString dn,
|
adamc@196
|
316 "__lwc_" ^ x ^ "_" ^ Int.toString n,
|
adamc@196
|
317 "lw_" ^ x)
|
adamc@185
|
318 end
|
adamc@186
|
319 | PConFfi {mod = m, datatyp, con, ...} =>
|
adamc@185
|
320 ("lw_" ^ m ^ "_" ^ datatyp,
|
adamc@196
|
321 "lw_" ^ m ^ "_" ^ con,
|
adamc@196
|
322 "lw_" ^ con)
|
adamc@185
|
323
|
adamc@182
|
324 fun p_exp' par env (e, loc) =
|
adamc@29
|
325 case e of
|
adamc@29
|
326 EPrim p => Prim.p_t p
|
adamc@29
|
327 | ERel n => p_rel env n
|
adamc@109
|
328 | ENamed n => p_enamed env n
|
adamc@188
|
329 | ECon (Enum, pc, _) => p_patCon env pc
|
adamc@198
|
330 | ECon (Option, pc, NONE) => string "NULL"
|
adamc@198
|
331 | ECon (Option, pc, SOME e) =>
|
adamc@198
|
332 let
|
adamc@198
|
333 val to = case pc of
|
adamc@198
|
334 PConVar n => #2 (E.lookupConstructor env n)
|
adamc@198
|
335 | PConFfi {arg, ...} => arg
|
adamc@198
|
336
|
adamc@198
|
337 val t = case to of
|
adamc@198
|
338 NONE => raise Fail "CjrPrint: ECon argument status mismatch"
|
adamc@198
|
339 | SOME t => t
|
adamc@198
|
340 in
|
adamc@198
|
341 case #1 t of
|
adamc@198
|
342 TDatatype _ => p_exp' par env e
|
adamc@199
|
343 | TFfi ("Basis", "string") => p_exp' par env e
|
adamc@198
|
344 | _ => box [string "({",
|
adamc@198
|
345 newline,
|
adamc@198
|
346 p_typ env t,
|
adamc@198
|
347 space,
|
adamc@198
|
348 string "*tmp",
|
adamc@198
|
349 space,
|
adamc@198
|
350 string "=",
|
adamc@198
|
351 space,
|
adamc@198
|
352 string "lw_malloc(ctx, sizeof(",
|
adamc@198
|
353 p_typ env t,
|
adamc@198
|
354 string "));",
|
adamc@198
|
355 newline,
|
adamc@198
|
356 string "*tmp",
|
adamc@198
|
357 space,
|
adamc@198
|
358 string "=",
|
adamc@198
|
359 p_exp' par env e,
|
adamc@198
|
360 string ";",
|
adamc@198
|
361 newline,
|
adamc@198
|
362 string "tmp;",
|
adamc@198
|
363 newline,
|
adamc@198
|
364 string "})"]
|
adamc@198
|
365 end
|
adamc@188
|
366 | ECon (Default, pc, eo) =>
|
adamc@181
|
367 let
|
adamc@196
|
368 val (xd, xc, xn) = patConInfo env pc
|
adamc@181
|
369 in
|
adamc@182
|
370 box [string "({",
|
adamc@181
|
371 newline,
|
adamc@181
|
372 string "struct",
|
adamc@181
|
373 space,
|
adamc@185
|
374 string xd,
|
adamc@181
|
375 space,
|
adamc@181
|
376 string "*tmp",
|
adamc@181
|
377 space,
|
adamc@181
|
378 string "=",
|
adamc@181
|
379 space,
|
adamc@185
|
380 string "lw_malloc(ctx, sizeof(struct ",
|
adamc@185
|
381 string xd,
|
adamc@181
|
382 string "));",
|
adamc@181
|
383 newline,
|
adamc@181
|
384 string "tmp->tag",
|
adamc@181
|
385 space,
|
adamc@181
|
386 string "=",
|
adamc@181
|
387 space,
|
adamc@185
|
388 string xc,
|
adamc@181
|
389 string ";",
|
adamc@181
|
390 newline,
|
adamc@181
|
391 case eo of
|
adamc@181
|
392 NONE => box []
|
adamc@185
|
393 | SOME e => box [string "tmp->data.",
|
adamc@196
|
394 string xn,
|
adamc@181
|
395 space,
|
adamc@181
|
396 string "=",
|
adamc@181
|
397 space,
|
adamc@181
|
398 p_exp env e,
|
adamc@181
|
399 string ";",
|
adamc@181
|
400 newline],
|
adamc@181
|
401 string "tmp;",
|
adamc@181
|
402 newline,
|
adamc@181
|
403 string "})"]
|
adamc@181
|
404 end
|
adamc@109
|
405
|
adamc@53
|
406 | EFfi (m, x) => box [string "lw_", string m, string "_", string x]
|
adamc@53
|
407 | EFfiApp (m, x, es) => box [string "lw_",
|
adamc@53
|
408 string m,
|
adamc@53
|
409 string "_",
|
adamc@53
|
410 string x,
|
adamc@117
|
411 string "(ctx, ",
|
adamc@53
|
412 p_list (p_exp env) es,
|
adamc@53
|
413 string ")"]
|
adamc@129
|
414 | EApp (e1, e2) =>
|
adamc@129
|
415 let
|
adamc@129
|
416 fun unravel (f, acc) =
|
adamc@129
|
417 case #1 f of
|
adamc@129
|
418 EApp (f', arg) => unravel (f', arg :: acc)
|
adamc@129
|
419 | _ => (f, acc)
|
adamc@129
|
420
|
adamc@129
|
421 val (f, args) = unravel (e1, [e2])
|
adamc@129
|
422 in
|
adamc@129
|
423 parenIf par (box [p_exp' true env e1,
|
adamc@129
|
424 string "(ctx,",
|
adamc@129
|
425 space,
|
adamc@129
|
426 p_list_sep (box [string ",", space]) (p_exp env) args,
|
adamc@129
|
427 string ")"])
|
adamc@129
|
428 end
|
adamc@29
|
429
|
adamc@29
|
430 | ERecord (i, xes) => box [string "({",
|
adamc@29
|
431 space,
|
adamc@29
|
432 string "struct",
|
adamc@29
|
433 space,
|
adamc@29
|
434 string ("__lws_" ^ Int.toString i),
|
adamc@29
|
435 space,
|
adamc@181
|
436 string "tmp",
|
adamc@29
|
437 space,
|
adamc@29
|
438 string "=",
|
adamc@29
|
439 space,
|
adamc@29
|
440 string "{",
|
adamc@29
|
441 p_list (fn (_, e) =>
|
adamc@29
|
442 p_exp env e) xes,
|
adamc@29
|
443 string "};",
|
adamc@29
|
444 space,
|
adamc@181
|
445 string "tmp;",
|
adamc@29
|
446 space,
|
adamc@29
|
447 string "})" ]
|
adamc@29
|
448 | EField (e, x) =>
|
adamc@29
|
449 box [p_exp' true env e,
|
adamc@182
|
450 string ".__lwf_",
|
adamc@29
|
451 string x]
|
adamc@29
|
452
|
adamc@182
|
453 | ECase (e, pes, {disc, result}) =>
|
adamc@182
|
454 let
|
adamc@182
|
455 val final = newGoto ()
|
adamc@182
|
456
|
adamc@182
|
457 val body = foldl (fn ((p, e), body) =>
|
adamc@182
|
458 let
|
adamc@182
|
459 val exit = newGoto ()
|
adamc@182
|
460 val (pr, _) = p_pat_preamble env p
|
adamc@182
|
461 val (p, env) = p_pat (env,
|
adamc@182
|
462 box [string "goto",
|
adamc@182
|
463 space,
|
adamc@182
|
464 exit,
|
adamc@182
|
465 string ";"],
|
adamc@182
|
466 0) p
|
adamc@182
|
467 in
|
adamc@182
|
468 box [body,
|
adamc@182
|
469 box [string "{",
|
adamc@182
|
470 newline,
|
adamc@182
|
471 pr,
|
adamc@182
|
472 newline,
|
adamc@182
|
473 p,
|
adamc@182
|
474 newline,
|
adamc@182
|
475 string "result",
|
adamc@182
|
476 space,
|
adamc@182
|
477 string "=",
|
adamc@182
|
478 space,
|
adamc@182
|
479 p_exp env e,
|
adamc@182
|
480 string ";",
|
adamc@182
|
481 newline,
|
adamc@182
|
482 string "goto",
|
adamc@182
|
483 space,
|
adamc@182
|
484 final,
|
adamc@182
|
485 string ";",
|
adamc@182
|
486 newline,
|
adamc@182
|
487 string "}"],
|
adamc@182
|
488 newline,
|
adamc@182
|
489 exit,
|
adamc@182
|
490 string ":",
|
adamc@182
|
491 newline]
|
adamc@182
|
492 end) (box []) pes
|
adamc@182
|
493 in
|
adamc@182
|
494 box [string "({",
|
adamc@182
|
495 newline,
|
adamc@182
|
496 p_typ env disc,
|
adamc@182
|
497 space,
|
adamc@182
|
498 string "disc0",
|
adamc@182
|
499 space,
|
adamc@182
|
500 string "=",
|
adamc@182
|
501 space,
|
adamc@182
|
502 p_exp env e,
|
adamc@182
|
503 string ";",
|
adamc@182
|
504 newline,
|
adamc@182
|
505 p_typ env result,
|
adamc@182
|
506 space,
|
adamc@182
|
507 string "result;",
|
adamc@182
|
508 newline,
|
adamc@182
|
509 body,
|
adamc@182
|
510 string "lw_error(ctx, FATAL, \"",
|
adamc@182
|
511 string (ErrorMsg.spanToString loc),
|
adamc@182
|
512 string ": pattern match failure\");",
|
adamc@182
|
513 newline,
|
adamc@182
|
514 final,
|
adamc@182
|
515 string ":",
|
adamc@182
|
516 space,
|
adamc@182
|
517 string "result;",
|
adamc@182
|
518 newline,
|
adamc@182
|
519 string "})"]
|
adamc@182
|
520 end
|
adamc@181
|
521
|
adamc@117
|
522 | EWrite e => box [string "(lw_write(ctx, ",
|
adamc@102
|
523 p_exp env e,
|
adamc@102
|
524 string "), lw_unit_v)"]
|
adamc@102
|
525
|
adamc@106
|
526 | ESeq (e1, e2) => box [string "(",
|
adamc@106
|
527 p_exp env e1,
|
adamc@106
|
528 string ",",
|
adamc@106
|
529 space,
|
adamc@106
|
530 p_exp env e2,
|
adamc@106
|
531 string ")"]
|
adamc@269
|
532 | ELet (x, t, e1, e2) => box [string "({",
|
adamc@269
|
533 newline,
|
adamc@269
|
534 p_typ env t,
|
adamc@269
|
535 space,
|
adamc@269
|
536 p_rel env 0,
|
adamc@269
|
537 space,
|
adamc@269
|
538 string "=",
|
adamc@269
|
539 space,
|
adamc@269
|
540 p_exp env e1,
|
adamc@269
|
541 string ";",
|
adamc@269
|
542 newline,
|
adamc@269
|
543 p_exp (E.pushERel env x t) e2,
|
adamc@269
|
544 string ";",
|
adamc@269
|
545 newline,
|
adamc@269
|
546 string "})"]
|
adamc@269
|
547
|
adamc@269
|
548 | EQuery {exps, tables, rnum, state, query, body, initial} =>
|
adamc@269
|
549 box [string "query[",
|
adamc@269
|
550 p_list (fn (x, t) => box [string x, space, string ":", space, p_typ env t]) exps,
|
adamc@269
|
551 string "] [",
|
adamc@269
|
552 p_list (fn (x, xts) => box [string x,
|
adamc@269
|
553 space,
|
adamc@269
|
554 string ":",
|
adamc@269
|
555 space,
|
adamc@269
|
556 string "{",
|
adamc@269
|
557 p_list (fn (x, t) => box [string x, space, string ":", space, p_typ env t]) xts,
|
adamc@269
|
558 string "}"]) tables,
|
adamc@269
|
559 string "] [",
|
adamc@269
|
560 p_typ env state,
|
adamc@269
|
561 string "] [",
|
adamc@269
|
562 string (Int.toString rnum),
|
adamc@269
|
563 string "]",
|
adamc@269
|
564 space,
|
adamc@269
|
565 p_exp env query,
|
adamc@269
|
566 space,
|
adamc@269
|
567 string "initial",
|
adamc@269
|
568 space,
|
adamc@269
|
569 p_exp env initial,
|
adamc@269
|
570 space,
|
adamc@269
|
571 string "in",
|
adamc@269
|
572 space,
|
adamc@269
|
573 p_exp (E.pushERel (E.pushERel env "r" dummyt) "acc" dummyt) body]
|
adamc@106
|
574
|
adamc@29
|
575 and p_exp env = p_exp' false env
|
adamc@29
|
576
|
adamc@129
|
577 fun p_fun env (fx, n, args, ran, e) =
|
adamc@129
|
578 let
|
adamc@129
|
579 val nargs = length args
|
adamc@129
|
580 val env' = foldl (fn ((x, dom), env) => E.pushERel env x dom) env args
|
adamc@129
|
581 in
|
adamc@129
|
582 box [string "static",
|
adamc@129
|
583 space,
|
adamc@129
|
584 p_typ env ran,
|
adamc@129
|
585 space,
|
adamc@129
|
586 string ("__lwn_" ^ fx ^ "_" ^ Int.toString n),
|
adamc@129
|
587 string "(",
|
adamc@129
|
588 p_list_sep (box [string ",", space]) (fn x => x)
|
adamc@129
|
589 (string "lw_context ctx" :: ListUtil.mapi (fn (i, (_, dom)) =>
|
adamc@129
|
590 box [p_typ env dom,
|
adamc@129
|
591 space,
|
adamc@129
|
592 p_rel env' (nargs - i - 1)]) args),
|
adamc@129
|
593 string ")",
|
adamc@129
|
594 space,
|
adamc@129
|
595 string "{",
|
adamc@129
|
596 newline,
|
adamc@129
|
597 box[string "return(",
|
adamc@129
|
598 p_exp env' e,
|
adamc@129
|
599 string ");"],
|
adamc@129
|
600 newline,
|
adamc@129
|
601 string "}"]
|
adamc@129
|
602 end
|
adamc@129
|
603
|
adamc@129
|
604 fun p_decl env (dAll as (d, _) : decl) =
|
adamc@29
|
605 case d of
|
adamc@29
|
606 DStruct (n, xts) =>
|
adamc@196
|
607 let
|
adamc@196
|
608 val env = E.declBinds env dAll
|
adamc@196
|
609 in
|
adamc@196
|
610 box [string "struct",
|
adamc@196
|
611 space,
|
adamc@196
|
612 string ("__lws_" ^ Int.toString n),
|
adamc@196
|
613 space,
|
adamc@196
|
614 string "{",
|
adamc@196
|
615 newline,
|
adamc@196
|
616 p_list_sep (box []) (fn (x, t) => box [p_typ env t,
|
adamc@196
|
617 space,
|
adamc@196
|
618 string "__lwf_",
|
adamc@196
|
619 string x,
|
adamc@196
|
620 string ";",
|
adamc@196
|
621 newline]) xts,
|
adamc@196
|
622 string "};"]
|
adamc@196
|
623 end
|
adamc@188
|
624 | DDatatype (Enum, x, n, xncs) =>
|
adamc@188
|
625 box [string "enum",
|
adamc@188
|
626 space,
|
adamc@188
|
627 string ("__lwe_" ^ x ^ "_" ^ Int.toString n),
|
adamc@188
|
628 space,
|
adamc@188
|
629 string "{",
|
adamc@188
|
630 space,
|
adamc@188
|
631 p_list_sep (box [string ",", space]) (fn (x, n, _) => string ("__lwc_" ^ x ^ "_" ^ Int.toString n)) xncs,
|
adamc@188
|
632 space,
|
adamc@188
|
633 string "};"]
|
adamc@198
|
634 | DDatatype (Option, _, _, _) => box []
|
adamc@188
|
635 | DDatatype (Default, x, n, xncs) =>
|
adamc@165
|
636 let
|
adamc@165
|
637 val xncsArgs = List.mapPartial (fn (x, n, NONE) => NONE
|
adamc@165
|
638 | (x, n, SOME t) => SOME (x, n, t)) xncs
|
adamc@165
|
639 in
|
adamc@165
|
640 box [string "enum",
|
adamc@165
|
641 space,
|
adamc@165
|
642 string ("__lwe_" ^ x ^ "_" ^ Int.toString n),
|
adamc@165
|
643 space,
|
adamc@165
|
644 string "{",
|
adamc@165
|
645 space,
|
adamc@165
|
646 p_list_sep (box [string ",", space]) (fn (x, n, _) => string ("__lwc_" ^ x ^ "_" ^ Int.toString n)) xncs,
|
adamc@165
|
647 space,
|
adamc@165
|
648 string "};",
|
adamc@165
|
649 newline,
|
adamc@165
|
650 newline,
|
adamc@165
|
651 string "struct",
|
adamc@165
|
652 space,
|
adamc@167
|
653 string ("__lwd_" ^ x ^ "_" ^ Int.toString n),
|
adamc@165
|
654 space,
|
adamc@165
|
655 string "{",
|
adamc@165
|
656 newline,
|
adamc@165
|
657 string "enum",
|
adamc@165
|
658 space,
|
adamc@165
|
659 string ("__lwe_" ^ x ^ "_" ^ Int.toString n),
|
adamc@165
|
660 space,
|
adamc@165
|
661 string "tag;",
|
adamc@165
|
662 newline,
|
adamc@165
|
663 box (case xncsArgs of
|
adamc@165
|
664 [] => []
|
adamc@165
|
665 | _ => [string "union",
|
adamc@165
|
666 space,
|
adamc@165
|
667 string "{",
|
adamc@165
|
668 newline,
|
adamc@165
|
669 p_list_sep newline (fn (x, n, t) => box [p_typ env t,
|
adamc@165
|
670 space,
|
adamc@196
|
671 string ("lw_" ^ x),
|
adamc@165
|
672 string ";"]) xncsArgs,
|
adamc@165
|
673 newline,
|
adamc@165
|
674 string "}",
|
adamc@165
|
675 space,
|
adamc@165
|
676 string "data;",
|
adamc@165
|
677 newline]),
|
adamc@165
|
678 string "};"]
|
adamc@188
|
679 end
|
adamc@29
|
680
|
adamc@196
|
681 | DDatatypeForward _ => box []
|
adamc@196
|
682
|
adamc@29
|
683 | DVal (x, n, t, e) =>
|
adamc@29
|
684 box [p_typ env t,
|
adamc@29
|
685 space,
|
adamc@29
|
686 string ("__lwn_" ^ x ^ "_" ^ Int.toString n),
|
adamc@29
|
687 space,
|
adamc@29
|
688 string "=",
|
adamc@29
|
689 space,
|
adamc@29
|
690 p_exp env e,
|
adamc@29
|
691 string ";"]
|
adamc@129
|
692 | DFun vi => p_fun env vi
|
adamc@129
|
693 | DFunRec vis =>
|
adamc@29
|
694 let
|
adamc@129
|
695 val env = E.declBinds env dAll
|
adamc@29
|
696 in
|
adamc@129
|
697 box [p_list_sep newline (fn (fx, n, args, ran, _) =>
|
adamc@129
|
698 box [string "static",
|
adamc@129
|
699 space,
|
adamc@129
|
700 p_typ env ran,
|
adamc@129
|
701 space,
|
adamc@129
|
702 string ("__lwn_" ^ fx ^ "_" ^ Int.toString n),
|
adamc@129
|
703 string "(lw_context,",
|
adamc@129
|
704 space,
|
adamc@129
|
705 p_list_sep (box [string ",", space])
|
adamc@129
|
706 (fn (_, dom) => p_typ env dom) args,
|
adamc@129
|
707 string ");"]) vis,
|
adamc@29
|
708 newline,
|
adamc@129
|
709 p_list_sep newline (p_fun env) vis,
|
adamc@129
|
710 newline]
|
adamc@29
|
711 end
|
adamc@29
|
712
|
adamc@144
|
713 datatype 'a search =
|
adamc@144
|
714 Found of 'a
|
adamc@144
|
715 | NotFound
|
adamc@144
|
716 | Error
|
adamc@120
|
717
|
adamc@101
|
718
|
adamc@101
|
719 fun p_file env (ds, ps) =
|
adamc@29
|
720 let
|
adamc@101
|
721 val (pds, env) = ListUtil.foldlMap (fn (d, env) =>
|
adamc@31
|
722 (p_decl env d,
|
adamc@31
|
723 E.declBinds env d))
|
adamc@101
|
724 env ds
|
adamc@144
|
725
|
adamc@144
|
726 val fields = foldl (fn ((ek, _, _, ts), fields) =>
|
adamc@144
|
727 case ek of
|
adamc@144
|
728 Core.Link => fields
|
adamc@144
|
729 | Core.Action =>
|
adamc@144
|
730 case List.last ts of
|
adamc@144
|
731 (TRecord i, _) =>
|
adamc@144
|
732 let
|
adamc@144
|
733 val xts = E.lookupStruct env i
|
adamc@144
|
734 val xtsSet = SS.addList (SS.empty, map #1 xts)
|
adamc@144
|
735 in
|
adamc@144
|
736 foldl (fn ((x, _), fields) =>
|
adamc@144
|
737 let
|
adamc@144
|
738 val xtsSet' = Option.getOpt (SM.find (fields, x), SS.empty)
|
adamc@144
|
739 in
|
adamc@144
|
740 SM.insert (fields, x, SS.union (SS.delete (xtsSet, x),
|
adamc@144
|
741 xtsSet'))
|
adamc@144
|
742 end) fields xts
|
adamc@144
|
743 end
|
adamc@144
|
744 | _ => raise Fail "CjrPrint: Last argument of action isn't record")
|
adamc@144
|
745 SM.empty ps
|
adamc@144
|
746
|
adamc@144
|
747 val fnums = SM.foldli (fn (x, xs, fnums) =>
|
adamc@144
|
748 let
|
adamc@144
|
749 val unusable = SS.foldl (fn (x', unusable) =>
|
adamc@144
|
750 case SM.find (fnums, x') of
|
adamc@144
|
751 NONE => unusable
|
adamc@144
|
752 | SOME n => IS.add (unusable, n))
|
adamc@144
|
753 IS.empty xs
|
adamc@144
|
754
|
adamc@144
|
755 fun findAvailable n =
|
adamc@144
|
756 if IS.member (unusable, n) then
|
adamc@144
|
757 findAvailable (n + 1)
|
adamc@144
|
758 else
|
adamc@144
|
759 n
|
adamc@144
|
760 in
|
adamc@144
|
761 SM.insert (fnums, x, findAvailable 0)
|
adamc@144
|
762 end)
|
adamc@144
|
763 SM.empty fields
|
adamc@144
|
764
|
adamc@144
|
765 fun makeSwitch (fnums, i) =
|
adamc@144
|
766 case SM.foldl (fn (n, NotFound) => Found n
|
adamc@144
|
767 | (n, Error) => Error
|
adamc@144
|
768 | (n, Found n') => if n = n' then
|
adamc@144
|
769 Found n'
|
adamc@144
|
770 else
|
adamc@144
|
771 Error) NotFound fnums of
|
adamc@144
|
772 NotFound => box [string "return",
|
adamc@144
|
773 space,
|
adamc@144
|
774 string "-1;"]
|
adamc@144
|
775 | Found n => box [string "return",
|
adamc@144
|
776 space,
|
adamc@144
|
777 string (Int.toString n),
|
adamc@144
|
778 string ";"]
|
adamc@144
|
779 | Error =>
|
adamc@144
|
780 let
|
adamc@144
|
781 val cmap = SM.foldli (fn (x, n, cmap) =>
|
adamc@144
|
782 let
|
adamc@144
|
783 val ch = if i < size x then
|
adamc@144
|
784 String.sub (x, i)
|
adamc@144
|
785 else
|
adamc@144
|
786 chr 0
|
adamc@144
|
787
|
adamc@144
|
788 val fnums = case CM.find (cmap, ch) of
|
adamc@144
|
789 NONE => SM.empty
|
adamc@144
|
790 | SOME fnums => fnums
|
adamc@144
|
791 val fnums = SM.insert (fnums, x, n)
|
adamc@144
|
792 in
|
adamc@144
|
793 CM.insert (cmap, ch, fnums)
|
adamc@144
|
794 end)
|
adamc@144
|
795 CM.empty fnums
|
adamc@144
|
796
|
adamc@144
|
797 val cmap = CM.listItemsi cmap
|
adamc@144
|
798 in
|
adamc@144
|
799 case cmap of
|
adamc@144
|
800 [(_, fnums)] =>
|
adamc@144
|
801 box [string "if",
|
adamc@144
|
802 space,
|
adamc@144
|
803 string "(name[",
|
adamc@144
|
804 string (Int.toString i),
|
adamc@144
|
805 string "]",
|
adamc@144
|
806 space,
|
adamc@144
|
807 string "==",
|
adamc@144
|
808 space,
|
adamc@144
|
809 string "0)",
|
adamc@144
|
810 space,
|
adamc@144
|
811 string "return",
|
adamc@144
|
812 space,
|
adamc@144
|
813 string "-1;",
|
adamc@144
|
814 newline,
|
adamc@144
|
815 makeSwitch (fnums, i+1)]
|
adamc@144
|
816 | _ =>
|
adamc@144
|
817 box [string "switch",
|
adamc@144
|
818 space,
|
adamc@144
|
819 string "(name[",
|
adamc@144
|
820 string (Int.toString i),
|
adamc@144
|
821 string "])",
|
adamc@144
|
822 space,
|
adamc@144
|
823 string "{",
|
adamc@144
|
824 newline,
|
adamc@144
|
825 box (map (fn (ch, fnums) =>
|
adamc@144
|
826 box [string "case",
|
adamc@144
|
827 space,
|
adamc@144
|
828 if ch = chr 0 then
|
adamc@144
|
829 string "0:"
|
adamc@144
|
830 else
|
adamc@144
|
831 box [string "'",
|
adamc@144
|
832 string (Char.toString ch),
|
adamc@144
|
833 string "':"],
|
adamc@144
|
834 newline,
|
adamc@144
|
835 makeSwitch (fnums, i+1),
|
adamc@144
|
836 newline]) cmap),
|
adamc@144
|
837 string "default:",
|
adamc@144
|
838 newline,
|
adamc@144
|
839 string "return",
|
adamc@144
|
840 space,
|
adamc@144
|
841 string "-1;",
|
adamc@144
|
842 newline,
|
adamc@144
|
843 string "}"]
|
adamc@144
|
844 end
|
adamc@144
|
845
|
adamc@186
|
846 fun capitalize s =
|
adamc@186
|
847 if s = "" then
|
adamc@186
|
848 ""
|
adamc@186
|
849 else
|
adamc@186
|
850 str (Char.toUpper (String.sub (s, 0))) ^ String.extract (s, 1, NONE)
|
adamc@186
|
851
|
adamc@144
|
852 fun unurlify (t, loc) =
|
adamc@144
|
853 case t of
|
adamc@186
|
854 TFfi (m, t) => string ("lw_" ^ m ^ "_unurlify" ^ capitalize t ^ "(ctx, &request)")
|
adamc@144
|
855
|
adamc@144
|
856 | TRecord 0 => string "lw_unit_v"
|
adamc@144
|
857 | TRecord i =>
|
adamc@144
|
858 let
|
adamc@144
|
859 val xts = E.lookupStruct env i
|
adamc@144
|
860 in
|
adamc@144
|
861 box [string "({",
|
adamc@144
|
862 newline,
|
adamc@144
|
863 box (map (fn (x, t) =>
|
adamc@144
|
864 box [p_typ env t,
|
adamc@144
|
865 space,
|
adamc@144
|
866 string x,
|
adamc@144
|
867 space,
|
adamc@144
|
868 string "=",
|
adamc@144
|
869 space,
|
adamc@144
|
870 unurlify t,
|
adamc@144
|
871 string ";",
|
adamc@144
|
872 newline]) xts),
|
adamc@144
|
873 string "struct",
|
adamc@144
|
874 space,
|
adamc@144
|
875 string "__lws_",
|
adamc@144
|
876 string (Int.toString i),
|
adamc@144
|
877 space,
|
adamc@181
|
878 string "tmp",
|
adamc@144
|
879 space,
|
adamc@144
|
880 string "=",
|
adamc@144
|
881 space,
|
adamc@144
|
882 string "{",
|
adamc@144
|
883 space,
|
adamc@144
|
884 p_list_sep (box [string ",", space]) (fn (x, _) => string x) xts,
|
adamc@144
|
885 space,
|
adamc@144
|
886 string "};",
|
adamc@144
|
887 newline,
|
adamc@181
|
888 string "tmp;",
|
adamc@144
|
889 newline,
|
adamc@144
|
890 string "})"]
|
adamc@144
|
891 end
|
adamc@144
|
892
|
adamc@188
|
893 | TDatatype (Enum, i, _) =>
|
adamc@188
|
894 let
|
adamc@188
|
895 val (x, xncs) = E.lookupDatatype env i
|
adamc@188
|
896
|
adamc@188
|
897 fun doEm xncs =
|
adamc@188
|
898 case xncs of
|
adamc@188
|
899 [] => string ("(lw_error(ctx, FATAL, \"Error unurlifying datatype " ^ x ^ "\"), (enum __lwe_"
|
adamc@188
|
900 ^ x ^ "_" ^ Int.toString i ^ ")0)")
|
adamc@188
|
901 | (x', n, to) :: rest =>
|
adamc@188
|
902 box [string "((!strncmp(request, \"",
|
adamc@188
|
903 string x',
|
adamc@188
|
904 string "\", ",
|
adamc@188
|
905 string (Int.toString (size x')),
|
adamc@188
|
906 string ") && (request[",
|
adamc@188
|
907 string (Int.toString (size x')),
|
adamc@188
|
908 string "] == 0 || request[",
|
adamc@188
|
909 string (Int.toString (size x')),
|
adamc@188
|
910 string ("] == '/')) ? __lwc_" ^ x' ^ "_" ^ Int.toString n),
|
adamc@188
|
911 space,
|
adamc@188
|
912 string ":",
|
adamc@188
|
913 space,
|
adamc@188
|
914 doEm rest,
|
adamc@188
|
915 string ")"]
|
adamc@188
|
916 in
|
adamc@188
|
917 doEm xncs
|
adamc@188
|
918 end
|
adamc@188
|
919
|
adamc@198
|
920 | TDatatype (Option, i, xncs) =>
|
adamc@198
|
921 let
|
adamc@198
|
922 val (x, _) = E.lookupDatatype env i
|
adamc@198
|
923
|
adamc@198
|
924 val (no_arg, has_arg, t) =
|
adamc@198
|
925 case !xncs of
|
adamc@198
|
926 [(no_arg, _, NONE), (has_arg, _, SOME t)] =>
|
adamc@198
|
927 (no_arg, has_arg, t)
|
adamc@198
|
928 | [(has_arg, _, SOME t), (no_arg, _, NONE)] =>
|
adamc@198
|
929 (no_arg, has_arg, t)
|
adamc@198
|
930 | _ => raise Fail "CjrPrint: unfooify misclassified Option datatype"
|
adamc@198
|
931 in
|
adamc@198
|
932 box [string "(request[0] == '/' ? ++request : request,",
|
adamc@198
|
933 newline,
|
adamc@198
|
934 string "((!strncmp(request, \"",
|
adamc@198
|
935 string no_arg,
|
adamc@198
|
936 string "\", ",
|
adamc@198
|
937 string (Int.toString (size no_arg)),
|
adamc@198
|
938 string ") && (request[",
|
adamc@198
|
939 string (Int.toString (size no_arg)),
|
adamc@198
|
940 string "] == 0 || request[",
|
adamc@198
|
941 string (Int.toString (size no_arg)),
|
adamc@198
|
942 string "] == '/')) ? (request",
|
adamc@198
|
943 space,
|
adamc@198
|
944 string "+=",
|
adamc@198
|
945 space,
|
adamc@198
|
946 string (Int.toString (size no_arg)),
|
adamc@198
|
947 string ", NULL) : ((!strncmp(request, \"",
|
adamc@198
|
948 string has_arg,
|
adamc@198
|
949 string "\", ",
|
adamc@198
|
950 string (Int.toString (size has_arg)),
|
adamc@198
|
951 string ") && (request[",
|
adamc@198
|
952 string (Int.toString (size has_arg)),
|
adamc@198
|
953 string "] == 0 || request[",
|
adamc@198
|
954 string (Int.toString (size has_arg)),
|
adamc@198
|
955 string "] == '/')) ? (request",
|
adamc@198
|
956 space,
|
adamc@198
|
957 string "+=",
|
adamc@198
|
958 space,
|
adamc@198
|
959 string (Int.toString (size has_arg)),
|
adamc@200
|
960 string ", (request[0] == '/' ? ++request : NULL), ",
|
adamc@200
|
961 newline,
|
adamc@198
|
962
|
adamc@198
|
963 case #1 t of
|
adamc@198
|
964 TDatatype _ => unurlify t
|
adamc@199
|
965 | TFfi ("Basis", "string") => unurlify t
|
adamc@198
|
966 | _ => box [string "({",
|
adamc@198
|
967 newline,
|
adamc@198
|
968 p_typ env t,
|
adamc@198
|
969 space,
|
adamc@198
|
970 string "*tmp",
|
adamc@198
|
971 space,
|
adamc@198
|
972 string "=",
|
adamc@198
|
973 space,
|
adamc@198
|
974 string "lw_malloc(ctx, sizeof(",
|
adamc@198
|
975 p_typ env t,
|
adamc@198
|
976 string "));",
|
adamc@198
|
977 newline,
|
adamc@198
|
978 string "*tmp",
|
adamc@198
|
979 space,
|
adamc@198
|
980 string "=",
|
adamc@198
|
981 space,
|
adamc@198
|
982 unurlify t,
|
adamc@198
|
983 string ";",
|
adamc@198
|
984 newline,
|
adamc@198
|
985 string "tmp;",
|
adamc@198
|
986 newline,
|
adamc@198
|
987 string "})"],
|
adamc@198
|
988 string ")",
|
adamc@198
|
989 newline,
|
adamc@198
|
990 string ":",
|
adamc@198
|
991 space,
|
adamc@198
|
992 string ("(lw_error(ctx, FATAL, \"Error unurlifying datatype " ^ x ^ "\"), NULL))))")]
|
adamc@198
|
993 end
|
adamc@198
|
994
|
adamc@188
|
995 | TDatatype (Default, i, _) =>
|
adamc@166
|
996 let
|
adamc@166
|
997 val (x, xncs) = E.lookupDatatype env i
|
adamc@166
|
998
|
adamc@166
|
999 fun doEm xncs =
|
adamc@166
|
1000 case xncs of
|
adamc@167
|
1001 [] => string ("(lw_error(ctx, FATAL, \"Error unurlifying datatype " ^ x ^ "\"), NULL)")
|
adamc@167
|
1002 | (x', n, to) :: rest =>
|
adamc@167
|
1003 box [string "((!strncmp(request, \"",
|
adamc@167
|
1004 string x',
|
adamc@167
|
1005 string "\", ",
|
adamc@167
|
1006 string (Int.toString (size x')),
|
adamc@167
|
1007 string ") && (request[",
|
adamc@167
|
1008 string (Int.toString (size x')),
|
adamc@167
|
1009 string "] == 0 || request[",
|
adamc@167
|
1010 string (Int.toString (size x')),
|
adamc@167
|
1011 string "] == '/')) ? ({",
|
adamc@166
|
1012 newline,
|
adamc@167
|
1013 string "struct",
|
adamc@167
|
1014 space,
|
adamc@166
|
1015 string ("__lwd_" ^ x ^ "_" ^ Int.toString i),
|
adamc@166
|
1016 space,
|
adamc@181
|
1017 string "*tmp = lw_malloc(ctx, sizeof(struct __lwd_",
|
adamc@167
|
1018 string x,
|
adamc@167
|
1019 string "_",
|
adamc@167
|
1020 string (Int.toString i),
|
adamc@167
|
1021 string "));",
|
adamc@166
|
1022 newline,
|
adamc@181
|
1023 string "tmp->tag",
|
adamc@166
|
1024 space,
|
adamc@166
|
1025 string "=",
|
adamc@166
|
1026 space,
|
adamc@167
|
1027 string ("__lwc_" ^ x' ^ "_" ^ Int.toString n),
|
adamc@166
|
1028 string ";",
|
adamc@166
|
1029 newline,
|
adamc@166
|
1030 string "request",
|
adamc@166
|
1031 space,
|
adamc@166
|
1032 string "+=",
|
adamc@166
|
1033 space,
|
adamc@167
|
1034 string (Int.toString (size x')),
|
adamc@166
|
1035 string ";",
|
adamc@166
|
1036 newline,
|
adamc@200
|
1037 string "if (request[0] == '/') ++request;",
|
adamc@200
|
1038 newline,
|
adamc@166
|
1039 case to of
|
adamc@166
|
1040 NONE => box []
|
adamc@197
|
1041 | SOME t => box [string "tmp->data.lw_",
|
adamc@167
|
1042 string x',
|
adamc@166
|
1043 space,
|
adamc@166
|
1044 string "=",
|
adamc@166
|
1045 space,
|
adamc@166
|
1046 unurlify t,
|
adamc@166
|
1047 string ";",
|
adamc@166
|
1048 newline],
|
adamc@181
|
1049 string "tmp;",
|
adamc@166
|
1050 newline,
|
adamc@166
|
1051 string "})",
|
adamc@166
|
1052 space,
|
adamc@166
|
1053 string ":",
|
adamc@166
|
1054 space,
|
adamc@166
|
1055 doEm rest,
|
adamc@166
|
1056 string ")"]
|
adamc@166
|
1057 in
|
adamc@166
|
1058 doEm xncs
|
adamc@166
|
1059 end
|
adamc@166
|
1060
|
adamc@144
|
1061 | _ => (ErrorMsg.errorAt loc "Unable to choose a URL decoding function";
|
adamc@144
|
1062 space)
|
adamc@144
|
1063
|
adamc@144
|
1064
|
adamc@144
|
1065 fun p_page (ek, s, n, ts) =
|
adamc@144
|
1066 let
|
adamc@144
|
1067 val (ts, defInputs, inputsVar) =
|
adamc@144
|
1068 case ek of
|
adamc@144
|
1069 Core.Link => (ts, string "", string "")
|
adamc@144
|
1070 | Core.Action =>
|
adamc@144
|
1071 case List.last ts of
|
adamc@144
|
1072 (TRecord i, _) =>
|
adamc@144
|
1073 let
|
adamc@144
|
1074 val xts = E.lookupStruct env i
|
adamc@144
|
1075 in
|
adamc@144
|
1076 (List.drop (ts, 1),
|
adamc@144
|
1077 box [box (map (fn (x, t) => box [p_typ env t,
|
adamc@144
|
1078 space,
|
adamc@144
|
1079 string "lw_input_",
|
adamc@144
|
1080 string x,
|
adamc@144
|
1081 string ";",
|
adamc@144
|
1082 newline]) xts),
|
adamc@144
|
1083 newline,
|
adamc@144
|
1084 box (map (fn (x, t) =>
|
adamc@144
|
1085 let
|
adamc@144
|
1086 val n = case SM.find (fnums, x) of
|
adamc@144
|
1087 NONE => raise Fail "CjrPrint: Can't find in fnums"
|
adamc@144
|
1088 | SOME n => n
|
adamc@190
|
1089
|
adamc@190
|
1090 val f = case t of
|
adamc@190
|
1091 (TFfi ("Basis", "bool"), _) => "optional_"
|
adamc@190
|
1092 | _ => ""
|
adamc@144
|
1093 in
|
adamc@190
|
1094 box [string "request = lw_get_",
|
adamc@190
|
1095 string f,
|
adamc@190
|
1096 string "input(ctx, ",
|
adamc@144
|
1097 string (Int.toString n),
|
adamc@144
|
1098 string ");",
|
adamc@144
|
1099 newline,
|
adamc@144
|
1100 string "if (request == NULL) {",
|
adamc@144
|
1101 newline,
|
adamc@144
|
1102 box [string "printf(\"Missing input ",
|
adamc@144
|
1103 string x,
|
adamc@144
|
1104 string "\\n\");",
|
adamc@144
|
1105 newline,
|
adamc@144
|
1106 string "exit(1);"],
|
adamc@144
|
1107 newline,
|
adamc@144
|
1108 string "}",
|
adamc@144
|
1109 newline,
|
adamc@144
|
1110 string "lw_input_",
|
adamc@144
|
1111 string x,
|
adamc@144
|
1112 space,
|
adamc@144
|
1113 string "=",
|
adamc@144
|
1114 space,
|
adamc@144
|
1115 unurlify t,
|
adamc@144
|
1116 string ";",
|
adamc@144
|
1117 newline]
|
adamc@144
|
1118 end) xts),
|
adamc@144
|
1119 string "struct __lws_",
|
adamc@144
|
1120 string (Int.toString i),
|
adamc@144
|
1121 space,
|
adamc@144
|
1122 string "lw_inputs",
|
adamc@144
|
1123 space,
|
adamc@144
|
1124 string "= {",
|
adamc@144
|
1125 newline,
|
adamc@144
|
1126 box (map (fn (x, _) => box [string "lw_input_",
|
adamc@144
|
1127 string x,
|
adamc@144
|
1128 string ",",
|
adamc@144
|
1129 newline]) xts),
|
adamc@144
|
1130 string "};",
|
adamc@144
|
1131 newline],
|
adamc@144
|
1132 box [string ",",
|
adamc@144
|
1133 space,
|
adamc@144
|
1134 string "lw_inputs"])
|
adamc@144
|
1135 end
|
adamc@144
|
1136
|
adamc@144
|
1137 | _ => raise Fail "CjrPrint: Last argument to an action isn't a record"
|
adamc@144
|
1138 in
|
adamc@144
|
1139 box [string "if (!strncmp(request, \"",
|
adamc@144
|
1140 string (String.toString s),
|
adamc@144
|
1141 string "\", ",
|
adamc@144
|
1142 string (Int.toString (size s)),
|
adamc@198
|
1143 string ") && (request[",
|
adamc@198
|
1144 string (Int.toString (size s)),
|
adamc@198
|
1145 string "] == 0 || request[",
|
adamc@198
|
1146 string (Int.toString (size s)),
|
adamc@198
|
1147 string "] == '/')) {",
|
adamc@144
|
1148 newline,
|
adamc@144
|
1149 string "request += ",
|
adamc@144
|
1150 string (Int.toString (size s)),
|
adamc@144
|
1151 string ";",
|
adamc@144
|
1152 newline,
|
adamc@144
|
1153 string "if (*request == '/') ++request;",
|
adamc@144
|
1154 newline,
|
adamc@144
|
1155 box [string "{",
|
adamc@144
|
1156 newline,
|
adamc@144
|
1157 box (ListUtil.mapi (fn (i, t) => box [p_typ env t,
|
adamc@144
|
1158 space,
|
adamc@144
|
1159 string "arg",
|
adamc@144
|
1160 string (Int.toString i),
|
adamc@144
|
1161 space,
|
adamc@144
|
1162 string "=",
|
adamc@144
|
1163 space,
|
adamc@144
|
1164 unurlify t,
|
adamc@144
|
1165 string ";",
|
adamc@144
|
1166 newline]) ts),
|
adamc@144
|
1167 defInputs,
|
adamc@144
|
1168 p_enamed env n,
|
adamc@144
|
1169 string "(",
|
adamc@144
|
1170 p_list_sep (box [string ",", space])
|
adamc@144
|
1171 (fn x => x)
|
adamc@144
|
1172 (string "ctx" :: ListUtil.mapi (fn (i, _) => string ("arg" ^ Int.toString i)) ts),
|
adamc@144
|
1173 inputsVar,
|
adamc@144
|
1174 string ");",
|
adamc@144
|
1175 newline,
|
adamc@144
|
1176 string "return;",
|
adamc@144
|
1177 newline,
|
adamc@144
|
1178 string "}",
|
adamc@144
|
1179 newline,
|
adamc@144
|
1180 string "}"]
|
adamc@144
|
1181 ]
|
adamc@144
|
1182 end
|
adamc@144
|
1183
|
adamc@144
|
1184 val pds' = map p_page ps
|
adamc@29
|
1185 in
|
adamc@144
|
1186 box [string "#include <stdio.h>",
|
adamc@144
|
1187 newline,
|
adamc@144
|
1188 string "#include <stdlib.h>",
|
adamc@144
|
1189 newline,
|
adamc@144
|
1190 newline,
|
adamc@244
|
1191 string "#include \"urweb.h\"",
|
adamc@101
|
1192 newline,
|
adamc@101
|
1193 newline,
|
adamc@101
|
1194 p_list_sep newline (fn x => x) pds,
|
adamc@101
|
1195 newline,
|
adamc@144
|
1196 string "int lw_inputs_len = ",
|
adamc@144
|
1197 string (Int.toString (SM.foldl Int.max 0 fnums + 1)),
|
adamc@144
|
1198 string ";",
|
adamc@144
|
1199 newline,
|
adamc@144
|
1200 newline,
|
adamc@144
|
1201 string "int lw_input_num(char *name) {",
|
adamc@144
|
1202 newline,
|
adamc@144
|
1203 makeSwitch (fnums, 0),
|
adamc@144
|
1204 string "}",
|
adamc@144
|
1205 newline,
|
adamc@144
|
1206 newline,
|
adamc@117
|
1207 string "void lw_handle(lw_context ctx, char *request) {",
|
adamc@101
|
1208 newline,
|
adamc@101
|
1209 p_list_sep newline (fn x => x) pds',
|
adamc@101
|
1210 newline,
|
adamc@101
|
1211 string "}",
|
adamc@101
|
1212 newline]
|
adamc@29
|
1213 end
|
adamc@29
|
1214
|
adamc@29
|
1215 end
|