1 implementation module IRC
14 import Text.Parsers.Simple.Core
15 import Text.Parsers.Simple.Chars
18 import Control.Applicative
19 from Data.Functor import <$>
21 from Data.Func import $
22 from Text import class Text(ltrim,indexOf,concat), instance Text String
24 from StdMisc import undef
29 derive gPrint IRCCommand, IRCReplies, IRCErrors, (,), Maybe, (), Either
31 //Start = runParser parseMessage $ fromString ":frobnicator!~frobnicat@92.110.128.124 QUIT\r\n"
32 //Start = runParser parseMessage $ fromString ":frobnicator!~frobnicat@92.110.128.124 AWAY test\r\n"
33 //Start = runParser parseMessage $ fromString ":frobnicator!~frobnicat@92.110.128.124 AWAY :test with spaces\r\n"
34 //Start = runParser parseMessage $ fromString ":cherryh.freenode.net NOTICE * :*** Found your hostname\r\n"
35 //Start = runParser parseMessage $ fromString ":cherryh.freenode.net QUIT :hoi hoi\r\n"
36 Start = parseIRCMessage ":clooglebot!~cloogle@dhcp-077-249-221-037.chello.nl QUIT\r\n"
38 (<+) infixr 5 :: a b -> String | toString a & toString b
39 (<+) a b = toString a +++ toString b
41 parseIRCMessage :: String -> Either [Error] IRCMessage
42 parseIRCMessage s = case runParser parsePrefix (fromString s) of
43 ([(prefix, rest):_], _) = case parse parseReply rest of
44 Left e = case parseCmd rest of
46 Right cmd = Right {IRCMessage | irc_prefix=prefix, irc_command=Right cmd}
47 Right repl = Right {IRCMessage | irc_prefix=prefix, irc_command=Left repl}
48 (_, es) = Left ["couldn't parse prefix":es]
51 parseCmd :: [Char] -> Either Error IRCCommand
53 | not (trace_tn $ toString cs) = undef
54 = processParse $ argfun $ 'Text'.split " " $ toString cs
56 argfun :: [String] -> [String]
60 | x.[0] == ':' = ['Text'.join " " [x:xs]]
61 | otherwise = [x:argfun xs]
63 command0 :: IRCCommand [String] -> Either Error IRCCommand
64 command0 c [] = Right c
65 command0 c x = Left $ toString c +++ " doesn't have arguments"
67 processParse :: [String] -> Either Error IRCCommand
68 processParse [] = Left "Empty list of arguments"
69 processParse [cmd:args] = case cmd of
70 //"ADMIN" = (Maybe String)
72 //"CONNECT" = String (Maybe (Int, Maybe String))
73 "DIE" = command0 DIE args
75 //"INFO" = (Maybe String)
76 //"INVITE" = String String
78 //"JOIN" = [(String, Maybe String)]
79 //"KICK" = String String (Maybe String)
80 //"KILL" = String String
81 //"LINKS" = (Maybe (Maybe String, String))
82 //"LIST" = (Maybe ([String], Maybe String))
83 //"LUSERS" = (Maybe (String, Maybe String))
84 //"MODE" = String String (Maybe String) (Maybe String) (Maybe String)
85 //"MOTD" = (Maybe String)
87 //"NICK" = String (Maybe String)
88 "NJOIN" = command0 NJOIN args
89 //"NOTICE" = String String
90 //"OPER" = String String
93 //"PING" = String (Maybe String)
94 //"PONG" = String (Maybe String)
95 //"PRIVMSG" = [String] String
97 [_,_:_] = Left $ "QUIT has too many arguments"
98 x = Right $ QUIT $ listToMaybe x
99 "REHASH" = command0 REHASH args
100 "RESTART" = command0 REHASH args
101 "SERVER" = command0 REHASH args
102 //"SERVICE" = String String String String
103 //"SERVLIST" = (Maybe (String, Maybe String))
104 //"SQUERY" = String String
105 "SQUIRT" = command0 REHASH args
106 //"SQUIT" = String String
107 //"STATS" = (Maybe (String, Maybe String))
108 //"SUMMON" = String (Maybe (String, Maybe String))
109 //"TIME" = (Maybe String)
110 //"TOPIC" = String (Maybe String)
111 //"TRACE" = (Maybe String)
112 //"USER" = String String String
113 //"USERHOST" = [String]
114 //"USERS" = (Maybe String)
115 //"VERSION" = (Maybe String)
117 //"WHO" = (Maybe String)
118 //"WHOIS" = (Maybe String) [String]
119 //"WHOWAS" = (Maybe String) [String]
120 _ = Left $ "Unknown command: " +++ cmd
122 parsePrefix :: Parser Char (Maybe (Either String IRCUser))
123 parsePrefix = optional (parseEither parseHost parseUser) <* spaceParser
125 parseReply :: Parser Char IRCNumReply
126 parseReply = (toString <$> pSome pDigit)
128 >>= \recipient->spaceParser >>| (toString <$> pSome (pNoneOf illegal))
129 >>= \msg->pure {IRCNumReply|irc_reply=fs rep,irc_recipient=recipient,irc_message=msg}
131 fs :: String -> IRCReplies
132 fs s = fromInt $ toInt s
134 spaceParser :: Parser Char [Char]
135 spaceParser = pMany $ pToken ' '
137 parseServer :: Parser Char String
140 parseEither :: (Parser a b) (Parser a c) -> Parser a (Either b c)
141 parseEither p q = Left <$> p <|> Right <$> q
143 parseUser :: Parser Char IRCUser
144 parseUser = pToken ':' >>| parseNick
145 >>= \nick->optional (pToken '!' >>| parseUsr)
146 >>= \muser->optional (pToken '@' >>| parseHost)
147 >>= \mhost->pure {IRCUser | irc_nick=nick, irc_user=muser, irc_host=mhost}
149 parseUsr :: Parser Char String
150 parseUsr = toString <$> pSome (pNoneOf [' ', '@':illegal])
152 parseNick :: Parser Char String
153 parseNick = pAlpha >>= \c->pMany (pAlpha <|> pDigit <|> pSpecial)
154 >>= \cs->pure (toString [c:cs])
156 pSpecial :: Parser Char Char
157 pSpecial = pOneOf ['-', '[', ']', '\\', '\`', '^', '{', '}']
159 parseHost :: Parser Char String
160 parseHost = pToken ':' >>| (concat <$> pSepBy parseName (pToken '.'))
162 parseName :: Parser Char String
163 parseName = toString <$> pSome (pAlpha <|> pDigit <|> pOneOf ['-'])
165 instance toString IRCNumReply where
166 toString m = toInt m.irc_reply <+ " " <+ m.irc_recipient <+ " " <+ formatMSG m.irc_message
167 instance toString IRCMessage where
168 toString m = maybe "" (\s->either id ((<+) ":") s <+ " ") m.irc_prefix
169 <+ either toString toString m.irc_command
171 instance toString IRCUser where
172 toString m = m.irc_nick <+ maybe "" ((<+) "!") m.irc_user
173 <+ maybe "" ((<+) "@") m.irc_host
178 pMiddle :: Parser Char String
179 pMiddle = fmap toString $
180 spaceParser >>| liftM2 cons (pNotSatisfy ((==)':')) (pMany $ pNoneOf [' ':illegal])
182 pTrailing :: Parser Char String
183 pTrailing = fmap toString $
184 spaceParser >>| pToken ':' >>| pMany (pNoneOf illegal)
186 pParam :: Parser Char String
187 pParam = pMiddle <|> pTrailing
189 pNoneOf :: [a] -> Parser a a | Eq a
190 pNoneOf l = pSatisfy (not o flip isMember l)
192 pNotSatisfy :: (a -> Bool) -> Parser a a | Eq a
193 pNotSatisfy f = pSatisfy (not o f)
195 pInt :: Parser Char Int
196 pInt = toInt o toString <$> (spaceParser >>| pSome pDigit)
199 illegal = ['\x00','\r','\n']
201 instance toString IRCCommand where
202 toString r = jon " " (print r) +++ "\r\n"
204 print :: IRCCommand -> [String]
206 ADMIN mm = ["ADMIN":maybeToList mm]
208 //CONNECT String (Maybe (Int, Maybe String))
211 //INFO (Maybe String)
212 //INVITE String String
214 JOIN chs = ["JOIN",if (isEmpty chs) "0"
215 (jon ", " [jon " " [ch:maybeToList mk]\\(ch, mk)<-chs])]
216 //KICK String String (Maybe String)
218 //LINKS (Maybe (Maybe String, String))
219 //LIST (Maybe ([String], Maybe String))
220 //LUSERS (Maybe (String, Maybe String))
221 //MODE String String (Maybe String) (Maybe String) (Maybe String)
222 //MOTD (Maybe String)
224 NICK n ms = ["NICK", n]
226 //NOTICE String String
230 PING a mb = ["PING",a:maybeToList mb]
231 PONG a mb = ["PONG",a:maybeToList mb]
232 PRIVMSG dest msg = ["PRIVMSG",jon "," dest,formatMSG msg]
233 QUIT msg = ["QUIT":maybeToList msg]
237 //SERVICE String String String String
238 //SERVLIST (Maybe (String, Maybe String))
239 //SQUERY String String
241 //SQUIT String String
242 //STATS (Maybe (String, Maybe String))
243 //SUMMON String (Maybe (String, Maybe String))
244 //TIME (Maybe String)
245 //TOPIC String (Maybe String)
246 //TRACE (Maybe String)
247 USER login mode rn = ["USER", login, mode, "*", ":"+++rn]
249 //USERS (Maybe String)
250 //VERSION (Maybe String)
253 //WHOIS (Maybe String) [String]
254 //WHOWAS (Maybe String) [String]
255 _ = [printToString r]
257 formatMSG :: String -> String
258 formatMSG s = if (indexOf " " s > 0 || indexOf " " s > 0) (":" +++ s) s
261 instance toString IRCReplies where toString r = printToString r
262 instance toString IRCErrors where toString r = printToString r
264 instance fromInt IRCReplies where
265 fromInt r = case r of
272 201 = RPL_TRACECONNECTING
273 202 = RPL_TRACEHANDSHAKE
274 203 = RPL_TRACEUNKNOWN
275 204 = RPL_TRACEOPERATOR
277 206 = RPL_TRACESERVER
278 207 = RPL_TRACESERVICE
279 208 = RPL_TRACENEWTYPE
281 210 = RPL_TRACERECONNECT
282 211 = RPL_STATSLINKINFO
283 212 = RPL_STATSCOMMANDS
287 235 = RPL_SERVLISTEND
288 242 = RPL_STATSUPTIME
290 251 = RPL_LUSERCLIENT
292 253 = RPL_LUSERUNKNOWN
293 254 = RPL_LUSERCHANNELS
308 312 = RPL_WHOISSERVER
309 313 = RPL_WHOISOPERATOR
314 319 = RPL_WHOISCHANNELS
318 324 = RPL_CHANNELMODEIS
325 347 = RPL_ENDOFINVITELIST
327 349 = RPL_ENDOFEXCEPTLIST
335 368 = RPL_ENDOFBANLIST
336 369 = RPL_ENDOFWHOWAS
344 383 = RPL_YOURESERVICE
352 instance toInt IRCReplies where
360 RPL_TRACECONNECTING = 201
361 RPL_TRACEHANDSHAKE = 202
362 RPL_TRACEUNKNOWN = 203
363 RPL_TRACEOPERATOR = 204
365 RPL_TRACESERVER = 206
366 RPL_TRACESERVICE = 207
367 RPL_TRACENEWTYPE = 208
369 RPL_TRACERECONNECT = 210
370 RPL_STATSLINKINFO = 211
371 RPL_STATSCOMMANDS = 212
375 RPL_SERVLISTEND = 234
376 RPL_STATSUPTIME = 242
378 RPL_LUSERCLIENT = 251
380 RPL_LUSERUNKNOWN = 253
381 RPL_LUSERCHANNELS = 254
396 RPL_WHOISSERVER = 312
397 RPL_WHOISOPERATOR = 313
402 RPL_WHOISCHANNELS = 319
406 RPL_CHANNELMODEIS = 324
413 RPL_ENDOFINVITELIST = 347
415 RPL_ENDOFEXCEPTLIST = 349
423 RPL_ENDOFBANLIST = 367
424 RPL_ENDOFWHOWAS = 369
432 RPL_YOURESERVICE = 383
439 instance fromInt IRCErrors where
440 fromInt r = case r of
442 402 = ERR_NOSUCHSERVER
443 403 = ERR_NOSUCHCHANNEL
444 404 = ERR_CANNOTSENDTOCHAN
445 405 = ERR_TOOMANYCHANNELS
446 406 = ERR_WASNOSUCHNICK
447 407 = ERR_TOOMANYTARGETS
448 408 = ERR_NOSUCHSERVICE
450 411 = ERR_NORECIPIENT
451 412 = ERR_NOTEXTTOSEND
453 414 = ERR_WILDTOPLEVEL
455 421 = ERR_UNKNOWNCOMMAND
457 423 = ERR_NOADMININFO
459 431 = ERR_NONICKNAMEGIVEN
460 432 = ERR_ERRONEUSNICKNAME
461 433 = ERR_NICKNAMEINUSE
462 436 = ERR_NICKCOLLISION
463 437 = ERR_UNAVAILRESOURCE
464 441 = ERR_USERNOTINCHANNEL
465 442 = ERR_NOTONCHANNEL
466 443 = ERR_USERONCHANNEL
468 445 = ERR_SUMMONDISABLED
469 446 = ERR_USERSDISABLED
470 451 = ERR_NOTREGISTERED
471 461 = ERR_NEEDMOREPARAMS
472 462 = ERR_ALREADYREGISTRED
473 463 = ERR_NOPERMFORHOST
474 464 = ERR_PASSWDMISMATCH
475 465 = ERR_YOUREBANNEDCREEP
476 466 = ERR_YOUWILLBEBANNED
478 471 = ERR_CHANNELISFULL
479 472 = ERR_UNKNOWNMODE
480 473 = ERR_INVITEONLYCHAN
481 474 = ERR_BANNEDFROMCHAN
482 475 = ERR_BADCHANNELKEY
483 476 = ERR_BADCHANMASK
484 477 = ERR_NOCHANMODES
485 478 = ERR_BANLISTFULL
486 481 = ERR_NOPRIVILEGES
487 482 = ERR_CHANOPRIVSNEEDED
488 483 = ERR_CANTKILLSERVER
490 485 = ERR_UNIQOPPRIVSNEEDED
492 501 = ERR_UMODEUNKNOWNFLAG
493 502 = ERR_USERSDONTMATCH
495 instance toInt IRCErrors where
498 ERR_NOSUCHSERVER = 402
499 ERR_NOSUCHCHANNEL = 403
500 ERR_CANNOTSENDTOCHAN = 404
501 ERR_TOOMANYCHANNELS = 405
502 ERR_WASNOSUCHNICK = 406
503 ERR_TOOMANYTARGETS = 407
504 ERR_NOSUCHSERVICE = 408
506 ERR_NORECIPIENT = 411
507 ERR_NOTEXTTOSEND = 412
509 ERR_WILDTOPLEVEL = 414
511 ERR_UNKNOWNCOMMAND = 421
513 ERR_NOADMININFO = 423
515 ERR_NONICKNAMEGIVEN = 431
516 ERR_ERRONEUSNICKNAME = 432
517 ERR_NICKNAMEINUSE = 433
518 ERR_NICKCOLLISION = 436
519 ERR_UNAVAILRESOURCE = 437
520 ERR_USERNOTINCHANNEL = 441
521 ERR_NOTONCHANNEL = 442
522 ERR_USERONCHANNEL = 443
524 ERR_SUMMONDISABLED = 445
525 ERR_USERSDISABLED = 446
526 ERR_NOTREGISTERED = 451
527 ERR_NEEDMOREPARAMS = 461
528 ERR_ALREADYREGISTRED = 462
529 ERR_NOPERMFORHOST = 463
530 ERR_PASSWDMISMATCH = 464
531 ERR_YOUREBANNEDCREEP = 465
532 ERR_YOUWILLBEBANNED = 466
534 ERR_CHANNELISFULL = 471
535 ERR_UNKNOWNMODE = 472
536 ERR_INVITEONLYCHAN = 473
537 ERR_BANNEDFROMCHAN = 474
538 ERR_BADCHANNELKEY = 475
539 ERR_BADCHANMASK = 476
540 ERR_NOCHANMODES = 477
541 ERR_BANLISTFULL = 478
542 ERR_NOPRIVILEGES = 481
543 ERR_CHANOPRIVSNEEDED = 482
544 ERR_CANTKILLSERVER = 483
546 ERR_UNIQOPPRIVSNEEDED = 485
548 ERR_UMODEUNKNOWNFLAG = 501
549 ERR_USERSDONTMATCH = 502