Improve code blocks

Cargo.lock

@@ -218,7 +218,7 @@ dependencies = [
 
 [[package]]
 name = "md-docs"
-version = "0.1.0"
+version = "0.1.2"
 dependencies = [
  "clap",
  "markdown",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "md-docs"
-version = "0.1.0"
+version = "0.1.2"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

src/generator/code.rs

@@ -15,9 +15,9 @@ impl Printable for Code {
                 return generate_nav_html(&self.value);
             }
 
-            format!("<pre class=\"language-{}\">{}</pre>", lang, code)
+            format!("<pre><code class=\"language-{}\">{}</code></pre>", lang, code)
         } else {
-            format!("<pre class=\"language-none\">{}</pre>", code)
+            format!("<pre><code>{}</code></pre>", code)
         }
     }
 

src/main.rs

@@ -9,6 +9,14 @@ mod sidebar;
 mod utils;
 
 fn main() {
+    // get args
+    let args: Vec<String> = std::env::args().collect();
+    // if first arg is --version, print version and exit
+    if args.len() > 1 && args[1] == "--version" {
+        println!("md-docs {}", env!("CARGO_PKG_VERSION"));
+        return;
+    }
+
     let config_file = Path::new(CONFIG_NAME);
     if !config_file.is_file() {
         eprintln!("A {} file was not found. Aborting.", CONFIG_NAME);