From dedc72789b6d09c131de4134b7137e73f895f0b6 Mon Sep 17 00:00:00 2001
From: Patrick Aldis <patricktaldis@gmail.com>
Date: Thu, 5 Mar 2026 16:05:15 +0000
Subject: [PATCH] Refactor cases to move out let binding

---
 datalog-lsp/src/Datalog/LSP/Highlight.hs | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/datalog-lsp/src/Datalog/LSP/Highlight.hs b/datalog-lsp/src/Datalog/LSP/Highlight.hs
index d627bed..bba76db 100644
--- a/datalog-lsp/src/Datalog/LSP/Highlight.hs
+++ b/datalog-lsp/src/Datalog/LSP/Highlight.hs
@@ -1,29 +1,35 @@
 module Datalog.LSP.Highlight where
 
+import Control.Concurrent.STM
+import Control.Monad.Trans
 import Data.Either (fromRight)
+import Data.Map qualified as M
 import Data.Text qualified as T
 import Datalog.LSP.Types (DLogLspM, LSPContext (..))
+import Datalog.LSP.Utils (currentBufferUri)
 import Datalog.Parser (SrcLoc (..))
 import Datalog.Syntax (Atom' (Atom), Program' (..), RelId (RelId), Rule' (..), Term' (..))
 import Language.LSP.Protocol.Message
 import Language.LSP.Protocol.Types
 import Language.LSP.Server
-import Control.Monad.Trans
-import Control.Concurrent.STM
-import qualified Data.Map as M
-import Datalog.LSP.Utils (currentBufferUri)
 
 tokenHandler :: Handlers DLogLspM
 tokenHandler = requestHandler SMethod_TextDocumentSemanticTokensFull $ \req responder -> do
     LSPContext parseRef <- getConfig
     p <- lift . readTVarIO $ parseRef
     case M.lookup (currentBufferUri req) p of
-        Nothing -> responder (Left $ TResponseError (InL LSPErrorCodes_RequestFailed) "Doc not bundle" Nothing)
+        Nothing ->
+            responder (Left $ TResponseError (InL LSPErrorCodes_RequestFailed) "Doc not bundle" Nothing)
         Just (Left _) ->
             responder (Left $ TResponseError (InL LSPErrorCodes_RequestFailed) "Failed to parse" Nothing)
         Just (Right prog) -> do
-            let semanticTokens = fromRight (error "") $ makeSemanticTokens defaultSemanticTokensLegend $ highlightProg prog
-            responder (Right $ InL semanticTokens)
+            responder
+                ( Right
+                    . InL
+                    . fromRight (error "")
+                    . makeSemanticTokens defaultSemanticTokensLegend
+                    $ highlightProg prog
+                )
 
 highlightProg :: Program' SrcLoc -> [SemanticTokenAbsolute]
 highlightProg (Program _ rs) = rs >>= highlightRule