|
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 |
|
|
10904
|
54 |
buildParseTree cmds = if isJust emptyNamed then cmdLeaf $ fromJust emptyNamed else subtree
|
|
|
55 |
where
|
|
|
56 |
emptyNamed = find (\(_, (Command n _:_)) -> null n) assocs
|
|
|
57 |
assocs = groupByFirstChar cmds
|
|
10906
|
58 |
subtree = map buildsub assocs
|
|
|
59 |
buildsub (c, cmds) = let st = buildParseTree cmds in if null $ drop 1 st then maybeMerge c st else PTPrefix [c] st
|
|
|
60 |
maybeMerge c cmd@[PTCommand _] = PTPrefix [c] cmd
|
|
|
61 |
maybeMerge c cmd@[PTPrefix s ss] = PTPrefix (c:s) ss
|
|
|
62 |
cmdLeaf (c, (hwc:_)) = [PTPrefix [c] [PTCommand hwc]]
|
|
|
63 |
|
|
|
64 |
dumpTree = vcat . map dt
|
|
|
65 |
where
|
|
|
66 |
dt (PTPrefix s st) = text s $$ (nest 1 $ vcat $ map dt st)
|
|
|
67 |
dt _ = empty
|
|
10904
|
68 |
|
|
10908
|
69 |
pas2 = buildSwitch $ buildParseTree commands
|
|
10906
|
70 |
where
|
|
|
71 |
buildSwitch cmds = text "case getNextChar of" $$ (nest 4 . vcat $ map buildCase cmds) $$ elsePart
|
|
|
72 |
buildCase (PTCommand _ ) = text "#10: <call cmd handler>;"
|
|
|
73 |
buildCase (PTPrefix (s:ss) cmds) = quotes (char s) <> text ": " <> consumePrefix ss (buildSwitch cmds)
|
|
|
74 |
consumePrefix "" = id
|
|
|
75 |
consumePrefix str = (text "consume" <> (parens . quotes $ text str) <> semi $$)
|
|
|
76 |
zeroChar = text "#0: state:= pstDisconnected;"
|
|
|
77 |
elsePart = text "else <unknown cmd> end;"
|
|
10904
|
78 |
|
|
10908
|
79 |
pas = text $ show $ buildTables $ buildParseTree commands
|
|
|
80 |
where
|
|
|
81 |
buildTables cmds = let (_, _, t1, t2) = foldl walk (0, 0, [], []) cmds in (reverse t1, reverse t2)
|
|
|
82 |
walk (lc, cc, tbl1, tbl2) (PTCommand _ ) = (lc, cc + 1, ("#10"):tbl1, (show $ -10 - cc):(tbl2))
|
|
|
83 |
walk lct (PTPrefix prefix cmds) = foldl walk (foldl fpf lct prefix) cmds
|
|
|
84 |
fpf (lc, cc, tbl1, tbl2) c = (lc + 1, cc, [c]:tbl1, (show lc):tbl2)
|
|
|
85 |
|
|
10900
|
86 |
main = putStrLn $ render pas
|