use std::env;
use std::env::VarError;
use std::ffi::{OsStr, OsString};
use std::path::PathBuf;

pub trait Environment {
  fn args_os(&self) -> Vec<OsString>;
  fn var<K>(&self, key: K) -> Result<String, VarError>
  where
    K: AsRef<OsStr>;
  fn set_var<K, V>(&self, key: K, value: V)
  where
    K: AsRef<OsStr>,
    V: AsRef<OsStr>;
  fn get_home_directory(&self) -> Option<PathBuf>;
}

#[derive(Debug, Default)]
pub struct Prod;

impl Environment for Prod {
  fn args_os(&self) -> Vec<OsString> {
    env::args_os().collect()
  }

  fn var<K>(&self, key: K) -> Result<String, VarError>
  where
    K: AsRef<OsStr>,
  {
    env::var(key)
  }

  fn set_var<K, V>(&self, key: K, value: V)
    where
      K: AsRef<OsStr>,
      V: AsRef<OsStr>
  {
    env::set_var(key, value);
  }

  fn get_home_directory(&self) -> Option<PathBuf> {
    homedir::my_home().ok().flatten()
  }
}