Mercurial > urweb

annotate src/compiler.sml @ 2211:ef766ef6e242

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge.
author Ziv Scully <ziv@mit.edu>
date Sat, 13 Sep 2014 19:16:07 -0400
parents 0ca11d57c175 88841212f0ba
children 365727ff68f4
rev   line source
adam@2067 1 (* Copyright (c) 2008-2012, 2014, 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
ziv@2202 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
ziv@2202 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,
adam@1725 47 linker : string option,
adamc@765 48 headers : string list,
adamc@766 49 scripts : string list,
adamc@765 50 clientToServer : Settings.ffi list,
adamc@765 51 effectful : Settings.ffi list,
adamc@1171 52 benignEffectful : Settings.ffi list,
adamc@765 53 clientOnly : Settings.ffi list,
adamc@765 54 serverOnly : Settings.ffi list,
adamc@768 55 jsFuncs : (Settings.ffi * string) list,
adamc@769 56 rewrites : Settings.rewrite list,
adamc@769 57 filterUrl : Settings.rule list,
adamc@866 58 filterMime : Settings.rule list,
adam@1465 59 filterRequest : Settings.rule list,
adam@1465 60 filterResponse : Settings.rule list,
adam@1799 61 filterEnv : Settings.rule list,
adamc@866 62 protocol : string option,
adamc@1164 63 dbms : string option,
adamc@1183 64 sigFile : string option,
adam@1294 65 safeGets : string list,
adam@1332 66 onError : (string * string list * string) option,
adam@1332 67 minHeap : int
adamc@270 68 }
adamc@201 69
adamc@201 70 type ('src, 'dst) phase = {
adamc@201 71 func : 'src -> 'dst,
adamc@201 72 print : 'dst -> Print.PD.pp_desc
adamc@201 73 }
adamc@201 74
adamc@201 75 type pmap = (string * Time.time) list
adamc@201 76
adamc@201 77 type ('src, 'dst) transform = {
adamc@201 78 func : 'src -> 'dst option,
adamc@201 79 print : 'dst -> Print.PD.pp_desc,
adamc@201 80 time : 'src * pmap -> 'dst option * pmap
adamc@201 81 }
adamc@201 82
adamc@1079 83 val debug = ref false
adam@1677 84 val dumpSource = ref false
adamc@1240 85 val doIflow = ref false
ziv@2209 86 val doSqlcache = ref false
adamc@1079 87
adam@1677 88 val doDumpSource = ref (fn () => ())
adam@1677 89
adam@1961 90 val stop = ref (NONE : string option)
adam@1961 91 fun setStop s = stop := SOME s
adam@1961 92
adamc@201 93 fun transform (ph : ('src, 'dst) phase) name = {
adamc@201 94 func = fn input => let
adamc@1079 95 val () = if !debug then
adamc@1079 96 print ("Starting " ^ name ^ "....\n")
adamc@1079 97 else
adamc@1079 98 ()
adamc@201 99 val v = #func ph input
adamc@201 100 in
adamc@1079 101 if !debug then
adamc@1079 102 print ("Finished " ^ name ^ ".\n")
adamc@1079 103 else
adamc@1079 104 ();
adamc@201 105 if ErrorMsg.anyErrors () then
adam@1677 106 (!doDumpSource ();
adam@1677 107 doDumpSource := (fn () => ());
adam@1677 108 NONE)
adam@1961 109 else if !stop = SOME name then
adam@1961 110 (Print.eprint (#print ph v);
adam@1961 111 ErrorMsg.error ("Stopped compilation after phase " ^ name);
adam@1961 112 NONE)
adamc@201 113 else
adam@1677 114 (if !dumpSource then
adam@1677 115 doDumpSource := (fn () => Print.eprint (#print ph v))
adam@1677 116 else
adam@1677 117 ();
adam@1677 118 SOME v)
adamc@201 119 end,
adamc@201 120 print = #print ph,
adamc@201 121 time = fn (input, pmap) => let
adamc@1186 122 val () = if !debug then
adamc@1186 123 print ("Starting " ^ name ^ "....\n")
adamc@1186 124 else
adamc@1186 125 ()
adamc@201 126 val befor = Time.now ()
adamc@201 127 val v = #func ph input
adamc@201 128 val elapsed = Time.- (Time.now (), befor)
adamc@201 129 in
adamc@1186 130 if !debug then
adamc@1186 131 print ("Finished " ^ name ^ ".\n")
adamc@1186 132 else
adamc@1186 133 ();
adamc@201 134 (if ErrorMsg.anyErrors () then
adamc@201 135 NONE
adamc@201 136 else
adamc@201 137 SOME v,
adamc@201 138 (name, elapsed) :: pmap)
adamc@201 139 end
adamc@201 140 }
adamc@201 141
adamc@346 142 fun check (tr : ('src, 'dst) transform) x = (ErrorMsg.resetErrors ();
adamc@346 143 ignore (#func tr x))
adamc@346 144
adamc@280 145 fun run (tr : ('src, 'dst) transform) x = (ErrorMsg.resetErrors ();
adamc@280 146 #func tr x)
adamc@201 147
adamc@201 148 fun runPrint (tr : ('src, 'dst) transform) input =
adamc@280 149 (ErrorMsg.resetErrors ();
adamc@280 150 case #func tr input of
adamc@280 151 NONE => print "Failure\n"
adamc@280 152 | SOME v =>
adamc@280 153 (print "Success\n";
adamc@280 154 Print.print (#print tr v);
adamc@280 155 print "\n"))
adamc@201 156
adam@1362 157 fun runPrintToFile (tr : ('src, 'dst) transform) input fname =
adam@1362 158 (ErrorMsg.resetErrors ();
adam@1362 159 case #func tr input of
adam@1362 160 NONE => print "Failure\n"
adam@1362 161 | SOME v =>
adam@1362 162 let
adam@1362 163 val outf = TextIO.openOut fname
adam@1362 164 val str = Print.openOut {dst = outf, wid = 80}
adam@1362 165 in
adam@1362 166 print "Success\n";
adam@1362 167 Print.fprint str (#print tr v);
adam@1362 168 Print.PD.PPS.closeStream str;
adam@1362 169 TextIO.closeOut outf
adam@1362 170 end)
adam@1362 171
adamc@201 172 fun time (tr : ('src, 'dst) transform) input =
adamc@55 173 let
adamc@201 174 val (_, pmap) = #time tr (input, [])
adamc@201 175 in
adamc@201 176 app (fn (name, time) =>
adamc@201 177 print (name ^ ": " ^ LargeReal.toString (Time.toReal time) ^ "\n")) (rev pmap);
adamc@201 178 print ("TOTAL: " ^ LargeReal.toString (Time.toReal (foldl Time.+ Time.zeroTime (map #2 pmap))) ^ "\n");
adamc@201 179 print "\n"
adamc@201 180 end
adamc@55 181
adamc@201 182 fun timePrint (tr : ('src, 'dst) transform) input =
adamc@201 183 let
adamc@201 184 val (ro, pmap) = #time tr (input, [])
adamc@55 185 in
adamc@201 186 app (fn (name, time) =>
adamc@201 187 print (name ^ ": " ^ LargeReal.toString (Time.toReal time) ^ "\n")) (rev pmap);
adamc@201 188 print ("TOTAL: " ^ LargeReal.toString (Time.toReal (foldl Time.+ Time.zeroTime (map #2 pmap))) ^ "\n");
adamc@201 189 print "\n";
adamc@201 190 case ro of
adamc@201 191 NONE => print "Failure\n"
adamc@201 192 | SOME v =>
adamc@201 193 (print "Success\n";
adamc@201 194 Print.print (#print tr v);
adamc@201 195 print "\n")
adamc@55 196 end
adamc@55 197
adam@1362 198 fun runPrintCoreFuncs (tr : ('src, Core.file) transform) input =
adam@1362 199 (ErrorMsg.resetErrors ();
adam@1362 200 case #func tr input of
adam@1362 201 NONE => print "Failure\n"
adam@1362 202 | SOME file =>
adam@1362 203 (print "Success\n";
adam@1362 204 app (fn (d, _) =>
adam@1362 205 case d of
adam@1362 206 Core.DVal (x, _, t, _, _) => Print.preface(x, CorePrint.p_con CoreEnv.empty t)
adam@1362 207 | Core.DValRec xts => app (fn (x, _, t, _, _) => Print.preface(x, CorePrint.p_con CoreEnv.empty t)) xts
adam@1362 208 | _ => ()) file))
adam@1362 209
adamc@244 210 val parseUrs =
adamc@201 211 {func = fn filename => let
adamc@201 212 val fname = OS.FileSys.tmpName ()
adamc@201 213 val outf = TextIO.openOut fname
adamc@201 214 val () = TextIO.output (outf, "sig\n")
adamc@201 215 val inf = TextIO.openIn filename
adamc@201 216 fun loop () =
adamc@201 217 case TextIO.inputLine inf of
adamc@201 218 NONE => ()
adamc@201 219 | SOME line => (TextIO.output (outf, line);
adamc@201 220 loop ())
adamc@201 221 val () = loop ()
adamc@201 222 val () = TextIO.closeIn inf
adamc@201 223 val () = TextIO.closeOut outf
adamc@201 224
adamc@201 225 val () = (ErrorMsg.resetErrors ();
adamc@201 226 ErrorMsg.resetPositioning filename;
adamc@201 227 Lex.UserDeclarations.initialize ())
adamc@201 228 val file = TextIO.openIn fname
adamc@201 229 fun get _ = TextIO.input file
adamc@201 230 fun parseerror (s, p1, p2) = ErrorMsg.errorAt' (p1, p2) s
adamc@201 231 val lexer = LrParser.Stream.streamify (Lex.makeLexer get)
adamc@244 232 val (absyn, _) = UrwebP.parse (30, lexer, parseerror, ())
adamc@201 233 in
adamc@201 234 TextIO.closeIn file;
adamc@201 235 case absyn of
adamc@201 236 [(Source.DSgn ("?", (Source.SgnConst sgis, _)), _)] => sgis
adamc@201 237 | _ => (ErrorMsg.errorAt {file = filename,
adamc@201 238 first = {line = 0,
adamc@201 239 char = 0},
adamc@201 240 last = {line = 0,
adamc@201 241 char = 0}} "Not a signature";
adamc@201 242 [])
adamc@201 243 end
adamc@201 244 handle LrParser.ParseError => [],
adamc@201 245 print = Print.p_list_sep Print.PD.newline SourcePrint.p_sgn_item}
adamc@55 246
adamc@1 247 (* The main parsing routine *)
adamc@244 248 val parseUr = {
adamc@201 249 func = fn filename =>
adamc@201 250 let
adamc@201 251 val () = (ErrorMsg.resetErrors ();
adamc@201 252 ErrorMsg.resetPositioning filename;
adamc@201 253 Lex.UserDeclarations.initialize ())
adamc@201 254 val file = TextIO.openIn filename
adamc@201 255 fun get _ = TextIO.input file
adamc@201 256 fun parseerror (s, p1, p2) = ErrorMsg.errorAt' (p1, p2) s
adamc@201 257 val lexer = LrParser.Stream.streamify (Lex.makeLexer get)
adamc@244 258 val (absyn, _) = UrwebP.parse (30, lexer, parseerror, ())
adamc@201 259 in
adamc@201 260 TextIO.closeIn file;
adamc@201 261 case absyn of
adamc@201 262 [(Source.DSgn ("?", _), _)] =>
adamc@201 263 (ErrorMsg.errorAt {file = filename,
adamc@201 264 first = {line = 0,
adamc@201 265 char = 0},
adamc@201 266 last = {line = 0,
adamc@201 267 char = 0}} "File starts with 'sig'";
adamc@201 268 [])
adamc@201 269 | _ => absyn
adamc@201 270 end
adamc@201 271 handle LrParser.ParseError => [],
ziv@2202 272 print = SourcePrint.p_file}
adamc@56 273
adamc@768 274 fun p_job ({prefix, database, exe, sql, sources, debug, profile,
adamc@768 275 timeout, ffi, link, headers, scripts,
adamc@1171 276 clientToServer, effectful, benignEffectful, clientOnly, serverOnly, jsFuncs, ...} : job) =
adamc@270 277 let
adamc@270 278 open Print.PD
adamc@270 279 open Print
adamc@765 280
adamc@765 281 fun p_ffi name = p_list_sep (box []) (fn (m, s) =>
adamc@765 282 box [string name, space, string m, string ".", string s, newline])
adamc@270 283 in
adamc@274 284 box [if debug then
adamc@274 285 box [string "DEBUG", newline]
adamc@274 286 else
adamc@274 287 box [],
adamc@502 288 if profile then
adamc@502 289 box [string "PROFILE", newline]
adamc@502 290 else
adamc@502 291 box [],
adamc@274 292 case database of
adamc@270 293 NONE => string "No database."
adamc@270 294 | SOME db => string ("Database: " ^ db),
adamc@270 295 newline,
adamc@274 296 string "Exe: ",
adamc@274 297 string exe,
adamc@274 298 newline,
adamc@274 299 case sql of
adamc@274 300 NONE => string "No SQL file."
adamc@274 301 | SOME sql => string ("SQL fle: " ^ sql),
adamc@673 302 newline,
adamc@673 303 string "Timeout: ",
adamc@673 304 string (Int.toString timeout),
adamc@673 305 newline,
adamc@764 306 p_list_sep (box []) (fn s => box [string "Ffi", space, string s, newline]) ffi,
adamc@764 307 p_list_sep (box []) (fn s => box [string "Header", space, string s, newline]) headers,
adamc@766 308 p_list_sep (box []) (fn s => box [string "Script", space, string s, newline]) scripts,
adamc@764 309 p_list_sep (box []) (fn s => box [string "Link", space, string s, newline]) link,
adamc@765 310 p_ffi "ClientToServer" clientToServer,
adamc@765 311 p_ffi "Effectful" effectful,
adamc@1171 312 p_ffi "BenignEffectful" benignEffectful,
adamc@765 313 p_ffi "ClientOnly" clientOnly,
adamc@765 314 p_ffi "ServerOnly" serverOnly,
adamc@765 315 p_list_sep (box []) (fn ((m, s), s') =>
adamc@765 316 box [string "JsFunc", space, string m, string ".", string s,
adamc@765 317 space, string "=", space, string s', newline]) jsFuncs,
adamc@270 318 string "Sources:",
adamc@270 319 p_list string sources,
adamc@270 320 newline]
adamc@270 321 end
adamc@270 322
adamc@270 323 fun trim s =
adamc@270 324 let
adamc@270 325 val (_, s) = Substring.splitl Char.isSpace s
adamc@270 326 val (s, _) = Substring.splitr Char.isSpace s
adamc@270 327 in
adamc@270 328 s
adamc@270 329 end
adamc@270 330
adam@1433 331 val trimS = Substring.string o trim o Substring.full
adam@1433 332
adamc@794 333 structure M = BinaryMapFn(struct
adamc@794 334 type ord_key = string
adamc@794 335 val compare = String.compare
adamc@794 336 end)
adamc@794 337
ezyang@1739 338 (* XXX ezyang: pathmap gets initialized /really early/, before
ezyang@1739 339 * we do any options parsing. So libUr will always point to the
ezyang@1739 340 * default. We override it explicitly in enableBoot *)
ezyang@1739 341 val pathmap = ref (M.insert (M.empty, "", Settings.libUr ()))
adamc@1089 342
adamc@1089 343 fun addPath (k, v) = pathmap := M.insert (!pathmap, k, v)
adamc@1089 344
ezyang@1739 345 (* XXX ezyang: this is not right; it probably doesn't work in
ezyang@1739 346 * the case of separate build and src trees *)
ezyang@1739 347 fun enableBoot () =
ezyang@1739 348 (Settings.configBin := OS.Path.joinDirFile {dir = Config.builddir, file = "bin"};
ezyang@1739 349 Settings.configSrcLib := OS.Path.joinDirFile {dir = Config.builddir, file = "lib"};
ezyang@1739 350 (* joinDirFile is annoying... (ArcError; it doesn't like
ezyang@1739 351 * slashes in file) *)
ezyang@1739 352 Settings.configLib := Config.builddir ^ "/src/c/.libs";
ezyang@1739 353 Settings.configInclude := OS.Path.joinDirFile {dir = Config.builddir ^ "/include", file = "urweb"};
ezyang@1739 354 Settings.configSitelisp := Config.builddir ^ "/src/elisp";
ezyang@1739 355 addPath ("", Settings.libUr ()))
ezyang@1739 356
adam@1296 357 fun capitalize "" = ""
adam@1296 358 | capitalize s = str (Char.toUpper (String.sub (s, 0))) ^ String.extract (s, 1, NONE)
adam@1296 359
adam@1296 360 fun institutionalizeJob (job : job) =
adam@1483 361 (Settings.setDebug (#debug job);
adam@1483 362 Settings.setUrlPrefix (#prefix job);
adam@1296 363 Settings.setTimeout (#timeout job);
adam@1296 364 Settings.setHeaders (#headers job);
adam@1296 365 Settings.setScripts (#scripts job);
adam@1296 366 Settings.setClientToServer (#clientToServer job);
adam@1296 367 Settings.setEffectful (#effectful job);
adam@1296 368 Settings.setBenignEffectful (#benignEffectful job);
adam@1296 369 Settings.setClientOnly (#clientOnly job);
adam@1296 370 Settings.setServerOnly (#serverOnly job);
adam@1296 371 Settings.setJsFuncs (#jsFuncs job);
adam@1296 372 Settings.setRewriteRules (#rewrites job);
adam@1296 373 Settings.setUrlRules (#filterUrl job);
adam@1296 374 Settings.setMimeRules (#filterMime job);
adam@1465 375 Settings.setRequestHeaderRules (#filterRequest job);
adam@1465 376 Settings.setResponseHeaderRules (#filterResponse job);
adam@1799 377 Settings.setEnvVarRules (#filterEnv job);
adam@1296 378 Option.app Settings.setProtocol (#protocol job);
adam@1296 379 Option.app Settings.setDbms (#dbms job);
adam@1296 380 Settings.setSafeGets (#safeGets job);
adam@1332 381 Settings.setOnError (#onError job);
adam@1408 382 Settings.setMinHeap (#minHeap job);
adam@1408 383 Settings.setSigFile (#sigFile job))
adam@1296 384
adam@1603 385 datatype commentableLine =
adam@1603 386 EndOfFile
adam@1603 387 | OnlyComment
adam@1603 388 | Content of string
adam@1603 389
adam@1331 390 fun inputCommentableLine inf =
adam@1603 391 case TextIO.inputLine inf of
adam@1603 392 NONE => EndOfFile
adam@1603 393 | SOME s =>
adam@1603 394 let
adam@1603 395 val (befor, after) = Substring.splitl (fn ch => ch <> #"#") (Substring.full s)
adam@1603 396 in
adam@1603 397 if not (Substring.isEmpty after)
adam@1603 398 andalso Substring.foldl (fn (ch, b) => b andalso Char.isSpace ch) true befor then
adam@1603 399 OnlyComment
adam@1603 400 else
adam@1603 401 let
adam@1603 402 val s = #1 (Substring.splitr (not o Char.isSpace) befor)
adam@1603 403 in
adam@1603 404 Content (Substring.string (if Substring.size s > 0 andalso Char.isSpace (Substring.sub (s, Substring.size s - 1)) then
adam@1603 405 if Substring.size s > 1 andalso Char.isSpace (Substring.sub (s, Substring.size s - 2)) then
adam@1603 406 Substring.trimr 2 s
adam@1603 407 else
adam@1603 408 Substring.trimr 1 s
adam@1603 409 else
adam@1603 410 s))
adam@1603 411 end
adam@1603 412 end
adam@1331 413
adam@1780 414 val lastUrp = ref ""
adam@1780 415
adamc@1082 416 fun parseUrp' accLibs fname =
adam@1780 417 (if !lastUrp = fname then
adam@1780 418 ()
adam@1780 419 else
adam@1780 420 ModDb.reset ();
adam@1780 421 lastUrp := fname;
adam@1780 422 if not (Posix.FileSys.access (fname ^ ".urp", []) orelse Posix.FileSys.access (fname ^ "/lib.urp", []))
adam@1780 423 andalso Posix.FileSys.access (fname ^ ".ur", []) then
adam@1780 424 let
adam@1780 425 val job = {prefix = "/",
adam@1780 426 database = NONE,
adam@1780 427 sources = [fname],
adam@1780 428 exe = fname ^ ".exe",
adam@1780 429 sql = NONE,
adam@1780 430 debug = Settings.getDebug (),
adam@1780 431 profile = false,
adam@1780 432 timeout = 60,
adam@1780 433 ffi = [],
adam@1780 434 link = [],
adam@1780 435 linker = NONE,
adam@1780 436 headers = [],
adam@1780 437 scripts = [],
adam@1780 438 clientToServer = [],
adam@1780 439 effectful = [],
adam@1780 440 benignEffectful = [],
adam@1780 441 clientOnly = [],
adam@1780 442 serverOnly = [],
adam@1780 443 jsFuncs = [],
adam@1780 444 rewrites = [{pkind = Settings.Any,
adam@1780 445 kind = Settings.Prefix,
adam@1780 446 from = capitalize (OS.Path.file fname) ^ "/", to = "",
adam@1780 447 hyphenate = false}],
adam@1780 448 filterUrl = [],
adam@1780 449 filterMime = [],
adam@1780 450 filterRequest = [],
adam@1780 451 filterResponse = [],
adam@1799 452 filterEnv = [],
adam@1780 453 protocol = NONE,
adam@1780 454 dbms = NONE,
adam@1780 455 sigFile = NONE,
adam@1780 456 safeGets = [],
adam@1780 457 onError = NONE,
adam@1780 458 minHeap = 0}
adam@1780 459 in
adam@1780 460 institutionalizeJob job;
adam@1780 461 {Job = job, Libs = []}
adam@1780 462 end
adam@1780 463 else
adam@1780 464 let
adam@2046 465 val thisPath = OS.Path.dir fname
adam@2046 466
adam@1780 467 val pathmap = ref (!pathmap)
adam@1780 468 val bigLibs = ref []
adamc@767 469
adam@1780 470 fun pu filename =
adam@1780 471 let
adam@1780 472 val filename = OS.Path.mkAbsolute {path = filename, relativeTo = OS.FileSys.getDir ()}
adam@1338 473
adam@1780 474 val dir = OS.Path.dir filename
adam@1780 475 fun opener () = TextIO.openIn (OS.Path.joinBaseExt {base = filename, ext = SOME "urp"})
adamc@1151 476
adam@1780 477 val inf = opener ()
adamc@1151 478
adam@1780 479 fun hasSpaceLine () =
adam@1780 480 case inputCommentableLine inf of
adam@1780 481 Content s => s = "debug" orelse s = "profile"
adam@1780 482 orelse CharVector.exists (fn ch => ch = #" " orelse ch = #"\t") s orelse hasSpaceLine ()
adam@1780 483 | EndOfFile => false
adam@1780 484 | OnlyComment => hasSpaceLine ()
adamc@1151 485
adam@1780 486 val hasBlankLine = hasSpaceLine ()
adamc@1151 487
adam@1780 488 val inf = (TextIO.closeIn inf; opener ())
adamc@767 489
adam@1780 490 fun pathify fname =
adam@1780 491 if size fname > 0 andalso String.sub (fname, 0) = #"$" then
adam@1780 492 let
adam@1780 493 val fname' = Substring.extract (fname, 1, NONE)
adam@1780 494 val (befor, after) = Substring.splitl (fn ch => ch <> #"/") fname'
adam@1780 495 in
adam@1780 496 case M.find (!pathmap, Substring.string befor) of
adam@1780 497 NONE => fname
adam@1780 498 | SOME rep => rep ^ Substring.string after
adam@1780 499 end
adam@1780 500 else
adam@1780 501 fname
adam@1296 502
adam@1780 503 fun relify fname =
adam@1780 504 let
adam@1780 505 val fname = pathify fname
adam@1780 506 in
adam@1780 507 OS.Path.concat (dir, fname)
adam@1780 508 handle OS.Path.Path => fname
adam@1780 509 end
adamc@767 510
adam@1780 511 fun libify path =
adam@1780 512 (if Posix.FileSys.access (path ^ ".urp", []) then
adam@1780 513 path
adam@1780 514 else
adam@1780 515 path ^ "/lib")
adam@1780 516 handle SysErr => path
adam@1780 517
adam@1780 518 fun libify' path =
adam@1780 519 (if Posix.FileSys.access (relify path ^ ".urp", []) then
adam@1780 520 path
adam@1780 521 else
adam@1780 522 path ^ "/lib")
adam@1780 523 handle SysErr => path
adam@1780 524
adam@1780 525 val absDir = OS.Path.mkAbsolute {path = dir, relativeTo = OS.FileSys.getDir ()}
adam@1780 526
adam@1780 527 fun relifyA fname =
adam@1780 528 OS.Path.mkAbsolute {path = pathify fname, relativeTo = absDir}
adam@1780 529
adam@1780 530 fun readSources acc =
adam@1780 531 case inputCommentableLine inf of
adam@1780 532 Content line =>
adam@1780 533 let
adam@1780 534 val acc = if CharVector.all Char.isSpace line then
adam@1780 535 acc
adam@1780 536 else
adam@1780 537 let
adam@1780 538 val fname = String.implode (List.filter (fn x => not (Char.isSpace x))
adam@1780 539 (String.explode line))
adam@1780 540 val fname = relifyA fname
adam@1780 541 in
adam@1780 542 fname :: acc
adam@1780 543 end
adam@1780 544 in
adam@1780 545 readSources acc
adam@1780 546 end
adam@1780 547 | OnlyComment => readSources acc
adam@1780 548 | EndOfFile => rev acc
adam@1780 549
adam@1780 550 val prefix = ref (case Settings.getUrlPrefixFull () of "/" => NONE | s => SOME s)
adam@1780 551 val database = ref (Settings.getDbstring ())
adam@1780 552 val exe = ref (Settings.getExe ())
adam@1780 553 val sql = ref (Settings.getSql ())
adam@1780 554 val debug = ref (Settings.getDebug ())
adam@1780 555 val profile = ref false
adam@1780 556 val timeout = ref NONE
adam@1780 557 val ffi = ref []
adam@1780 558 val link = ref []
adam@1780 559 val linker = ref NONE
adam@1780 560 val headers = ref []
adam@1780 561 val scripts = ref []
adam@1780 562 val clientToServer = ref []
adam@1780 563 val effectful = ref []
adam@1780 564 val benignEffectful = ref []
adam@1780 565 val clientOnly = ref []
adam@1780 566 val serverOnly = ref []
adam@1780 567 val jsFuncs = ref []
adam@1780 568 val rewrites = ref []
adam@1780 569 val url = ref []
adam@1780 570 val mime = ref []
adam@1780 571 val request = ref []
adam@1780 572 val response = ref []
adam@1799 573 val env = ref []
adam@1780 574 val libs = ref []
adam@1780 575 val protocol = ref NONE
adam@1780 576 val dbms = ref NONE
adam@1780 577 val sigFile = ref (Settings.getSigFile ())
adam@1780 578 val safeGets = ref []
adam@1780 579 val onError = ref NONE
adam@1780 580 val minHeap = ref 0
adam@1780 581
adam@1780 582 fun finish sources =
adam@1780 583 let
adam@1780 584 val job = {
adam@1780 585 prefix = Option.getOpt (!prefix, "/"),
adam@1780 586 database = !database,
adam@1780 587 exe = Option.getOpt (!exe, OS.Path.joinBaseExt {base = OS.Path.base filename,
adam@1780 588 ext = SOME "exe"}),
adam@1780 589 sql = !sql,
adam@1780 590 debug = !debug,
adam@1780 591 profile = !profile,
adam@1780 592 timeout = Option.getOpt (!timeout, 60),
adam@1780 593 ffi = rev (!ffi),
adam@1780 594 link = rev (!link),
adam@1780 595 linker = !linker,
adam@1780 596 headers = rev (!headers),
adam@1780 597 scripts = rev (!scripts),
adam@1780 598 clientToServer = rev (!clientToServer),
adam@1780 599 effectful = rev (!effectful),
adam@1780 600 benignEffectful = rev (!benignEffectful),
adam@1780 601 clientOnly = rev (!clientOnly),
adam@1780 602 serverOnly = rev (!serverOnly),
adam@1780 603 jsFuncs = rev (!jsFuncs),
adam@1780 604 rewrites = rev (!rewrites),
adam@1780 605 filterUrl = rev (!url),
adam@1780 606 filterMime = rev (!mime),
adam@1780 607 filterRequest = rev (!request),
adam@1780 608 filterResponse = rev (!response),
adam@1799 609 filterEnv = rev (!env),
adam@1780 610 sources = sources,
adam@1780 611 protocol = !protocol,
adam@1780 612 dbms = !dbms,
adam@1780 613 sigFile = !sigFile,
adam@1780 614 safeGets = rev (!safeGets),
adam@1780 615 onError = !onError,
adam@1780 616 minHeap = !minHeap
adam@1780 617 }
adam@1780 618
adam@1780 619 fun mergeO f (old, new) =
adam@1780 620 case (old, new) of
adam@1780 621 (NONE, _) => new
adam@1780 622 | (_, NONE) => old
adam@1780 623 | (SOME v1, SOME v2) => SOME (f (v1, v2))
adam@1780 624
adam@1780 625 fun same desc = mergeO (fn (x : string, y) =>
adam@1780 626 (if x = y then
adam@1780 627 ()
adam@1780 628 else
adam@1780 629 ErrorMsg.error ("Multiple "
adam@1780 630 ^ desc ^ " values that don't agree");
adam@1780 631 x))
adam@1780 632
adam@1780 633 fun merge (old : job, new : job) = {
adam@1780 634 prefix = case #prefix old of
adam@1780 635 "/" => #prefix new
adam@1780 636 | pold => case #prefix new of
adam@1780 637 "/" => pold
adam@1780 638 | pnew => (if pold = pnew then
adam@1780 639 ()
adam@1780 640 else
adam@1780 641 ErrorMsg.error ("Multiple prefix values that don't agree: "
adam@1780 642 ^ pold ^ ", " ^ pnew);
adam@1780 643 pold),
adam@1780 644 database = mergeO (fn (old, _) => old) (#database old, #database new),
adam@1780 645 exe = #exe old,
adam@1780 646 sql = #sql old,
adam@1780 647 debug = #debug old orelse #debug new,
adam@1780 648 profile = #profile old orelse #profile new,
adam@1780 649 timeout = #timeout old,
adam@1780 650 ffi = #ffi old @ #ffi new,
adam@1780 651 link = #link old @ #link new,
adam@1780 652 linker = mergeO (fn (_, new) => new) (#linker old, #linker new),
adam@1780 653 headers = #headers old @ #headers new,
adam@1780 654 scripts = #scripts old @ #scripts new,
adam@1780 655 clientToServer = #clientToServer old @ #clientToServer new,
adam@1780 656 effectful = #effectful old @ #effectful new,
adam@1780 657 benignEffectful = #benignEffectful old @ #benignEffectful new,
adam@1780 658 clientOnly = #clientOnly old @ #clientOnly new,
adam@1780 659 serverOnly = #serverOnly old @ #serverOnly new,
adam@1780 660 jsFuncs = #jsFuncs old @ #jsFuncs new,
adam@1780 661 rewrites = #rewrites old @ #rewrites new,
adam@1780 662 filterUrl = #filterUrl old @ #filterUrl new,
adam@1780 663 filterMime = #filterMime old @ #filterMime new,
adam@1780 664 filterRequest = #filterRequest old @ #filterRequest new,
adam@1780 665 filterResponse = #filterResponse old @ #filterResponse new,
adam@1799 666 filterEnv = #filterEnv old @ #filterEnv new,
adam@1780 667 sources = #sources new
adam@1780 668 @ List.filter (fn s => List.all (fn s' => s' <> s) (#sources new))
adam@1780 669 (#sources old),
adam@1780 670 protocol = mergeO #2 (#protocol old, #protocol new),
adam@1780 671 dbms = mergeO #2 (#dbms old, #dbms new),
adam@1780 672 sigFile = mergeO #2 (#sigFile old, #sigFile new),
adam@1780 673 safeGets = #safeGets old @ #safeGets new,
adam@1780 674 onError = mergeO #2 (#onError old, #onError new),
adam@1780 675 minHeap = Int.max (#minHeap old, #minHeap new)
adam@1780 676 }
adam@1780 677 in
adam@1780 678 if accLibs then
adam@1780 679 foldr (fn (job', job) => merge (job, job')) job (!libs)
adam@1780 680 else
adam@1780 681 job
adam@1780 682 end
adam@1780 683
adam@1780 684 fun parsePkind s =
adam@1780 685 case s of
adam@1780 686 "all" => Settings.Any
adam@1780 687 | "url" => Settings.Url
adam@1780 688 | "table" => Settings.Table
adam@1780 689 | "sequence" => Settings.Sequence
adam@1780 690 | "view" => Settings.View
adam@1780 691 | "relation" => Settings.Relation
adam@1780 692 | "cookie" => Settings.Cookie
adam@1780 693 | "style" => Settings.Style
adam@1780 694 | _ => (ErrorMsg.error "Bad path kind spec";
adam@1780 695 Settings.Any)
adam@1780 696
adam@1780 697 fun parseFrom s =
adam@1780 698 if size s > 1 andalso String.sub (s, size s - 2) = #"/" andalso String.sub (s, size s - 1) = #"*" then
adam@1780 699 (Settings.Prefix, String.substring (s, 0, size s - 1))
adam@1296 700 else
adam@1780 701 (Settings.Exact, s)
adamc@767 702
adam@1780 703 fun parseFkind s =
adam@1780 704 case s of
adam@1780 705 "url" => url
adam@1780 706 | "mime" => mime
adam@1780 707 | "requestHeader" => request
adam@1780 708 | "responseHeader" => response
adam@1799 709 | "env" => env
adam@1780 710 | _ => (ErrorMsg.error "Bad filter kind";
adam@1780 711 url)
adam@1780 712
adam@1780 713 fun parsePattern s =
adam@1780 714 if size s > 0 andalso String.sub (s, size s - 1) = #"*" then
adam@1780 715 (Settings.Prefix, String.substring (s, 0, size s - 1))
adam@1296 716 else
adam@1780 717 (Settings.Exact, s)
adamc@767 718
adam@1780 719 fun read () =
adam@1780 720 case inputCommentableLine inf of
adam@1780 721 EndOfFile => finish []
adam@1780 722 | OnlyComment => read ()
adam@1780 723 | Content "" => finish (readSources [])
adam@1780 724 | Content line =>
adam@1780 725 let
adam@1780 726 val (cmd, arg) = Substring.splitl (fn x => not (Char.isSpace x)) (Substring.full line)
adam@1780 727 val cmd = Substring.string (trim cmd)
adam@1780 728 val arg = Substring.string (trim arg)
adamc@794 729
adam@1780 730 fun ffiS () =
adam@1780 731 case String.fields (fn ch => ch = #".") arg of
adam@1780 732 [m, x] => (m, x)
adam@1780 733 | _ => (ErrorMsg.error (cmd ^ " argument not of the form Module.func");
adam@1780 734 ("", ""))
adamc@794 735
adam@1780 736 fun ffiM () =
adam@1780 737 case String.fields (fn ch => ch = #"=") arg of
adam@1780 738 [f, s] =>
adam@1780 739 let
adam@1780 740 val f = trimS f
adam@1780 741 val s = trimS s
adam@1780 742 in
adam@1780 743 case String.fields (fn ch => ch = #".") f of
adam@1780 744 [m, x] => ((m, x), s)
adam@1780 745 | _ => (ErrorMsg.error (cmd ^ " argument not of the form Module.func=func'");
adam@1780 746 (("", ""), ""))
adam@1780 747 end
adam@1780 748 | _ => (ErrorMsg.error (cmd ^ " argument not of the form Module.func=func'");
adam@1780 749 (("", ""), ""))
adam@1780 750 in
adam@1780 751 case cmd of
adam@1780 752 "prefix" => prefix := SOME arg
adam@1780 753 | "database" =>
adam@1780 754 (case !database of
adam@1780 755 NONE => database := SOME arg
adam@1780 756 | SOME _ => ())
adam@1780 757 | "dbms" =>
adam@1780 758 (case !dbms of
adam@1780 759 NONE => dbms := SOME arg
adam@1780 760 | SOME _ => ())
adam@1780 761 | "sigfile" =>
adam@1780 762 (case !sigFile of
adam@1780 763 NONE => sigFile := SOME arg
adam@1780 764 | SOME _ => ())
adam@1780 765 | "exe" =>
adam@1780 766 (case !exe of
adam@1780 767 NONE => exe := SOME (relify arg)
adam@1780 768 | SOME _ => ())
adam@1780 769 | "sql" =>
adam@1780 770 (case !sql of
adam@1780 771 NONE => sql := SOME (relify arg)
adam@1780 772 | SOME _ => ())
adam@1780 773 | "debug" => debug := true
adam@1780 774 | "profile" => profile := true
adam@1780 775 | "timeout" =>
adam@1780 776 (case !timeout of
adam@1780 777 NONE => ()
adam@1780 778 | SOME _ => ErrorMsg.error "Duplicate 'timeout' directive";
adam@1780 779 timeout := SOME (valOf (Int.fromString arg)))
adam@1780 780 | "ffi" => ffi := relify arg :: !ffi
adam@1780 781 | "link" => let
adam@1780 782 val arg = if size arg >= 1
adam@1780 783 andalso String.sub (arg, 0) = #"-" then
adam@1780 784 arg
adam@1780 785 else
adam@1780 786 relifyA arg
adam@1780 787 in
adam@1780 788 link := arg :: !link
adam@1780 789 end
adam@1780 790 | "linker" => linker := SOME arg
adam@1780 791 | "include" => headers := relifyA arg :: !headers
adam@1780 792 | "script" => scripts := arg :: !scripts
adam@1780 793 | "clientToServer" => clientToServer := ffiS () :: !clientToServer
adam@1780 794 | "safeGet" => safeGets := arg :: !safeGets
adam@1780 795 | "effectful" => effectful := ffiS () :: !effectful
adam@1780 796 | "benignEffectful" => benignEffectful := ffiS () :: !benignEffectful
adam@1780 797 | "clientOnly" => clientOnly := ffiS () :: !clientOnly
adam@1780 798 | "serverOnly" => serverOnly := ffiS () :: !serverOnly
adam@1780 799 | "jsFunc" => jsFuncs := ffiM () :: !jsFuncs
adam@1780 800 | "rewrite" =>
adam@1780 801 let
adam@1780 802 fun doit (pkind, from, to, hyph) =
adam@1780 803 let
adam@1780 804 val pkind = parsePkind pkind
adam@1780 805 val (kind, from) = parseFrom from
adam@1780 806 in
adam@1780 807 rewrites := {pkind = pkind, kind = kind, from = from, to = to, hyphenate = hyph} :: !rewrites
adam@1780 808 end
adam@1780 809 in
adam@1780 810 case String.tokens Char.isSpace arg of
adam@1780 811 [pkind, from, to, "[-]"] => doit (pkind, from, to, true)
adam@1780 812 | [pkind, from, "[-]"] => doit (pkind, from, "", true)
adam@1780 813 | [pkind, from, to] => doit (pkind, from, to, false)
adam@1780 814 | [pkind, from] => doit (pkind, from, "", false)
adam@1780 815 | _ => ErrorMsg.error "Bad 'rewrite' syntax"
adam@1780 816 end
adam@1780 817 | "allow" =>
adam@1780 818 (case String.tokens Char.isSpace arg of
adam@1780 819 [fkind, pattern] =>
adam@1780 820 let
adam@1780 821 val fkind = parseFkind fkind
adam@1780 822 val (kind, pattern) = parsePattern pattern
adam@1780 823 in
adam@1780 824 fkind := {action = Settings.Allow, kind = kind, pattern = pattern} :: !fkind
adam@1780 825 end
adam@1780 826 | _ => ErrorMsg.error "Bad 'allow' syntax")
adam@1780 827 | "deny" =>
adam@1780 828 (case String.tokens Char.isSpace arg of
adam@1780 829 [fkind, pattern] =>
adam@1780 830 let
adam@1780 831 val fkind = parseFkind fkind
adam@1780 832 val (kind, pattern) = parsePattern pattern
adam@1780 833 in
adam@1780 834 fkind := {action = Settings.Deny, kind = kind, pattern = pattern} :: !fkind
adam@1780 835 end
adam@1780 836 | _ => ErrorMsg.error "Bad 'deny' syntax")
adam@1780 837 | "library" => if accLibs then
adam@1780 838 libs := pu (libify (relify arg)) :: !libs
adam@1780 839 else
adam@1780 840 bigLibs := libify' arg :: !bigLibs
adam@1780 841 | "path" =>
adam@1780 842 (case String.fields (fn ch => ch = #"=") arg of
adam@1780 843 [n, v] => ((pathmap := M.insert (!pathmap, n, OS.Path.mkAbsolute {path = v, relativeTo = dir}))
adam@1780 844 handle OS.Path.Path => ErrorMsg.error "Invalid 'path' directory argument")
adam@1780 845 | _ => ErrorMsg.error "path argument not of the form name=value'")
adam@1780 846 | "onError" =>
adam@1780 847 (case String.fields (fn ch => ch = #".") arg of
adam@1780 848 m1 :: (fs as _ :: _) =>
adam@1780 849 onError := SOME (m1, List.take (fs, length fs - 1), List.last fs)
adam@1780 850 | _ => ErrorMsg.error "invalid 'onError' argument")
adam@1780 851 | "limit" =>
adam@1780 852 (case String.fields Char.isSpace arg of
adam@1780 853 [class, num] =>
adam@1780 854 (case Int.fromString num of
adam@1780 855 NONE => ErrorMsg.error ("invalid limit number '" ^ num ^ "'")
adam@1780 856 | SOME n =>
adam@1780 857 if n < 0 then
adam@1780 858 ErrorMsg.error ("invalid limit number '" ^ num ^ "'")
adam@1780 859 else
adam@1780 860 Settings.addLimit (class, n))
adam@1780 861 | _ => ErrorMsg.error "invalid 'limit' arguments")
adam@1780 862 | "minHeap" =>
adam@1780 863 (case Int.fromString arg of
adam@1780 864 NONE => ErrorMsg.error ("invalid min heap '" ^ arg ^ "'")
adam@1780 865 | SOME n => minHeap := n)
vshabanoff@1815 866 | "coreInline" =>
vshabanoff@1815 867 (case Int.fromString arg of
vshabanoff@1815 868 NONE => ErrorMsg.error ("invalid core inline level '" ^ arg ^ "'")
vshabanoff@1815 869 | SOME n => Settings.setCoreInline n)
vshabanoff@1815 870 | "monoInline" =>
vshabanoff@1815 871 (case Int.fromString arg of
vshabanoff@1815 872 NONE => ErrorMsg.error ("invalid mono inline level '" ^ arg ^ "'")
vshabanoff@1815 873 | SOME n => Settings.setMonoInline n)
adam@1780 874 | "alwaysInline" => Settings.addAlwaysInline arg
adam@1966 875 | "neverInline" => Settings.addNeverInline arg
adam@1780 876 | "noXsrfProtection" => Settings.addNoXsrfProtection arg
adam@1780 877 | "timeFormat" => Settings.setTimeFormat arg
adam@1953 878 | "noMangleSql" => Settings.setMangleSql false
adam@1956 879 | "html5" => Settings.setIsHtml5 true
adam@2010 880 | "lessSafeFfi" => Settings.setLessSafeFfi true
adamc@794 881
adam@2046 882 | "file" =>
adam@2046 883 (case String.fields Char.isSpace arg of
adam@2046 884 [uri, fname] => (Settings.setFilePath thisPath;
adam@2046 885 Settings.addFile {Uri = uri,
adam@2046 886 LoadFromFilename = fname})
adam@2046 887 | _ => ErrorMsg.error "Bad 'file' arguments")
adam@2046 888
adam@1780 889 | _ => ErrorMsg.error ("Unrecognized command '" ^ cmd ^ "'");
adam@1780 890 read ()
adam@1780 891 end
adamc@794 892
adam@1780 893 val job = if hasBlankLine then
adam@1780 894 read ()
adam@1780 895 else
adam@1780 896 finish (readSources [])
adam@1780 897 in
adam@1780 898 TextIO.closeIn inf;
adam@1780 899 institutionalizeJob job;
adam@1780 900 job
adam@1780 901 end
adam@1780 902 in
adam@1780 903 {Job = pu fname, Libs = !bigLibs}
adam@1780 904 end)
adamc@767 905
adamc@1082 906 fun p_job' {Job = j, Libs = _ : string list} = p_job j
adamc@1082 907
adamc@270 908 val parseUrp = {
adamc@1083 909 func = #Job o parseUrp' true,
adamc@270 910 print = p_job
adamc@270 911 }
adamc@270 912
adamc@1082 913 val parseUrp' = {
adamc@1083 914 func = parseUrp' false,
adamc@1082 915 print = p_job'
adamc@1082 916 }
adamc@1082 917
adamc@270 918 val toParseJob = transform parseUrp "parseJob"
adamc@1082 919 val toParseJob' = transform parseUrp' "parseJob'"
adamc@1082 920
adamc@1082 921 fun op o (tr2 : ('b, 'c) transform, tr1 : ('a, 'b) transform) = {
adamc@1082 922 func = fn input => case #func tr1 input of
adamc@1082 923 NONE => NONE
adamc@1082 924 | SOME v => #func tr2 v,
adamc@1082 925 print = #print tr2,
adamc@1082 926 time = fn (input, pmap) => let
adamc@1082 927 val (ro, pmap) = #time tr1 (input, pmap)
adamc@1082 928 in
adamc@1082 929 case ro of
adamc@1082 930 NONE => (NONE, pmap)
adamc@1082 931 | SOME v => #time tr2 (v, pmap)
adamc@1082 932 end
adamc@1082 933 }
adamc@270 934
adamc@1090 935 structure SM = BinaryMapFn(struct
adamc@1090 936 type ord_key = string
adamc@1090 937 val compare = String.compare
adamc@1090 938 end)
adamc@1090 939
adamc@1090 940 val moduleRoots = ref ([] : (string * string) list)
adam@1868 941 fun addModuleRoot (k, v) = moduleRoots :=
adam@1868 942 (OS.Path.mkAbsolute {path = k,
adam@1868 943 relativeTo = OS.FileSys.getDir ()},
adam@1868 944 v) :: !moduleRoots
adamc@1090 945
adam@1743 946 structure SK = struct
adam@1743 947 type ord_key = string
adam@1743 948 val compare = String.compare
adam@1743 949 end
adam@1743 950
adam@1743 951 structure SS = BinarySetFn(SK)
adam@1743 952 structure SM = BinaryMapFn(SK)
adamc@1090 953
adam@1773 954 exception MissingFile of string
adam@1773 955
adamc@201 956 val parse = {
adam@1294 957 func = fn {database, sources = fnames, ffi, onError, ...} : job =>
adamc@201 958 let
adamc@1090 959 val mrs = !moduleRoots
adamc@1090 960
adamc@834 961 val anyErrors = ref false
adamc@834 962 fun checkErrors () = anyErrors := (!anyErrors orelse ErrorMsg.anyErrors ())
adam@1343 963 fun nameOf fname =
adam@1343 964 let
adam@1343 965 val fname = OS.Path.file fname
adam@1343 966 val fst = if size fname = 0 then #"!" else String.sub (fname, 0)
adam@1343 967 in
adam@1343 968 if not (Char.isAlpha fst) then
adam@1343 969 ErrorMsg.error ("Filename doesn't start with letter: " ^ fname)
adam@1343 970 else if CharVector.exists (fn ch => not (Char.isAlphaNum ch) andalso ch <> #"_") fname then
adam@1343 971 ErrorMsg.error ("Filename contains a character that isn't alphanumeric or underscore: " ^ fname)
adam@1343 972 else
adam@1343 973 ();
adam@1343 974 capitalize fname
adam@1343 975 end
adamc@109 976
adamc@764 977 fun parseFfi fname =
adamc@764 978 let
adamc@764 979 val mname = nameOf fname
adamc@764 980 val urs = OS.Path.joinBaseExt {base = fname, ext = SOME "urs"}
adamc@764 981
adamc@764 982 val loc = {file = urs,
adamc@764 983 first = ErrorMsg.dummyPos,
adamc@764 984 last = ErrorMsg.dummyPos}
adamc@764 985
adamc@764 986 val sgn = (Source.SgnConst (#func parseUrs urs), loc)
adamc@764 987 in
adamc@834 988 checkErrors ();
adam@1733 989 (Source.DFfiStr (mname, sgn, if !Elaborate.incremental then SOME (OS.FileSys.modTime urs) else NONE), loc)
adamc@764 990 end
adamc@764 991
adamc@1090 992 val defed = ref SS.empty
adamc@1092 993 val fulls = ref SS.empty
adamc@1090 994
adamc@201 995 fun parseOne fname =
adamc@201 996 let
adamc@201 997 val mname = nameOf fname
adamc@244 998 val ur = OS.Path.joinBaseExt {base = fname, ext = SOME "ur"}
adamc@244 999 val urs = OS.Path.joinBaseExt {base = fname, ext = SOME "urs"}
adamc@56 1000
adam@1773 1001 val () = if Posix.FileSys.access (ur, []) then
adam@1773 1002 ()
adam@1773 1003 else
adam@1773 1004 raise MissingFile ur
adam@1773 1005
adamc@201 1006 val sgnO =
adamc@244 1007 if Posix.FileSys.access (urs, []) then
adamc@244 1008 SOME (Source.SgnConst (#func parseUrs urs),
adamc@244 1009 {file = urs,
adamc@201 1010 first = ErrorMsg.dummyPos,
adamc@201 1011 last = ErrorMsg.dummyPos})
adamc@834 1012 before checkErrors ()
adamc@201 1013 else
adamc@201 1014 NONE
adamc@56 1015
adamc@244 1016 val loc = {file = ur,
adamc@201 1017 first = ErrorMsg.dummyPos,
adamc@201 1018 last = ErrorMsg.dummyPos}
adamc@56 1019
adam@1738 1020 val urt = OS.FileSys.modTime ur
adam@1738 1021 val urst = (OS.FileSys.modTime urs) handle _ => urt
adam@1738 1022
adamc@244 1023 val ds = #func parseUr ur
adam@1738 1024 val d = (Source.DStr (mname, sgnO, if !Elaborate.incremental then SOME (if Time.> (urt, urst) then urt else urst) else NONE,
adam@1868 1025 (Source.StrConst ds, loc), false), loc)
adamc@1090 1026
adamc@1126 1027 val fname = OS.Path.mkCanonical fname
adamc@1090 1028 val d = case List.find (fn (root, name) =>
adamc@1090 1029 String.isPrefix (root ^ "/") fname) mrs of
adamc@1090 1030 NONE => d
adamc@1090 1031 | SOME (root, name) =>
adamc@1090 1032 let
adamc@1090 1033 val fname = String.extract (fname, size root + 1, NONE)
adamc@1090 1034 val pieces = name :: String.tokens (fn ch => ch = #"/") fname
adamc@1090 1035 val pieces = List.filter (fn s => size s > 0
adamc@1090 1036 andalso Char.isAlpha (String.sub (s, 0)))
adamc@1090 1037 pieces
adamc@1090 1038 val pieces = map capitalize pieces
adamc@1092 1039 val full = String.concatWith "." pieces
adamc@1090 1040
adam@1868 1041 fun makeD first prefix pieces =
adamc@1090 1042 case pieces of
adamc@1090 1043 [] => (ErrorMsg.error "Empty module path";
adamc@1090 1044 (Source.DStyle "Boo", loc))
adamc@1090 1045 | [_] => d
adamc@1090 1046 | piece :: pieces =>
adamc@1090 1047 let
adamc@1146 1048 val this = case prefix of
adamc@1146 1049 "" => piece
adamc@1146 1050 | _ => prefix ^ "." ^ piece
adamc@1090 1051 val old = SS.member (!defed, this)
adamc@1146 1052
adamc@1146 1053 fun notThere (ch, s) =
adamc@1146 1054 Substring.isEmpty (#2 (Substring.splitl
adamc@1146 1055 (fn ch' => ch' <> ch) s))
adamc@1146 1056
adamc@1146 1057 fun simOpen () =
adamc@1146 1058 SS.foldl (fn (full, ds) =>
adamc@1146 1059 if String.isPrefix (this ^ ".") full
adamc@1146 1060 andalso notThere (#".",
adamc@1146 1061 Substring.extract (full,
adamc@1146 1062 size
adamc@1146 1063 this + 1,
adamc@1146 1064 NONE)) then
adamc@1146 1065 let
adamc@1146 1066 val parts = String.tokens
adamc@1146 1067 (fn ch => ch = #".") full
adamc@1146 1068
adamc@1146 1069 val part = List.last parts
adamc@1146 1070
adamc@1146 1071 val imp = if length parts >= 2 then
adamc@1146 1072 (Source.StrProj
adamc@1146 1073 ((Source.StrVar
adamc@1146 1074 (List.nth (parts,
adamc@1146 1075 length
adamc@1146 1076 parts
adamc@1146 1077 - 2)),
adamc@1146 1078 loc),
adamc@1146 1079 part), loc)
adamc@1146 1080 else
adamc@1146 1081 (Source.StrVar part, loc)
adamc@1146 1082 in
adam@1868 1083 (Source.DStr (part, NONE, NONE, imp, false),
adamc@1146 1084 loc) :: ds
adamc@1146 1085 end
adamc@1146 1086 else
adamc@1146 1087 ds) [] (!fulls)
adamc@1090 1088 in
adamc@1090 1089 defed := SS.add (!defed, this);
adam@1732 1090 (Source.DStr (piece, NONE, NONE,
adamc@1090 1091 (Source.StrConst (if old then
adamc@1146 1092 simOpen ()
adam@1868 1093 @ [makeD false this pieces]
adamc@1090 1094 else
adam@1868 1095 [makeD false this pieces]),
adam@1868 1096 loc), first andalso old),
adamc@1090 1097 loc)
adamc@1090 1098 end
adamc@1090 1099 in
adamc@1092 1100 if SS.member (!fulls, full) then
adamc@1092 1101 ErrorMsg.error ("Rooted module " ^ full ^ " has multiple versions.")
adamc@1092 1102 else
adamc@1092 1103 ();
ziv@2202 1104
adam@1868 1105 makeD true "" pieces
adamc@1146 1106 before ignore (foldl (fn (new, path) =>
adamc@1146 1107 let
adamc@1146 1108 val new' = case path of
adamc@1146 1109 "" => new
adamc@1146 1110 | _ => path ^ "." ^ new
adamc@1146 1111 in
adamc@1146 1112 fulls := SS.add (!fulls, new');
adamc@1146 1113 new'
adamc@1146 1114 end) "" pieces)
adamc@1090 1115 end
adamc@201 1116 in
adamc@834 1117 checkErrors ();
adamc@1090 1118 d
adam@1773 1119 end handle MissingFile fname => (ErrorMsg.error ("Missing source file: " ^ fname);
adam@1773 1120 (Source.DSequence "", ErrorMsg.dummySpan))
adamc@56 1121
adamc@764 1122 val dsFfi = map parseFfi ffi
adamc@201 1123 val ds = map parseOne fnames
adamc@1090 1124 val loc = ErrorMsg.dummySpan
adamc@201 1125 in
adamc@834 1126 if !anyErrors then
adamc@834 1127 ErrorMsg.error "Parse failure"
adamc@834 1128 else
adamc@834 1129 ();
adamc@834 1130
adamc@201 1131 let
adamc@1126 1132 val final = List.last fnames
adamc@1126 1133 val final = case List.find (fn (root, name) =>
adamc@1126 1134 String.isPrefix (root ^ "/") final) mrs of
adamc@1126 1135 NONE => (Source.StrVar (nameOf final), loc)
adamc@1126 1136 | SOME (root, name) =>
adamc@1126 1137 let
adamc@1126 1138 val m = (Source.StrVar name, loc)
adamc@1126 1139 val final = String.extract (final, size root + 1, NONE)
adamc@1264 1140 val fields = String.fields (fn ch => ch = #"/") final
adamc@1264 1141 val fields = List.filter (fn s => size s = 0
adamc@1264 1142 orelse not (Char.isDigit (String.sub (s, 0))))
adamc@1264 1143 fields
adamc@1126 1144 in
adamc@1126 1145 foldl (fn (x, m) => (Source.StrProj (m, capitalize x), loc))
adamc@1264 1146 m fields
adamc@1126 1147 end
adamc@271 1148
adamc@764 1149 val ds = dsFfi @ ds
adamc@1126 1150 @ [(Source.DExport final, loc)]
adamc@1090 1151
adamc@1090 1152 val ds = case database of
adamc@1090 1153 NONE => ds
adamc@1090 1154 | SOME s => (Source.DDatabase s, loc) :: ds
adam@1294 1155
adam@1294 1156 val ds = case onError of
adam@1294 1157 NONE => ds
adam@1294 1158 | SOME v => ds @ [(Source.DOnError v, loc)]
adam@1868 1159
adam@1868 1160 fun dummy fname = {file = Settings.libFile fname,
adam@1868 1161 first = ErrorMsg.dummyPos,
adam@1868 1162 last = ErrorMsg.dummyPos}
adam@1868 1163
adam@1868 1164 val used = SM.insert (SM.empty, "Basis", dummy "basis.urs")
adam@1868 1165 val used = SM.insert (used, "Top", dummy "top.urs")
adamc@201 1166 in
adam@1738 1167 ignore (List.foldl (fn (d, used) =>
adam@1738 1168 case #1 d of
adam@1868 1169 Source.DStr (x, _, _, _, false) =>
adam@1743 1170 (case SM.find (used, x) of
adam@1743 1171 SOME loc =>
adam@1743 1172 (ErrorMsg.error ("Duplicate top-level module name " ^ x);
adam@1743 1173 Print.prefaces "Files" [("Previous", Print.PD.string (ErrorMsg.spanToString loc)),
adam@1743 1174 ("Current", Print.PD.string (ErrorMsg.spanToString (#2 d)))];
adam@1743 1175 used)
adam@1743 1176 | NONE =>
adam@1743 1177 SM.insert (used, x, #2 d))
adam@1868 1178 | _ => used) used ds);
adamc@1090 1179 ds
adamc@201 1180 end handle Empty => ds
adamc@201 1181 end,
adamc@201 1182 print = SourcePrint.p_file
adamc@201 1183 }
adamc@56 1184
adamc@270 1185 val toParse = transform parse "parse" o toParseJob
adamc@38 1186
adamc@201 1187 val elaborate = {
adamc@201 1188 func = fn file => let
ezyang@1739 1189 val basisF = Settings.libFile "basis.urs"
ezyang@1739 1190 val topF = Settings.libFile "top.urs"
ezyang@1739 1191 val topF' = Settings.libFile "top.ur"
adam@1732 1192
adam@1732 1193 val basis = #func parseUrs basisF
adam@1732 1194 val topSgn = #func parseUrs topF
adam@1732 1195 val topStr = #func parseUr topF'
adam@1732 1196
adam@1732 1197 val tm1 = OS.FileSys.modTime topF
adam@1732 1198 val tm2 = OS.FileSys.modTime topF'
adamc@201 1199 in
adam@1732 1200 Elaborate.elabFile basis (OS.FileSys.modTime basisF)
adam@1732 1201 topStr topSgn (if Time.< (tm1, tm2) then tm2 else tm1)
adam@1732 1202 ElabEnv.empty file
adamc@201 1203 end,
adamc@201 1204 print = ElabPrint.p_file ElabEnv.empty
adamc@201 1205 }
adamc@5 1206
adamc@270 1207 val toElaborate = transform elaborate "elaborate" o toParse
adamc@201 1208
adamc@448 1209 val unnest = {
adamc@448 1210 func = Unnest.unnest,
adamc@448 1211 print = ElabPrint.p_file ElabEnv.empty
adamc@448 1212 }
adamc@448 1213
adamc@448 1214 val toUnnest = transform unnest "unnest" o toElaborate
adamc@448 1215
adamc@313 1216 val termination = {
adamc@313 1217 func = (fn file => (Termination.check file; file)),
adamc@313 1218 print = ElabPrint.p_file ElabEnv.empty
adamc@313 1219 }
adamc@313 1220
adamc@448 1221 val toTermination = transform termination "termination" o toUnnest
adamc@313 1222
adamc@201 1223 val explify = {
adamc@201 1224 func = Explify.explify,
adamc@201 1225 print = ExplPrint.p_file ExplEnv.empty
adamc@201 1226 }
adamc@201 1227
adamc@625 1228 val toExplify = transform explify "explify" o toUnnest
adamc@201 1229
adamc@201 1230 val corify = {
adamc@201 1231 func = Corify.corify,
adamc@201 1232 print = CorePrint.p_file CoreEnv.empty
adamc@201 1233 }
adamc@201 1234
adamc@270 1235 val toCorify = transform corify "corify" o toExplify
adamc@201 1236
adamc@482 1237 (*val reduce_local = {
adamc@482 1238 func = ReduceLocal.reduce,
adamc@482 1239 print = CorePrint.p_file CoreEnv.empty
adamc@482 1240 }
adamc@482 1241
adamc@482 1242 val toReduce_local = transform reduce_local "reduce_local" o toCorify*)
adamc@482 1243
adamc@443 1244 val especialize = {
adamc@443 1245 func = ESpecialize.specialize,
adamc@443 1246 print = CorePrint.p_file CoreEnv.empty
adamc@443 1247 }
adamc@443 1248
adamc@454 1249 val core_untangle = {
adamc@454 1250 func = CoreUntangle.untangle,
adamc@454 1251 print = CorePrint.p_file CoreEnv.empty
adamc@454 1252 }
adamc@454 1253
adamc@794 1254 val toCore_untangle = transform core_untangle "core_untangle" o toCorify
adamc@454 1255
adamc@202 1256 val shake = {
adamc@202 1257 func = Shake.shake,
adamc@202 1258 print = CorePrint.p_file CoreEnv.empty
adamc@202 1259 }
adamc@39 1260
adamc@454 1261 val toShake1 = transform shake "shake1" o toCore_untangle
adamc@110 1262
adam@1362 1263 val toEspecialize1' = transform especialize "especialize1'" o toShake1
adam@1362 1264 val toShake1' = transform shake "shake1'" o toEspecialize1'
adam@1362 1265
adamc@607 1266 val rpcify = {
adamc@607 1267 func = Rpcify.frob,
adamc@607 1268 print = CorePrint.p_file CoreEnv.empty
adamc@607 1269 }
adamc@607 1270
adam@1362 1271 val toRpcify = transform rpcify "rpcify" o toShake1'
adamc@607 1272
adamc@642 1273 val toCore_untangle2 = transform core_untangle "core_untangle2" o toRpcify
adamc@642 1274 val toShake2 = transform shake "shake2" o toCore_untangle2
adamc@1181 1275
adamc@1186 1276 val toEspecialize1 = transform especialize "especialize1" o toShake2
adamc@1181 1277
adamc@1062 1278 val toCore_untangle3 = transform core_untangle "core_untangle3" o toEspecialize1
adamc@1062 1279 val toShake3 = transform shake "shake3" o toCore_untangle3
adamc@642 1280
adamc@202 1281 val tag = {
adamc@202 1282 func = Tag.tag,
adamc@202 1283 print = CorePrint.p_file CoreEnv.empty
adamc@202 1284 }
adamc@193 1285
adamc@1062 1286 val toTag = transform tag "tag" o toShake3
adamc@20 1287
adamc@202 1288 val reduce = {
adamc@202 1289 func = Reduce.reduce,
adamc@202 1290 print = CorePrint.p_file CoreEnv.empty
adamc@202 1291 }
adamc@25 1292
adamc@692 1293 val toReduce = transform reduce "reduce" o toTag
adamc@23 1294
adamc@1186 1295 val toShakey = transform shake "shakey" o toReduce
adamc@1186 1296
adamc@1186 1297 val unpoly = {
adamc@1186 1298 func = Unpoly.unpoly,
adamc@1186 1299 print = CorePrint.p_file CoreEnv.empty
adamc@1186 1300 }
adamc@1186 1301
adamc@1186 1302 val toUnpoly = transform unpoly "unpoly" o toShakey
adamc@315 1303
adamc@202 1304 val specialize = {
adamc@202 1305 func = Specialize.specialize,
adamc@202 1306 print = CorePrint.p_file CoreEnv.empty
adamc@202 1307 }
adamc@132 1308
adamc@1186 1309 val toSpecialize = transform specialize "specialize" o toUnpoly
adamc@131 1310
adamc@1062 1311 val toShake4 = transform shake "shake4" o toSpecialize
adamc@133 1312
adamc@1062 1313 val toEspecialize2 = transform especialize "especialize2" o toShake4
adamc@1272 1314 val toShake4' = transform shake "shake4'" o toEspecialize2
adamc@1272 1315 val toUnpoly2 = transform unpoly "unpoly2" o toShake4'
adamc@1276 1316 val toSpecialize2 = transform specialize "specialize2" o toUnpoly2
adamc@1276 1317 val toShake4'' = transform shake "shake4'" o toSpecialize2
adamc@1272 1318 val toEspecialize3 = transform especialize "especialize3" o toShake4''
adamc@794 1319
adamc@1272 1320 val toReduce2 = transform reduce "reduce2" o toEspecialize3
adamc@898 1321
adamc@1062 1322 val toShake5 = transform shake "shake5" o toReduce2
adamc@794 1323
adamc@692 1324 val marshalcheck = {
adamc@692 1325 func = (fn file => (MarshalCheck.check file; file)),
adamc@692 1326 print = CorePrint.p_file CoreEnv.empty
adamc@692 1327 }
adamc@692 1328
adamc@1062 1329 val toMarshalcheck = transform marshalcheck "marshalcheck" o toShake5
adamc@692 1330
adamc@732 1331 val effectize = {
adamc@732 1332 func = Effective.effectize,
adamc@732 1333 print = CorePrint.p_file CoreEnv.empty
adamc@732 1334 }
adamc@732 1335
adamc@732 1336 val toEffectize = transform effectize "effectize" o toMarshalcheck
adamc@732 1337
adamc@1170 1338 val css = {
adamc@1170 1339 func = Css.summarize,
adamc@1170 1340 print = fn _ => Print.box []
adamc@1170 1341 }
adamc@1170 1342
adamc@1170 1343 val toCss = transform css "css" o toShake5
adamc@1170 1344
adamc@202 1345 val monoize = {
adamc@202 1346 func = Monoize.monoize CoreEnv.empty,
adamc@202 1347 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1348 }
adamc@134 1349
adamc@732 1350 val toMonoize = transform monoize "monoize" o toEffectize
adamc@96 1351
adamc@202 1352 val mono_opt = {
adamc@910 1353 func = MonoOpt.optimize,
adamc@202 1354 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1355 }
adamc@29 1356
adamc@270 1357 val toMono_opt1 = transform mono_opt "mono_opt1" o toMonoize
adamc@5 1358
adamc@202 1359 val untangle = {
adamc@202 1360 func = Untangle.untangle,
adamc@202 1361 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1362 }
adamc@1 1363
adamc@270 1364 val toUntangle = transform untangle "untangle" o toMono_opt1
adamc@38 1365
adamc@202 1366 val mono_reduce = {
adamc@202 1367 func = MonoReduce.reduce,
adamc@202 1368 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1369 }
adamc@16 1370
adamc@270 1371 val toMono_reduce = transform mono_reduce "mono_reduce" o toUntangle
adamc@39 1372
adamc@202 1373 val mono_shake = {
adamc@202 1374 func = MonoShake.shake,
adamc@202 1375 print = MonoPrint.p_file MonoEnv.empty
adamc@202 1376 }
adamc@110 1377
adamc@270 1378 val toMono_shake = transform mono_shake "mono_shake1" o toMono_reduce
adamc@193 1379
adamc@572 1380 val toMono_opt2 = transform mono_opt "mono_opt2" o toMono_shake
adamc@572 1381
adamc@1200 1382 val iflow = {
adamc@1235 1383 func = (fn file => (if !doIflow then Iflow.check file else (); file)),
adamc@1200 1384 print = MonoPrint.p_file MonoEnv.empty
adamc@1200 1385 }
adamc@1200 1386
adamc@1200 1387 val toIflow = transform iflow "iflow" o toMono_opt2
adamc@1200 1388
adam@1800 1389 val namejs = {
adam@1800 1390 func = NameJS.rewrite,
adam@1800 1391 print = MonoPrint.p_file MonoEnv.empty
adam@1800 1392 }
adam@1800 1393
adam@1800 1394 val toNamejs = transform namejs "namejs" o toIflow
adam@1800 1395
adam@1800 1396 val toNamejs_untangle = transform untangle "namejs_untangle" o toNamejs
adam@1800 1397
adam@1845 1398 val scriptcheck = {
adam@1845 1399 func = ScriptCheck.classify,
adam@1845 1400 print = MonoPrint.p_file MonoEnv.empty
adam@1845 1401 }
adam@1845 1402
adam@1845 1403 val toScriptcheck = transform scriptcheck "scriptcheck" o toNamejs_untangle
adam@1845 1404
adam@2056 1405 val dbmodecheck = {
adam@2056 1406 func = DbModeCheck.classify,
adam@2056 1407 print = MonoPrint.p_file MonoEnv.empty
adam@2056 1408 }
adam@2056 1409
adam@2056 1410 val toDbmodecheck = transform dbmodecheck "dbmodecheck" o toScriptcheck
adam@2056 1411
adamc@567 1412 val jscomp = {
adamc@567 1413 func = JsComp.process,
adamc@567 1414 print = MonoPrint.p_file MonoEnv.empty
adamc@567 1415 }
adamc@567 1416
adam@2056 1417 val toJscomp = transform jscomp "jscomp" o toDbmodecheck
adamc@567 1418
adamc@910 1419 val toMono_opt3 = transform mono_opt "mono_opt3" o toJscomp
adamc@20 1420
adamc@506 1421 val fuse = {
adamc@506 1422 func = Fuse.fuse,
adamc@506 1423 print = MonoPrint.p_file MonoEnv.empty
adamc@506 1424 }
adamc@506 1425
adamc@572 1426 val toFuse = transform fuse "fuse" o toMono_opt3
adamc@506 1427
adamc@506 1428 val toUntangle2 = transform untangle "untangle2" o toFuse
adamc@506 1429
adamc@601 1430 val toMono_reduce2 = transform mono_reduce "mono_reduce2" o toUntangle2
adamc@601 1431 val toMono_shake2 = transform mono_shake "mono_shake2" o toMono_reduce2
adamc@916 1432 val toMono_opt4 = transform mono_opt "mono_opt4" o toMono_shake2
adamc@1017 1433 val toMono_reduce3 = transform mono_reduce "mono_reduce3" o toMono_opt4
adamc@1185 1434 val toFuse2 = transform fuse "fuse2" o toMono_reduce3
adamc@916 1435 val toUntangle3 = transform untangle "untangle3" o toFuse2
adamc@916 1436 val toMono_shake3 = transform mono_shake "mono_shake3" o toUntangle3
adamc@506 1437
adamc@377 1438 val pathcheck = {
adamc@377 1439 func = (fn file => (PathCheck.check file; file)),
adamc@377 1440 print = MonoPrint.p_file MonoEnv.empty
adamc@377 1441 }
adamc@377 1442
adamc@916 1443 val toPathcheck = transform pathcheck "pathcheck" o toMono_shake3
adamc@377 1444
adam@1595 1445 val sidecheck = {
adam@1595 1446 func = SideCheck.check,
adam@1595 1447 print = MonoPrint.p_file MonoEnv.empty
adam@1595 1448 }
adam@1595 1449
adam@1595 1450 val toSidecheck = transform sidecheck "sidecheck" o toPathcheck
adam@1595 1451
adam@1856 1452 val sigcheck = {
adam@1856 1453 func = SigCheck.check,
adam@1856 1454 print = MonoPrint.p_file MonoEnv.empty
adam@1856 1455 }
adam@1856 1456
adam@1856 1457 val toSigcheck = transform sigcheck "sigcheck" o toSidecheck
adam@1856 1458
ziv@2209 1459 val sqlcache = {
ziv@2209 1460 func = (fn file => (if !doSqlcache then Sqlcache.go file else file)),
ziv@2202 1461 print = MonoPrint.p_file MonoEnv.empty
ziv@2202 1462 }
ziv@2202 1463
ziv@2209 1464 val toSqlcache = transform sqlcache "sqlcache" o toSigcheck
ziv@2202 1465
adamc@202 1466 val cjrize = {
adamc@202 1467 func = Cjrize.cjrize,
adamc@202 1468 print = CjrPrint.p_file CjrEnv.empty
adamc@202 1469 }
adamc@23 1470
ziv@2209 1471 val toCjrize = transform cjrize "cjrize" o toSqlcache
adamc@29 1472
adamc@282 1473 val prepare = {
adamc@282 1474 func = Prepare.prepare,
adamc@282 1475 print = CjrPrint.p_file CjrEnv.empty
adamc@282 1476 }
adamc@282 1477
adam@1845 1478 val toPrepare = transform prepare "prepare" o toCjrize
adamc@282 1479
adamc@879 1480 val checknest = {
adamc@879 1481 func = fn f => if #supportsNestedPrepared (Settings.currentDbms ()) then f else Checknest.annotate f,
adamc@879 1482 print = CjrPrint.p_file CjrEnv.empty
adamc@879 1483 }
adamc@879 1484
adamc@879 1485 val toChecknest = transform checknest "checknest" o toPrepare
adamc@879 1486
adamc@274 1487 val sqlify = {
adamc@274 1488 func = Cjrize.cjrize,
adamc@274 1489 print = CjrPrint.p_sql CjrEnv.empty
adamc@274 1490 }
adamc@274 1491
adamc@274 1492 val toSqlify = transform sqlify "sqlify" o toMono_opt2
adamc@274 1493
adam@2067 1494 fun escapeFilename s =
adam@2067 1495 "\""
adam@2067 1496 ^ String.translate (fn #"\"" => "\\\"" | #"\\" => "\\\\" | ch => str ch) s
adam@2067 1497 ^ "\""
adam@1473 1498
adam@1540 1499 val beforeC = ref (fn () => ())
adam@1540 1500
adam@1822 1501 structure StringSet = BinarySetFn(struct
adam@1822 1502 type ord_key = string
adam@1822 1503 val compare = String.compare
adam@1822 1504 end)
adam@1822 1505
adam@1725 1506 fun compileC {cname, oname, ename, libs, profile, debug, linker, link = link'} =
adamc@183 1507 let
adamc@855 1508 val proto = Settings.currentProtocol ()
adamc@1095 1509
adam@1368 1510 val lib = if Settings.getStaticLinking () then
adam@1846 1511 " -static " ^ !Settings.configLib ^ "/" ^ #linkStatic proto ^ " " ^ !Settings.configLib ^ "/liburweb.a"
adam@1368 1512 else
ezyang@1739 1513 "-L" ^ !Settings.configLib ^ " " ^ #linkDynamic proto ^ " -lurweb"
adamc@378 1514
adam@1558 1515 val opt = if debug then
adam@1558 1516 ""
adam@1558 1517 else
adam@1558 1518 " -O3"
adam@1558 1519
grrwlf@1871 1520 val compile = (Settings.getCCompiler ()) ^ " " ^ Config.ccArgs ^ " " ^ Config.pthreadCflags ^ " -Wimplicit -Werror -Wno-unused-value"
ezyang@1739 1521 ^ opt ^ " -I " ^ !Settings.configInclude
adamc@1096 1522 ^ " " ^ #compile proto
adam@1473 1523 ^ " -c " ^ escapeFilename cname ^ " -o " ^ escapeFilename oname
adamc@1096 1524
grrwlf@1871 1525 val linker = Option.getOpt (linker, (Settings.getCCompiler ()) ^ " -Werror" ^ opt ^ " " ^ Config.ccArgs ^ " " ^ Config.pthreadCflags ^ " " ^ Config.pthreadLibs)
adam@1725 1526
adam@1846 1527 val ssl = if Settings.getStaticLinking () then
adam@1846 1528 Config.openssl ^ " -ldl -lz"
adam@1846 1529 else
adam@1846 1530 Config.openssl
adam@1846 1531
adam@1725 1532 val link = linker
andersk@1893 1533 ^ " " ^ escapeFilename oname ^ " " ^ lib ^ " -lm " ^ ssl ^ " " ^ libs ^ " -o " ^ escapeFilename ename
adamc@502 1534
adamc@502 1535 val (compile, link) =
adamc@502 1536 if profile then
adamc@502 1537 (compile ^ " -pg", link ^ " -pg")
adamc@502 1538 else
adamc@502 1539 (compile, link)
adamc@742 1540
adamc@742 1541 val (compile, link) =
adamc@742 1542 if debug then
adamc@742 1543 (compile ^ " -g", link ^ " -g")
adamc@742 1544 else
adamc@742 1545 (compile, link)
adamc@764 1546
adam@1822 1547 val link = #1 (foldl
adam@1822 1548 (fn (s, (link, set)) =>
adam@1822 1549 if StringSet.member (set, s) then
adam@1822 1550 (link, set)
adam@1822 1551 else
adam@1822 1552 ((link ^ " " ^ s), StringSet.add (set, s)))
adam@1822 1553 (link, StringSet.empty)
adam@1822 1554 link')
adam@1394 1555
adam@1394 1556 fun system s =
adam@1394 1557 (if debug then
adam@1394 1558 print (s ^ "\n")
adam@1394 1559 else
adam@1394 1560 ();
adam@1394 1561 OS.Process.isSuccess (OS.Process.system s))
adamc@183 1562 in
adam@1540 1563 !beforeC ();
adam@1394 1564 system compile andalso system link
adamc@183 1565 end
adamc@183 1566
adamc@202 1567 fun compile job =
adamc@879 1568 case run toChecknest job of
adamc@1045 1569 NONE => false
adamc@29 1570 | SOME file =>
adamc@202 1571 let
adamc@274 1572 val job = valOf (run (transform parseUrp "parseUrp") job)
adamc@102 1573
adamc@274 1574 val (cname, oname, cleanup) =
adamc@274 1575 if #debug job then
adamc@457 1576 ("/tmp/webapp.c", "/tmp/webapp.o", fn () => ())
adamc@274 1577 else
adamc@274 1578 let
adamc@274 1579 val dir = OS.FileSys.tmpName ()
adamc@403 1580 val () = if OS.FileSys.access (dir, []) then
adamc@403 1581 OS.FileSys.remove dir
adamc@403 1582 else
adamc@403 1583 ()
adamc@457 1584 val cname = OS.Path.joinDirFile {dir = dir, file = "webapp.c"}
adamc@457 1585 val oname = OS.Path.joinDirFile {dir = dir, file = "webapp.o"}
adamc@274 1586 in
adamc@274 1587 OS.FileSys.mkDir dir;
adamc@274 1588 (cname, oname,
adam@1700 1589 fn () => (if OS.Process.isSuccess (OS.Process.system ("rm -rf " ^ dir)) then
adam@1700 1590 ()
adam@1700 1591 else
adam@1700 1592 raise Fail ("Unable to delete temporary directory " ^ dir)))
adamc@274 1593 end
adamc@274 1594 val ename = #exe job
adamc@202 1595 in
adamc@274 1596 let
adamc@274 1597 val outf = TextIO.openOut cname
adamc@274 1598 val s = TextIOPP.openOut {dst = outf, wid = 80}
adamc@432 1599
adamc@432 1600 val hasDb = List.exists (fn (Cjr.DDatabase _, _) => true | _ => false) (#1 file)
adamc@432 1601 val libs =
adamc@432 1602 if hasDb then
adamc@866 1603 #link (Settings.currentDbms ())
adamc@432 1604 else
adamc@432 1605 ""
adamc@274 1606 in
adamc@274 1607 Print.fprint s (CjrPrint.p_file CjrEnv.empty file);
adamc@415 1608 TextIO.output1 (outf, #"\n");
adamc@274 1609 TextIO.closeOut outf;
adamc@102 1610
adam@1381 1611 if ErrorMsg.anyErrors () then
adam@1381 1612 false
adam@1381 1613 else
adam@1381 1614 (case #sql job of
adam@1381 1615 NONE => ()
adam@1381 1616 | SOME sql =>
adam@1381 1617 let
adam@1381 1618 val outf = TextIO.openOut sql
adam@1381 1619 val s = TextIOPP.openOut {dst = outf, wid = 80}
adam@1381 1620 in
adam@1381 1621 Print.fprint s (CjrPrint.p_sql CjrEnv.empty file);
adam@1381 1622 TextIO.closeOut outf
adam@1381 1623 end;
adamc@274 1624
adam@1381 1625 compileC {cname = cname, oname = oname, ename = ename, libs = libs,
adam@1725 1626 profile = #profile job, debug = #debug job, linker = #linker job, link = #link job}
ziv@2202 1627
adam@1381 1628 before cleanup ())
adamc@274 1629 end
adamc@274 1630 handle ex => (((cleanup ()) handle _ => ()); raise ex)
adamc@202 1631 end
adamc@29 1632
adamc@1045 1633 fun compiler job =
adamc@1045 1634 if compile job then
adamc@1045 1635 ()
adamc@1045 1636 else
adamc@1045 1637 OS.Process.exit OS.Process.failure
adamc@1045 1638
adamc@1266 1639 fun moduleOf fname =
adamc@1266 1640 let
adamc@1266 1641 val mrs = !moduleRoots
adamc@1266 1642 val fname = OS.Path.mkCanonical fname
adamc@1266 1643 in
adamc@1266 1644 case List.find (fn (root, _) => String.isPrefix (root ^ "/") fname) mrs of
adamc@1266 1645 NONE => capitalize (OS.Path.base (OS.Path.file fname))
adamc@1266 1646 | SOME (root, name) =>
adamc@1266 1647 let
adamc@1266 1648 val fname = OS.Path.base fname
adamc@1266 1649 val fname = String.extract (fname, size root + 1, NONE)
adamc@1266 1650 val fs = String.fields (fn ch => ch = #"/") fname
adamc@1266 1651 val fs = List.filter (CharVector.exists (fn ch => not (Char.isDigit ch))) fs
adamc@1266 1652 val fs = map capitalize fs
adamc@1266 1653 in
adamc@1266 1654 String.concatWith "." (name :: fs)
adamc@1266 1655 end
adamc@1266 1656 end
adamc@1266 1657
adamc@1 1658 end