zsh support

.gitignore

@@ -1,3 +1,5 @@
+.vscode/launch.json
+
 # Cargo build dir
 /target
 

src/config.rs

@@ -294,6 +294,7 @@ enum Shell {
     #[default]
     Bash,
     Python,
+    Zsh,
     #[value(skip)]
     Custom {
         program: String,
@@ -315,6 +316,7 @@ impl Display for Shell {
         match self {
             Self::Bash => f.write_str("bash"),
             Self::Python => f.write_str("python"),
+            Self::Zsh => f.write_str("zsh"),
             Self::Custom { program, args, .. } => f.write_str(
                 &iter::once(program)
                     .chain(args)
@@ -342,7 +344,7 @@ impl Merge for Shell {
 impl Shell {
     fn line_split(&self) -> &str {
         match self {
-            Self::Bash | Self::Python => " \\",
+            Self::Bash | Self::Python | Self::Zsh => " \\",
             Self::Custom { line_split, .. } => line_split,
         }
     }
@@ -351,6 +353,7 @@ impl Shell {
         match self {
             Self::Bash => "bash",
             Self::Python => "python",
+            Self::Zsh => "zsh",
             Self::Custom { program, .. } => program,
         }
     }
@@ -370,6 +373,7 @@ impl Shell {
         match self {
             Self::Bash => spawn::bash(timeout, environment, width, height),
             Self::Python => spawn::python(timeout, environment, width, height),
+            Self::Zsh => spawn::zsh(timeout, environment, width, height),
             Self::Custom {
                 program,
                 args,

src/config/de/shell.rs

@@ -12,6 +12,7 @@ use crate::config::Shell;
 enum Variant {
     Bash,
     Python,
+    Zsh,
     Custom,
 }
 
@@ -41,6 +42,7 @@ impl<'de> de::Visitor<'de> for Visitor {
         match v {
             "bash" | "Bash" => Ok(Shell::Bash),
             "python" | "Python" => Ok(Shell::Python),
+            "zsh" | "Zsh" => Ok(Shell::Zsh),
             _ => Err(E::invalid_value(
                 de::Unexpected::Str(v),
                 &"supported shell (e.g. bash or python) or a custom shell",
@@ -57,6 +59,7 @@ impl<'de> de::Visitor<'de> for Visitor {
         match tag {
             Variant::Bash => variant.unit_variant().map(|_| Shell::Bash),
             Variant::Python => variant.unit_variant().map(|_| Shell::Python),
+            Variant::Zsh => variant.unit_variant().map(|_| Shell::Zsh),
             Variant::Custom => variant.struct_variant(CUSTOM_FIELDS, CustomVisitor),
         }
     }
@@ -109,6 +112,7 @@ mod tests {
     fn visit_str() -> serde_yaml::Result<()> {
         assert_eq!(serde_yaml::from_str::<Shell>("bash")?, Shell::Bash);
         assert_eq!(serde_yaml::from_str::<Shell>("python")?, Shell::Python);
+        assert_eq!(serde_yaml::from_str::<Shell>("zsh")?, Shell::Zsh);
         assert!(serde_yaml::from_str::<Shell>("custom").is_err());
         Ok(())
     }

src/config/spawn.rs

@@ -1,7 +1,7 @@
 use std::{
     ffi::OsStr,
     io::{self, BufRead, BufReader, Read, Write},
-    ops::{Deref, DerefMut},
+    ops::DerefMut,
     process::Command,
     time::{Duration, Instant},
 };
@@ -19,6 +19,38 @@ use os_str_bytes::OsStrBytes;
 
 use crate::asciicast::Event;
 
+pub(super) fn zsh<I, K, V>(
+    timeout: Duration,
+    environment: I,
+    width: u16,
+    height: u16,
+) -> color_eyre::Result<ShellSession>
+where
+    I: IntoIterator<Item = (K, V)>,
+    K: AsRef<OsStr>,
+    V: AsRef<OsStr>,
+{
+    const PROMPT: &str = "AUTOCAST_PROMPT";
+    const PROMPT_COMMAND: &str =
+        "PS1=AUTOCAST_PROMPT; unset PROMPT_COMMAND; unset zle_bracketed_paste";
+
+    let mut command = Command::new("zsh");
+    command.arg("--no-rcs");
+    command
+        .envs(environment)
+        .env("PS1", PROMPT)
+        .env("PROMPT_COMMAND", PROMPT_COMMAND);
+
+    ShellSession::spawn(
+        command,
+        width,
+        height,
+        String::from(PROMPT),
+        Some(String::from("exit")),
+        timeout,
+    )
+}
+
 pub(super) fn bash<I, K, V>(
     timeout: Duration,
     environment: I,
@@ -256,7 +288,15 @@ pub trait Wait: Process {
 #[cfg(unix)]
 impl Wait for UnixProcess {
     fn wait(&self, _: Duration) -> color_eyre::Result<()> {
-        self.deref().wait().map(|_| ()).map_err(Into::into)
+        #[cfg(target_os = "macos")]
+        {
+            Ok(())
+        }
+
+        #[cfg(target_os = "linux")]
+        {
+            self.deref().wait().map(|_| ()).map_err(Into::into)
+        }
     }
 }