annotate src/compiler.sml @ 1181:618f9f458da9

Got split1 working, but noticed a nasty type inference bug with transplanted unification variables
author Adam Chlipala <adamc@hcoop.net>
date Sat, 06 Mar 2010 19:14:48 -0500
parents 983d9b38abc7
children 9d3ccb8b39ac
rev   line source
adamc@1132 1 (* Copyright (c) 2008-2010, Adam Chlipala
adamc@1 2 * All rights reserved.
adamc@1 3 *
adamc@1 4 * Redistribution and use in source and binary forms, with or without
adamc@1 5 * modification, are permitted provided that the following conditions are met:
adamc@1 6 *
adamc@1 7 * - Redistributions of source code must retain the above copyright notice,
adamc@1 8 * this list of conditions and the following disclaimer.
adamc@1 9 * - Redistributions in binary form must reproduce the above copyright notice,
adamc@1 10 * this list of conditions and the following disclaimer in the documentation
adamc@1 11 * and/or other materials provided with the distribution.
adamc@1 12 * - The names of contributors may not be used to endorse or promote products
adamc@1 13 * derived from this software without specific prior written permission.
adamc@1 14 *
adamc@1 15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
adamc@1 16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
adamc@1 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
adamc@1 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
adamc@1 19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
adamc@1 20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
adamc@1 21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
adamc@1 22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
adamc@1 23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
adamc@1 24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
adamc@1 25 * POSSIBILITY OF SUCH DAMAGE.
adamc@1 26 *)
adamc@1 27
adamc@1 28 structure Compiler :> COMPILER = struct
adamc@1 29
adamc@244 30 structure UrwebLrVals = UrwebLrValsFn(structure Token = LrParser.Token)
adamc@244 31 structure Lex = UrwebLexFn(structure Tokens = UrwebLrVals.Tokens)
adamc@244 32 structure UrwebP = Join(structure ParserData = UrwebLrVals.ParserData
adamc@1 33 structure Lex = Lex
adamc@1 34 structure LrParser = LrParser)
adamc@1 35
adamc@270 36 type job = {
adamc@385 37 prefix : string,
adamc@270 38 database : string option,
adamc@274 39 sources : string list,
adamc@274 40 exe : string,
adamc@274 41 sql : string option,
adamc@502 42 debug : bool,
adamc@673 43 profile : bool,
adamc@764 44 timeout : int,
adamc@764 45 ffi : string list,
adamc@764 46 link : string list,
adamc@765 47 headers : string list,
adamc@766 48 scripts : string list,
adamc@765 49 clientToServer : Settings.ffi list,
adamc@765 50 effectful : Settings.ffi list,
adamc@1171 51 benignEffectful : Settings.ffi list,
adamc@765 52 clientOnly : Settings.ffi list,
adamc@765 53 serverOnly : Settings.ffi list,
adamc@768 54 jsFuncs : (Settings.ffi * string) list,
adamc@769 55 rewrites : Settings.rewrite list,
adamc@769 56 filterUrl : Settings.rule list,
adamc@866 57 filterMime : Settings.rule list,
adamc@866 58 protocol : string option,
adamc@1164 59 dbms : string option,
adamc@1164 60 sigFile : string option
adamc@270 61 }
adamc@201 62
adamc@201 63 type ('src, 'dst) phase = {
adamc@201 64 func : 'src -> 'dst,
adamc@201 65 print : 'dst -> Print.PD.pp_desc
adamc@201 66 }
adamc@201 67
adamc@201 68 type pmap = (string * Time.time) list
adamc@201 69
adamc@201 70 type ('src, 'dst) transform = {
adamc@201 71 func : 'src -> 'dst option,
adamc@201 72 print : 'dst -> Print.PD.pp_desc,
adamc@201 73 time : 'src * pmap -> 'dst option * pmap
adamc@201 74 }
adamc@201 75
adamc@1079 76 val debug = ref false
adamc@1079 77
adamc@201 78 fun transform (ph : ('src, 'dst) phase) name = {
adamc@201 79 func = fn input => let
adamc@1079 80 val () = if !debug then
adamc@1079 81 print ("Starting " ^ name ^ "....\n")
adamc@1079 82 else
adamc@1079 83 ()
adamc@201 84 val v = #func ph input
adamc@201 85 in
adamc@1079 86 if !debug then
adamc@1079 87 print ("Finished " ^ name ^ ".\n")
adamc@1079 88 else
adamc@1079 89 ();
adamc@201 90 if ErrorMsg.anyErrors () then
adamc@201 91 NONE
adamc@201 92 else
adamc@201 93 SOME v
adamc@201 94 end,
adamc@201 95 print = #print ph,
adamc@201 96 time = fn (input, pmap) => let
adamc@201 97 val befor = Time.now ()
adamc@201 98 val v = #func ph input
adamc@201 99 val elapsed = Time.- (Time.now (), befor)
adamc@201 100 in
adamc@201 101 (if ErrorMsg.anyErrors () then
adamc@201 102 NONE
adamc@201 103 else
adamc@201 104 SOME v,
adamc@201 105 (name, elapsed) :: pmap)
adamc@201 106 end
adamc@201 107 }
adamc@201 108
adamc@346 109 fun check (tr : ('src, 'dst) transform) x = (ErrorMsg.resetErrors ();
adamc@346 110 ignore (#func tr x))
adamc@346 111
adamc@280 112 fun run (tr : ('src, 'dst) transform) x = (ErrorMsg.resetErrors ();
adamc@280 113 #func tr x)
adamc@201 114
adamc@201 115 fun runPrint (tr : ('src, 'dst) transform) input =
adamc@280 116 (ErrorMsg.resetErrors ();
adamc@280 117 case #func tr input of
adamc@280 118 NONE => print "Failure\n"
adamc@280 119 | SOME v =>
adamc@280 120 (print "Success\n";
adamc@280 121 Print.print (#print tr v);
adamc@280 122 print "\n"))
adamc@201 123
adamc@201 124 fun time (tr : ('src, 'dst) transform) input =
adamc@55 125 let
adamc@201 126 val (_, pmap) = #time tr (input, [])
adamc@201 127 in
adamc@201 128 app (fn (name, time) =>
adamc@201 129 print (name ^ ": " ^ LargeReal.toString (Time.toReal time) ^ "\n")) (rev pmap);
adamc@201 130 print ("TOTAL: " ^ LargeReal.toString (Time.toReal (foldl Time.+ Time.zeroTime (map #2 pmap))) ^ "\n");
adamc@201 131 print "\n"
adamc@201 132 end
adamc@55 133
adamc@201 134 fun timePrint (tr : ('src, 'dst) transform) input =
adamc@201 135 let
adamc@201 136 val (ro, pmap) = #time tr (input, [])
adamc@55 137 in
adamc@201 138 app (fn (name, time) =>
adamc@201 139 print (name ^ ": " ^ LargeReal.toString (Time.toReal time) ^ "\n")) (rev pmap);
adamc@201 140 print ("TOTAL: " ^ LargeReal.toString (Time.toReal (foldl Time.+ Time.zeroTime (map #2 pmap))) ^ "\n");
adamc@201 141 print "\n";
adamc@201 142 case ro of
adamc@201 143 NONE => print "Failure\n"
adamc@201 144 | SOME v =>
adamc@201 145 (print "Success\n";
adamc@201 146 Print.print (#print tr v);
adamc@201 147 print "\n")
adamc@55 148 end
adamc@55 149
adamc@244 150 val parseUrs =
adamc@201 151 {func = fn filename => let
adamc@201 152 val fname = OS.FileSys.tmpName ()
adamc@201 153 val outf = TextIO.openOut fname
adamc@201 154 val () = TextIO.output (outf, "sig\n")
adamc@201 155 val inf = TextIO.openIn filename
adamc@201 156 fun loop () =
adamc@201 157 case TextIO.inputLine inf of
adamc@201 158 NONE => ()
adamc@201 159 | SOME line => (TextIO.output (outf, line);
adamc@201 160 loop ())
adamc@201 161 val () = loop ()
adamc@201 162 val () = TextIO.closeIn inf
adamc@201 163 val () = TextIO.closeOut outf
adamc@201 164
adamc@201 165 val () = (ErrorMsg.resetErrors ();
adamc@201 166 ErrorMsg.resetPositioning filename;
adamc@201 167 Lex.UserDeclarations.initialize ())
adamc@201 168 val file = TextIO.openIn fname
adamc@201 169 fun get _ = TextIO.input file
adamc@201 170 fun parseerror (s, p1, p2) = ErrorMsg.errorAt' (p1, p2) s
adamc@201 171 val lexer = LrParser.Stream.streamify (Lex.makeLexer get)
adamc@244 172 val (absyn, _) = UrwebP.parse (30, lexer, parseerror, ())
adamc@201 173 in
adamc@201 174 TextIO.closeIn file;
adamc@201 175 case absyn of
adamc@201 176 [(Source.DSgn ("?", (Source.SgnConst sgis, _)), _)] => sgis
adamc@201 177 | _ => (ErrorMsg.errorAt {file = filename,
adamc@201 178 first = {line = 0,
adamc@201 179 char = 0},
adamc@201 180 last = {line = 0,
adamc@201 181 char = 0}} "Not a signature";
adamc@201 182 [])
adamc@201 183 end
adamc@201 184 handle LrParser.ParseError => [],
adamc@201 185 print = Print.p_list_sep Print.PD.newline SourcePrint.p_sgn_item}
adamc@55 186
adamc@1 187 (* The main parsing routine *)
adamc@244 188 val parseUr = {
adamc@201 189 func = fn filename =>
adamc@201 190 let
adamc@201 191 val () = (ErrorMsg.resetErrors ();
adamc@201 192 ErrorMsg.resetPositioning filename;
adamc@201 193 Lex.UserDeclarations.initialize ())
adamc@201 194 val file = TextIO.openIn filename
adamc@201 195 fun get _ = TextIO.input file
adamc@201 196 fun parseerror (s, p1, p2) = ErrorMsg.errorAt' (p1, p2) s
adamc@201 197 val lexer = LrParser.Stream.streamify (Lex.makeLexer get)
adamc@244 198 val (absyn, _) = UrwebP.parse (30, lexer, parseerror, ())
adamc@201 199 in
adamc@201 200 TextIO.closeIn file;
adamc@201 201 case absyn of
adamc@201 202 [(Source.DSgn ("?", _), _)] =>
adamc@201 203 (ErrorMsg.errorAt {file = filename,
adamc@201 204 first = {line = 0,
adamc@201 205 char = 0},
adamc@201 206 last = {line = 0,
adamc@201 207 char = 0}} "File starts with 'sig'";
adamc@201 208 [])
adamc@201 209 | _ => absyn
adamc@201 210 end
adamc@201 211 handle LrParser.ParseError => [],
adamc@201 212 print = SourcePrint.p_file}
adamc@56 213
adamc@768 214 fun p_job ({prefix, database, exe, sql, sources, debug, profile,
adamc@768 215 timeout, ffi, link, headers, scripts,
adamc@1171 216 clientToServer, effectful, benignEffectful, clientOnly, serverOnly, jsFuncs, ...} : job) =
adamc@270 217 let
adamc@270 218 open Print.PD
adamc@270 219 open Print
adamc@765 220
adamc@765 221 fun p_ffi name = p_list_sep (box []) (fn (m, s) =>
adamc@765 222 box [string name, space, string m, string ".", string s, newline])
adamc@270 223 in
adamc@274 224 box [if debug then
adamc@274 225 box [string "DEBUG", newline]
adamc@274 226 else
adamc@274 227 box [],
adamc@502 228 if profile then
adamc@502 229 box [string "PROFILE", newline]
adamc@502 230 else
adamc@502 231 box [],
adamc@274 232 case database of
adamc@270 233 NONE => string "No database."
adamc@270 234 | SOME db => string ("Database: " ^ db),
adamc@270 235 newline,
adamc@274 236 string "Exe: ",
adamc@274 237 string exe,
adamc@274 238 newline,
adamc@274 239 case sql of
adamc@274 240 NONE => string "No SQL file."
adamc@274 241 | SOME sql => string ("SQL fle: " ^ sql),
adamc@673 242 newline,
adamc@673 243 string "Timeout: ",
adamc@673 244 string (Int.toString timeout),
adamc@673 245 newline,
adamc@764 246 p_list_sep (box []) (fn s => box [string "Ffi", space, string s, newline]) ffi,
adamc@764 247 p_list_sep (box []) (fn s => box [string "Header", space, string s, newline]) headers,
adamc@766 248 p_list_sep (box []) (fn s => box [string "Script", space, string s, newline]) scripts,
adamc@764 249 p_list_sep (box []) (fn s => box [string "Link", space, string s, newline]) link,
adamc@765 250 p_ffi "ClientToServer" clientToServer,
adamc@765 251 p_ffi "Effectful" effectful,
adamc@1171 252 p_ffi "BenignEffectful" benignEffectful,
adamc@765 253 p_ffi "ClientOnly" clientOnly,
adamc@765 254 p_ffi "ServerOnly" serverOnly,
adamc@765 255 p_list_sep (box []) (fn ((m, s), s') =>
adamc@765 256 box [string "JsFunc", space, string m, string ".", string s,
adamc@765 257 space, string "=", space, string s', newline]) jsFuncs,
adamc@270 258 string "Sources:",
adamc@270 259 p_list string sources,
adamc@270 260 newline]
adamc@270 261 end
adamc@270 262
adamc@270 263 fun trim s =
adamc@270 264 let
adamc@270 265 val (_, s) = Substring.splitl Char.isSpace s
adamc@270 266 val (s, _) = Substring.splitr Char.isSpace s
adamc@270 267 in
adamc@270 268 s
adamc@270 269 end
adamc@270 270
adamc@794 271 structure M = BinaryMapFn(struct
adamc@794 272 type ord_key = string
adamc@794 273 val compare = String.compare
adamc@794 274 end)
adamc@794 275
adamc@1089 276 val pathmap = ref (M.insert (M.empty, "", Config.libUr))
adamc@1089 277
adamc@1089 278 fun addPath (k, v) = pathmap := M.insert (!pathmap, k, v)
adamc@1089 279
adamc@1082 280 fun parseUrp' accLibs fname =
adamc@767 281 let
adamc@1089 282 val pathmap = ref (!pathmap)
adamc@1082 283 val bigLibs = ref []
adamc@767 284
adamc@794 285 fun pu filename =
adamc@794 286 let
adamc@794 287 val dir = OS.Path.dir filename
adamc@1151 288 fun opener () = TextIO.openIn (OS.Path.joinBaseExt {base = filename, ext = SOME "urp"})
adamc@1151 289
adamc@1151 290 val inf = opener ()
adamc@1151 291
adamc@1157 292 fun hasSpaceLine () =
adamc@1151 293 case TextIO.inputLine inf of
adamc@1151 294 NONE => false
adamc@1173 295 | SOME s => s = "debug\n" orelse s = "profile\n"
adamc@1173 296 orelse CharVector.exists (fn ch => ch = #" " orelse ch = #"\t") s orelse hasSpaceLine ()
adamc@1151 297
adamc@1157 298 val hasBlankLine = hasSpaceLine ()
adamc@1151 299
adamc@1151 300 val inf = (TextIO.closeIn inf; opener ())
adamc@767 301
adamc@794 302 fun pathify fname =
adamc@794 303 if size fname > 0 andalso String.sub (fname, 0) = #"$" then
adamc@794 304 let
adamc@794 305 val fname' = Substring.extract (fname, 1, NONE)
adamc@794 306 val (befor, after) = Substring.splitl (fn ch => ch <> #"/") fname'
adamc@794 307 in
adamc@794 308 if Substring.isEmpty after then
adamc@794 309 fname
adamc@794 310 else
adamc@794 311 case M.find (!pathmap, Substring.string befor) of
adamc@794 312 NONE => fname
adamc@794 313 | SOME rep => rep ^ Substring.string after
adamc@794 314 end
adamc@794 315 else
adamc@794 316 fname
adamc@767 317
adamc@794 318 fun relify fname =
adamc@794 319 let
adamc@794 320 val fname = pathify fname
adamc@794 321 in
adamc@794 322 OS.Path.concat (dir, fname)
adamc@794 323 handle OS.Path.Path => fname
adamc@794 324 end
adamc@767 325
adamc@1089 326 fun libify path =
adamc@1089 327 (if Posix.FileSys.access (path ^ ".urp", []) then
adamc@1089 328 path
adamc@1089 329 else
adamc@1089 330 path ^ "/lib")
adamc@1089 331 handle SysErr => path
adamc@1089 332
adamc@1089 333 fun libify' path =
adamc@1089 334 (if Posix.FileSys.access (relify path ^ ".urp", []) then
adamc@1089 335 path
adamc@1089 336 else
adamc@1089 337 path ^ "/lib")
adamc@1089 338 handle SysErr => path
adamc@1089 339
adamc@794 340 val absDir = OS.Path.mkAbsolute {path = dir, relativeTo = OS.FileSys.getDir ()}
adamc@767 341
adamc@794 342 fun relifyA fname =
adamc@794 343 OS.Path.mkAbsolute {path = pathify fname, relativeTo = absDir}
adamc@767 344
adamc@794 345 fun readSources acc =
adamc@794 346 case TextIO.inputLine inf of
adamc@794 347 NONE => rev acc
adamc@794 348 | SOME line =>
adamc@794 349 let
adamc@794 350 val acc = if CharVector.all Char.isSpace line then
adamc@794 351 acc
adamc@794 352 else
adamc@794 353 let
adamc@794 354 val fname = String.implode (List.filter (fn x => not (Char.isSpace x))
adamc@794 355 (String.explode line))
adamc@1126 356 val fname = relifyA fname
adamc@794 357 in
adamc@794 358 fname :: acc
adamc@794 359 end
adamc@794 360 in
adamc@794 361 readSources acc
adamc@794 362 end
adamc@767 363
adamc@794 364 val prefix = ref NONE
adamc@891 365 val database = ref (Settings.getDbstring ())
adamc@891 366 val exe = ref (Settings.getExe ())
adamc@891 367 val sql = ref (Settings.getSql ())
adamc@857 368 val debug = ref (Settings.getDebug ())
adamc@794 369 val profile = ref false
adamc@794 370 val timeout = ref NONE
adamc@794 371 val ffi = ref []
adamc@794 372 val link = ref []
adamc@794 373 val headers = ref []
adamc@794 374 val scripts = ref []
adamc@794 375 val clientToServer = ref []
adamc@794 376 val effectful = ref []
adamc@1171 377 val benignEffectful = ref []
adamc@794 378 val clientOnly = ref []
adamc@794 379 val serverOnly = ref []
adamc@794 380 val jsFuncs = ref []
adamc@794 381 val rewrites = ref []
adamc@794 382 val url = ref []
adamc@794 383 val mime = ref []
adamc@794 384 val libs = ref []
adamc@866 385 val protocol = ref NONE
adamc@866 386 val dbms = ref NONE
adamc@1164 387 val sigFile = ref (Settings.getSigFile ())
adamc@767 388
adamc@794 389 fun finish sources =
adamc@794 390 let
adamc@794 391 val job = {
adamc@794 392 prefix = Option.getOpt (!prefix, "/"),
adamc@794 393 database = !database,
adamc@794 394 exe = Option.getOpt (!exe, OS.Path.joinBaseExt {base = OS.Path.base filename,
adamc@794 395 ext = SOME "exe"}),
adamc@794 396 sql = !sql,
adamc@794 397 debug = !debug,
adamc@794 398 profile = !profile,
adamc@794 399 timeout = Option.getOpt (!timeout, 60),
adamc@794 400 ffi = rev (!ffi),
adamc@794 401 link = rev (!link),
adamc@794 402 headers = rev (!headers),
adamc@794 403 scripts = rev (!scripts),
adamc@794 404 clientToServer = rev (!clientToServer),
adamc@794 405 effectful = rev (!effectful),
adamc@1171 406 benignEffectful = rev (!benignEffectful),
adamc@794 407 clientOnly = rev (!clientOnly),
adamc@794 408 serverOnly = rev (!serverOnly),
adamc@794 409 jsFuncs = rev (!jsFuncs),
adamc@794 410 rewrites = rev (!rewrites),
adamc@794 411 filterUrl = rev (!url),
adamc@794 412 filterMime = rev (!mime),
adamc@866 413 sources = sources,
adamc@866 414 protocol = !protocol,
adamc@1164 415 dbms = !dbms,
adamc@1164 416 sigFile = !sigFile
adamc@794 417 }
adamc@767 418
adamc@794 419 fun mergeO f (old, new) =
adamc@794 420 case (old, new) of
adamc@794 421 (NONE, _) => new
adamc@794 422 | (_, NONE) => old
adamc@794 423 | (SOME v1, SOME v2) => SOME (f (v1, v2))
adamc@794 424
adamc@794 425 fun same desc = mergeO (fn (x : string, y) =>
adamc@794 426 (if x = y then
adamc@794 427 ()
adamc@794 428 else
adamc@794 429 ErrorMsg.error ("Multiple "
adamc@794 430 ^ desc ^ " values that don't agree");
adamc@794 431 x))
adamc@794 432
adamc@794 433 fun merge (old : job, new : job) = {
adamc@794 434 prefix = #prefix old,
adamc@891 435 database = mergeO (fn (old, _) => old) (#database old, #database new),
adamc@794 436 exe = #exe old,
adamc@794 437 sql = #sql old,
adamc@794 438 debug = #debug old orelse #debug new,
adamc@794 439 profile = #profile old orelse #profile new,
adamc@794 440 timeout = #timeout old,
adamc@794 441 ffi = #ffi old @ #ffi new,
adamc@794 442 link = #link old @ #link new,
adamc@794 443 headers = #headers old @ #headers new,
adamc@794 444 scripts = #scripts old @ #scripts new,
adamc@794 445 clientToServer = #clientToServer old @ #clientToServer new,
adamc@794 446 effectful = #effectful old @ #effectful new,
adamc@1171 447 benignEffectful = #benignEffectful old @ #benignEffectful new,
adamc@794 448 clientOnly = #clientOnly old @ #clientOnly new,
adamc@794 449 serverOnly = #serverOnly old @ #serverOnly new,
adamc@794 450 jsFuncs = #jsFuncs old @ #jsFuncs new,
adamc@794 451 rewrites = #rewrites old @ #rewrites new,
adamc@794 452 filterUrl = #filterUrl old @ #filterUrl new,
adamc@794 453 filterMime = #filterMime old @ #filterMime new,
adamc@1092 454 sources = #sources new
adamc@1092 455 @ List.filter (fn s => List.all (fn s' => s' <> s) (#sources new))
adamc@1092 456 (#sources old),
adamc@866 457 protocol = mergeO #2 (#protocol old, #protocol new),
adamc@1164 458 dbms = mergeO #2 (#dbms old, #dbms new),
adamc@1164 459 sigFile = mergeO #2 (#sigFile old, #sigFile new)
adamc@794 460 }
adamc@794 461 in
adamc@1082 462 if accLibs then
adamc@1082 463 foldr (fn (job', job) => merge (job, job')) job (!libs)
adamc@1082 464 else
adamc@1082 465 job
adamc@794 466 end
adamc@794 467
adamc@794 468 fun parsePkind s =
adamc@794 469 case s of
adamc@794 470 "all" => Settings.Any
adamc@794 471 | "url" => Settings.Url
adamc@794 472 | "table" => Settings.Table
adamc@794 473 | "sequence" => Settings.Sequence
adamc@794 474 | "view" => Settings.View
adamc@794 475 | "relation" => Settings.Relation
adamc@794 476 | "cookie" => Settings.Cookie
adamc@794 477 | "style" => Settings.Style
adamc@794 478 | _ => (ErrorMsg.error "Bad path kind spec";
adamc@794 479 Settings.Any)
adamc@794 480
adamc@794 481 fun parseFrom s =
adamc@794 482 if size s > 1 andalso String.sub (s, size s - 2) = #"/" andalso String.sub (s, size s - 1) = #"*" then
adamc@794 483 (Settings.Prefix, String.substring (s, 0, size s - 1))
adamc@794 484 else
adamc@794 485 (Settings.Exact, s)
adamc@794 486
adamc@794 487 fun parseFkind s =
adamc@794 488 case s of
adamc@794 489 "url" => url
adamc@794 490 | "mime" => mime
adamc@794 491 | _ => (ErrorMsg.error "Bad filter kind";
adamc@794 492 url)
adamc@794 493
adamc@794 494 fun parsePattern s =
adamc@794 495 if size s > 0 andalso String.sub (s, size s - 1) = #"*" then
adamc@794 496 (Settings.Prefix, String.substring (s, 0, size s - 1))
adamc@794 497 else
adamc@794 498 (Settings.Exact, s)
adamc@794 499
adamc@794 500 fun read () =
adamc@794 501 case TextIO.inputLine inf of
adamc@794 502 NONE => finish []
adamc@794 503 | SOME "\n" => finish (readSources [])
adamc@794 504 | SOME line =>
adamc@794 505 let
adamc@794 506 val (cmd, arg) = Substring.splitl (fn x => not (Char.isSpace x)) (Substring.full line)
adamc@794 507 val cmd = Substring.string (trim cmd)
adamc@794 508 val arg = Substring.string (trim arg)
adamc@794 509
adamc@794 510 fun ffiS () =
adamc@794 511 case String.fields (fn ch => ch = #".") arg of
adamc@794 512 [m, x] => (m, x)
adamc@794 513 | _ => (ErrorMsg.error (cmd ^ " argument not of the form Module.func");
adamc@794 514 ("", ""))
adamc@794 515
adamc@794 516 fun ffiM () =
adamc@794 517 case String.fields (fn ch => ch = #"=") arg of
adamc@794 518 [f, s] =>
adamc@794 519 (case String.fields (fn ch => ch = #".") f of
adamc@794 520 [m, x] => ((m, x), s)
adamc@794 521 | _ => (ErrorMsg.error (cmd ^ " argument not of the form Module.func=func'");
adamc@794 522 (("", ""), "")))
adamc@794 523 | _ => (ErrorMsg.error (cmd ^ " argument not of the form Module.func=func'");
adamc@794 524 (("", ""), ""))
adamc@794 525 in
adamc@794 526 case cmd of
adamc@794 527 "prefix" =>
adamc@794 528 (case !prefix of
adamc@794 529 NONE => ()
adamc@794 530 | SOME _ => ErrorMsg.error "Duplicate 'prefix' directive";
adamc@794 531 prefix := SOME arg)
adamc@794 532 | "database" =>
adamc@794 533 (case !database of
adamc@891 534 NONE => database := SOME arg
adamc@891 535 | SOME _ => ())
adamc@1164 536 | "dbms" =>
adamc@1164 537 (case !dbms of
adamc@1164 538 NONE => dbms := SOME arg
adamc@1164 539 | SOME _ => ())
adamc@1164 540 | "sigfile" =>
adamc@1164 541 (case !sigFile of
adamc@1164 542 NONE => sigFile := SOME arg
adamc@1164 543 | SOME _ => ())
adamc@794 544 | "exe" =>
adamc@794 545 (case !exe of
adamc@891 546 NONE => exe := SOME (relify arg)
adamc@891 547 | SOME _ => ())
adamc@794 548 | "sql" =>
adamc@794 549 (case !sql of
adamc@891 550 NONE => sql := SOME (relify arg)
adamc@891 551 | SOME _ => ())
adamc@794 552 | "debug" => debug := true
adamc@794 553 | "profile" => profile := true
adamc@794 554 | "timeout" =>
adamc@794 555 (case !timeout of
adamc@794 556 NONE => ()
adamc@794 557 | SOME _ => ErrorMsg.error "Duplicate 'timeout' directive";
adamc@794 558 timeout := SOME (valOf (Int.fromString arg)))
adamc@794 559 | "ffi" => ffi := relify arg :: !ffi
adamc@1061 560 | "link" => let
adamc@1096 561 val arg = if size arg >= 1
adamc@1096 562 andalso String.sub (arg, 0) = #"-" then
adamc@1061 563 arg
adamc@1061 564 else
adamc@1061 565 relifyA arg
adamc@1061 566 in
adamc@1061 567 link := arg :: !link
adamc@1061 568 end
adamc@794 569 | "include" => headers := relifyA arg :: !headers
adamc@794 570 | "script" => scripts := arg :: !scripts
adamc@794 571 | "clientToServer" => clientToServer := ffiS () :: !clientToServer
adamc@794 572 | "effectful" => effectful := ffiS () :: !effectful
adamc@1171 573 | "benignEffectful" => benignEffectful := ffiS () :: !benignEffectful
adamc@794 574 | "clientOnly" => clientOnly := ffiS () :: !clientOnly
adamc@794 575 | "serverOnly" => serverOnly := ffiS () :: !serverOnly
adamc@794 576 | "jsFunc" => jsFuncs := ffiM () :: !jsFuncs
adamc@794 577 | "rewrite" =>
adamc@794 578 let
adamc@794 579 fun doit (pkind, from, to) =
adamc@794 580 let
adamc@794 581 val pkind = parsePkind pkind
adamc@794 582 val (kind, from) = parseFrom from
adamc@794 583 in
adamc@794 584 rewrites := {pkind = pkind, kind = kind, from = from, to = to} :: !rewrites
adamc@794 585 end
adamc@794 586 in
adamc@794 587 case String.tokens Char.isSpace arg of
adamc@794 588 [pkind, from, to] => doit (pkind, from, to)
adamc@794 589 | [pkind, from] => doit (pkind, from, "")
adamc@794 590 | _ => ErrorMsg.error "Bad 'rewrite' syntax"
adamc@794 591 end
adamc@794 592 | "allow" =>
adamc@794 593 (case String.tokens Char.isSpace arg of
adamc@794 594 [fkind, pattern] =>
adamc@794 595 let
adamc@794 596 val fkind = parseFkind fkind
adamc@794 597 val (kind, pattern) = parsePattern pattern
adamc@794 598 in
adamc@794 599 fkind := {action = Settings.Allow, kind = kind, pattern = pattern} :: !fkind
adamc@794 600 end
adamc@794 601 | _ => ErrorMsg.error "Bad 'allow' syntax")
adamc@794 602 | "deny" =>
adamc@794 603 (case String.tokens Char.isSpace arg of
adamc@794 604 [fkind, pattern] =>
adamc@794 605 let
adamc@794 606 val fkind = parseFkind fkind
adamc@794 607 val (kind, pattern) = parsePattern pattern
adamc@794 608 in
adamc@794 609 fkind := {action = Settings.Deny, kind = kind, pattern = pattern} :: !fkind
adamc@794 610 end
adamc@794 611 | _ => ErrorMsg.error "Bad 'deny' syntax")
adamc@1082 612 | "library" => if accLibs then
adamc@1089 613 libs := pu (libify (relify arg)) :: !libs
adamc@1082 614 else
adamc@1089 615 bigLibs := libify' arg :: !bigLibs
adamc@794 616 | "path" =>
adamc@794 617 (case String.fields (fn ch => ch = #"=") arg of
adamc@794 618 [n, v] => pathmap := M.insert (!pathmap, n, v)
adamc@794 619 | _ => ErrorMsg.error "path argument not of the form name=value'")
adamc@794 620 | _ => ErrorMsg.error ("Unrecognized command '" ^ cmd ^ "'");
adamc@794 621 read ()
adamc@794 622 end
adamc@794 623
adamc@1151 624 val job = if hasBlankLine then
adamc@1151 625 read ()
adamc@1151 626 else
adamc@1151 627 finish (readSources [])
adamc@767 628 in
adamc@794 629 TextIO.closeIn inf;
adamc@794 630 Settings.setUrlPrefix (#prefix job);
adamc@794 631 Settings.setTimeout (#timeout job);
adamc@794 632 Settings.setHeaders (#headers job);
adamc@794 633 Settings.setScripts (#scripts job);
adamc@794 634 Settings.setClientToServer (#clientToServer job);
adamc@794 635 Settings.setEffectful (#effectful job);
adamc@1171 636 Settings.setBenignEffectful (#benignEffectful job);
adamc@794 637 Settings.setClientOnly (#clientOnly job);
adamc@794 638 Settings.setServerOnly (#serverOnly job);
adamc@794 639 Settings.setJsFuncs (#jsFuncs job);
adamc@794 640 Settings.setRewriteRules (#rewrites job);
adamc@794 641 Settings.setUrlRules (#filterUrl job);
adamc@794 642 Settings.setMimeRules (#filterMime job);
adamc@866 643 Option.app Settings.setProtocol (#protocol job);
adamc@866 644 Option.app Settings.setDbms (#dbms job);
adamc@794 645 job
adamc@767 646 end
adamc@767 647 in
adamc@1082 648 {Job = pu fname, Libs = !bigLibs}
adamc@767 649 end
adamc@767 650
adamc@1082 651 fun p_job' {Job = j, Libs = _ : string list} = p_job j
adamc@1082 652
adamc@270 653 val parseUrp = {
adamc@1083 654 func = #Job o parseUrp' true,
adamc@270 655 print = p_job
adamc@270 656 }
adamc@270 657
adamc@1082 658 val parseUrp' = {
adamc@1083 659 func = parseUrp' false,
adamc@1082 660 print = p_job'
adamc@1082 661 }
adamc@1082 662
adamc@270 663 val toParseJob = transform parseUrp "parseJob"
adamc@1082 664 val toParseJob' = transform parseUrp' "parseJob'"
adamc@1082 665
adamc@1082 666 fun op o (tr2 : ('b, 'c) transform, tr1 : ('a, 'b) transform) = {
adamc@1082 667 func = fn input => case #func tr1 input of
adamc@1082 668 NONE => NONE
adamc@1082 669 | SOME v => #func tr2 v,
adamc@1082 670 print = #print tr2,
adamc@1082 671 time = fn (input, pmap) => let
adamc@1082 672 val (ro, pmap) = #time tr1 (input, pmap)
adamc@1082 673 in
adamc@1082 674 case ro of
adamc@1082 675 NONE => (NONE, pmap)
adamc@1082 676 | SOME v => #time tr2 (v, pmap)
adamc@1082 677 end
adamc@1082 678 }
adamc@270 679
adamc@56 680 fun capitalize "" = ""
adamc@56 681 | capitalize s = str (Char.toUpper (String.sub (s, 0))) ^ String.extract (s, 1, NONE)
adamc@56 682
adamc@1090 683 structure SM = BinaryMapFn(struct
adamc@1090 684 type ord_key = string
adamc@1090 685 val compare = String.compare
adamc@1090 686 end)
adamc@1090 687
adamc@1090 688 val moduleRoots = ref ([] : (string * string) list)
adamc@1090 689 fun addModuleRoot (k, v) = moduleRoots := (k, v) :: !moduleRoots
adamc@1090 690
adamc@1090 691 structure SS = BinarySetFn(struct
adamc@1090 692 type ord_key = string
adamc@1090 693 val compare = String.compare
adamc@1090 694 end)
adamc@1090 695
adamc@201 696 val parse = {
adamc@764 697 func = fn {database, sources = fnames, ffi, ...} : job =>
adamc@201 698 let
adamc@1090 699 val mrs = !moduleRoots
adamc@1090 700
adamc@834 701 val anyErrors = ref false
adamc@834 702 fun checkErrors () = anyErrors := (!anyErrors orelse ErrorMsg.anyErrors ())
adamc@201 703 fun nameOf fname = capitalize (OS.Path.file fname)
adamc@109 704
adamc@764 705 fun parseFfi fname =
adamc@764 706 let
adamc@764 707 val mname = nameOf fname
adamc@764 708 val urs = OS.Path.joinBaseExt {base = fname, ext = SOME "urs"}
adamc@764 709
adamc@764 710 val loc = {file = urs,
adamc@764 711 first = ErrorMsg.dummyPos,
adamc@764 712 last = ErrorMsg.dummyPos}
adamc@764 713
adamc@764 714 val sgn = (Source.SgnConst (#func parseUrs urs), loc)
adamc@764 715 in
adamc@834 716 checkErrors ();
adamc@764 717 (Source.DFfiStr (mname, sgn), loc)
adamc@764 718 end
adamc@764 719
adamc@1090 720 val defed = ref SS.empty
adamc@1092 721 val fulls = ref SS.empty
adamc@1090 722
adamc@201 723 fun parseOne fname =
adamc@201 724 let
adamc@201 725 val mname = nameOf fname
adamc@244 726 val ur = OS.Path.joinBaseExt {base = fname, ext = SOME "ur"}
adamc@244 727 val urs = OS.Path.joinBaseExt {base = fname, ext = SOME "urs"}
adamc@56 728
adamc@201 729 val sgnO =
adamc@244 730 if Posix.FileSys.access (urs, []) then
adamc@244 731 SOME (Source.SgnConst (#func parseUrs urs),
adamc@244 732 {file = urs,
adamc@201 733 first = ErrorMsg.dummyPos,
adamc@201 734 last = ErrorMsg.dummyPos})
adamc@834 735 before checkErrors ()
adamc@201 736 else
adamc@201 737 NONE
adamc@56 738
adamc@244 739 val loc = {file = ur,
adamc@201 740 first = ErrorMsg.dummyPos,
adamc@201 741 last = ErrorMsg.dummyPos}
adamc@56 742
adamc@244 743 val ds = #func parseUr ur
adamc@1090 744 val d = (Source.DStr (mname, sgnO, (Source.StrConst ds, loc)), loc)
adamc@1090 745
adamc@1126 746 val fname = OS.Path.mkCanonical fname
adamc@1090 747 val d = case List.find (fn (root, name) =>
adamc@1090 748 String.isPrefix (root ^ "/") fname) mrs of
adamc@1090 749 NONE => d
adamc@1090 750 | SOME (root, name) =>
adamc@1090 751 let
adamc@1090 752 val fname = String.extract (fname, size root + 1, NONE)
adamc@1090 753 val pieces = name :: String.tokens (fn ch => ch = #"/") fname
adamc@1090 754 val pieces = List.filter (fn s => size s > 0
adamc@1090 755 andalso Char.isAlpha (String.sub (s, 0)))
adamc@1090 756 pieces
adamc@1090 757 val pieces = map capitalize pieces
adamc@1092 758 val full = String.concatWith "." pieces
adamc@1090 759
adamc@1090 760 fun makeD prefix pieces =
adamc@1090 761 case pieces of
adamc@1090 762 [] => (ErrorMsg.error "Empty module path";
adamc@1090 763 (Source.DStyle "Boo", loc))
adamc@1090 764 | [_] => d
adamc@1090 765 | piece :: pieces =>
adamc@1090 766 let
adamc@1146 767 val this = case prefix of
adamc@1146 768 "" => piece
adamc@1146 769 | _ => prefix ^ "." ^ piece
adamc@1090 770 val old = SS.member (!defed, this)
adamc@1146 771
adamc@1146 772 fun notThere (ch, s) =
adamc@1146 773 Substring.isEmpty (#2 (Substring.splitl
adamc@1146 774 (fn ch' => ch' <> ch) s))
adamc@1146 775
adamc@1146 776 fun simOpen () =
adamc@1146 777 SS.foldl (fn (full, ds) =>
adamc@1146 778 if String.isPrefix (this ^ ".") full
adamc@1146 779 andalso notThere (#".",
adamc@1146 780 Substring.extract (full,
adamc@1146 781 size
adamc@1146 782 this + 1,
adamc@1146 783 NONE)) then
adamc@1146 784 let
adamc@1146 785 val parts = String.tokens
adamc@1146 786 (fn ch => ch = #".") full
adamc@1146 787
adamc@1146 788 val part = List.last parts
adamc@1146 789
adamc@1146 790 val imp = if length parts >= 2 then
adamc@1146 791 (Source.StrProj
adamc@1146 792 ((Source.StrVar
adamc@1146 793 (List.nth (parts,
adamc@1146 794 length
adamc@1146 795 parts
adamc@1146 796 - 2)),
adamc@1146 797 loc),
adamc@1146 798 part), loc)
adamc@1146 799 else
adamc@1146 800 (Source.StrVar part, loc)
adamc@1146 801 in
adamc@1146 802 (Source.DStr (part, NONE, imp),
adamc@1146 803 loc) :: ds
adamc@1146 804 end
adamc@1146 805 else
adamc@1146 806 ds) [] (!fulls)
adamc@1090 807 in
adamc@1090 808 defed := SS.add (!defed, this);
adamc@1090 809 (Source.DStr (piece, NONE,
adamc@1090 810 (Source.StrConst (if old then
adamc@1146 811 simOpen ()
adamc@1146 812 @ [makeD this pieces]
adamc@1090 813 else
adamc@1146 814 [makeD this pieces]), loc)),
adamc@1090 815 loc)
adamc@1090 816 end
adamc@1090 817 in
adamc@1092 818 if SS.member (!fulls, full) then
adamc@1092 819 ErrorMsg.error ("Rooted module " ^ full ^ " has multiple versions.")
adamc@1092 820 else
adamc@1092 821 ();
adamc@1146 822
adamc@1090 823 makeD "" pieces
adamc@1146 824 before ignore (foldl (fn (new, path) =>
adamc@1146 825 let
adamc@1146 826 val new' = case path of
adamc@1146 827 "" => new
adamc@1146 828 | _ => path ^ "." ^ new
adamc@1146 829 in
adamc@1146 830 fulls := SS.add (!fulls, new');
adamc@1146 831 new'
adamc@1146 832 end) "" pieces)
adamc@1090 833 end
adamc@201 834 in
adamc@834 835 checkErrors ();
adamc@1090 836 d
adamc@201 837 end
adamc@56 838
adamc@764 839 val dsFfi = map parseFfi ffi
adamc@201 840 val ds = map parseOne fnames
adamc@1090 841 val loc = ErrorMsg.dummySpan
adamc@201 842 in
adamc@834 843 if !anyErrors then
adamc@834 844 ErrorMsg.error "Parse failure"
adamc@834 845 else
adamc@834 846 ();
adamc@834 847
adamc@201 848 let
adamc@1126 849 val final = List.last fnames
adamc@1126 850 val final = case List.find (fn (root, name) =>
adamc@1126 851 String.isPrefix (root ^ "/") final) mrs of
adamc@1126 852 NONE => (Source.StrVar (nameOf final), loc)
adamc@1126 853 | SOME (root, name) =>
adamc@1126 854 let
adamc@1126 855 val m = (Source.StrVar name, loc)
adamc@1126 856 val final = String.extract (final, size root + 1, NONE)
adamc@1126 857 in
adamc@1126 858 foldl (fn (x, m) => (Source.StrProj (m, capitalize x), loc))
adamc@1126 859 m (String.fields (fn ch => ch = #"/") final)
adamc@1126 860 end
adamc@271 861
adamc@764 862 val ds = dsFfi @ ds
adamc@1126 863 @ [(Source.DExport final, loc)]
adamc@1090 864
adamc@1090 865 val ds = case database of
adamc@1090 866 NONE => ds
adamc@1090 867 | SOME s => (Source.DDatabase s, loc) :: ds
adamc@201 868 in
adamc@1090 869 ds
adamc@201 870 end handle Empty => ds
adamc@201 871 end,
adamc@201 872 print = SourcePrint.p_file
adamc@201 873 }
adamc@56 874
adamc@270 875 val toParse = transform parse "parse" o toParseJob
adamc@38 876
adamc@378 877 fun libFile s = OS.Path.joinDirFile {dir = Config.libUr,
adamc@378 878 file = s}
adamc@378 879 fun clibFile s = OS.Path.joinDirFile {dir = Config.libC,
adamc@378 880 file = s}
adamc@378 881
adamc@201 882 val elaborate = {
adamc@201 883 func = fn file => let
adamc@378 884 val basis = #func parseUrs (libFile "basis.urs")
adamc@378 885 val topSgn = #func parseUrs (libFile "top.urs")
adamc@378 886 val topStr = #func parseUr (libFile "top.ur")
adamc@201 887 in
adamc@325 888 Elaborate.elabFile basis topStr topSgn ElabEnv.empty file
adamc@201 889 end,
adamc@201 890 print = ElabPrint.p_file ElabEnv.empty
adamc@201 891 }
adamc@5 892
adamc@270 893 val toElaborate = transform elaborate "elaborate" o toParse
adamc@201 894
adamc@448 895 val unnest = {
adamc@448 896 func = Unnest.unnest,
adamc@448 897 print = ElabPrint.p_file ElabEnv.empty
adamc@448 898 }
adamc@448 899
adamc@448 900 val toUnnest = transform unnest "unnest" o toElaborate
adamc@448 901
adamc@313 902 val termination = {
adamc@313 903 func = (fn file => (Termination.check file; file)),
adamc@313 904 print = ElabPrint.p_file ElabEnv.empty
adamc@313 905 }
adamc@313 906
adamc@448 907 val toTermination = transform termination "termination" o toUnnest
adamc@313 908
adamc@201 909 val explify = {
adamc@201 910 func = Explify.explify,
adamc@201 911 print = ExplPrint.p_file ExplEnv.empty
adamc@201 912 }
adamc@201 913
adamc@625 914 val toExplify = transform explify "explify" o toUnnest
adamc@201 915
adamc@201 916 val corify = {
adamc@201 917 func = Corify.corify,
adamc@201 918 print = CorePrint.p_file CoreEnv.empty
adamc@201 919 }
adamc@201 920
adamc@270 921 val toCorify = transform corify "corify" o toExplify
adamc@201 922
adamc@482 923 (*val reduce_local = {
adamc@482 924 func = ReduceLocal.reduce,
adamc@482 925 print = CorePrint.p_file CoreEnv.empty
adamc@482 926 }
adamc@482 927
adamc@482 928 val toReduce_local = transform reduce_local "reduce_local" o toCorify*)
adamc@482 929
adamc@443 930 val especialize = {
adamc@443 931 func = ESpecialize.specialize,
adamc@443 932 print = CorePrint.p_file CoreEnv.empty
adamc@443 933 }
adamc@443 934
adamc@454 935 val core_untangle = {
adamc@454 936 func = CoreUntangle.untangle,
adamc@454 937 print = CorePrint.p_file CoreEnv.empty
adamc@454 938 }
adamc@454 939
adamc@794 940 val toCore_untangle = transform core_untangle "core_untangle" o toCorify
adamc@454 941
adamc@202 942 val shake = {
adamc@202 943 func = Shake.shake,
adamc@202 944 print = CorePrint.p_file CoreEnv.empty
adamc@202 945 }
adamc@39 946
adamc@454 947 val toShake1 = transform shake "shake1" o toCore_untangle
adamc@110 948
adamc@607 949 val rpcify = {
adamc@607 950 func = Rpcify.frob,
adamc@607 951 print = CorePrint.p_file CoreEnv.empty
adamc@607 952 }
adamc@607 953
adamc@607 954 val toRpcify = transform rpcify "rpcify" o toShake1
adamc@607 955
adamc@642 956 val toCore_untangle2 = transform core_untangle "core_untangle2" o toRpcify
adamc@642 957 val toShake2 = transform shake "shake2" o toCore_untangle2
adamc@1181 958
adamc@1181 959 val unpoly = {
adamc@1181 960 func = Unpoly.unpoly,
adamc@1181 961 print = CorePrint.p_file CoreEnv.empty
adamc@1181 962 }
adamc@1181 963
adamc@1181 964 val toUnpoly1 = transform unpoly "unpoly1" o toShake2
adamc@1181 965
adamc@1181 966 val toEspecialize1 = transform especialize "especialize1" o toUnpoly1
adamc@1181 967
adamc@1062 968 val toCore_untangle3 = transform core_untangle "core_untangle3" o toEspecialize1
adamc@1062 969 val toShake3 = transform shake "shake3" o toCore_untangle3
adamc@642 970
adamc@202 971 val tag = {
adamc@202 972 func = Tag.tag,
adamc@202 973 print = CorePrint.p_file CoreEnv.empty
adamc@202 974 }
adamc@193 975
adamc@1062 976 val toTag = transform tag "tag" o toShake3
adamc@20 977
adamc@202 978 val reduce = {
adamc@202 979 func = Reduce.reduce,
adamc@202 980 print = CorePrint.p_file CoreEnv.empty
adamc@202 981 }
adamc@25 982
adamc@692 983 val toReduce = transform reduce "reduce" o toTag
adamc@23 984
adamc@1181 985 val toUnpoly2 = transform unpoly "unpoly2" o toReduce
adamc@315 986
adamc@202 987 val specialize = {
adamc@202 988 func = Specialize.specialize,
adamc@202 989 print = CorePrint.p_file CoreEnv.empty
adamc@202 990 }
adamc@132 991
adamc@1181 992 val toSpecialize = transform specialize "specialize" o toUnpoly2
adamc@131 993
adamc@1062 994 val toShake4 = transform shake "shake4" o toSpecialize
adamc@133 995
adamc@1062 996 val toEspecialize2 = transform especialize "especialize2" o toShake4
adamc@794 997
adamc@1062 998 val toReduce2 = transform reduce "reduce2" o toEspecialize2
adamc@898 999
adamc@1062 1000 val toShake5 = transform shake "shake5" o toReduce2
adamc@794 1001
adamc@692 1002 val marshalcheck = {
adamc@692 1003 func = (fn file => (MarshalCheck.check file; file)),
adamc@692 1004 print = CorePrint.p_file CoreEnv.empty
adamc@692 1005 }
adamc@692 1006
adamc@1062 1007 val toMarshalcheck = transform marshalcheck "marshalcheck" o toShake5
adamc@692 1008
adamc@732 1009 val effectize = {
adamc@732 1010 func = Effective.effectize,
adamc@732 1011 print = CorePrint.p_file CoreEnv.empty
adamc@732 1012 }
adamc@732 1013
adamc@732 1014 val toEffectize = transform effectize "effectize" o toMarshalcheck
adamc@732 1015
adamc@1170 1016 val css = {
adamc@1170 1017 func = Css.summarize,
adamc@1170 1018 print = fn _ => Print.box []
adamc@1170 1019 }
adamc@1170 1020
adamc@1170 1021 val toCss = transform css "css" o toShake5
adamc@1170 1022
adamc@202 1023 val monoize = {
adamc@202 1024 func = Monoize.monoize CoreEnv.empty,
adamc@202 1025 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1026 }
adamc@134 1027
adamc@732 1028 val toMonoize = transform monoize "monoize" o toEffectize
adamc@96 1029
adamc@202 1030 val mono_opt = {
adamc@910 1031 func = MonoOpt.optimize,
adamc@202 1032 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1033 }
adamc@29 1034
adamc@270 1035 val toMono_opt1 = transform mono_opt "mono_opt1" o toMonoize
adamc@5 1036
adamc@202 1037 val untangle = {
adamc@202 1038 func = Untangle.untangle,
adamc@202 1039 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1040 }
adamc@1 1041
adamc@270 1042 val toUntangle = transform untangle "untangle" o toMono_opt1
adamc@38 1043
adamc@202 1044 val mono_reduce = {
adamc@202 1045 func = MonoReduce.reduce,
adamc@202 1046 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1047 }
adamc@16 1048
adamc@270 1049 val toMono_reduce = transform mono_reduce "mono_reduce" o toUntangle
adamc@39 1050
adamc@202 1051 val mono_shake = {
adamc@202 1052 func = MonoShake.shake,
adamc@202 1053 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1054 }
adamc@110 1055
adamc@270 1056 val toMono_shake = transform mono_shake "mono_shake1" o toMono_reduce
adamc@193 1057
adamc@572 1058 val toMono_opt2 = transform mono_opt "mono_opt2" o toMono_shake
adamc@572 1059
adamc@567 1060 val jscomp = {
adamc@567 1061 func = JsComp.process,
adamc@567 1062 print = MonoPrint.p_file MonoEnv.empty
adamc@567 1063 }
adamc@567 1064
adamc@572 1065 val toJscomp = transform jscomp "jscomp" o toMono_opt2
adamc@567 1066
adamc@910 1067 val toMono_opt3 = transform mono_opt "mono_opt3" o toJscomp
adamc@20 1068
adamc@506 1069 val fuse = {
adamc@506 1070 func = Fuse.fuse,
adamc@506 1071 print = MonoPrint.p_file MonoEnv.empty
adamc@506 1072 }
adamc@506 1073
adamc@572 1074 val toFuse = transform fuse "fuse" o toMono_opt3
adamc@506 1075
adamc@506 1076 val toUntangle2 = transform untangle "untangle2" o toFuse
adamc@506 1077
adamc@601 1078 val toMono_reduce2 = transform mono_reduce "mono_reduce2" o toUntangle2
adamc@601 1079 val toMono_shake2 = transform mono_shake "mono_shake2" o toMono_reduce2
adamc@916 1080 val toMono_opt4 = transform mono_opt "mono_opt4" o toMono_shake2
adamc@1017 1081 val toMono_reduce3 = transform mono_reduce "mono_reduce3" o toMono_opt4
adamc@1017 1082 val toFuse2 = transform fuse "shake2" o toMono_reduce3
adamc@916 1083 val toUntangle3 = transform untangle "untangle3" o toFuse2
adamc@916 1084 val toMono_shake3 = transform mono_shake "mono_shake3" o toUntangle3
adamc@506 1085
adamc@377 1086 val pathcheck = {
adamc@377 1087 func = (fn file => (PathCheck.check file; file)),
adamc@377 1088 print = MonoPrint.p_file MonoEnv.empty
adamc@377 1089 }
adamc@377 1090
adamc@916 1091 val toPathcheck = transform pathcheck "pathcheck" o toMono_shake3
adamc@377 1092
adamc@202 1093 val cjrize = {
adamc@202 1094 func = Cjrize.cjrize,
adamc@202 1095 print = CjrPrint.p_file CjrEnv.empty
adamc@202 1096 }
adamc@23 1097
adamc@377 1098 val toCjrize = transform cjrize "cjrize" o toPathcheck
adamc@29 1099
adamc@643 1100 val scriptcheck = {
adamc@643 1101 func = ScriptCheck.classify,
adamc@643 1102 print = CjrPrint.p_file CjrEnv.empty
adamc@643 1103 }
adamc@643 1104
adamc@643 1105 val toScriptcheck = transform scriptcheck "scriptcheck" o toCjrize
adamc@643 1106
adamc@282 1107 val prepare = {
adamc@282 1108 func = Prepare.prepare,
adamc@282 1109 print = CjrPrint.p_file CjrEnv.empty
adamc@282 1110 }
adamc@282 1111
adamc@643 1112 val toPrepare = transform prepare "prepare" o toScriptcheck
adamc@282 1113
adamc@879 1114 val checknest = {
adamc@879 1115 func = fn f => if #supportsNestedPrepared (Settings.currentDbms ()) then f else Checknest.annotate f,
adamc@879 1116 print = CjrPrint.p_file CjrEnv.empty
adamc@879 1117 }
adamc@879 1118
adamc@879 1119 val toChecknest = transform checknest "checknest" o toPrepare
adamc@879 1120
adamc@274 1121 val sqlify = {
adamc@274 1122 func = Cjrize.cjrize,
adamc@274 1123 print = CjrPrint.p_sql CjrEnv.empty
adamc@274 1124 }
adamc@274 1125
adamc@274 1126 val toSqlify = transform sqlify "sqlify" o toMono_opt2
adamc@274 1127
adamc@764 1128 fun compileC {cname, oname, ename, libs, profile, debug, link = link'} =
adamc@183 1129 let
adamc@855 1130 val proto = Settings.currentProtocol ()
adamc@1095 1131
adamc@1095 1132 val lib = if Settings.getStaticLinking () then
adamc@1132 1133 #linkStatic proto ^ " " ^ Config.lib ^ "/../liburweb.a"
adamc@1095 1134 else
adamc@1106 1135 "-L" ^ Config.lib ^ "/.. -lurweb " ^ #linkDynamic proto
adamc@378 1136
adamc@1094 1137 val compile = "gcc " ^ Config.gccArgs ^ " -Wimplicit -Werror -O3 -fno-inline -I " ^ Config.includ
adamc@1096 1138 ^ " " ^ #compile proto
adamc@832 1139 ^ " -c " ^ cname ^ " -o " ^ oname
adamc@1096 1140
adamc@1095 1141 val link = "gcc -Werror -O3 -lm -lmhash -pthread " ^ Config.gccArgs ^ " " ^ libs ^ " " ^ lib ^ " " ^ oname
adamc@1095 1142 ^ " -o " ^ ename
adamc@502 1143
adamc@502 1144 val (compile, link) =
adamc@502 1145 if profile then
adamc@502 1146 (compile ^ " -pg", link ^ " -pg")
adamc@502 1147 else
adamc@502 1148 (compile, link)
adamc@742 1149
adamc@742 1150 val (compile, link) =
adamc@742 1151 if debug then
adamc@742 1152 (compile ^ " -g", link ^ " -g")
adamc@742 1153 else
adamc@742 1154 (compile, link)
adamc@764 1155
adamc@764 1156 val link = foldl (fn (s, link) => link ^ " " ^ s) link link'
adamc@183 1157 in
adamc@1045 1158 OS.Process.isSuccess (OS.Process.system compile)
adamc@1045 1159 andalso OS.Process.isSuccess (OS.Process.system link)
adamc@183 1160 end
adamc@183 1161
adamc@202 1162 fun compile job =
adamc@879 1163 case run toChecknest job of
adamc@1045 1164 NONE => false
adamc@29 1165 | SOME file =>
adamc@202 1166 let
adamc@274 1167 val job = valOf (run (transform parseUrp "parseUrp") job)
adamc@102 1168
adamc@274 1169 val (cname, oname, cleanup) =
adamc@274 1170 if #debug job then
adamc@457 1171 ("/tmp/webapp.c", "/tmp/webapp.o", fn () => ())
adamc@274 1172 else
adamc@274 1173 let
adamc@274 1174 val dir = OS.FileSys.tmpName ()
adamc@403 1175 val () = if OS.FileSys.access (dir, []) then
adamc@403 1176 OS.FileSys.remove dir
adamc@403 1177 else
adamc@403 1178 ()
adamc@457 1179 val cname = OS.Path.joinDirFile {dir = dir, file = "webapp.c"}
adamc@457 1180 val oname = OS.Path.joinDirFile {dir = dir, file = "webapp.o"}
adamc@274 1181 in
adamc@274 1182 OS.FileSys.mkDir dir;
adamc@274 1183 (cname, oname,
adamc@274 1184 fn () => (OS.FileSys.remove cname;
adamc@274 1185 OS.FileSys.remove oname;
adamc@473 1186 OS.FileSys.rmDir dir)
adamc@473 1187 handle OS.SysErr _ => OS.FileSys.rmDir dir)
adamc@274 1188 end
adamc@274 1189 val ename = #exe job
adamc@202 1190 in
adamc@274 1191 let
adamc@274 1192 val outf = TextIO.openOut cname
adamc@274 1193 val s = TextIOPP.openOut {dst = outf, wid = 80}
adamc@432 1194
adamc@432 1195 val hasDb = List.exists (fn (Cjr.DDatabase _, _) => true | _ => false) (#1 file)
adamc@432 1196 val libs =
adamc@432 1197 if hasDb then
adamc@866 1198 #link (Settings.currentDbms ())
adamc@432 1199 else
adamc@432 1200 ""
adamc@274 1201 in
adamc@274 1202 Print.fprint s (CjrPrint.p_file CjrEnv.empty file);
adamc@415 1203 TextIO.output1 (outf, #"\n");
adamc@274 1204 TextIO.closeOut outf;
adamc@102 1205
adamc@274 1206 case #sql job of
adamc@274 1207 NONE => ()
adamc@274 1208 | SOME sql =>
adamc@274 1209 let
adamc@274 1210 val outf = TextIO.openOut sql
adamc@274 1211 val s = TextIOPP.openOut {dst = outf, wid = 80}
adamc@274 1212 in
adamc@274 1213 Print.fprint s (CjrPrint.p_sql CjrEnv.empty file);
adamc@274 1214 TextIO.closeOut outf
adamc@274 1215 end;
adamc@274 1216
adamc@742 1217 compileC {cname = cname, oname = oname, ename = ename, libs = libs,
adamc@1045 1218 profile = #profile job, debug = #debug job, link = #link job}
adamc@274 1219
adamc@1045 1220 before cleanup ()
adamc@274 1221 end
adamc@274 1222 handle ex => (((cleanup ()) handle _ => ()); raise ex)
adamc@202 1223 end
adamc@29 1224
adamc@1045 1225 fun compiler job =
adamc@1045 1226 if compile job then
adamc@1045 1227 ()
adamc@1045 1228 else
adamc@1045 1229 OS.Process.exit OS.Process.failure
adamc@1045 1230
adamc@1 1231 end