Add support for the bitwise NOT(~) unary operator (#7418)

Author: cometkim

CHANGELOG.md

@@ -21,6 +21,7 @@
 
 - Add shift (`<<`, `>>`, `>>>`) operators for `int` and `bigint`. https://github.com/rescript-lang/rescript/pull/7183
 - Add bitwise AND (`&`) operator for `int` and `bigint`. https://github.com/rescript-lang/rescript/pull/7415
+- Add bitwise NOT (`~`) operator for `int` and `bigint`. https://github.com/rescript-lang/rescript/pull/7418
 - Significantly reduced the download size by splitting binaries into optional platform-specific dependencies (e.g, `@rescript/linux-x64`). https://github.com/rescript-lang/rescript/pull/7395
 - JSX: do not error on ref as prop anymore (which is allowed in React 19). https://github.com/rescript-lang/rescript/pull/7420
 

compiler/core/j.ml

@@ -88,6 +88,7 @@ and expression_desc =
   | Typeof of expression
   | In of expression * expression (* prop in obj *)
   | Js_not of expression (* !v *)
+  | Js_bnot of expression (* ~v *)
   (* TODO: Add some primitives so that [js inliner] can do a better job *)
   | Seq of expression * expression
   | Cond of expression * expression * expression

compiler/core/js_analyzer.ml

@@ -110,7 +110,7 @@ let rec no_side_effect_expression_desc (x : J.expression_desc) =
     no_side_effect call_expr
     && Ext_list.for_all strings no_side_effect
     && Ext_list.for_all values no_side_effect
-  | Js_not e -> no_side_effect e
+  | Js_not e | Js_bnot e -> no_side_effect e
   | In (prop, obj) -> no_side_effect prop && no_side_effect obj
   | Cond (a, b, c) -> no_side_effect a && no_side_effect b && no_side_effect c
   | Call ({expression_desc = Str {txt = "Array.isArray"}}, [e], _) ->
@@ -228,8 +228,8 @@ let rec eq_expression ({expression_desc = x0} : J.expression)
       eq_expression_list ls0 ls1 && flag0 = flag1 && eq_expression tag0 tag1
     | _ -> false)
   | Length _ | Is_null_or_undefined _ | String_append _ | Typeof _ | Js_not _
-  | In _ | Cond _ | FlatCall _ | New _ | Fun _ | Raw_js_code _ | Array _
-  | Caml_block_tag _ | Object _ | Tagged_template _ | Await _ ->
+  | Js_bnot _ | In _ | Cond _ | FlatCall _ | New _ | Fun _ | Raw_js_code _
+  | Array _ | Caml_block_tag _ | Object _ | Tagged_template _ | Await _ ->
     false
   | Spread _ -> false
 

compiler/core/js_dump.ml

@@ -166,8 +166,8 @@ let rec exp_need_paren ?(arrow = false) (e : J.expression) =
   | Length _ | Call _ | Caml_block_tag _ | Seq _ | Static_index _ | Cond _
   | Bin _ | Is_null_or_undefined _ | String_index _ | Array_index _
   | String_append _ | Var _ | Undefined _ | Null | Str _ | Array _
-  | Caml_block _ | FlatCall _ | Typeof _ | Number _ | Js_not _ | In _ | Bool _
-  | New _ ->
+  | Caml_block _ | FlatCall _ | Typeof _ | Number _ | Js_not _ | Js_bnot _
+  | In _ | Bool _ | New _ ->
     false
   | Await _ -> false
   | Spread _ -> false
@@ -677,6 +677,10 @@ and expression_desc cxt ~(level : int) f x : cxt =
     P.cond_paren_group f (level > 13) (fun _ ->
         P.string f "!";
         expression ~level:13 cxt f e)
+  | Js_bnot e ->
+    P.cond_paren_group f (level > 13) (fun _ ->
+        P.string f "~";
+        expression ~level:13 cxt f e)
   | In (prop, obj) ->
     P.cond_paren_group f (level > 12) (fun _ ->
         let cxt = expression ~level:0 cxt f prop in

compiler/core/js_exp_make.ml

@@ -1616,6 +1616,11 @@ let int32_mul ?comment (e1 : J.expression) (e2 : J.expression) : J.expression =
 let unchecked_int32_mul ?comment e1 e2 : J.expression =
   {comment; expression_desc = Bin (Mul, e1, e2)}
 
+let int_bnot ?comment (e : t) : J.expression =
+  match e.expression_desc with
+  | Number (Int {i}) -> int ?comment (Int32.lognot i)
+  | _ -> {comment; expression_desc = Js_bnot e}
+
 let int32_pow ?comment (e1 : t) (e2 : t) : J.expression =
   match (e1.expression_desc, e2.expression_desc) with
   | Number (Int {i = i1}), Number (Int {i = i2}) ->

compiler/core/js_exp_make.mli

@@ -210,6 +210,8 @@ val float_equal : ?comment:string -> t -> t -> t
 
 val int_equal : ?comment:string -> t -> t -> t
 
+val int_bnot : ?comment:string -> t -> t
+
 val string_equal : ?comment:string -> t -> t -> t
 
 val eq_null_undefined_boolean : ?comment:string -> t -> t -> t

compiler/core/js_fold.ml

@@ -103,6 +103,9 @@ class fold =
       | Js_not _x0 ->
         let _self = _self#expression _x0 in
         _self
+      | Js_bnot _x0 ->
+        let _self = _self#expression _x0 in
+        _self
       | In (_x0, _x1) ->
         let _self = _self#expression _x0 in
         let _self = _self#expression _x1 in

compiler/core/js_op.ml

@@ -38,6 +38,7 @@ type binop =
   | Le
   | Gt
   | Ge
+  | Bnot
   | Bor
   | Bxor
   | Band

compiler/core/js_op_util.ml

@@ -39,7 +39,7 @@ let op_prec (op : Js_op.binop) =
   | Bxor -> (6, 6, 6)
   | Band -> (7, 7, 7)
   | Lsl | Lsr | Asr -> (10, 10, 11)
-  | Plus | Minus -> (11, 11, 12)
+  | Bnot | Plus | Minus -> (11, 11, 12)
   | Mul | Div | Mod -> (12, 12, 13)
   | Pow -> (13, 14, 12)
 
@@ -55,6 +55,7 @@ let op_int_prec (op : Js_op.int_op) =
 
 let op_str (op : Js_op.binop) =
   match op with
+  | Bnot -> "~"
   | Bor -> "|"
   | Bxor -> "^"
   | Band -> "&"

compiler/core/js_record_fold.ml

@@ -109,6 +109,9 @@ let expression_desc : 'a. ('a, expression_desc) fn =
   | Js_not _x0 ->
     let st = _self.expression _self st _x0 in
     st
+  | Js_bnot _x0 ->
+    let st = _self.expression _self st _x0 in
+    st
   | In (_x0, _x1) ->
     let st = _self.expression _self st _x0 in
     let st = _self.expression _self st _x1 in