Move parseHeader up in the chain - parseDocument can either parse

headers or a paragraph

src/MdToHTML.hs

@@ -59,9 +59,6 @@ instance Show MdToken where
 
 -- ---------------
 -- Helpers
-mustBeHash :: ReadP Char
-mustBeHash = satisfy (\x -> x == '#')
-
 leftmostLongest :: (Foldable t) => [(a, t b)] -> (a, t b)
 leftmostLongest xs =
   let lastElem = (last xs)
@@ -115,8 +112,8 @@ parseItalic :: ReadP MdToken
 parseItalic = do
   text <-
     choice
-      [ (between (string "_") (string "_") (munch1 (/= '_'))),
-        (between (string "*") (string "*") (munch1 (/= '*')))
+      [ between (string "_") (string "_") (munch1 (/= '_')),
+        between (string "*") (string "*") (munch1 (/= '*'))
       ]
   let parsedText = fst $ leftmostLongestParse parseLine text
   return (Italic parsedText)
@@ -143,7 +140,6 @@ parseString = do
 
 lineParsers :: [ReadP MdToken]
 lineParsers = [parseLinebreak, parseSingleNewline, parseBold, parseItalic, parseString] -- A 'line' doesn't include a 'header'
---lineParsers = [parseSingleNewline, parseString]
 
 -- List of all parsers
 allParsers :: [ReadP MdToken]
@@ -158,9 +154,7 @@ parseLine :: ReadP MdToken
 parseLine = do
   skipSpaces
   -- Fail if we have reached the end of the document.
-  remaining <- look
-  when (null remaining) pfail
-  parsed <- parseMany parseLineToken
+  parsed <- manyTill parseLineToken eof
   return (Line parsed)
 
 -- Parse a paragraph, which is a 'Line' (can span multiple actual lines), separated by double-newlines.
@@ -172,7 +166,7 @@ parsePara = do
   --  string "\n\n" <|> (eof >> return "") -- Consume the next double-newline or EOF.
   text <- (manyTill get ((string "\n\n") <|> (eof >> return "")))
   when (null text) pfail
-  let parsedText = fst $ leftmostLongestParse (parseHeader <|> parseLine) text -- Parse either a line or a header.
+  let parsedText = fst $ leftmostLongestParse parseLine text -- Parse either a line or a header.
   -- If the paragraph is a header, return a Header token. Otheriwse return a Para token.
   case parsedText of
     Header level token -> return (Header level token)
@@ -180,4 +174,6 @@ parsePara = do
 
 -- Parse a document, which is multiple paragraphs.
 parseDocument :: ReadP MdToken
-parseDocument = (many1 parsePara) >>= (\res -> return (Document (res)))
+parseDocument = do
+  res <- manyTill (parseHeader <++ parsePara) (eof)
+  return (Document res)