1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
-- | Module providing pattern matching and capturing on 'Identifier's.
--
-- A very simple pattern could be, for example:
--
-- > foo/bar
--
-- This pattern will only match the @foo\/bar@ identifier.
--
-- To match more than one identifier, there are different captures that one can
-- use:
--
-- * @*@: matches exactly one element of an identifier;
--
-- * @**@: matches one or more elements of an identifier.
--
-- Some examples:
--
-- * @foo\/*@ will match @foo\/bar@ and @foo\/foo@, but not @foo\/bar\/qux@ nor
-- @foo@;
--
-- * @**@ will match any non-empty identifier;
--
-- * @foo\/**@ will match @foo\/bar@ and @foo\/bar\/qux@, but not @bar\/foo@ nor
-- @foo@;
--
-- A small warning: patterns are not globs. Using @foo\/*.markdown@ will not do
-- what you probably intended, as it will only match the file which is literally
-- called @foo\/*.markdown@. Remember that these captures only work on elements
-- of identifiers as a whole; not on parts of these elements.
--
-- Furthermore, the 'match' function allows the user to get access to the
-- elements captured by the capture elements in the pattern.
--
module Hakyll.Core.Identifier.Pattern
( Pattern
, parsePattern
, match
, doesMatch
, matches
) where
import Data.List (intercalate)
import Control.Monad (msum)
import Data.Maybe (isJust)
import GHC.Exts (IsString, fromString)
import Hakyll.Core.Identifier
-- | One base element of a pattern
--
data PatternComponent = CaptureOne
| CaptureMany
| Literal String
deriving (Eq)
instance Show PatternComponent where
show CaptureOne = "*"
show CaptureMany = "**"
show (Literal s) = s
-- | Type that allows matching on identifiers
--
newtype Pattern = Pattern {unPattern :: [PatternComponent]}
deriving (Eq)
instance Show Pattern where
show = intercalate "/" . map show . unPattern
instance IsString Pattern where
fromString = parsePattern
-- | Parse a pattern from a string
--
parsePattern :: String -> Pattern
parsePattern = Pattern . map toPattern . unIdentifier . parseIdentifier
where
toPattern x | x == "*" = CaptureOne
| x == "**" = CaptureMany
| otherwise = Literal x
-- | Match an identifier against a pattern, generating a list of captures
--
match :: Pattern -> Identifier -> Maybe [[String]]
match (Pattern p) (Identifier i) = match' p i
-- | Check if an identifier matches a pattern
--
doesMatch :: Pattern -> Identifier -> Bool
doesMatch p = isJust . match p
-- | Given a list of identifiers, retain only those who match the given pattern
--
matches :: Pattern -> [Identifier] -> [Identifier]
matches p = filter (doesMatch p)
-- | Split a list at every possible point, generate a list of (init, tail) cases
--
splits :: [a] -> [([a], [a])]
splits ls = reverse $ splits' [] ls
where
splits' lx ly = (lx, ly) : case ly of
[] -> []
(y : ys) -> splits' (lx ++ [y]) ys
-- | Internal verion of 'match'
--
match' :: [PatternComponent] -> [String] -> Maybe [[String]]
match' [] [] = Just [] -- An empty match
match' [] _ = Nothing -- No match
match' _ [] = Nothing -- No match
match' (m : ms) (s : ss) = case m of
-- Take one string and one literal, fail on mismatch
Literal l -> if s == l then match' ms ss else Nothing
-- Take one string and one capture
CaptureOne -> fmap ([s] :) $ match' ms ss
-- Take one string, and one or many captures
CaptureMany ->
let take' (i, t) = fmap (i :) $ match' ms t
in msum $ map take' $ splits (s : ss)
|