annotate src/monoize.sml @ 263:02e1870a0516

Monoize OFFSET
author Adam Chlipala <adamc@hcoop.net>
date Sun, 31 Aug 2008 16:11:15 -0400
parents a6cb33f49366
children 35ed669a253f
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@25 28 structure Monoize :> MONOIZE = struct
adamc@25 29
adamc@25 30 structure E = ErrorMsg
adamc@25 31 structure Env = CoreEnv
adamc@25 32
adamc@25 33 structure L = Core
adamc@25 34 structure L' = Mono
adamc@25 35
adamc@196 36 structure IM = IntBinaryMap
adamc@196 37
adamc@196 38 val dummyTyp = (L'.TDatatype (0, ref (L'.Enum, [])), E.dummySpan)
adamc@25 39
adamc@252 40 structure U = MonoUtil
adamc@252 41
adamc@252 42 val liftExpInExp =
adamc@252 43 U.Exp.mapB {typ = fn t => t,
adamc@252 44 exp = fn bound => fn e =>
adamc@252 45 case e of
adamc@252 46 L'.ERel xn =>
adamc@252 47 if xn < bound then
adamc@252 48 e
adamc@252 49 else
adamc@252 50 L'.ERel (xn + 1)
adamc@252 51 | _ => e,
adamc@252 52 bind = fn (bound, U.Exp.RelE _) => bound + 1
adamc@252 53 | (bound, _) => bound}
adamc@252 54
adamc@25 55 fun monoName env (all as (c, loc)) =
adamc@25 56 let
adamc@25 57 fun poly () =
adamc@25 58 (E.errorAt loc "Unsupported name constructor";
adamc@25 59 Print.eprefaces' [("Constructor", CorePrint.p_con env all)];
adamc@25 60 "")
adamc@25 61 in
adamc@25 62 case c of
adamc@25 63 L.CName s => s
adamc@25 64 | _ => poly ()
adamc@25 65 end
adamc@25 66
adamc@196 67 fun monoType env =
adamc@25 68 let
adamc@196 69 fun mt env dtmap (all as (c, loc)) =
adamc@196 70 let
adamc@196 71 fun poly () =
adamc@196 72 (E.errorAt loc "Unsupported type constructor";
adamc@196 73 Print.eprefaces' [("Constructor", CorePrint.p_con env all)];
adamc@196 74 dummyTyp)
adamc@196 75 in
adamc@196 76 case c of
adamc@196 77 L.TFun (c1, c2) => (L'.TFun (mt env dtmap c1, mt env dtmap c2), loc)
adamc@196 78 | L.TCFun _ => poly ()
adamc@196 79 | L.TRecord (L.CRecord ((L.KType, _), xcs), _) =>
adamc@196 80 (L'.TRecord (map (fn (x, t) => (monoName env x, mt env dtmap t)) xcs), loc)
adamc@196 81 | L.TRecord _ => poly ()
adamc@196 82
adamc@196 83 | L.CApp ((L.CApp ((L.CApp ((L.CFfi ("Basis", "xml"), _), _), _), _), _), _) =>
adamc@196 84 (L'.TFfi ("Basis", "string"), loc)
adamc@196 85 | L.CApp ((L.CApp ((L.CFfi ("Basis", "xhtml"), _), _), _), _) =>
adamc@196 86 (L'.TFfi ("Basis", "string"), loc)
adamc@196 87
adamc@251 88 | L.CApp ((L.CFfi ("Basis", "transaction"), _), t) =>
adamc@252 89 (L'.TFun ((L'.TRecord [], loc), mt env dtmap t), loc)
adamc@252 90 | L.CApp ((L.CFfi ("Basis", "sql_table"), _), _) =>
adamc@252 91 (L'.TFfi ("Basis", "string"), loc)
adamc@252 92 | L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_query"), _), _), _), _) =>
adamc@252 93 (L'.TFfi ("Basis", "string"), loc)
adamc@252 94 | L.CApp ((L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_query1"), _), _), _), _), _), _) =>
adamc@252 95 (L'.TFfi ("Basis", "string"), loc)
adamc@252 96 | L.CApp ((L.CApp ((L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_exp"), _), _), _), _), _), _), _), _) =>
adamc@252 97 (L'.TFfi ("Basis", "string"), loc)
adamc@252 98
adamc@252 99 | L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_subset"), _), _), _), _) =>
adamc@252 100 (L'.TRecord [], loc)
adamc@252 101 | L.CFfi ("Basis", "sql_relop") =>
adamc@252 102 (L'.TFfi ("Basis", "string"), loc)
adamc@252 103 | L.CFfi ("Basis", "sql_direction") =>
adamc@252 104 (L'.TFfi ("Basis", "string"), loc)
adamc@252 105 | L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_order_by"), _), _), _), _) =>
adamc@252 106 (L'.TFfi ("Basis", "string"), loc)
adamc@252 107 | L.CFfi ("Basis", "sql_limit") =>
adamc@252 108 (L'.TFfi ("Basis", "string"), loc)
adamc@252 109 | L.CFfi ("Basis", "sql_offset") =>
adamc@252 110 (L'.TFfi ("Basis", "string"), loc)
adamc@252 111
adamc@252 112 | L.CApp ((L.CFfi ("Basis", "sql_injectable"), _), t) =>
adamc@252 113 (L'.TFun (mt env dtmap t, (L'.TFfi ("Basis", "string"), loc)), loc)
adamc@252 114 | L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_unary"), _), _), _), _) =>
adamc@252 115 (L'.TFfi ("Basis", "string"), loc)
adamc@252 116 | L.CApp ((L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_binary"), _), _), _), _), _), _) =>
adamc@252 117 (L'.TFfi ("Basis", "string"), loc)
adamc@252 118 | L.CFfi ("Basis", "sql_comparison") =>
adamc@252 119 (L'.TFfi ("Basis", "string"), loc)
adamc@252 120 | L.CApp ((L.CFfi ("Basis", "sql_aggregate"), _), t) =>
adamc@252 121 (L'.TFfi ("Basis", "string"), loc)
adamc@252 122 | L.CApp ((L.CFfi ("Basis", "sql_summable"), _), _) =>
adamc@252 123 (L'.TRecord [], loc)
adamc@252 124 | L.CApp ((L.CFfi ("Basis", "sql_maxable"), _), _) =>
adamc@252 125 (L'.TRecord [], loc)
adamc@251 126
adamc@196 127 | L.CRel _ => poly ()
adamc@196 128 | L.CNamed n =>
adamc@196 129 (case IM.find (dtmap, n) of
adamc@196 130 SOME r => (L'.TDatatype (n, r), loc)
adamc@196 131 | NONE =>
adamc@196 132 let
adamc@196 133 val r = ref (L'.Default, [])
adamc@196 134 val (_, xs, xncs) = Env.lookupDatatype env n
adamc@196 135
adamc@196 136 val dtmap' = IM.insert (dtmap, n, r)
adamc@196 137
adamc@196 138 val xncs = map (fn (x, n, to) => (x, n, Option.map (mt env dtmap') to)) xncs
adamc@196 139 in
adamc@196 140 case xs of
adamc@198 141 [] =>(r := (ElabUtil.classifyDatatype xncs, xncs);
adamc@196 142 (L'.TDatatype (n, r), loc))
adamc@196 143 | _ => poly ()
adamc@196 144 end)
adamc@196 145 | L.CFfi mx => (L'.TFfi mx, loc)
adamc@196 146 | L.CApp _ => poly ()
adamc@196 147 | L.CAbs _ => poly ()
adamc@196 148
adamc@196 149 | L.CName _ => poly ()
adamc@196 150
adamc@196 151 | L.CRecord _ => poly ()
adamc@196 152 | L.CConcat _ => poly ()
adamc@196 153 | L.CFold _ => poly ()
adamc@196 154 | L.CUnit => poly ()
adamc@214 155
adamc@214 156 | L.CTuple _ => poly ()
adamc@214 157 | L.CProj _ => poly ()
adamc@196 158 end
adamc@25 159 in
adamc@196 160 mt env IM.empty
adamc@25 161 end
adamc@25 162
adamc@25 163 val dummyExp = (L'.EPrim (Prim.Int 0), E.dummySpan)
adamc@25 164
adamc@179 165 structure IM = IntBinaryMap
adamc@179 166
adamc@179 167 datatype foo_kind =
adamc@179 168 Attr
adamc@179 169 | Url
adamc@179 170
adamc@179 171 fun fk2s fk =
adamc@179 172 case fk of
adamc@179 173 Attr => "attr"
adamc@179 174 | Url => "url"
adamc@179 175
adamc@179 176 structure Fm :> sig
adamc@179 177 type t
adamc@179 178
adamc@179 179 val empty : int -> t
adamc@179 180
adamc@179 181 val lookup : t -> foo_kind -> int -> (int -> t -> L'.decl * t) -> t * int
adamc@179 182 val enter : t -> t
adamc@179 183 val decls : t -> L'.decl list
adamc@179 184 end = struct
adamc@179 185
adamc@179 186 structure M = BinaryMapFn(struct
adamc@179 187 type ord_key = foo_kind
adamc@179 188 fun compare x =
adamc@179 189 case x of
adamc@179 190 (Attr, Attr) => EQUAL
adamc@179 191 | (Attr, _) => LESS
adamc@179 192 | (_, Attr) => GREATER
adamc@179 193
adamc@179 194 | (Url, Url) => EQUAL
adamc@179 195 end)
adamc@179 196
adamc@179 197 type t = {
adamc@179 198 count : int,
adamc@179 199 map : int IM.map M.map,
adamc@179 200 decls : L'.decl list
adamc@179 201 }
adamc@179 202
adamc@179 203 fun empty count = {
adamc@179 204 count = count,
adamc@179 205 map = M.empty,
adamc@179 206 decls = []
adamc@179 207 }
adamc@179 208
adamc@179 209 fun enter ({count, map, ...} : t) = {count = count, map = map, decls = []}
adamc@179 210 fun decls ({decls, ...} : t) = decls
adamc@179 211
adamc@179 212 fun lookup (t as {count, map, decls}) k n thunk =
adamc@120 213 let
adamc@179 214 val im = Option.getOpt (M.find (map, k), IM.empty)
adamc@179 215 in
adamc@179 216 case IM.find (im, n) of
adamc@179 217 NONE =>
adamc@179 218 let
adamc@179 219 val n' = count
adamc@179 220 val (d, {count, map, decls}) = thunk count {count = count + 1,
adamc@179 221 map = M.insert (map, k, IM.insert (im, n, n')),
adamc@179 222 decls = decls}
adamc@179 223 in
adamc@179 224 ({count = count,
adamc@179 225 map = map,
adamc@179 226 decls = d :: decls}, n')
adamc@179 227 end
adamc@179 228 | SOME n' => (t, n')
adamc@179 229 end
adamc@179 230
adamc@179 231 end
adamc@185 232
adamc@185 233
adamc@185 234 fun capitalize s =
adamc@185 235 if s = "" then
adamc@185 236 s
adamc@185 237 else
adamc@185 238 str (Char.toUpper (String.sub (s, 0))) ^ String.extract (s, 1, NONE)
adamc@179 239
adamc@179 240 fun fooifyExp fk env =
adamc@179 241 let
adamc@179 242 fun fooify fm (e, tAll as (t, loc)) =
adamc@120 243 case #1 e of
adamc@120 244 L'.EClosure (fnam, [(L'.ERecord [], _)]) =>
adamc@120 245 let
adamc@120 246 val (_, _, _, s) = Env.lookupENamed env fnam
adamc@120 247 in
adamc@183 248 ((L'.EPrim (Prim.String ("/" ^ s)), loc), fm)
adamc@120 249 end
adamc@120 250 | L'.EClosure (fnam, args) =>
adamc@120 251 let
adamc@120 252 val (_, ft, _, s) = Env.lookupENamed env fnam
adamc@120 253 val ft = monoType env ft
adamc@111 254
adamc@179 255 fun attrify (args, ft, e, fm) =
adamc@120 256 case (args, ft) of
adamc@179 257 ([], _) => (e, fm)
adamc@120 258 | (arg :: args, (L'.TFun (t, ft), _)) =>
adamc@179 259 let
adamc@179 260 val (arg', fm) = fooify fm (arg, t)
adamc@179 261 in
adamc@179 262 attrify (args, ft,
adamc@179 263 (L'.EStrcat (e,
adamc@179 264 (L'.EStrcat ((L'.EPrim (Prim.String "/"), loc),
adamc@179 265 arg'), loc)), loc),
adamc@179 266 fm)
adamc@179 267 end
adamc@120 268 | _ => (E.errorAt loc "Type mismatch encoding attribute";
adamc@179 269 (e, fm))
adamc@120 270 in
adamc@183 271 attrify (args, ft, (L'.EPrim (Prim.String ("/" ^ s)), loc), fm)
adamc@120 272 end
adamc@120 273 | _ =>
adamc@120 274 case t of
adamc@185 275 L'.TFfi (m, x) => ((L'.EFfiApp (m, fk2s fk ^ "ify" ^ capitalize x, [e]), loc), fm)
adamc@200 276
adamc@179 277 | L'.TRecord [] => ((L'.EPrim (Prim.String ""), loc), fm)
adamc@200 278 | L'.TRecord ((x, t) :: xts) =>
adamc@200 279 let
adamc@200 280 val (se, fm) = fooify fm ((L'.EField (e, x), loc), t)
adamc@200 281 in
adamc@200 282 foldl (fn ((x, t), (se, fm)) =>
adamc@200 283 let
adamc@200 284 val (se', fm) = fooify fm ((L'.EField (e, x), loc), t)
adamc@200 285 in
adamc@200 286 ((L'.EStrcat (se,
adamc@200 287 (L'.EStrcat ((L'.EPrim (Prim.String "/"), loc),
adamc@200 288 se'), loc)), loc),
adamc@200 289 fm)
adamc@200 290 end) (se, fm) xts
adamc@200 291 end
adamc@111 292
adamc@196 293 | L'.TDatatype (i, ref (dk, _)) =>
adamc@179 294 let
adamc@179 295 fun makeDecl n fm =
adamc@179 296 let
adamc@193 297 val (x, _, xncs) = Env.lookupDatatype env i
adamc@179 298
adamc@179 299 val (branches, fm) =
adamc@179 300 ListUtil.foldlMap
adamc@179 301 (fn ((x, n, to), fm) =>
adamc@179 302 case to of
adamc@179 303 NONE =>
adamc@188 304 (((L'.PCon (dk, L'.PConVar n, NONE), loc),
adamc@179 305 (L'.EPrim (Prim.String x), loc)),
adamc@179 306 fm)
adamc@179 307 | SOME t =>
adamc@179 308 let
adamc@182 309 val t = monoType env t
adamc@182 310 val (arg, fm) = fooify fm ((L'.ERel 0, loc), t)
adamc@179 311 in
adamc@188 312 (((L'.PCon (dk, L'.PConVar n, SOME (L'.PVar ("a", t), loc)), loc),
adamc@179 313 (L'.EStrcat ((L'.EPrim (Prim.String (x ^ "/")), loc),
adamc@179 314 arg), loc)),
adamc@179 315 fm)
adamc@179 316 end)
adamc@179 317 fm xncs
adamc@179 318
adamc@179 319 val dom = tAll
adamc@179 320 val ran = (L'.TFfi ("Basis", "string"), loc)
adamc@179 321 in
adamc@179 322 ((L'.DValRec [(fk2s fk ^ "ify_" ^ x,
adamc@179 323 n,
adamc@179 324 (L'.TFun (dom, ran), loc),
adamc@179 325 (L'.EAbs ("x",
adamc@179 326 dom,
adamc@179 327 ran,
adamc@179 328 (L'.ECase ((L'.ERel 0, loc),
adamc@179 329 branches,
adamc@182 330 {disc = dom,
adamc@182 331 result = ran}), loc)), loc),
adamc@179 332 "")], loc),
adamc@179 333 fm)
adamc@179 334 end
adamc@179 335
adamc@179 336 val (fm, n) = Fm.lookup fm fk i makeDecl
adamc@179 337 in
adamc@179 338 ((L'.EApp ((L'.ENamed n, loc), e), loc), fm)
adamc@179 339 end
adamc@164 340
adamc@120 341 | _ => (E.errorAt loc "Don't know how to encode attribute type";
adamc@120 342 Print.eprefaces' [("Type", MonoPrint.p_typ MonoEnv.empty tAll)];
adamc@179 343 (dummyExp, fm))
adamc@120 344 in
adamc@120 345 fooify
adamc@120 346 end
adamc@120 347
adamc@179 348 val attrifyExp = fooifyExp Attr
adamc@179 349 val urlifyExp = fooifyExp Url
adamc@105 350
adamc@143 351 datatype 'a failable_search =
adamc@143 352 Found of 'a
adamc@143 353 | NotFound
adamc@143 354 | Error
adamc@143 355
adamc@153 356 structure St :> sig
adamc@153 357 type t
adamc@153 358
adamc@153 359 val empty : t
adamc@153 360
adamc@153 361 val radioGroup : t -> string option
adamc@153 362 val setRadioGroup : t * string -> t
adamc@153 363 end = struct
adamc@153 364
adamc@153 365 type t = {
adamc@153 366 radioGroup : string option
adamc@153 367 }
adamc@153 368
adamc@153 369 val empty = {radioGroup = NONE}
adamc@153 370
adamc@153 371 fun radioGroup (t : t) = #radioGroup t
adamc@153 372
adamc@153 373 fun setRadioGroup (t : t, x) = {radioGroup = SOME x}
adamc@153 374
adamc@153 375 end
adamc@153 376
adamc@186 377 fun monoPatCon env pc =
adamc@178 378 case pc of
adamc@178 379 L.PConVar n => L'.PConVar n
adamc@188 380 | L.PConFfi {mod = m, datatyp, con, arg, ...} => L'.PConFfi {mod = m, datatyp = datatyp, con = con,
adamc@188 381 arg = Option.map (monoType env) arg}
adamc@178 382
adamc@193 383 val dummyPat = (L'.PPrim (Prim.Int 0), ErrorMsg.dummySpan)
adamc@193 384
adamc@193 385 fun monoPat env (all as (p, loc)) =
adamc@193 386 let
adamc@193 387 fun poly () =
adamc@193 388 (E.errorAt loc "Unsupported pattern";
adamc@193 389 Print.eprefaces' [("Pattern", CorePrint.p_pat env all)];
adamc@193 390 dummyPat)
adamc@193 391 in
adamc@193 392 case p of
adamc@193 393 L.PWild => (L'.PWild, loc)
adamc@193 394 | L.PVar (x, t) => (L'.PVar (x, monoType env t), loc)
adamc@193 395 | L.PPrim p => (L'.PPrim p, loc)
adamc@193 396 | L.PCon (dk, pc, [], po) => (L'.PCon (dk, monoPatCon env pc, Option.map (monoPat env) po), loc)
adamc@193 397 | L.PCon _ => poly ()
adamc@193 398 | L.PRecord xps => (L'.PRecord (map (fn (x, p, t) => (x, monoPat env p, monoType env t)) xps), loc)
adamc@193 399 end
adamc@178 400
adamc@252 401 fun strcat loc es =
adamc@252 402 case es of
adamc@252 403 [] => (L'.EPrim (Prim.String ""), loc)
adamc@252 404 | [e] => e
adamc@252 405 | _ =>
adamc@252 406 let
adamc@252 407 val e2 = List.last es
adamc@252 408 val es = List.take (es, length es - 1)
adamc@252 409 val e1 = List.last es
adamc@252 410 val es = List.take (es, length es - 1)
adamc@252 411 in
adamc@252 412 foldr (fn (e, e') => (L'.EStrcat (e, e'), loc))
adamc@252 413 (L'.EStrcat (e1, e2), loc) es
adamc@252 414 end
adamc@252 415
adamc@252 416 fun strcatComma loc es =
adamc@252 417 case es of
adamc@252 418 [] => (L'.EPrim (Prim.String ""), loc)
adamc@252 419 | [e] => e
adamc@252 420 | _ =>
adamc@252 421 let
adamc@252 422 val e1 = List.last es
adamc@252 423 val es = List.take (es, length es - 1)
adamc@252 424 in
adamc@252 425 foldr (fn (e, e') =>
adamc@252 426 case e of
adamc@252 427 (L'.EPrim (Prim.String ""), _) => e'
adamc@252 428 | _ =>
adamc@252 429 (L'.EStrcat (e,
adamc@252 430 (L'.EStrcat ((L'.EPrim (Prim.String ", "), loc), e'), loc)), loc))
adamc@252 431 e1 es
adamc@252 432 end
adamc@252 433
adamc@252 434 fun strcatR loc e xs = strcatComma loc (map (fn (x, _) => (L'.EField (e, x), loc)) xs)
adamc@252 435
adamc@179 436 fun monoExp (env, st, fm) (all as (e, loc)) =
adamc@25 437 let
adamc@25 438 fun poly () =
adamc@25 439 (E.errorAt loc "Unsupported expression";
adamc@25 440 Print.eprefaces' [("Expression", CorePrint.p_exp env all)];
adamc@179 441 (dummyExp, fm))
adamc@25 442 in
adamc@25 443 case e of
adamc@179 444 L.EPrim p => ((L'.EPrim p, loc), fm)
adamc@179 445 | L.ERel n => ((L'.ERel n, loc), fm)
adamc@179 446 | L.ENamed n => ((L'.ENamed n, loc), fm)
adamc@193 447 | L.ECon (dk, pc, [], eo) =>
adamc@193 448 let
adamc@179 449 val (eo, fm) =
adamc@179 450 case eo of
adamc@179 451 NONE => (NONE, fm)
adamc@179 452 | SOME e =>
adamc@179 453 let
adamc@179 454 val (e, fm) = monoExp (env, st, fm) e
adamc@179 455 in
adamc@179 456 (SOME e, fm)
adamc@179 457 end
adamc@179 458 in
adamc@188 459 ((L'.ECon (dk, monoPatCon env pc, eo), loc), fm)
adamc@193 460 end
adamc@193 461 | L.ECon _ => poly ()
adamc@94 462
adamc@251 463 | L.ECApp ((L.EFfi ("Basis", "return"), _), t) =>
adamc@252 464 let
adamc@252 465 val t = monoType env t
adamc@252 466 in
adamc@252 467 ((L'.EAbs ("x", t,
adamc@252 468 (L'.TFun ((L'.TRecord [], loc), t), loc),
adamc@252 469 (L'.EAbs ("_", (L'.TRecord [], loc), t,
adamc@252 470 (L'.ERel 1, loc)), loc)), loc), fm)
adamc@252 471 end
adamc@251 472 | L.ECApp ((L.ECApp ((L.EFfi ("Basis", "bind"), _), t1), _), t2) =>
adamc@251 473 let
adamc@251 474 val t1 = monoType env t1
adamc@251 475 val t2 = monoType env t2
adamc@251 476 val un = (L'.TRecord [], loc)
adamc@252 477 val mt1 = (L'.TFun (un, t1), loc)
adamc@252 478 val mt2 = (L'.TFun (un, t2), loc)
adamc@251 479 in
adamc@252 480 ((L'.EAbs ("m1", mt1, (L'.TFun (mt1, (L'.TFun (mt2, (L'.TFun (un, un), loc)), loc)), loc),
adamc@252 481 (L'.EAbs ("m2", mt2, (L'.TFun (un, un), loc),
adamc@252 482 (L'.EAbs ("_", un, un,
adamc@252 483 (L'.ELet ("r", t1, (L'.EApp ((L'.ERel 2, loc),
adamc@252 484 (L'.ERecord [], loc)), loc),
adamc@252 485 (L'.EApp (
adamc@252 486 (L'.EApp ((L'.ERel 2, loc), (L'.ERel 0, loc)), loc),
adamc@252 487 (L'.ERecord [], loc)),
adamc@252 488 loc)), loc)), loc)), loc)), loc),
adamc@251 489 fm)
adamc@251 490 end
adamc@251 491
adamc@252 492 | L.ECApp (
adamc@252 493 (L.ECApp (
adamc@252 494 (L.ECApp ((L.EFfi ("Basis", "query"), _), (L.CRecord (_, tables), _)), _),
adamc@252 495 exps), _),
adamc@252 496 state) =>
adamc@252 497 (case monoType env (L.TRecord exps, loc) of
adamc@252 498 (L'.TRecord exps, _) =>
adamc@252 499 let
adamc@252 500 val tables = map (fn ((L.CName x, _), xts) =>
adamc@252 501 (case monoType env (L.TRecord xts, loc) of
adamc@252 502 (L'.TRecord xts, _) => SOME (x, xts)
adamc@252 503 | _ => NONE)
adamc@252 504 | _ => NONE) tables
adamc@252 505 in
adamc@252 506 if List.exists (fn x => x = NONE) tables then
adamc@252 507 poly ()
adamc@252 508 else
adamc@252 509 let
adamc@252 510 val tables = List.mapPartial (fn x => x) tables
adamc@252 511 val state = monoType env state
adamc@252 512 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@252 513 val un = (L'.TRecord [], loc)
adamc@252 514
adamc@252 515 val rt = exps @ map (fn (x, xts) => (x, (L'.TRecord xts, loc))) tables
adamc@252 516 val ft = (L'.TFun ((L'.TRecord rt, loc),
adamc@252 517 (L'.TFun (state,
adamc@252 518 (L'.TFun (un, state), loc)),
adamc@252 519 loc)), loc)
adamc@252 520
adamc@252 521 val body' = (L'.EAbs ("r", (L'.TRecord rt, loc),
adamc@252 522 (L'.TFun (state, state), loc),
adamc@252 523 (L'.EAbs ("acc", state, state,
adamc@252 524 (L'.EApp (
adamc@252 525 (L'.EApp (
adamc@252 526 (L'.EApp ((L'.ERel 4, loc),
adamc@252 527 (L'.ERel 1, loc)), loc),
adamc@252 528 (L'.ERel 0, loc)), loc),
adamc@252 529 (L'.ERecord [], loc)), loc)), loc)), loc)
adamc@252 530
adamc@252 531 val body = (L'.EQuery {exps = exps,
adamc@252 532 tables = tables,
adamc@252 533 state = state,
adamc@252 534 query = (L'.ERel 3, loc),
adamc@252 535 body = body',
adamc@252 536 initial = (L'.ERel 1, loc)},
adamc@252 537 loc)
adamc@252 538 in
adamc@252 539 ((L'.EAbs ("q", s, (L'.TFun (ft, (L'.TFun (state, (L'.TFun (un, state), loc)), loc)), loc),
adamc@252 540 (L'.EAbs ("f", ft, (L'.TFun (state, (L'.TFun (un, state), loc)), loc),
adamc@252 541 (L'.EAbs ("i", state, (L'.TFun (un, state), loc),
adamc@252 542 (L'.EAbs ("_", un, state,
adamc@252 543 body), loc)), loc)), loc)), loc), fm)
adamc@252 544 end
adamc@252 545 end
adamc@252 546 | _ => poly ())
adamc@252 547
adamc@252 548 | L.ECApp ((L.ECApp ((L.ECApp ((L.EFfi ("Basis", "sql_query"), _), _), _), _), _), _) =>
adamc@252 549 let
adamc@252 550 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@252 551 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@252 552 fun gf s = (L'.EField ((L'.ERel 0, loc), s), loc)
adamc@252 553 in
adamc@252 554 ((L'.EAbs ("r",
adamc@252 555 (L'.TRecord [("Rows", s), ("OrderBy", s), ("Limit", s), ("Offset", s)], loc),
adamc@252 556 s,
adamc@252 557 strcat loc [gf "Rows",
adamc@261 558 (L'.ECase (gf "OrderBy",
adamc@261 559 [((L'.PPrim (Prim.String ""), loc), sc ""),
adamc@261 560 ((L'.PWild, loc),
adamc@261 561 strcat loc [sc " ORDER BY ",
adamc@261 562 gf "OrderBy"])],
adamc@261 563 {disc = s, result = s}), loc),
adamc@252 564 gf "Limit",
adamc@252 565 gf "Offset"]), loc), fm)
adamc@252 566 end
adamc@252 567
adamc@252 568 | L.ECApp (
adamc@252 569 (L.ECApp (
adamc@252 570 (L.ECApp (
adamc@252 571 (L.ECApp (
adamc@252 572 (L.EFfi ("Basis", "sql_query1"), _),
adamc@252 573 (L.CRecord (_, tables), _)), _),
adamc@252 574 (L.CRecord (_, grouped), _)), _),
adamc@252 575 (L.CRecord (_, stables), _)), _),
adamc@252 576 sexps) =>
adamc@252 577 let
adamc@252 578 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@252 579 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@252 580 val un = (L'.TRecord [], loc)
adamc@252 581 fun gf s = (L'.EField ((L'.ERel 0, loc), s), loc)
adamc@252 582
adamc@252 583 fun doTables tables =
adamc@252 584 let
adamc@252 585 val tables = map (fn ((L.CName x, _), xts) =>
adamc@252 586 (case monoType env (L.TRecord xts, loc) of
adamc@252 587 (L'.TRecord xts, _) => SOME (x, xts)
adamc@252 588 | _ => NONE)
adamc@252 589 | _ => NONE) tables
adamc@252 590 in
adamc@252 591 if List.exists (fn x => x = NONE) tables then
adamc@252 592 NONE
adamc@252 593 else
adamc@260 594 let
adamc@260 595 val tables = List.mapPartial (fn x => x) tables
adamc@260 596 val tables = ListMergeSort.sort
adamc@260 597 (fn ((x, _), (y, _)) => String.compare (x, y) = GREATER)
adamc@260 598 tables
adamc@260 599 val tables = map (fn (x, xts) =>
adamc@260 600 (x, ListMergeSort.sort
adamc@260 601 (fn ((x, _), (y, _)) => String.compare (x, y) = GREATER)
adamc@260 602 xts)) tables
adamc@260 603 in
adamc@260 604 SOME tables
adamc@260 605 end
adamc@252 606 end
adamc@252 607 in
adamc@252 608 case (doTables tables, doTables grouped, doTables stables, monoType env (L.TRecord sexps, loc)) of
adamc@252 609 (SOME tables, SOME grouped, SOME stables, (L'.TRecord sexps, _)) =>
adamc@252 610 ((L'.EAbs ("r",
adamc@252 611 (L'.TRecord [("From", (L'.TRecord (map (fn (x, _) => (x, s)) tables), loc)),
adamc@252 612 ("Where", s),
adamc@252 613 ("GroupBy", un),
adamc@252 614 ("Having", s),
adamc@252 615 ("SelectFields", un),
adamc@252 616 ("SelectExps", (L'.TRecord (map (fn (x, _) => (x, s)) sexps), loc))],
adamc@252 617 loc),
adamc@252 618 s,
adamc@252 619 strcat loc [sc "SELECT ",
adamc@261 620 strcatComma loc (map (fn (x, t) =>
adamc@261 621 strcat loc [
adamc@261 622 (L'.EField (gf "SelectExps", x), loc),
adamc@261 623 sc (" AS _" ^ x)
adamc@261 624 ]) sexps),
adamc@252 625 case sexps of
adamc@252 626 [] => sc ""
adamc@252 627 | _ => sc ", ",
adamc@252 628 strcatComma loc (map (fn (x, xts) =>
adamc@252 629 strcatComma loc
adamc@252 630 (map (fn (x', _) =>
adamc@252 631 sc (x ^ "." ^ x'))
adamc@252 632 xts)) stables),
adamc@252 633 sc " FROM ",
adamc@252 634 strcatComma loc (map (fn (x, _) => strcat loc [(L'.EField (gf "From", x), loc),
adamc@253 635 sc (" AS " ^ x)]) tables),
adamc@258 636 (L'.ECase (gf "Where",
adamc@258 637 [((L'.PPrim (Prim.String "TRUE"), loc),
adamc@258 638 sc ""),
adamc@258 639 ((L'.PWild, loc),
adamc@258 640 strcat loc [sc " WHERE ", gf "Where"])],
adamc@258 641 {disc = s,
adamc@258 642 result = s}), loc),
adamc@258 643
adamc@255 644 if List.all (fn (x, xts) =>
adamc@255 645 case List.find (fn (x', _) => x' = x) grouped of
adamc@255 646 NONE => List.null xts
adamc@255 647 | SOME (_, xts') =>
adamc@255 648 List.all (fn (x, _) =>
adamc@255 649 List.exists (fn (x', _) => x' = x)
adamc@255 650 xts') xts) tables then
adamc@255 651 sc ""
adamc@255 652 else
adamc@255 653 strcat loc [
adamc@255 654 sc " GROUP BY ",
adamc@255 655 strcatComma loc (map (fn (x, xts) =>
adamc@255 656 strcatComma loc
adamc@255 657 (map (fn (x', _) =>
adamc@255 658 sc (x ^ "." ^ x'))
adamc@255 659 xts)) grouped)
adamc@259 660 ],
adamc@259 661
adamc@259 662 (L'.ECase (gf "Having",
adamc@259 663 [((L'.PPrim (Prim.String "TRUE"), loc),
adamc@259 664 sc ""),
adamc@259 665 ((L'.PWild, loc),
adamc@259 666 strcat loc [sc " HAVING ", gf "Having"])],
adamc@259 667 {disc = s,
adamc@259 668 result = s}), loc)
adamc@252 669 ]), loc),
adamc@252 670 fm)
adamc@252 671 | _ => poly ()
adamc@252 672 end
adamc@252 673
adamc@252 674 | L.ECApp (
adamc@252 675 (L.ECApp (
adamc@252 676 (L.ECApp (
adamc@252 677 (L.ECApp (
adamc@252 678 (L.EFfi ("Basis", "sql_inject"), _),
adamc@252 679 _), _),
adamc@252 680 _), _),
adamc@252 681 _), _),
adamc@252 682 t) =>
adamc@252 683 let
adamc@252 684 val t = monoType env t
adamc@252 685 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@252 686 in
adamc@252 687 ((L'.EAbs ("f", (L'.TFun (t, s), loc), (L'.TFun (t, s), loc),
adamc@252 688 (L'.ERel 0, loc)), loc), fm)
adamc@252 689 end
adamc@252 690
adamc@253 691 | L.EFfi ("Basis", "sql_int") =>
adamc@253 692 ((L'.EAbs ("x", (L'.TFfi ("Basis", "int"), loc), (L'.TFfi ("Basis", "string"), loc),
adamc@253 693 (L'.EFfiApp ("Basis", "sqlifyInt", [(L'.ERel 0, loc)]), loc)), loc),
adamc@253 694 fm)
adamc@253 695 | L.EFfi ("Basis", "sql_float") =>
adamc@253 696 ((L'.EAbs ("x", (L'.TFfi ("Basis", "float"), loc), (L'.TFfi ("Basis", "string"), loc),
adamc@253 697 (L'.EFfiApp ("Basis", "sqlifyFloat", [(L'.ERel 0, loc)]), loc)), loc),
adamc@253 698 fm)
adamc@253 699 | L.EFfi ("Basis", "sql_bool") =>
adamc@253 700 ((L'.EAbs ("x", (L'.TFfi ("Basis", "bool"), loc), (L'.TFfi ("Basis", "string"), loc),
adamc@253 701 (L'.EFfiApp ("Basis", "sqlifyBool", [(L'.ERel 0, loc)]), loc)), loc),
adamc@253 702 fm)
adamc@253 703 | L.EFfi ("Basis", "sql_string") =>
adamc@253 704 ((L'.EAbs ("x", (L'.TFfi ("Basis", "string"), loc), (L'.TFfi ("Basis", "string"), loc),
adamc@253 705 (L'.EFfiApp ("Basis", "sqlifyString", [(L'.ERel 0, loc)]), loc)), loc),
adamc@253 706 fm)
adamc@253 707
adamc@252 708 | L.ECApp ((L.EFfi ("Basis", "sql_subset"), _), _) =>
adamc@252 709 ((L'.ERecord [], loc), fm)
adamc@252 710 | L.ECApp ((L.EFfi ("Basis", "sql_subset_all"), _), _) =>
adamc@252 711 ((L'.ERecord [], loc), fm)
adamc@252 712
adamc@252 713 | L.ECApp ((L.ECApp ((L.EFfi ("Basis", "sql_order_by_Nil"), _), _), _), _) =>
adamc@252 714 ((L'.EPrim (Prim.String ""), loc), fm)
adamc@261 715 | L.ECApp (
adamc@261 716 (L.ECApp (
adamc@261 717 (L.ECApp (
adamc@261 718 (L.EFfi ("Basis", "sql_order_by_Cons"), _),
adamc@261 719 _), _),
adamc@261 720 _), _),
adamc@261 721 _) =>
adamc@261 722 let
adamc@261 723 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@261 724 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@261 725 in
adamc@261 726 ((L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@261 727 (L'.EAbs ("e2", s, s,
adamc@261 728 (L'.ECase ((L'.ERel 0, loc),
adamc@261 729 [((L'.PPrim (Prim.String ""), loc),
adamc@261 730 (L'.ERel 1, loc)),
adamc@261 731 ((L'.PWild, loc),
adamc@261 732 strcat loc [(L'.ERel 1, loc),
adamc@261 733 sc ", ",
adamc@261 734 (L'.ERel 0, loc),
adamc@261 735 sc ")"])],
adamc@261 736 {disc = s, result = s}), loc)), loc)), loc),
adamc@261 737 fm)
adamc@261 738 end
adamc@252 739
adamc@252 740 | L.EFfi ("Basis", "sql_no_limit") =>
adamc@252 741 ((L'.EPrim (Prim.String ""), loc), fm)
adamc@262 742 | L.EFfiApp ("Basis", "sql_limit", [e]) =>
adamc@262 743 let
adamc@262 744 val (e, fm) = monoExp (env, st, fm) e
adamc@262 745 in
adamc@262 746 (strcat loc [
adamc@262 747 (L'.EPrim (Prim.String " LIMIT "), loc),
adamc@262 748 (L'.EFfiApp ("Basis", "sqlifyInt", [e]), loc)
adamc@262 749 ],
adamc@262 750 fm)
adamc@262 751 end
adamc@262 752
adamc@252 753 | L.EFfi ("Basis", "sql_no_offset") =>
adamc@252 754 ((L'.EPrim (Prim.String ""), loc), fm)
adamc@263 755 | L.EFfiApp ("Basis", "sql_offset", [e]) =>
adamc@263 756 let
adamc@263 757 val (e, fm) = monoExp (env, st, fm) e
adamc@263 758 in
adamc@263 759 (strcat loc [
adamc@263 760 (L'.EPrim (Prim.String " OFFSET "), loc),
adamc@263 761 (L'.EFfiApp ("Basis", "sqlifyInt", [e]), loc)
adamc@263 762 ],
adamc@263 763 fm)
adamc@263 764 end
adamc@253 765
adamc@253 766 | L.EFfi ("Basis", "sql_eq") =>
adamc@253 767 ((L'.EPrim (Prim.String "="), loc), fm)
adamc@253 768 | L.EFfi ("Basis", "sql_ne") =>
adamc@253 769 ((L'.EPrim (Prim.String "<>"), loc), fm)
adamc@253 770 | L.EFfi ("Basis", "sql_lt") =>
adamc@253 771 ((L'.EPrim (Prim.String "<"), loc), fm)
adamc@253 772 | L.EFfi ("Basis", "sql_le") =>
adamc@253 773 ((L'.EPrim (Prim.String "<="), loc), fm)
adamc@253 774 | L.EFfi ("Basis", "sql_gt") =>
adamc@253 775 ((L'.EPrim (Prim.String ">"), loc), fm)
adamc@253 776 | L.EFfi ("Basis", "sql_ge") =>
adamc@253 777 ((L'.EPrim (Prim.String ">="), loc), fm)
adamc@253 778
adamc@253 779 | L.ECApp (
adamc@253 780 (L.ECApp (
adamc@253 781 (L.ECApp (
adamc@253 782 (L.ECApp (
adamc@254 783 (L.ECApp (
adamc@254 784 (L.ECApp (
adamc@254 785 (L.EFfi ("Basis", "sql_binary"), _),
adamc@254 786 _), _),
adamc@254 787 _), _),
adamc@254 788 _), _),
adamc@254 789 _), _),
adamc@254 790 _), _),
adamc@254 791 _) =>
adamc@254 792 let
adamc@254 793 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@254 794 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@254 795 in
adamc@254 796 ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@254 797 (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@254 798 (L'.EAbs ("e2", s, s,
adamc@254 799 strcat loc [sc "(",
adamc@254 800 (L'.ERel 1, loc),
adamc@254 801 sc " ",
adamc@254 802 (L'.ERel 2, loc),
adamc@254 803 sc " ",
adamc@254 804 (L'.ERel 0, loc),
adamc@254 805 sc ")"]), loc)), loc)), loc),
adamc@254 806 fm)
adamc@254 807 end
adamc@254 808 | L.EFfi ("Basis", "sql_and") => ((L'.EPrim (Prim.String "AND"), loc), fm)
adamc@254 809 | L.EFfi ("Basis", "sql_or") => ((L'.EPrim (Prim.String "OR"), loc), fm)
adamc@254 810
adamc@254 811 | L.ECApp (
adamc@254 812 (L.ECApp (
adamc@254 813 (L.ECApp (
adamc@254 814 (L.ECApp (
adamc@253 815 (L.EFfi ("Basis", "sql_comparison"), _),
adamc@253 816 _), _),
adamc@253 817 _), _),
adamc@253 818 _), _),
adamc@253 819 _) =>
adamc@253 820 let
adamc@253 821 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@253 822 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@253 823 in
adamc@253 824 ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@253 825 (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@253 826 (L'.EAbs ("e2", s, s,
adamc@254 827 strcat loc [sc "(",
adamc@254 828 (L'.ERel 1, loc),
adamc@253 829 sc " ",
adamc@253 830 (L'.ERel 2, loc),
adamc@253 831 sc " ",
adamc@254 832 (L'.ERel 0, loc),
adamc@254 833 sc ")"]), loc)), loc)), loc),
adamc@253 834 fm)
adamc@253 835 end
adamc@253 836
adamc@253 837 | L.ECApp (
adamc@253 838 (L.ECApp (
adamc@253 839 (L.ECApp (
adamc@253 840 (L.ECApp (
adamc@253 841 (L.ECApp (
adamc@253 842 (L.ECApp (
adamc@253 843 (L.ECApp (
adamc@253 844 (L.EFfi ("Basis", "sql_field"), _),
adamc@253 845 _), _),
adamc@253 846 _), _),
adamc@253 847 _), _),
adamc@253 848 _), _),
adamc@253 849 _), _),
adamc@253 850 (L.CName tab, _)), _),
adamc@253 851 (L.CName field, _)) => ((L'.EPrim (Prim.String (tab ^ "." ^ field)), loc), fm)
adamc@260 852
adamc@260 853 | L.ECApp (
adamc@260 854 (L.ECApp (
adamc@260 855 (L.ECApp (
adamc@260 856 (L.ECApp (
adamc@261 857 (L.ECApp (
adamc@261 858 (L.EFfi ("Basis", "sql_exp"), _),
adamc@261 859 _), _),
adamc@261 860 _), _),
adamc@261 861 _), _),
adamc@261 862 _), _),
adamc@261 863 (L.CName nm, _)) => ((L'.EPrim (Prim.String ("_" ^ nm)), loc), fm)
adamc@261 864
adamc@261 865 | L.ECApp (
adamc@261 866 (L.ECApp (
adamc@261 867 (L.ECApp (
adamc@261 868 (L.ECApp (
adamc@260 869 (L.EFfi ("Basis", "sql_relop"), _),
adamc@260 870 _), _),
adamc@260 871 _), _),
adamc@260 872 _), _),
adamc@260 873 _) =>
adamc@260 874 let
adamc@260 875 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@260 876 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@260 877 in
adamc@260 878 ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@260 879 (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@260 880 (L'.EAbs ("e2", s, s,
adamc@260 881 strcat loc [sc "((",
adamc@260 882 (L'.ERel 1, loc),
adamc@260 883 sc ") ",
adamc@260 884 (L'.ERel 2, loc),
adamc@260 885 sc " (",
adamc@260 886 (L'.ERel 0, loc),
adamc@260 887 sc "))"]), loc)), loc)), loc),
adamc@260 888 fm)
adamc@260 889 end
adamc@260 890
adamc@260 891 | L.EFfi ("Basis", "sql_union") => ((L'.EPrim (Prim.String "UNION"), loc), fm)
adamc@260 892 | L.EFfi ("Basis", "sql_intersect") => ((L'.EPrim (Prim.String "INTERSECT"), loc), fm)
adamc@260 893 | L.EFfi ("Basis", "sql_except") => ((L'.EPrim (Prim.String "EXCEPT"), loc), fm)
adamc@260 894
adamc@139 895 | L.EApp (
adamc@139 896 (L.ECApp (
adamc@141 897 (L.ECApp ((L.EFfi ("Basis", "cdata"), _), _), _),
adamc@139 898 _), _),
adamc@179 899 se) =>
adamc@179 900 let
adamc@179 901 val (se, fm) = monoExp (env, st, fm) se
adamc@179 902 in
adamc@179 903 ((L'.EFfiApp ("Basis", "htmlifyString", [se]), loc), fm)
adamc@179 904 end
adamc@179 905
adamc@95 906 | L.EApp (
adamc@95 907 (L.EApp (
adamc@95 908 (L.ECApp (
adamc@95 909 (L.ECApp (
adamc@95 910 (L.ECApp (
adamc@139 911 (L.ECApp (
adamc@140 912 (L.EFfi ("Basis", "join"),
adamc@139 913 _), _), _),
adamc@139 914 _), _),
adamc@95 915 _), _),
adamc@95 916 _), _),
adamc@95 917 xml1), _),
adamc@179 918 xml2) =>
adamc@179 919 let
adamc@179 920 val (xml1, fm) = monoExp (env, st, fm) xml1
adamc@179 921 val (xml2, fm) = monoExp (env, st, fm) xml2
adamc@179 922 in
adamc@179 923 ((L'.EStrcat (xml1, xml2), loc), fm)
adamc@179 924 end
adamc@95 925
adamc@95 926 | L.EApp (
adamc@95 927 (L.EApp (
adamc@104 928 (L.EApp (
adamc@95 929 (L.ECApp (
adamc@104 930 (L.ECApp (
adamc@104 931 (L.ECApp (
adamc@104 932 (L.ECApp (
adamc@139 933 (L.ECApp (
adamc@139 934 (L.ECApp (
adamc@139 935 (L.ECApp (
adamc@139 936 (L.ECApp (
adamc@139 937 (L.EFfi ("Basis", "tag"),
adamc@139 938 _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _),
adamc@104 939 attrs), _),
adamc@95 940 tag), _),
adamc@95 941 xml) =>
adamc@95 942 let
adamc@140 943 fun getTag' (e, _) =
adamc@140 944 case e of
adamc@143 945 L.EFfi ("Basis", tag) => (tag, [])
adamc@143 946 | L.ECApp (e, t) => let
adamc@143 947 val (tag, ts) = getTag' e
adamc@143 948 in
adamc@143 949 (tag, ts @ [t])
adamc@143 950 end
adamc@140 951 | _ => (E.errorAt loc "Non-constant XML tag";
adamc@140 952 Print.eprefaces' [("Expression", CorePrint.p_exp env tag)];
adamc@143 953 ("", []))
adamc@140 954
adamc@95 955 fun getTag (e, _) =
adamc@95 956 case e of
adamc@143 957 L.EFfiApp ("Basis", tag, [(L.ERecord [], _)]) => (tag, [])
adamc@140 958 | L.EApp (e, (L.ERecord [], _)) => getTag' e
adamc@95 959 | _ => (E.errorAt loc "Non-constant XML tag";
adamc@95 960 Print.eprefaces' [("Expression", CorePrint.p_exp env tag)];
adamc@143 961 ("", []))
adamc@95 962
adamc@143 963 val (tag, targs) = getTag tag
adamc@95 964
adamc@179 965 val (attrs, fm) = monoExp (env, st, fm) attrs
adamc@104 966
adamc@143 967 fun tagStart tag =
adamc@104 968 case #1 attrs of
adamc@104 969 L'.ERecord xes =>
adamc@104 970 let
adamc@104 971 fun lowercaseFirst "" = ""
adamc@143 972 | lowercaseFirst s = str (Char.toLower (String.sub (s, 0)))
adamc@143 973 ^ String.extract (s, 1, NONE)
adamc@104 974
adamc@104 975 val s = (L'.EPrim (Prim.String (String.concat ["<", tag])), loc)
adamc@104 976 in
adamc@179 977 foldl (fn ((x, e, t), (s, fm)) =>
adamc@104 978 let
adamc@104 979 val xp = " " ^ lowercaseFirst x ^ "=\""
adamc@120 980
adamc@120 981 val fooify =
adamc@120 982 case x of
adamc@185 983 "Href" => urlifyExp
adamc@185 984 | "Link" => urlifyExp
adamc@143 985 | "Action" => urlifyExp
adamc@120 986 | _ => attrifyExp
adamc@179 987
adamc@179 988 val (e, fm) = fooify env fm (e, t)
adamc@104 989 in
adamc@179 990 ((L'.EStrcat (s,
adamc@179 991 (L'.EStrcat ((L'.EPrim (Prim.String xp), loc),
adamc@179 992 (L'.EStrcat (e,
adamc@179 993 (L'.EPrim (Prim.String "\""),
adamc@179 994 loc)),
adamc@179 995 loc)),
adamc@179 996 loc)), loc),
adamc@179 997 fm)
adamc@104 998 end)
adamc@179 999 (s, fm) xes
adamc@104 1000 end
adamc@143 1001 | _ => raise Fail "Non-record attributes!"
adamc@104 1002
adamc@143 1003 fun input typ =
adamc@143 1004 case targs of
adamc@155 1005 [_, (L.CName name, _)] =>
adamc@179 1006 let
adamc@179 1007 val (ts, fm) = tagStart "input"
adamc@179 1008 in
adamc@179 1009 ((L'.EStrcat (ts,
adamc@179 1010 (L'.EPrim (Prim.String (" type=\"" ^ typ ^ "\" name=\"" ^ name ^ "\"/>")),
adamc@179 1011 loc)), loc), fm)
adamc@179 1012 end
adamc@143 1013 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153 1014 raise Fail "No name passed to input tag")
adamc@104 1015
adamc@152 1016 fun normal (tag, extra) =
adamc@143 1017 let
adamc@179 1018 val (tagStart, fm) = tagStart tag
adamc@152 1019 val tagStart = case extra of
adamc@152 1020 NONE => tagStart
adamc@152 1021 | SOME extra => (L'.EStrcat (tagStart, extra), loc)
adamc@152 1022
adamc@143 1023 fun normal () =
adamc@179 1024 let
adamc@179 1025 val (xml, fm) = monoExp (env, st, fm) xml
adamc@179 1026 in
adamc@179 1027 ((L'.EStrcat ((L'.EStrcat (tagStart, (L'.EPrim (Prim.String ">"), loc)), loc),
adamc@179 1028 (L'.EStrcat (xml,
adamc@179 1029 (L'.EPrim (Prim.String (String.concat ["</", tag, ">"])),
adamc@179 1030 loc)), loc)),
adamc@179 1031 loc),
adamc@179 1032 fm)
adamc@179 1033 end
adamc@143 1034 in
adamc@143 1035 case xml of
adamc@143 1036 (L.EApp ((L.ECApp (
adamc@143 1037 (L.ECApp ((L.EFfi ("Basis", "cdata"), _),
adamc@143 1038 _), _),
adamc@143 1039 _), _),
adamc@143 1040 (L.EPrim (Prim.String s), _)), _) =>
adamc@143 1041 if CharVector.all Char.isSpace s then
adamc@179 1042 ((L'.EStrcat (tagStart, (L'.EPrim (Prim.String "/>"), loc)), loc), fm)
adamc@143 1043 else
adamc@143 1044 normal ()
adamc@143 1045 | _ => normal ()
adamc@143 1046 end
adamc@152 1047 in
adamc@152 1048 case tag of
adamc@179 1049 "submit" => ((L'.EPrim (Prim.String "<input type=\"submit\"/>"), loc), fm)
adamc@152 1050
adamc@152 1051 | "textbox" =>
adamc@152 1052 (case targs of
adamc@152 1053 [_, (L.CName name, _)] =>
adamc@179 1054 let
adamc@179 1055 val (ts, fm) = tagStart "input"
adamc@179 1056 in
adamc@179 1057 ((L'.EStrcat (ts,
adamc@179 1058 (L'.EPrim (Prim.String (" name=\"" ^ name ^ "\"/>")),
adamc@179 1059 loc)), loc), fm)
adamc@179 1060 end
adamc@152 1061 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153 1062 raise Fail "No name passed to textarea tag"))
adamc@155 1063 | "password" => input "password"
adamc@152 1064 | "ltextarea" =>
adamc@152 1065 (case targs of
adamc@152 1066 [_, (L.CName name, _)] =>
adamc@179 1067 let
adamc@179 1068 val (ts, fm) = tagStart "textarea"
adamc@179 1069 val (xml, fm) = monoExp (env, st, fm) xml
adamc@179 1070 in
adamc@179 1071 ((L'.EStrcat ((L'.EStrcat (ts,
adamc@179 1072 (L'.EPrim (Prim.String (" name=\"" ^ name ^ "\">")), loc)), loc),
adamc@179 1073 (L'.EStrcat (xml,
adamc@179 1074 (L'.EPrim (Prim.String "</textarea>"),
adamc@179 1075 loc)), loc)),
adamc@179 1076 loc), fm)
adamc@179 1077 end
adamc@152 1078 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153 1079 raise Fail "No name passed to ltextarea tag"))
adamc@153 1080
adamc@190 1081 | "checkbox" => input "checkbox"
adamc@190 1082
adamc@153 1083 | "radio" =>
adamc@153 1084 (case targs of
adamc@153 1085 [_, (L.CName name, _)] =>
adamc@179 1086 monoExp (env, St.setRadioGroup (st, name), fm) xml
adamc@153 1087 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153 1088 raise Fail "No name passed to radio tag"))
adamc@153 1089 | "radioOption" =>
adamc@153 1090 (case St.radioGroup st of
adamc@153 1091 NONE => raise Fail "No name for radioGroup"
adamc@153 1092 | SOME name =>
adamc@153 1093 normal ("input",
adamc@153 1094 SOME (L'.EPrim (Prim.String (" type=\"radio\" name=\"" ^ name ^ "\"")), loc)))
adamc@152 1095
adamc@154 1096 | "lselect" =>
adamc@154 1097 (case targs of
adamc@154 1098 [_, (L.CName name, _)] =>
adamc@179 1099 let
adamc@179 1100 val (ts, fm) = tagStart "select"
adamc@179 1101 val (xml, fm) = monoExp (env, st, fm) xml
adamc@179 1102 in
adamc@179 1103 ((L'.EStrcat ((L'.EStrcat (ts,
adamc@179 1104 (L'.EPrim (Prim.String (" name=\"" ^ name ^ "\">")), loc)), loc),
adamc@179 1105 (L'.EStrcat (xml,
adamc@179 1106 (L'.EPrim (Prim.String "</select>"),
adamc@179 1107 loc)), loc)),
adamc@179 1108 loc),
adamc@179 1109 fm)
adamc@179 1110 end
adamc@154 1111 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@154 1112 raise Fail "No name passed to lselect tag"))
adamc@154 1113
adamc@154 1114 | "loption" => normal ("option", NONE)
adamc@154 1115
adamc@152 1116 | _ => normal (tag, NONE)
adamc@95 1117 end
adamc@94 1118
adamc@141 1119 | L.EApp ((L.ECApp (
adamc@141 1120 (L.ECApp ((L.EFfi ("Basis", "lform"), _), _), _),
adamc@141 1121 _), _),
adamc@141 1122 xml) =>
adamc@143 1123 let
adamc@143 1124 fun findSubmit (e, _) =
adamc@143 1125 case e of
adamc@143 1126 L.EApp (
adamc@143 1127 (L.EApp (
adamc@143 1128 (L.ECApp (
adamc@143 1129 (L.ECApp (
adamc@143 1130 (L.ECApp (
adamc@143 1131 (L.ECApp (
adamc@143 1132 (L.EFfi ("Basis", "join"),
adamc@143 1133 _), _), _),
adamc@143 1134 _), _),
adamc@143 1135 _), _),
adamc@143 1136 _), _),
adamc@143 1137 xml1), _),
adamc@143 1138 xml2) => (case findSubmit xml1 of
adamc@143 1139 Error => Error
adamc@143 1140 | NotFound => findSubmit xml2
adamc@143 1141 | Found e =>
adamc@143 1142 case findSubmit xml2 of
adamc@143 1143 NotFound => Found e
adamc@143 1144 | _ => Error)
adamc@143 1145 | L.EApp (
adamc@143 1146 (L.EApp (
adamc@143 1147 (L.EApp (
adamc@143 1148 (L.ECApp (
adamc@143 1149 (L.ECApp (
adamc@143 1150 (L.ECApp (
adamc@143 1151 (L.ECApp (
adamc@143 1152 (L.ECApp (
adamc@143 1153 (L.ECApp (
adamc@143 1154 (L.ECApp (
adamc@143 1155 (L.ECApp (
adamc@143 1156 (L.EFfi ("Basis", "tag"),
adamc@143 1157 _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _),
adamc@143 1158 attrs), _),
adamc@143 1159 _), _),
adamc@143 1160 xml) =>
adamc@143 1161 (case #1 attrs of
adamc@143 1162 L.ERecord xes =>
adamc@143 1163 (case ListUtil.search (fn ((L.CName "Action", _), e, t) => SOME (e, t)
adamc@143 1164 | _ => NONE) xes of
adamc@143 1165 NONE => findSubmit xml
adamc@143 1166 | SOME et =>
adamc@143 1167 case findSubmit xml of
adamc@143 1168 NotFound => Found et
adamc@143 1169 | _ => Error)
adamc@143 1170 | _ => findSubmit xml)
adamc@143 1171 | _ => NotFound
adamc@143 1172
adamc@143 1173 val (action, actionT) = case findSubmit xml of
adamc@143 1174 NotFound => raise Fail "No submit found"
adamc@143 1175 | Error => raise Fail "Not ready for multi-submit lforms yet"
adamc@143 1176 | Found et => et
adamc@143 1177
adamc@143 1178 val actionT = monoType env actionT
adamc@179 1179 val (action, fm) = monoExp (env, st, fm) action
adamc@179 1180 val (action, fm) = urlifyExp env fm (action, actionT)
adamc@179 1181 val (xml, fm) = monoExp (env, st, fm) xml
adamc@143 1182 in
adamc@179 1183 ((L'.EStrcat ((L'.EStrcat ((L'.EPrim (Prim.String "<form action=\""), loc),
adamc@179 1184 (L'.EStrcat (action,
adamc@179 1185 (L'.EPrim (Prim.String "\">"), loc)), loc)), loc),
adamc@179 1186 (L'.EStrcat (xml,
adamc@179 1187 (L'.EPrim (Prim.String "</form>"), loc)), loc)), loc),
adamc@179 1188 fm)
adamc@143 1189 end
adamc@141 1190
adamc@148 1191 | L.EApp ((L.ECApp (
adamc@148 1192 (L.ECApp (
adamc@148 1193 (L.ECApp (
adamc@148 1194 (L.ECApp (
adamc@148 1195 (L.EFfi ("Basis", "useMore"), _), _), _),
adamc@148 1196 _), _),
adamc@148 1197 _), _),
adamc@148 1198 _), _),
adamc@179 1199 xml) => monoExp (env, st, fm) xml
adamc@148 1200
adamc@179 1201 | L.EApp (e1, e2) =>
adamc@179 1202 let
adamc@179 1203 val (e1, fm) = monoExp (env, st, fm) e1
adamc@179 1204 val (e2, fm) = monoExp (env, st, fm) e2
adamc@179 1205 in
adamc@179 1206 ((L'.EApp (e1, e2), loc), fm)
adamc@179 1207 end
adamc@26 1208 | L.EAbs (x, dom, ran, e) =>
adamc@179 1209 let
adamc@179 1210 val (e, fm) = monoExp (Env.pushERel env x dom, st, fm) e
adamc@179 1211 in
adamc@179 1212 ((L'.EAbs (x, monoType env dom, monoType env ran, e), loc), fm)
adamc@179 1213 end
adamc@25 1214 | L.ECApp _ => poly ()
adamc@25 1215 | L.ECAbs _ => poly ()
adamc@25 1216
adamc@252 1217 | L.EFfi mx => ((L'.EFfi mx, loc), fm)
adamc@252 1218 | L.EFfiApp (m, x, es) =>
adamc@252 1219 let
adamc@252 1220 val (es, fm) = ListUtil.foldlMap (fn (e, fm) => monoExp (env, st, fm) e) fm es
adamc@252 1221 in
adamc@252 1222 ((L'.EFfiApp (m, x, es), loc), fm)
adamc@252 1223 end
adamc@252 1224
adamc@179 1225 | L.ERecord xes =>
adamc@179 1226 let
adamc@179 1227 val (xes, fm) = ListUtil.foldlMap
adamc@179 1228 (fn ((x, e, t), fm) =>
adamc@179 1229 let
adamc@179 1230 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1231 in
adamc@179 1232 ((monoName env x,
adamc@179 1233 e,
adamc@179 1234 monoType env t), fm)
adamc@179 1235 end) fm xes
adamc@179 1236 in
adamc@179 1237 ((L'.ERecord xes, loc), fm)
adamc@179 1238 end
adamc@179 1239 | L.EField (e, x, _) =>
adamc@179 1240 let
adamc@179 1241 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1242 in
adamc@179 1243 ((L'.EField (e, monoName env x), loc), fm)
adamc@179 1244 end
adamc@149 1245 | L.ECut _ => poly ()
adamc@73 1246 | L.EFold _ => poly ()
adamc@177 1247
adamc@182 1248 | L.ECase (e, pes, {disc, result}) =>
adamc@179 1249 let
adamc@179 1250 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1251 val (pes, fm) = ListUtil.foldlMap
adamc@179 1252 (fn ((p, e), fm) =>
adamc@179 1253 let
adamc@179 1254 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1255 in
adamc@182 1256 ((monoPat env p, e), fm)
adamc@179 1257 end) fm pes
adamc@179 1258 in
adamc@182 1259 ((L'.ECase (e, pes, {disc = monoType env disc, result = monoType env result}), loc), fm)
adamc@179 1260 end
adamc@177 1261
adamc@179 1262 | L.EWrite e =>
adamc@179 1263 let
adamc@179 1264 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1265 in
adamc@252 1266 ((L'.EAbs ("_", (L'.TRecord [], loc), (L'.TRecord [], loc),
adamc@252 1267 (L'.EWrite (liftExpInExp 0 e), loc)), loc), fm)
adamc@179 1268 end
adamc@110 1269
adamc@179 1270 | L.EClosure (n, es) =>
adamc@179 1271 let
adamc@179 1272 val (es, fm) = ListUtil.foldlMap (fn (e, fm) =>
adamc@179 1273 monoExp (env, st, fm) e)
adamc@179 1274 fm es
adamc@179 1275 in
adamc@179 1276 ((L'.EClosure (n, es), loc), fm)
adamc@179 1277 end
adamc@25 1278 end
adamc@25 1279
adamc@179 1280 fun monoDecl (env, fm) (all as (d, loc)) =
adamc@25 1281 let
adamc@25 1282 fun poly () =
adamc@25 1283 (E.errorAt loc "Unsupported declaration";
adamc@25 1284 Print.eprefaces' [("Declaration", CorePrint.p_decl env all)];
adamc@25 1285 NONE)
adamc@25 1286 in
adamc@25 1287 case d of
adamc@25 1288 L.DCon _ => NONE
adamc@193 1289 | L.DDatatype (x, n, [], xncs) =>
adamc@193 1290 let
adamc@196 1291 val env' = Env.declBinds env all
adamc@196 1292 val d = (L'.DDatatype (x, n, map (fn (x, n, to) => (x, n, Option.map (monoType env') to)) xncs), loc)
adamc@164 1293 in
adamc@196 1294 SOME (env', fm, d)
adamc@193 1295 end
adamc@193 1296 | L.DDatatype _ => poly ()
adamc@179 1297 | L.DVal (x, n, t, e, s) =>
adamc@179 1298 let
adamc@179 1299 val (e, fm) = monoExp (env, St.empty, fm) e
adamc@179 1300 in
adamc@179 1301 SOME (Env.pushENamed env x n t NONE s,
adamc@179 1302 fm,
adamc@179 1303 (L'.DVal (x, n, monoType env t, e, s), loc))
adamc@179 1304 end
adamc@128 1305 | L.DValRec vis =>
adamc@128 1306 let
adamc@128 1307 val env = foldl (fn ((x, n, t, e, s), env) => Env.pushENamed env x n t NONE s) env vis
adamc@179 1308
adamc@179 1309 val (vis, fm) = ListUtil.foldlMap
adamc@179 1310 (fn ((x, n, t, e, s), fm) =>
adamc@179 1311 let
adamc@179 1312 val (e, fm) = monoExp (env, St.empty, fm) e
adamc@179 1313 in
adamc@179 1314 ((x, n, monoType env t, e, s), fm)
adamc@179 1315 end)
adamc@179 1316 fm vis
adamc@128 1317 in
adamc@128 1318 SOME (env,
adamc@179 1319 fm,
adamc@179 1320 (L'.DValRec vis, loc))
adamc@128 1321 end
adamc@144 1322 | L.DExport (ek, n) =>
adamc@115 1323 let
adamc@120 1324 val (_, t, _, s) = Env.lookupENamed env n
adamc@120 1325
adamc@120 1326 fun unwind (t, _) =
adamc@120 1327 case t of
adamc@120 1328 L.TFun (dom, ran) => dom :: unwind ran
adamc@120 1329 | _ => []
adamc@120 1330
adamc@120 1331 val ts = map (monoType env) (unwind t)
adamc@115 1332 in
adamc@179 1333 SOME (env, fm, (L'.DExport (ek, s, n, ts), loc))
adamc@115 1334 end
adamc@251 1335 | L.DTable (x, n, _, s) =>
adamc@251 1336 let
adamc@251 1337 val t = (L.CFfi ("Basis", "string"), loc)
adamc@251 1338 val t' = (L'.TFfi ("Basis", "string"), loc)
adamc@251 1339 val e = (L'.EPrim (Prim.String s), loc)
adamc@251 1340 in
adamc@251 1341 SOME (Env.pushENamed env x n t NONE s,
adamc@251 1342 fm,
adamc@251 1343 (L'.DVal (x, n, t', e, s), loc))
adamc@251 1344 end
adamc@25 1345 end
adamc@25 1346
adamc@25 1347 fun monoize env ds =
adamc@25 1348 let
adamc@179 1349 val (_, _, ds) = List.foldl (fn (d, (env, fm, ds)) =>
adamc@179 1350 case monoDecl (env, fm) d of
adamc@179 1351 NONE => (env, fm, ds)
adamc@179 1352 | SOME (env, fm, d) =>
adamc@179 1353 (env,
adamc@179 1354 Fm.enter fm,
adamc@179 1355 d :: Fm.decls fm @ ds))
adamc@179 1356 (env, Fm.empty (CoreUtil.File.maxName ds + 1), []) ds
adamc@25 1357 in
adamc@25 1358 rev ds
adamc@25 1359 end
adamc@25 1360
adamc@25 1361 end