Add nicer API

[opengl.git] / src / Language / Haskell / LSP / Test / Parsing.hs

diff --git a/src/Language/Haskell/LSP/Test/Parsing.hs b/src/Language/Haskell/LSP/Test/Parsing.hs
index fdc2e958bafc45eae1e02deceba80ba4e03657b4..614495b27e68c051249fa10beb0bef0e5a11b788 100644 (file)
--- a/src/Language/Haskell/LSP/Test/Parsing.hs
+++ b/src/Language/Haskell/LSP/Test/Parsing.hs
@@ -6,6 +6,7 @@ module Language.Haskell.LSP.Test.Parsing where
 
 import Control.Applicative
 import Control.Concurrent
+import Control.Lens
 import Control.Monad.IO.Class
 import Control.Monad.Trans.Class
 import Data.Aeson
@@ -13,12 +14,13 @@ import qualified Data.ByteString.Lazy.Char8 as B
 import Data.Conduit.Parser
 import Data.Maybe
 import Language.Haskell.LSP.Messages
-import Language.Haskell.LSP.Types hiding (error)
+import Language.Haskell.LSP.Types as LSP hiding (error)
 import Language.Haskell.LSP.Test.Exceptions
 import Language.Haskell.LSP.Test.Messages
 import Language.Haskell.LSP.Test.Session
+import System.Console.ANSI
 
-satisfy :: (MonadIO m, MonadSessionConfig m) => (a -> Bool) -> ConduitParser a m a
+satisfy :: (MonadIO m, MonadSessionConfig m) => (FromServerMessage -> Bool) -> ConduitParser FromServerMessage m FromServerMessage
 satisfy pred = do
   timeout <- timeout <$> lift sessionConfig
   tId <- liftIO myThreadId
@@ -27,8 +29,14 @@ satisfy pred = do
     throwTo tId TimeoutException
   x <- await
   liftIO $ killThread timeoutThread
+
   if pred x
-    then return x
+    then do
+      liftIO $ do
+        setSGR [SetColor Foreground Vivid Magenta]
+        putStrLn $ "<-- " ++ B.unpack (encodeMsg x)
+        setSGR [Reset]
+      return x
     else empty
 
 -- | Matches if the message is a notification.
@@ -61,6 +69,15 @@ response = named "Response" $ do
   x <- satisfy (isJust . parser)
   return $ castMsg x
 
+responseForId :: forall m a. (MonadIO m, MonadSessionConfig m, FromJSON a) => LspId -> ConduitParser FromServerMessage m (ResponseMessage a)
+responseForId lid = named "Response for id" $ do
+  let parser = decode . encodeMsg :: FromServerMessage -> Maybe (ResponseMessage a)
+  x <- satisfy (maybe False (\z -> z ^. LSP.id == responseId lid) . parser)
+  return $ castMsg x
+
+anyMessage :: (MonadIO m, MonadSessionConfig m) => ConduitParser FromServerMessage m FromServerMessage
+anyMessage = satisfy (const True)
+
 -- | A stupid method for getting out the inner message.
 castMsg :: FromJSON a => FromServerMessage -> a
 castMsg = fromMaybe (error "Failed casting a message") . decode . encodeMsg