module Yi.Keymap.Vim.StyledRegion
( StyledRegion(..)
, normalizeRegion
, transformCharactersInRegionB
, transformCharactersInLineN
) where
import Control.Monad (forM_)
import qualified Data.Text as T (map)
import Yi.Buffer.Adjusted
import qualified Yi.Rope as R (withText)
import Yi.Utils (SemiNum ((-~)))
data StyledRegion = StyledRegion !RegionStyle !Region
normalizeRegion :: StyledRegion -> BufferM StyledRegion
normalizeRegion sr@(StyledRegion style reg) =
if style == Exclusive
then do
let end = regionEnd reg
(_, endColumn) <- getLineAndColOfPoint end
return (if endColumn == 0
then StyledRegion Inclusive $ reg { regionEnd = end -~ 2 }
else sr)
else return sr
transformCharactersInRegionB :: StyledRegion -> (Char -> Char) -> BufferM ()
transformCharactersInRegionB (StyledRegion Block reg) f = do
subregions <- splitBlockRegionToContiguousSubRegionsB reg
forM_ subregions $ \sr ->
transformCharactersInRegionB (StyledRegion Exclusive sr) f
case subregions of
(sr:_) -> moveTo (regionStart sr)
[] -> error "Should never happen"
transformCharactersInRegionB (StyledRegion style reg) f = do
reg' <- convertRegionToStyleB reg style
s <- readRegionB reg'
replaceRegionB reg' (R.withText (T.map f) s)
moveTo (regionStart reg')
transformCharactersInLineN :: Int -> (Char -> Char) -> BufferM ()
transformCharactersInLineN count action = do
p0 <- pointB
moveXorEol count
p1 <- pointB
let sreg = StyledRegion Exclusive $ mkRegion p0 p1
transformCharactersInRegionB sreg action
moveTo p1