adamc@25: (* Copyright (c) 2008, Adam Chlipala
adamc@25: * All rights reserved.
adamc@25: *
adamc@25: * Redistribution and use in source and binary forms, with or without
adamc@25: * modification, are permitted provided that the following conditions are met:
adamc@25: *
adamc@25: * - Redistributions of source code must retain the above copyright notice,
adamc@25: * this list of conditions and the following disclaimer.
adamc@25: * - Redistributions in binary form must reproduce the above copyright notice,
adamc@25: * this list of conditions and the following disclaimer in the documentation
adamc@25: * and/or other materials provided with the distribution.
adamc@25: * - The names of contributors may not be used to endorse or promote products
adamc@25: * derived from this software without specific prior written permission.
adamc@25: *
adamc@25: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@25: * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@25: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@25: * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
adamc@25: * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
adamc@25: * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@25: * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@25: * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@25: * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@25: * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@25: * POSSIBILITY OF SUCH DAMAGE.
adamc@25: *)
adamc@25:
adamc@25: structure Monoize :> MONOIZE = struct
adamc@25:
adamc@25: structure E = ErrorMsg
adamc@25: structure Env = CoreEnv
adamc@25:
adamc@25: structure L = Core
adamc@25: structure L' = Mono
adamc@25:
adamc@196: structure IM = IntBinaryMap
adamc@196:
adamc@196: val dummyTyp = (L'.TDatatype (0, ref (L'.Enum, [])), E.dummySpan)
adamc@25:
adamc@252: structure U = MonoUtil
adamc@252:
adamc@252: val liftExpInExp =
adamc@252: U.Exp.mapB {typ = fn t => t,
adamc@252: exp = fn bound => fn e =>
adamc@252: case e of
adamc@252: L'.ERel xn =>
adamc@252: if xn < bound then
adamc@252: e
adamc@252: else
adamc@252: L'.ERel (xn + 1)
adamc@252: | _ => e,
adamc@252: bind = fn (bound, U.Exp.RelE _) => bound + 1
adamc@252: | (bound, _) => bound}
adamc@252:
adamc@25: fun monoName env (all as (c, loc)) =
adamc@25: let
adamc@25: fun poly () =
adamc@25: (E.errorAt loc "Unsupported name constructor";
adamc@25: Print.eprefaces' [("Constructor", CorePrint.p_con env all)];
adamc@25: "")
adamc@25: in
adamc@25: case c of
adamc@25: L.CName s => s
adamc@25: | _ => poly ()
adamc@25: end
adamc@25:
adamc@196: fun monoType env =
adamc@25: let
adamc@196: fun mt env dtmap (all as (c, loc)) =
adamc@196: let
adamc@196: fun poly () =
adamc@196: (E.errorAt loc "Unsupported type constructor";
adamc@196: Print.eprefaces' [("Constructor", CorePrint.p_con env all)];
adamc@196: dummyTyp)
adamc@196: in
adamc@196: case c of
adamc@196: L.TFun (c1, c2) => (L'.TFun (mt env dtmap c1, mt env dtmap c2), loc)
adamc@196: | L.TCFun _ => poly ()
adamc@196: | L.TRecord (L.CRecord ((L.KType, _), xcs), _) =>
adamc@196: (L'.TRecord (map (fn (x, t) => (monoName env x, mt env dtmap t)) xcs), loc)
adamc@196: | L.TRecord _ => poly ()
adamc@196:
adamc@196: | L.CApp ((L.CApp ((L.CApp ((L.CFfi ("Basis", "xml"), _), _), _), _), _), _) =>
adamc@196: (L'.TFfi ("Basis", "string"), loc)
adamc@196: | L.CApp ((L.CApp ((L.CFfi ("Basis", "xhtml"), _), _), _), _) =>
adamc@196: (L'.TFfi ("Basis", "string"), loc)
adamc@196:
adamc@251: | L.CApp ((L.CFfi ("Basis", "transaction"), _), t) =>
adamc@252: (L'.TFun ((L'.TRecord [], loc), mt env dtmap t), loc)
adamc@252: | L.CApp ((L.CFfi ("Basis", "sql_table"), _), _) =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_query"), _), _), _), _) =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CApp ((L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_query1"), _), _), _), _), _), _) =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CApp ((L.CApp ((L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_exp"), _), _), _), _), _), _), _), _) =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252:
adamc@252: | L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_subset"), _), _), _), _) =>
adamc@252: (L'.TRecord [], loc)
adamc@252: | L.CFfi ("Basis", "sql_relop") =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CFfi ("Basis", "sql_direction") =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_order_by"), _), _), _), _) =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CFfi ("Basis", "sql_limit") =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CFfi ("Basis", "sql_offset") =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252:
adamc@252: | L.CApp ((L.CFfi ("Basis", "sql_injectable"), _), t) =>
adamc@252: (L'.TFun (mt env dtmap t, (L'.TFfi ("Basis", "string"), loc)), loc)
adamc@252: | L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_unary"), _), _), _), _) =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CApp ((L.CApp ((L.CApp ((L.CFfi ("Basis", "sql_binary"), _), _), _), _), _), _) =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CFfi ("Basis", "sql_comparison") =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CApp ((L.CFfi ("Basis", "sql_aggregate"), _), t) =>
adamc@252: (L'.TFfi ("Basis", "string"), loc)
adamc@252: | L.CApp ((L.CFfi ("Basis", "sql_summable"), _), _) =>
adamc@252: (L'.TRecord [], loc)
adamc@252: | L.CApp ((L.CFfi ("Basis", "sql_maxable"), _), _) =>
adamc@252: (L'.TRecord [], loc)
adamc@251:
adamc@196: | L.CRel _ => poly ()
adamc@196: | L.CNamed n =>
adamc@196: (case IM.find (dtmap, n) of
adamc@196: SOME r => (L'.TDatatype (n, r), loc)
adamc@196: | NONE =>
adamc@196: let
adamc@196: val r = ref (L'.Default, [])
adamc@196: val (_, xs, xncs) = Env.lookupDatatype env n
adamc@196:
adamc@196: val dtmap' = IM.insert (dtmap, n, r)
adamc@196:
adamc@196: val xncs = map (fn (x, n, to) => (x, n, Option.map (mt env dtmap') to)) xncs
adamc@196: in
adamc@196: case xs of
adamc@198: [] =>(r := (ElabUtil.classifyDatatype xncs, xncs);
adamc@196: (L'.TDatatype (n, r), loc))
adamc@196: | _ => poly ()
adamc@196: end)
adamc@196: | L.CFfi mx => (L'.TFfi mx, loc)
adamc@196: | L.CApp _ => poly ()
adamc@196: | L.CAbs _ => poly ()
adamc@196:
adamc@196: | L.CName _ => poly ()
adamc@196:
adamc@196: | L.CRecord _ => poly ()
adamc@196: | L.CConcat _ => poly ()
adamc@196: | L.CFold _ => poly ()
adamc@196: | L.CUnit => poly ()
adamc@214:
adamc@214: | L.CTuple _ => poly ()
adamc@214: | L.CProj _ => poly ()
adamc@196: end
adamc@25: in
adamc@196: mt env IM.empty
adamc@25: end
adamc@25:
adamc@25: val dummyExp = (L'.EPrim (Prim.Int 0), E.dummySpan)
adamc@25:
adamc@179: structure IM = IntBinaryMap
adamc@179:
adamc@179: datatype foo_kind =
adamc@179: Attr
adamc@179: | Url
adamc@179:
adamc@179: fun fk2s fk =
adamc@179: case fk of
adamc@179: Attr => "attr"
adamc@179: | Url => "url"
adamc@179:
adamc@179: structure Fm :> sig
adamc@179: type t
adamc@179:
adamc@179: val empty : int -> t
adamc@179:
adamc@179: val lookup : t -> foo_kind -> int -> (int -> t -> L'.decl * t) -> t * int
adamc@179: val enter : t -> t
adamc@179: val decls : t -> L'.decl list
adamc@179: end = struct
adamc@179:
adamc@179: structure M = BinaryMapFn(struct
adamc@179: type ord_key = foo_kind
adamc@179: fun compare x =
adamc@179: case x of
adamc@179: (Attr, Attr) => EQUAL
adamc@179: | (Attr, _) => LESS
adamc@179: | (_, Attr) => GREATER
adamc@179:
adamc@179: | (Url, Url) => EQUAL
adamc@179: end)
adamc@179:
adamc@179: type t = {
adamc@179: count : int,
adamc@179: map : int IM.map M.map,
adamc@179: decls : L'.decl list
adamc@179: }
adamc@179:
adamc@179: fun empty count = {
adamc@179: count = count,
adamc@179: map = M.empty,
adamc@179: decls = []
adamc@179: }
adamc@179:
adamc@179: fun enter ({count, map, ...} : t) = {count = count, map = map, decls = []}
adamc@179: fun decls ({decls, ...} : t) = decls
adamc@179:
adamc@179: fun lookup (t as {count, map, decls}) k n thunk =
adamc@120: let
adamc@179: val im = Option.getOpt (M.find (map, k), IM.empty)
adamc@179: in
adamc@179: case IM.find (im, n) of
adamc@179: NONE =>
adamc@179: let
adamc@179: val n' = count
adamc@179: val (d, {count, map, decls}) = thunk count {count = count + 1,
adamc@179: map = M.insert (map, k, IM.insert (im, n, n')),
adamc@179: decls = decls}
adamc@179: in
adamc@179: ({count = count,
adamc@179: map = map,
adamc@179: decls = d :: decls}, n')
adamc@179: end
adamc@179: | SOME n' => (t, n')
adamc@179: end
adamc@179:
adamc@179: end
adamc@185:
adamc@185:
adamc@185: fun capitalize s =
adamc@185: if s = "" then
adamc@185: s
adamc@185: else
adamc@185: str (Char.toUpper (String.sub (s, 0))) ^ String.extract (s, 1, NONE)
adamc@179:
adamc@179: fun fooifyExp fk env =
adamc@179: let
adamc@179: fun fooify fm (e, tAll as (t, loc)) =
adamc@120: case #1 e of
adamc@120: L'.EClosure (fnam, [(L'.ERecord [], _)]) =>
adamc@120: let
adamc@120: val (_, _, _, s) = Env.lookupENamed env fnam
adamc@120: in
adamc@183: ((L'.EPrim (Prim.String ("/" ^ s)), loc), fm)
adamc@120: end
adamc@120: | L'.EClosure (fnam, args) =>
adamc@120: let
adamc@120: val (_, ft, _, s) = Env.lookupENamed env fnam
adamc@120: val ft = monoType env ft
adamc@111:
adamc@179: fun attrify (args, ft, e, fm) =
adamc@120: case (args, ft) of
adamc@179: ([], _) => (e, fm)
adamc@120: | (arg :: args, (L'.TFun (t, ft), _)) =>
adamc@179: let
adamc@179: val (arg', fm) = fooify fm (arg, t)
adamc@179: in
adamc@179: attrify (args, ft,
adamc@179: (L'.EStrcat (e,
adamc@179: (L'.EStrcat ((L'.EPrim (Prim.String "/"), loc),
adamc@179: arg'), loc)), loc),
adamc@179: fm)
adamc@179: end
adamc@120: | _ => (E.errorAt loc "Type mismatch encoding attribute";
adamc@179: (e, fm))
adamc@120: in
adamc@183: attrify (args, ft, (L'.EPrim (Prim.String ("/" ^ s)), loc), fm)
adamc@120: end
adamc@120: | _ =>
adamc@120: case t of
adamc@185: L'.TFfi (m, x) => ((L'.EFfiApp (m, fk2s fk ^ "ify" ^ capitalize x, [e]), loc), fm)
adamc@200:
adamc@179: | L'.TRecord [] => ((L'.EPrim (Prim.String ""), loc), fm)
adamc@200: | L'.TRecord ((x, t) :: xts) =>
adamc@200: let
adamc@200: val (se, fm) = fooify fm ((L'.EField (e, x), loc), t)
adamc@200: in
adamc@200: foldl (fn ((x, t), (se, fm)) =>
adamc@200: let
adamc@200: val (se', fm) = fooify fm ((L'.EField (e, x), loc), t)
adamc@200: in
adamc@200: ((L'.EStrcat (se,
adamc@200: (L'.EStrcat ((L'.EPrim (Prim.String "/"), loc),
adamc@200: se'), loc)), loc),
adamc@200: fm)
adamc@200: end) (se, fm) xts
adamc@200: end
adamc@111:
adamc@196: | L'.TDatatype (i, ref (dk, _)) =>
adamc@179: let
adamc@179: fun makeDecl n fm =
adamc@179: let
adamc@193: val (x, _, xncs) = Env.lookupDatatype env i
adamc@179:
adamc@179: val (branches, fm) =
adamc@179: ListUtil.foldlMap
adamc@179: (fn ((x, n, to), fm) =>
adamc@179: case to of
adamc@179: NONE =>
adamc@188: (((L'.PCon (dk, L'.PConVar n, NONE), loc),
adamc@179: (L'.EPrim (Prim.String x), loc)),
adamc@179: fm)
adamc@179: | SOME t =>
adamc@179: let
adamc@182: val t = monoType env t
adamc@182: val (arg, fm) = fooify fm ((L'.ERel 0, loc), t)
adamc@179: in
adamc@188: (((L'.PCon (dk, L'.PConVar n, SOME (L'.PVar ("a", t), loc)), loc),
adamc@179: (L'.EStrcat ((L'.EPrim (Prim.String (x ^ "/")), loc),
adamc@179: arg), loc)),
adamc@179: fm)
adamc@179: end)
adamc@179: fm xncs
adamc@179:
adamc@179: val dom = tAll
adamc@179: val ran = (L'.TFfi ("Basis", "string"), loc)
adamc@179: in
adamc@179: ((L'.DValRec [(fk2s fk ^ "ify_" ^ x,
adamc@179: n,
adamc@179: (L'.TFun (dom, ran), loc),
adamc@179: (L'.EAbs ("x",
adamc@179: dom,
adamc@179: ran,
adamc@179: (L'.ECase ((L'.ERel 0, loc),
adamc@179: branches,
adamc@182: {disc = dom,
adamc@182: result = ran}), loc)), loc),
adamc@179: "")], loc),
adamc@179: fm)
adamc@179: end
adamc@179:
adamc@179: val (fm, n) = Fm.lookup fm fk i makeDecl
adamc@179: in
adamc@179: ((L'.EApp ((L'.ENamed n, loc), e), loc), fm)
adamc@179: end
adamc@164:
adamc@120: | _ => (E.errorAt loc "Don't know how to encode attribute type";
adamc@120: Print.eprefaces' [("Type", MonoPrint.p_typ MonoEnv.empty tAll)];
adamc@179: (dummyExp, fm))
adamc@120: in
adamc@120: fooify
adamc@120: end
adamc@120:
adamc@179: val attrifyExp = fooifyExp Attr
adamc@179: val urlifyExp = fooifyExp Url
adamc@105:
adamc@143: datatype 'a failable_search =
adamc@143: Found of 'a
adamc@143: | NotFound
adamc@143: | Error
adamc@143:
adamc@153: structure St :> sig
adamc@153: type t
adamc@153:
adamc@153: val empty : t
adamc@153:
adamc@153: val radioGroup : t -> string option
adamc@153: val setRadioGroup : t * string -> t
adamc@153: end = struct
adamc@153:
adamc@153: type t = {
adamc@153: radioGroup : string option
adamc@153: }
adamc@153:
adamc@153: val empty = {radioGroup = NONE}
adamc@153:
adamc@153: fun radioGroup (t : t) = #radioGroup t
adamc@153:
adamc@153: fun setRadioGroup (t : t, x) = {radioGroup = SOME x}
adamc@153:
adamc@153: end
adamc@153:
adamc@186: fun monoPatCon env pc =
adamc@178: case pc of
adamc@178: L.PConVar n => L'.PConVar n
adamc@188: | L.PConFfi {mod = m, datatyp, con, arg, ...} => L'.PConFfi {mod = m, datatyp = datatyp, con = con,
adamc@188: arg = Option.map (monoType env) arg}
adamc@178:
adamc@193: val dummyPat = (L'.PPrim (Prim.Int 0), ErrorMsg.dummySpan)
adamc@193:
adamc@193: fun monoPat env (all as (p, loc)) =
adamc@193: let
adamc@193: fun poly () =
adamc@193: (E.errorAt loc "Unsupported pattern";
adamc@193: Print.eprefaces' [("Pattern", CorePrint.p_pat env all)];
adamc@193: dummyPat)
adamc@193: in
adamc@193: case p of
adamc@193: L.PWild => (L'.PWild, loc)
adamc@193: | L.PVar (x, t) => (L'.PVar (x, monoType env t), loc)
adamc@193: | L.PPrim p => (L'.PPrim p, loc)
adamc@193: | L.PCon (dk, pc, [], po) => (L'.PCon (dk, monoPatCon env pc, Option.map (monoPat env) po), loc)
adamc@193: | L.PCon _ => poly ()
adamc@193: | L.PRecord xps => (L'.PRecord (map (fn (x, p, t) => (x, monoPat env p, monoType env t)) xps), loc)
adamc@193: end
adamc@178:
adamc@252: fun strcat loc es =
adamc@252: case es of
adamc@252: [] => (L'.EPrim (Prim.String ""), loc)
adamc@252: | [e] => e
adamc@252: | _ =>
adamc@252: let
adamc@252: val e2 = List.last es
adamc@252: val es = List.take (es, length es - 1)
adamc@252: val e1 = List.last es
adamc@252: val es = List.take (es, length es - 1)
adamc@252: in
adamc@252: foldr (fn (e, e') => (L'.EStrcat (e, e'), loc))
adamc@252: (L'.EStrcat (e1, e2), loc) es
adamc@252: end
adamc@252:
adamc@252: fun strcatComma loc es =
adamc@252: case es of
adamc@252: [] => (L'.EPrim (Prim.String ""), loc)
adamc@252: | [e] => e
adamc@252: | _ =>
adamc@252: let
adamc@252: val e1 = List.last es
adamc@252: val es = List.take (es, length es - 1)
adamc@252: in
adamc@252: foldr (fn (e, e') =>
adamc@265: case (e, e') of
adamc@265: ((L'.EPrim (Prim.String ""), _), _) => e'
adamc@265: | (_, (L'.EPrim (Prim.String ""), _)) => e
adamc@252: | _ =>
adamc@252: (L'.EStrcat (e,
adamc@252: (L'.EStrcat ((L'.EPrim (Prim.String ", "), loc), e'), loc)), loc))
adamc@252: e1 es
adamc@252: end
adamc@252:
adamc@252: fun strcatR loc e xs = strcatComma loc (map (fn (x, _) => (L'.EField (e, x), loc)) xs)
adamc@252:
adamc@179: fun monoExp (env, st, fm) (all as (e, loc)) =
adamc@25: let
adamc@25: fun poly () =
adamc@25: (E.errorAt loc "Unsupported expression";
adamc@25: Print.eprefaces' [("Expression", CorePrint.p_exp env all)];
adamc@179: (dummyExp, fm))
adamc@25: in
adamc@25: case e of
adamc@179: L.EPrim p => ((L'.EPrim p, loc), fm)
adamc@179: | L.ERel n => ((L'.ERel n, loc), fm)
adamc@179: | L.ENamed n => ((L'.ENamed n, loc), fm)
adamc@193: | L.ECon (dk, pc, [], eo) =>
adamc@193: let
adamc@179: val (eo, fm) =
adamc@179: case eo of
adamc@179: NONE => (NONE, fm)
adamc@179: | SOME e =>
adamc@179: let
adamc@179: val (e, fm) = monoExp (env, st, fm) e
adamc@179: in
adamc@179: (SOME e, fm)
adamc@179: end
adamc@179: in
adamc@188: ((L'.ECon (dk, monoPatCon env pc, eo), loc), fm)
adamc@193: end
adamc@193: | L.ECon _ => poly ()
adamc@94:
adamc@251: | L.ECApp ((L.EFfi ("Basis", "return"), _), t) =>
adamc@252: let
adamc@252: val t = monoType env t
adamc@252: in
adamc@252: ((L'.EAbs ("x", t,
adamc@252: (L'.TFun ((L'.TRecord [], loc), t), loc),
adamc@252: (L'.EAbs ("_", (L'.TRecord [], loc), t,
adamc@252: (L'.ERel 1, loc)), loc)), loc), fm)
adamc@252: end
adamc@251: | L.ECApp ((L.ECApp ((L.EFfi ("Basis", "bind"), _), t1), _), t2) =>
adamc@251: let
adamc@251: val t1 = monoType env t1
adamc@251: val t2 = monoType env t2
adamc@251: val un = (L'.TRecord [], loc)
adamc@252: val mt1 = (L'.TFun (un, t1), loc)
adamc@252: val mt2 = (L'.TFun (un, t2), loc)
adamc@251: in
adamc@252: ((L'.EAbs ("m1", mt1, (L'.TFun (mt1, (L'.TFun (mt2, (L'.TFun (un, un), loc)), loc)), loc),
adamc@252: (L'.EAbs ("m2", mt2, (L'.TFun (un, un), loc),
adamc@252: (L'.EAbs ("_", un, un,
adamc@252: (L'.ELet ("r", t1, (L'.EApp ((L'.ERel 2, loc),
adamc@252: (L'.ERecord [], loc)), loc),
adamc@252: (L'.EApp (
adamc@252: (L'.EApp ((L'.ERel 2, loc), (L'.ERel 0, loc)), loc),
adamc@252: (L'.ERecord [], loc)),
adamc@252: loc)), loc)), loc)), loc)), loc),
adamc@251: fm)
adamc@251: end
adamc@251:
adamc@252: | L.ECApp (
adamc@252: (L.ECApp (
adamc@252: (L.ECApp ((L.EFfi ("Basis", "query"), _), (L.CRecord (_, tables), _)), _),
adamc@252: exps), _),
adamc@252: state) =>
adamc@252: (case monoType env (L.TRecord exps, loc) of
adamc@252: (L'.TRecord exps, _) =>
adamc@252: let
adamc@252: val tables = map (fn ((L.CName x, _), xts) =>
adamc@252: (case monoType env (L.TRecord xts, loc) of
adamc@252: (L'.TRecord xts, _) => SOME (x, xts)
adamc@252: | _ => NONE)
adamc@252: | _ => NONE) tables
adamc@252: in
adamc@252: if List.exists (fn x => x = NONE) tables then
adamc@252: poly ()
adamc@252: else
adamc@252: let
adamc@252: val tables = List.mapPartial (fn x => x) tables
adamc@252: val state = monoType env state
adamc@252: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@252: val un = (L'.TRecord [], loc)
adamc@252:
adamc@252: val rt = exps @ map (fn (x, xts) => (x, (L'.TRecord xts, loc))) tables
adamc@252: val ft = (L'.TFun ((L'.TRecord rt, loc),
adamc@252: (L'.TFun (state,
adamc@252: (L'.TFun (un, state), loc)),
adamc@252: loc)), loc)
adamc@252:
adamc@267: val body' = (L'.EApp (
adamc@267: (L'.EApp (
adamc@267: (L'.EApp ((L'.ERel 4, loc),
adamc@267: (L'.ERel 1, loc)), loc),
adamc@267: (L'.ERel 0, loc)), loc),
adamc@267: (L'.ERecord [], loc)), loc)
adamc@252:
adamc@252: val body = (L'.EQuery {exps = exps,
adamc@252: tables = tables,
adamc@252: state = state,
adamc@252: query = (L'.ERel 3, loc),
adamc@252: body = body',
adamc@252: initial = (L'.ERel 1, loc)},
adamc@252: loc)
adamc@252: in
adamc@252: ((L'.EAbs ("q", s, (L'.TFun (ft, (L'.TFun (state, (L'.TFun (un, state), loc)), loc)), loc),
adamc@252: (L'.EAbs ("f", ft, (L'.TFun (state, (L'.TFun (un, state), loc)), loc),
adamc@252: (L'.EAbs ("i", state, (L'.TFun (un, state), loc),
adamc@252: (L'.EAbs ("_", un, state,
adamc@252: body), loc)), loc)), loc)), loc), fm)
adamc@252: end
adamc@252: end
adamc@252: | _ => poly ())
adamc@252:
adamc@252: | L.ECApp ((L.ECApp ((L.ECApp ((L.EFfi ("Basis", "sql_query"), _), _), _), _), _), _) =>
adamc@252: let
adamc@252: fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@252: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@252: fun gf s = (L'.EField ((L'.ERel 0, loc), s), loc)
adamc@252: in
adamc@252: ((L'.EAbs ("r",
adamc@252: (L'.TRecord [("Rows", s), ("OrderBy", s), ("Limit", s), ("Offset", s)], loc),
adamc@252: s,
adamc@252: strcat loc [gf "Rows",
adamc@261: (L'.ECase (gf "OrderBy",
adamc@261: [((L'.PPrim (Prim.String ""), loc), sc ""),
adamc@261: ((L'.PWild, loc),
adamc@261: strcat loc [sc " ORDER BY ",
adamc@261: gf "OrderBy"])],
adamc@261: {disc = s, result = s}), loc),
adamc@252: gf "Limit",
adamc@252: gf "Offset"]), loc), fm)
adamc@252: end
adamc@252:
adamc@252: | L.ECApp (
adamc@252: (L.ECApp (
adamc@252: (L.ECApp (
adamc@252: (L.ECApp (
adamc@252: (L.EFfi ("Basis", "sql_query1"), _),
adamc@252: (L.CRecord (_, tables), _)), _),
adamc@252: (L.CRecord (_, grouped), _)), _),
adamc@252: (L.CRecord (_, stables), _)), _),
adamc@252: sexps) =>
adamc@252: let
adamc@252: fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@252: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@252: val un = (L'.TRecord [], loc)
adamc@252: fun gf s = (L'.EField ((L'.ERel 0, loc), s), loc)
adamc@252:
adamc@252: fun doTables tables =
adamc@252: let
adamc@252: val tables = map (fn ((L.CName x, _), xts) =>
adamc@252: (case monoType env (L.TRecord xts, loc) of
adamc@252: (L'.TRecord xts, _) => SOME (x, xts)
adamc@252: | _ => NONE)
adamc@252: | _ => NONE) tables
adamc@252: in
adamc@252: if List.exists (fn x => x = NONE) tables then
adamc@252: NONE
adamc@252: else
adamc@260: let
adamc@260: val tables = List.mapPartial (fn x => x) tables
adamc@260: val tables = ListMergeSort.sort
adamc@260: (fn ((x, _), (y, _)) => String.compare (x, y) = GREATER)
adamc@260: tables
adamc@260: val tables = map (fn (x, xts) =>
adamc@260: (x, ListMergeSort.sort
adamc@260: (fn ((x, _), (y, _)) => String.compare (x, y) = GREATER)
adamc@260: xts)) tables
adamc@260: in
adamc@260: SOME tables
adamc@260: end
adamc@252: end
adamc@252: in
adamc@252: case (doTables tables, doTables grouped, doTables stables, monoType env (L.TRecord sexps, loc)) of
adamc@252: (SOME tables, SOME grouped, SOME stables, (L'.TRecord sexps, _)) =>
adamc@252: ((L'.EAbs ("r",
adamc@252: (L'.TRecord [("From", (L'.TRecord (map (fn (x, _) => (x, s)) tables), loc)),
adamc@252: ("Where", s),
adamc@252: ("GroupBy", un),
adamc@252: ("Having", s),
adamc@252: ("SelectFields", un),
adamc@252: ("SelectExps", (L'.TRecord (map (fn (x, _) => (x, s)) sexps), loc))],
adamc@252: loc),
adamc@252: s,
adamc@252: strcat loc [sc "SELECT ",
adamc@261: strcatComma loc (map (fn (x, t) =>
adamc@261: strcat loc [
adamc@261: (L'.EField (gf "SelectExps", x), loc),
adamc@261: sc (" AS _" ^ x)
adamc@265: ]) sexps
adamc@265: @ map (fn (x, xts) =>
adamc@265: strcatComma loc
adamc@265: (map (fn (x', _) =>
adamc@277: sc (x ^ ".lw_" ^ x'))
adamc@265: xts)) stables),
adamc@252: sc " FROM ",
adamc@252: strcatComma loc (map (fn (x, _) => strcat loc [(L'.EField (gf "From", x), loc),
adamc@253: sc (" AS " ^ x)]) tables),
adamc@258: (L'.ECase (gf "Where",
adamc@258: [((L'.PPrim (Prim.String "TRUE"), loc),
adamc@258: sc ""),
adamc@258: ((L'.PWild, loc),
adamc@258: strcat loc [sc " WHERE ", gf "Where"])],
adamc@258: {disc = s,
adamc@258: result = s}), loc),
adamc@258:
adamc@255: if List.all (fn (x, xts) =>
adamc@255: case List.find (fn (x', _) => x' = x) grouped of
adamc@255: NONE => List.null xts
adamc@255: | SOME (_, xts') =>
adamc@255: List.all (fn (x, _) =>
adamc@255: List.exists (fn (x', _) => x' = x)
adamc@255: xts') xts) tables then
adamc@255: sc ""
adamc@255: else
adamc@255: strcat loc [
adamc@255: sc " GROUP BY ",
adamc@255: strcatComma loc (map (fn (x, xts) =>
adamc@255: strcatComma loc
adamc@255: (map (fn (x', _) =>
adamc@277: sc (x ^ ".lw_" ^ x'))
adamc@255: xts)) grouped)
adamc@259: ],
adamc@259:
adamc@259: (L'.ECase (gf "Having",
adamc@259: [((L'.PPrim (Prim.String "TRUE"), loc),
adamc@259: sc ""),
adamc@259: ((L'.PWild, loc),
adamc@259: strcat loc [sc " HAVING ", gf "Having"])],
adamc@259: {disc = s,
adamc@259: result = s}), loc)
adamc@252: ]), loc),
adamc@252: fm)
adamc@252: | _ => poly ()
adamc@252: end
adamc@252:
adamc@252: | L.ECApp (
adamc@252: (L.ECApp (
adamc@252: (L.ECApp (
adamc@252: (L.ECApp (
adamc@252: (L.EFfi ("Basis", "sql_inject"), _),
adamc@252: _), _),
adamc@252: _), _),
adamc@252: _), _),
adamc@252: t) =>
adamc@252: let
adamc@252: val t = monoType env t
adamc@252: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@252: in
adamc@252: ((L'.EAbs ("f", (L'.TFun (t, s), loc), (L'.TFun (t, s), loc),
adamc@252: (L'.ERel 0, loc)), loc), fm)
adamc@252: end
adamc@252:
adamc@253: | L.EFfi ("Basis", "sql_int") =>
adamc@253: ((L'.EAbs ("x", (L'.TFfi ("Basis", "int"), loc), (L'.TFfi ("Basis", "string"), loc),
adamc@253: (L'.EFfiApp ("Basis", "sqlifyInt", [(L'.ERel 0, loc)]), loc)), loc),
adamc@253: fm)
adamc@253: | L.EFfi ("Basis", "sql_float") =>
adamc@253: ((L'.EAbs ("x", (L'.TFfi ("Basis", "float"), loc), (L'.TFfi ("Basis", "string"), loc),
adamc@253: (L'.EFfiApp ("Basis", "sqlifyFloat", [(L'.ERel 0, loc)]), loc)), loc),
adamc@253: fm)
adamc@253: | L.EFfi ("Basis", "sql_bool") =>
adamc@253: ((L'.EAbs ("x", (L'.TFfi ("Basis", "bool"), loc), (L'.TFfi ("Basis", "string"), loc),
adamc@253: (L'.EFfiApp ("Basis", "sqlifyBool", [(L'.ERel 0, loc)]), loc)), loc),
adamc@253: fm)
adamc@253: | L.EFfi ("Basis", "sql_string") =>
adamc@253: ((L'.EAbs ("x", (L'.TFfi ("Basis", "string"), loc), (L'.TFfi ("Basis", "string"), loc),
adamc@253: (L'.EFfiApp ("Basis", "sqlifyString", [(L'.ERel 0, loc)]), loc)), loc),
adamc@253: fm)
adamc@253:
adamc@252: | L.ECApp ((L.EFfi ("Basis", "sql_subset"), _), _) =>
adamc@252: ((L'.ERecord [], loc), fm)
adamc@252: | L.ECApp ((L.EFfi ("Basis", "sql_subset_all"), _), _) =>
adamc@252: ((L'.ERecord [], loc), fm)
adamc@252:
adamc@252: | L.ECApp ((L.ECApp ((L.EFfi ("Basis", "sql_order_by_Nil"), _), _), _), _) =>
adamc@252: ((L'.EPrim (Prim.String ""), loc), fm)
adamc@261: | L.ECApp (
adamc@261: (L.ECApp (
adamc@261: (L.ECApp (
adamc@261: (L.EFfi ("Basis", "sql_order_by_Cons"), _),
adamc@261: _), _),
adamc@261: _), _),
adamc@261: _) =>
adamc@261: let
adamc@261: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@261: fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@261: in
adamc@268: ((L'.EAbs ("e1", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@268: (L'.EAbs ("d", s, (L'.TFun (s, s), loc),
adamc@268: (L'.EAbs ("e2", s, s,
adamc@268: (L'.ECase ((L'.ERel 0, loc),
adamc@268: [((L'.PPrim (Prim.String ""), loc),
adamc@268: strcat loc [(L'.ERel 2, loc),
adamc@268: (L'.ERel 1, loc)]),
adamc@268: ((L'.PWild, loc),
adamc@268: strcat loc [(L'.ERel 2, loc),
adamc@268: (L'.ERel 1, loc),
adamc@268: sc ", ",
adamc@268: (L'.ERel 0, loc)])],
adamc@268: {disc = s, result = s}), loc)), loc)), loc)), loc),
adamc@261: fm)
adamc@261: end
adamc@252:
adamc@252: | L.EFfi ("Basis", "sql_no_limit") =>
adamc@252: ((L'.EPrim (Prim.String ""), loc), fm)
adamc@262: | L.EFfiApp ("Basis", "sql_limit", [e]) =>
adamc@262: let
adamc@262: val (e, fm) = monoExp (env, st, fm) e
adamc@262: in
adamc@262: (strcat loc [
adamc@262: (L'.EPrim (Prim.String " LIMIT "), loc),
adamc@262: (L'.EFfiApp ("Basis", "sqlifyInt", [e]), loc)
adamc@262: ],
adamc@262: fm)
adamc@262: end
adamc@262:
adamc@252: | L.EFfi ("Basis", "sql_no_offset") =>
adamc@252: ((L'.EPrim (Prim.String ""), loc), fm)
adamc@263: | L.EFfiApp ("Basis", "sql_offset", [e]) =>
adamc@263: let
adamc@263: val (e, fm) = monoExp (env, st, fm) e
adamc@263: in
adamc@263: (strcat loc [
adamc@263: (L'.EPrim (Prim.String " OFFSET "), loc),
adamc@263: (L'.EFfiApp ("Basis", "sqlifyInt", [e]), loc)
adamc@263: ],
adamc@263: fm)
adamc@263: end
adamc@253:
adamc@253: | L.EFfi ("Basis", "sql_eq") =>
adamc@253: ((L'.EPrim (Prim.String "="), loc), fm)
adamc@253: | L.EFfi ("Basis", "sql_ne") =>
adamc@253: ((L'.EPrim (Prim.String "<>"), loc), fm)
adamc@253: | L.EFfi ("Basis", "sql_lt") =>
adamc@253: ((L'.EPrim (Prim.String "<"), loc), fm)
adamc@253: | L.EFfi ("Basis", "sql_le") =>
adamc@253: ((L'.EPrim (Prim.String "<="), loc), fm)
adamc@253: | L.EFfi ("Basis", "sql_gt") =>
adamc@253: ((L'.EPrim (Prim.String ">"), loc), fm)
adamc@253: | L.EFfi ("Basis", "sql_ge") =>
adamc@253: ((L'.EPrim (Prim.String ">="), loc), fm)
adamc@253:
adamc@253: | L.ECApp (
adamc@253: (L.ECApp (
adamc@253: (L.ECApp (
adamc@253: (L.ECApp (
adamc@254: (L.ECApp (
adamc@264: (L.EFfi ("Basis", "sql_unary"), _),
adamc@264: _), _),
adamc@264: _), _),
adamc@264: _), _),
adamc@264: _), _),
adamc@264: _) =>
adamc@264: let
adamc@264: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@264: fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@264: in
adamc@264: ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@264: (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@264: strcat loc [sc "(",
adamc@264: (L'.ERel 1, loc),
adamc@264: sc " ",
adamc@264: (L'.ERel 0, loc),
adamc@264: sc ")"]), loc)), loc),
adamc@264: fm)
adamc@264: end
adamc@264: | L.EFfi ("Basis", "sql_not") => ((L'.EPrim (Prim.String "NOT"), loc), fm)
adamc@264:
adamc@264: | L.ECApp (
adamc@264: (L.ECApp (
adamc@264: (L.ECApp (
adamc@264: (L.ECApp (
adamc@264: (L.ECApp (
adamc@254: (L.ECApp (
adamc@254: (L.EFfi ("Basis", "sql_binary"), _),
adamc@254: _), _),
adamc@254: _), _),
adamc@254: _), _),
adamc@254: _), _),
adamc@254: _), _),
adamc@254: _) =>
adamc@254: let
adamc@254: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@254: fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@254: in
adamc@254: ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@254: (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@254: (L'.EAbs ("e2", s, s,
adamc@254: strcat loc [sc "(",
adamc@254: (L'.ERel 1, loc),
adamc@254: sc " ",
adamc@254: (L'.ERel 2, loc),
adamc@254: sc " ",
adamc@254: (L'.ERel 0, loc),
adamc@254: sc ")"]), loc)), loc)), loc),
adamc@254: fm)
adamc@254: end
adamc@254: | L.EFfi ("Basis", "sql_and") => ((L'.EPrim (Prim.String "AND"), loc), fm)
adamc@254: | L.EFfi ("Basis", "sql_or") => ((L'.EPrim (Prim.String "OR"), loc), fm)
adamc@254:
adamc@254: | L.ECApp (
adamc@254: (L.ECApp (
adamc@254: (L.ECApp (
adamc@254: (L.ECApp (
adamc@253: (L.EFfi ("Basis", "sql_comparison"), _),
adamc@253: _), _),
adamc@253: _), _),
adamc@253: _), _),
adamc@253: _) =>
adamc@253: let
adamc@253: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@253: fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@253: in
adamc@253: ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@253: (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@253: (L'.EAbs ("e2", s, s,
adamc@254: strcat loc [sc "(",
adamc@254: (L'.ERel 1, loc),
adamc@253: sc " ",
adamc@253: (L'.ERel 2, loc),
adamc@253: sc " ",
adamc@254: (L'.ERel 0, loc),
adamc@254: sc ")"]), loc)), loc)), loc),
adamc@253: fm)
adamc@253: end
adamc@253:
adamc@253: | L.ECApp (
adamc@253: (L.ECApp (
adamc@253: (L.ECApp (
adamc@253: (L.ECApp (
adamc@253: (L.ECApp (
adamc@253: (L.ECApp (
adamc@253: (L.ECApp (
adamc@253: (L.EFfi ("Basis", "sql_field"), _),
adamc@253: _), _),
adamc@253: _), _),
adamc@253: _), _),
adamc@253: _), _),
adamc@253: _), _),
adamc@253: (L.CName tab, _)), _),
adamc@277: (L.CName field, _)) => ((L'.EPrim (Prim.String (tab ^ ".lw_" ^ field)), loc), fm)
adamc@260:
adamc@260: | L.ECApp (
adamc@260: (L.ECApp (
adamc@260: (L.ECApp (
adamc@260: (L.ECApp (
adamc@261: (L.ECApp (
adamc@261: (L.EFfi ("Basis", "sql_exp"), _),
adamc@261: _), _),
adamc@261: _), _),
adamc@261: _), _),
adamc@261: _), _),
adamc@261: (L.CName nm, _)) => ((L'.EPrim (Prim.String ("_" ^ nm)), loc), fm)
adamc@261:
adamc@261: | L.ECApp (
adamc@261: (L.ECApp (
adamc@261: (L.ECApp (
adamc@261: (L.ECApp (
adamc@260: (L.EFfi ("Basis", "sql_relop"), _),
adamc@260: _), _),
adamc@260: _), _),
adamc@260: _), _),
adamc@260: _) =>
adamc@260: let
adamc@260: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@260: fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@260: in
adamc@260: ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@260: (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@260: (L'.EAbs ("e2", s, s,
adamc@260: strcat loc [sc "((",
adamc@260: (L'.ERel 1, loc),
adamc@260: sc ") ",
adamc@260: (L'.ERel 2, loc),
adamc@260: sc " (",
adamc@260: (L'.ERel 0, loc),
adamc@260: sc "))"]), loc)), loc)), loc),
adamc@260: fm)
adamc@260: end
adamc@260:
adamc@260: | L.EFfi ("Basis", "sql_union") => ((L'.EPrim (Prim.String "UNION"), loc), fm)
adamc@260: | L.EFfi ("Basis", "sql_intersect") => ((L'.EPrim (Prim.String "INTERSECT"), loc), fm)
adamc@260: | L.EFfi ("Basis", "sql_except") => ((L'.EPrim (Prim.String "EXCEPT"), loc), fm)
adamc@260:
adamc@265: | L.ECApp (
adamc@265: (L.ECApp (
adamc@265: (L.ECApp (
adamc@265: (L.EFfi ("Basis", "sql_count"), _),
adamc@265: _), _),
adamc@265: _), _),
adamc@265: _) => ((L'.EAbs ("_", (L'.TRecord [], loc), (L'.TFfi ("Basis", "string"), loc),
adamc@265: (L'.EPrim (Prim.String "COUNT(*)"), loc)), loc),
adamc@265: fm)
adamc@266:
adamc@266: | L.ECApp (
adamc@266: (L.ECApp (
adamc@266: (L.ECApp (
adamc@266: (L.ECApp (
adamc@266: (L.EFfi ("Basis", "sql_aggregate"), _),
adamc@266: _), _),
adamc@266: _), _),
adamc@266: _), _),
adamc@266: _) =>
adamc@266: let
adamc@266: val s = (L'.TFfi ("Basis", "string"), loc)
adamc@266: fun sc s = (L'.EPrim (Prim.String s), loc)
adamc@266: in
adamc@266: ((L'.EAbs ("c", s, (L'.TFun (s, (L'.TFun (s, s), loc)), loc),
adamc@266: (L'.EAbs ("e1", s, (L'.TFun (s, s), loc),
adamc@266: strcat loc [(L'.ERel 1, loc),
adamc@266: sc "(",
adamc@266: (L'.ERel 0, loc),
adamc@266: sc ")"]), loc)), loc),
adamc@266: fm)
adamc@266: end
adamc@266:
adamc@266: | L.EFfi ("Basis", "sql_summable_int") => ((L'.ERecord [], loc), fm)
adamc@266: | L.EFfi ("Basis", "sql_summable_float") => ((L'.ERecord [], loc), fm)
adamc@266: | L.ECApp ((L.EFfi ("Basis", "sql_avg"), _), _) =>
adamc@266: ((L'.EAbs ("_", (L'.TRecord [], loc), (L'.TFfi ("Basis", "string"), loc),
adamc@266: (L'.EPrim (Prim.String "AVG"), loc)), loc),
adamc@266: fm)
adamc@266: | L.ECApp ((L.EFfi ("Basis", "sql_sum"), _), _) =>
adamc@266: ((L'.EAbs ("_", (L'.TRecord [], loc), (L'.TFfi ("Basis", "string"), loc),
adamc@266: (L'.EPrim (Prim.String "SUM"), loc)), loc),
adamc@266: fm)
adamc@266:
adamc@266: | L.EFfi ("Basis", "sql_maxable_int") => ((L'.ERecord [], loc), fm)
adamc@266: | L.EFfi ("Basis", "sql_maxable_float") => ((L'.ERecord [], loc), fm)
adamc@266: | L.EFfi ("Basis", "sql_maxable_string") => ((L'.ERecord [], loc), fm)
adamc@266: | L.ECApp ((L.EFfi ("Basis", "sql_max"), _), _) =>
adamc@266: ((L'.EAbs ("_", (L'.TRecord [], loc), (L'.TFfi ("Basis", "string"), loc),
adamc@266: (L'.EPrim (Prim.String "MAX"), loc)), loc),
adamc@266: fm)
adamc@266: | L.ECApp ((L.EFfi ("Basis", "sql_min"), _), _) =>
adamc@266: ((L'.EAbs ("_", (L'.TRecord [], loc), (L'.TFfi ("Basis", "string"), loc),
adamc@266: (L'.EPrim (Prim.String "MIN"), loc)), loc),
adamc@266: fm)
adamc@266:
adamc@268: | L.EFfi ("Basis", "sql_asc") => ((L'.EPrim (Prim.String ""), loc), fm)
adamc@268: | L.EFfi ("Basis", "sql_desc") => ((L'.EPrim (Prim.String " DESC"), loc), fm)
adamc@268:
adamc@139: | L.EApp (
adamc@139: (L.ECApp (
adamc@141: (L.ECApp ((L.EFfi ("Basis", "cdata"), _), _), _),
adamc@139: _), _),
adamc@179: se) =>
adamc@179: let
adamc@179: val (se, fm) = monoExp (env, st, fm) se
adamc@179: in
adamc@179: ((L'.EFfiApp ("Basis", "htmlifyString", [se]), loc), fm)
adamc@179: end
adamc@179:
adamc@95: | L.EApp (
adamc@95: (L.EApp (
adamc@95: (L.ECApp (
adamc@95: (L.ECApp (
adamc@95: (L.ECApp (
adamc@139: (L.ECApp (
adamc@140: (L.EFfi ("Basis", "join"),
adamc@139: _), _), _),
adamc@139: _), _),
adamc@95: _), _),
adamc@95: _), _),
adamc@95: xml1), _),
adamc@179: xml2) =>
adamc@179: let
adamc@179: val (xml1, fm) = monoExp (env, st, fm) xml1
adamc@179: val (xml2, fm) = monoExp (env, st, fm) xml2
adamc@179: in
adamc@179: ((L'.EStrcat (xml1, xml2), loc), fm)
adamc@179: end
adamc@95:
adamc@95: | L.EApp (
adamc@95: (L.EApp (
adamc@104: (L.EApp (
adamc@95: (L.ECApp (
adamc@104: (L.ECApp (
adamc@104: (L.ECApp (
adamc@104: (L.ECApp (
adamc@139: (L.ECApp (
adamc@139: (L.ECApp (
adamc@139: (L.ECApp (
adamc@139: (L.ECApp (
adamc@139: (L.EFfi ("Basis", "tag"),
adamc@139: _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _),
adamc@104: attrs), _),
adamc@95: tag), _),
adamc@95: xml) =>
adamc@95: let
adamc@140: fun getTag' (e, _) =
adamc@140: case e of
adamc@143: L.EFfi ("Basis", tag) => (tag, [])
adamc@143: | L.ECApp (e, t) => let
adamc@143: val (tag, ts) = getTag' e
adamc@143: in
adamc@143: (tag, ts @ [t])
adamc@143: end
adamc@140: | _ => (E.errorAt loc "Non-constant XML tag";
adamc@140: Print.eprefaces' [("Expression", CorePrint.p_exp env tag)];
adamc@143: ("", []))
adamc@140:
adamc@95: fun getTag (e, _) =
adamc@95: case e of
adamc@143: L.EFfiApp ("Basis", tag, [(L.ERecord [], _)]) => (tag, [])
adamc@140: | L.EApp (e, (L.ERecord [], _)) => getTag' e
adamc@95: | _ => (E.errorAt loc "Non-constant XML tag";
adamc@95: Print.eprefaces' [("Expression", CorePrint.p_exp env tag)];
adamc@143: ("", []))
adamc@95:
adamc@143: val (tag, targs) = getTag tag
adamc@95:
adamc@179: val (attrs, fm) = monoExp (env, st, fm) attrs
adamc@104:
adamc@143: fun tagStart tag =
adamc@104: case #1 attrs of
adamc@104: L'.ERecord xes =>
adamc@104: let
adamc@104: fun lowercaseFirst "" = ""
adamc@143: | lowercaseFirst s = str (Char.toLower (String.sub (s, 0)))
adamc@143: ^ String.extract (s, 1, NONE)
adamc@104:
adamc@104: val s = (L'.EPrim (Prim.String (String.concat ["<", tag])), loc)
adamc@104: in
adamc@179: foldl (fn ((x, e, t), (s, fm)) =>
adamc@104: let
adamc@104: val xp = " " ^ lowercaseFirst x ^ "=\""
adamc@120:
adamc@120: val fooify =
adamc@120: case x of
adamc@185: "Href" => urlifyExp
adamc@185: | "Link" => urlifyExp
adamc@143: | "Action" => urlifyExp
adamc@120: | _ => attrifyExp
adamc@179:
adamc@179: val (e, fm) = fooify env fm (e, t)
adamc@104: in
adamc@179: ((L'.EStrcat (s,
adamc@179: (L'.EStrcat ((L'.EPrim (Prim.String xp), loc),
adamc@179: (L'.EStrcat (e,
adamc@179: (L'.EPrim (Prim.String "\""),
adamc@179: loc)),
adamc@179: loc)),
adamc@179: loc)), loc),
adamc@179: fm)
adamc@104: end)
adamc@179: (s, fm) xes
adamc@104: end
adamc@143: | _ => raise Fail "Non-record attributes!"
adamc@104:
adamc@143: fun input typ =
adamc@143: case targs of
adamc@155: [_, (L.CName name, _)] =>
adamc@179: let
adamc@179: val (ts, fm) = tagStart "input"
adamc@179: in
adamc@179: ((L'.EStrcat (ts,
adamc@179: (L'.EPrim (Prim.String (" type=\"" ^ typ ^ "\" name=\"" ^ name ^ "\"/>")),
adamc@179: loc)), loc), fm)
adamc@179: end
adamc@143: | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153: raise Fail "No name passed to input tag")
adamc@104:
adamc@152: fun normal (tag, extra) =
adamc@143: let
adamc@179: val (tagStart, fm) = tagStart tag
adamc@152: val tagStart = case extra of
adamc@152: NONE => tagStart
adamc@152: | SOME extra => (L'.EStrcat (tagStart, extra), loc)
adamc@152:
adamc@143: fun normal () =
adamc@179: let
adamc@179: val (xml, fm) = monoExp (env, st, fm) xml
adamc@179: in
adamc@179: ((L'.EStrcat ((L'.EStrcat (tagStart, (L'.EPrim (Prim.String ">"), loc)), loc),
adamc@179: (L'.EStrcat (xml,
adamc@179: (L'.EPrim (Prim.String (String.concat [""])),
adamc@179: loc)), loc)),
adamc@179: loc),
adamc@179: fm)
adamc@179: end
adamc@143: in
adamc@143: case xml of
adamc@143: (L.EApp ((L.ECApp (
adamc@143: (L.ECApp ((L.EFfi ("Basis", "cdata"), _),
adamc@143: _), _),
adamc@143: _), _),
adamc@143: (L.EPrim (Prim.String s), _)), _) =>
adamc@143: if CharVector.all Char.isSpace s then
adamc@179: ((L'.EStrcat (tagStart, (L'.EPrim (Prim.String "/>"), loc)), loc), fm)
adamc@143: else
adamc@143: normal ()
adamc@143: | _ => normal ()
adamc@143: end
adamc@152: in
adamc@152: case tag of
adamc@179: "submit" => ((L'.EPrim (Prim.String ""), loc), fm)
adamc@152:
adamc@152: | "textbox" =>
adamc@152: (case targs of
adamc@152: [_, (L.CName name, _)] =>
adamc@179: let
adamc@179: val (ts, fm) = tagStart "input"
adamc@179: in
adamc@179: ((L'.EStrcat (ts,
adamc@179: (L'.EPrim (Prim.String (" name=\"" ^ name ^ "\"/>")),
adamc@179: loc)), loc), fm)
adamc@179: end
adamc@152: | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153: raise Fail "No name passed to textarea tag"))
adamc@155: | "password" => input "password"
adamc@152: | "ltextarea" =>
adamc@152: (case targs of
adamc@152: [_, (L.CName name, _)] =>
adamc@179: let
adamc@179: val (ts, fm) = tagStart "textarea"
adamc@179: val (xml, fm) = monoExp (env, st, fm) xml
adamc@179: in
adamc@179: ((L'.EStrcat ((L'.EStrcat (ts,
adamc@179: (L'.EPrim (Prim.String (" name=\"" ^ name ^ "\">")), loc)), loc),
adamc@179: (L'.EStrcat (xml,
adamc@179: (L'.EPrim (Prim.String ""),
adamc@179: loc)), loc)),
adamc@179: loc), fm)
adamc@179: end
adamc@152: | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153: raise Fail "No name passed to ltextarea tag"))
adamc@153:
adamc@190: | "checkbox" => input "checkbox"
adamc@190:
adamc@153: | "radio" =>
adamc@153: (case targs of
adamc@153: [_, (L.CName name, _)] =>
adamc@179: monoExp (env, St.setRadioGroup (st, name), fm) xml
adamc@153: | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@153: raise Fail "No name passed to radio tag"))
adamc@153: | "radioOption" =>
adamc@153: (case St.radioGroup st of
adamc@153: NONE => raise Fail "No name for radioGroup"
adamc@153: | SOME name =>
adamc@153: normal ("input",
adamc@153: SOME (L'.EPrim (Prim.String (" type=\"radio\" name=\"" ^ name ^ "\"")), loc)))
adamc@152:
adamc@154: | "lselect" =>
adamc@154: (case targs of
adamc@154: [_, (L.CName name, _)] =>
adamc@179: let
adamc@179: val (ts, fm) = tagStart "select"
adamc@179: val (xml, fm) = monoExp (env, st, fm) xml
adamc@179: in
adamc@179: ((L'.EStrcat ((L'.EStrcat (ts,
adamc@179: (L'.EPrim (Prim.String (" name=\"" ^ name ^ "\">")), loc)), loc),
adamc@179: (L'.EStrcat (xml,
adamc@179: (L'.EPrim (Prim.String ""),
adamc@179: loc)), loc)),
adamc@179: loc),
adamc@179: fm)
adamc@179: end
adamc@154: | _ => (Print.prefaces "Targs" (map (fn t => ("T", CorePrint.p_con env t)) targs);
adamc@154: raise Fail "No name passed to lselect tag"))
adamc@154:
adamc@154: | "loption" => normal ("option", NONE)
adamc@154:
adamc@152: | _ => normal (tag, NONE)
adamc@95: end
adamc@94:
adamc@141: | L.EApp ((L.ECApp (
adamc@141: (L.ECApp ((L.EFfi ("Basis", "lform"), _), _), _),
adamc@141: _), _),
adamc@141: xml) =>
adamc@143: let
adamc@143: fun findSubmit (e, _) =
adamc@143: case e of
adamc@143: L.EApp (
adamc@143: (L.EApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.EFfi ("Basis", "join"),
adamc@143: _), _), _),
adamc@143: _), _),
adamc@143: _), _),
adamc@143: _), _),
adamc@143: xml1), _),
adamc@143: xml2) => (case findSubmit xml1 of
adamc@143: Error => Error
adamc@143: | NotFound => findSubmit xml2
adamc@143: | Found e =>
adamc@143: case findSubmit xml2 of
adamc@143: NotFound => Found e
adamc@143: | _ => Error)
adamc@143: | L.EApp (
adamc@143: (L.EApp (
adamc@143: (L.EApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.ECApp (
adamc@143: (L.EFfi ("Basis", "tag"),
adamc@143: _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _), _),
adamc@143: attrs), _),
adamc@143: _), _),
adamc@143: xml) =>
adamc@143: (case #1 attrs of
adamc@143: L.ERecord xes =>
adamc@143: (case ListUtil.search (fn ((L.CName "Action", _), e, t) => SOME (e, t)
adamc@143: | _ => NONE) xes of
adamc@143: NONE => findSubmit xml
adamc@143: | SOME et =>
adamc@143: case findSubmit xml of
adamc@143: NotFound => Found et
adamc@143: | _ => Error)
adamc@143: | _ => findSubmit xml)
adamc@143: | _ => NotFound
adamc@143:
adamc@143: val (action, actionT) = case findSubmit xml of
adamc@143: NotFound => raise Fail "No submit found"
adamc@143: | Error => raise Fail "Not ready for multi-submit lforms yet"
adamc@143: | Found et => et
adamc@143:
adamc@143: val actionT = monoType env actionT
adamc@179: val (action, fm) = monoExp (env, st, fm) action
adamc@179: val (action, fm) = urlifyExp env fm (action, actionT)
adamc@179: val (xml, fm) = monoExp (env, st, fm) xml
adamc@143: in
adamc@179: ((L'.EStrcat ((L'.EStrcat ((L'.EPrim (Prim.String ""), loc)), loc)), loc),
adamc@179: fm)
adamc@143: end
adamc@141:
adamc@148: | L.EApp ((L.ECApp (
adamc@148: (L.ECApp (
adamc@148: (L.ECApp (
adamc@148: (L.ECApp (
adamc@148: (L.EFfi ("Basis", "useMore"), _), _), _),
adamc@148: _), _),
adamc@148: _), _),
adamc@148: _), _),
adamc@179: xml) => monoExp (env, st, fm) xml
adamc@148:
adamc@179: | L.EApp (e1, e2) =>
adamc@179: let
adamc@179: val (e1, fm) = monoExp (env, st, fm) e1
adamc@179: val (e2, fm) = monoExp (env, st, fm) e2
adamc@179: in
adamc@179: ((L'.EApp (e1, e2), loc), fm)
adamc@179: end
adamc@26: | L.EAbs (x, dom, ran, e) =>
adamc@179: let
adamc@179: val (e, fm) = monoExp (Env.pushERel env x dom, st, fm) e
adamc@179: in
adamc@179: ((L'.EAbs (x, monoType env dom, monoType env ran, e), loc), fm)
adamc@179: end
adamc@25: | L.ECApp _ => poly ()
adamc@25: | L.ECAbs _ => poly ()
adamc@25:
adamc@252: | L.EFfi mx => ((L'.EFfi mx, loc), fm)
adamc@252: | L.EFfiApp (m, x, es) =>
adamc@252: let
adamc@252: val (es, fm) = ListUtil.foldlMap (fn (e, fm) => monoExp (env, st, fm) e) fm es
adamc@252: in
adamc@252: ((L'.EFfiApp (m, x, es), loc), fm)
adamc@252: end
adamc@252:
adamc@179: | L.ERecord xes =>
adamc@179: let
adamc@179: val (xes, fm) = ListUtil.foldlMap
adamc@179: (fn ((x, e, t), fm) =>
adamc@179: let
adamc@179: val (e, fm) = monoExp (env, st, fm) e
adamc@179: in
adamc@179: ((monoName env x,
adamc@179: e,
adamc@179: monoType env t), fm)
adamc@179: end) fm xes
adamc@179: in
adamc@179: ((L'.ERecord xes, loc), fm)
adamc@179: end
adamc@179: | L.EField (e, x, _) =>
adamc@179: let
adamc@179: val (e, fm) = monoExp (env, st, fm) e
adamc@179: in
adamc@179: ((L'.EField (e, monoName env x), loc), fm)
adamc@179: end
adamc@149: | L.ECut _ => poly ()
adamc@73: | L.EFold _ => poly ()
adamc@177:
adamc@182: | L.ECase (e, pes, {disc, result}) =>
adamc@179: let
adamc@179: val (e, fm) = monoExp (env, st, fm) e
adamc@179: val (pes, fm) = ListUtil.foldlMap
adamc@179: (fn ((p, e), fm) =>
adamc@179: let
adamc@179: val (e, fm) = monoExp (env, st, fm) e
adamc@179: in
adamc@182: ((monoPat env p, e), fm)
adamc@179: end) fm pes
adamc@179: in
adamc@182: ((L'.ECase (e, pes, {disc = monoType env disc, result = monoType env result}), loc), fm)
adamc@179: end
adamc@177:
adamc@179: | L.EWrite e =>
adamc@179: let
adamc@179: val (e, fm) = monoExp (env, st, fm) e
adamc@179: in
adamc@252: ((L'.EAbs ("_", (L'.TRecord [], loc), (L'.TRecord [], loc),
adamc@252: (L'.EWrite (liftExpInExp 0 e), loc)), loc), fm)
adamc@179: end
adamc@110:
adamc@179: | L.EClosure (n, es) =>
adamc@179: let
adamc@179: val (es, fm) = ListUtil.foldlMap (fn (e, fm) =>
adamc@179: monoExp (env, st, fm) e)
adamc@179: fm es
adamc@179: in
adamc@179: ((L'.EClosure (n, es), loc), fm)
adamc@179: end
adamc@25: end
adamc@25:
adamc@179: fun monoDecl (env, fm) (all as (d, loc)) =
adamc@25: let
adamc@25: fun poly () =
adamc@25: (E.errorAt loc "Unsupported declaration";
adamc@25: Print.eprefaces' [("Declaration", CorePrint.p_decl env all)];
adamc@25: NONE)
adamc@25: in
adamc@25: case d of
adamc@25: L.DCon _ => NONE
adamc@193: | L.DDatatype (x, n, [], xncs) =>
adamc@193: let
adamc@196: val env' = Env.declBinds env all
adamc@196: val d = (L'.DDatatype (x, n, map (fn (x, n, to) => (x, n, Option.map (monoType env') to)) xncs), loc)
adamc@164: in
adamc@273: SOME (env', fm, [d])
adamc@193: end
adamc@193: | L.DDatatype _ => poly ()
adamc@179: | L.DVal (x, n, t, e, s) =>
adamc@179: let
adamc@179: val (e, fm) = monoExp (env, St.empty, fm) e
adamc@179: in
adamc@179: SOME (Env.pushENamed env x n t NONE s,
adamc@179: fm,
adamc@273: [(L'.DVal (x, n, monoType env t, e, s), loc)])
adamc@179: end
adamc@128: | L.DValRec vis =>
adamc@128: let
adamc@128: val env = foldl (fn ((x, n, t, e, s), env) => Env.pushENamed env x n t NONE s) env vis
adamc@179:
adamc@179: val (vis, fm) = ListUtil.foldlMap
adamc@179: (fn ((x, n, t, e, s), fm) =>
adamc@179: let
adamc@179: val (e, fm) = monoExp (env, St.empty, fm) e
adamc@179: in
adamc@179: ((x, n, monoType env t, e, s), fm)
adamc@179: end)
adamc@179: fm vis
adamc@128: in
adamc@128: SOME (env,
adamc@179: fm,
adamc@273: [(L'.DValRec vis, loc)])
adamc@128: end
adamc@144: | L.DExport (ek, n) =>
adamc@115: let
adamc@120: val (_, t, _, s) = Env.lookupENamed env n
adamc@120:
adamc@120: fun unwind (t, _) =
adamc@120: case t of
adamc@120: L.TFun (dom, ran) => dom :: unwind ran
adamc@120: | _ => []
adamc@120:
adamc@120: val ts = map (monoType env) (unwind t)
adamc@115: in
adamc@273: SOME (env, fm, [(L'.DExport (ek, s, n, ts), loc)])
adamc@115: end
adamc@273: | L.DTable (x, n, (L.CRecord (_, xts), _), s) =>
adamc@251: let
adamc@251: val t = (L.CFfi ("Basis", "string"), loc)
adamc@251: val t' = (L'.TFfi ("Basis", "string"), loc)
adamc@274: val s = "lw_" ^ s
adamc@251: val e = (L'.EPrim (Prim.String s), loc)
adamc@273:
adamc@273: val xts = map (fn (x, t) => (monoName env x, monoType env t)) xts
adamc@251: in
adamc@251: SOME (Env.pushENamed env x n t NONE s,
adamc@251: fm,
adamc@273: [(L'.DTable (s, xts), loc),
adamc@273: (L'.DVal (x, n, t', e, s), loc)])
adamc@251: end
adamc@273: | L.DTable _ => poly ()
adamc@273: | L.DDatabase s => SOME (env, fm, [(L'.DDatabase s, loc)])
adamc@25: end
adamc@25:
adamc@25: fun monoize env ds =
adamc@25: let
adamc@179: val (_, _, ds) = List.foldl (fn (d, (env, fm, ds)) =>
adamc@179: case monoDecl (env, fm) d of
adamc@179: NONE => (env, fm, ds)
adamc@273: | SOME (env, fm, ds') =>
adamc@179: (env,
adamc@179: Fm.enter fm,
adamc@273: ds' @ Fm.decls fm @ ds))
adamc@179: (env, Fm.empty (CoreUtil.File.maxName ds + 1), []) ds
adamc@25: in
adamc@25: rev ds
adamc@25: end
adamc@25:
adamc@25: end