|
10898
|
1 |
module Main where
|
|
|
2 |
|
|
|
3 |
import Text.PrettyPrint.HughesPJ
|
|
10904
|
4 |
import qualified Data.MultiMap as MM
|
|
|
5 |
import Data.Maybe
|
|
|
6 |
import Data.List
|
|
10898
|
7 |
|
|
|
8 |
data HWProtocol = Command String [CmdParam]
|
|
|
9 |
data CmdParam = Skip
|
|
|
10 |
| SS
|
|
|
11 |
| LS
|
|
|
12 |
| IntP
|
|
|
13 |
| Many [CmdParam]
|
|
|
14 |
data ClientStates = NotConnected
|
|
|
15 |
| JustConnected
|
|
|
16 |
| ServerAuth
|
|
|
17 |
| Lobby
|
|
|
18 |
|
|
10906
|
19 |
data ParseTree = PTPrefix String [ParseTree]
|
|
10902
|
20 |
| PTCommand HWProtocol
|
|
|
21 |
|
|
10898
|
22 |
cmd = Command
|
|
|
23 |
cmd1 s p = Command s [p]
|
|
|
24 |
cmd2 s p1 p2 = Command s [p1, p2]
|
|
|
25 |
|
|
10902
|
26 |
breakCmd (Command (c:cs) params) = (c, Command cs params)
|
|
|
27 |
|
|
10898
|
28 |
commands = [
|
|
|
29 |
cmd "CONNECTED" [Skip, IntP]
|
|
|
30 |
, cmd1 "NICK" SS
|
|
|
31 |
, cmd1 "PROTO" IntP
|
|
|
32 |
, cmd1 "ASKPASSWORD" SS
|
|
|
33 |
, cmd1 "SERVER_AUTH" SS
|
|
10904
|
34 |
, cmd1 "JOINING" SS
|
|
|
35 |
, cmd1 "BANLIST" $ Many [SS]
|
|
|
36 |
, cmd1 "JOINED" $ Many [SS]
|
|
10898
|
37 |
, cmd1 "LOBBY:JOINED" $ Many [SS]
|
|
10904
|
38 |
, cmd2 "LOBBY:LEFT" SS LS
|
|
|
39 |
, cmd2 "CLIENT_FLAGS" SS $ Many [SS]
|
|
|
40 |
, cmd2 "LEFT" SS $ Many [SS]
|
|
10902
|
41 |
, cmd1 "SERVER_MESSAGE" LS
|
|
10904
|
42 |
, cmd1 "EM" $ Many [LS]
|
|
|
43 |
, cmd1 "PING" $ Many [SS]
|
|
|
44 |
, cmd2 "CHAT" SS LS
|
|
|
45 |
, cmd2 "SERVER_VARS" SS LS
|
|
|
46 |
, cmd2 "BYE" SS LS
|
|
10908
|
47 |
, cmd1 "INFO" $ Many [SS]
|
|
10904
|
48 |
, cmd "KICKED" []
|
|
10898
|
49 |
]
|
|
|
50 |
|
|
10904
|
51 |
groupByFirstChar :: [HWProtocol] -> [(Char, [HWProtocol])]
|
|
|
52 |
groupByFirstChar = MM.assocs . MM.fromList . map breakCmd
|
|
10902
|
53 |
|
|
10925
|
54 |
buildParseTree cmds = [PTPrefix "!" $ bpt cmds]
|
|
|
55 |
bpt cmds = if isJust emptyNamed then cmdLeaf $ fromJust emptyNamed else subtree
|
|
10904
|
56 |
where
|
|
|
57 |
emptyNamed = find (\(_, (Command n _:_)) -> null n) assocs
|
|
|
58 |
assocs = groupByFirstChar cmds
|
|
10906
|
59 |
subtree = map buildsub assocs
|
|
10925
|
60 |
buildsub (c, cmds) = let st = bpt cmds in if null $ drop 1 st then maybeMerge c st else PTPrefix [c] st
|
|
10906
|
61 |
maybeMerge c cmd@[PTCommand _] = PTPrefix [c] cmd
|
|
|
62 |
maybeMerge c cmd@[PTPrefix s ss] = PTPrefix (c:s) ss
|
|
|
63 |
cmdLeaf (c, (hwc:_)) = [PTPrefix [c] [PTCommand hwc]]
|
|
|
64 |
|
|
|
65 |
dumpTree = vcat . map dt
|
|
|
66 |
where
|
|
|
67 |
dt (PTPrefix s st) = text s $$ (nest 1 $ vcat $ map dt st)
|
|
|
68 |
dt _ = empty
|
|
10904
|
69 |
|
|
10908
|
70 |
pas2 = buildSwitch $ buildParseTree commands
|
|
10906
|
71 |
where
|
|
|
72 |
buildSwitch cmds = text "case getNextChar of" $$ (nest 4 . vcat $ map buildCase cmds) $$ elsePart
|
|
|
73 |
buildCase (PTCommand _ ) = text "#10: <call cmd handler>;"
|
|
|
74 |
buildCase (PTPrefix (s:ss) cmds) = quotes (char s) <> text ": " <> consumePrefix ss (buildSwitch cmds)
|
|
|
75 |
consumePrefix "" = id
|
|
|
76 |
consumePrefix str = (text "consume" <> (parens . quotes $ text str) <> semi $$)
|
|
|
77 |
zeroChar = text "#0: state:= pstDisconnected;"
|
|
|
78 |
elsePart = text "else <unknown cmd> end;"
|
|
10904
|
79 |
|
|
10925
|
80 |
renderArrays letters commands = l $+$ s
|
|
10908
|
81 |
where
|
|
10925
|
82 |
maybeQuotes s = if null $ tail s then quotes $ text s else text s
|
|
|
83 |
l = text "const letters: array[0.." <> (int $ length letters - 1) <> text "] of char = "
|
|
|
84 |
<> parens (hsep . punctuate comma $ map maybeQuotes letters) <> semi
|
|
|
85 |
s = text "const commands: array[0.." <> (int $ length commands - 1) <> text "] of integer = "
|
|
|
86 |
<> parens (hsep . punctuate comma $ map text commands) <> semi
|
|
10908
|
87 |
|
|
10925
|
88 |
pas = uncurry renderArrays $ buildTables $ buildParseTree commands
|
|
|
89 |
where
|
|
|
90 |
buildTables cmds = let (_, _, _, t1, t2) = foldr walk (0, [0], -10, [], []) cmds in (tail t1, tail t2)
|
|
|
91 |
walk (PTCommand _ ) (lc, s:sh, pc, tbl1, tbl2) = (lc, 2:sh, pc - 1, "#10":"0":tbl1, "0":(show pc):tbl2)
|
|
|
92 |
walk (PTPrefix prefix cmds) l = lvldown $ foldr fpf (foldr walk (lvlup l) cmds) prefix
|
|
|
93 |
lvlup (lc, sh, pc, tbl1, tbl2) = (lc, 0:sh, pc, tbl1, tbl2)
|
|
|
94 |
lvldown (lc, s1:s2:sh, pc, tbl1, t:tbl2) = (lc, s1+s2:sh, pc, tbl1, show s1:tbl2)
|
|
|
95 |
fpf c (lc, s:sh, pc, tbl1, tbl2) = (lc + 1, s+1:sh, pc, [c]:tbl1, "0":tbl2)
|
|
|
96 |
|
|
|
97 |
main = putStrLn $ renderStyle style{lineLength = 80} pas
|