Merge pull request purescript#1117 from purescript/write-multiple-doc-files

garyb · garyb · commit ddae724ede8b · 2015-05-13T20:31:04.000+01:00
Allow psc-docs to write multiple modules to a file
diff --git a/psc-docs/Main.hs b/psc-docs/Main.hs
@@ -16,11 +16,13 @@
 module Main where
 
 import Control.Applicative
-import Control.Monad.Writer
 import Control.Arrow (first, second)
 import Control.Category ((>>>))
+import Control.Monad.Writer
+import Data.Function (on)
 import Data.List
 import Data.Maybe (fromMaybe)
+import Data.Tuple (swap)
 import Data.Version (showVersion)
 
 import Options.Applicative
@@ -75,7 +77,7 @@ docgen (PSCDocsOptions fmt input output) =
         Left (D.DesugarError err) -> do
           hPutStrLn stderr $ P.prettyPrintMultipleErrors False err
           exitFailure
-        Right ms' -> do
+        Right ms' ->
           case output of
             EverythingToStdOut ->
               putStrLn (D.renderModulesAsMarkdown ms')
@@ -86,9 +88,11 @@ docgen (PSCDocsOptions fmt input output) =
             ToFiles names -> do
               let (ms, missing) = takeModulesByName' ms' names
               guardMissing missing
-              forM_ ms $ \(m, fp) -> do
+              let ms'' = groupBy ((==) `on` fst) . sortBy (compare `on` fst) $ map swap ms
+              forM_ ms'' $ \grp -> do
+                let fp = fst (head grp)
                 createDirectoryIfMissing True (takeDirectory fp)
-                writeFile fp (D.renderModulesAsMarkdown [m])
+                writeFile fp (D.renderModulesAsMarkdown $ snd `map` grp)
 
   where
   guardMissing [] = return ()
