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
|
{-# LANGUAGE NoImplicitPrelude #-}
{- |
Module : Text.Pandoc.Writers.Roff
Copyright : Copyright (C) 2007-2019 John MacFarlane
License : GNU GPL, version 2 or above
Maintainer : John MacFarlane <jgm@berkeley.edu>
Stability : alpha
Portability : portable
Common functions for roff writers (man, ms).
-}
module Text.Pandoc.Writers.Roff (
WriterState(..)
, defaultWriterState
, MS
, Note
, EscapeMode(..)
, escapeString
, withFontFeature
) where
import Prelude
import Data.Char (ord, isAscii)
import Control.Monad.State.Strict
import qualified Data.Map as Map
import Data.Maybe (fromMaybe, isJust, catMaybes)
import Text.Pandoc.Class (PandocMonad)
import Text.Pandoc.Definition
import Text.Pandoc.Pretty
import Text.Printf (printf)
import Text.Pandoc.RoffChar (standardEscapes,
characterCodes, combiningAccents)
data WriterState = WriterState { stHasInlineMath :: Bool
, stFirstPara :: Bool
, stNotes :: [Note]
, stSmallCaps :: Bool
, stHighlighting :: Bool
, stInHeader :: Bool
, stFontFeatures :: Map.Map Char Bool
, stHasTables :: Bool
}
defaultWriterState :: WriterState
defaultWriterState = WriterState{ stHasInlineMath = False
, stFirstPara = True
, stNotes = []
, stSmallCaps = False
, stHighlighting = False
, stInHeader = False
, stFontFeatures = Map.fromList [
('I',False)
, ('B',False)
, ('C',False)
]
, stHasTables = False
}
type Note = [Block]
type MS = StateT WriterState
data EscapeMode = AllowUTF8 -- ^ use preferred man escapes
| AsciiOnly -- ^ escape everything
deriving Show
combiningAccentsMap :: Map.Map Char String
combiningAccentsMap = Map.fromList combiningAccents
essentialEscapes :: Map.Map Char String
essentialEscapes = Map.fromList standardEscapes
-- | Escape special characters for roff.
escapeString :: EscapeMode -> String -> String
escapeString _ [] = []
escapeString escapeMode ('\n':'.':xs) =
'\n':'\\':'&':'.':escapeString escapeMode xs
escapeString escapeMode (x:xs) =
case Map.lookup x essentialEscapes of
Just s -> s ++ escapeString escapeMode xs
Nothing
| isAscii x -> x : escapeString escapeMode xs
| otherwise ->
case escapeMode of
AllowUTF8 -> x : escapeString escapeMode xs
AsciiOnly ->
let accents = catMaybes $ takeWhile isJust
(map (\c -> Map.lookup c combiningAccentsMap) xs)
rest = drop (length accents) xs
s = case Map.lookup x characterCodeMap of
Just t -> "\\[" <> unwords (t:accents) <> "]"
Nothing -> "\\[" <> unwords
(printf "u%04X" (ord x) : accents) <> "]"
in s ++ escapeString escapeMode rest
characterCodeMap :: Map.Map Char String
characterCodeMap = Map.fromList characterCodes
fontChange :: PandocMonad m => MS m Doc
fontChange = do
features <- gets stFontFeatures
inHeader <- gets stInHeader
let filling = ['C' | fromMaybe False $ Map.lookup 'C' features] ++
['B' | inHeader ||
fromMaybe False (Map.lookup 'B' features)] ++
['I' | fromMaybe False $ Map.lookup 'I' features]
return $
if null filling
then text "\\f[R]"
else text $ "\\f[" ++ filling ++ "]"
withFontFeature :: PandocMonad m => Char -> MS m Doc -> MS m Doc
withFontFeature c action = do
modify $ \st -> st{ stFontFeatures = Map.adjust not c $ stFontFeatures st }
begin <- fontChange
d <- action
modify $ \st -> st{ stFontFeatures = Map.adjust not c $ stFontFeatures st }
end <- fontChange
return $ begin <> d <> end
|