Allow formatting of multiple files.

diku-dk · Nov 4, 2024 · 09c4284 · 09c4284
1 parent b061618
commit 09c4284
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/Futhark/CLI/Fmt.hs b/src/Futhark/CLI/Fmt.hs
@@ -1,6 +1,7 @@
 -- | @futhark fmt@
 module Futhark.CLI.Fmt (main) where
 
+import Control.Monad (forM_)
 import Data.Text.IO qualified as T
 import Futhark.Fmt.Printer
 import Futhark.Util.Options
@@ -12,15 +13,13 @@ import System.IO
 
 -- | Run @futhark fmt@.
 main :: String -> [String] -> IO ()
-main = mainWithOptions () [] "program" $ \args () ->
+main = mainWithOptions () [] "[FILES" $ \args () ->
   case args of
     [] -> Just $ putDoc =<< onInput =<< T.getContents
-    [file] ->
-      Just $ do
+    files ->
+      Just $ forM_ files $ \file -> do
         doc <- onInput =<< T.readFile file
-        withFile file WriteMode $ \h ->
-          hPutDoc h doc
-    _any -> Nothing
+        withFile file WriteMode $ \h -> hPutDoc h doc
   where
     onInput s = do
       case fmtToDoc "<stdin>" s of