-
Notifications
You must be signed in to change notification settings - Fork 125
/
ofile.ml
74 lines (63 loc) · 3.67 KB
/
ofile.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
(*
Copyright © 2011 MLstate
This file is part of Opa.
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*)
(* depends *)
module String = BaseString
(** Simplier version of ofile + retourned type is an option (safer) *)
let extension = "of"
let error s = prerr_endline (Printf.sprintf "[!] ofile : %s" s); exit 1
let do_verbose = ref false
let verbose s =
if !do_verbose then prerr_endline (Printf.sprintf "ofile : %s" s)
let safe_content file =
try (String.escaped (File.content file))
with Unix.Unix_error(e,s0,s1) -> error (Printf.sprintf "cannot load \"%s\" from \"%s\" [%s,%s,%s]" file (Sys.getcwd ()) (Unix.error_message e) s0 s1)
let win2unix f =
let res = String.copy f in
let n = String.length res in
for i = 0 to n-1 do
if res.[i] = '\\' then res.[i] <- '/'
done;
res
let iter_of path file =
if File.extension file <> extension
then error (Printf.sprintf "input file \"%s\" must have .%s extension" file extension)
else
if Sys.file_exists file && not (Sys.is_directory file)
then
let buf = FBuffer.create 1024 in
let buf = FBuffer.addln buf "let get_file = function" in
let files_list = List.rev
(File.lines_fold
(fun acc line -> let t = String.trim line in if t = "" then acc else t::acc)
[] file) in
let portable_path _path file =
let path_file =(*_path^*)file in
(Mlstate_platform.platform_dependent
~unix:PathTransform.string_to_unix
~windows:PathTransform.string_to_windows ()) path_file
in
let buf = List.fold_left
(fun buf t -> FBuffer.addln buf (Printf.sprintf " | %S -> Some \"%s\"" t (safe_content (portable_path path t) ))) buf files_list in
let buf = FBuffer.addln buf " | _ -> None\n" in
let buf = FBuffer.addln buf "let file_list =" in
let buf = FBuffer.addln buf (String.concat_map ~left:"[" ~right:"]" "; " (fun f -> Printf.sprintf "%S" f) files_list) in
let out = (File.chop_extension file)^".ml" in
if File.output out (FBuffer.contents buf)
then verbose (Printf.sprintf "write file \"%s\"" out)
else error (Printf.sprintf "cannot write file \"%s\" from \"%s\"" out (Sys.getcwd ()))
let _ =
(** please, keep the possibility of this option for other potential users of this generic application *)
let path = ref "" and empty = ref true in
Arg.parse
[ "-v", Arg.Set do_verbose, " verbose";
"-path", Arg.Set_string path, "<dir> prefix directory for files"
]
(fun t -> iter_of !path t; empty := false)
(Printf.sprintf "%s: embedded file ocaml preprocessor\nUsage: %s [options]\nOptions:" Sys.argv.(0) Sys.argv.(0));
if !empty
then error (Printf.sprintf "no input files (give me some \"%s\" file)" extension)