1 files changed, 81 insertions, 0 deletions
diff --git a/crates/turtle/src/command/client/setup.rs b/crates/turtle/src/command/client/setup.rs
new file mode 100644
index 00000000..b32ceb97
--- /dev/null
+++ b/crates/turtle/src/command/client/setup.rs
@@ -0,0 +1,81 @@
+use crate::atuin_client::settings::Settings;
+
+use colored::Colorize;
+use eyre::Result;
+use std::io::{self, Write};
+use toml_edit::{DocumentMut, value};
+
+pub async fn run(_settings: &Settings) -> Result<()> {
+    let enable_ai = prompt(
+        "Atuin AI",
+        "This will enable command generation and other AI features via the question mark key",
+        Some(
+            "By default, Atuin AI only has access to the name and version of your operating system and shell - your shell history is not sent to the AI.",
+        ),
+    )?;
+
+    let enable_daemon = prompt(
+        "Atuin Daemon",
+        "This will enable improved search and history sync using a persistent background process",
+        None,
+    )?;
+
+    let config_file = Settings::get_config_path()?;
+    let config_str = tokio::fs::read_to_string(&config_file).await?;
+    let mut doc = config_str.parse::<DocumentMut>()?;
+
+    let mut changed = false;
+    if enable_ai {
+        changed = true;
+        if !doc.contains_key("ai") {
+            doc["ai"] = toml_edit::table();
+        }
+        doc["ai"]["enabled"] = value(true);
+    }
+
+    if enable_daemon {
+        changed = true;
+        if !doc.contains_key("daemon") {
+            doc["daemon"] = toml_edit::table();
+        }
+        doc["daemon"]["enabled"] = value(true);
+        doc["daemon"]["autostart"] = value(true);
+        doc["search_mode"] = value("daemon-fuzzy");
+    }
+
+    if changed {
+        tokio::fs::write(config_file, doc.to_string()).await?;
+
+        println!(
+            "{check} Settings updated successfully",
+            check = "✓".bold().bright_green()
+        );
+    } else {
+        println!(
+            "{check} No settings changed",
+            check = "✓".bold().bright_green()
+        );
+    }
+
+    Ok(())
+}
+
+pub fn prompt(feature: &str, description: &str, note: Option<&str>) -> Result<bool> {
+    println!(
+        "> Enable {feature}?",
+        feature = feature.bold().bright_blue()
+    );
+    if let Some(note) = note {
+        println!("  {description}");
+        print!("  {note} {q} ", q = "[Y/n]".bold());
+    } else {
+        print!("  {description} {q} ", q = "[Y/n]".bold());
+    }
+
+    io::stdout().flush().ok();
+
+    let mut input = String::new();
+    io::stdin().read_line(&mut input)?;
+    let answer = input.trim().to_lowercase();
+    Ok(answer.is_empty() || answer == "y" || answer == "yes")
+}