feat: override codeblock when `.override` class is provided

1 files changed, 13 insertions, 13 deletions

diff --git a/src/main.rs b/src/main.rs
index 3890ae9..2deced8 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -92,19 +92,19 @@ fn main() -> Result<()> {
                         .find_map(|(k,v)| if k == "path" { Some(v.clone()) } else { None })
                         .unwrap_or(String::from(""));
                     key.push_str(&attr.0);
-                    /* Insert (or replace) block of code.
-                     * NOTE: we are always "borrowing" the snippets of code with `Cow`, i.e., it
-                     * will always be a `Cow::Borrowed` value, until we process the string
-                     * substituting macros. */
-                    //blocks.insert(key, Cow::from(code));
-                    blocks.entry(key)
-                          .and_modify(|s| {
-                              let mut concat = s.to_string();
-                              concat.push_str("\n");
-                              concat.push_str(code);
-                              *s = Cow::from(concat);
-                          })
-                          .or_insert(Cow::from(code));
+                    /* Insert (or replace) block of code. */
+                    if attr.1.iter().any(|c| c == "override") {
+                        blocks.insert(key, Cow::from(code));
+                    } else {
+                        blocks.entry(key)
+                              .and_modify(|s| {
+                                  let mut concat = s.to_string();
+                                  concat.push_str("\n");
+                                  concat.push_str(code);
+                                  *s = Cow::from(concat);
+                              })
+                              .or_insert(Cow::from(code));
+                    }
                 } else {
                     // println!("The following code has no ID:");
                     // code.lines().for_each(|l| println!("    {}", l));