Autodetect elm version per file

Resolves avh4#561

elm-format.cabal

@@ -180,6 +180,7 @@ test-Suite elm-format-tests
         Util.ListTest
 
     build-depends:
+        filepath >= 1.4.2.1 && < 2,
         tasty >= 1.2 && < 2,
         tasty-golden >= 2.3.2 && < 3,
         tasty-hunit >= 0.10.0.1 && < 0.11,

src/ElmFormat.hs

@@ -20,6 +20,7 @@ import qualified AST.Json
 import qualified AST.Module
 import qualified Flags
 import qualified Data.Text as Text
+import qualified Data.Maybe as Maybe
 import qualified ElmFormat.Execute as Execute
 import qualified ElmFormat.InputConsole as InputConsole
 import qualified ElmFormat.Parse as Parse
@@ -34,18 +35,21 @@ import qualified Reporting.Result as Result
 import qualified Text.JSON
 
 
-resolveFile :: FileStore f => FilePath -> Free f (Either InputFileMessage [FilePath])
+resolveFile :: FileStore f => FilePath -> Free f (Either InputFileMessage [ElmFile])
 resolveFile path =
     do
         fileType <- FileStore.stat path
 
         case fileType of
             FileStore.IsFile ->
-                return $ Right [path]
+                do
+                    elmFile <- FS.addElmVersion path
+                    return $ Right [elmFile]
 
             FileStore.IsDirectory ->
                 do
-                    elmFiles <- FS.findAllElmFiles path
+                    files <- FS.findAllElmFiles path
+                    elmFiles <- sequence $ fmap FS.addElmVersion files
                     case elmFiles of
                         [] -> return $ Left $ NoElmFiles path
                         _ -> return $ Right elmFiles
@@ -74,7 +78,7 @@ collectErrors list =
         foldl step (Right []) list
 
 
-resolveFiles :: FileStore f => [FilePath] -> Free f (Either [InputFileMessage] [FilePath])
+resolveFiles :: FileStore f => [FilePath] -> Free f (Either [InputFileMessage] [ElmFile])
 resolveFiles inputFiles =
     do
         result <- collectErrors <$> mapM resolveFile inputFiles
@@ -83,23 +87,26 @@ resolveFiles inputFiles =
                 return $ Left ls
 
             Right files ->
-                return $ Right $ concat files
+                return $ Right $ concat $ files
+
 
+type ElmFile
+    = (FilePath, ElmVersion)
 
 data WhatToDo
-    = FormatToFile FilePath FilePath
+    = FormatToFile ElmFile FilePath
     | StdinToFile FilePath
-    | FormatInPlace FilePath [FilePath]
+    | FormatInPlace ElmFile [ElmFile]
     | StdinToStdout
     | ValidateStdin
-    | ValidateFiles FilePath [FilePath]
-    | FileToJson FilePath
+    | ValidateFiles ElmFile [ElmFile]
+    | FileToJson ElmFile
     | StdinToJson
 
 
 data Source
     = Stdin
-    | FromFiles FilePath [FilePath]
+    | FromFiles ElmFile [ElmFile]
 
 
 data Destination
@@ -109,7 +116,7 @@ data Destination
     | ToJson
 
 
-determineSource :: Bool -> Either [InputFileMessage] [FilePath] -> Either ErrorMessage Source
+determineSource :: Bool -> Either [InputFileMessage] [ElmFile] -> Either ErrorMessage Source
 determineSource stdin inputFiles =
     case ( stdin, inputFiles ) of
         ( _, Left fileErrors ) -> Left $ BadInputFiles fileErrors
@@ -145,7 +152,7 @@ determineWhatToDo source destination =
         ( FromFiles _ _, ToJson ) -> Left SingleOutputWithMultipleInputs
 
 
-determineWhatToDoFromConfig :: Flags.Config -> Either [InputFileMessage] [FilePath] -> Either ErrorMessage WhatToDo
+determineWhatToDoFromConfig :: Flags.Config -> Either [InputFileMessage] [(FilePath, ElmVersion)] -> Either ErrorMessage WhatToDo
 determineWhatToDoFromConfig config resolvedInputFiles =
     do
         source <- determineSource (Flags._stdin config) resolvedInputFiles
@@ -223,6 +230,7 @@ main'' elmFormatVersion_ experimental_ args =
                 do
                     let autoYes = Flags._yes config
                     resolvedInputFiles <- Execute.run (Execute.forHuman autoYes) $ resolveFiles (Flags._input config)
+                    detectedElmVersion <- Execute.run (Execute.forHuman autoYes) $ ElmVersion.fromFile "."
 
                     case determineWhatToDoFromConfig config resolvedInputFiles of
                         Left NoInputs ->
@@ -234,44 +242,25 @@ main'' elmFormatVersion_ experimental_ args =
                             exitWithError message
 
                         Right whatToDo -> do
-                            elmVersionChoice <- case Flags._elmVersion config of
-                                Just v -> return $ Right v
-                                Nothing -> autoDetectElmVersion
+                            let elmVersionFlag = Flags._elmVersion config
+                            let elmVersionChoice = case elmVersionFlag of
+                                    Just v -> v
+                                    Nothing -> detectedElmVersion
 
-                            case elmVersionChoice of
+                            let elmVersionResult = determineVersion elmVersionChoice (Flags._upgrade config)
+                            case elmVersionResult of
                                 Left message ->
-                                    putStr message *> exitFailure
-
-                                Right elmVersionChoice' -> do
-                                    let elmVersionResult = determineVersion elmVersionChoice' (Flags._upgrade config)
+                                    exitWithError message
 
-                                    case elmVersionResult of
-                                        Left message ->
-                                            exitWithError message
-
-                                        Right elmVersion ->
-                                            do
-                                                let run = case (Flags._validate config) of
-                                                        True -> Execute.run $ Execute.forMachine elmVersion True
-                                                        False -> Execute.run $ Execute.forHuman autoYes
-                                                result <-  run $ doIt elmVersion whatToDo
-                                                if result
-                                                    then exitSuccess
-                                                    else exitFailure
-
-
-autoDetectElmVersion :: World m => m (Either String ElmVersion)
-autoDetectElmVersion =
-    do
-        hasElmPackageJson <- doesFileExist "elm-package.json"
-        if hasElmPackageJson
-            then
-                do
-                    hasElmJson <- doesFileExist "elm.json"
-                    if hasElmJson
-                        then return $ Right Elm_0_19
-                        else return $ Right Elm_0_18
-            else return $ Right Elm_0_19
+                                Right elmVersion ->
+                                    do
+                                        let run = case (Flags._validate config) of
+                                                True -> Execute.run $ Execute.forMachine elmVersion True
+                                                False -> Execute.run $ Execute.forHuman autoYes
+                                        result <-  run $ doIt elmVersion elmVersionFlag whatToDo
+                                        if result
+                                            then exitSuccess
+                                            else exitFailure
 
 
 validate :: ElmVersion -> (FilePath, Text.Text) -> Either InfoMessage ()
@@ -363,33 +352,41 @@ logErrorOr fn result =
             fn value *> return True
 
 
-doIt :: (InputConsole f, OutputConsole f, InfoFormatter f, FileStore f, FileWriter f) => ElmVersion -> WhatToDo -> Free f Bool
-doIt elmVersion whatToDo =
+
+doIt :: (InputConsole f, OutputConsole f, InfoFormatter f, FileStore f, FileWriter f) => ElmVersion -> Maybe ElmVersion -> WhatToDo -> Free f Bool
+doIt elmVersion elmVersionFlag whatToDo =
+    let
+        getVersion fileDetectedElmVersion =
+            case elmVersion of
+                Elm_0_18_Upgrade -> Elm_0_18_Upgrade
+                Elm_0_19_Upgrade -> Elm_0_19_Upgrade
+                _ -> Maybe.fromMaybe fileDetectedElmVersion elmVersionFlag
+    in
     case whatToDo of
         ValidateStdin ->
             (validate elmVersion <$> readStdin) >>= logError
 
         ValidateFiles first rest ->
             all id <$> mapM validateFile (first:rest)
-            where validateFile file = (validate elmVersion <$> ElmFormat.readFile file) >>= logError
+            where validateFile (file, fileElmVersion) = (validate (getVersion fileElmVersion) <$> ElmFormat.readFile file) >>= logError
 
         StdinToStdout ->
             (fmap getOutputText <$> format elmVersion <$> readStdin) >>= logErrorOr OutputConsole.writeStdout
 
         StdinToFile outputFile ->
             (fmap getOutputText <$> format elmVersion <$> readStdin) >>= logErrorOr (FileWriter.overwriteFile outputFile)
 
-        FormatToFile inputFile outputFile ->
-            (fmap getOutputText <$> format elmVersion <$> ElmFormat.readFile inputFile) >>= logErrorOr (FileWriter.overwriteFile outputFile)
+        FormatToFile (inputFile, fileElmVersion) outputFile ->
+            (fmap getOutputText <$> format (getVersion fileElmVersion) <$> ElmFormat.readFile inputFile) >>= logErrorOr (FileWriter.overwriteFile outputFile)
 
         FormatInPlace first rest ->
             do
-                canOverwrite <- approve $ FilesWillBeOverwritten (first:rest)
+                canOverwrite <- approve $ FilesWillBeOverwritten $ fmap fst $ (first:rest)
                 if canOverwrite
                     then all id <$> mapM formatFile (first:rest)
                     else return True
             where
-                formatFile file = (format elmVersion <$> ElmFormat.readFile file) >>= logErrorOr ElmFormat.updateFile
+                formatFile (file, fileElmVersion) = (format (getVersion fileElmVersion) <$> ElmFormat.readFile file) >>= logErrorOr ElmFormat.updateFile
 
         StdinToJson ->
             (fmap (Text.pack . Text.JSON.encode . AST.Json.showModule) <$> parseModule elmVersion <$> readStdin) >>= logErrorOr OutputConsole.writeStdout

src/ElmFormat/Filesystem.hs

@@ -3,6 +3,7 @@ module ElmFormat.Filesystem where
 import Control.Monad.Free
 import ElmFormat.FileStore
 import System.FilePath ((</>))
+import ElmVersion
 import qualified System.FilePath as FilePath
 
 
@@ -63,3 +64,7 @@ findAllElmFiles inputFile =
 hasFilename :: String -> FilePath -> Bool
 hasFilename name path =
     name == FilePath.takeFileName path
+
+addElmVersion :: FileStore f => FilePath -> Free f (FilePath, ElmVersion)
+addElmVersion path =
+    fmap ((,) path) $ ElmVersion.fromFile path

src/ElmVersion.hs

@@ -1,6 +1,9 @@
 {-# OPTIONS_GHC -Wall #-}
 module ElmVersion where
 
+import Control.Monad.Free
+import System.FilePath ((</>), takeDirectory)
+import ElmFormat.FileStore
 
 data ElmVersion
   = Elm_0_16 -- TODO: remove 0_16
@@ -58,3 +61,22 @@ style_0_19_cannotExposeOpenListing elmVersion =
         Elm_0_18 -> False
         Elm_0_18_Upgrade -> False
         _ -> True
+
+fromFile :: FileStore f => FilePath -> Free f ElmVersion
+fromFile path =
+    do
+        let dir = takeDirectory (path)
+        elmPackageJson <- stat (path </> "elm-package.json")
+        case elmPackageJson of
+            IsFile ->
+                do
+                    elmJson <- stat (path </> "elm.json")
+                    return $ case elmJson of
+                        IsFile -> Elm_0_19
+                        _ -> Elm_0_18
+
+            _ | path == dir ->
+                return Elm_0_19
+
+            _ ->
+                fromFile $ dir

tests/ElmFormat/TestWorld.hs

@@ -14,6 +14,7 @@ import qualified Data.Map.Strict as Dict
 import qualified Data.Text.Lazy as Text
 import qualified Data.Text.Lazy.Encoding as Text
 import qualified Data.Text as StrictText
+import qualified System.FilePath as FilePath
 
 
 data TestWorldState =
@@ -48,7 +49,7 @@ instance World (State.State TestWorldState) where
     doesFileExist path =
         do
             state <- State.get
-            return $ Dict.member path (filesystem state)
+            return $ Dict.member (FilePath.normalise path) (filesystem state)
 
     doesDirectoryExist _path =
         return False