36 lines
1.1 KiB
OCaml
36 lines
1.1 KiB
OCaml
|
open Ast_simplified
|
||
|
|
open Trace
|
||
|
|
|
||
|
|
let peephole_expression : expression -> expression result = fun e ->
|
||
|
|
let return expression = ok { e with expression } in
|
||
|
|
match e.expression with
|
||
|
|
| E_constant ("MAP_LITERAL" , lst) -> (
|
||
|
|
let%bind elt =
|
||
|
|
trace_option (simple_error "map literal expects a single parameter") @@
|
||
|
|
List.to_singleton lst
|
||
|
|
in
|
||
|
|
let%bind lst =
|
||
|
|
trace (simple_error "map literal expects a list as parameter") @@
|
||
|
|
get_e_list elt.expression
|
||
|
|
in
|
||
|
|
let aux = fun (e : expression) ->
|
||
|
|
trace (simple_error "map literal expects a list of pairs as parameter") @@
|
||
|
|
let%bind tpl = get_e_tuple e.expression in
|
||
|
|
let%bind (a , b) =
|
||
|
|
trace_option (simple_error "of pairs") @@
|
||
|
|
List.to_pair tpl
|
||
|
|
in
|
||
|
|
ok (a , b)
|
||
|
|
in
|
||
|
|
let%bind pairs = bind_map_list aux lst in
|
||
|
|
return @@ E_map pairs
|
||
|
|
)
|
||
|
|
| E_constant ("MAP_EMPTY" , lst) -> (
|
||
|
|
let%bind () =
|
||
|
|
trace_strong (simple_error "MAP_EMPTY expects no parameter") @@
|
||
|
|
Assert.assert_list_empty lst
|
||
|
|
in
|
||
|
|
return @@ E_map []
|
||
|
|
)
|
||
|
|
| e -> return e
|