raftcli 1.14.1

Command line interface for raft framework and serial monitoring
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

// RaftCLI: Command History Module
// Rob Dobson 2024

use std::fs::{File, OpenOptions};
use std::io::{BufRead, BufReader, Write};
use std::path::Path;

pub struct CommandHistory {
    history: Vec<String>,
    position: usize,
    history_file_path: String,
}

impl CommandHistory {
    pub fn new(history_file_path: &str) -> CommandHistory {
        let mut history = Vec::new();

        // Load history from the file if it exists
        if Path::new(history_file_path).exists() {
            if let Ok(file) = File::open(history_file_path) {
                let reader = BufReader::new(file);
                for line in reader.lines() {
                    if let Ok(command) = line {
                        history.push(command);
                    }
                }
            }
        }

        let position = history.len();

        CommandHistory {
            history,
            position,
            history_file_path: history_file_path.to_string(),
        }
    }

    pub fn add_command(&mut self, command: &str) {
        if !command.is_empty() {
            // Avoid duplicate consecutive entries
            if self.history.is_empty() || self.history.last().unwrap() != command {
                self.history.push(command.to_string());
                self.position = self.history.len();

                // Append command to history file
                if let Ok(mut file) = OpenOptions::new()
                    .create(true)
                    .append(true)
                    .open(&self.history_file_path)
                {
                    writeln!(file, "{}", command).unwrap();
                }
            }
        }
    }

    pub fn move_up(&mut self) {
        if self.position > 0 {
            self.position -= 1;
        } 
    }

    pub fn move_down(&mut self) {
        if self.position < self.history.len() {
            self.position += 1;
        }
    }

    pub fn get_current(&self) -> String {
        if self.position < self.history.len() {
            self.history[self.position].clone()
        } else {
            "".to_string()
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::fs;

    #[test]
    fn test_command_history() {
        let test_history_path = "test_raftcli_history.txt";
        let _ = fs::remove_file(test_history_path);

        let mut command_history = CommandHistory::new(test_history_path);

        command_history.add_command("first command");
        command_history.add_command("second command");
        command_history.add_command("third command");

        command_history.move_up();
        assert_eq!(command_history.get_current(), "third command");
        command_history.move_up();
        assert_eq!(command_history.get_current(), "second command");
        command_history.move_up();
        assert_eq!(command_history.get_current(), "first command");
        command_history.move_up();
        assert_eq!(command_history.get_current(), "first command");

        command_history.move_down();
        assert_eq!(command_history.get_current(), "second command");
        command_history.move_down();
        assert_eq!(command_history.get_current(), "third command");
        command_history.move_down();
        assert_eq!(command_history.get_current(), "third command");
        command_history.move_down();

        // Cleanup
        let _ = fs::remove_file(test_history_path);
    }
}