Retrieve and parse --print=check-cfg in target info

Author: Urgau

src/cargo/core/compiler/build_context/target_info.rs

@@ -14,7 +14,7 @@ use crate::util::context::{GlobalContext, StringList, TargetConfig};
 use crate::util::interning::InternedString;
 use crate::util::{CargoResult, Rustc};
 use anyhow::Context as _;
-use cargo_platform::{Cfg, CfgExpr};
+use cargo_platform::{Cfg, CfgExpr, CheckCfg};
 use cargo_util::{ProcessBuilder, paths};
 use serde::{Deserialize, Serialize};
 use std::cell::RefCell;
@@ -43,6 +43,8 @@ pub struct TargetInfo {
     crate_types: RefCell<HashMap<CrateType, Option<(String, String)>>>,
     /// `cfg` information extracted from `rustc --print=cfg`.
     cfg: Vec<Cfg>,
+    /// `check-cfg` informations extracted from `rustc --print=check-cfg`.
+    check_cfg: Option<CheckCfg>,
     /// `supports_std` information extracted from `rustc --print=target-spec-json`
     pub supports_std: Option<bool>,
     /// Supported values for `-Csplit-debuginfo=` flag, queried from rustc
@@ -208,6 +210,14 @@ impl TargetInfo {
             process.arg("--print=crate-name"); // `___` as a delimiter.
             process.arg("--print=cfg");
 
+            if gctx.cli_unstable().check_target_cfgs {
+                process.arg("--print=crate-name"); // `___` as a delimiter.
+                process.arg("--print=check-cfg");
+
+                process.arg("--check-cfg=cfg()"); // otherwise `--print=check-cfg` won't output
+                process.arg("-Zunstable-options"); // required by `--print=check-cfg`
+            }
+
             // parse_crate_type() relies on "unsupported/unknown crate type" error message,
             // so make warnings always emitted as warnings.
             process.arg("-Wwarnings");
@@ -261,16 +271,42 @@ impl TargetInfo {
                 res
             };
 
-            let cfg = lines
-                .map(|line| Ok(Cfg::from_str(line)?))
-                .filter(TargetInfo::not_user_specific_cfg)
-                .collect::<CargoResult<Vec<_>>>()
-                .with_context(|| {
-                    format!(
-                        "failed to parse the cfg from `rustc --print=cfg`, got:\n{}",
-                        output
-                    )
-                })?;
+            let cfg = {
+                let mut res = Vec::new();
+                for line in &mut lines {
+                    // HACK: abuse `--print=crate-name` to use `___` as a delimiter.
+                    if line == "___" {
+                        break;
+                    }
+
+                    let cfg = Cfg::from_str(line).with_context(|| {
+                        format!(
+                            "failed to parse the cfg from `rustc --print=cfg`, got:\n{}",
+                            output
+                        )
+                    })?;
+                    if TargetInfo::not_user_specific_cfg(&cfg) {
+                        res.push(cfg);
+                    }
+                }
+                res
+            };
+
+            let check_cfg = if gctx.cli_unstable().check_target_cfgs {
+                let mut check_cfg = CheckCfg::default();
+                check_cfg.exhaustive = true;
+
+                for line in lines {
+                    check_cfg
+                        .parse_print_check_cfg_line(line)
+                        .with_context(|| {
+                            format!("unable to parse a line from `--print=check-cfg`")
+                        })?;
+                }
+                Some(check_cfg)
+            } else {
+                None
+            };
 
             // recalculate `rustflags` from above now that we have `cfg`
             // information
@@ -356,14 +392,15 @@ impl TargetInfo {
                 )?
                 .into(),
                 cfg,
+                check_cfg,
                 supports_std,
                 support_split_debuginfo,
             });
         }
     }
 
-    fn not_user_specific_cfg(cfg: &CargoResult<Cfg>) -> bool {
-        if let Ok(Cfg::Name(cfg_name)) = cfg {
+    fn not_user_specific_cfg(cfg: &Cfg) -> bool {
+        if let Cfg::Name(cfg_name) = cfg {
             // This should also include "debug_assertions", but it causes
             // regressions. Maybe some day in the distant future it can be
             // added (and possibly change the warning to an error).
@@ -379,6 +416,11 @@ impl TargetInfo {
         &self.cfg
     }
 
+    /// The [`CheckCfg`] settings.
+    pub fn check_cfg(&self) -> &Option<CheckCfg> {
+        &self.check_cfg
+    }
+
     /// Returns the list of file types generated by the given crate type.
     ///
     /// Returns `None` if the target does not support the given crate type.