|
|
|
(*
|
|
|
|
* Copyright (c) 2016 - present Facebook, Inc.
|
|
|
|
* All rights reserved.
|
|
|
|
*
|
|
|
|
* This source code is licensed under the BSD style license found in the
|
|
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
*)
|
|
|
|
|
|
|
|
open! IStd
|
|
|
|
|
|
|
|
type compilation_data = {
|
|
|
|
dir : string;
|
|
|
|
command : string;
|
|
|
|
args : string;
|
|
|
|
}
|
|
|
|
|
|
|
|
type t = compilation_data SourceFile.Map.t ref
|
|
|
|
let empty () = ref SourceFile.Map.empty
|
|
|
|
|
|
|
|
let get_size database = SourceFile.Map.cardinal !database
|
|
|
|
|
|
|
|
let iter database f = SourceFile.Map.iter f !database
|
|
|
|
|
|
|
|
let find database key = SourceFile.Map.find key !database
|
|
|
|
|
|
|
|
let parse_command_and_arguments command_and_arguments =
|
|
|
|
let regexp = Str.regexp "[^\\][ ]" in
|
|
|
|
let index = Str.search_forward regexp command_and_arguments 0 in
|
|
|
|
let command = Str.string_before command_and_arguments (index+1) in
|
|
|
|
let arguments = Str.string_after command_and_arguments (index+1) in
|
|
|
|
command, arguments
|
|
|
|
|
|
|
|
(** Parse the compilation database json file into the compilationDatabase
|
|
|
|
map. The json file consists of an array of json objects that contain the file
|
|
|
|
to be compiled, the directory to be compiled in, and the compilation command as a list
|
|
|
|
and as a string. We pack this information into the compilationDatabase map, and remove the
|
|
|
|
clang invocation part, because we will use a clang wrapper. *)
|
|
|
|
let decode_json_file (database : t) json_path =
|
|
|
|
let exit_format_error () =
|
|
|
|
failwith ("Json file doesn't have the expected format") in
|
|
|
|
let json = Yojson.Basic.from_file json_path in
|
|
|
|
let get_dir el =
|
|
|
|
match el with
|
|
|
|
| ("directory", `String dir) -> Some dir
|
|
|
|
| _ -> None in
|
|
|
|
let get_file el =
|
|
|
|
match el with
|
|
|
|
| ("file", `String file) -> Some file
|
|
|
|
| _ -> None in
|
|
|
|
let get_cmd el =
|
|
|
|
match el with
|
|
|
|
| ("command", `String cmd) -> Some cmd
|
|
|
|
| _ -> None in
|
|
|
|
let rec parse_json json =
|
|
|
|
match json with
|
|
|
|
| `List arguments ->
|
|
|
|
IList.iter parse_json arguments
|
|
|
|
| `Assoc l ->
|
|
|
|
let dir = match IList.find_map_opt get_dir l with
|
|
|
|
| Some dir -> dir
|
|
|
|
| None -> exit_format_error () in
|
|
|
|
let file = match IList.find_map_opt get_file l with
|
|
|
|
| Some file -> file
|
|
|
|
| None -> exit_format_error () in
|
|
|
|
let cmd = match IList.find_map_opt get_cmd l with
|
|
|
|
| Some cmd -> cmd
|
|
|
|
| None -> exit_format_error () in
|
|
|
|
let command, args = parse_command_and_arguments cmd in
|
|
|
|
let compilation_data = { dir; command; args;} in
|
|
|
|
let source_file = SourceFile.from_abs_path file in
|
|
|
|
database := SourceFile.Map.add source_file compilation_data !database
|
|
|
|
| _ -> exit_format_error () in
|
|
|
|
parse_json json
|
|
|
|
|
|
|
|
let from_json_files db_json_files =
|
|
|
|
let db = empty () in
|
|
|
|
IList.iter (decode_json_file db) db_json_files;
|
|
|
|
db
|