add subRange and positionInRange

Author: kokobd

lsp-types/src/Language/LSP/Types/Location.hs

@@ -1,11 +1,11 @@
-{-# LANGUAGE TemplateHaskell            #-}
-{-# LANGUAGE DeriveGeneric              #-}
+{-# LANGUAGE DeriveGeneric   #-}
+{-# LANGUAGE TemplateHaskell #-}
 module Language.LSP.Types.Location where
 
 import           Control.DeepSeq
 import           Data.Aeson.TH
 import           Data.Hashable
-import           GHC.Generics hiding (UInt)
+import           GHC.Generics              hiding (UInt)
 import           Language.LSP.Types.Common
 import           Language.LSP.Types.Uri
 import           Language.LSP.Types.Utils
@@ -33,8 +33,8 @@ instance Hashable Position
 
 data Range =
   Range
-    { _start :: Position -- ^ The range's start position.
-    , _end   :: Position -- ^ The range's end position.
+    { _start :: Position -- ^ The range's start position. (inclusive)
+    , _end   :: Position -- ^ The range's end position. (exclusive)
     } deriving (Show, Read, Eq, Ord, Generic)
 
 instance NFData Range
@@ -65,12 +65,12 @@ data LocationLink =
     -- range at the mouse position.
     _originSelectionRange :: Maybe Range
     -- | The target resource identifier of this link.
-  , _targetUri :: Uri
+  , _targetUri            :: Uri
     -- | The full target range of this link. If the target for example is a
     -- symbol then target range is the range enclosing this symbol not including
     -- leading/trailing whitespace but everything else like comments. This
     -- information is typically used to highlight the range in the editor.
-  , _targetRange :: Range
+  , _targetRange          :: Range
     -- | The range that should be selected and revealed when this link is being
     -- followed, e.g the name of a function. Must be contained by the the
     -- 'targetRange'. See also @DocumentSymbol._range@
@@ -84,3 +84,9 @@ deriveJSON lspOptions ''LocationLink
 -- prop> mkRange l c l' c' = Range (Position l c) (Position l' c')
 mkRange :: UInt -> UInt -> UInt -> UInt -> Range
 mkRange l c l' c' = Range (Position l c) (Position l' c')
+
+subRange :: Range -> Range -> Bool
+subRange smallRange range = _start smallRange >= _start range && _end smallRange <= _end range
+
+positionInRange :: Position -> Range -> Bool
+positionInRange p (Range sp ep) = sp <= p && p < ep

lsp-types/test/LocationSpec.hs

@@ -0,0 +1,35 @@
+{-# LANGUAGE OverloadedStrings #-}
+
+module LocationSpec where
+
+import           Language.LSP.Types
+import           Test.Hspec
+
+main :: IO ()
+main = hspec spec
+
+spec :: Spec
+spec = do
+  describe "subRange" $ do
+    it "is true if the first range is totally inside the second range" $
+      subRange (mkRange 1 2 1 5) (mkRange 1 1 1 6) `shouldBe` True
+    it "is true if two ranges equal" $
+      subRange (mkRange 1 2 1 5) (mkRange 1 2 1 5) `shouldBe` True
+    it "is false if the first range is outside of the second" $
+      subRange (mkRange 1 1 1 5) (mkRange 1 2 1 5) `shouldBe` False
+
+  describe "positionInRange" $ do
+    it "is false if position is after the end of a single line range" $
+      positionInRange (Position 1 10) (Range (Position 1 1) (Position 1 3)) `shouldBe` False
+    it "is false if position is before the begining of a single line range" $
+      positionInRange (Position 1 0) (Range (Position 1 1) (Position 1 6)) `shouldBe` False
+    it "is true if position is in a single line range" $
+      positionInRange (Position 1 5) (Range (Position 1 1) (Position 1 6)) `shouldBe` True
+    it "is false if position is right at the end of the range" $
+      positionInRange (Position 1 5) (Range (Position 1 1) (Position 1 5)) `shouldBe` False
+    it "is true if position is in the middle of a multiline range" $
+      positionInRange (Position 3 5) (Range (Position 1 1) (Position 5 6)) `shouldBe` True
+    it "is false if position is before the beginning of a multiline range" $
+      positionInRange (Position 3 5) (Range (Position 3 6) (Position 4 10)) `shouldBe` False
+    it "is false if position is right at the end of a multiline range" $
+      positionInRange (Position 4 10) (Range (Position 3 6) (Position 4 10)) `shouldBe` False