Add independent flags to C or C++ files

Signed-off-by: Varphone Wong <varphone@qq.com>

Author: varphone

src/lib.rs

@@ -265,6 +265,8 @@ pub struct Build {
     known_flag_support_status: Arc<Mutex<HashMap<String, bool>>>,
     ar_flags: Vec<Arc<str>>,
     asm_flags: Vec<Arc<str>>,
+    c_flags: Vec<Arc<str>>,
+    cpp_flags: Vec<Arc<str>>,
     no_default_flags: bool,
     files: Vec<Arc<Path>>,
     cpp: bool,
@@ -388,6 +390,8 @@ impl Build {
             known_flag_support_status: Arc::new(Mutex::new(HashMap::new())),
             ar_flags: Vec::new(),
             asm_flags: Vec::new(),
+            c_flags: Vec::new(),
+            cpp_flags: Vec::new(),
             no_default_flags: false,
             files: Vec::new(),
             shared_flag: None,
@@ -564,6 +568,36 @@ impl Build {
         self
     }
 
+    /// Add an arbitrary flag to the invocation of the compiler for c files
+    ///
+    /// # Example
+    ///
+    /// ```no_run
+    /// cc::Build::new()
+    ///     .file("src/foo.c")
+    ///     .c_flag("-std=c99")
+    ///     .compile("foo");
+    /// ```
+    pub fn c_flag(&mut self, flag: &str) -> &mut Build {
+        self.c_flags.push(flag.into());
+        self
+    }
+
+    /// Add an arbitrary flag to the invocation of the compiler for cpp files
+    ///
+    /// # Example
+    ///
+    /// ```no_run
+    /// cc::Build::new()
+    ///     .file("src/foo.cpp")
+    ///     .cpp_flag("-std=c++17")
+    ///     .compile("foo");
+    /// ```
+    pub fn cpp_flag(&mut self, flag: &str) -> &mut Build {
+        self.cpp_flags.push(flag.into());
+        self
+    }
+
     fn ensure_check_file(&self) -> Result<PathBuf, Error> {
         let out_dir = self.get_out_dir()?;
         let src = if self.cuda {
@@ -1790,6 +1824,14 @@ impl Build {
             cmd.push_cc_arg(warnings_to_errors_flag);
         }
 
+        for flag in self.c_flags.iter() {
+            cmd.c_args.push((**flag).into());
+        }
+
+        for flag in self.cpp_flags.iter() {
+            cmd.cpp_args.push((**flag).into());
+        }
+
         Ok(cmd)
     }
 

src/tool.rs

@@ -29,6 +29,8 @@ pub struct Tool {
     pub(crate) cc_wrapper_path: Option<PathBuf>,
     pub(crate) cc_wrapper_args: Vec<OsString>,
     pub(crate) args: Vec<OsString>,
+    pub(crate) c_args: Vec<OsString>,
+    pub(crate) cpp_args: Vec<OsString>,
     pub(crate) env: Vec<(OsString, OsString)>,
     pub(crate) family: ToolFamily,
     pub(crate) cuda: bool,
@@ -77,6 +79,8 @@ impl Tool {
             cc_wrapper_path: None,
             cc_wrapper_args: Vec::new(),
             args: Vec::new(),
+            c_args: Vec::new(),
+            cpp_args: Vec::new(),
             env: Vec::new(),
             family,
             cuda: false,
@@ -193,6 +197,8 @@ impl Tool {
             cc_wrapper_path: None,
             cc_wrapper_args: Vec::new(),
             args: Vec::new(),
+            c_args: Vec::new(),
+            cpp_args: Vec::new(),
             env: Vec::new(),
             family,
             cuda,