intermodal/src/input_target.rs

use crate::common::*;

#[derive(PartialEq, Debug, Clone)]
pub(crate) enum InputTarget {
  Path(PathBuf),
  Stdin,
}

impl InputTarget {
  pub(crate) fn resolve(&self, env: &Env) -> Self {
    match self {
      Self::Path(path) => Self::Path(env.resolve(path)),
      Self::Stdin => Self::Stdin,
    }
  }
}

impl From<&OsStr> for InputTarget {
  fn from(text: &OsStr) -> Self {
    if text == OsStr::new("-") {
      Self::Stdin
    } else {
      Self::Path(text.into())
    }
  }
}

impl Display for InputTarget {
  fn fmt(&self, f: &mut Formatter) -> fmt::Result {
    match self {
      Self::Stdin => write!(f, "standard input"),
      Self::Path(path) => write!(f, "`{}`", path.display()),
    }
  }
}

impl<P: AsRef<Path>> PartialEq<P> for InputTarget {
  fn eq(&self, other: &P) -> bool {
    match self {
      Self::Path(path) => path == other.as_ref(),
      Self::Stdin => Path::new("-") == other.as_ref(),
    }
  }
}

#[cfg(test)]
mod tests {
  use super::*;

  #[test]
  fn file() {
    assert_eq!(
      InputTarget::from(OsStr::new("foo")),
      InputTarget::Path("foo".into())
    );
  }

  #[test]
  fn stdio() {
    assert_eq!(InputTarget::from(OsStr::new("-")), InputTarget::Stdin);
  }

  #[test]
  fn display_file() {
    let path = PathBuf::from("./path");
    let have = InputTarget::Path(path).to_string();
    let want = "`./path`";
    assert_eq!(have, want);
  }

  #[test]
  fn display_stdio() {
    let have = InputTarget::Stdin.to_string();
    let want = "standard input";
    assert_eq!(have, want);
  }
}