html decode the input before highlighting

1 files changed, 3 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index d96cc46..0904b55 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -148,6 +148,8 @@ fn get_config<'a>(
 
 fn main() {
     if let [_, lang, code] = args().collect::<Vec<String>>().as_slice() {
+        // decode code as it may be html encoded
+        let code = html_escape::decode_html_entities(code);
         let mut highlighter = Highlighter::new();
         let configs = RefCell::new(HashMap::new());
 
@@ -174,6 +176,6 @@ fn main() {
         }).unwrap();
         print!("{}", String::from_utf8(renderer.html).unwrap())
     } else {
-        panic!("Need <lang> <code> as arguments")
+        panic!("Need <lang> <code> as arguments, got {:?}", args().collect::<Vec<String>>())
     }
 }