2 * Copyright (C) 2014-2015 Red Hat Inc.
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
26 try Sys.getenv "MCLU_PATH"
27 with Not_found -> Config.pkgdatadir // "templates"
30 let files = Sys.readdir template_dir in
31 let files = Array.to_list files in
32 let files = List.map ((//) template_dir) files in
34 List.filter (fun name -> Filename.check_suffix name ".template") files in
37 let name = Filename.basename filename in
38 let name = Filename.chop_suffix name ".template" in
42 let template_names () = List.map fst (templates ())
44 let run_template ~verbose filename subcmd args =
46 sprintf "%s %s %s" (quote filename) (quote subcmd)
47 (String.concat " " (List.map quote args)) in
48 if verbose then printf "%s\n%!" cmd;
49 let chan = Unix.open_process_in cmd in
51 (try while true do lines := input_line chan :: !lines done
52 with End_of_file -> ());
53 let lines = List.rev !lines in
54 let stat = Unix.close_process_in chan in
56 | Unix.WEXITED 0 -> Some lines
57 | Unix.WEXITED 2 -> None
59 eprintf "mclu: template '%s' subcmd '%s' exited with error %d\n"
63 eprintf "mclu: template '%s' subcmd '%s' killed by signal %d\n"
67 eprintf "mclu: template '%s' subcmd '%s' stopped by signal %d\n"
72 type template_info = {
74 minimum_memory : int64 option;
75 recommended_memory : int64 option;
76 disk_bus : string option;
77 network_model : string option;
80 let probe ?(verbose = false) filename =
81 (* Check the template is a template. *)
82 (match run_template ~verbose filename "probe" [] with
83 | Some ["hello"] -> ()
85 eprintf "mclu: file %s is not an mclu template\n" filename;
89 (* Probe for various properties. *)
91 match run_template ~verbose filename "base-image" [] with
92 | Some [answer] -> answer
94 eprintf "mclu: cannot parse '%s base-image'\n" filename;
97 match run_template ~verbose filename "minimum-memory" [] with
99 (try Some (bytes_of_human_size memory)
101 eprintf "mclu: cannot parse output of '%s minimum-memory'\n"
106 let recommended_memory =
107 match run_template ~verbose filename "recommended-memory" [] with
109 (try Some (bytes_of_human_size memory)
111 eprintf "mclu: cannot parse output of '%s recommended-memory'\n"
117 match run_template ~verbose filename "disk-bus" [] with
118 | Some [answer] -> Some answer
121 match run_template ~verbose filename "network-model" [] with
122 | Some [answer] -> Some answer
125 { base_image = base_image;
126 minimum_memory = minimum_memory;
127 recommended_memory = recommended_memory;
129 network_model = network_model }