1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
open Ppxlib
let ext t ~locality extension_kind =
Extension.declare_with_path_arg
(Ppx_let_expander.ext_full_name t ~locality extension_kind)
Extension.Context.expression
Ast_pattern.(single_expr_payload __)
(fun ~loc:_ ~path:_ ~arg expr ->
Ppx_let_expander.expand t extension_kind ~modul:arg ~locality expr)
;;
open Ppx_let_expander
module List = struct
include List
let concat_map list ~f = List.concat_map f list
let map list ~f = List.map f list
end
let () =
let extensions =
List.concat_map [ bind; map ] ~f:(fun t ->
List.concat_map [ `local; `global ] ~f:(fun locality ->
List.map
Extension_kind.[ default; default_open; n; n_open ]
~f:(fun kind -> ext t ~locality kind)))
in
Driver.register_transformation "let" ~extensions
;;