Source file visitor_js.ml

(* Yoann Padioleau
 *
 * Copyright (C) 2010 Facebook
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public License
 * version 2.1 as published by the Free Software Foundation, with the
 * special exception on linking described in file license.txt.
 *
 * This library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the file
 * license.txt for more details.
 *)
open Common
open Ocaml
open Cst_js

(*****************************************************************************)
(* Prelude *)
(*****************************************************************************)

(* hooks *)
type visitor_in = {
  kexpr: (expr  -> unit) * visitor_out -> expr  -> unit;
  kstmt: (stmt  -> unit) * visitor_out -> stmt  -> unit;
  kprop: (property -> unit) * visitor_out -> property -> unit;
  kinfo: (tok -> unit)  * visitor_out -> tok  -> unit;
}
and visitor_out = any -> unit

let default_visitor =
  { kexpr   = (fun (k,_) x -> k x);
    kstmt   = (fun (k,_) x -> k x);
    kinfo   = (fun (k,_) x -> k x);
    kprop = (fun (k,_) x -> k x);
  }

let (mk_visitor: visitor_in -> visitor_out) = fun vin ->

(* start of auto generation *)


let rec v_info x =
  let k x = match x with { Parse_info.
     token = _v_pinfox; transfo = _v_transfo
    } ->
(*
    let arg = Parse_info.v_pinfo v_pinfox in
    let arg = v_unit v_comments in
    let arg = Parse_info.v_transformation v_transfo in
*)
    ()
  in
  vin.kinfo (k, all_functions) x

and v_tok v = v_info v

and v_wrap: 'a. ('a -> unit) -> 'a wrap -> unit = fun _of_a (v1, v2) ->
  let v1 = _of_a v1 and v2 = v_info v2 in ()
and v_angle: 'a. ('a -> unit) -> 'a angle -> unit = fun _of_a (v1, v2, v3) ->
  let v1 = v_tok v1 and v2 = _of_a v2 and v3 = v_tok v3 in ()
and v_paren: 'a. ('a -> unit) -> 'a paren -> unit = fun _of_a (v1, v2, v3) ->
  let v1 = v_tok v1 and v2 = _of_a v2 and v3 = v_tok v3 in ()
and v_brace: 'a. ('a -> unit) -> 'a brace -> unit = fun _of_a (v1, v2, v3) ->
  let v1 = v_tok v1 and v2 = _of_a v2 and v3 = v_tok v3 in ()
and v_bracket: 'a. ('a -> unit) -> 'a bracket -> unit = fun _of_a (v1, v2, v3)->
  let v1 = v_tok v1 and v2 = _of_a v2 and v3 = v_tok v3 in ()

and v_comma x = v_tok x

and v_comma_list: 'a. ('a -> unit) -> 'a comma_list -> unit = fun _of_a xs ->
  xs +> List.iter (function | Left x -> _of_a x | Right info -> v_comma info)

and v_sc v = v_option v_tok v

and v_name v = v_wrap v_string v
and v_module_path v = v_wrap v_string v

and v_expr (x: expr) =
  (* tweak *)
  let k x =  match x with
  | L v1 -> let v1 = v_litteral v1 in ()
  | V v1 -> let v1 = v_name v1 in ()
  | This v1 -> let v1 = v_tok v1 in ()
  | Super v1 -> let v1 = v_tok v1 in ()
  | NewTarget ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_tok v2 and v3 = v_tok v3 in ()
  | U ((v1, v2)) -> let v1 = v_wrap v_unop v1 and v2 = v_expr v2 in ()
  | B ((v1, v2, v3)) ->
      let v1 = v_expr v1 and v2 = v_wrap v_binop v2 and v3 = v_expr v3 in ()
  | Bracket ((v1, v2)) ->
      let v1 = v_expr v1 and v2 = v_bracket v_expr v2 in ()
  | Period ((v1, v2, v3)) ->
      let v1 = v_expr v1 and v2 = v_tok v2 and v3 = v_name v3 in ()
  | Object v1 ->
      let v1 =
        v_brace
          (v_comma_list v_property

             )
          v1
      in ()
  | Array v1 -> let v1 = v_bracket (v_comma_list v_expr) v1 in ()
  | Apply ((v1, v2)) ->
      let v1 = v_expr v1 and v2 = v_paren (v_comma_list v_expr) v2 in ()
  | Conditional ((v1, v2, v3, v4, v5)) ->
      let v1 = v_expr v1
      and v2 = v_tok v2
      and v3 = v_expr v3
      and v4 = v_tok v4
      and v5 = v_expr v5
      in ()
  | Assign ((v1, v2, v3)) ->
      let v1 = v_expr v1
      and v2 = v_wrap v_assignment_operator v2
      and v3 = v_expr v3
      in ()
  | Seq ((v1, v2, v3)) ->
      let v1 = v_expr v1 and v2 = v_tok v2 and v3 = v_expr v3 in ()
  | Function v1 -> let v1 = v_func_decl v1 in ()
  | Class v1 -> let v1 = v_class_decl v1 in ()
  | Arrow v1 -> let v1 = v_arrow_func v1 in ()
  | Paren v1 -> let v1 = v_paren v_expr v1 in ()
  | XhpHtml v1 -> let v1 = v_xhp_html v1 in ()
  | Encaps ((v1, v2, v3, v4)) ->
      let v1 = v_option v_name v1
      and v2 = v_tok v2
      and v3 = v_list v_encaps v3
      and v4 = v_tok v4
      in ()
  | Yield ((v1, v2, v3)) ->
      let v1 = v_tok v1
      and v2 = v_option v_tok v2
      and v3 = v_option v_expr v3
      in ()
  | Await ((v1, v2)) -> let v1 = v_tok v1 and v2 = v_expr v2 in ()


  in
  vin.kexpr (k, all_functions) x

and v_property x =
  let k = function
    | P_field (v1, v2, v3) ->
        let v1 = v_property_name v1
        and v2 = v_tok v2
        and v3 = v_expr v3
        in ()
    | P_method v1 ->
        let v1 = v_func_decl v1
        in ()
    | P_shorthand v1 -> let v1 = v_name v1 in ()
    | P_spread (v1, v2) ->
        let v1 = v_tok v1 in
        let v2 = v_expr v2 in
        ()

  in
  vin.kprop (k, all_functions) x

and v_litteral =
  function
  | Bool v1 -> let v1 = v_wrap v_bool v1 in ()
  | Num v1 -> let v1 = v_wrap v_string v1 in ()
  | String v1 -> let v1 = v_wrap v_string v1 in ()
  | Regexp v1 -> let v1 = v_wrap v_string v1 in ()
  | Null v1 -> let v1 = v_tok v1 in ()
and v_unop =
  function
  | U_new -> ()
  | U_delete -> ()
  | U_void -> ()
  | U_typeof -> ()
  | U_bitnot -> ()
  | U_pre_increment -> ()
  | U_pre_decrement -> ()
  | U_post_increment -> ()
  | U_post_decrement -> ()
  | U_plus -> ()
  | U_minus -> ()
  | U_not -> ()
  | U_spread -> ()
and v_binop =
  function
  | B_instanceof -> ()
  | B_in -> ()
  | B_mul -> ()
  | B_expo -> ()
  | B_div -> ()
  | B_mod -> ()
  | B_add -> ()
  | B_sub -> ()
  | B_le -> ()
  | B_ge -> ()
  | B_lt -> ()
  | B_gt -> ()
  | B_lsr -> ()
  | B_asr -> ()
  | B_lsl -> ()
  | B_equal -> ()
  | B_notequal -> ()
  | B_physequal -> ()
  | B_physnotequal -> ()
  | B_bitand -> ()
  | B_bitor -> ()
  | B_bitxor -> ()
  | B_and -> ()
  | B_or -> ()
and v_property_name =
  function
  | PN_Id v1 -> let v1 = v_name v1 in ()
  | PN_String v1 -> let v1 = v_name v1 in ()
  | PN_Num v1 -> let v1 = v_wrap v_string v1 in ()
  | PN_Computed v1 -> let v1 = v_bracket v_expr v1 in ()
and v_assignment_operator =
  function
  | A_eq -> ()
  | A_mul -> ()
  | A_div -> ()
  | A_mod -> ()
  | A_add -> ()
  | A_sub -> ()
  | A_lsl -> ()
  | A_lsr -> ()
  | A_asr -> ()
  | A_and -> ()
  | A_xor -> ()
  | A_or -> ()
and v_xhp_html =
  function
  | Xhp ((v1, v2, v3, v4, v5)) ->
      let v1 = v_wrap v_xhp_tag v1
      and v2 = v_list v_xhp_attribute v2
      and v3 = v_tok v3
      and v4 = v_list v_xhp_body v4
      and v5 = v_wrap (v_option v_xhp_tag) v5
      in ()
  | XhpSingleton ((v1, v2, v3)) ->
      let v1 = v_wrap v_xhp_tag v1
      and v2 = v_list v_xhp_attribute v2
      and v3 = v_tok v3
      in ()
and v_xhp_attribute =
  function
  | XhpAttrValue ((v1, v2, v3)) ->
      let v1 = v_xhp_attr_name v1
      and v2 = v_tok v2
      and v3 = v_xhp_attr_value v3
      in ()
  | XhpAttrNoValue ((v1)) ->
      let v1 = v_xhp_attr_name v1
      in ()
  | XhpAttrSpread v1 ->
      let v1 =
        v_brace
          (fun (v1, v2) -> let v1 = v_tok v1 and v2 = v_expr v2 in ()) v1
      in ()
and v_xhp_attr_name v = v_wrap v_string v
and v_xhp_attr_value =
  function
  | XhpAttrString v1 -> let v1 = v_wrap v_string v1 in ()
  | XhpAttrExpr v1 -> let v1 = v_brace v_expr v1 in ()
and v_xhp_body =
  function
  | XhpText v1 -> let v1 = v_wrap v_string v1 in ()
  | XhpExpr v1 -> let v1 = v_brace (v_option v_expr) v1 in ()
  | XhpNested v1 -> let v1 = v_xhp_html v1 in ()
and v_xhp_tag x = v_string x
and v_encaps =
  function
  | EncapsString v1 -> let v1 = v_wrap v_string v1 in ()
  | EncapsExpr ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_expr v2 and v3 = v_tok v3 in ()

and v_st x =
  let k x = match x with
  | VarsDecl ((v1, v2, v3)) ->
      let v1 = v_wrap v_var_kind v1
      and v2 = v_comma_list v_var_binding v2
      and v3 = v_sc v3
      in ()
  | Block v1 -> let v1 = v_brace (v_list v_item) v1 in ()
  | Nop v1 -> let v1 = v_sc v1 in ()
  | ExprStmt ((v1, v2)) -> let v1 = v_expr v1 and v2 = v_sc v2 in ()
  | If ((v1, v2, v3, v4)) ->
      let v1 = v_tok v1
      and v2 = v_paren v_expr v2
      and v3 = v_st v3
      and v4 =
        v_option (fun (v1, v2) -> let v1 = v_tok v1 and v2 = v_st v2 in ())
          v4
      in ()
  | Do ((v1, v2, v3, v4, v5)) ->
      let v1 = v_tok v1
      and v2 = v_st v2
      and v3 = v_tok v3
      and v4 = v_paren v_expr v4
      and v5 = v_sc v5
      in ()
  | While ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_paren v_expr v2 and v3 = v_st v3 in ()
  | For ((v1, v2, v3, v4, v5, v6, v7, v8, v9)) ->
      let v1 = v_tok v1
      and v2 = v_tok v2
      and v3 = v_option v_lhs_or_vars v3
      and v4 = v_tok v4
      and v5 = v_option v_expr v5
      and v6 = v_tok v6
      and v7 = v_option v_expr v7
      and v8 = v_tok v8
      and v9 = v_st v9
      in ()
  | ForIn ((v1, v2, v3, v4, v5, v6, v7)) ->
      let v1 = v_tok v1
      and v2 = v_tok v2
      and v3 = v_lhs_or_var v3
      and v4 = v_tok v4
      and v5 = v_expr v5
      and v6 = v_tok v6
      and v7 = v_st v7
      in ()
  | ForOf ((v1, v2, v3, v4, v5, v6, v7)) ->
      let v1 = v_tok v1
      and v2 = v_tok v2
      and v3 = v_lhs_or_var v3
      and v4 = v_tok v4
      and v5 = v_expr v5
      and v6 = v_tok v6
      and v7 = v_st v7
      in ()
  | Switch ((v1, v2, v3)) ->
      let v1 = v_tok v1
      and v2 = v_paren v_expr v2
      and v3 = v_brace (v_list v_case_clause) v3
      in ()
  | Continue ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_option v_label v2 and v3 = v_sc v3 in ()
  | Break ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_option v_label v2 and v3 = v_sc v3 in ()
  | Return ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_option v_expr v2 and v3 = v_sc v3 in ()
  | With ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_paren v_expr v2 and v3 = v_st v3 in ()
  | Labeled ((v1, v2, v3)) ->
      let v1 = v_label v1 and v2 = v_tok v2 and v3 = v_st v3 in ()
  | Throw ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_expr v2 and v3 = v_sc v3 in ()
  | Try ((v1, v2, v3, v4)) ->
      let v1 = v_tok v1
      and v2 = v_st v2
      and v3 =
        v_option
          (fun (v1, v2, v3) ->
             let v1 = v_tok v1
             and v2 = v_paren v_arg v2
             and v3 = v_st v3
             in ())
          v3
      and v4 =
        v_option (fun (v1, v2) -> let v1 = v_tok v1 and v2 = v_st v2 in ())
          v4
      in ()
  in
  vin.kstmt (k, all_functions) x

and v_label v = v_wrap v_string v
and v_lhs_or_vars =
  function
  | LHS1 v1 -> let v1 = v_expr v1 in ()
  | ForVars ((v1, v2)) ->
      let v1 = v_wrap v_var_kind v1 
      and v2 = v_comma_list v_var_binding v2 
      in ()
and v_lhs_or_var =
  function
  | LHS2 v1 -> let v1 = v_expr v1 in ()
  | ForVar ((v1, v2)) ->
      let v1 = v_wrap v_var_kind v1 
      and v2 = v_var_binding v2 
      in ()
and v_case_clause =
  function
  | Default ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_tok v2 and v3 = v_list v_item v3 in ()
  | Case ((v1, v2, v3, v4)) ->
      let v1 = v_tok v1
      and v2 = v_expr v2
      and v3 = v_tok v3
      and v4 = v_list v_item v4
      in ()
and v_arg v = v_wrap v_string v
and v_type_ =
  function
  | TTodo -> ()
  | TName v1 -> let v1 = v_nominal_type v1 in ()
  | TQuestion ((v1, v2)) -> let v1 = v_tok v1 and v2 = v_type_ v2 in ()
  | TFun ((v1, v2, v3)) ->
      let v1 =
        v_paren
          (v_comma_list
             (fun (v1, v2) ->
                let v1 = v_param_name v1
                and v2 = v_annotation v2
                in ()))
          v1
      and v2 = v_tok v2
      and v3 = v_type_ v3
      in ()
  | TObj v1 ->
      let v1 =
        v_brace
          (v_list
             (fun (v1, v2, v3) ->
                let v1 = v_property_name v1
                and v2 = v_annotation v2
                and v3 = v_sc v3
                in ()))
          v1
      in ()
and v_param_name = function
  | RequiredParam(v1) ->
      let v1 = v_name v1
      in ()
  | OptionalParam(v1,v2) ->
      let v1 = v_name v1
      and v2 = v_tok v2
      in ()
  | RestParam(v1,v2) ->
      let v1 = v_tok v1
      and v2 = v_name v2
      in ()

and v_nominal_type ((v1,v2)) =
  let v1 = v_expr v1 in
  let v2 = v_option (v_angle (v_comma_list v_type_)) v2 in
  ()
and v_type_opt v =
  v_option v_annotation v
and v_annotation = function
  | TAnnot (v1,v2) ->
      let v1 = v_tok v1 in
      let v2 = v_type_ v2 in
      ()
  | TFunAnnot (v1,v2,v3,v4) ->
      let v1 = v_option (v_angle (v_comma_list v_name)) v1 in
      let v2 =
        v_paren
          (v_comma_list
             (fun (v1, v2) ->
                let v1 = v_param_name v1
                and v2 = v_annotation v2
                in ()))
          v2
      and v3 = v_tok v3
      and v4 = v_type_ v4
      in ()

and
  v_func_decl {
                f_kind = v_f_kind;
                f_properties = v_f_properties;
                f_params = v_f_params;
                f_body = v_f_body;
                f_type_params = v_f_type_params;
                f_return_type = v_f_return_type
              } =
  let arg = v_func_kind v_f_kind in
  let arg = v_list v_func_property v_f_properties in
  let arg = v_paren (v_comma_list v_parameter_binding) v_f_params in
  let arg = v_brace (v_list v_item) v_f_body in
  let arg = v_option v_type_parameters v_f_type_params in
  let arg = v_type_opt v_f_return_type in ()
and v_type_parameters v = v_angle (v_comma_list v_type_parameter) v
and v_type_parameter v = v_name v

and v_func_kind =
  function
  | F_func ((v1, v2)) -> let v1 = v_tok v1 and v2 = v_option v_name v2 in ()
  | F_method v1 -> let v1 = v_property_name v1 in ()
  | F_get ((v1, v2)) -> let v1 = v_tok v1 and v2 = v_property_name v2 in ()
  | F_set ((v1, v2)) -> let v1 = v_tok v1 and v2 = v_property_name v2 in ()
and v_func_property =
  function
  | Generator v1 -> let v1 = v_tok v1 in ()
  | Async v1 -> let v1 = v_tok v1 in ()

and v_parameter_binding =
  function
  | ParamClassic v1 -> let v1 = v_parameter v1 in ()
  | ParamPattern v1 -> let v1 = v_parameter_pattern v1 in ()
and
  v_parameter_pattern {
                        ppat = v_ppat;
                        ppat_type = v_ppat_type;
                        ppat_default = v_ppat_default
                      } =
  let arg = v_pattern v_ppat in
  let arg = v_type_opt v_ppat_type in
  let arg =
    v_option (fun (v1, v2) -> let v1 = v_tok v1 and v2 = v_expr v2 in ())
      v_ppat_default
  in ()


and v_parameter { p_name = v_p_name; p_type = v_p_type; p_default; p_dots } =
  let arg = v_option v_default p_default in
  let arg = v_option v_tok p_dots in
  let arg = v_name v_p_name in
  let arg = v_type_opt v_p_type in
  ()
and v_default = function
  | DNone v1 -> let _ = v_tok v1 in ()
  | DSome (v1,v2) -> let _ = v_tok v1 and _ = v_expr v2 in ()
and
  v_arrow_func { a_params = v_a_params; a_return_type = v_a_return_type;
                 a_tok = v_a_tok; a_body = v_a_body }
               =
  let arg = v_arrow_params v_a_params in
  let arg = v_type_opt v_a_return_type in
  let arg = v_tok v_a_tok in let arg = v_arrow_body v_a_body in ()
and v_arrow_params =
  function
  | ASingleParam v1 -> let v1 = v_parameter_binding v1 in ()
  | AParams v1 -> let v1 = v_paren (v_comma_list v_parameter_binding) v1 in ()
and v_arrow_body =
  function
  | AExpr v1 -> let v1 = v_expr v1 in ()
  | ABody v1 -> let v1 = v_brace (v_list v_item) v1 in ()
and v_var_kind = function | Var -> () | Const -> () | Let -> ()

and v_var_binding =
  function
  | VarClassic v1 -> let v1 = v_variable_declaration v1 in ()
  | VarPattern v1 -> let v1 = v_variable_declaration_pattern v1 in ()

and v_init (v1, v2) = let v1 = v_tok v1 and v2 = v_expr v2 in ()
and
  v_variable_declaration_pattern {
                                   vpat = v_vpat;
                                   vpat_init = v_vpat_init;
                                   vpat_type = v_vpat_type
                                 } =
  let arg = v_pattern v_vpat in
  let arg = v_option v_init v_vpat_init in
  let arg = v_type_opt v_vpat_type in ()
and v_pattern =
  function
  | PatObj v1 -> let v1 = v_brace (v_comma_list v_pattern) v1 in ()
  | PatArr v1 -> let v1 = v_bracket (v_comma_list v_pattern) v1 in ()
  | PatId ((v1, v2)) -> let v1 = v_name v1 and v2 = v_option v_init v2 in ()
  | PatProp ((v1, v2, v3)) ->
      let v1 = v_property_name v1
      and v2 = v_tok v2
      and v3 = v_pattern v3
      in ()
  | PatDots ((v1, v2)) -> let v1 = v_tok v1 and v2 = v_pattern v2 in ()
  | PatNest ((v1, v2)) ->
      let v1 = v_pattern v1 and v2 = v_option v_init v2 in ()

and v_variable_declaration {
                           v_name = v_v_name;
                           v_init = v_v_init;
                           v_type = v_v_type
                         } =
  let arg = v_name v_v_name in
  let arg =
    v_option (fun (v1, v2) -> let v1 = v_tok v1 and v2 = v_expr v2 in ())
      v_v_init in
  let arg = v_type_opt v_v_type in ()


and
  v_class_decl {
                 c_tok = v_c_tok;
                 c_name = v_c_name;
                 c_type_params = v_c_type_params;
                 c_extends = v_c_extends;
                 c_body = v_c_body
               } =
  let arg = v_tok v_c_tok in
  let arg = v_option v_name v_c_name in
  let arg = v_option (v_angle (v_comma_list v_name)) v_c_type_params in
  let arg =
    v_option
      (fun (v1, v2) -> let v1 = v_tok v1 and v2 = v_nominal_type v2 in ())
      v_c_extends in
  let arg = v_brace (v_list v_class_stmt) v_c_body in
  ()
and
  v_interface_decl {
                 i_tok = v_i_tok;
                 i_name = v_i_name;
                 i_type_params = v_i_type_params;
                 i_type = v_i_type;
               } =
  let arg = v_tok v_i_tok in
  let arg = v_name v_i_name in
  let arg = v_option (v_angle (v_comma_list v_name)) v_i_type_params in
  let arg = v_type_ v_i_type in
  ()

and v_static_opt v = v_option v_tok v
and
  v_field_decl {
                 fld_static = v_fld_static;
                 fld_name = v_fld_name;
                 fld_type = v_fld_type;
                 fld_init = v_fld_init
               } =
  let arg = v_static_opt v_fld_static in
  let arg = v_property_name v_fld_name in
  let arg = v_type_opt v_fld_type in
  let arg = v_option v_init v_fld_init in ()

and v_class_stmt =
  function
  | C_field ((v1, v2)) -> let v1 = v_field_decl v1 and v2 = v_sc v2 in ()
  | C_method ((v1, v2)) ->
      let v1 = v_option v_tok v1 and v2 = v_func_decl v2 in ()
  | C_extrasemicolon v1 -> let v1 = v_sc v1 in ()

and v_item =
  function
  | ItemTodo v1 -> let v1 = v_tok v1 in ()
  | St v1 -> let v1 = v_st v1 in ()
  | FunDecl v1 -> let v1 = v_func_decl v1 in ()
  | ClassDecl v1 -> let v1 = v_class_decl v1 in ()
  | InterfaceDecl v1 -> let v1 = v_interface_decl v1 in ()

and v_import =
  function
  | ImportFrom v1 ->
      let v1 =
        (match v1 with
         | (v1, v2) ->
             let v1 = v_import_clause v1
             and v2 =
               (match v2 with
                | (v1, v2) ->
                    let v1 = v_tok v1 and v2 = v_module_path v2 in ())
             in ())
      in ()
  | ImportEffect v1 -> let v1 = v_module_path v1 in ()
and v_import_clause (v1, v2) =
  let v1 = v_option v_import_default v1
  and v2 = v_option v_name_import v2
  in ()
and v_name_import =
  function
  | ImportNamespace ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_tok v2 and v3 = v_name v3 in ()
  | ImportNames v1 ->
      let v1 = v_brace (v_comma_list v_import_name) v1 in ()
  | ImportTypes (v1, v2) ->
      let v1 = v_tok v1 and v2 = v_brace (v_comma_list v_import_name) v2 in ()
and v_import_default v = v_name v
and v_import_name (v1, v2) =
  let v1 = v_name v1
  and v2 =
    v_option (fun (v1, v2) -> let v1 = v_tok v1 and v2 = v_name v2 in ()) v2
  in ()
and v_export =
  function
  | ExportDecl v1 -> let v1 = v_item v1 in ()
  | ExportDefaultDecl ((v1, v2)) ->
      let v1 = v_tok v1 and v2 = v_item v2 in ()
  | ExportDefaultExpr ((v1, v2, v3)) ->
      let v1 = v_tok v1 and v2 = v_expr v2 and v3 = v_sc v3 in ()
  | ExportNames ((v1, v2)) ->
      let v1 = v_brace (v_comma_list v_import_name) v1
      and v2 = v_sc v2
      in ()
  | ReExportNamespace ((v1, v2, v3)) ->
      let v1 = v_tok v1
      and v2 =
        (match v2 with
         | (v1, v2) -> let v1 = v_tok v1 and v2 = v_module_path v2 in ())
      and v3 = v_sc v3
      in ()
  | ReExportNames ((v1, v2, v3)) ->
      let v1 = v_brace (v_comma_list v_import_name) v1
      and v2 =
        (match v2 with
         | (v1, v2) -> let v1 = v_tok v1 and v2 = v_module_path v2 in ())
      and v3 = v_sc v3
      in ()

        
and v_module_item =
  function
  | It v1 -> let v1 = v_item v1 in ()
  | Import v1 ->
      let v1 =
        (match v1 with
         | (v1, v2, v3) ->
             let v1 = v_tok v1 and v2 = v_import v2 and v3 = v_sc v3 in ())
      in ()
  | Export v1 ->
      let v1 =
        (match v1 with
         | (v1, v2) -> let v1 = v_tok v1 and v2 = v_export v2 in ())
      in ()


and v_program v = v_list v_module_item v

and v_any =  function
  | Expr v1 -> let v1 = v_expr v1 in ()
  | Stmt v1 -> let v1 = v_st v1 in ()
  | Pattern v1 -> let v1 = v_pattern v1 in ()
  | Item v1 -> let v1 = v_item v1 in ()
  | Program v1 -> let v1 = v_program v1 in ()

and all_functions x = v_any x
in
all_functions

(*****************************************************************************)
(* Helpers *)
(*****************************************************************************)

let do_visit_with_ref mk_hooks = fun any ->
  let res = ref [] in
  let hooks = mk_hooks res in
  let vout = mk_visitor hooks in
  vout any;
  List.rev !res