import Data.Foldable
import Data.List
import qualified Data.Map as Map
+import qualified Data.Set as Set
import qualified Data.Text as T
import qualified Data.Text.IO as T
import qualified Data.HashMap.Strict as HashMap
, curDynCaps :: Map.Map T.Text SomeRegistration
-- ^ The capabilities that the server has dynamically registered with us so
-- far
+ , curProgressSessions :: Set.Set ProgressToken
}
class Monad m => HasState s m where
mainThreadId <- myThreadId
let context = SessionContext serverIn absRootDir messageChan timeoutIdVar reqMap initRsp config caps
- initState vfs = SessionState 0 vfs mempty False Nothing mempty
+ initState vfs = SessionState 0 vfs mempty False Nothing mempty mempty
runSession' ses = initVFS $ \vfs -> runSessionMonad context (initState vfs) ses
errorHandler = throwTo mainThreadId :: SessionException -> IO ()
updateStateC :: ConduitM FromServerMessage FromServerMessage (StateT SessionState (ReaderT SessionContext IO)) ()
updateStateC = awaitForever $ \msg -> do
updateState msg
+ respond msg
yield msg
+ where
+ respond :: (MonadIO m, HasReader SessionContext m) => FromServerMessage -> m ()
+ respond (FromServerMess SWindowWorkDoneProgressCreate req) =
+ sendMessage $ ResponseMessage "2.0" (Just $ req ^. LSP.id) (Right ())
+ respond (FromServerMess SWorkspaceApplyEdit r) = do
+ sendMessage $ ResponseMessage "2.0" (Just $ r ^. LSP.id) (Right $ ApplyWorkspaceEditResponseBody True Nothing)
+ respond _ = pure ()
+
-- extract Uri out from DocumentChange
-- didn't put this in `lsp-types` because TH was getting in the way
updateState :: (MonadIO m, HasReader SessionContext m, HasState SessionState m)
=> FromServerMessage -> m ()
+updateState (FromServerMess SProgress req) = case req ^. params . value of
+ Begin _ ->
+ modify $ \s -> s { curProgressSessions = Set.insert (req ^. params . token) $ curProgressSessions s }
+ End _ ->
+ modify $ \s -> s { curProgressSessions = Set.delete (req ^. params . token) $ curProgressSessions s }
+ _ -> pure ()
-- Keep track of dynamic capability registration
updateState (FromServerMess SClientRegisterCapability req) = do