From dedadcfd30516c40692fe495a6ad10aea7c050de Mon Sep 17 00:00:00 2001
From: Rory Dudley
Date: Mon, 20 May 2024 23:54:01 -0600
Subject: Added the 'which' built-in command

Added a built-in which command (for MacOS and BSD), which can check
aliases, and other shell built-in commands, in addition to bins on the
$PATH.
---
 src/poem/anthology/which.rs | 61 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 src/poem/anthology/which.rs

(limited to 'src/poem/anthology')

diff --git a/src/poem/anthology/which.rs b/src/poem/anthology/which.rs
new file mode 100644
index 0000000..515b52e
--- /dev/null
+++ b/src/poem/anthology/which.rs
@@ -0,0 +1,61 @@
+use crate::compose::Environment;
+use crate::poem::Verse;
+
+pub fn incant(verse: &Verse, out: &mut Vec<u8>, env: &Environment) -> i32 {
+    let mut status = 0;
+    match verse.clause() {
+        Some(clause) => {
+            let mut output: String;
+            for word in clause {
+                // Check if it's an alias
+                if env.aliases.contains_key(&word) {
+                    output = format!("{}: aliased to {}\n", word, env.aliases.get(&word).unwrap());
+                    if verse.couplet > 0 {
+                        out.append(&mut output.as_bytes().to_vec());
+                    } else {
+                        print!("{}", output);
+                    }
+                    continue;
+                }
+
+                // Check if it's a built-in
+                match super::lookup(&word) {
+                    Some(_) => {
+                        output = format!("{}: shell built-in command\n", word);
+                        if verse.couplet > 0 {
+                            out.append(&mut output.as_bytes().to_vec());
+                        } else {
+                            print!("{}", output);
+                        }
+                        continue;
+                    }
+                    None => {}
+                }
+
+                // Manually check the path
+                let mut verb = Verse::new();
+                verb.push(word.clone());
+                match verb.spellcheck(&env.bins) {
+                    Some(i) => {
+                        output = format!("{}\n", env.bins[i]);
+                        if verse.couplet > 0 {
+                            out.append(&mut output.as_bytes().to_vec());
+                        } else {
+                            print!("{}", output);
+                        }
+                    }
+                    None => {
+                        output = format!("{} not found\n", word);
+                        status = 1;
+                        eprint!("{}", output);
+                    }
+                }
+            }
+        }
+        None => {
+            eprintln!("which: not enough arguments");
+            return 1;
+        }
+    }
+    status
+}
-- 
cgit v1.2.3