Add --did-save-period-ms option

Author: thomasjm

app/Main.hs

@@ -52,6 +52,7 @@ data Options = Options {
   optWrappedLanguageServer :: Maybe FilePath
   , optWrappedArgs :: Maybe Text
   , optShadowDirTemplate :: Maybe FilePath
+  , optDidSaveDebouncePeriodMs :: Int
   , optLogLevel :: Maybe Text
   }
 
@@ -60,6 +61,7 @@ options = Options
   <$> optional (strOption (long "wrapped-server" <> help "Wrapped rust-analyzer binary"))
   <*> optional (strOption (long "wrapped-args" <> help "Extra arguments to rust-analyzer"))
   <*> optional (strOption (long "shadow-dir-template" <> help "Template for making a shadow project directory"))
+  <*> option auto (long "did-save-period-ms" <> showDefault <> help "Debounce period for sending textDocument/didSave notifications after changes" <> value 1000 <> metavar "INT")
   <*> optional (strOption (long "log-level" <> help "Log level (debug, info, warn, error)"))
 
 fullOpts :: ParserInfo Options
@@ -134,7 +136,7 @@ main = do
         }
 
   withMaybeShadowDir optShadowDirTemplate $ \maybeShadowDir -> do
-    transformerState <- newTransformerState maybeShadowDir
+    transformerState <- newTransformerState maybeShadowDir optDidSaveDebouncePeriodMs
 
     flip runLoggingT logFn $ filterLogger logFilterFn $ flip runReaderT transformerState $
       withAsync (readWrappedOut clientReqMap serverReqMap wrappedOut sendToStdout) $ \_wrappedOutAsync ->

default.nix

@@ -19,14 +19,14 @@ mkDerivation {
     unliftio-core uuid
   ];
   executableHaskellDepends = [
-    aeson auto-update base bytestring lens lsp-types monad-logger mtl
+    aeson base bytestring lens lsp-types monad-logger mtl
     optparse-applicative process retry safe string-interpolate text
-    unix unliftio unliftio-core uuid
+    unix unliftio unliftio-core
   ];
   testHaskellDepends = [
-    auto-update base exceptions lsp-types monad-logger myers-diff
-    QuickCheck quickcheck-instances row-types sandwich
-    sandwich-quickcheck string-interpolate text text-rope unliftio uuid
+    base exceptions lsp-types monad-logger myers-diff QuickCheck
+    quickcheck-instances row-types sandwich sandwich-quickcheck
+    string-interpolate text text-rope unliftio
   ];
   doCheck = false;
   license = "unknown";

package.yaml

@@ -6,15 +6,13 @@ extra-source-files:
 - CHANGELOG.md
 
 dependencies:
-- auto-update
 - base >= 4.7 && < 5
 # - language-rust
 - lsp-types >= 2.0.0.1
 - monad-logger
 - string-interpolate
 - text
 - unliftio
-- uuid
 
 default-extensions:
 - OverloadedStrings
@@ -40,6 +38,7 @@ library:
   source-dirs: src
   dependencies:
   - aeson
+  - auto-update
   - containers
   - filepath
   - lens
@@ -54,6 +53,7 @@ library:
   - text-rope
   - time
   - unliftio-core
+  - uuid
 
 executables:
   rust-notebook-language-server:

rust-notebook-language-server.cabal

@@ -100,7 +100,6 @@ executable rust-notebook-language-server
   ghc-options: -Wunused-packages -Wall -fno-warn-name-shadowing -threaded -rtsopts -with-rtsopts=-N -O2 -W
   build-depends:
       aeson
-    , auto-update
     , base >=4.7 && <5
     , bytestring
     , lens
@@ -117,7 +116,6 @@ executable rust-notebook-language-server
     , unix
     , unliftio
     , unliftio-core
-    , uuid
   default-language: Haskell2010
 
 test-suite rust-notebook-language-server-test
@@ -151,7 +149,6 @@ test-suite rust-notebook-language-server-test
   ghc-options: -Wunused-packages -Wall -fno-warn-name-shadowing -threaded -rtsopts -with-rtsopts=-N -O2 -W
   build-depends:
       QuickCheck
-    , auto-update
     , base >=4.7 && <5
     , exceptions
     , lsp-types >=2.0.0.1
@@ -166,5 +163,4 @@ test-suite rust-notebook-language-server-test
     , text
     , text-rope
     , unliftio
-    , uuid
   default-language: Haskell2010

src/Transform/ClientNot.hs

@@ -86,7 +86,7 @@ transformClientNot' sendExtraNotification SMethod_TextDocumentDidOpen params = w
                 | documentUuid == uuid -> runInIO $ doDidSave ds sendExtraNotification
                 | otherwise -> return ()
               _ -> return ()
-        , debounceFreq = 5_000_000
+        , debounceFreq = transformerDidSaveDebouncePeriodMs * 1000
         , debounceEdge = trailingEdge
         })
 

src/Transform/Util.hs

@@ -110,16 +110,18 @@ data TransformerState = TransformerState {
   , transformerInitializeParams :: MVar (Maybe InitializeParams)
   , transformerInitializeResult :: MVar (Maybe InitializeResult)
   , transformerShadowDir :: FilePath
+  , transformerDidSaveDebouncePeriodMs :: Int
   }
 
 -- * Transformers
 
-newTransformerState :: (MonadIO m) => FilePath -> m TransformerState
-newTransformerState shadowDir = TransformerState
+newTransformerState :: (MonadIO m) => FilePath -> Int -> m TransformerState
+newTransformerState shadowDir didSaveDebouncePeriodMs = TransformerState
   <$> newMVar mempty
   <*> newMVar Nothing
   <*> newMVar Nothing
   <*> pure shadowDir
+  <*> pure didSaveDebouncePeriodMs
 
 lookupTransformer :: TransformerMonad m => Uri -> m (Maybe DocumentState)
 lookupTransformer uri = do