git_su/
switcher.rs

1// Switcher: request switch, print current, list, add, clear, edit
2
3use crate::config_repository::ConfigRepository;
4use crate::git::{Git, Scope};
5use crate::user::User;
6use crate::user_list::UserList;
7use std::path::Path;
8
9pub struct Switcher<'a> {
10    user_list: &'a UserList,
11}
12
13impl<'a> Switcher<'a> {
14    pub fn new(user_list: &'a UserList) -> Self {
15        Switcher { user_list }
16    }
17
18    pub fn request(&self, scope: Scope, user_strings: &[String], output: &mut impl std::io::Write) {
19        let (parsed, not_parsed): (Vec<User>, Vec<String>) = user_strings
20            .iter()
21            .map(|s| {
22                User::parse(s)
23                    .map(|u| (Some(u), None))
24                    .unwrap_or_else(|_| (None, Some(s.clone())))
25            })
26            .fold(
27                (vec![], vec![]),
28                |(mut parsed, mut not_parsed), (p, n)| {
29                    if let Some(u) = p {
30                        parsed.push(u);
31                    }
32                    if let Some(s) = n {
33                        not_parsed.push(s);
34                    }
35                    (parsed, not_parsed)
36                },
37            );
38
39        let found = match self.user_list.find(&not_parsed) {
40            Ok(users) => users,
41            Err(e) => {
42                let _ = writeln!(output, "{}", e);
43                return;
44            }
45        };
46
47        let group_email = ConfigRepository::group_email_address();
48        let all_users: Vec<User> = parsed
49            .iter()
50            .chain(found.iter())
51            .cloned()
52            .collect();
53        let combined = all_users
54            .iter()
55            .fold(User::none(), |acc, u| acc.combine(u, &group_email));
56
57        let actual_scope = if scope == Scope::Default || scope == Scope::Derived {
58            ConfigRepository::default_select_scope()
59        } else {
60            scope
61        };
62
63        if Git::select_user(&combined, actual_scope) {
64            let _ = writeln!(
65                output,
66                "Switched {} user to {}",
67                actual_scope.display_name(),
68                Git::render(&combined)
69            );
70        }
71
72        for u in &parsed {
73            if !u.is_none() && !self.user_list.list().iter().any(|l| l == u) {
74                self.user_list.add(u);
75            }
76        }
77    }
78
79    pub fn print_current(&self, scopes: &[Scope], output: &mut impl std::io::Write) {
80        const LABEL_WIDTH: usize = 6; // Local, Global, System
81        let show_all = scopes.is_empty();
82        if show_all {
83            let _ = writeln!(output, "Current user: {}", Git::render(&Git::selected_user(Scope::Derived)));
84            let _ = writeln!(output);
85            let _ = writeln!(output, "{:>1$}: {2}", "Local", LABEL_WIDTH, Git::render(&Git::selected_user(Scope::Local)));
86            let _ = writeln!(output, "{:>1$}: {2}", "Global", LABEL_WIDTH, Git::render(&Git::selected_user(Scope::Global)));
87            let _ = writeln!(output, "{:>1$}: {2}", "System", LABEL_WIDTH, Git::render(&Git::selected_user(Scope::System)));
88        } else {
89            for scope in scopes {
90                let label = scope.display_name();
91                let w = label.len().max(LABEL_WIDTH);
92                let u = Git::selected_user(*scope);
93                if !u.is_none() {
94                    let _ = writeln!(output, "{:<1$}: {2}", label, w, Git::render(&u));
95                }
96            }
97        }
98    }
99
100    pub fn clear(&self, scopes: &[Scope], output: &mut impl std::io::Write) {
101        let scopes: Vec<Scope> = if scopes.is_empty() {
102            vec![Scope::Local, Scope::Global, Scope::System]
103        } else {
104            scopes.to_vec()
105        };
106        let scope_names: Vec<&str> = scopes.iter().map(|s| s.as_flag()).collect();
107        let _ = writeln!(
108            output,
109            "Clearing Git user in {} scope(s)",
110            scope_names.join(", ")
111        );
112        for scope in scopes {
113            Git::clear_user(scope);
114        }
115    }
116
117    pub fn list(&self, output: &mut impl std::io::Write) {
118        for u in self.user_list.list() {
119            let _ = writeln!(output, "{}", Git::render(&u));
120        }
121    }
122
123    pub fn add(&self, user_string: &str, output: &mut impl std::io::Write) {
124        let user = match User::parse(user_string) {
125            Ok(u) => u,
126            Err(e) => {
127                let _ = writeln!(output, "{}", e);
128                return;
129            }
130        };
131        if self.user_list.list().iter().any(|u| u == &user) {
132            let _ = writeln!(
133                output,
134                "User '{}' already in user list (try switching to them with 'git su {}')",
135                user,
136                user.initials()
137            );
138        } else {
139            self.user_list.add(&user);
140            let _ = writeln!(output, "User '{}' added to users", user);
141        }
142    }
143
144    pub fn edit_config(&self, path: &Path) -> bool {
145        Git::edit_gitsu_config(path)
146    }
147}
git_su/switcher.rs

git_su/
switcher.rs
