allow highlighting to be disabled, once again

1 files changed, 10 insertions, 9 deletions

diff --git a/src/line/mod.rs b/src/line/mod.rs
index 510c480..6e24f14 100644
--- a/src/line/mod.rs
+++ b/src/line/mod.rs
@@ -41,19 +41,20 @@ impl Line {
         }
         self.dirty = false;
 
-        // TODO this is currently needed to reprint the prompt upon modification so we can't skip it.
-        // if !h.enabled {
-        //     return Ok(());
-        // }
+        let buf = self.into_bytes();
 
-        use crate::parse::{self, Parse};
+        let highlights = if h.enabled {
+            use crate::parse::{self, Parse};
+            let mut parser = parse::Cursor::new(&buf, parse::ParseMode::Completion);
+            let _ = parse::Ast::parse(&mut parser);
+            parser.highlights
+        } else {
+            Vec::new()
+        };
 
-        let buf = self.into_bytes();
-        let mut parser = parse::Cursor::new(&buf, parse::ParseMode::Completion);
-        let _ = parse::Ast::parse(&mut parser);
         let mut stdout = io::stdout().lock();
         self.line_start.f_go_to(&mut stdout)?;
-        h.pretty_print(&buf, self.left_len(), parser.highlights, &mut stdout)?;
+        h.pretty_print(&buf, self.left_len(), highlights, &mut stdout)?;
         stdout.flush()?;
 
         Ok(())