feature: automation of shell settings on alx init

Cargo.toml

@@ -25,6 +25,7 @@ anyhow = "1.0"
 thiserror = "2.0.17"
 dirs = "6.0.0"
 comfy-table = "7.1"
+dialoguer = "0.12.0"
 
 [dev-dependencies]
 tempfile = "3.8"

README.md

@@ -24,33 +24,41 @@ cargo install alx
 
 ## Quick Start
 
-### 1. Initialize
+### Initialize
 
 ```bash
 alx init
 ```
 
-### 2. Add to your shell config
+Choose your shell.
 
-First, run the `alx info` command, and then check the path for `Shell aliases`.
-
-Add the following line to your shell configuration file:
+```bash
+Initialized alx configuration at: "/your_alx_path/alx"
+Select your shell:
+> zsh (default)
+  bash
+  fis
+```
 
-**Bash** (`~/.bashrc`):
+Choose whether to write the 'alx' alias configuration to your shell file. The default is No.
 
 ```bash
-[ -f ~/your_alx_path/aliases.sh ] && source ~/your_alx_path/aliases.sh
+To enable aliases, add the following line to your shell config:
+     # Add to '/your/.zshrc'
+     [ -f '/your_alx_path/alx/shell/aliases.sh' ] && source '/your_alx_path/alx/shell/aliases.sh'
+Do you want to add this line to '/your/.zshrc' automatically? [y/N]
 ```
 
-**Zsh** (`~/.zshrc`):
+See below for manual setup.
 
 ```bash
-[ -f ~/your_alias_path/aliases.sh ] && source ~/your_alias_path/aliases.sh
-```
+# Bash (`~/.bashrc`):
+[ -f ~/your_alx_path/aliases.sh ] && source ~/your_alx_path/aliases.sh
 
-**Fish** (`~/.config/fish/config.fish`):
+# Zsh (`~/.zshrc`):
+[ -f ~/your_alias_path/aliases.sh ] && source ~/your_alias_path/aliases.sh
 
-```fish
+# Fish (`~/.config/fish/config.fish`):
 source ~/your_alias_path/aliases.sh
 ```
 

src/command.rs

@@ -11,6 +11,7 @@ use crate::shell::{ShellHandler, ShellType};
 use comfy_table::{
     Cell, ContentArrangement, Table, modifiers::UTF8_ROUND_CORNERS, presets::UTF8_BORDERS_ONLY,
 };
+use dialoguer::{Confirm, Select};
 use std::fs;
 
 fn sync_aliases() -> Result<()> {
@@ -49,32 +50,112 @@ pub fn init() -> Result<()> {
         "✓ Initialized alx configuration at: {:?}",
         config_manager.config_dir()
     );
-    println!("\nNext steps:");
-    println!("  1. Add aliases with: alx add <name> <command>");
-    println!("  2. Add the following line to your shell config file:");
 
-    let shell = ShellDetector::detect().ok();
-    if let Some(shell_type) = shell {
-        let handler: Box<dyn ShellHandler> = match shell_type {
-            ShellType::Bash => Box::new(BashHandler::new()),
-            ShellType::Zsh => Box::new(ZshHandler::new()),
-            ShellType::Fish => Box::new(FishHandler::new()),
-        };
+    // Detect default shell
+    let default_shell = ShellDetector::detect().ok();
+
+    // Create shell options with default shell at the top
+    let mut shell_options = vec![];
+    if let Some(default) = default_shell {
+        shell_options.push(format!("{} (default)", default.as_str()));
+    }
+
+    // Add other shells
+    let all_shells = vec![ShellType::Bash, ShellType::Zsh, ShellType::Fish];
+    for shell in all_shells {
+        if Some(shell) != default_shell {
+            shell_options.push(shell.as_str().to_string());
+        }
+    }
 
-        if let Ok(config_file) = handler.config_file_path() {
-            println!("\n     # Add to {:?}", config_file);
+    // Prompt user to select shell
+    let selection = Select::new()
+        .with_prompt("Select your shell")
+        .items(&shell_options)
+        .default(0)
+        .interact()
+        .map_err(|e| error::AlxError::ConfigError(format!("Failed to select shell: {}", e)))?;
+
+    // Parse selected shell
+    let selected_shell = if let (0, Some(default)) = (selection, default_shell) {
+        default
+    } else {
+        let shell_name = shell_options[selection].split_whitespace().next().unwrap();
+        match shell_name {
+            "bash" => ShellType::Bash,
+            "zsh" => ShellType::Zsh,
+            "fish" => ShellType::Fish,
+            _ => unreachable!(),
         }
+    };
+
+    let handler: Box<dyn ShellHandler> = match selected_shell {
+        ShellType::Bash => Box::new(BashHandler::new()),
+        ShellType::Zsh => Box::new(ZshHandler::new()),
+        ShellType::Fish => Box::new(FishHandler::new()),
+    };
+
+    let shell_aliases_file = config_manager.shell_aliases_file();
+    let aliases_path = shell_aliases_file.display();
+
+    let source_line = if selected_shell == ShellType::Fish {
+        format!("source '{}'", aliases_path)
+    } else {
+        format!("[ -f '{}' ] && source '{}'", aliases_path, aliases_path)
+    };
 
-        let shell_aliases_file = config_manager.shell_aliases_file();
-        let aliases_path = shell_aliases_file.display();
+    // Ask if user wants to add source line automatically
+    let config_file = handler.config_file_path()?;
+
+    println!("\nTo enable aliases, add the following line to your shell config:");
+    println!("     # Add to '{}'", config_file.display());
+    println!("     {}", source_line);
+
+    let should_add = Confirm::new()
+        .with_prompt(format!(
+            "Do you want to add this line to '{}' automatically?",
+            config_file.display()
+        ))
+        .default(false)
+        .interact()
+        .map_err(|e| error::AlxError::ConfigError(format!("Failed to confirm: {}", e)))?;
+
+    if should_add {
+        // Add source line to shell config
+        let mut file_content = if config_file.exists() {
+            fs::read_to_string(&config_file)?
+        } else {
+            String::new()
+        };
 
-        if shell_type == ShellType::Fish {
-            println!("     source {}", aliases_path);
+        // Check if the line already exists
+        if file_content.contains(&source_line) {
+            println!(
+                "✓ Source line already exists in '{}'",
+                config_file.display()
+            );
         } else {
-            println!("     [ -f {} ] && source {}", aliases_path, aliases_path);
+            // Add newline if file doesn't end with one
+            if !file_content.is_empty() && !file_content.ends_with('\n') {
+                file_content.push('\n');
+            }
+
+            // Add comment and source line
+            file_content.push_str("\n# alx - alias manager\n");
+            file_content.push_str(&source_line);
+            file_content.push('\n');
+
+            fs::write(&config_file, file_content)?;
+            println!("✓ Added source line to '{}'", config_file.display());
+            println!("\nPlease restart your shell or run:");
+            println!("     source '{}'", config_file.display());
         }
     }
 
+    println!("\nNext steps:");
+    println!("  1. Add aliases with: alx add <name> <command>");
+    println!("  2. Run 'alx list' to see your aliases");
+
     Ok(())
 }