Mercurial > urweb

annotate src/compiler.sml @ 1343:0a2b630f2463

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