blob: 2e3d6db957b81a6af9b6cc3d0ebccde2b91fb77c (
plain)
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
|
module Text.Pandoc.Readers.Docx.StyleMap ( StyleMap
, ParaStyleMap
, CharStyleMap
, StyleMaps(..)
, defaultStyleMaps
, getStyleMaps
, getStyleId
, hasStyleName
) where
import Text.XML.Light
import Text.Pandoc.Readers.Docx.Util
import Control.Monad.State
import Data.Char (toLower)
import Data.Maybe (fromMaybe)
import qualified Data.Map as M
newtype ParaStyleMap = ParaStyleMap ( M.Map String String )
newtype CharStyleMap = CharStyleMap ( M.Map String String )
class StyleMap a where
alterMap :: (M.Map String String -> M.Map String String) -> a -> a
getMap :: a -> M.Map String String
instance StyleMap ParaStyleMap where
alterMap f (ParaStyleMap m) = ParaStyleMap $ f m
getMap (ParaStyleMap m) = m
instance StyleMap CharStyleMap where
alterMap f (CharStyleMap m) = CharStyleMap $ f m
getMap (CharStyleMap m) = m
insert :: (StyleMap a) => String -> String -> a -> a
insert k v = alterMap $ M.insert k v
getStyleId :: (StyleMap a) => String -> a -> String
getStyleId s = M.findWithDefault (filter (/=' ') s) (map toLower s) . getMap
hasStyleName :: (StyleMap a) => String -> a -> Bool
hasStyleName styleName = M.member (map toLower styleName) . getMap
data StyleMaps = StyleMaps { sNameSpaces :: NameSpaces
, sParaStyleMap :: ParaStyleMap
, sCharStyleMap :: CharStyleMap
}
data StyleType = ParaStyle | CharStyle
defaultStyleMaps :: StyleMaps
defaultStyleMaps = StyleMaps { sNameSpaces = []
, sParaStyleMap = ParaStyleMap M.empty
, sCharStyleMap = CharStyleMap M.empty
}
type StateM a = StateT StyleMaps Maybe a
getStyleMaps :: Element -> StyleMaps
getStyleMaps docElem = fromMaybe state' $ execStateT genStyleMap state'
where
state' = defaultStyleMaps {sNameSpaces = elemToNameSpaces docElem}
insertPara key val = modify $ \s ->
s { sParaStyleMap = insert key val $ sParaStyleMap s }
insertChar key val = modify $ \s ->
s { sCharStyleMap = insert key val $ sCharStyleMap s }
genStyleItem e = do
styleType <- getStyleType e
nameVal <- getNameVal e
styleId <- getAttrStyleId e
let nameValLC = map toLower nameVal
case styleType of
ParaStyle -> insertPara nameValLC styleId
CharStyle -> insertChar nameValLC styleId
genStyleMap = do
style <- elemName' "style"
let styles = findChildren style docElem
forM_ styles genStyleItem
getStyleType :: Element -> StateM StyleType
getStyleType e = do
styleTypeStr <- getAttrType e
case styleTypeStr of
"paragraph" -> return ParaStyle
"character" -> return CharStyle
_ -> lift Nothing
getAttrType :: Element -> StateM String
getAttrType el = do
name <- elemName' "type"
lift $ findAttr name el
getAttrStyleId :: Element -> StateM String
getAttrStyleId el = do
name <- elemName' "styleId"
lift $ findAttr name el
getNameVal :: Element -> StateM String
getNameVal el = do
name <- elemName' "name"
val <- elemName' "val"
lift $ findChild name el >>= findAttr val
elemName' :: String -> StateM QName
elemName' name = do
namespaces <- gets sNameSpaces
return $ elemName namespaces "w" name
|
