From 06cee7edd4c2ebf280f18c62f896e869ac23c299 Mon Sep 17 00:00:00 2001
From: Fabian Greimel <fabgrei@gmail.com>
Date: Sat, 29 Feb 2020 16:25:49 +0100
Subject: [PATCH] remove trailing lines as well

---
 src/Literate.jl | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/Literate.jl b/src/Literate.jl
index 2aa693a..9ced72d 100644
--- a/src/Literate.jl
+++ b/src/Literate.jl
@@ -146,8 +146,8 @@ function replace_default(content, sym;
     push!(repls, "\r\n" => "\n") # normalize line endings
 
     # unconditionally remove #src lines
-    push!(repls, r"^#src.*\n?"m => "")
-    push!(repls, r".*#src$\n?"m => "")
+    push!(repls, r"^#src.*\n?"m => "") # remove leadig #src lines
+    push!(repls, r".*#src$\n?"m => "") # remove trailing #src lines
 
     if sym === :md
         push!(repls, r"^#md "m => "")      # remove leading #md
@@ -155,17 +155,21 @@ function replace_default(content, sym;
         push!(repls, r"^#nb.*\n?"m => "")  # remove #nb lines
         push!(repls, r"^#!nb "m => "")     # remove leading #!nb
         push!(repls, r"^#jl.*\n?"m => "")  # remove leading #jl lines
+        push!(repls, r".*#jl$\n?"m => "")  # remove trailing #jl lines
         push!(repls, r"^#!jl "m => "")     # remove leading #!jl
     elseif sym === :nb
-        push!(repls, r"^#md.*\n?"m => "")  # remove #md lines
+        push!(repls, r"^#md.*\n?"m => "")  # remove leading #md lines
+        push!(repls, r".*#md$\n?"m => "")  # remove trailing #md lines
         push!(repls, r"^#!md "m => "")     # remove leading #!md
         push!(repls, r"^#nb "m => "")      # remove leading #nb
         push!(repls, r"^#!nb.*\n?"m => "") # remove #!nb lines
         push!(repls, r"^#jl.*\n?"m => "")  # remove leading #jl lines
+        push!(repls, r".*#jl$\n?"m => "")  # remove trailing #jl lines
         push!(repls, r"^#!jl "m => "")     # remove leading #!jl
         push!(repls, r"```math(.*?)```"s => s"$$\1$$")
     else # sym === :jl
-        push!(repls, r"^#md.*\n?"m => "")  # remove #md lines
+        push!(repls, r"^#md.*\n?"m => "")  # remove leading #md lines
+        push!(repls, r".*#md$\n?"m => "")  # remove trailing #md lines
         push!(repls, r"^#!md "m => "")     # remove leading #!md
         push!(repls, r"^#nb.*\n?"m => "")  # remove #nb lines
         push!(repls, r"^#!nb "m => "")     # remove leading #!nb