Muse reader: simplify museToPandocTable

to get rid of partial functions (head, tail, maximum)

1 files changed, 4 insertions, 6 deletions

diff --git a/src/Text/Pandoc/Readers/Muse.hs b/src/Text/Pandoc/Readers/Muse.hs
index 88c78d450..818f66e20 100644
--- a/src/Text/Pandoc/Readers/Muse.hs
+++ b/src/Text/Pandoc/Readers/Muse.hs
@@ -45,7 +45,7 @@ import Control.Monad.Reader
 import Control.Monad.Except (throwError)
 import Data.Bifunctor
 import Data.Default
-import Data.List (intercalate, transpose)
+import Data.List (intercalate, transpose, uncons)
 import Data.List.Split (splitOn)
 import qualified Data.Map as M
 import qualified Data.Set as Set
@@ -640,11 +640,9 @@ data MuseTableElement = MuseHeaderRow [Blocks]
 
 museToPandocTable :: MuseTable -> Blocks
 museToPandocTable (MuseTable caption headers body footers) =
-  B.table caption attrs headRow rows
-  where ncol = maximum (0 : map length (headers ++ body ++ footers))
-        attrs = replicate ncol (AlignDefault, 0.0)
-        headRow = if null headers then [] else head headers
-        rows = (if null headers then [] else tail headers) ++ body ++ footers
+  B.table caption attrs headRow (rows ++ body ++ footers)
+  where attrs = const (AlignDefault, 0.0) <$> transpose (headers ++ body ++ footers)
+        (headRow, rows) = fromMaybe ([], []) $ uncons headers
 
 museAppendElement :: MuseTableElement
                   -> MuseTable