Initial Retrie plugin

README.md

@@ -71,6 +71,10 @@ This is *very* early stage software.
 
   This will cause compilation errors if a dependency contains invalid Haddock markup, though in a future version of GHC (hopefully 8.12), [these will be demoted to warnings](https://gitlab.haskell.org/ghc/ghc/-/merge_requests/2377).
 
+ - Integration with [retrie](https://hackage.haskell.org/package/retrie)
+
+   ![Retrie](https://i.imgur.com/VFKx1He.gif)
+
  - Many more (TBD)
 
 ## Installation

exe/Main.hs

@@ -69,6 +69,7 @@ import Ide.Plugin.GhcIde                  as GhcIde
 import Ide.Plugin.Floskell                as Floskell
 import Ide.Plugin.Ormolu                  as Ormolu
 import Ide.Plugin.StylishHaskell          as StylishHaskell
+import Ide.Plugin.Retrie                  as Retrie
 #if AGPL
 import Ide.Plugin.Brittany                as Brittany
 #endif
@@ -105,6 +106,7 @@ idePlugins includeExamples = pluginDescToIdePlugins allPlugins
       -- , ghcmodDescriptor      "ghcmod"
       , Ormolu.descriptor   "ormolu"
       , StylishHaskell.descriptor "stylish-haskell"
+      , Retrie.descriptor "retrie"
 #if AGPL
       , Brittany.descriptor    "brittany"
 #endif

haskell-language-server.cabal

@@ -47,6 +47,7 @@ library
       Ide.Plugin.GhcIde
       Ide.Plugin.Ormolu
       Ide.Plugin.Pragmas
+      Ide.Plugin.Retrie
       Ide.Plugin.Floskell
       Ide.Plugin.Formatter
       Ide.Plugin.StylishHaskell
@@ -83,6 +84,8 @@ library
     , optparse-simple
     , process
     , regex-tdfa >= 1.3.1.0
+    , retrie >= 0.1.1.0
+    , safe-exceptions
     , shake >= 0.17.5
     , stylish-haskell == 0.11.*
     , temporary

shell.nix

@@ -17,7 +17,8 @@ let defaultCompiler = "ghc" + lib.replaceStrings ["."] [""] haskellPackages.ghc.
             # for all other compilers there is no Nix cache so dont bother building deps with NIx
             else haskell.packages.${compiler}.ghcWithPackages (_: []);
 
-   compilerWithPackages = haskellPackagesForProject(p:
+    retrie = with haskell.lib; dontCheck(disableLibraryProfiling(haskellPackages.retrie));
+    compilerWithPackages = haskellPackagesForProject(p:
         with p;
         [
           Diff
@@ -66,6 +67,7 @@ let defaultCompiler = "ghc" + lib.replaceStrings ["."] [""] haskellPackages.ghc.
           primes
           psqueues
           regex-tdfa
+          retrie
           rope-utf16-splay
           safe-exceptions
           shake

src/Ide/Plugin.hs

@@ -18,6 +18,7 @@ module Ide.Plugin
     , responseError
     ) where
 
+import           Control.Exception(SomeException, catch)
 import           Control.Lens ( (^.) )
 import           Control.Monad
 import qualified Data.Aeson as J
@@ -206,7 +207,7 @@ executeCommandHandlers ecs = PartialHandlers $ \WithMessage{..} x -> return x{
 --                             -> ExecuteCommandParams
 --                             -> IO (Either ResponseError Value, Maybe (ServerMethod, ApplyWorkspaceEditParams))
 makeExecuteCommands :: [(PluginId, [PluginCommand])] -> LSP.LspFuncs Config -> ExecuteCommandProvider
-makeExecuteCommands ecs lf ide = do
+makeExecuteCommands ecs lf ide = wrapUnhandledExceptions $ do
   let
       pluginMap = Map.fromList ecs
       parseCmdId :: T.Text -> Maybe (PluginId, CommandId)
@@ -334,6 +335,14 @@ makeExecuteCommands ecs lf ide = do
 
 -- | Runs a plugin command given a PluginId, CommandId and
 -- arguments in the form of a JSON object.
+wrapUnhandledExceptions ::
+    (a -> IO (Either ResponseError J.Value, Maybe b)) ->
+       a -> IO (Either ResponseError J.Value, Maybe b)
+wrapUnhandledExceptions action input =
+    catch (action input) $ \(e::SomeException) -> do
+        let resp = ResponseError InternalError (T.pack $ show e) Nothing
+        return (Left resp, Nothing)
+
 runPluginCommand :: Map.Map PluginId [PluginCommand]
                  -> LSP.LspFuncs Config
                  -> IdeState