add missing operator checks for update assignment.

3 files changed, 11 insertions, 11 deletions

diff --git a/makefile b/makefile
index 913d70b..11e59f3 100644
--- a/makefile
+++ b/makefile
@@ -192,10 +192,11 @@ endif
         @$(END_TIME)
 
 .PHONY: wasm
-wasm:
+wasm: clean
         @$(MAKE) generic CC='emcc -s WASM=1' AR='emar rcu' RANLIB='emranlib' -C $(SRC_PATH)/3rdParty/lua
         @mkdir -p doc/docs/.vuepress/public/js
         @emcc $(SRC_PATH)/yue_wasm.cpp $(SRC_PATH)/yuescript/ast.cpp $(SRC_PATH)/yuescript/parser.cpp $(SRC_PATH)/yuescript/yue_compiler.cpp $(SRC_PATH)/yuescript/yue_parser.cpp $(SRC_PATH)/yuescript/yuescript.cpp $(SRC_PATH)/3rdParty/lua/liblua.a -s WASM=1 -O2 -o doc/docs/.vuepress/public/js/yuescript.js -I $(SRC_PATH) -I $(SRC_PATH)/3rdParty/lua -std=c++17 --bind -fexceptions
+        @${MAKE} clean
 
 # Debug build for gdb debugging
 .PHONY: debug
diff --git a/src/yuescript/yue_compiler.cpp b/src/yuescript/yue_compiler.cpp
index 1a389b0..6213631 100755
--- a/src/yuescript/yue_compiler.cpp
+++ b/src/yuescript/yue_compiler.cpp
@@ -2431,6 +2431,7 @@ private:
                                 chain->items.clear();
                                 chain->items.dup(tmpChain->items);
                                 auto op = _parser.toString(update->op);
+                                checkOperatorAvailable(op, update->op);
                                 if (op == "??"sv) {
                                         auto defs = getPreDefineLine(assignment);
                                         temp.push_back(defs);
@@ -5802,8 +5803,7 @@ private:
                 out.push_back(join(temp));
         }
 
-        void transformBinaryOperator(BinaryOperator_t* node, str_list& out) {
-                auto op = _parser.toString(node);
+        void checkOperatorAvailable(const std::string& op, ast_node* node) {
                 if (op == "&"sv ||
                         op == "~"sv ||
                         op == "|"sv ||
@@ -5817,6 +5817,11 @@ private:
                                 throw std::logic_error(_info.errorMessage("floor division is not available when not targeting Lua version 5.3 or higher"sv, node));
                         }
                 }
+        }
+
+        void transformBinaryOperator(BinaryOperator_t* node, str_list& out) {
+                auto op = _parser.toString(node);
+                checkOperatorAvailable(op, node);
                 out.push_back(op == "!="sv ? "~="s : op);
         }
 
diff --git a/src/yuescript/yue_parser.cpp b/src/yuescript/yue_parser.cpp
index 94ce550..31a2560 100755
--- a/src/yuescript/yue_parser.cpp
+++ b/src/yuescript/yue_parser.cpp
@@ -328,19 +328,13 @@ YueParser::YueParser() {
 
         update_op =
                 expr("..") |
-                expr("+") |
-                expr("-") |
-                expr("*") |
                 expr("//") |
-                expr("/") |
-                expr("%") |
                 expr("or") |
                 expr("and") |
-                expr("&") |
-                expr("|") |
                 expr(">>") |
                 expr("<<") |
-                expr("??");
+                expr("??") |
+                set("+-*/%&|");
 
         Update = Space >> update_op >> expr("=") >> Exp;