# HG changeset patch
# User Adam Chlipala <adamc@hcoop.net>
# Date 1220816108 14400
# Node ID 0aee86b8a6d6c823f29b397bbbbfa0104552ded0
# Parent  ea62b15da922def6d136da95c395ccf2387ce640
Automatically add table annotations in UPDATE and DELETE

diff -r ea62b15da922 -r 0aee86b8a6d6 src/urweb.grm
--- a/src/urweb.grm	Sun Sep 07 15:16:10 2008 -0400
+++ b/src/urweb.grm	Sun Sep 07 15:35:08 2008 -0400
@@ -150,6 +150,8 @@
         (EApp (e, e2), loc)
     end
 
+val inDml = ref false
+
 %%
 %header (functor UrwebLrValsFn(structure Token : TOKEN))
 
@@ -284,6 +286,8 @@
  | fields of con list
  | sqlexps of exp list
  | fsets of (con * exp) list
+ | enterDml of unit
+ | leaveDml of unit
 
 
 %verbose                                (* print summary of errors *)
@@ -748,7 +752,7 @@
                                                  ();
                                              (EApp (e, (ERecord (ListPair.zip (fields, sqlexps)), loc)), loc)
                                          end)
-       | LPAREN UPDATE texp SET fsets CWHERE sqlexp RPAREN
+       | LPAREN enterDml UPDATE texp SET fsets CWHERE sqlexp leaveDml RPAREN
                                         (let
                                              val loc = s (LPARENleft, RPARENright)
 
@@ -758,7 +762,7 @@
                                          in
                                              (EApp (e, sqlexp), loc)
                                          end)
-       | LPAREN DELETE FROM texp CWHERE sqlexp RPAREN
+       | LPAREN enterDml DELETE FROM texp CWHERE sqlexp leaveDml RPAREN
                                         (let
                                              val loc = s (LPARENleft, RPARENright)
 
@@ -770,6 +774,9 @@
 
        | UNDER                          (EWild, s (UNDERleft, UNDERright))
 
+enterDml :                              (inDml := true)
+leaveDml :                              (inDml := false)
+
 texp   : SYMBOL                         (EVar ([], SYMBOL), s (SYMBOLleft, SYMBOLright))
        | LBRACE LBRACE eexp RBRACE RBRACE (eexp)
 
@@ -1026,10 +1033,21 @@
                                          end)
        | CSYMBOL                         (let
                                              val loc = s (CSYMBOLleft, CSYMBOLright)
-                                             val e = (EVar (["Basis"], "sql_exp"), loc)
-                                         in
-                                             (ECApp (e, (CName CSYMBOL, loc)), loc)
-                                         end)
+                                          in
+                                              if !inDml then
+                                                  let
+                                                      val e = (EVar (["Basis"], "sql_field"), loc)
+                                                      val e = (ECApp (e, (CName "T", loc)), loc)
+                                                  in
+                                                      (ECApp (e, (CName CSYMBOL, loc)), loc)
+                                                  end
+                                              else
+                                                  let
+                                                      val e = (EVar (["Basis"], "sql_exp"), loc)
+                                                  in
+                                                      (ECApp (e, (CName CSYMBOL, loc)), loc)
+                                                  end
+                                          end)
 
        | sqlexp EQ sqlexp               (sql_compare ("eq", sqlexp1, sqlexp2, s (sqlexp1left, sqlexp2right)))
        | sqlexp NE sqlexp               (sql_compare ("ne", sqlexp1, sqlexp2, s (sqlexp1left, sqlexp2right)))
diff -r ea62b15da922 -r 0aee86b8a6d6 tests/delete.ur
--- a/tests/delete.ur	Sun Sep 07 15:16:10 2008 -0400
+++ b/tests/delete.ur	Sun Sep 07 15:35:08 2008 -0400
@@ -1,5 +1,5 @@
 table t1 : {A : int, B : string, C : float, D : bool}
 
 fun main () : transaction page =
-        () <- dml (DELETE FROM t1 WHERE T.A = 5);
+        () <- dml (DELETE FROM t1 WHERE A = 5);
         return <html><body>Deleted.</body></html>
diff -r ea62b15da922 -r 0aee86b8a6d6 tests/update.ur
--- a/tests/update.ur	Sun Sep 07 15:16:10 2008 -0400
+++ b/tests/update.ur	Sun Sep 07 15:35:08 2008 -0400
@@ -1,5 +1,5 @@
 table t1 : {A : int, B : string, C : float, D : bool}
 
 fun main () : transaction page =
-        () <- dml (UPDATE t1 SET B = 'Hi', C = 12.34 WHERE T.A = 5);
+        () <- dml (UPDATE t1 SET B = 'Hi', C = 12.34 WHERE A = 5);
         return <html><body>Updated.</body></html>