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@25: val dummyTyp = (L'.TNamed 0, E.dummySpan) adamc@25: 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@25: fun monoType env (all as (c, loc)) = adamc@25: let adamc@25: fun poly () = adamc@25: (E.errorAt loc "Unsupported type constructor"; adamc@25: Print.eprefaces' [("Constructor", CorePrint.p_con env all)]; adamc@25: dummyTyp) adamc@25: in adamc@25: case c of adamc@25: L.TFun (c1, c2) => (L'.TFun (monoType env c1, monoType env c2), loc) adamc@25: | L.TCFun _ => poly () adamc@25: | L.TRecord (L.CRecord ((L.KType, _), xcs), _) => adamc@25: (L'.TRecord (map (fn (x, t) => (monoName env x, monoType env t)) xcs), loc) adamc@25: | L.TRecord _ => poly () adamc@25: adamc@94: | L.CApp ((L.CFfi ("Basis", "xml"), _), _) => (L'.TFfi ("Basis", "string"), loc) adamc@94: adamc@25: | L.CRel _ => poly () adamc@25: | L.CNamed n => (L'.TNamed n, loc) adamc@51: | L.CFfi mx => (L'.TFfi mx, loc) adamc@25: | L.CApp _ => poly () adamc@25: | L.CAbs _ => poly () adamc@25: adamc@25: | L.CName _ => poly () adamc@25: adamc@25: | L.CRecord _ => poly () adamc@25: | L.CConcat _ => poly () adamc@69: | L.CFold _ => poly () adamc@87: | L.CUnit => poly () adamc@25: end adamc@25: adamc@25: val dummyExp = (L'.EPrim (Prim.Int 0), E.dummySpan) adamc@25: adamc@120: fun fooifyExp name env = adamc@120: let adamc@120: fun fooify (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@120: (L'.EPrim (Prim.String s), loc) 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@120: fun attrify (args, ft, e) = adamc@120: case (args, ft) of adamc@120: ([], _) => e adamc@120: | (arg :: args, (L'.TFun (t, ft), _)) => adamc@121: attrify (args, ft, adamc@121: (L'.EStrcat (e, adamc@121: (L'.EStrcat ((L'.EPrim (Prim.String "/"), loc), adamc@121: fooify (arg, t)), loc)), loc)) adamc@120: | _ => (E.errorAt loc "Type mismatch encoding attribute"; adamc@120: e) adamc@120: in adamc@120: attrify (args, ft, (L'.EPrim (Prim.String s), loc)) adamc@120: end adamc@120: | _ => adamc@120: case t of adamc@120: L'.TFfi ("Basis", "string") => (L'.EFfiApp ("Basis", name ^ "ifyString", [e]), loc) adamc@120: | L'.TFfi ("Basis", "int") => (L'.EFfiApp ("Basis", name ^ "ifyInt", [e]), loc) adamc@120: | L'.TFfi ("Basis", "float") => (L'.EFfiApp ("Basis", name ^ "ifyFloat", [e]), loc) adamc@120: | L'.TRecord [] => (L'.EPrim (Prim.String ""), loc) adamc@111: 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@120: dummyExp) adamc@120: in adamc@120: fooify adamc@120: end adamc@120: adamc@120: val attrifyExp = fooifyExp "attr" adamc@120: val urlifyExp = fooifyExp "url" adamc@105: adamc@25: fun monoExp env (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@25: dummyExp) adamc@25: in adamc@25: case e of adamc@25: L.EPrim p => (L'.EPrim p, loc) adamc@25: | L.ERel n => (L'.ERel n, loc) adamc@25: | L.ENamed n => (L'.ENamed n, loc) adamc@51: | L.EFfi mx => (L'.EFfi mx, loc) adamc@51: | L.EFfiApp (m, x, es) => (L'.EFfiApp (m, x, map (monoExp env) es), loc) adamc@94: adamc@139: | L.EApp ( adamc@139: (L.ECApp ( adamc@139: (L.ECApp ( adamc@139: (L.ECApp ((L.EFfi ("Basis", "cdata"), _), _), _), adamc@139: _), _), adamc@139: _), _), adamc@139: se) => (L'.EFfiApp ("Basis", "htmlifyString", [monoExp env se]), loc) 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@139: (L.ECApp ( adamc@139: (L.ECApp ( adamc@139: (L.ECApp ( adamc@139: (L.EFfi ("Basis", "join"), adamc@139: _), _), _), adamc@139: _), _), adamc@139: _), _), adamc@139: _), _), adamc@139: _), _), adamc@95: _), _), adamc@95: _), _), adamc@95: xml1), _), adamc@95: xml2) => (L'.EStrcat (monoExp env xml1, monoExp env xml2), loc) 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@95: fun getTag (e, _) = adamc@95: case e of adamc@95: L.EFfi ("Basis", tag) => tag adamc@95: | _ => (E.errorAt loc "Non-constant XML tag"; adamc@95: Print.eprefaces' [("Expression", CorePrint.p_exp env tag)]; adamc@95: "") adamc@95: adamc@95: val tag = getTag tag adamc@95: adamc@104: val attrs = monoExp env attrs adamc@104: adamc@104: val tagStart = adamc@104: case #1 attrs of adamc@104: L'.ERecord xes => adamc@104: let adamc@104: fun lowercaseFirst "" = "" adamc@104: | lowercaseFirst s = str (Char.toLower (String.sub (s, 0))) ^ String.extract (s, 1, NONE) adamc@104: adamc@104: val s = (L'.EPrim (Prim.String (String.concat ["<", tag])), loc) adamc@104: in adamc@105: foldl (fn ((x, e, t), s) => adamc@104: let adamc@104: val xp = " " ^ lowercaseFirst x ^ "=\"" adamc@120: adamc@120: val fooify = adamc@120: case x of adamc@120: "Link" => urlifyExp adamc@120: | _ => attrifyExp adamc@104: in adamc@104: (L'.EStrcat (s, adamc@104: (L'.EStrcat ((L'.EPrim (Prim.String xp), loc), adamc@120: (L'.EStrcat (fooify env (e, t), adamc@104: (L'.EPrim (Prim.String "\""), loc)), adamc@104: loc)), adamc@104: loc)), loc) adamc@104: end) adamc@104: s xes adamc@104: end adamc@104: | _ => raise Fail "Attributes!" adamc@104: adamc@95: fun normal () = adamc@104: (L'.EStrcat ((L'.EStrcat (tagStart, (L'.EPrim (Prim.String ">"), loc)), loc), adamc@95: (L'.EStrcat (monoExp env xml, adamc@95: (L'.EPrim (Prim.String (String.concat [""])), loc)), loc)), adamc@95: loc) adamc@104: adamc@104: adamc@95: in adamc@95: case xml of adamc@95: (L.EApp ((L.ECApp ((L.EFfi ("Basis", "cdata"), _), adamc@95: _), _), (L.EPrim (Prim.String s), _)), _) => adamc@95: if CharVector.all Char.isSpace s then adamc@104: (L'.EStrcat (tagStart, (L'.EPrim (Prim.String "/>"), loc)), loc) adamc@95: else adamc@95: normal () adamc@95: | _ => normal () adamc@95: end adamc@94: adamc@25: | L.EApp (e1, e2) => (L'.EApp (monoExp env e1, monoExp env e2), loc) adamc@26: | L.EAbs (x, dom, ran, e) => adamc@26: (L'.EAbs (x, monoType env dom, monoType env ran, monoExp (Env.pushERel env x dom) e), loc) adamc@25: | L.ECApp _ => poly () adamc@25: | L.ECAbs _ => poly () adamc@25: adamc@29: | L.ERecord xes => (L'.ERecord (map (fn (x, e, t) => (monoName env x, monoExp env e, monoType env t)) xes), loc) adamc@25: | L.EField (e, x, _) => (L'.EField (monoExp env e, monoName env x), loc) adamc@73: | L.EFold _ => poly () adamc@102: | L.EWrite e => (L'.EWrite (monoExp env e), loc) adamc@110: adamc@111: | L.EClosure (n, es) => (L'.EClosure (n, map (monoExp env) es), loc) adamc@25: end adamc@25: adamc@25: fun monoDecl env (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@109: | L.DVal (x, n, t, e, s) => SOME (Env.pushENamed env x n t (SOME e) s, adamc@109: (L'.DVal (x, n, monoType env t, monoExp env e, s), loc)) 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@128: in adamc@128: SOME (env, adamc@128: (L'.DValRec (map (fn (x, n, t, e, s) => (x, n, monoType env t, monoExp env e, s)) vis), loc)) adamc@128: end adamc@115: | L.DExport 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@120: SOME (env, (L'.DExport (s, n, ts), loc)) adamc@115: end adamc@25: end adamc@25: adamc@25: fun monoize env ds = adamc@25: let adamc@25: val (_, ds) = List.foldl (fn (d, (env, ds)) => adamc@25: case monoDecl env d of adamc@25: NONE => (env, ds) adamc@25: | SOME (env, d) => (env, d :: ds)) (env, []) ds adamc@25: in adamc@25: rev ds adamc@25: end adamc@25: adamc@25: end