More flexible "start of code block" in Markdown mode.

This should fix #595.
GaloisInc · May 14, 2019 · f607b40 · f607b40
1 parent 843fe31
commit f607b40
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/Cryptol/Parser/Unlit.hs b/src/Cryptol/Parser/Unlit.hs
@@ -99,12 +99,17 @@ markdown = blanks []
     | otherwise         = fenced op (l : current) ls
 
 
-  isOpenFence l  | "```cryptol" == l'         = Just Code
-                 | "```"        == l'         = Just Code
-                 | "```" `Text.isPrefixOf` l' = Just Comment
-                 | otherwise                  = Nothing
+  isOpenFence l
+    | "```" `Text.isPrefixOf` l' =
+      Just $ case Text.drop 3 l' of
+               l'' | "cryptol" `Text.isPrefixOf` l'' -> Code
+                   | isBlank l''                     -> Code
+                   | otherwise                       -> Comment
+
+    | otherwise = Nothing
     where
     l' = Text.dropWhile isSpace l
+
   isCloseFence l = "```" `Text.isPrefixOf` l
   isBlank l      = Text.all isSpace l
   isCodeLine l   = "\t" `Text.isPrefixOf` l || "    " `Text.isPrefixOf` l