fix: fix replace error

The Regex was too greedy, resulting in the replacement of all content between openening and closing
comments when multiple comments were present. For now, only 1 (the first) match is replaced and the
regex is non-greedy so it matches the first end tag.

1 files changed, 3 insertions, 2 deletions

diff --git a/src/lib.rs b/src/lib.rs
index af9ac20..f691ea3 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -120,9 +120,10 @@ pub fn make_toc(
 ) -> Result<String, ::std::io::Error> {
     let content = read_file(file_path_in)?;
     let new_toc = generate_toc(content.to_owned(), min_depth, max_depth);
-    let re_toc = regex::Regex::new(r"(?ms)^(<!-- BEGIN mktoc).*(END mktoc -->)").unwrap();
+    let re_toc =
+        regex::Regex::new(r"(?ms)^(<!-- BEGIN mktoc -->)(.*?)(<!-- END mktoc -->)").unwrap();
     let res: String = re_toc
-        .replace_all(content.as_str(), new_toc.as_str())
+        .replacen(content.as_str(), 1, new_toc.as_str())
         .into_owned();
 
     Ok(res)