use crate::common::*;

pub(crate) struct TestEnvBuilder {
  args: Vec<String>,
  out_is_term: bool,
  use_color: bool,
}

impl TestEnvBuilder {
  pub(crate) fn new() -> TestEnvBuilder {
    TestEnvBuilder {
      args: Vec::new(),
      out_is_term: false,
      use_color: false,
    }
  }

  pub(crate) fn out_is_term(mut self) -> Self {
    self.out_is_term = true;
    self
  }

  pub(crate) fn arg(mut self, arg: impl Into<String>) -> Self {
    self.args.push(arg.into());
    self
  }

  pub(crate) fn args(mut self, args: impl IntoIterator<Item = impl Into<String>>) -> Self {
    for arg in args {
      self.args.push(arg.into());
    }
    self
  }

  pub(crate) fn arg_slice(mut self, args: &[&str]) -> Self {
    for arg in args.iter().cloned() {
      self.args.push(arg.to_owned());
    }
    self
  }

  pub(crate) fn build(self) -> TestEnv {
    let err = Capture::new();
    let out = Capture::new();

    let env = Env::new(
      tempfile::tempdir().unwrap(),
      out.clone(),
      if self.use_color && self.out_is_term {
        Style::active()
      } else {
        Style::inactive()
      },
      self.out_is_term,
      err.clone(),
      Style::inactive(),
      self.args,
    );

    TestEnv::new(env, err, out)
  }
}