annotate src/mono_print.sml @ 821:395a5d450cc0

Chars and more string operations
author Adam Chlipala <adamc@hcoop.net>
date Tue, 26 May 2009 12:25:06 -0400
parents 493f44759879
children 44c2c089ca15
rev   line source
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@757 68 | TList t => box [string "list(",
adamc@757 69 p_typ env t,
adamc@757 70 string ")"]
adamc@577 71 | TSource => string "source"
adamc@568 72 | TSignal t => box [string "signal(",
adamc@568 73 p_typ env t,
adamc@568 74 string ")"]
adamc@25 75
adamc@25 76 and p_typ env = p_typ' false env
adamc@25 77
adamc@109 78 fun p_enamed env n =
adamc@178 79 (if !debug then
adamc@178 80 string (#1 (E.lookupENamed env n) ^ "__" ^ Int.toString n)
adamc@178 81 else
adamc@178 82 string (#1 (E.lookupENamed env n)))
adamc@178 83 handle E.UnboundNamed _ => string ("UNBOUNDN_" ^ Int.toString n)
adamc@178 84
adamc@178 85 fun p_con_named env n =
adamc@178 86 (if !debug then
adamc@178 87 string (#1 (E.lookupConstructor env n) ^ "__" ^ Int.toString n)
adamc@178 88 else
adamc@178 89 string (#1 (E.lookupConstructor env n)))
adamc@178 90 handle E.UnboundNamed _ => string ("CONSTRUCTOR_" ^ Int.toString n)
adamc@178 91
adamc@178 92 fun p_patCon env pc =
adamc@178 93 case pc of
adamc@178 94 PConVar n => p_con_named env n
adamc@186 95 | PConFfi {mod = m, con, ...} => box [string "FFIC(",
adamc@185 96 string m,
adamc@185 97 string ".",
adamc@185 98 string con,
adamc@185 99 string ")"]
adamc@178 100
adamc@178 101 fun p_pat' par env (p, _) =
adamc@178 102 case p of
adamc@178 103 PWild => string "_"
adamc@182 104 | PVar (s, _) => string s
adamc@178 105 | PPrim p => Prim.p_t p
adamc@188 106 | PCon (_, n, NONE) => p_patCon env n
adamc@188 107 | PCon (_, n, SOME p) => parenIf par (box [p_patCon env n,
adamc@288 108 space,
adamc@288 109 p_pat' true env p])
adamc@178 110 | PRecord xps =>
adamc@178 111 box [string "{",
adamc@182 112 p_list_sep (box [string ",", space]) (fn (x, p, _) =>
adamc@178 113 box [string x,
adamc@178 114 space,
adamc@178 115 string "=",
adamc@178 116 space,
adamc@178 117 p_pat env p]) xps,
adamc@178 118 string "}"]
adamc@288 119 | PNone _ => string "None"
adamc@757 120 | PSome (t, p) =>
adamc@757 121 if !debug then
adamc@757 122 box [string "Some[",
adamc@757 123 p_typ env t,
adamc@757 124 string "]",
adamc@757 125 space,
adamc@757 126 p_pat' true env p]
adamc@757 127 else
adamc@757 128 box [string "Some",
adamc@757 129 space,
adamc@757 130 p_pat' true env p]
adamc@178 131
adamc@178 132 and p_pat x = p_pat' false x
adamc@109 133
adamc@591 134 fun p_mode env m =
adamc@591 135 case m of
adamc@591 136 Attribute => string "Attribute"
adamc@591 137 | Script => string "Script"
adamc@591 138 | Source t => box [string "Source", space, p_typ env t]
adamc@591 139
adamc@25 140 fun p_exp' par env (e, _) =
adamc@25 141 case e of
adamc@25 142 EPrim p => Prim.p_t p
adamc@25 143 | ERel n =>
adamc@178 144 ((if !debug then
adamc@178 145 string (#1 (E.lookupERel env n) ^ "_" ^ Int.toString n)
adamc@178 146 else
adamc@178 147 string (#1 (E.lookupERel env n)))
adamc@178 148 handle E.UnboundRel _ => string ("UNBOUND_" ^ Int.toString n))
adamc@109 149 | ENamed n => p_enamed env n
adamc@188 150 | ECon (_, pc, NONE) => p_patCon env pc
adamc@188 151 | ECon (_, pc, SOME e) => parenIf par (box [p_patCon env pc,
adamc@188 152 space,
adamc@188 153 p_exp' true env e])
adamc@297 154 | ENone _ => string "None"
adamc@290 155 | ESome (_, e) => parenIf par (box [string "Some",
adamc@290 156 space,
adamc@290 157 p_exp' true env e])
adamc@109 158
adamc@51 159 | EFfi (m, x) => box [string "FFI(", string m, string ".", string x, string ")"]
adamc@51 160 | EFfiApp (m, x, es) => box [string "FFI(",
adamc@51 161 string m,
adamc@51 162 string ".",
adamc@51 163 string x,
adamc@51 164 string "(",
adamc@51 165 p_list (p_exp env) es,
adamc@51 166 string "))"]
adamc@25 167 | EApp (e1, e2) => parenIf par (box [p_exp env e1,
adamc@25 168 space,
adamc@25 169 p_exp' true env e2])
adamc@252 170 | EAbs (x, t, _, e) => parenIf true (box [string "fn",
adamc@252 171 space,
adamc@252 172 string x,
adamc@252 173 space,
adamc@252 174 string ":",
adamc@252 175 space,
adamc@252 176 p_typ env t,
adamc@252 177 space,
adamc@252 178 string "=>",
adamc@252 179 space,
adamc@252 180 p_exp (E.pushERel env x t NONE) e])
adamc@25 181
adamc@387 182 | EUnop (s, e) => parenIf true (box [string s,
adamc@387 183 space,
adamc@387 184 p_exp' true env e])
adamc@387 185 | EBinop (s, e1, e2) => parenIf true (box [p_exp' true env e1,
adamc@387 186 space,
adamc@387 187 string s,
adamc@387 188 space,
adamc@387 189 p_exp' true env e2])
adamc@387 190
adamc@25 191 | ERecord xes => box [string "{",
adamc@29 192 p_list (fn (x, e, _) =>
adamc@25 193 box [string x,
adamc@25 194 space,
adamc@25 195 string "=",
adamc@25 196 space,
adamc@25 197 p_exp env e]) xes,
adamc@25 198 string "}"]
adamc@25 199 | EField (e, x) =>
adamc@25 200 box [p_exp' true env e,
adamc@25 201 string ".",
adamc@25 202 string x]
adamc@25 203
adamc@252 204 | ECase (e, pes, _) => parenIf true (box [string "case",
adamc@252 205 space,
adamc@252 206 p_exp env e,
adamc@252 207 space,
adamc@252 208 string "of",
adamc@252 209 space,
adamc@252 210 p_list_sep (box [space, string "|", space])
adamc@252 211 (fn (p, e) => box [p_pat env p,
adamc@252 212 space,
adamc@252 213 string "=>",
adamc@252 214 space,
adamc@252 215 p_exp (E.patBinds env p) e]) pes])
adamc@94 216
adamc@283 217 | EError (e, t) => box [string "(error",
adamc@283 218 space,
adamc@283 219 p_exp env e,
adamc@283 220 space,
adamc@283 221 string ":",
adamc@283 222 space,
adamc@283 223 p_typ env t,
adamc@283 224 string ")"]
adamc@741 225 | EReturnBlob {blob, mimeType, t} => box [string "(blob",
adamc@741 226 space,
adamc@741 227 p_exp env blob,
adamc@741 228 space,
adamc@741 229 string "in",
adamc@741 230 space,
adamc@741 231 p_exp env mimeType,
adamc@741 232 space,
adamc@741 233 string ":",
adamc@741 234 space,
adamc@741 235 p_typ env t,
adamc@741 236 string ")"]
adamc@283 237
adamc@331 238 | EStrcat (e1, e2) => parenIf par (box [p_exp' true env e1,
adamc@331 239 space,
adamc@331 240 string "^",
adamc@331 241 space,
adamc@331 242 p_exp env e2])
adamc@94 243
adamc@102 244 | EWrite e => box [string "write(",
adamc@102 245 p_exp env e,
adamc@102 246 string ")"]
adamc@102 247
adamc@578 248 | ESeq (e1, e2) => box [string "(",
adamc@578 249 p_exp env e1,
adamc@106 250 string ";",
adamc@106 251 space,
adamc@578 252 p_exp env e2,
adamc@578 253 string ")"]
adamc@252 254 | ELet (x, t, e1, e2) => box [string "(let",
adamc@251 255 space,
adamc@251 256 string x,
adamc@251 257 space,
adamc@251 258 string ":",
adamc@251 259 space,
adamc@251 260 p_typ env t,
adamc@251 261 space,
adamc@251 262 string "=",
adamc@251 263 space,
adamc@252 264 string "(",
adamc@251 265 p_exp env e1,
adamc@252 266 string ")",
adamc@251 267 space,
adamc@251 268 string "in",
adamc@251 269 space,
adamc@252 270 string "(",
adamc@252 271 p_exp (E.pushERel env x t NONE) e2,
adamc@252 272 string "))"]
adamc@106 273
adamc@111 274 | EClosure (n, es) => box [string "CLOSURE(",
adamc@111 275 p_enamed env n,
adamc@111 276 p_list_sep (string "") (fn e => box [string ", ",
adamc@111 277 p_exp env e]) es,
adamc@111 278 string ")"]
adamc@111 279
adamc@252 280 | EQuery {exps, tables, state, query, body, initial} =>
adamc@252 281 box [string "query[",
adamc@252 282 p_list (fn (x, t) => box [string x, space, string ":", space, p_typ env t]) exps,
adamc@252 283 string "] [",
adamc@252 284 p_list (fn (x, xts) => box [string x,
adamc@252 285 space,
adamc@252 286 string ":",
adamc@252 287 space,
adamc@252 288 string "{",
adamc@252 289 p_list (fn (x, t) => box [string x, space, string ":", space, p_typ env t]) xts,
adamc@252 290 string "}"]) tables,
adamc@252 291 string "] [",
adamc@252 292 p_typ env state,
adamc@252 293 string "]",
adamc@252 294 space,
adamc@252 295 p_exp env query,
adamc@252 296 space,
adamc@252 297 string "initial",
adamc@252 298 space,
adamc@252 299 p_exp env initial,
adamc@252 300 space,
adamc@252 301 string "in",
adamc@252 302 space,
adamc@252 303 p_exp (E.pushERel (E.pushERel env "r" dummyt NONE) "acc" dummyt NONE) body]
adamc@307 304 | EDml e => box [string "dml(",
adamc@307 305 p_exp env e,
adamc@307 306 string ")"]
adamc@338 307 | ENextval e => box [string "nextval(",
adamc@338 308 p_exp env e,
adamc@338 309 string ")"]
adamc@463 310 | EUnurlify (e, _) => box [string "unurlify(",
adamc@463 311 p_exp env e,
adamc@463 312 string ")"]
adamc@815 313 | EJavaScript (m, e) => box [string "JavaScript(",
adamc@815 314 p_mode env m,
adamc@815 315 string ",",
adamc@815 316 space,
adamc@815 317 p_exp env e,
adamc@815 318 string ")"]
adamc@568 319
adamc@568 320 | ESignalReturn e => box [string "Return(",
adamc@568 321 p_exp env e,
adamc@568 322 string ")"]
adamc@574 323 | ESignalBind (e1, e2) => box [string "Bind(",
adamc@572 324 p_exp env e1,
adamc@572 325 string ",",
adamc@572 326 space,
adamc@572 327 p_exp env e2,
adamc@572 328 string ")"]
adamc@574 329 | ESignalSource e => box [string "Source(",
adamc@574 330 p_exp env e,
adamc@574 331 string ")"]
adamc@252 332
adamc@736 333 | EServerCall (n, e, _, _) => box [string "Server(",
adamc@736 334 p_exp env n,
adamc@736 335 string ")[",
adamc@736 336 p_exp env e,
adamc@736 337 string "]"]
adamc@670 338 | ERecv (n, e, _) => box [string "Recv(",
adamc@670 339 p_exp env n,
adamc@670 340 string ")[",
adamc@670 341 p_exp env e,
adamc@670 342 string "]"]
adamc@695 343 | ESleep (n, e) => box [string "Sleep(",
adamc@695 344 p_exp env n,
adamc@695 345 string ")[",
adamc@695 346 p_exp env e,
adamc@695 347 string "]"]
adamc@608 348
adamc@25 349 and p_exp env = p_exp' false env
adamc@25 350
adamc@126 351 fun p_vali env (x, n, t, e, s) =
adamc@126 352 let
adamc@126 353 val xp = if !debug then
adamc@126 354 box [string x,
adamc@126 355 string "__",
adamc@126 356 string (Int.toString n)]
adamc@126 357 else
adamc@126 358 string x
adamc@126 359 in
adamc@126 360 box [xp,
adamc@126 361 space,
adamc@126 362 string "as",
adamc@126 363 space,
adamc@126 364 string s,
adamc@126 365 space,
adamc@126 366 string ":",
adamc@126 367 space,
adamc@126 368 p_typ env t,
adamc@126 369 space,
adamc@126 370 string "=",
adamc@126 371 space,
adamc@126 372 p_exp env e]
adamc@126 373 end
adamc@126 374
adamc@164 375 fun p_datatype env (x, n, cons) =
adamc@164 376 let
adamc@168 377 val env = E.pushDatatype env x n cons
adamc@164 378 in
adamc@808 379 box [string x,
adamc@164 380 space,
adamc@164 381 string "=",
adamc@164 382 space,
adamc@164 383 p_list_sep (box [space, string "|", space])
adamc@164 384 (fn (x, n, NONE) => if !debug then (string (x ^ "__" ^ Int.toString n))
adamc@164 385 else string x
adamc@164 386 | (x, _, SOME t) => box [if !debug then (string (x ^ "__" ^ Int.toString n))
adamc@164 387 else string x, space, string "of", space, p_typ env t])
adamc@164 388 cons]
adamc@164 389 end
adamc@164 390
adamc@126 391 fun p_decl env (dAll as (d, _) : decl) =
adamc@25 392 case d of
adamc@808 393 DDatatype x => box [string "datatype",
adamc@808 394 space,
adamc@808 395 p_list_sep (box [space, string "and", space]) (p_datatype (E.declBinds env dAll)) x]
adamc@164 396 | DVal vi => box [string "val",
adamc@126 397 space,
adamc@126 398 p_vali env vi]
adamc@126 399 | DValRec vis =>
adamc@25 400 let
adamc@126 401 val env = E.declBinds env dAll
adamc@25 402 in
adamc@25 403 box [string "val",
adamc@25 404 space,
adamc@126 405 string "rec",
adamc@25 406 space,
adamc@126 407 p_list_sep (box [newline, string "and", space]) (p_vali env) vis]
adamc@25 408 end
adamc@109 409
adamc@609 410 | DExport (ek, s, n, ts, t) => box [string "export",
adamc@609 411 space,
adamc@732 412 Export.p_export_kind ek,
adamc@609 413 space,
adamc@609 414 p_enamed env n,
adamc@609 415 space,
adamc@609 416 string "as",
adamc@609 417 space,
adamc@609 418 string s,
adamc@609 419 p_list_sep (string "") (fn t => box [space,
adamc@609 420 string "(",
adamc@609 421 p_typ env t,
adamc@609 422 string ")"]) ts,
adamc@609 423 space,
adamc@609 424 string "->",
adamc@609 425 space,
adamc@609 426 p_typ env t]
adamc@271 427
adamc@707 428 | DTable (s, xts, pe, ce) => box [string "(* SQL table ",
adamc@707 429 string s,
adamc@707 430 space,
adamc@707 431 string ":",
adamc@707 432 space,
adamc@707 433 p_list (fn (x, t) => box [string x,
adamc@707 434 space,
adamc@707 435 string ":",
adamc@707 436 space,
adamc@707 437 p_typ env t]) xts,
adamc@707 438 space,
adamc@707 439 string "keys",
adamc@707 440 space,
adamc@707 441 p_exp env pe,
adamc@707 442 space,
adamc@707 443 string "constraints",
adamc@707 444 space,
adamc@707 445 p_exp env ce,
adamc@707 446 space,
adamc@707 447 string "*)"]
adamc@338 448 | DSequence s => box [string "(* SQL sequence ",
adamc@338 449 string s,
adamc@338 450 string "*)"]
adamc@754 451 | DView (s, _, e) => box [string "(* SQL view ",
adamc@754 452 string s,
adamc@754 453 space,
adamc@754 454 string "as",
adamc@754 455 space,
adamc@754 456 p_exp env e,
adamc@754 457 string "*)"]
adamc@687 458 | DDatabase {name, expunge, initialize} => box [string "database",
adamc@687 459 space,
adamc@687 460 string name,
adamc@687 461 space,
adamc@687 462 string "(",
adamc@687 463 p_enamed env expunge,
adamc@687 464 string ",",
adamc@687 465 space,
adamc@687 466 p_enamed env initialize,
adamc@687 467 string ")"]
adamc@569 468 | DJavaScript s => box [string "JavaScript(",
adamc@569 469 string s,
adamc@569 470 string ")"]
adamc@569 471
adamc@725 472 | DCookie s => box [string "cookie",
adamc@725 473 space,
adamc@725 474 string s]
adamc@720 475 | DStyle s => box [string "style",
adamc@720 476 space,
adamc@720 477 string s]
adamc@718 478
adamc@100 479
adamc@25 480 fun p_file env file =
adamc@25 481 let
adamc@31 482 val (pds, _) = ListUtil.foldlMap (fn (d, env) =>
adamc@31 483 (p_decl env d,
adamc@31 484 E.declBinds env d))
adamc@25 485 env file
adamc@25 486 in
adamc@25 487 p_list_sep newline (fn x => x) pds
adamc@25 488 end
adamc@25 489
adamc@25 490 end