From 195c2b1d7d84eec66788702d3bc124abaa76a56b Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Sun, 21 Jul 2019 21:55:42 +0800
Subject: [PATCH] add get_home(), get_absolute_path()

---
 src/lib.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 4d8b3d1..c5a2c78 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,7 +1,10 @@
 extern crate term;
 
 use std::{
+    env,
+    fs,
     io::{self, Write, Error, ErrorKind},
+    path::PathBuf,
     process::Command,
 };
 
@@ -34,6 +37,19 @@ pub fn is_macos_or_linux() -> bool {
     is_macos() || is_linux()
 }
 
+pub fn get_home() -> Option<String> {
+    env::var("HOME").ok()
+}
+
+pub fn get_absolute_path(path: &str) -> Option<PathBuf> {
+    if path == "~" {
+        return Some(PathBuf::from(get_home()?));
+    } else if path.starts_with("~/") {
+        return Some(PathBuf::from(&format!("{}/{}", get_home()?, &path[2..])));
+    }
+    fs::canonicalize(path).ok()
+}
+
 pub enum MessageType { INFO, OK, WARN, ERROR, }
 
 pub fn print_message_ex(color: Option<term::color::Color>, h: &str, message: &str) {