eliprompt 0.3.0

Elegant and informative shell prompt
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

// Copyright (C) 2020 Stephane Raux. Distributed under the 0BSD license.

use crate::Error;
use git2::Repository;
use once_cell::sync::OnceCell;
use std::{
    env,
    fmt::{self, Debug},
    path::{Path, PathBuf},
    time::Duration,
};

pub struct Environment {
    working_dir: PathBuf,
    prev_exit_code: i32,
    repo: OnceCell<Option<Repository>>,
    prev_cmd_duration: Option<Duration>,
    force_alternative_prompt: bool,
}

impl Environment {
    pub fn new<P: Into<PathBuf>>(working_dir: P) -> Result<Self, Error> {
        Ok(Environment {
            working_dir: working_dir.into(),
            prev_exit_code: 0,
            repo: OnceCell::new(),
            prev_cmd_duration: None,
            force_alternative_prompt: false,
        })
    }
    pub fn current() -> Result<Self, Error> {
        let working_dir = env::current_dir().map_err(Error::GettingPwdFailed)?;
        Self::new(working_dir)
    }

    pub fn with_prev_exit_code(self, code: i32) -> Self {
        Self {
            prev_exit_code: code,
            ..self
        }
    }

    pub fn with_prev_cmd_duration(self, d: Duration) -> Self {
        Self {
            prev_cmd_duration: Some(d),
            ..self
        }
    }

    pub fn force_alternative_prompt(self, yes: bool) -> Self {
        Self {
            force_alternative_prompt: yes,
            ..self
        }
    }

    pub fn alternative_prompt_is_used(&self) -> bool {
        if self.force_alternative_prompt {
            return true;
        }
        let alternative_requested = env::var("ELIPROMPT_ALTERNATIVE_PROMPT").is_ok();
        let terms_using_alternative = ["linux"];
        let term_uses_alternative =
            env::var("TERM").map_or(false, |term| terms_using_alternative.contains(&&*term));
        alternative_requested || term_uses_alternative
    }

    pub fn working_dir(&self) -> &Path {
        &self.working_dir
    }

    pub fn repo(&self) -> Option<&Repository> {
        let repo = self
            .repo
            .get_or_init(|| match Repository::discover(&self.working_dir) {
                Ok(repo) => Some(repo),
                Err(e) if e.code() == git2::ErrorCode::NotFound => None,
                Err(e) => {
                    tracing::error!("Failed to open git repository: {}", e);
                    None
                }
            });
        repo.as_ref()
    }

    pub fn prev_exit_code(&self) -> i32 {
        self.prev_exit_code
    }

    pub fn prev_cmd_duration(&self) -> Option<Duration> {
        self.prev_cmd_duration
    }
}

impl Debug for Environment {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("Environment")
            .field("working_dir", &self.working_dir)
            .field("prev_exit_code", &self.prev_exit_code)
            .field("prev_cmd_duration", &self.prev_cmd_duration)
            .finish()
    }
}