check + expand: Allow passing in multiple files

Language/Haskell/GhcMod/Check.hs

@@ -16,9 +16,14 @@ import Prelude
 --   Warnings and errors are returned.
 checkSyntax :: Options
             -> Cradle
-            -> FilePath  -- ^ A target file
+            -> [FilePath]  -- ^ The target files
             -> IO String
-checkSyntax opt cradle file = unlines <$> withGHC file (check opt cradle file)
+checkSyntax _   _      []    = error "ghc-mod: checkSyntax: No files given"
+checkSyntax opt cradle files = unlines <$> withGHC sessionName (check opt cradle files)
+  where
+    sessionName = case files of
+      [file] -> file
+      _      -> "MultipleFiles"
 
 ----------------------------------------------------------------
 
@@ -26,13 +31,14 @@ checkSyntax opt cradle file = unlines <$> withGHC file (check opt cradle file)
 --   Warnings and errors are returned.
 check :: Options
       -> Cradle
-      -> FilePath  -- ^ A target file
+      -> [FilePath]  -- ^ The target files
       -> Ghc [String]
-check opt cradle fileName = checkIt `gcatch` handleErrMsg ls
+check _   _      []        = error "ghc-mod: check: No files given"
+check opt cradle fileNames = checkIt `gcatch` handleErrMsg ls
   where
     checkIt = do
         readLog <- initializeFlagsWithCradle opt cradle options True
-        setTargetFile fileName
+        setTargetFiles fileNames
         checkSlowAndSet
         void $ load LoadAllTargets
         liftIO readLog

Language/Haskell/GhcMod/Debug.hs

@@ -36,7 +36,7 @@ debug opt cradle ver fileName = do
             return (ghcOpts opt, [], [])
     [fast] <- do
         void $ initializeFlagsWithCradle opt cradle gopts True
-        setTargetFile fileName
+        setTargetFiles [fileName]
         pure . canCheckFast <$> depanal [] False
     return [
         "GHC version:         " ++ ver

Language/Haskell/GhcMod/GHCApi.hs

@@ -5,7 +5,7 @@ module Language.Haskell.GhcMod.GHCApi (
   , withGHCDummyFile
   , initializeFlags
   , initializeFlagsWithCradle
-  , setTargetFile
+  , setTargetFiles
   , getDynamicFlags
   , setSlowDynFlags
   , checkSlowAndSet
@@ -154,11 +154,12 @@ modifyFlagsWithOpts dflags cmdOpts =
 
 ----------------------------------------------------------------
 
--- | Set the file that GHC will load / compile
+-- | Set the files that GHC will load / compile
-setTargetFile :: (GhcMonad m) => String -> m ()
+setTargetFiles :: (GhcMonad m) => [String] -> m ()
-setTargetFile file = do
+setTargetFiles [] = error "ghc-mod: setTargetFiles: No target files given"
-    target <- guessTarget file Nothing
+setTargetFiles files = do
-    setTargets [target]
+    targets <- forM files $ \file -> guessTarget file Nothing
+    setTargets targets
 
 ----------------------------------------------------------------
 

Language/Haskell/GhcMod/Info.hs

@@ -148,7 +148,7 @@ inModuleContext cmd opt cradle file modstr action errmsg =
     valid = do
         void $ initializeFlagsWithCradle opt cradle ["-w:"] False
         when (cmd == Info) setSlowDynFlags
-        setTargetFile file
+        setTargetFiles [file]
         checkSlowAndSet
         void $ load LoadAllTargets
         doif setContextFromTarget action

Language/Haskell/GhcMod/Internal.hs

@@ -5,7 +5,7 @@ module Language.Haskell.GhcMod.Internal (
     LogReader
   , GHCOption
   , initializeFlagsWithCradle
-  , setTargetFile
+  , setTargetFiles
   , checkSlowAndSet
   , getDynamicFlags
   ) where

src/GHCMod.hs

@@ -4,6 +4,7 @@ module Main where
 
 import Control.Applicative
 import Control.Exception
+import Control.Monad
 import Data.Typeable
 import Data.Version
 import Language.Haskell.GhcMod
@@ -27,8 +28,8 @@ usage =    "ghc-mod version " ++ showVersion version ++ "\n"
         ++ "\t ghc-mod lang [-l]\n"
         ++ "\t ghc-mod flag [-l]\n"
         ++ "\t ghc-mod browse" ++ ghcOptHelp ++ "[-l] [-o] [-d] <module> [<module> ...]\n"
-        ++ "\t ghc-mod check" ++ ghcOptHelp ++ "<HaskellFile>\n"
+        ++ "\t ghc-mod check" ++ ghcOptHelp ++ "<HaskellFiles...>\n"
-        ++ "\t ghc-mod expand" ++ ghcOptHelp ++ "<HaskellFile>\n"
+        ++ "\t ghc-mod expand" ++ ghcOptHelp ++ "<HaskellFiles...>\n"
         ++ "\t ghc-mod debug" ++ ghcOptHelp ++ "<HaskellFile>\n"
         ++ "\t ghc-mod info" ++ ghcOptHelp ++ "<HaskellFile> <module> <expression>\n"
         ++ "\t ghc-mod type" ++ ghcOptHelp ++ "<HaskellFile> <module> <line-no> <column-no>\n"
@@ -102,8 +103,8 @@ main = flip catches handlers $ do
     res <- case cmdArg0 of
       "browse" -> concat <$> mapM (browseModule opt) remainingArgs
       "list"   -> listModules opt
-      "check"  -> nArgs 1 $ checkSyntax opt cradle cmdArg1
+      "check"  -> checkSyntax opt cradle remainingArgs
-      "expand" -> nArgs 1 $ checkSyntax opt { expandSplice = True } cradle cmdArg1
+      "expand" -> checkSyntax opt { expandSplice = True } cradle remainingArgs
       "debug"  -> nArgs 1 $ debugInfo opt cradle strVer cmdArg1
       "type"   -> nArgs 4 $ typeExpr opt cradle cmdArg1 cmdArg2 (read cmdArg3) (read cmdArg4)
       "info"   -> nArgs 3 infoExpr opt cradle cmdArg1 cmdArg2 cmdArg3

test/CheckSpec.hs

@@ -14,24 +14,24 @@ spec = do
             withDirectory_ "test/data/ghc-mod-check" $ do
                 (strVer,_) <- getGHCVersion
                 cradle <- findCradle Nothing strVer
-                res <- checkSyntax defaultOptions cradle "main.hs"
+                res <- checkSyntax defaultOptions cradle ["main.hs"]
                 res `shouldBe` "main.hs:5:1:Warning: Top-level binding with no type signature: main :: IO ()\NUL\n"
 
         it "can check even if a test module imports another test module located at different directory" $ do
             withDirectory_ "test/data/check-test-subdir" $ do
                 cradle <- getGHCVersion >>= findCradle Nothing . fst
-                res <- checkSyntax defaultOptions cradle "test/Bar/Baz.hs"
+                res <- checkSyntax defaultOptions cradle ["test/Bar/Baz.hs"]
                 res `shouldSatisfy` (("test" </> "Foo.hs:3:1:Warning: Top-level binding with no type signature: foo :: [Char]\NUL\n") `isSuffixOf`)
 
         it "can detect mutually imported modules" $ do
             withDirectory_ "test/data" $ do
                 (strVer,_) <- getGHCVersion
                 cradle <- findCradle Nothing strVer
-                res <- checkSyntax defaultOptions cradle "Mutual1.hs"
+                res <- checkSyntax defaultOptions cradle ["Mutual1.hs"]
                 res `shouldSatisfy` ("Module imports form a cycle" `isInfixOf`)
 
         it "can check a module using QuasiQuotes" $ do
             withDirectory_ "test/data" $ do
                 cradle <- getGHCVersion >>= findCradle Nothing . fst
-                res <- checkSyntax defaultOptions cradle "Baz.hs"
+                res <- checkSyntax defaultOptions cradle ["Baz.hs"]
                 res `shouldSatisfy` ("Baz.hs:5:1:Warning:" `isPrefixOf`)