blob: e6a325dea7f3a8955d75edec549c135783d7b330 [file] [log] [blame]
Mårten Kongstad867a3492023-04-25 15:06:30 +02001/*
2 * Copyright (C) 2023 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17//! `aconfig` is a build time tool to manage build time configurations, such as feature flags.
18
Mårten Kongstad403658f2023-06-14 09:51:56 +020019use anyhow::{anyhow, bail, ensure, Result};
Mårten Kongstad98b0eeb2023-05-08 11:25:30 +020020use clap::{builder::ArgAction, builder::EnumValueParser, Arg, ArgMatches, Command};
Mårten Kongstad6b9e3822023-05-16 11:19:58 +020021use core::any::Any;
Mårten Kongstad4d2b4b02023-04-27 16:05:58 +020022use std::fs;
Mårten Kongstada1029092023-05-08 11:51:59 +020023use std::io;
24use std::io::Write;
Mårten Kongstadd42eeeb2023-05-12 10:01:00 +020025use std::path::{Path, PathBuf};
Mårten Kongstad867a3492023-04-25 15:06:30 +020026
Mårten Kongstad00cf0452023-05-26 16:48:01 +020027mod codegen;
Dennis Shen1dc9ad42023-05-12 00:21:55 +000028mod codegen_cpp;
Zhi Doueb744892023-05-10 04:02:33 +000029mod codegen_java;
Mårten Kongstadf73b9632023-05-24 15:43:47 +020030mod codegen_rust;
Mårten Kongstad4d2b4b02023-04-27 16:05:58 +020031mod commands;
Mårten Kongstadfe753f52023-04-26 09:13:03 +020032mod protos;
Mårten Kongstadfe753f52023-04-26 09:13:03 +020033
Mårten Kongstad83a87602023-06-02 11:20:15 +020034#[cfg(test)]
35mod test;
36
Mårten Kongstad403658f2023-06-14 09:51:56 +020037use commands::{DumpFormat, Input, OutputFile};
Mårten Kongstad4d2b4b02023-04-27 16:05:58 +020038
39fn cli() -> Command {
40 Command::new("aconfig")
41 .subcommand_required(true)
42 .subcommand(
43 Command::new("create-cache")
Mårten Kongstad9fb58962023-05-31 13:02:13 +020044 .arg(Arg::new("package").long("package").required(true))
Mårten Kongstadfa23d292023-05-11 14:47:02 +020045 .arg(Arg::new("declarations").long("declarations").action(ArgAction::Append))
46 .arg(Arg::new("values").long("values").action(ArgAction::Append))
Mårten Kongstad4d2b4b02023-04-27 16:05:58 +020047 .arg(Arg::new("cache").long("cache").required(true)),
48 )
49 .subcommand(
Zhi Doueb744892023-05-10 04:02:33 +000050 Command::new("create-java-lib")
51 .arg(Arg::new("cache").long("cache").required(true))
52 .arg(Arg::new("out").long("out").required(true)),
53 )
54 .subcommand(
Dennis Shen1dc9ad42023-05-12 00:21:55 +000055 Command::new("create-cpp-lib")
56 .arg(Arg::new("cache").long("cache").required(true))
57 .arg(Arg::new("out").long("out").required(true)),
58 )
59 .subcommand(
Mårten Kongstadf73b9632023-05-24 15:43:47 +020060 Command::new("create-rust-lib")
61 .arg(Arg::new("cache").long("cache").required(true))
62 .arg(Arg::new("out").long("out").required(true)),
63 )
64 .subcommand(
Mårten Kongstadf02734e2023-06-02 11:34:24 +020065 Command::new("create-device-config-defaults")
66 .arg(Arg::new("cache").long("cache").action(ArgAction::Append).required(true))
67 .arg(Arg::new("out").long("out").default_value("-")),
68 )
69 .subcommand(
Mårten Kongstadc31a6ff2023-06-02 11:54:36 +020070 Command::new("create-device-config-sysprops")
71 .arg(Arg::new("cache").long("cache").action(ArgAction::Append).required(true))
72 .arg(Arg::new("out").long("out").default_value("-")),
73 )
74 .subcommand(
Mårten Kongstada1029092023-05-08 11:51:59 +020075 Command::new("dump")
Mårten Kongstadaf677032023-05-17 16:18:25 +020076 .arg(Arg::new("cache").long("cache").action(ArgAction::Append).required(true))
Mårten Kongstada1029092023-05-08 11:51:59 +020077 .arg(
78 Arg::new("format")
79 .long("format")
Mårten Kongstadba94e6a2023-05-16 11:00:16 +020080 .value_parser(EnumValueParser::<commands::DumpFormat>::new())
Mårten Kongstada1029092023-05-08 11:51:59 +020081 .default_value("text"),
82 )
83 .arg(Arg::new("out").long("out").default_value("-")),
Mårten Kongstad4d2b4b02023-04-27 16:05:58 +020084 )
85}
Mårten Kongstad867a3492023-04-25 15:06:30 +020086
Mårten Kongstad6b9e3822023-05-16 11:19:58 +020087fn get_required_arg<'a, T>(matches: &'a ArgMatches, arg_name: &str) -> Result<&'a T>
88where
89 T: Any + Clone + Send + Sync + 'static,
90{
91 matches
92 .get_one::<T>(arg_name)
93 .ok_or(anyhow!("internal error: required argument '{}' not found", arg_name))
94}
95
Mårten Kongstad98b0eeb2023-05-08 11:25:30 +020096fn open_zero_or_more_files(matches: &ArgMatches, arg_name: &str) -> Result<Vec<Input>> {
97 let mut opened_files = vec![];
98 for path in matches.get_many::<String>(arg_name).unwrap_or_default() {
99 let file = Box::new(fs::File::open(path)?);
Mårten Kongstad403658f2023-06-14 09:51:56 +0200100 opened_files.push(Input { source: path.to_string(), reader: file });
Mårten Kongstad98b0eeb2023-05-08 11:25:30 +0200101 }
102 Ok(opened_files)
103}
104
Mårten Kongstad403658f2023-06-14 09:51:56 +0200105fn open_single_file(matches: &ArgMatches, arg_name: &str) -> Result<Input> {
106 let Some(path) = matches.get_one::<String>(arg_name) else {
107 bail!("missing argument {}", arg_name);
108 };
109 let file = Box::new(fs::File::open(path)?);
110 Ok(Input { source: path.to_string(), reader: file })
111}
112
Mårten Kongstadd42eeeb2023-05-12 10:01:00 +0200113fn write_output_file_realtive_to_dir(root: &Path, output_file: &OutputFile) -> Result<()> {
114 ensure!(
115 root.is_dir(),
116 "output directory {} does not exist or is not a directory",
117 root.display()
118 );
119 let path = root.join(output_file.path.clone());
120 let parent = path
121 .parent()
122 .ok_or(anyhow!("unable to locate parent of output file {}", path.display()))?;
123 fs::create_dir_all(parent)?;
124 let mut file = fs::File::create(path)?;
125 file.write_all(&output_file.contents)?;
126 Ok(())
127}
128
Mårten Kongstadf02734e2023-06-02 11:34:24 +0200129fn write_output_to_file_or_stdout(path: &str, data: &[u8]) -> Result<()> {
130 if path == "-" {
131 io::stdout().write_all(data)?;
132 } else {
133 fs::File::create(path)?.write_all(data)?;
134 }
135 Ok(())
136}
137
Mårten Kongstadbb520722023-04-26 13:16:41 +0200138fn main() -> Result<()> {
Mårten Kongstad4d2b4b02023-04-27 16:05:58 +0200139 let matches = cli().get_matches();
140 match matches.subcommand() {
141 Some(("create-cache", sub_matches)) => {
Mårten Kongstad9fb58962023-05-31 13:02:13 +0200142 let package = get_required_arg::<String>(sub_matches, "package")?;
Mårten Kongstadfa23d292023-05-11 14:47:02 +0200143 let declarations = open_zero_or_more_files(sub_matches, "declarations")?;
144 let values = open_zero_or_more_files(sub_matches, "values")?;
Mårten Kongstad403658f2023-06-14 09:51:56 +0200145 let output = commands::parse_flags(package, declarations, values)?;
Mårten Kongstad6b9e3822023-05-16 11:19:58 +0200146 let path = get_required_arg::<String>(sub_matches, "cache")?;
Mårten Kongstad403658f2023-06-14 09:51:56 +0200147 write_output_to_file_or_stdout(path, &output)?;
Mårten Kongstad4d2b4b02023-04-27 16:05:58 +0200148 }
Zhi Doueb744892023-05-10 04:02:33 +0000149 Some(("create-java-lib", sub_matches)) => {
Mårten Kongstad403658f2023-06-14 09:51:56 +0200150 let cache = open_single_file(sub_matches, "cache")?;
Zhi Dou4655c962023-06-12 15:56:03 +0000151 let generated_files = commands::create_java_lib(cache)?;
Mårten Kongstad403658f2023-06-14 09:51:56 +0200152 let dir = PathBuf::from(get_required_arg::<String>(sub_matches, "out")?);
Zhi Dou4655c962023-06-12 15:56:03 +0000153 generated_files
154 .iter()
155 .try_for_each(|file| write_output_file_realtive_to_dir(&dir, file))?;
Dennis Shen1dc9ad42023-05-12 00:21:55 +0000156 }
157 Some(("create-cpp-lib", sub_matches)) => {
Mårten Kongstad403658f2023-06-14 09:51:56 +0200158 let cache = open_single_file(sub_matches, "cache")?;
Mårten Kongstadb27f2ce2023-06-02 11:43:21 +0200159 let generated_file = commands::create_cpp_lib(cache)?;
Mårten Kongstad403658f2023-06-14 09:51:56 +0200160 let dir = PathBuf::from(get_required_arg::<String>(sub_matches, "out")?);
Mårten Kongstadd42eeeb2023-05-12 10:01:00 +0200161 write_output_file_realtive_to_dir(&dir, &generated_file)?;
Zhi Doueb744892023-05-10 04:02:33 +0000162 }
Mårten Kongstadf73b9632023-05-24 15:43:47 +0200163 Some(("create-rust-lib", sub_matches)) => {
Mårten Kongstad403658f2023-06-14 09:51:56 +0200164 let cache = open_single_file(sub_matches, "cache")?;
Mårten Kongstadb27f2ce2023-06-02 11:43:21 +0200165 let generated_file = commands::create_rust_lib(cache)?;
Mårten Kongstad403658f2023-06-14 09:51:56 +0200166 let dir = PathBuf::from(get_required_arg::<String>(sub_matches, "out")?);
Mårten Kongstadf73b9632023-05-24 15:43:47 +0200167 write_output_file_realtive_to_dir(&dir, &generated_file)?;
168 }
Mårten Kongstadf02734e2023-06-02 11:34:24 +0200169 Some(("create-device-config-defaults", sub_matches)) => {
Mårten Kongstad403658f2023-06-14 09:51:56 +0200170 let cache = open_single_file(sub_matches, "cache")?;
171 let output = commands::create_device_config_defaults(cache)?;
Mårten Kongstadf02734e2023-06-02 11:34:24 +0200172 let path = get_required_arg::<String>(sub_matches, "out")?;
173 write_output_to_file_or_stdout(path, &output)?;
174 }
Mårten Kongstadc31a6ff2023-06-02 11:54:36 +0200175 Some(("create-device-config-sysprops", sub_matches)) => {
Mårten Kongstad403658f2023-06-14 09:51:56 +0200176 let cache = open_single_file(sub_matches, "cache")?;
177 let output = commands::create_device_config_sysprops(cache)?;
Mårten Kongstadc31a6ff2023-06-02 11:54:36 +0200178 let path = get_required_arg::<String>(sub_matches, "out")?;
179 write_output_to_file_or_stdout(path, &output)?;
180 }
Mårten Kongstad4d2b4b02023-04-27 16:05:58 +0200181 Some(("dump", sub_matches)) => {
Mårten Kongstad403658f2023-06-14 09:51:56 +0200182 let input = open_zero_or_more_files(sub_matches, "cache")?;
Mårten Kongstad6b9e3822023-05-16 11:19:58 +0200183 let format = get_required_arg::<DumpFormat>(sub_matches, "format")?;
Mårten Kongstad403658f2023-06-14 09:51:56 +0200184 let output = commands::dump_parsed_flags(input, *format)?;
Mårten Kongstad6b9e3822023-05-16 11:19:58 +0200185 let path = get_required_arg::<String>(sub_matches, "out")?;
Mårten Kongstadf02734e2023-06-02 11:34:24 +0200186 write_output_to_file_or_stdout(path, &output)?;
Mårten Kongstad4d2b4b02023-04-27 16:05:58 +0200187 }
188 _ => unreachable!(),
189 }
Mårten Kongstadbb520722023-04-26 13:16:41 +0200190 Ok(())
Mårten Kongstad867a3492023-04-25 15:06:30 +0200191}