From 2d473af0fe018ae48ebe54eac0b2e7ff52f6b4ca Mon Sep 17 00:00:00 2001
From: Emile <git@emile.space>
Date: Thu, 3 Feb 2022 21:55:10 +0100
Subject: improved the code rendering

---
 src/main.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 80f0f83..55dd485 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -407,8 +407,9 @@ fn write_readme_content(file: &mut File, in_path: &Path, raw_path: &Path)
             <h1><a href="#{a}">{b}</a></h1>
             <pre>"##, a = heading_sanitized, b = heading).as_bytes())?;
 
-        } else if line.starts_with(">") {
-            let line = line.get(1..).unwrap();
+        } else if line.starts_with("> ") {
+            let line = line.replace("<", "&lt");
+            let line = line.get(2..).unwrap();
             file.write_all(format!("</pre><pre class=\"code\">{}</pre><pre>\n", line).as_bytes())?;
 
         } else if line.starts_with(":::tree") {
-- 
cgit 1.4.1