From bbdf38018b47328b5faa2cef635c37095045be72 Mon Sep 17 00:00:00 2001 From: Benedikt Peetz Date: Fri, 12 Jun 2026 01:54:21 +0200 Subject: feat(server): Really make users stateless (with tests) This commit also remove another load of unneeded features. --- crates/turtle/src/atuin_common/shell.rs | 137 +------------------------------- 1 file changed, 3 insertions(+), 134 deletions(-) (limited to 'crates/turtle/src/atuin_common/shell.rs') diff --git a/crates/turtle/src/atuin_common/shell.rs b/crates/turtle/src/atuin_common/shell.rs index dbd9b982..d259b99e 100644 --- a/crates/turtle/src/atuin_common/shell.rs +++ b/crates/turtle/src/atuin_common/shell.rs @@ -1,9 +1,3 @@ -use std::{ffi::OsStr, path::Path, process::Command}; - -use serde::Serialize; -use sysinfo::{Process, System, get_current_pid}; -use thiserror::Error; - #[derive(PartialEq)] pub(crate) enum Shell { Sh, @@ -35,91 +29,15 @@ impl std::fmt::Display for Shell { } } -#[derive(Debug, Error, Serialize)] -pub(crate) enum ShellError { - #[error("shell not supported")] - NotSupported, - - #[error("failed to execute shell command: {0}")] - ExecError(String), -} - impl Shell { - pub(crate) fn current() -> Shell { - let sys = System::new_all(); - - let process = sys - .process(get_current_pid().expect("Failed to get current PID")) - .expect("Process with current pid does not exist"); - - let parent = sys - .process(process.parent().expect("Atuin running with no parent!")) - .expect("Process with parent pid does not exist"); - - let shell = parent.name().trim().to_lowercase(); - let shell = shell.strip_prefix('-').unwrap_or(&shell); - - Shell::from_string(shell.to_string()) - } - pub(crate) fn from_env() -> Shell { std::env::var("ATUIN_SHELL").map_or(Shell::Unknown, |shell| { - Shell::from_string(shell.trim().to_lowercase()) + Shell::from_string(shell.trim().to_lowercase().as_str()) }) } - pub(crate) fn config_file(&self) -> Option { - let mut path = if let Some(base) = directories::BaseDirs::new() { - base.home_dir().to_owned() - } else { - return None; - }; - - // TODO: handle all shells - match self { - Shell::Bash => path.push(".bashrc"), - Shell::Zsh => path.push(".zshrc"), - Shell::Fish => path.push(".config/fish/config.fish"), - - _ => return None, - }; - - Some(path) - } - - /// Best-effort attempt to determine the default shell - /// This implementation will be different across different platforms - /// Caller should ensure to handle Shell::Unknown correctly - pub(crate) fn default_shell() -> Result { - let sys = System::name().unwrap_or("".to_string()).to_lowercase(); - - // TODO: Support Linux - // I'm pretty sure we can use /etc/passwd there, though there will probably be some issues - let path = if sys.contains("darwin") { - // This works in my testing so far - Shell::Sh.run_interactive([ - "dscl localhost -read \"/Local/Default/Users/$USER\" shell | awk '{print $2}'", - ])? - } else if cfg!(windows) { - return Ok(Shell::Powershell); - } else { - Shell::Sh.run_interactive(["getent passwd $LOGNAME | cut -d: -f7"])? - }; - - let path = Path::new(path.trim()); - let shell = path.file_name(); - - if shell.is_none() { - return Err(ShellError::NotSupported); - } - - Ok(Shell::from_string( - shell.unwrap().to_string_lossy().to_string(), - )) - } - - pub(crate) fn from_string(name: String) -> Shell { - match name.as_str() { + pub(crate) fn from_string(name: &str) -> Shell { + match name { "bash" => Shell::Bash, "fish" => Shell::Fish, "zsh" => Shell::Zsh, @@ -131,53 +49,4 @@ impl Shell { _ => Shell::Unknown, } } - - /// Returns true if the shell is posix-like - /// Note that while fish is not posix compliant, it behaves well enough for our current - /// featureset that this does not matter. - pub(crate) fn is_posixish(&self) -> bool { - matches!(self, Shell::Bash | Shell::Fish | Shell::Zsh) - } - - pub(crate) fn run_interactive(&self, args: I) -> Result - where - I: IntoIterator, - S: AsRef, - { - let shell = self.to_string(); - let output = if self == &Self::Powershell { - Command::new(shell) - .args(args) - .output() - .map_err(|e| ShellError::ExecError(e.to_string()))? - } else { - Command::new(shell) - .arg("-ic") - .args(args) - .output() - .map_err(|e| ShellError::ExecError(e.to_string()))? - }; - - Ok(String::from_utf8(output.stdout).unwrap()) - } -} - -pub(crate) fn shell_name(parent: Option<&Process>) -> String { - let sys = System::new_all(); - - let parent = if let Some(parent) = parent { - parent - } else { - let process = sys - .process(get_current_pid().expect("Failed to get current PID")) - .expect("Process with current pid does not exist"); - - sys.process(process.parent().expect("Atuin running with no parent!")) - .expect("Process with parent pid does not exist") - }; - - let shell = parent.name().trim().to_lowercase(); - let shell = shell.strip_prefix('-').unwrap_or(&shell); - - shell.to_string() } -- cgit v1.3.1