Merge pull request #4 from Kristories/develop

Fix navigation and refactoring

Author: Kristories

README.md

@@ -1,12 +1,40 @@
 # MySSH
 
-Checking for existing SSH keys.
+MySSH is a simple tool to check for existing SSH keys in your `.ssh` directory and view their contents.
 
 ![Cover](art/cover.png)
 
 ## Installation
 
-Install via brew
+You can install MySSH using Homebrew:
+
+### Step 1: Add the formula to Homebrew
+
+If this is your first time installing from the `devtical` tap, you'll need to add it:
+
+```bash
+brew tap devtical/formulae
+```
+
+### Step 2: Install MySSH
+
+After adding the tap, install MySSH by running:
+
 ```bash
 brew install devtical/formulae/myssh
-```
+```
+
+## Usage
+
+Once installed, you can run MySSH directly from your terminal:
+
+```bash
+myssh
+```
+
+This will display a list of SSH keys available in your `.ssh` directory. Select any key to view its contents.
+
+## License
+
+This project is licensed under the Apache License 2.0. See the [LICENSE](LICENSE) file for more details.
+

src/main.rs

@@ -6,36 +6,49 @@ use std::fs;
 use std::path::Path;
 
 fn main() {
-    let home = std::env::var("HOME").unwrap();
-    let path = Path::new(&home).join(".ssh");
-    let mut arr = Vec::new();
+    let mut siv = cursive::default();
+    show_file_selection(&mut siv);
+    siv.run();
+}
 
-    for entry in fs::read_dir(path).expect("Unable to list") {
-        let entry = entry.expect("unable to get entry");
+fn show_file_selection(siv: &mut Cursive) {
+    let home = std::env::var("HOME").expect("HOME environment variable not set");
+    let ssh_path = Path::new(&home).join(".ssh");
+    let mut keys = Vec::new();
+
+    for entry in fs::read_dir(&ssh_path).expect("Unable to list directory") {
+        let entry = entry.expect("Unable to read entry");
         let path = entry.path();
-        let str = path.display().to_string();
+        let file_path = path.display().to_string();
 
-        if !str.contains("known_hosts") {
-            arr.push(str);
+        if !file_path.contains("known_hosts") {
+            keys.push(file_path);
         }
     }
 
     let mut select = SelectView::new().h_align(HAlign::Center).autojump();
-
-    select.add_all_str(arr);
+    select.add_all_str(keys);
     select.set_on_submit(show_next_window);
 
-    let mut siv = cursive::default();
-
-    siv.add_layer(Dialog::around(select.scrollable().fixed_size((40, 10))).title("My Keys"));
-    siv.run();
+    siv.add_layer(
+        Dialog::around(select.scrollable().fixed_size((40, 10)))
+            .title("My SSH Keys")
+            .button("Quit", |s| s.quit()),
+    );
 }
 
-fn show_next_window(siv: &mut Cursive, str: &str) {
+fn show_next_window(siv: &mut Cursive, file_path: &str) {
     siv.pop_layer();
 
-    let contents = fs::read_to_string(str).expect("Should have been able to read the file");
-    let text = contents;
-
-    siv.add_layer(Dialog::around(TextView::new(text)).button("Quit", |s| s.quit()));
+    let contents = fs::read_to_string(file_path).expect("Unable to read the file");
+    let text_view = TextView::new(contents);
+
+    siv.add_layer(
+        Dialog::around(text_view)
+            .button("Back", move |s| {
+                s.pop_layer();
+                show_file_selection(s);
+            })
+            .button("Quit", |s| s.quit()),
+    );
 }