Use haskell-src-exts

1 files changed, 20 insertions, 7 deletions

diff --git a/src/Main.hs b/src/Main.hs
index 7a325f3..d975f5f 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -1,21 +1,34 @@
+--------------------------------------------------------------------------------
 module Main
     ( main
     ) where
 
 
 --------------------------------------------------------------------------------
-import           Control.Monad         ((>=>))
-import           System.Environment    (getArgs)
+import           Control.Applicative    ((<$>))
+import           Data.Maybe             (listToMaybe)
+import           System.Environment     (getArgs)
+
+
+--------------------------------------------------------------------------------
+import           StylishHaskell.Editor
+import           StylishHaskell.Imports
+import           StylishHaskell.Parse
 
 
 --------------------------------------------------------------------------------
-import           StylishHaskellImports
+stylishHaskell :: Maybe FilePath -> String -> IO ()
+stylishHaskell mfp string = case parseModule mfp string of
+    Left err      -> error err
+    Right module' -> do
+        let ls      = lines string
+            changes = stylish ls module'
+        putStr $ unlines $ applyChanges changes ls
 
 
 --------------------------------------------------------------------------------
 main :: IO ()
 main = do
-    args <- getArgs
-    case args of
-        [] -> interact stylishHaskellImports
-        _  -> mapM_ (readFile >=> putStr . stylishHaskellImports) args
+    filePath <- listToMaybe <$> getArgs
+    contents <- maybe getContents readFile filePath
+    stylishHaskell filePath contents