support < > in for loop

Author: bobzhang

jscomp/core/j.ml

@@ -94,7 +94,7 @@ and exception_ident = ident
 
 and for_ident = ident 
 
-and for_direction = Asttypes.direction_flag
+and for_direction = Js_op.direction_flag
 
 and property_map = 
     (property_name * expression) list

jscomp/core/js_dump.ml

@@ -265,7 +265,9 @@ let pp_var_declare cxt f id =
 
 let pp_direction f (direction : J.for_direction) =   
   match direction with
+  | Up
   | Upto -> P.string f L.plus_plus
+  | Down
   | Downto -> P.string f L.minus_minus 
 
 let return_sp f = 
@@ -1171,10 +1173,18 @@ and statement_desc top cxt f (s : J.statement_desc) : cxt =
                       let (_,_,right) = Js_op_util.op_prec Le  in
                       P.string f L.le;
                       right
+                    | Up ->  
+                      let (_,_,right) = Js_op_util.op_prec Lt  in
+                      P.string f L.lt;
+                      right
                     | Downto ->
                       let (_,_,right) = Js_op_util.op_prec Ge in
                       P.string f L.ge ;
                       right
+                    | Down ->
+                      let (_,_,right) = Js_op_util.op_prec Gt in
+                      P.string f L.gt ;
+                      right                      
                   in
                   P.space f ;
                   let cxt  =

jscomp/core/js_dump_lit.ml

@@ -81,7 +81,9 @@ let string_cap = "String"
 let fromCharcode = "fromCharCode"
 let eq = "="
 let le = "<="
+let lt = "<"
 let ge = ">="
+let gt = ">"
 let plus_plus = "++" 
 (*  FIXME: use (i = i + 1 | 0) instead  *)
 let minus_minus = "--"

jscomp/core/js_op.ml

@@ -184,7 +184,11 @@ type mutable_flag =
   | Mutable
   | Immutable
   | NA
-
+type direction_flag = 
+  | Upto
+  | Downto
+  | Up
+  | Down
 (* 
     {[
     let rec x = 1 :: y 

jscomp/core/lam_compile.ml

@@ -954,7 +954,13 @@ and compile_while (predicate : Lam.t) (body : Lam.t) (lambda_cxt : Lam_compile_c
 *)
 
 and compile_for 
-    id start finish direction body (lambda_cxt : Lam_compile_context.t) = 
+    (id : J.for_ident) 
+    (start : Lam.t) 
+    (finish : Lam.t) 
+    (direction : Js_op.direction_flag) 
+    (body : Lam.t)
+    (lambda_cxt : Lam_compile_context.t) 
+    = 
     let new_cxt = {lambda_cxt with continuation = NeedValue Not_tail} in 
     let block =
       match compile_lambda new_cxt start,
@@ -1599,8 +1605,13 @@ and compile_lambda
       compile_staticcatch cur_lam lambda_cxt      
     | Lwhile(p,body) ->
       compile_while p body lambda_cxt
-    | Lfor (id,start,finish,direction,body) ->
-      compile_for id start finish direction body lambda_cxt
+    | Lfor (id,start,finish,direction,body) ->  
+      begin match direction with 
+        | Upto ->                 
+          compile_for id start finish Upto body lambda_cxt
+        | Downto -> 
+          compile_for id start finish Downto body lambda_cxt
+      end
     | Lassign(id,lambda) ->
       compile_assign id lambda lambda_cxt
     | Ltrywith(lam,id, catch) ->  (* generate documentation *)