add host selection menu

src/main.rs

@@ -173,4 +173,31 @@ fn main() {
     grab_home();
 
     nix_gen();
+
+    // Host selection
+    loop {
+        println!("");
+        println!("Host selection");
+        println!("1. Device 1");
+        println!("2. Device 2");
+        println!("3. Device 3");
+        println!("");
+
+        println!("Enter your host for installation:");
+
+        let mut choice = String::new();
+        io::stdin().read_line(&mut choice).unwrap();
+
+        match choice.trim() {
+            "1" => println!("Hello, world!"),
+            "2" => {
+                println!("2 + 2 = {}", 2 + 2);
+            },
+            "3" => {
+                println!("Goodbye!");
+                break;
+                },
+            _ => println!("Invalid choice, try again."),
+        }
+    }
 }