From d61e0ff1e192c62085e842ba878cdb10bacd4c9e Mon Sep 17 00:00:00 2001
From: NerdyPepper <akshayoppiliappan@gmail.com>
Date: Mon, 20 May 2019 19:10:40 +0530
Subject: add pista and pista_minimal prompts

---
 src/main.rs | 33 ++++++++++++++++++++++++++++-----
 1 file changed, 28 insertions(+), 5 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index bfa47a0..e0b8f41 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,11 +3,34 @@ mod prompt_char;
 mod vcs;
 mod venv;
 
-use colored::*;
-
 fn main() {
-    print!("{}", cwd::cwd());
+    println!("{}", pista());
+}
+
+fn pista() -> String {
+    let cwd = cwd::cwd();
+    let (branch, status) = vcs::vcs_status().unwrap_or(("".into(), "".into()));
+    let venv = venv::get_name();
+    let prompt_char = prompt_char::get_char();
+    format!("{cwd} {branch} {status}\n{venv}{pchar} ",
+            cwd=cwd,
+            branch=branch,
+            status=status,
+            venv=venv,
+            pchar=prompt_char
+            )
+}
+
+fn pista_minimal() -> String {
+    let cwd = cwd::cwd();
     let (branch, status) = vcs::vcs_status().unwrap_or(("".into(), "".into()));
-    println!(" {} {}", branch, status.dimmed());
-    print!("{}{} ", venv::get_name(), prompt_char::prompt_char());
+    let venv = venv::get_name();
+    let prompt_char = prompt_char::get_char();
+    format!("{cwd} {branch}{status}{venv}{pchar}",
+            cwd=cwd,
+            branch=branch,
+            status=status,
+            venv=venv,
+            pchar=prompt_char
+            )
 }
-- 
cgit v1.2.3