bgit 0.4.2

User-friendly Git wrapper for beginners, automating essential tasks like adding, committing, and pushing changes. It includes smart rules to avoid common pitfalls, such as accidentally adding sensitive files or directories and has exclusive support for portable hooks!
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

use crate::config::global::BGitGlobalConfig;
use crate::config::local::{StepFlags, WorkflowRules};
use crate::events::git_add::{AddMode, GitAdd};
use crate::events::{AtomicEvent, git_status};
use crate::rules::Rule;
use crate::rules::a12_no_secrets_staged::NoSecretsStaged;
use crate::rules::a12b_no_secret_files_staged::NoSecretFilesStaged;
use crate::rules::a16_no_large_file::NoLargeFile;
use crate::step::ActionStep;
use crate::step::Task::ActionStepTask;
use crate::workflows::default::action::ta07_has_uncommitted::HasUncommitted;
use crate::{
    bgit_error::{BGitError, BGitErrorWorkflowType, NO_EVENT, NO_RULE},
    step::{PromptStep, Step},
};

use dialoguer::{MultiSelect, Select, theme::ColorfulTheme};
pub(crate) struct AskAddMode {
    name: String,
}

impl PromptStep for AskAddMode {
    fn new() -> Self
    where
        Self: Sized,
    {
        AskAddMode {
            name: "ask_add_mode".to_owned(),
        }
    }

    fn get_name(&self) -> &str {
        &self.name
    }

    fn execute(
        &self,
        _step_config_flags: Option<&StepFlags>,
        workflow_rules_config: Option<&WorkflowRules>,
        global_config: &BGitGlobalConfig,
    ) -> Result<Step, Box<BGitError>> {
        let options = vec!["Add all unstaged files", "Select specific files to add"];
        let selection = Select::with_theme(&ColorfulTheme::default())
            .with_prompt("Choose add mode:")
            .default(0)
            .items(&options)
            .interact()
            .map_err(|e| {
                Box::new(BGitError::new(
                    "BGitError",
                    &format!("Failed to get user selection: {e}"),
                    BGitErrorWorkflowType::AtomicEvent,
                    NO_EVENT,
                    &self.name,
                    NO_RULE,
                ))
            })?;

        let add_mode = match selection {
            0 => AddMode::All,
            1 => {
                // Handle selective mode with file selection prompt
                let selected_files = self.prompt_file_selection(global_config)?;
                if selected_files.is_empty() {
                    println!("No files selected.");
                    return Ok(Step::Stop);
                }
                AddMode::Selective(selected_files)
            }
            _ => AddMode::All,
        };

        // Create GitAdd instance with the selected mode and execute
        let mut git_add = GitAdd::new(global_config).with_add_mode(add_mode);

        git_add.add_pre_check_rule(Box::new(NoSecretsStaged::new(workflow_rules_config)));
        git_add.add_pre_check_rule(Box::new(NoSecretFilesStaged::new(workflow_rules_config)));
        git_add.add_pre_check_rule(Box::new(NoLargeFile::new(workflow_rules_config)));

        git_add.execute()?;

        Ok(Step::Task(ActionStepTask(Box::new(HasUncommitted::new()))))
    }
}

impl AskAddMode {
    fn prompt_file_selection(
        &self,
        global_config: &BGitGlobalConfig,
    ) -> Result<Vec<String>, Box<BGitError>> {
        let unstaged_files = git_status::GitStatus::new(global_config).get_unstaged_files_list()?;

        if unstaged_files.is_empty() {
            println!("No unstaged files found.");
            return Ok(vec![]);
        }

        let file_display: Vec<String> = unstaged_files
            .iter()
            .map(|file| format!("{} ({})", file.path, file.status_type))
            .collect();

        if file_display.is_empty() {
            println!("No files to select.");
            return Ok(vec![]);
        }

        let selections = MultiSelect::with_theme(&ColorfulTheme::default())
            .with_prompt("Select files to add (use Space to select, Enter to confirm):")
            .items(&file_display)
            .interact()
            .map_err(|e| {
                Box::new(BGitError::new(
                    "BGitError",
                    &format!("Failed to get file selections: {e}"),
                    BGitErrorWorkflowType::AtomicEvent,
                    NO_EVENT,
                    &self.name,
                    NO_RULE,
                ))
            })?;

        if selections.is_empty() {
            return Ok(vec![]);
        }

        let selected_files: Vec<String> = selections
            .iter()
            .map(|&i| unstaged_files[i].path.clone())
            .collect();

        Ok(selected_files)
    }
}