Source file `parse_generic.ml`

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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
(* Yoann Padioleau
 *
 * Copyright (C) 2019 r2c
 *
 * 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

(*****************************************************************************)
(* Prelude *)
(*****************************************************************************)
(* Wrappers around many languages to transform them in a generic AST
 * (see ast_generic.ml)
 *)

(*****************************************************************************)
(* Entry point *)
(*****************************************************************************)

let parse_with_lang lang file = 
  match lang with
  | Lang.Python ->
    let ast = Parse_python.parse_program file in
    Resolve_python.resolve ast;
    Python_to_generic.program ast
  | Lang.Javascript ->
    let cst = Parse_js.parse_program file in
    let ast = Ast_js_build.program cst in
    Js_to_generic.program ast
  | Lang.C ->
    (* this internally uses the CST for c++ *)
    let ast = Parse_c.parse_program file in
    C_to_generic.program ast
  | Lang.Java ->
    let ast = Parse_java.parse_program file in
    Java_to_generic.program ast
  | Lang.ML ->
    let cst = Parse_ml.parse_program file in
    let ast = Ast_ml_build.program cst in
    let _ = Ml_to_generic.program ast in
    raise Todo

let parse_program file =
  match Lang.lang_of_filename_opt file with
  | Some x -> parse_with_lang x file
  | None -> failwith (spf "unsupported file for AST generic: %s" file)


let parse_pattern lang str =
  Common.save_excursion Flag_parsing.sgrep_mode true (fun () ->
  match lang with
  | Lang.Python ->
      let any = Parse_python.any_of_string str in
      Python_to_generic.any any
  | Lang.Javascript ->
      let any_cst = Parse_js.any_of_string str in
      let any = Ast_js_build.any any_cst in
      Js_to_generic.any any
  | Lang.C ->
      let any = Parse_c.any_of_string str in
      C_to_generic.any any
  | Lang.Java ->
      let any = Parse_java.any_of_string str in
      Java_to_generic.any any
  | Lang.ML -> 
      raise Todo
  )