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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
|
{-
Code generator for function documentation tables
(Function / Return Type / Description / Example / Result).
Usage: runghc doc-to-sgml.hs < documentation > out
-}
import Control.Applicative
import Data.Char
import Data.List
import Data.Maybe
trim :: String -> String
trim = f . f
where f = reverse . dropWhile isSpace
list_find :: (Eq a) => [a] -> [a] -> Maybe Int
list_find [] _ = error "list_find: Empty search string"
list_find needle@(x:xs) haystack = f 0 haystack where
len = length needle - 1
f i [] = Nothing
f i (y:ys) =
if x == y && (xs == take len ys)
then Just i
else f (i+1) ys
explode :: (Eq a) => [a] -> [a] -> [[a]]
explode delim str = f str where
len = length delim
f str = fromMaybe [str] $ do
pos <- list_find delim str
let (a,b) = splitAt pos str
return (a : f (drop len b))
data Line = Line Int Int String
deriving (Show)
subLine :: Line -> String -> Line
subLine (Line number indent text) newtext = Line number indent newtext
line_no :: Line -> Int
line_no (Line n _ _) = n
line_indent :: Line -> Int
line_indent (Line _ n _) = n
line_str :: Line -> String
line_str (Line _ _ str) = str
line_error :: String -> Line -> a
line_error info line = error (info ++ ": error at line " ++ show (line_no line))
line_split :: Line -> [String] -> [String]
line_split line splitters = f (line_str line) splitters where
f str [] = [str]
f str (x:xs) = a : f (drop (length x) b) xs where
(a,b) = splitAt (fromMaybe err $ list_find x str) str
err = line_error ("line_split " ++ show (line_str line) ++ " " ++ show splitters) line
lines' :: String -> [Line]
lines' str = filter noblank $ zipWith mkLine [1..] (lines str)
where
noblank line = trim (line_str line) /= ""
mkLine num line = Line num (indentOf line) (trim line)
indentOf line = length $ takeWhile (== '\t') line
data DocEntry = DocEntry {
function :: String,
returns :: Type,
description :: String,
examples :: [(String, Result)]
} deriving (Show)
data Type = Type String | Enum String [String]
deriving (Show)
data Result = Result String | ResultSet [String]
deriving (Show)
entriesToSgml :: String -> String -> [DocEntry] -> String
entriesToSgml table_id title entries =
" <table id=\"" ++ table_id ++ "\">\n" ++
" <title>" ++ title ++ "</title>\n" ++
"\n" ++
" <tgroup cols=\"5\">\n" ++
" <thead>\n" ++
" <row>\n" ++
" <entry>Function</entry>\n" ++
" <entry>Return Type</entry>\n" ++
" <entry>Description</entry>\n" ++
" <entry>Example</entry>\n" ++
" <entry>Result</entry>\n" ++
" </row>\n" ++
" </thead>\n" ++
"\n" ++
" <tbody>\n" ++
concatMap entryToSgml entries ++
" </tbody>\n" ++
" </tgroup>\n" ++
" </table>"
typeToSgml :: Type -> String
typeToSgml (Type t) = "<type>" ++ t ++ "</type>"
typeToSgml (Enum t xs) =
"<type>" ++ t ++ "</type>" ++
" - one of:\n" ++
"<programlisting>\n" ++
unlines xs ++
"</programlisting>\n" ++
entryIndent ""
resultToSgml :: Result -> String
resultToSgml (Result str) = "<literal>" ++ str ++ "</literal>"
resultToSgml (ResultSet strs) =
"\n<programlisting>\n" ++
(unlines . map (" " ++)) strs ++
"(" ++ show len ++ (if len == 1 then " row" else " rows") ++ ")\n" ++
"</programlisting>\n" ++ entryIndent ""
where
len = length strs
rowIndent :: String -> String
rowIndent x = " " ++ x
entryIndent :: String -> String
entryIndent x = " " ++ x
exampleToSgml :: Maybe (String, Result) -> String
exampleToSgml (Just (example, result))
= entryIndent "<entry><literal>" ++ example ++ "</literal></entry>\n" ++
entryIndent "<entry>" ++ resultToSgml result ++ "</entry>\n"
exampleToSgml Nothing
= entryIndent "<entry></entry>\n" ++
entryIndent "<entry></entry>\n"
examplesToSgml :: [(String, Result)] -> String
examplesToSgml xs = do
x <- xs
rowIndent "<row>\n" ++ (exampleToSgml (Just x)) ++ rowIndent "</row>\n"
entryToSgml :: DocEntry -> String
entryToSgml ent =
rowIndent "<row>\n" ++
entry ("<function>" ++ function ent ++ "</function>") ++
entry (typeToSgml (returns ent)) ++
entry (description ent) ++
exampleToSgml (if exlen > 0 then Just ex else Nothing) ++
rowIndent "</row>\n" ++
(if exlen > 1 then examplesToSgml exs else "")
where
(ex:exs) = examples ent
exlen = length (examples ent)
entry x = entryIndent
(if exlen > 1
then "<entry morerows=\"" ++ show (exlen - 1) ++ "\">"
else "<entry>") ++ x ++ "</entry>\n"
parseExamples :: [Line] -> [(String, Result)]
parseExamples [] = []
parseExamples (x:xs) =
if line_indent x /= 1
then line_error "parseExamples" x
else if result == ""
then (example, ResultSet (map line_str set)) : parseExamples after_set
else (example, Result result) : parseExamples xs
where
[example, result] = map trim $ line_split x ["=>"]
[set, after_set] = [takeWhile f xs, dropWhile f xs]
f = (>= 2) . line_indent
parseType :: Line -> Type
parseType line =
if elem '{' str
then if c /= ""
then line_error "parseType" line
else Enum a (map trim $ explode "|" b)
else Type str
where
str = line_str line
[a,b,c] = map trim $ line_split line ["{", "}"]
parseDocEntry :: [Line] -> (DocEntry, [Line])
parseDocEntry (x:xs) =
if line_indent x /= 0
then line_error "parseDocEntry" x
else (DocEntry {
function = a,
returns = parseType (subLine x b),
description = c,
examples = parseExamples xlines
}, rest)
where
[a,b,c] = map trim $ line_split x ["->", ":"]
f = (>= 1) . line_indent
xlines = takeWhile f xs
rest = dropWhile f xs
parseDocEntries :: String -> [DocEntry]
parseDocEntries str = f (lines' str) where
f [] = []
f lines = docEntry : f rest where
(docEntry, rest) = parseDocEntry lines
main = do
entries <- parseDocEntries <$> getContents
--putStr $ unlines $ map show entries
putStr $ entriesToSgml "json-func-table" "<type>json</type> Functions" entries
|