summaryrefslogtreecommitdiffstats
path: root/src/poem/elements
diff options
context:
space:
mode:
Diffstat (limited to 'src/poem/elements')
-rw-r--r--src/poem/elements/verse.rs31
1 files changed, 23 insertions, 8 deletions
diff --git a/src/poem/elements/verse.rs b/src/poem/elements/verse.rs
index 42d7c6a..b4f41ba 100644
--- a/src/poem/elements/verse.rs
+++ b/src/poem/elements/verse.rs
@@ -1,6 +1,7 @@
use super::rune::Rune;
use super::stanza::Stanza;
use super::word::Word;
+use crate::poem::anthology;
use crate::poem::anthology::Anthology;
use crate::poem::Poem;
mod logic;
@@ -32,6 +33,13 @@ pub struct Verse {
pub meter: Rune,
}
+#[derive(Debug, Clone)]
+pub enum Spelling {
+ FullPath,
+ OnPath(usize),
+ BuiltIn(usize),
+}
+
pub trait Forkable<T> {
fn fork(&mut self, env: &mut Environment) -> Result<T, io::Error>;
}
@@ -216,7 +224,7 @@ impl Verse {
/// stanza_success.spellcheck(bins) // -> Some(i)
/// stanza_fail.spellcheck(bins) // -> None
/// ```
- pub fn spellcheck(&self, bins: &Vec<String>) -> Option<usize> {
+ pub fn spellcheck(&self, bins: &Vec<String>) -> Option<Spelling> {
// An empty verb (i.e. the empty string) cannot be a program, so
// return false
// Thanks to the parsing in Poem::read, however, it's
@@ -225,6 +233,13 @@ impl Verse {
return None;
}
+ // Check for built-ins
+ let i = anthology::lookup(self.verb().as_str());
+ match i {
+ Some(i) => return Some(Spelling::BuiltIn(i)),
+ None => {}
+ };
+
// Only search the $PATH if a full or relative path was not given, or
// if the path given does not exist
if !Path::new(self.verb().as_str()).exists() {
@@ -232,7 +247,7 @@ impl Verse {
// Searches in $PATH order
for (i, path) in bins.iter().enumerate() {
if path.split('/').last().unwrap() == self.verb() {
- return Some(i);
+ return Some(Spelling::OnPath(i));
}
}
@@ -241,7 +256,7 @@ impl Verse {
}
// Return the length of bins if the full path or relative path exists
- Some(bins.len())
+ Some(Spelling::FullPath)
}
/// Run a command
@@ -256,7 +271,7 @@ impl Verse {
out: &mut Vec<u8>,
pids: &mut Arc<Mutex<Vec<i32>>>,
env: &mut Environment,
- anthology: Option<usize>,
+ spell: Option<Spelling>,
) -> Result<i32, io::Error> {
// Read files into 'out' if Rune::Read is present in the verse's IO
if self.io.contains(&Rune::Read) {
@@ -274,12 +289,12 @@ impl Verse {
}
// Build and run the command
- match anthology {
- Some(_) => {
- let mut command = Anthology::new(self.verb());
+ match spell {
+ Some(Spelling::BuiltIn(i)) => {
+ let mut command = Anthology::new(i);
incant!(self.verb(), command, out, pids, env, self)
}
- None => {
+ _ => {
let mut command = Command::new(self.verb());
incant!(self.verb(), command, out, pids, env, self)
}