Reviewed By: jvillard Differential Revision: D4299387 fbshipit-source-id: ee8ea3dmaster
Andrzej Kotulski
8 years ago
committed by
Facebook Github Bot
parent
e5863f5420
commit
56f8757337
@ -0,0 +1,163 @@
|
|||||||
|
(*
|
||||||
|
* 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! Utils
|
||||||
|
|
||||||
|
let count_newlines (path: string): int =
|
||||||
|
let open Core.Std in
|
||||||
|
let f file = In_channel.fold_lines file ~init:0 ~f:(fun i _ -> i + 1) in
|
||||||
|
In_channel.with_file path ~f
|
||||||
|
|
||||||
|
type t =
|
||||||
|
| Absolute of string
|
||||||
|
| RelativeProjectRoot of string (* relative to project root *)
|
||||||
|
| RelativeInferModel of string (* relative to infer models *)
|
||||||
|
[@@deriving compare]
|
||||||
|
|
||||||
|
let equal sf1 sf2 =
|
||||||
|
compare sf1 sf2 = 0
|
||||||
|
|
||||||
|
module OrderedSourceFile =
|
||||||
|
struct
|
||||||
|
type _t = t [@@deriving compare]
|
||||||
|
type t = _t [@@deriving compare]
|
||||||
|
end
|
||||||
|
|
||||||
|
module Map = Map.Make(OrderedSourceFile)
|
||||||
|
|
||||||
|
module Set = Set.Make(OrderedSourceFile)
|
||||||
|
|
||||||
|
let rel_path_from_abs_path root fname =
|
||||||
|
let relative_complemented_fname = filename_to_relative root fname in
|
||||||
|
if string_is_prefix root fname &&
|
||||||
|
Filename.is_relative relative_complemented_fname then
|
||||||
|
Some relative_complemented_fname
|
||||||
|
else None (* The project root is not a prefix of the file name *)
|
||||||
|
|
||||||
|
let from_abs_path fname =
|
||||||
|
if Filename.is_relative fname then
|
||||||
|
(failwithf
|
||||||
|
"ERROR: Path %s is relative, when absolute path was expected .@."
|
||||||
|
fname);
|
||||||
|
(* try to get realpath of source file. Use original if it fails *)
|
||||||
|
let fname_real = try realpath fname with Unix.Unix_error _ -> fname in
|
||||||
|
let project_root_real = realpath Config.project_root in
|
||||||
|
let models_dir_real = Config.models_src_dir in
|
||||||
|
match rel_path_from_abs_path project_root_real fname_real with
|
||||||
|
| Some path -> RelativeProjectRoot path
|
||||||
|
| None -> (
|
||||||
|
match rel_path_from_abs_path models_dir_real fname_real with
|
||||||
|
| Some path -> RelativeInferModel path
|
||||||
|
| None -> Absolute fname (* fname is absolute already *)
|
||||||
|
)
|
||||||
|
|
||||||
|
let curr_encoding = `Enc_crc
|
||||||
|
|
||||||
|
let to_string fname =
|
||||||
|
match fname with
|
||||||
|
| RelativeInferModel path -> "INFER_MODEL/" ^ path
|
||||||
|
| RelativeProjectRoot path
|
||||||
|
| Absolute path -> path
|
||||||
|
|
||||||
|
let pp fmt fname =
|
||||||
|
Format.fprintf fmt "%s" (to_string fname)
|
||||||
|
|
||||||
|
(* Checking if the path exists may be needed only in some cases, hence the flag check_exists *)
|
||||||
|
let to_abs_path fname =
|
||||||
|
match fname with
|
||||||
|
| RelativeProjectRoot path -> Filename.concat Config.project_root path
|
||||||
|
| RelativeInferModel path -> Filename.concat Config.models_src_dir path
|
||||||
|
| Absolute path -> path
|
||||||
|
|
||||||
|
let line_count source_file =
|
||||||
|
let abs_path = to_abs_path source_file in
|
||||||
|
count_newlines abs_path
|
||||||
|
|
||||||
|
let to_rel_path fname =
|
||||||
|
match fname with
|
||||||
|
| RelativeProjectRoot path -> path
|
||||||
|
| _ -> to_abs_path fname
|
||||||
|
|
||||||
|
(** string encoding of a source file (including path) as a single filename *)
|
||||||
|
let encoding source_file =
|
||||||
|
let prefix = match source_file with
|
||||||
|
| RelativeProjectRoot _ -> "P"
|
||||||
|
| RelativeInferModel _ -> "MOD"
|
||||||
|
| Absolute _ -> "ABS" in
|
||||||
|
let source_file_s = to_string source_file in
|
||||||
|
match curr_encoding with
|
||||||
|
| `Enc_base ->
|
||||||
|
Filename.basename source_file_s
|
||||||
|
| `Enc_path_with_underscores ->
|
||||||
|
prefix ^ Escape.escape_path source_file_s
|
||||||
|
| `Enc_crc ->
|
||||||
|
let base = Filename.basename source_file_s in
|
||||||
|
let dir = prefix ^ Filename.dirname source_file_s in
|
||||||
|
string_append_crc_cutoff ~key:dir base
|
||||||
|
|
||||||
|
let empty = Absolute ""
|
||||||
|
|
||||||
|
let is_infer_model source_file = match source_file with
|
||||||
|
| RelativeProjectRoot _ | Absolute _ -> false
|
||||||
|
| RelativeInferModel _ -> true
|
||||||
|
|
||||||
|
(** Returns true if the file is a C++ model *)
|
||||||
|
let is_cpp_model file =
|
||||||
|
match file with
|
||||||
|
| RelativeInferModel path ->
|
||||||
|
string_is_prefix Config.relative_cpp_models_dir path
|
||||||
|
| _ -> false
|
||||||
|
|
||||||
|
let is_under_project_root = function
|
||||||
|
| RelativeProjectRoot _ -> true
|
||||||
|
| Absolute _ | RelativeInferModel _ -> false
|
||||||
|
|
||||||
|
let exists_cache = Hashtbl.create 256
|
||||||
|
|
||||||
|
let path_exists abs_path =
|
||||||
|
try Hashtbl.find exists_cache abs_path
|
||||||
|
with Not_found ->
|
||||||
|
let result = Sys.file_exists abs_path in
|
||||||
|
Hashtbl.add exists_cache abs_path result;
|
||||||
|
result
|
||||||
|
|
||||||
|
let of_header header_file =
|
||||||
|
let abs_path = to_abs_path header_file in
|
||||||
|
let source_exts = ["c"; "cc"; "cpp"; "cxx"; "m"; "mm"] in
|
||||||
|
let header_exts = ["h"; "hh"; "hpp"; "hxx"] in
|
||||||
|
let file_no_ext, ext_opt = Core.Std.Filename.split_extension abs_path in
|
||||||
|
let file_opt = match ext_opt with
|
||||||
|
| Some ext when IList.mem Core.Std.String.equal ext header_exts -> (
|
||||||
|
let possible_files = IList.map (fun ext -> file_no_ext ^ "." ^ ext) source_exts in
|
||||||
|
try Some (IList.find path_exists possible_files)
|
||||||
|
with Not_found -> None
|
||||||
|
)
|
||||||
|
| _ -> None in
|
||||||
|
Option.map from_abs_path file_opt
|
||||||
|
|
||||||
|
let changed_files_set =
|
||||||
|
let create_source_file path =
|
||||||
|
if Filename.is_relative path then
|
||||||
|
(* sources in changed-files-index may be specified relative to project root *)
|
||||||
|
RelativeProjectRoot path
|
||||||
|
else
|
||||||
|
from_abs_path path in
|
||||||
|
Option.map_default read_file None Config.changed_files_index |>
|
||||||
|
Option.map (
|
||||||
|
IList.fold_left
|
||||||
|
(fun changed_files line ->
|
||||||
|
let source_file = create_source_file line in
|
||||||
|
let changed_files' = Set.add source_file changed_files in
|
||||||
|
(* Add source corresponding to changed header if it exists *)
|
||||||
|
match of_header source_file with
|
||||||
|
| Some src -> Set.add src changed_files'
|
||||||
|
| None -> changed_files'
|
||||||
|
)
|
||||||
|
Set.empty
|
||||||
|
)
|
||||||
@ -0,0 +1,62 @@
|
|||||||
|
(*
|
||||||
|
* 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.
|
||||||
|
*)
|
||||||
|
|
||||||
|
type t [@@deriving compare]
|
||||||
|
|
||||||
|
(** equality of source files *)
|
||||||
|
val equal : t -> t -> bool
|
||||||
|
|
||||||
|
(** Maps from source_file *)
|
||||||
|
module Map : Map.S with type key = t
|
||||||
|
|
||||||
|
(** Set of source files *)
|
||||||
|
module Set : Set.S with type elt = t
|
||||||
|
|
||||||
|
(** compute line count of a source file *)
|
||||||
|
val line_count : t -> int
|
||||||
|
|
||||||
|
(** empty source file *)
|
||||||
|
val empty : t
|
||||||
|
|
||||||
|
(** create source file from absolute path *)
|
||||||
|
val from_abs_path : string -> t
|
||||||
|
|
||||||
|
(** string encoding of a source file (including path) as a single filename *)
|
||||||
|
val encoding : t -> string
|
||||||
|
|
||||||
|
(** convert a source file to a string
|
||||||
|
WARNING: result may not be valid file path, do not use this function to perform operations
|
||||||
|
on filenames *)
|
||||||
|
val to_string : t -> string
|
||||||
|
|
||||||
|
(** pretty print t *)
|
||||||
|
val pp : Format.formatter -> t -> unit
|
||||||
|
|
||||||
|
(** get the full path of a source file *)
|
||||||
|
val to_abs_path : t -> string
|
||||||
|
|
||||||
|
(** get the relative path of a source file *)
|
||||||
|
val to_rel_path : t -> string
|
||||||
|
|
||||||
|
val is_infer_model : t -> bool
|
||||||
|
|
||||||
|
(** Returns true if the file is a C++ model *)
|
||||||
|
val is_cpp_model : t -> bool
|
||||||
|
|
||||||
|
(** Returns true if the file is in project root *)
|
||||||
|
val is_under_project_root : t -> bool
|
||||||
|
|
||||||
|
(** Return approximate source file corresponding to the parameter if it's header file and
|
||||||
|
file exists. returns None otherwise *)
|
||||||
|
val of_header : t -> t option
|
||||||
|
|
||||||
|
(** Set of files read from --changed-files-index file, None if option not specified
|
||||||
|
NOTE: it may include extra source_files if --changed-files-index contains paths to
|
||||||
|
header files *)
|
||||||
|
val changed_files_set : Set.t option
|
||||||
Loading…
Reference in new issue