Lua writer: Convert not (x == y) to x ~= y

minoki · Sep 9, 2024 · 720b4a1 · 720b4a1
1 parent 3bc588a
commit 720b4a1
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/src/lua-syntax.sml b/src/lua-syntax.sml
@@ -831,6 +831,32 @@ struct
                       @ Fragment (" " ^ luaop ^ " ") :: paren prec exp2
                   }
             end
+        | doExp
+            (LuaSyntax.UnaryExp
+               (LuaSyntax.NOT, LuaSyntax.BinExp (LuaSyntax.EQUAL, exp1, exp2))) =
+            let
+              val exp1 = doExp exp1
+              val exp2 = doExp exp2
+              val prec = 10
+            in
+              { prec = prec
+              , exp = paren prec exp1 @ Fragment " ~= " :: paren (prec - 1) exp2
+              }
+            end
+        | doExp
+            (LuaSyntax.UnaryExp
+               ( LuaSyntax.NOT
+               , LuaSyntax.BinExp (LuaSyntax.NOTEQUAL, exp1, exp2)
+               )) =
+            let
+              val exp1 = doExp exp1
+              val exp2 = doExp exp2
+              val prec = 10
+            in
+              { prec = prec
+              , exp = paren prec exp1 @ Fragment " == " :: paren (prec - 1) exp2
+              }
+            end
         | doExp (LuaSyntax.UnaryExp (unOp, exp)) =
             let
               val unOp =