From: Ting-Gian LUA <banacorn@gmail.com>
Date: Fri, 11 Dec 2020 10:13:08 +0000 (+0800)
Subject: Fix DocumentChange related stuff in Test.hs
X-Git-Tag: 0.13.0.0~6^2~2
X-Git-Url: https://git.lukelau.me/?p=lsp-test.git;a=commitdiff_plain;h=56c39e493dfc0bf610e3807f69ccc7af8549d3e3

Fix DocumentChange related stuff in Test.hs
---

diff --git a/src/Language/LSP/Test.hs b/src/Language/LSP/Test.hs
index 3eda63e..a068b0e 100644
--- a/src/Language/LSP/Test.hs
+++ b/src/Language/LSP/Test.hs
@@ -283,9 +283,16 @@ getDocumentEdit doc = do
 
   documentContents doc
   where
+    -- extract Uri out from DocumentChange
+    documentChangeUri :: DocumentChange -> Uri
+    documentChangeUri (InL x) = x ^. textDocument . uri
+    documentChangeUri (InR (InL x)) = x ^. uri
+    documentChangeUri (InR (InR (InL x))) = x ^. oldUri
+    documentChangeUri (InR (InR (InR x))) = x ^. uri
+
     checkDocumentChanges req =
       let changes = req ^. params . edit . documentChanges
-          maybeDocs = fmap (fmap (^. textDocument . uri)) changes
+          maybeDocs = fmap (fmap documentChangeUri) changes
       in case maybeDocs of
         Just docs -> (doc ^. uri) `elem` docs
         Nothing -> False
@@ -583,7 +590,7 @@ applyEdit doc edit = do
   let wEdit = if supportsDocChanges
       then
         let docEdit = TextDocumentEdit verDoc (List [edit])
-        in WorkspaceEdit Nothing (Just (List [docEdit]))
+        in WorkspaceEdit Nothing (Just (List [InL docEdit]))
       else
         let changes = HashMap.singleton (doc ^. uri) (List [edit])
         in WorkspaceEdit (Just changes) Nothing