{-# OPTIONS_GHC -fno-warn-name-shadowing #-}
module Text.KSRegex (
Regex,
mkRegex,
mkRegexWithOpts,
matchRegex,
matchRegexAll,
subRegex,
splitRegex
) where
import Data.Array((!))
import Text.Regex.Base(RegexMaker(makeRegexOpts),defaultCompOpt,defaultExecOpt,RegexLike(matchAll,matchAllText),RegexContext(matchM),MatchText)
import Text.Regex.TDFA(Regex,caseSensitive,multiline,newSyntax)
mkRegex :: String -> Regex
mkRegex :: String -> Regex
mkRegex String
s = forall regex compOpt execOpt source.
RegexMaker regex compOpt execOpt source =>
compOpt -> execOpt -> source -> regex
makeRegexOpts CompOption
opt forall regex compOpt execOpt.
RegexOptions regex compOpt execOpt =>
execOpt
defaultExecOpt String
s
where opt :: CompOption
opt = forall regex compOpt execOpt.
RegexOptions regex compOpt execOpt =>
compOpt
defaultCompOpt { newSyntax :: Bool
newSyntax = Bool
True, multiline :: Bool
multiline = Bool
True }
mkRegexWithOpts
:: String
-> Bool
-> Bool
-> Regex
mkRegexWithOpts :: String -> Bool -> Bool -> Regex
mkRegexWithOpts String
s Bool
single_line Bool
case_sensitive
= let opt :: CompOption
opt = forall regex compOpt execOpt.
RegexOptions regex compOpt execOpt =>
compOpt
defaultCompOpt
{ multiline :: Bool
multiline = (if Bool
single_line then Bool
True else Bool
False)
, caseSensitive :: Bool
caseSensitive = (if Bool
case_sensitive then Bool
True else Bool
False)
, newSyntax :: Bool
newSyntax = Bool
True }
in forall regex compOpt execOpt source.
RegexMaker regex compOpt execOpt source =>
compOpt -> execOpt -> source -> regex
makeRegexOpts CompOption
opt forall regex compOpt execOpt.
RegexOptions regex compOpt execOpt =>
execOpt
defaultExecOpt String
s
matchRegex
:: Regex
-> String
-> Maybe [String]
matchRegex :: Regex -> String -> Maybe [String]
matchRegex Regex
p String
str = forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (\(String
_,String
_,String
_,[String]
str) -> [String]
str) (Regex -> String -> Maybe (String, String, String, [String])
matchRegexAll Regex
p String
str)
matchRegexAll
:: Regex
-> String
-> Maybe ( String, String, String, [String] )
matchRegexAll :: Regex -> String -> Maybe (String, String, String, [String])
matchRegexAll Regex
p String
str = forall regex source target (m :: * -> *).
(RegexContext regex source target, MonadFail m) =>
regex -> source -> m target
matchM Regex
p String
str
subRegex :: Regex
-> String
-> String
-> String
subRegex :: Regex -> String -> String -> String
subRegex Regex
_ String
"" String
_ = String
""
subRegex Regex
regexp String
inp String
repl =
let compile :: Int
-> String
-> [(String, (Int, Int))]
-> Array i (String, b)
-> String
-> String
compile Int
_i String
str [] = \ Array i (String, b)
_m -> (String
strforall a. [a] -> [a] -> [a]
++)
compile Int
i String
str ((String
"\\",(Int
off,Int
len)):[(String, (Int, Int))]
rest) =
let i' :: Int
i' = Int
offforall a. Num a => a -> a -> a
+Int
len
pre :: String
pre = forall a. Int -> [a] -> [a]
take (Int
offforall a. Num a => a -> a -> a
-Int
i) String
str
str' :: String
str' = forall a. Int -> [a] -> [a]
drop (Int
i'forall a. Num a => a -> a -> a
-Int
i) String
str
in if forall (t :: * -> *) a. Foldable t => t a -> Bool
null String
str' then \ Array i (String, b)
_m -> (String
pre forall a. [a] -> [a] -> [a]
++) forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Char
'\\'forall a. a -> [a] -> [a]
:)
else \ Array i (String, b)
m -> (String
pre forall a. [a] -> [a] -> [a]
++) forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Char
'\\' forall a. a -> [a] -> [a]
:) forall b c a. (b -> c) -> (a -> b) -> a -> c
. Int
-> String
-> [(String, (Int, Int))]
-> Array i (String, b)
-> String
-> String
compile Int
i' String
str' [(String, (Int, Int))]
rest Array i (String, b)
m
compile Int
i String
str ((String
xstr,(Int
off,Int
len)):[(String, (Int, Int))]
rest) =
let i' :: Int
i' = Int
offforall a. Num a => a -> a -> a
+Int
len
pre :: String
pre = forall a. Int -> [a] -> [a]
take (Int
offforall a. Num a => a -> a -> a
-Int
i) String
str
str' :: String
str' = forall a. Int -> [a] -> [a]
drop (Int
i'forall a. Num a => a -> a -> a
-Int
i) String
str
x :: i
x = forall a. Read a => String -> a
read String
xstr
in if forall (t :: * -> *) a. Foldable t => t a -> Bool
null String
str' then \ Array i (String, b)
m -> (String
preforall a. [a] -> [a] -> [a]
++) forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((forall a b. (a, b) -> a
fst (Array i (String, b)
mforall i e. Ix i => Array i e -> i -> e
!i
x))forall a. [a] -> [a] -> [a]
++)
else \ Array i (String, b)
m -> (String
preforall a. [a] -> [a] -> [a]
++) forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((forall a b. (a, b) -> a
fst (Array i (String, b)
mforall i e. Ix i => Array i e -> i -> e
!i
x))forall a. [a] -> [a] -> [a]
++) forall b c a. (b -> c) -> (a -> b) -> a -> c
. Int
-> String
-> [(String, (Int, Int))]
-> Array i (String, b)
-> String
-> String
compile Int
i' String
str' [(String, (Int, Int))]
rest Array i (String, b)
m
compiled :: MatchText String -> String -> String
compiled :: MatchText String -> String -> String
compiled = forall {i} {b}.
(Ix i, Read i) =>
Int
-> String
-> [(String, (Int, Int))]
-> Array i (String, b)
-> String
-> String
compile Int
0 String
repl [(String, (Int, Int))]
findrefs where
bre :: Regex
bre = String -> Regex
mkRegex String
"\\\\(\\\\|[0-9]+)"
findrefs :: [(String, (Int, Int))]
findrefs = forall a b. (a -> b) -> [a] -> [b]
map (\MatchText String
m -> (forall a b. (a, b) -> a
fst (MatchText String
mforall i e. Ix i => Array i e -> i -> e
!Int
1),forall a b. (a, b) -> b
snd (MatchText String
mforall i e. Ix i => Array i e -> i -> e
!Int
0))) (forall regex source.
RegexLike regex source =>
regex -> source -> [MatchText source]
matchAllText Regex
bre String
repl)
go :: Int -> String -> [MatchText String] -> String
go Int
_i String
str [] = String
str
go Int
i String
str (MatchText String
m:[MatchText String]
ms) =
let (String
_,(Int
off,Int
len)) = MatchText String
mforall i e. Ix i => Array i e -> i -> e
!Int
0
i' :: Int
i' = Int
offforall a. Num a => a -> a -> a
+Int
len
pre :: String
pre = forall a. Int -> [a] -> [a]
take (Int
offforall a. Num a => a -> a -> a
-Int
i) String
str
str' :: String
str' = forall a. Int -> [a] -> [a]
drop (Int
i'forall a. Num a => a -> a -> a
-Int
i) String
str
in if forall (t :: * -> *) a. Foldable t => t a -> Bool
null String
str' then String
pre forall a. [a] -> [a] -> [a]
++ (MatchText String -> String -> String
compiled MatchText String
m String
"")
else String
pre forall a. [a] -> [a] -> [a]
++ (MatchText String -> String -> String
compiled MatchText String
m (Int -> String -> [MatchText String] -> String
go Int
i' String
str' [MatchText String]
ms))
in Int -> String -> [MatchText String] -> String
go Int
0 String
inp (forall regex source.
RegexLike regex source =>
regex -> source -> [MatchText source]
matchAllText Regex
regexp String
inp)
splitRegex :: Regex -> String -> [String]
splitRegex :: Regex -> String -> [String]
splitRegex Regex
_ [] = []
splitRegex Regex
delim String
strIn =
let matches :: [(Int, Int)]
matches = forall a b. (a -> b) -> [a] -> [b]
map (forall i e. Ix i => Array i e -> i -> e
! Int
0) (forall regex source.
RegexLike regex source =>
regex -> source -> [Array Int (Int, Int)]
matchAll Regex
delim String
strIn)
go :: Int -> String -> [(Int, Int)] -> [String]
go Int
_i String
str [] = String
str forall a. a -> [a] -> [a]
: []
go Int
i String
str ((Int
off,Int
len):[(Int, Int)]
rest) =
let i' :: Int
i' = Int
offforall a. Num a => a -> a -> a
+Int
len
firstline :: String
firstline = forall a. Int -> [a] -> [a]
take (Int
offforall a. Num a => a -> a -> a
-Int
i) String
str
remainder :: String
remainder = forall a. Int -> [a] -> [a]
drop (Int
i'forall a. Num a => a -> a -> a
-Int
i) String
str
in seq :: forall a b. a -> b -> b
seq Int
i' forall a b. (a -> b) -> a -> b
$
if forall (t :: * -> *) a. Foldable t => t a -> Bool
null String
remainder then [String
firstline,String
""]
else String
firstline forall a. a -> [a] -> [a]
: Int -> String -> [(Int, Int)] -> [String]
go Int
i' String
remainder [(Int, Int)]
rest
in Int -> String -> [(Int, Int)] -> [String]
go Int
0 String
strIn [(Int, Int)]
matches