module System.FilePath.GlobPattern (
GlobPattern
, (~~)
, (/~)
) where
import Control.Arrow (second)
import Control.Monad (msum)
import Data.Ix (Ix, inRange)
import Data.List (nub)
import Data.Maybe (isJust)
import System.FilePath (pathSeparator)
type GlobPattern = String
spanClass :: Char -> String -> (String, String)
spanClass :: Char -> String -> (String, String)
spanClass Char
c = String -> String -> (String, String)
gs []
where gs :: String -> String -> (String, String)
gs String
_ [] = String -> (String, String)
forall a. HasCallStack => String -> a
error String
"unterminated character class"
gs String
acc (Char
d:String
ds) | Char
d Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== Char
c = (String -> String
forall a. [a] -> [a]
reverse String
acc, String
ds)
| Char
d Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== Char
'\\' = case String
ds of
(Char
e:String
es) -> String -> String -> (String, String)
gs (Char
eChar -> String -> String
forall a. a -> [a] -> [a]
:Char
'\\'Char -> String -> String
forall a. a -> [a] -> [a]
:String
acc) String
es
String
_ -> String -> (String, String)
forall a. HasCallStack => String -> a
error String
"unterminated escape"
| Bool
otherwise = String -> String -> (String, String)
gs (Char
dChar -> String -> String
forall a. a -> [a] -> [a]
:String
acc) String
ds
data Ix a => SRange a = SRange [a] [(a, a)]
deriving (Int -> SRange a -> String -> String
[SRange a] -> String -> String
SRange a -> String
(Int -> SRange a -> String -> String)
-> (SRange a -> String)
-> ([SRange a] -> String -> String)
-> Show (SRange a)
forall a. (Ix a, Show a) => Int -> SRange a -> String -> String
forall a. (Ix a, Show a) => [SRange a] -> String -> String
forall a. (Ix a, Show a) => SRange a -> String
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
showList :: [SRange a] -> String -> String
$cshowList :: forall a. (Ix a, Show a) => [SRange a] -> String -> String
show :: SRange a -> String
$cshow :: forall a. (Ix a, Show a) => SRange a -> String
showsPrec :: Int -> SRange a -> String -> String
$cshowsPrec :: forall a. (Ix a, Show a) => Int -> SRange a -> String -> String
Show)
inSRange :: Ix a => a -> SRange a -> Bool
inSRange :: a -> SRange a -> Bool
inSRange a
c (SRange [a]
d [(a, a)]
s) = a
c a -> [a] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [a]
d Bool -> Bool -> Bool
|| ((a, a) -> Bool) -> [(a, a)] -> Bool
forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Bool
any (((a, a) -> a -> Bool) -> a -> (a, a) -> Bool
forall a b c. (a -> b -> c) -> b -> a -> c
flip (a, a) -> a -> Bool
forall a. Ix a => (a, a) -> a -> Bool
inRange a
c) [(a, a)]
s
type CharClass = SRange Char
makeClass :: String -> CharClass
makeClass :: String -> CharClass
makeClass = [(Char, Char)] -> String -> String -> CharClass
makeClass' [] []
where makeClass' :: [(Char, Char)] -> [Char] -> String -> CharClass
makeClass' :: [(Char, Char)] -> String -> String -> CharClass
makeClass' [(Char, Char)]
dense String
sparse [] = String -> [(Char, Char)] -> CharClass
forall a. [a] -> [(a, a)] -> SRange a
SRange String
sparse [(Char, Char)]
dense
makeClass' [(Char, Char)]
dense String
sparse (Char
a:Char
'-':Char
b:String
cs) =
[(Char, Char)] -> String -> String -> CharClass
makeClass' ((Char
a,Char
b)(Char, Char) -> [(Char, Char)] -> [(Char, Char)]
forall a. a -> [a] -> [a]
:[(Char, Char)]
dense) String
sparse String
cs
makeClass' [(Char, Char)]
dense String
sparse (Char
c:String
cs) = [(Char, Char)] -> String -> String -> CharClass
makeClass' [(Char, Char)]
dense (Char
cChar -> String -> String
forall a. a -> [a] -> [a]
:String
sparse) String
cs
data MatchTerm = MatchLiteral String
| MatchAny
| MatchDir
| MatchChar
| MatchClass Bool CharClass
| MatchGroup [String]
deriving (Int -> MatchTerm -> String -> String
[MatchTerm] -> String -> String
MatchTerm -> String
(Int -> MatchTerm -> String -> String)
-> (MatchTerm -> String)
-> ([MatchTerm] -> String -> String)
-> Show MatchTerm
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
showList :: [MatchTerm] -> String -> String
$cshowList :: [MatchTerm] -> String -> String
show :: MatchTerm -> String
$cshow :: MatchTerm -> String
showsPrec :: Int -> MatchTerm -> String -> String
$cshowsPrec :: Int -> MatchTerm -> String -> String
Show)
parseGlob :: GlobPattern -> [MatchTerm]
parseGlob :: String -> [MatchTerm]
parseGlob [] = []
parseGlob (Char
'*':Char
'*':String
cs) = MatchTerm
MatchAny MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: String -> [MatchTerm]
parseGlob String
cs
parseGlob (Char
'*':String
cs) = MatchTerm
MatchDir MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: String -> [MatchTerm]
parseGlob String
cs
parseGlob (Char
'?':String
cs) = MatchTerm
MatchChar MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: String -> [MatchTerm]
parseGlob String
cs
parseGlob (Char
'[':String
cs) = let (String
cc, String
ccs) = Char -> String -> (String, String)
spanClass Char
']' String
cs
cls :: MatchTerm
cls = case String
cc of
(Char
'!':String
ccs') -> Bool -> CharClass -> MatchTerm
MatchClass Bool
False (CharClass -> MatchTerm) -> CharClass -> MatchTerm
forall a b. (a -> b) -> a -> b
$ String -> CharClass
makeClass String
ccs'
String
_ -> Bool -> CharClass -> MatchTerm
MatchClass Bool
True (CharClass -> MatchTerm) -> CharClass -> MatchTerm
forall a b. (a -> b) -> a -> b
$ String -> CharClass
makeClass String
cc
in MatchTerm
cls MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: String -> [MatchTerm]
parseGlob String
ccs
parseGlob (Char
'(':String
cs) = let (String
gg, String
ggs) = Char -> String -> (String, String)
spanClass Char
')' String
cs
in [String] -> MatchTerm
MatchGroup (String -> String -> [String]
breakGroup [] String
gg) MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: String -> [MatchTerm]
parseGlob String
ggs
where breakGroup :: String -> String -> [String]
breakGroup :: String -> String -> [String]
breakGroup String
acc [] = [String -> String
forall a. [a] -> [a]
reverse String
acc]
breakGroup String
_ [Char
'\\'] = String -> [String]
forall a. HasCallStack => String -> a
error String
"group: unterminated escape"
breakGroup String
acc (Char
'\\':Char
c:String
cs') = String -> String -> [String]
breakGroup (Char
cChar -> String -> String
forall a. a -> [a] -> [a]
:String
acc) String
cs'
breakGroup String
acc (Char
'|':String
cs') = String -> String
forall a. [a] -> [a]
reverse String
acc String -> [String] -> [String]
forall a. a -> [a] -> [a]
: String -> String -> [String]
breakGroup [] String
cs'
breakGroup String
acc (Char
c:String
cs') = String -> String -> [String]
breakGroup (Char
cChar -> String -> String
forall a. a -> [a] -> [a]
:String
acc) String
cs'
parseGlob [Char
'\\'] = String -> [MatchTerm]
forall a. HasCallStack => String -> a
error String
"glob: unterminated escape"
parseGlob (Char
'\\':Char
c:String
cs) = String -> MatchTerm
MatchLiteral [Char
c] MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: String -> [MatchTerm]
parseGlob String
cs
parseGlob (Char
c:String
cs) = String -> MatchTerm
MatchLiteral [Char
c] MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: String -> [MatchTerm]
parseGlob String
cs
simplifyTerms :: [MatchTerm] -> [MatchTerm]
simplifyTerms :: [MatchTerm] -> [MatchTerm]
simplifyTerms [] = []
simplifyTerms (MatchLiteral []:[MatchTerm]
as) = [MatchTerm] -> [MatchTerm]
simplifyTerms [MatchTerm]
as
simplifyTerms (m :: MatchTerm
m@(MatchLiteral String
a):[MatchTerm]
as) =
case [MatchTerm] -> [MatchTerm]
simplifyTerms [MatchTerm]
as of
(MatchLiteral String
b:[MatchTerm]
bs) -> String -> MatchTerm
MatchLiteral (String
a String -> String -> String
forall a. [a] -> [a] -> [a]
++ String
b) MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: [MatchTerm]
bs
[MatchTerm]
bs -> MatchTerm
m MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: [MatchTerm]
bs
simplifyTerms (MatchClass Bool
True (SRange [] []):[MatchTerm]
as) = [MatchTerm] -> [MatchTerm]
simplifyTerms [MatchTerm]
as
simplifyTerms (MatchClass Bool
True (SRange a :: String
a@[Char
_] []):[MatchTerm]
as) =
[MatchTerm] -> [MatchTerm]
simplifyTerms ([MatchTerm] -> [MatchTerm]) -> [MatchTerm] -> [MatchTerm]
forall a b. (a -> b) -> a -> b
$ String -> MatchTerm
MatchLiteral String
a MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: [MatchTerm]
as
simplifyTerms (MatchGroup []:[MatchTerm]
as) = [MatchTerm] -> [MatchTerm]
simplifyTerms [MatchTerm]
as
simplifyTerms (MatchGroup [String]
gs:[MatchTerm]
as) =
case [String] -> (String, [String])
commonPrefix [String]
gs of
(String
p ,[]) -> [MatchTerm] -> [MatchTerm]
simplifyTerms (String -> MatchTerm
MatchLiteral String
p MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: [MatchTerm]
as)
(String
"",[String]
ss) -> [String] -> MatchTerm
MatchGroup [String]
ss MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: [MatchTerm] -> [MatchTerm]
simplifyTerms [MatchTerm]
as
(String
p ,[String]
ss) -> [MatchTerm] -> [MatchTerm]
simplifyTerms (String -> MatchTerm
MatchLiteral String
p MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: [String] -> MatchTerm
MatchGroup [String]
ss MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: [MatchTerm]
as)
simplifyTerms (MatchTerm
a:[MatchTerm]
as) = MatchTerm
aMatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
:[MatchTerm] -> [MatchTerm]
simplifyTerms [MatchTerm]
as
commonPrefix :: [String] -> (String, [String])
commonPrefix :: [String] -> (String, [String])
commonPrefix = ([String] -> [String]) -> (String, [String]) -> (String, [String])
forall (a :: * -> * -> *) b c d.
Arrow a =>
a b c -> a (d, b) (d, c)
second [String] -> [String]
forall a. Eq a => [a] -> [a]
nub ((String, [String]) -> (String, [String]))
-> ([String] -> (String, [String]))
-> [String]
-> (String, [String])
forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> [String] -> (String, [String])
pfx String
""
where pfx :: String -> [String] -> (String, [String])
pfx String
_ [] = (String
"", [])
pfx String
acc [String]
ss | (String -> Bool) -> [String] -> Bool
forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Bool
any String -> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null [String]
ss = (String -> String
forall a. [a] -> [a]
reverse String
acc, [String]
ss)
| Bool
otherwise = let hs :: String
hs = (String -> Char) -> [String] -> String
forall a b. (a -> b) -> [a] -> [b]
map String -> Char
forall a. [a] -> a
head [String]
ss
h :: Char
h = String -> Char
forall a. [a] -> a
head String
hs
in if (Char -> Bool) -> String -> Bool
forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Bool
all (Char
hChar -> Char -> Bool
forall a. Eq a => a -> a -> Bool
==) (String -> Bool) -> String -> Bool
forall a b. (a -> b) -> a -> b
$ String -> String
forall a. [a] -> [a]
tail String
hs
then String -> [String] -> (String, [String])
pfx (Char
hChar -> String -> String
forall a. a -> [a] -> [a]
:String
acc) ([String] -> (String, [String])) -> [String] -> (String, [String])
forall a b. (a -> b) -> a -> b
$ (String -> String) -> [String] -> [String]
forall a b. (a -> b) -> [a] -> [b]
map String -> String
forall a. [a] -> [a]
tail [String]
ss
else (String -> String
forall a. [a] -> [a]
reverse String
acc, [String]
ss)
matchTerms :: [MatchTerm] -> String -> Maybe ()
matchTerms :: [MatchTerm] -> String -> Maybe ()
matchTerms [] [] = () -> Maybe ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
matchTerms [] String
_ = String -> Maybe ()
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"residual string"
matchTerms (MatchLiteral String
m:[MatchTerm]
ts) String
cs = String -> String -> Maybe String
forall a (m :: * -> *). (Eq a, MonadFail m) => [a] -> [a] -> m [a]
matchLiteral String
m String
cs Maybe String -> (String -> Maybe ()) -> Maybe ()
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= [MatchTerm] -> String -> Maybe ()
matchTerms [MatchTerm]
ts
where matchLiteral :: [a] -> [a] -> m [a]
matchLiteral (a
a:[a]
as) (a
b:[a]
bs) | a
a a -> a -> Bool
forall a. Eq a => a -> a -> Bool
== a
b = [a] -> [a] -> m [a]
matchLiteral [a]
as [a]
bs
matchLiteral [] [a]
as = [a] -> m [a]
forall (m :: * -> *) a. Monad m => a -> m a
return [a]
as
matchLiteral [a]
_ [a]
_ = String -> m [a]
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"not a prefix"
matchTerms (MatchClass Bool
k CharClass
c:[MatchTerm]
ts) String
cs = String -> Maybe String
forall (m :: * -> *). MonadFail m => String -> m String
matchClass String
cs Maybe String -> (String -> Maybe ()) -> Maybe ()
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= [MatchTerm] -> String -> Maybe ()
matchTerms [MatchTerm]
ts
where matchClass :: String -> m String
matchClass (Char
b:String
bs) | (Bool
inClass Bool -> Bool -> Bool
&& Bool
k) Bool -> Bool -> Bool
|| Bool -> Bool
not (Bool
inClass Bool -> Bool -> Bool
|| Bool
k) = String -> m String
forall (m :: * -> *) a. Monad m => a -> m a
return String
bs
where inClass :: Bool
inClass = Char
b Char -> CharClass -> Bool
forall a. Ix a => a -> SRange a -> Bool
`inSRange` CharClass
c
matchClass String
_ = String -> m String
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"no match"
matchTerms (MatchGroup [String]
g:[MatchTerm]
ts) String
cs = [Maybe ()] -> Maybe ()
forall (t :: * -> *) (m :: * -> *) a.
(Foldable t, MonadPlus m) =>
t (m a) -> m a
msum ((String -> Maybe ()) -> [String] -> [Maybe ()]
forall a b. (a -> b) -> [a] -> [b]
map String -> Maybe ()
matchGroup [String]
g)
where matchGroup :: String -> Maybe ()
matchGroup String
g = [MatchTerm] -> String -> Maybe ()
matchTerms (String -> MatchTerm
MatchLiteral String
g MatchTerm -> [MatchTerm] -> [MatchTerm]
forall a. a -> [a] -> [a]
: [MatchTerm]
ts) String
cs
matchTerms [MatchTerm
MatchAny] String
_ = () -> Maybe ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
matchTerms (MatchTerm
MatchAny:[MatchTerm]
ts) String
cs = String -> Maybe String
forall (m :: * -> *). MonadFail m => String -> m String
matchAny String
cs Maybe String -> (String -> Maybe ()) -> Maybe ()
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= [MatchTerm] -> String -> Maybe ()
matchTerms [MatchTerm]
ts
where matchAny :: String -> m String
matchAny [] = String -> m String
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"no match"
matchAny String
cs' = case [MatchTerm] -> String -> Maybe ()
matchTerms [MatchTerm]
ts String
cs' of
Maybe ()
Nothing -> String -> m String
matchAny (String -> String
forall a. [a] -> [a]
tail String
cs')
Maybe ()
_ -> String -> m String
forall (m :: * -> *) a. Monad m => a -> m a
return String
cs'
matchTerms [MatchTerm
MatchDir] String
cs | Char
pathSeparator Char -> String -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` String
cs = String -> Maybe ()
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"path separator"
| Bool
otherwise = () -> Maybe ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
matchTerms (MatchTerm
MatchDir:[MatchTerm]
ts) String
cs = String -> Maybe String
forall (m :: * -> *). MonadFail m => String -> m String
matchDir String
cs Maybe String -> (String -> Maybe ()) -> Maybe ()
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= [MatchTerm] -> String -> Maybe ()
matchTerms [MatchTerm]
ts
where matchDir :: String -> m String
matchDir [] = String -> m String
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"no match"
matchDir (Char
c:String
_) | Char
c Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== Char
pathSeparator = String -> m String
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"path separator"
matchDir String
cs' = case [MatchTerm] -> String -> Maybe ()
matchTerms [MatchTerm]
ts String
cs' of
Maybe ()
Nothing -> String -> m String
matchDir (String -> m String) -> String -> m String
forall a b. (a -> b) -> a -> b
$ String -> String
forall a. [a] -> [a]
tail String
cs'
Maybe ()
_ -> String -> m String
forall (m :: * -> *) a. Monad m => a -> m a
return String
cs'
matchTerms (MatchTerm
MatchChar:[MatchTerm]
_) [] = String -> Maybe ()
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"end of input"
matchTerms (MatchTerm
MatchChar:[MatchTerm]
ts) (Char
_:String
cs) = [MatchTerm] -> String -> Maybe ()
matchTerms [MatchTerm]
ts String
cs
(~~) :: FilePath -> GlobPattern -> Bool
String
name ~~ :: String -> String -> Bool
~~ String
pat = let terms :: [MatchTerm]
terms = [MatchTerm] -> [MatchTerm]
simplifyTerms (String -> [MatchTerm]
parseGlob String
pat)
in (Maybe () -> Bool
forall a. Maybe a -> Bool
isJust (Maybe () -> Bool) -> (String -> Maybe ()) -> String -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. [MatchTerm] -> String -> Maybe ()
matchTerms [MatchTerm]
terms) String
name
(/~) :: FilePath -> GlobPattern -> Bool
/~ :: String -> String -> Bool
(/~) = (Bool -> Bool
not (Bool -> Bool) -> (String -> Bool) -> String -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ) ((String -> Bool) -> String -> Bool)
-> (String -> String -> Bool) -> String -> String -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> String -> Bool
(~~)