Make the Ormolu plugin respect .ormolu fixity files

Author: amesgen

plugins/hls-ormolu-plugin/src/Ide/Plugin/Ormolu.hs

@@ -1,3 +1,4 @@
+{-# LANGUAGE CPP               #-}
 {-# LANGUAGE OverloadedStrings #-}
 {-# LANGUAGE TypeApplications  #-}
 
@@ -7,7 +8,8 @@ module Ide.Plugin.Ormolu
   )
 where
 
-import           Control.Exception               (try)
+import           Control.Exception               (Handler (..), IOException,
+                                                  SomeException (..), catches)
 import           Control.Monad.IO.Class          (liftIO)
 import qualified Data.Text                       as T
 import           Development.IDE                 hiding (pluginHandlers)
@@ -22,6 +24,11 @@ import           Language.LSP.Types
 import           Ormolu
 import           System.FilePath                 (takeFileName)
 
+#if MIN_VERSION_ormolu(0,5,0)
+import           Ormolu.Utils.Cabal              (getCabalInfoForSourceFile)
+import           Ormolu.Utils.Fixity             (getFixityOverridesForSourceFile)
+#endif
+
 -- ---------------------------------------------------------------------
 
 descriptor :: PluginId -> PluginDescriptor IdeState
@@ -43,9 +50,21 @@ provider ideState typ contents fp _ = withIndefiniteProgress title Cancellable $
     fullRegion = RegionIndices Nothing Nothing
     rangeRegion s e = RegionIndices (Just $ s + 1) (Just $ e + 1)
     mkConf o region = defaultConfig { cfgDynOptions = o, cfgRegion = region }
-    fmt :: T.Text -> Config RegionIndices -> IO (Either OrmoluException T.Text)
-    fmt cont conf =
-      try @OrmoluException $ ormolu conf (fromNormalizedFilePath fp) $ T.unpack cont
+    fmt :: T.Text -> Config RegionIndices -> IO (Either SomeException T.Text)
+    fmt cont conf = flip catches handlers $ do
+      let fp' = fromNormalizedFilePath fp
+#if MIN_VERSION_ormolu(0,5,0)
+      fixityOverrides <-
+        getFixityOverridesForSourceFile =<< getCabalInfoForSourceFile fp'
+      let conf' = conf { cfgFixityOverrides = fixityOverrides }
+#else
+      let conf' = conf
+#endif
+      Right <$> ormolu conf' fp' (T.unpack cont)
+    handlers =
+      [ Handler $ pure . Left . SomeException @OrmoluException
+      , Handler $ pure . Left . SomeException @IOException
+      ]
 
   case typ of
     FormatText -> ret <$> fmt contents (mkConf fileOpts fullRegion)
@@ -54,7 +73,7 @@ provider ideState typ contents fp _ = withIndefiniteProgress title Cancellable $
  where
    title = T.pack $ "Formatting " <> takeFileName (fromNormalizedFilePath fp)
 
-   ret :: Either OrmoluException T.Text -> Either ResponseError (List TextEdit)
+   ret :: Either SomeException T.Text -> Either ResponseError (List TextEdit)
    ret (Left err)  = Left . responseError . T.pack $ "ormoluCmd: " ++ show err
    ret (Right new) = Right $ makeDiffTextEdit contents new
 

plugins/hls-ormolu-plugin/test/Main.hs

@@ -20,6 +20,8 @@ tests = testGroup "ormolu"
       formatDoc doc (FormattingOptions 4 True Nothing Nothing Nothing)
   , goldenWithOrmolu "formats imports correctly" "Ormolu2" "formatted" $ \doc -> do
       formatDoc doc (FormattingOptions 4 True Nothing Nothing Nothing)
+  , goldenWithOrmolu "formats operators correctly" "Ormolu3" "formatted" $ \doc -> do
+      formatDoc doc (FormattingOptions 4 True Nothing Nothing Nothing)
   ]
 
 goldenWithOrmolu :: TestName -> FilePath -> FilePath -> (TextDocumentIdentifier -> Session ()) -> TestTree

plugins/hls-ormolu-plugin/test/testdata/.ormolu

@@ -0,0 +1 @@
+infixl 7 .=?

plugins/hls-ormolu-plugin/test/testdata/Ormolu3.expected.hs

@@ -0,0 +1,5 @@
+foo :: String
+foo =
+  "a" .=? "b"
+    <> "c" .=? "d"
+    <> "e" .=? "f"

plugins/hls-ormolu-plugin/test/testdata/Ormolu3.formatted.hs

@@ -0,0 +1,5 @@
+foo :: String
+foo =
+  "a" .=? "b"
+    <> "c" .=? "d"
+    <> "e" .=? "f"

plugins/hls-ormolu-plugin/test/testdata/Ormolu3.hs

@@ -0,0 +1,3 @@
+foo :: String
+foo = "a" .=? "b"
+    <> "c" .=? "d" <> "e" .=? "f"

plugins/hls-ormolu-plugin/test/testdata/test.cabal

@@ -0,0 +1,3 @@
+cabal-version: 3.0
+name: test
+version: 0