1 files changed, 40 insertions, 0 deletions
diff --git a/src/run/builtin.rs b/src/run/builtin.rs
index c946472..2255997 100644
--- a/src/run/builtin.rs
+++ b/src/run/builtin.rs
@@ -967,6 +967,46 @@ impl Builtin for export {
     }
 }
 
+#[derive(Copy, Clone)]
+pub struct pish_theme;
+
+impl Builtin for pish_theme {
+    fn name(&self) -> &str {
+        "pish_theme"
+    }
+
+    fn io(
+        &self,
+        session: Arc<Mutex<Session>>,
+        args: &[BString],
+        _stdin: &mut dyn Read,
+        stdout: &mut dyn Write,
+    ) -> Result {
+        if args.len() != 2 {
+            stdout.write_all(b"usage: pish_theme <kind> <color>\nwhere color is an ansi escape code,\nand where kind is one of the following: ")?;
+            for ident in crate::parse::HighlightKind::all_identifiers() {
+                stdout.write_all(&ident)?;
+                stdout.write_all(b" ")?;
+            }
+            stdout.write_all(b"\n")?;
+            return Err(Error::Exit(-1));
+        }
+
+        let mut se = session.lock().unwrap();
+        match se.highlighter.set_color(&args[0], &args[1]) {
+            Ok(_) => Ok(()),
+            Err(e) => match e {
+                syntax_highlighting::SetColorError::NoSuchKeyword => {
+                    stdout.write_all(b"no such kind: ")?;
+                    stdout.write_all(&args[0])?;
+                    stdout.write_all(b"\n")?;
+                    Err(Error::Exit(-1))
+                }
+            },
+        }
+    }
+}
+
 #[cfg(debug_assertions)]
 mod dbg {
     use super::*;