Changed handling of response headers to allow multiple values; added documentation...

[bpt/mlt.git] / src / lib / main.sml
diff --git a/src/lib/main.sml b/src/lib/main.sml

index b058cce..139686a 100644 (file)
--- a/src/lib/main.sml
+++ b/src/lib/main.sml
@@ -29,7 +29,9 @@ struct
             val cgiFields = Cgi.cgi_fieldnames ()
             fun mapper name = (name, Cgi.cgi_field_strings name)
         in
+           Web.addHeader ("Content-type", "text/html");
             Web.pushParams (map mapper cgiFields);
+           Templates.beforeFn ();
             case args of
                 [] => (print "Status: 500\nContent-type: text/html\n\n<html><head><title>500 error</title></head><body><h2>500 error</h2>No template was specified</body></html>\n";
                        OS.Process.failure)
@@ -38,12 +40,16 @@ struct
                      NONE => (print "Status: 404\nContent-type: text/html\n\n<html><head><title>404 error</title></head><body><h2>404 error</h2>Template not found</body></html>\n";
                               OS.Process.failure)
                    | SOME f => (f ();
+                               Templates.afterFn ();
                                 Web.output ();
                                 OS.Process.success))
-       end handle Fail msg => (print "Status: 500\nContent-type: text/plain\n\nFatal error: \n\n";
-                               print msg;
-                               OS.Process.failure)
-                | ex => (print "Status: 500\nContent-type: text/plain\n\nAn exception!\n\n";
-                         app (fn s => print (s ^ "\n")) (SMLofNJ.exnHistory ex);
+       end handle ex => (Web.setExn ex;
+                         Web.clear ();
+                         Templates.exnFn ();
+                         if Web.noOutput () then
+                             (print "Status: 500\nContent-type: text/plain\n\nAn exception!\n\n";
+                              app (fn s => print (s ^ "\n")) (SMLofNJ.exnHistory ex))
+                         else
+                             Web.output ();
                           OS.Process.failure)
  end
 \ No newline at end of file