src/tools/rust-analyzer/crates/project-model/src/cfg_flag.rs

   1 //! Parsing of CfgFlags as command line arguments, as in
   2 //!
   3 //! rustc main.rs --cfg foo --cfg 'feature="bar"'
   4 use std::{fmt, str::FromStr};
   5
   6 use cfg::CfgOptions;
   7
   8 #[derive(Clone, Eq, PartialEq, Debug)]
   9 pub enum CfgFlag {
  10     Atom(String),
  11     KeyValue { key: String, value: String },
  12 }
  13
  14 impl FromStr for CfgFlag {
  15     type Err = String;
  16     fn from_str(s: &str) -> Result<Self, Self::Err> {
  17         let res = match s.split_once('=') {
  18             Some((key, value)) => {
  19                 if !(value.starts_with('"') && value.ends_with('"')) {
  20                     return Err(format!("Invalid cfg ({:?}), value should be in quotes", s));
  21                 }
  22                 let key = key.to_string();
  23                 let value = value[1..value.len() - 1].to_string();
  24                 CfgFlag::KeyValue { key, value }
  25             }
  26             None => CfgFlag::Atom(s.into()),
  27         };
  28         Ok(res)
  29     }
  30 }
  31
  32 impl<'de> serde::Deserialize<'de> for CfgFlag {
  33     fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
  34     where
  35         D: serde::Deserializer<'de>,
  36     {
  37         String::deserialize(deserializer)?.parse().map_err(serde::de::Error::custom)
  38     }
  39 }
  40
  41 impl Extend<CfgFlag> for CfgOptions {
  42     fn extend<T: IntoIterator<Item = CfgFlag>>(&mut self, iter: T) {
  43         for cfg_flag in iter {
  44             match cfg_flag {
  45                 CfgFlag::Atom(it) => self.insert_atom(it.into()),
  46                 CfgFlag::KeyValue { key, value } => self.insert_key_value(key.into(), value.into()),
  47             }
  48         }
  49     }
  50 }
  51
  52 impl fmt::Display for CfgFlag {
  53     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
  54         match self {
  55             CfgFlag::Atom(atom) => f.write_str(atom),
  56             CfgFlag::KeyValue { key, value } => {
  57                 f.write_str(key)?;
  58                 f.write_str("=")?;
  59                 f.write_str(value)
  60             }
  61         }
  62     }
  63 }