compiler/rustc_driver/src/args.rs

   1 use std::error;
   2 use std::fmt;
   3 use std::fs;
   4 use std::io;
   5
   6 fn arg_expand(arg: String) -> Result<Vec<String>, Error> {
   7     if let Some(path) = arg.strip_prefix('@') {
   8         let file = match fs::read_to_string(path) {
   9             Ok(file) => file,
  10             Err(ref err) if err.kind() == io::ErrorKind::InvalidData => {
  11                 return Err(Error::Utf8Error(Some(path.to_string())));
  12             }
  13             Err(err) => return Err(Error::IOError(path.to_string(), err)),
  14         };
  15         Ok(file.lines().map(ToString::to_string).collect())
  16     } else {
  17         Ok(vec![arg])
  18     }
  19 }
  20
  21 pub fn arg_expand_all(at_args: &[String]) -> Vec<String> {
  22     let mut args = Vec::new();
  23     for arg in at_args {
  24         match arg_expand(arg.clone()) {
  25             Ok(arg) => args.extend(arg),
  26             Err(err) => rustc_session::early_error(
  27                 rustc_session::config::ErrorOutputType::default(),
  28                 &format!("Failed to load argument file: {err}"),
  29             ),
  30         }
  31     }
  32     args
  33 }
  34
  35 #[derive(Debug)]
  36 pub enum Error {
  37     Utf8Error(Option<String>),
  38     IOError(String, io::Error),
  39 }
  40
  41 impl fmt::Display for Error {
  42     fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
  43         match self {
  44             Error::Utf8Error(None) => write!(fmt, "Utf8 error"),
  45             Error::Utf8Error(Some(path)) => write!(fmt, "Utf8 error in {path}"),
  46             Error::IOError(path, err) => write!(fmt, "IO Error: {path}: {err}"),
  47         }
  48     }
  49 }
  50
  51 impl error::Error for Error {}