From 7ced31c354bdfb2e256de3ecc49bcc4f379f78af Mon Sep 17 00:00:00 2001
From: Ellie Huxtable <ellie@elliehuxtable.com>
Date: Wed, 10 Apr 2024 13:01:48 +0100
Subject: feat(dotfiles): add alias import (#1938)

* feat(dotfiles): add alias import

* things

* clippy clappy
---
 atuin-common/src/lib.rs   |  1 +
 atuin-common/src/shell.rs | 66 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 67 insertions(+)
 create mode 100644 atuin-common/src/shell.rs

(limited to 'atuin-common/src')

diff --git a/atuin-common/src/lib.rs b/atuin-common/src/lib.rs
index d4513ee0..2d848f6f 100644
--- a/atuin-common/src/lib.rs
+++ b/atuin-common/src/lib.rs
@@ -54,4 +54,5 @@ macro_rules! new_uuid {
 
 pub mod api;
 pub mod record;
+pub mod shell;
 pub mod utils;
diff --git a/atuin-common/src/shell.rs b/atuin-common/src/shell.rs
new file mode 100644
index 00000000..2c0f3534
--- /dev/null
+++ b/atuin-common/src/shell.rs
@@ -0,0 +1,66 @@
+use sysinfo::{get_current_pid, Process, System};
+use thiserror::Error;
+
+pub enum Shell {
+    Sh,
+    Bash,
+    Fish,
+    Zsh,
+    Xonsh,
+    Nu,
+
+    Unknown,
+}
+
+#[derive(Debug, Error)]
+pub enum ShellError {
+    #[error("shell not supported")]
+    NotSupported,
+
+    #[error("failed to execute shell command: {0}")]
+    ExecError(String),
+}
+
+pub fn shell() -> Shell {
+    let name = shell_name(None);
+
+    match name.as_str() {
+        "bash" => Shell::Bash,
+        "fish" => Shell::Fish,
+        "zsh" => Shell::Zsh,
+        "xonsh" => Shell::Xonsh,
+        "nu" => Shell::Nu,
+        "sh" => Shell::Sh,
+
+        _ => Shell::Unknown,
+    }
+}
+
+impl Shell {
+    /// 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 fn is_posixish(&self) -> bool {
+        matches!(self, Shell::Bash | Shell::Fish | Shell::Zsh)
+    }
+}
+
+pub 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