Mercurial > urweb

annotate src/monoize.sml @ 264:35ed669a253f

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Monoize unops
author Adam Chlipala <adamc@hcoop.net>
date Sun, 31 Aug 2008 16:13:07 -0400
parents 02e1870a0516
children b66e692422a4
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@264 784 (L.EFfi ("Basis", "sql_unary"), _),
adamc@264 785 _), _),
adamc@264 786 _), _),
adamc@264 787 _), _),
adamc@264 788 _), _),
adamc@264 789 _) =>
adamc@264 790 let
adamc@264 791 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@264 792 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@264 793 in
adamc@264 794 ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@264 795 (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@264 796 strcat loc [sc "(",
adamc@264 797 (L'.ERel 1, loc),
adamc@264 798 sc " ",
adamc@264 799 (L'.ERel 0, loc),
adamc@264 800 sc ")"]), loc)), loc),
adamc@264 801 fm)
adamc@264 802 end
adamc@264 803 | L.EFfi ("Basis", "sql_not") => ((L'.EPrim (Prim.String "NOT"), loc), fm)
adamc@264 804
adamc@264 805 | L.ECApp (
adamc@264 806 (L.ECApp (
adamc@264 807 (L.ECApp (
adamc@264 808 (L.ECApp (
adamc@264 809 (L.ECApp (
adamc@254 810 (L.ECApp (
adamc@254 811 (L.EFfi ("Basis", "sql_binary"), _),
adamc@254 812 _), _),
adamc@254 813 _), _),
adamc@254 814 _), _),
adamc@254 815 _), _),
adamc@254 816 _), _),
adamc@254 817 _) =>
adamc@254 818 let
adamc@254 819 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@254 820 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@254 821 in
adamc@254 822 ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@254 823 (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@254 824 (L'.EAbs ("e2", s, s,
adamc@254 825 strcat loc [sc "(",
adamc@254 826 (L'.ERel 1, loc),
adamc@254 827 sc " ",
adamc@254 828 (L'.ERel 2, loc),
adamc@254 829 sc " ",
adamc@254 830 (L'.ERel 0, loc),
adamc@254 831 sc ")"]), loc)), loc)), loc),
adamc@254 832 fm)
adamc@254 833 end
adamc@254 834 | L.EFfi ("Basis", "sql_and") => ((L'.EPrim (Prim.String "AND"), loc), fm)
adamc@254 835 | L.EFfi ("Basis", "sql_or") => ((L'.EPrim (Prim.String "OR"), loc), fm)
adamc@254 836
adamc@254 837 | L.ECApp (
adamc@254 838 (L.ECApp (
adamc@254 839 (L.ECApp (
adamc@254 840 (L.ECApp (
adamc@253 841 (L.EFfi ("Basis", "sql_comparison"), _),
adamc@253 842 _), _),
adamc@253 843 _), _),
adamc@253 844 _), _),
adamc@253 845 _) =>
adamc@253 846 let
adamc@253 847 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@253 848 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@253 849 in
adamc@253 850 ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@253 851 (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@253 852 (L'.EAbs ("e2", s, s,
adamc@254 853 strcat loc [sc "(",
adamc@254 854 (L'.ERel 1, loc),
adamc@253 855 sc " ",
adamc@253 856 (L'.ERel 2, loc),
adamc@253 857 sc " ",
adamc@254 858 (L'.ERel 0, loc),
adamc@254 859 sc ")"]), loc)), loc)), loc),
adamc@253 860 fm)
adamc@253 861 end
adamc@253 862
adamc@253 863 | L.ECApp (
adamc@253 864 (L.ECApp (
adamc@253 865 (L.ECApp (
adamc@253 866 (L.ECApp (
adamc@253 867 (L.ECApp (
adamc@253 868 (L.ECApp (
adamc@253 869 (L.ECApp (
adamc@253 870 (L.EFfi ("Basis", "sql_field"), _),
adamc@253 871 _), _),
adamc@253 872 _), _),
adamc@253 873 _), _),
adamc@253 874 _), _),
adamc@253 875 _), _),
adamc@253 876 (L.CName tab, _)), _),
adamc@253 877 (L.CName field, _)) => ((L'.EPrim (Prim.String (tab ^ "." ^ field)), loc), fm)
adamc@260 878
adamc@260 879 | L.ECApp (
adamc@260 880 (L.ECApp (
adamc@260 881 (L.ECApp (
adamc@260 882 (L.ECApp (
adamc@261 883 (L.ECApp (
adamc@261 884 (L.EFfi ("Basis", "sql_exp"), _),
adamc@261 885 _), _),
adamc@261 886 _), _),
adamc@261 887 _), _),
adamc@261 888 _), _),
adamc@261 889 (L.CName nm, _)) => ((L'.EPrim (Prim.String ("_" ^ nm)), loc), fm)
adamc@261 890
adamc@261 891 | L.ECApp (
adamc@261 892 (L.ECApp (
adamc@261 893 (L.ECApp (
adamc@261 894 (L.ECApp (
adamc@260 895 (L.EFfi ("Basis", "sql_relop"), _),
adamc@260 896 _), _),
adamc@260 897 _), _),
adamc@260 898 _), _),
adamc@260 899 _) =>
adamc@260 900 let
adamc@260 901 val s = (L'.TFfi ("Basis", "string"), loc)
adamc@260 902 fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@260 903 in
adamc@260 904 ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@260 905 (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@260 906 (L'.EAbs ("e2", s, s,
adamc@260 907 strcat loc [sc "((",
adamc@260 908 (L'.ERel 1, loc),
adamc@260 909 sc ") ",
adamc@260 910 (L'.ERel 2, loc),
adamc@260 911 sc " (",
adamc@260 912 (L'.ERel 0, loc),
adamc@260 913 sc "))"]), loc)), loc)), loc),
adamc@260 914 fm)
adamc@260 915 end
adamc@260 916
adamc@260 917 | L.EFfi ("Basis", "sql_union") => ((L'.EPrim (Prim.String "UNION"), loc), fm)
adamc@260 918 | L.EFfi ("Basis", "sql_intersect") => ((L'.EPrim (Prim.String "INTERSECT"), loc), fm)
adamc@260 919 | L.EFfi ("Basis", "sql_except") => ((L'.EPrim (Prim.String "EXCEPT"), loc), fm)
adamc@260 920
adamc@139 921 | L.EApp (
adamc@139 922 (L.ECApp (
adamc@141 923 (L.ECApp ((L.EFfi ("Basis", "cdata"), _), _), _),
adamc@139 924 _), _),
adamc@179 925 se) =>
adamc@179 926 let
adamc@179 927 val (se, fm) = monoExp (env, st, fm) se
adamc@179 928 in
adamc@179 929 ((L'.EFfiApp ("Basis", "htmlifyString", [se]), loc), fm)
adamc@179 930 end
adamc@179 931
adamc@95 932 | L.EApp (
adamc@95 933 (L.EApp (
adamc@95 934 (L.ECApp (
adamc@95 935 (L.ECApp (
adamc@95 936 (L.ECApp (
adamc@139 937 (L.ECApp (
adamc@140 938 (L.EFfi ("Basis", "join"),
adamc@139 939 _), _), _),
adamc@139 940 _), _),
adamc@95 941 _), _),
adamc@95 942 _), _),
adamc@95 943 xml1), _),
adamc@179 944 xml2) =>
adamc@179 945 let
adamc@179 946 val (xml1, fm) = monoExp (env, st, fm) xml1
adamc@179 947 val (xml2, fm) = monoExp (env, st, fm) xml2
adamc@179 948 in
adamc@179 949 ((L'.EStrcat (xml1, xml2), loc), fm)
adamc@179 950 end
adamc@95 951
adamc@95 952 | L.EApp (
adamc@95 953 (L.EApp (
adamc@104 954 (L.EApp (
adamc@95 955 (L.ECApp (
adamc@104 956 (L.ECApp (
adamc@104 957 (L.ECApp (
adamc@104 958 (L.ECApp (
adamc@139 959 (L.ECApp (
adamc@139 960 (L.ECApp (
adamc@139 961 (L.ECApp (
adamc@139 962 (L.ECApp (
adamc@139 963 (L.EFfi ("Basis", "tag"),
adamc@139 964 _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _),
adamc@104 965 attrs), _),
adamc@95 966 tag), _),
adamc@95 967 xml) =>
adamc@95 968 let
adamc@140 969 fun getTag' (e, _) =
adamc@140 970 case e of
adamc@143 971 L.EFfi ("Basis", tag) => (tag, [])
adamc@143 972 | L.ECApp (e, t) => let
adamc@143 973 val (tag, ts) = getTag' e
adamc@143 974 in
adamc@143 975 (tag, ts @ [t])
adamc@143 976 end
adamc@140 977 | _ => (E.errorAt loc "Non-constant XML tag";
adamc@140 978 Print.eprefaces' [("Expression", CorePrint.p_exp env tag)];
adamc@143 979 ("", []))
adamc@140 980
adamc@95 981 fun getTag (e, _) =
adamc@95 982 case e of
adamc@143 983 L.EFfiApp ("Basis", tag, [(L.ERecord [], _)]) => (tag, [])
adamc@140 984 | L.EApp (e, (L.ERecord [], _)) => getTag' e
adamc@95 985 | _ => (E.errorAt loc "Non-constant XML tag";
adamc@95 986 Print.eprefaces' [("Expression", CorePrint.p_exp env tag)];
adamc@143 987 ("", []))
adamc@95 988
adamc@143 989 val (tag, targs) = getTag tag
adamc@95 990
adamc@179 991 val (attrs, fm) = monoExp (env, st, fm) attrs
adamc@104 992
adamc@143 993 fun tagStart tag =
adamc@104 994 case #1 attrs of
adamc@104 995 L'.ERecord xes =>
adamc@104 996 let
adamc@104 997 fun lowercaseFirst "" = ""
adamc@143 998 | lowercaseFirst s = str (Char.toLower (String.sub (s, 0)))
adamc@143 999 ^ String.extract (s, 1, NONE)
adamc@104 1000
adamc@104 1001 val s = (L'.EPrim (Prim.String (String.concat ["<", tag])), loc)
adamc@104 1002 in
adamc@179 1003 foldl (fn ((x, e, t), (s, fm)) =>
adamc@104 1004 let
adamc@104 1005 val xp = " " ^ lowercaseFirst x ^ "=\""
adamc@120 1006
adamc@120 1007 val fooify =
adamc@120 1008 case x of
adamc@185 1009 "Href" => urlifyExp
adamc@185 1010 | "Link" => urlifyExp
adamc@143 1011 | "Action" => urlifyExp
adamc@120 1012 | _ => attrifyExp
adamc@179 1013
adamc@179 1014 val (e, fm) = fooify env fm (e, t)
adamc@104 1015 in
adamc@179 1016 ((L'.EStrcat (s,
adamc@179 1017 (L'.EStrcat ((L'.EPrim (Prim.String xp), loc),
adamc@179 1018 (L'.EStrcat (e,
adamc@179 1019 (L'.EPrim (Prim.String "\""),
adamc@179 1020 loc)),
adamc@179 1021 loc)),
adamc@179 1022 loc)), loc),
adamc@179 1023 fm)
adamc@104 1024 end)
adamc@179 1025 (s, fm) xes
adamc@104 1026 end
adamc@143 1027 | _ => raise Fail "Non-record attributes!"
adamc@104 1028
adamc@143 1029 fun input typ =
adamc@143 1030 case targs of
adamc@155 1031 [_, (L.CName name, _)] =>
adamc@179 1032 let
adamc@179 1033 val (ts, fm) = tagStart "input"
adamc@179 1034 in
adamc@179 1035 ((L'.EStrcat (ts,
adamc@179 1036 (L'.EPrim (Prim.String (" type=\"" ^ typ ^ "\" name=\"" ^ name ^ "\"/>")),
adamc@179 1037 loc)), loc), fm)
adamc@179 1038 end
adamc@143 1039 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153 1040 raise Fail "No name passed to input tag")
adamc@104 1041
adamc@152 1042 fun normal (tag, extra) =
adamc@143 1043 let
adamc@179 1044 val (tagStart, fm) = tagStart tag
adamc@152 1045 val tagStart = case extra of
adamc@152 1046 NONE => tagStart
adamc@152 1047 | SOME extra => (L'.EStrcat (tagStart, extra), loc)
adamc@152 1048
adamc@143 1049 fun normal () =
adamc@179 1050 let
adamc@179 1051 val (xml, fm) = monoExp (env, st, fm) xml
adamc@179 1052 in
adamc@179 1053 ((L'.EStrcat ((L'.EStrcat (tagStart, (L'.EPrim (Prim.String ">"), loc)), loc),
adamc@179 1054 (L'.EStrcat (xml,
adamc@179 1055 (L'.EPrim (Prim.String (String.concat ["</", tag, ">"])),
adamc@179 1056 loc)), loc)),
adamc@179 1057 loc),
adamc@179 1058 fm)
adamc@179 1059 end
adamc@143 1060 in
adamc@143 1061 case xml of
adamc@143 1062 (L.EApp ((L.ECApp (
adamc@143 1063 (L.ECApp ((L.EFfi ("Basis", "cdata"), _),
adamc@143 1064 _), _),
adamc@143 1065 _), _),
adamc@143 1066 (L.EPrim (Prim.String s), _)), _) =>
adamc@143 1067 if CharVector.all Char.isSpace s then
adamc@179 1068 ((L'.EStrcat (tagStart, (L'.EPrim (Prim.String "/>"), loc)), loc), fm)
adamc@143 1069 else
adamc@143 1070 normal ()
adamc@143 1071 | _ => normal ()
adamc@143 1072 end
adamc@152 1073 in
adamc@152 1074 case tag of
adamc@179 1075 "submit" => ((L'.EPrim (Prim.String "<input type=\"submit\"/>"), loc), fm)
adamc@152 1076
adamc@152 1077 | "textbox" =>
adamc@152 1078 (case targs of
adamc@152 1079 [_, (L.CName name, _)] =>
adamc@179 1080 let
adamc@179 1081 val (ts, fm) = tagStart "input"
adamc@179 1082 in
adamc@179 1083 ((L'.EStrcat (ts,
adamc@179 1084 (L'.EPrim (Prim.String (" name=\"" ^ name ^ "\"/>")),
adamc@179 1085 loc)), loc), fm)
adamc@179 1086 end
adamc@152 1087 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153 1088 raise Fail "No name passed to textarea tag"))
adamc@155 1089 | "password" => input "password"
adamc@152 1090 | "ltextarea" =>
adamc@152 1091 (case targs of
adamc@152 1092 [_, (L.CName name, _)] =>
adamc@179 1093 let
adamc@179 1094 val (ts, fm) = tagStart "textarea"
adamc@179 1095 val (xml, fm) = monoExp (env, st, fm) xml
adamc@179 1096 in
adamc@179 1097 ((L'.EStrcat ((L'.EStrcat (ts,
adamc@179 1098 (L'.EPrim (Prim.String (" name=\"" ^ name ^ "\">")), loc)), loc),
adamc@179 1099 (L'.EStrcat (xml,
adamc@179 1100 (L'.EPrim (Prim.String "</textarea>"),
adamc@179 1101 loc)), loc)),
adamc@179 1102 loc), fm)
adamc@179 1103 end
adamc@152 1104 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153 1105 raise Fail "No name passed to ltextarea tag"))
adamc@153 1106
adamc@190 1107 | "checkbox" => input "checkbox"
adamc@190 1108
adamc@153 1109 | "radio" =>
adamc@153 1110 (case targs of
adamc@153 1111 [_, (L.CName name, _)] =>
adamc@179 1112 monoExp (env, St.setRadioGroup (st, name), fm) xml
adamc@153 1113 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153 1114 raise Fail "No name passed to radio tag"))
adamc@153 1115 | "radioOption" =>
adamc@153 1116 (case St.radioGroup st of
adamc@153 1117 NONE => raise Fail "No name for radioGroup"
adamc@153 1118 | SOME name =>
adamc@153 1119 normal ("input",
adamc@153 1120 SOME (L'.EPrim (Prim.String (" type=\"radio\" name=\"" ^ name ^ "\"")), loc)))
adamc@152 1121
adamc@154 1122 | "lselect" =>
adamc@154 1123 (case targs of
adamc@154 1124 [_, (L.CName name, _)] =>
adamc@179 1125 let
adamc@179 1126 val (ts, fm) = tagStart "select"
adamc@179 1127 val (xml, fm) = monoExp (env, st, fm) xml
adamc@179 1128 in
adamc@179 1129 ((L'.EStrcat ((L'.EStrcat (ts,
adamc@179 1130 (L'.EPrim (Prim.String (" name=\"" ^ name ^ "\">")), loc)), loc),
adamc@179 1131 (L'.EStrcat (xml,
adamc@179 1132 (L'.EPrim (Prim.String "</select>"),
adamc@179 1133 loc)), loc)),
adamc@179 1134 loc),
adamc@179 1135 fm)
adamc@179 1136 end
adamc@154 1137 | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@154 1138 raise Fail "No name passed to lselect tag"))
adamc@154 1139
adamc@154 1140 | "loption" => normal ("option", NONE)
adamc@154 1141
adamc@152 1142 | _ => normal (tag, NONE)
adamc@95 1143 end
adamc@94 1144
adamc@141 1145 | L.EApp ((L.ECApp (
adamc@141 1146 (L.ECApp ((L.EFfi ("Basis", "lform"), _), _), _),
adamc@141 1147 _), _),
adamc@141 1148 xml) =>
adamc@143 1149 let
adamc@143 1150 fun findSubmit (e, _) =
adamc@143 1151 case e of
adamc@143 1152 L.EApp (
adamc@143 1153 (L.EApp (
adamc@143 1154 (L.ECApp (
adamc@143 1155 (L.ECApp (
adamc@143 1156 (L.ECApp (
adamc@143 1157 (L.ECApp (
adamc@143 1158 (L.EFfi ("Basis", "join"),
adamc@143 1159 _), _), _),
adamc@143 1160 _), _),
adamc@143 1161 _), _),
adamc@143 1162 _), _),
adamc@143 1163 xml1), _),
adamc@143 1164 xml2) => (case findSubmit xml1 of
adamc@143 1165 Error => Error
adamc@143 1166 | NotFound => findSubmit xml2
adamc@143 1167 | Found e =>
adamc@143 1168 case findSubmit xml2 of
adamc@143 1169 NotFound => Found e
adamc@143 1170 | _ => Error)
adamc@143 1171 | L.EApp (
adamc@143 1172 (L.EApp (
adamc@143 1173 (L.EApp (
adamc@143 1174 (L.ECApp (
adamc@143 1175 (L.ECApp (
adamc@143 1176 (L.ECApp (
adamc@143 1177 (L.ECApp (
adamc@143 1178 (L.ECApp (
adamc@143 1179 (L.ECApp (
adamc@143 1180 (L.ECApp (
adamc@143 1181 (L.ECApp (
adamc@143 1182 (L.EFfi ("Basis", "tag"),
adamc@143 1183 _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _),
adamc@143 1184 attrs), _),
adamc@143 1185 _), _),
adamc@143 1186 xml) =>
adamc@143 1187 (case #1 attrs of
adamc@143 1188 L.ERecord xes =>
adamc@143 1189 (case ListUtil.search (fn ((L.CName "Action", _), e, t) => SOME (e, t)
adamc@143 1190 | _ => NONE) xes of
adamc@143 1191 NONE => findSubmit xml
adamc@143 1192 | SOME et =>
adamc@143 1193 case findSubmit xml of
adamc@143 1194 NotFound => Found et
adamc@143 1195 | _ => Error)
adamc@143 1196 | _ => findSubmit xml)
adamc@143 1197 | _ => NotFound
adamc@143 1198
adamc@143 1199 val (action, actionT) = case findSubmit xml of
adamc@143 1200 NotFound => raise Fail "No submit found"
adamc@143 1201 | Error => raise Fail "Not ready for multi-submit lforms yet"
adamc@143 1202 | Found et => et
adamc@143 1203
adamc@143 1204 val actionT = monoType env actionT
adamc@179 1205 val (action, fm) = monoExp (env, st, fm) action
adamc@179 1206 val (action, fm) = urlifyExp env fm (action, actionT)
adamc@179 1207 val (xml, fm) = monoExp (env, st, fm) xml
adamc@143 1208 in
adamc@179 1209 ((L'.EStrcat ((L'.EStrcat ((L'.EPrim (Prim.String "<form action=\""), loc),
adamc@179 1210 (L'.EStrcat (action,
adamc@179 1211 (L'.EPrim (Prim.String "\">"), loc)), loc)), loc),
adamc@179 1212 (L'.EStrcat (xml,
adamc@179 1213 (L'.EPrim (Prim.String "</form>"), loc)), loc)), loc),
adamc@179 1214 fm)
adamc@143 1215 end
adamc@141 1216
adamc@148 1217 | L.EApp ((L.ECApp (
adamc@148 1218 (L.ECApp (
adamc@148 1219 (L.ECApp (
adamc@148 1220 (L.ECApp (
adamc@148 1221 (L.EFfi ("Basis", "useMore"), _), _), _),
adamc@148 1222 _), _),
adamc@148 1223 _), _),
adamc@148 1224 _), _),
adamc@179 1225 xml) => monoExp (env, st, fm) xml
adamc@148 1226
adamc@179 1227 | L.EApp (e1, e2) =>
adamc@179 1228 let
adamc@179 1229 val (e1, fm) = monoExp (env, st, fm) e1
adamc@179 1230 val (e2, fm) = monoExp (env, st, fm) e2
adamc@179 1231 in
adamc@179 1232 ((L'.EApp (e1, e2), loc), fm)
adamc@179 1233 end
adamc@26 1234 | L.EAbs (x, dom, ran, e) =>
adamc@179 1235 let
adamc@179 1236 val (e, fm) = monoExp (Env.pushERel env x dom, st, fm) e
adamc@179 1237 in
adamc@179 1238 ((L'.EAbs (x, monoType env dom, monoType env ran, e), loc), fm)
adamc@179 1239 end
adamc@25 1240 | L.ECApp _ => poly ()
adamc@25 1241 | L.ECAbs _ => poly ()
adamc@25 1242
adamc@252 1243 | L.EFfi mx => ((L'.EFfi mx, loc), fm)
adamc@252 1244 | L.EFfiApp (m, x, es) =>
adamc@252 1245 let
adamc@252 1246 val (es, fm) = ListUtil.foldlMap (fn (e, fm) => monoExp (env, st, fm) e) fm es
adamc@252 1247 in
adamc@252 1248 ((L'.EFfiApp (m, x, es), loc), fm)
adamc@252 1249 end
adamc@252 1250
adamc@179 1251 | L.ERecord xes =>
adamc@179 1252 let
adamc@179 1253 val (xes, fm) = ListUtil.foldlMap
adamc@179 1254 (fn ((x, e, t), fm) =>
adamc@179 1255 let
adamc@179 1256 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1257 in
adamc@179 1258 ((monoName env x,
adamc@179 1259 e,
adamc@179 1260 monoType env t), fm)
adamc@179 1261 end) fm xes
adamc@179 1262 in
adamc@179 1263 ((L'.ERecord xes, loc), fm)
adamc@179 1264 end
adamc@179 1265 | L.EField (e, x, _) =>
adamc@179 1266 let
adamc@179 1267 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1268 in
adamc@179 1269 ((L'.EField (e, monoName env x), loc), fm)
adamc@179 1270 end
adamc@149 1271 | L.ECut _ => poly ()
adamc@73 1272 | L.EFold _ => poly ()
adamc@177 1273
adamc@182 1274 | L.ECase (e, pes, {disc, result}) =>
adamc@179 1275 let
adamc@179 1276 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1277 val (pes, fm) = ListUtil.foldlMap
adamc@179 1278 (fn ((p, e), fm) =>
adamc@179 1279 let
adamc@179 1280 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1281 in
adamc@182 1282 ((monoPat env p, e), fm)
adamc@179 1283 end) fm pes
adamc@179 1284 in
adamc@182 1285 ((L'.ECase (e, pes, {disc = monoType env disc, result = monoType env result}), loc), fm)
adamc@179 1286 end
adamc@177 1287
adamc@179 1288 | L.EWrite e =>
adamc@179 1289 let
adamc@179 1290 val (e, fm) = monoExp (env, st, fm) e
adamc@179 1291 in
adamc@252 1292 ((L'.EAbs ("_", (L'.TRecord [], loc), (L'.TRecord [], loc),
adamc@252 1293 (L'.EWrite (liftExpInExp 0 e), loc)), loc), fm)
adamc@179 1294 end
adamc@110 1295
adamc@179 1296 | L.EClosure (n, es) =>
adamc@179 1297 let
adamc@179 1298 val (es, fm) = ListUtil.foldlMap (fn (e, fm) =>
adamc@179 1299 monoExp (env, st, fm) e)
adamc@179 1300 fm es
adamc@179 1301 in
adamc@179 1302 ((L'.EClosure (n, es), loc), fm)
adamc@179 1303 end
adamc@25 1304 end
adamc@25 1305
adamc@179 1306 fun monoDecl (env, fm) (all as (d, loc)) =
adamc@25 1307 let
adamc@25 1308 fun poly () =
adamc@25 1309 (E.errorAt loc "Unsupported declaration";
adamc@25 1310 Print.eprefaces' [("Declaration", CorePrint.p_decl env all)];
adamc@25 1311 NONE)
adamc@25 1312 in
adamc@25 1313 case d of
adamc@25 1314 L.DCon _ => NONE
adamc@193 1315 | L.DDatatype (x, n, [], xncs) =>
adamc@193 1316 let
adamc@196 1317 val env' = Env.declBinds env all
adamc@196 1318 val d = (L'.DDatatype (x, n, map (fn (x, n, to) => (x, n, Option.map (monoType env') to)) xncs), loc)
adamc@164 1319 in
adamc@196 1320 SOME (env', fm, d)
adamc@193 1321 end
adamc@193 1322 | L.DDatatype _ => poly ()
adamc@179 1323 | L.DVal (x, n, t, e, s) =>
adamc@179 1324 let
adamc@179 1325 val (e, fm) = monoExp (env, St.empty, fm) e
adamc@179 1326 in
adamc@179 1327 SOME (Env.pushENamed env x n t NONE s,
adamc@179 1328 fm,
adamc@179 1329 (L'.DVal (x, n, monoType env t, e, s), loc))
adamc@179 1330 end
adamc@128 1331 | L.DValRec vis =>
adamc@128 1332 let
adamc@128 1333 val env = foldl (fn ((x, n, t, e, s), env) => Env.pushENamed env x n t NONE s) env vis
adamc@179 1334
adamc@179 1335 val (vis, fm) = ListUtil.foldlMap
adamc@179 1336 (fn ((x, n, t, e, s), fm) =>
adamc@179 1337 let
adamc@179 1338 val (e, fm) = monoExp (env, St.empty, fm) e
adamc@179 1339 in
adamc@179 1340 ((x, n, monoType env t, e, s), fm)
adamc@179 1341 end)
adamc@179 1342 fm vis
adamc@128 1343 in
adamc@128 1344 SOME (env,
adamc@179 1345 fm,
adamc@179 1346 (L'.DValRec vis, loc))
adamc@128 1347 end
adamc@144 1348 | L.DExport (ek, n) =>
adamc@115 1349 let
adamc@120 1350 val (_, t, _, s) = Env.lookupENamed env n
adamc@120 1351
adamc@120 1352 fun unwind (t, _) =
adamc@120 1353 case t of
adamc@120 1354 L.TFun (dom, ran) => dom :: unwind ran
adamc@120 1355 | _ => []
adamc@120 1356
adamc@120 1357 val ts = map (monoType env) (unwind t)
adamc@115 1358 in
adamc@179 1359 SOME (env, fm, (L'.DExport (ek, s, n, ts), loc))
adamc@115 1360 end
adamc@251 1361 | L.DTable (x, n, _, s) =>
adamc@251 1362 let
adamc@251 1363 val t = (L.CFfi ("Basis", "string"), loc)
adamc@251 1364 val t' = (L'.TFfi ("Basis", "string"), loc)
adamc@251 1365 val e = (L'.EPrim (Prim.String s), loc)
adamc@251 1366 in
adamc@251 1367 SOME (Env.pushENamed env x n t NONE s,
adamc@251 1368 fm,
adamc@251 1369 (L'.DVal (x, n, t', e, s), loc))
adamc@251 1370 end
adamc@25 1371 end
adamc@25 1372
adamc@25 1373 fun monoize env ds =
adamc@25 1374 let
adamc@179 1375 val (_, _, ds) = List.foldl (fn (d, (env, fm, ds)) =>
adamc@179 1376 case monoDecl (env, fm) d of
adamc@179 1377 NONE => (env, fm, ds)
adamc@179 1378 | SOME (env, fm, d) =>
adamc@179 1379 (env,
adamc@179 1380 Fm.enter fm,
adamc@179 1381 d :: Fm.decls fm @ ds))
adamc@179 1382 (env, Fm.empty (CoreUtil.File.maxName ds + 1), []) ds
adamc@25 1383 in
adamc@25 1384 rev ds
adamc@25 1385 end
adamc@25 1386
adamc@25 1387 end