1 implementation module IRC
3 import StdList, StdTuple, StdOverloaded, StdFunc, StdString, StdChar, StdBool
9 import Control.Applicative
14 import Text.Parsers.Simple.Chars
15 import Text.Parsers.Simple.Core
19 from Data.Functor import <$>
20 from Data.Func import $
21 from StdMisc import undef, abort
22 from Text import class Text(lpad,trim,rtrim,split,indexOf,concat),
28 derive gPrint IRCErrors, IRCReplies, Maybe, Either, IRCUser, IRCNumReply
30 Start = (map (fmap toString) msgs, msgs)
33 [ parseIRCMessage ":clooglebot!~cloogle@dhcp-077-249-221-037.chello.nl QUIT\r\n"
34 , parseIRCMessage ":clooglebot!~cloogle QUIT\r\n"
35 , parseIRCMessage ":frobnicator!~frobnicat@92.110.128.124 QUIT\r\n"
36 , parseIRCMessage ":frobnicator!~frobnicat@92.110.128.124 AWAY test\r\n"
37 , parseIRCMessage ":frobnicator!~frobnicat@92.110.128.124 AWAY :test with spaces\r\n"
38 , parseIRCMessage ":cherryh.freenode.net NOTICE * :*** Found your hostname\r\n"
39 , parseIRCMessage ":cherryh.freenode.net QUIT :hoi hoi\r\n"
40 , parseIRCMessage ":cherryh.freenode.net JOIN #cha,#ch-b #twilight\r\n"
41 , parseIRCMessage ":cherryh.freenode.net ISON a b c d e f :g h\r\n"
42 , parseIRCMessage ":wilhelm.freenode.net 001 clooglebot :Welcome to the freenode Internet Relay Chat Network clooglebot\r\n"
43 , parseIRCMessage "PING :orwell.freenode.net\r\n"
44 , parseIRCMessage ":ChanServ!ChanServ@services. MODE #cloogle +o frobnicator\r\n"
45 , parseIRCMessage ":qbot_v01!~qbot@ip-213-124-170-20.ip.prioritytelecom.net PRIVMSG ##chinees :[link] Cloogle - https://cloogle.org"
48 parseIRCMessage :: String -> Either [Error] IRCMessage
49 parseIRCMessage s = case runParser parsePrefix (fromString s) of
51 ([(prefix, rest):_], _)
52 //Try parsing a numeric reply
53 = case parse parseReply rest of
54 //Try a normal command
55 Left e = case parseCmd rest of
57 Right cmd = Right {IRCMessage | irc_prefix=prefix, irc_command=Right cmd}
58 Right repl = Right {IRCMessage | irc_prefix=prefix, irc_command=Left repl}
59 // Error parsing prefix
60 (_, es) = Left ["Error parsing prefix"]
63 parsePrefix :: Parser Char (Maybe (Either IRCUser String))
65 = optional (pToken ':' >>| parseEither parseUser parseHost <* pToken ' ')
67 parseEither :: (Parser a b) (Parser a c) -> Parser a (Either b c)
68 parseEither p q = Left <$> p <|> Right <$> q
70 parseUser :: Parser Char IRCUser
72 >>= \nick->optional (pToken '!' >>| parseUsr)
73 >>= \muser->optional (pToken '@' >>| parseHost)
74 >>= \mhost->pure {IRCUser
75 | irc_nick=nick, irc_user=muser, irc_host=mhost}
77 parseUsr :: Parser Char String
78 parseUsr = toString <$> pSome (pNoneOf [' ', '@':illegal])
80 parseNick :: Parser Char String
82 >>= \c ->pMany (pAlpha <|> pDigit <|> pOneOf (fromString "_-[]\\`^{}"))
83 >>= \cs->pure (toString [c:cs])
85 parseHost :: Parser Char String
86 parseHost = jon "." <$> (pSepBy parseName (pToken '.'))
87 >>= \s->optional (pToken '.') >>= pure o maybe s (\p->s+++toString s)
89 parseName :: Parser Char String
90 parseName = toString <$> pSome (pAlpha <|> pDigit <|> pOneOf ['-', '/'])
93 parseCmd :: [Char] -> Either Error IRCCommand
94 parseCmd cs = fst $ gIRCParse{|*|} $ argfun $ split " " $ toString cs
96 argfun :: [String] -> [String]
100 | x.[0] == ':' = [jon " " $ [x % (1, size x):map rtrim xs]]
101 | otherwise = [x:argfun xs]
104 parseReply :: Parser Char IRCNumReply
105 parseReply = spaceParser
106 >>| (pMany (pToken '0') >>| pSome pDigit <* spaceParser)
107 >>= \rep->(toString <$> pSome (pNoneOf [' ':illegal]) <* spaceParser)
108 >>= \rec->(toString <$> pSome (pNoneOf illegal))
109 >>= \msg->pure {IRCNumReply
110 | irc_reply = fromInt $ toInt $ toString rep
111 , irc_recipient = rec
112 , irc_message = msg % (if (msg.[0] == ':') 1 0, size msg)
114 <* pToken '\r' <* pToken '\n'
116 spaceParser :: Parser Char [Char]
117 spaceParser = pMany $ pToken ' '
120 pNoneOf :: [a] -> Parser a a | Eq a
121 pNoneOf l = pSatisfy (not o flip isMember l)
124 illegal = ['\x00','\r','\n']
126 instance toString IRCNumReply where
127 toString m = lpad (toString $ toInt m.irc_reply) 3 '0' <+ " " <+
128 m.irc_recipient <+ " " <+ concat (gIRCPrint{|*|} m.irc_message)
129 instance toString IRCMessage where
130 toString m = maybe "" (\s->either ((<+) ":") id s <+ " ") m.irc_prefix
131 <+ either toString toString m.irc_command
132 instance toString IRCUser where
133 toString m = m.irc_nick <+ maybe "" ((<+) "!") m.irc_user
134 <+ maybe "" ((<+) "@") m.irc_host
135 instance toString IRCCommand where
136 toString m = jon " " (gIRCPrint{|*|} m) +++ "\r\n"
137 instance toString IRCReplies where toString r = printToString r
138 instance toString IRCErrors where toString r = printToString r
140 (<+) infixr 5 :: a b -> String | toString a & toString b
141 (<+) a b = toString a +++ toString b
143 instance fromInt IRCReplies where
144 fromInt r = case r of
151 201 = RPL_TRACECONNECTING
152 202 = RPL_TRACEHANDSHAKE
153 203 = RPL_TRACEUNKNOWN
154 204 = RPL_TRACEOPERATOR
156 206 = RPL_TRACESERVER
157 207 = RPL_TRACESERVICE
158 208 = RPL_TRACENEWTYPE
160 210 = RPL_TRACERECONNECT
161 211 = RPL_STATSLINKINFO
162 212 = RPL_STATSCOMMANDS
166 235 = RPL_SERVLISTEND
167 242 = RPL_STATSUPTIME
169 251 = RPL_LUSERCLIENT
171 253 = RPL_LUSERUNKNOWN
172 254 = RPL_LUSERCHANNELS
187 312 = RPL_WHOISSERVER
188 313 = RPL_WHOISOPERATOR
193 319 = RPL_WHOISCHANNELS
197 324 = RPL_CHANNELMODEIS
204 347 = RPL_ENDOFINVITELIST
206 349 = RPL_ENDOFEXCEPTLIST
214 368 = RPL_ENDOFBANLIST
215 369 = RPL_ENDOFWHOWAS
223 383 = RPL_YOURESERVICE
231 instance toInt IRCReplies where
239 RPL_TRACECONNECTING = 201
240 RPL_TRACEHANDSHAKE = 202
241 RPL_TRACEUNKNOWN = 203
242 RPL_TRACEOPERATOR = 204
244 RPL_TRACESERVER = 206
245 RPL_TRACESERVICE = 207
246 RPL_TRACENEWTYPE = 208
248 RPL_TRACERECONNECT = 210
249 RPL_STATSLINKINFO = 211
250 RPL_STATSCOMMANDS = 212
254 RPL_SERVLISTEND = 234
255 RPL_STATSUPTIME = 242
257 RPL_LUSERCLIENT = 251
259 RPL_LUSERUNKNOWN = 253
260 RPL_LUSERCHANNELS = 254
275 RPL_WHOISSERVER = 312
276 RPL_WHOISOPERATOR = 313
281 RPL_WHOISCHANNELS = 319
285 RPL_CHANNELMODEIS = 324
292 RPL_ENDOFINVITELIST = 347
294 RPL_ENDOFEXCEPTLIST = 349
302 RPL_ENDOFBANLIST = 367
303 RPL_ENDOFWHOWAS = 369
311 RPL_YOURESERVICE = 383
319 instance fromInt IRCErrors where
320 fromInt r = case r of
322 402 = ERR_NOSUCHSERVER
323 403 = ERR_NOSUCHCHANNEL
324 404 = ERR_CANNOTSENDTOCHAN
325 405 = ERR_TOOMANYCHANNELS
326 406 = ERR_WASNOSUCHNICK
327 407 = ERR_TOOMANYTARGETS
328 408 = ERR_NOSUCHSERVICE
330 411 = ERR_NORECIPIENT
331 412 = ERR_NOTEXTTOSEND
333 414 = ERR_WILDTOPLEVEL
335 421 = ERR_UNKNOWNCOMMAND
337 423 = ERR_NOADMININFO
339 431 = ERR_NONICKNAMEGIVEN
340 432 = ERR_ERRONEUSNICKNAME
341 433 = ERR_NICKNAMEINUSE
342 436 = ERR_NICKCOLLISION
343 437 = ERR_UNAVAILRESOURCE
344 441 = ERR_USERNOTINCHANNEL
345 442 = ERR_NOTONCHANNEL
346 443 = ERR_USERONCHANNEL
348 445 = ERR_SUMMONDISABLED
349 446 = ERR_USERSDISABLED
350 451 = ERR_NOTREGISTERED
351 461 = ERR_NEEDMOREPARAMS
352 462 = ERR_ALREADYREGISTRED
353 463 = ERR_NOPERMFORHOST
354 464 = ERR_PASSWDMISMATCH
355 465 = ERR_YOUREBANNEDCREEP
356 466 = ERR_YOUWILLBEBANNED
358 471 = ERR_CHANNELISFULL
359 472 = ERR_UNKNOWNMODE
360 473 = ERR_INVITEONLYCHAN
361 474 = ERR_BANNEDFROMCHAN
362 475 = ERR_BADCHANNELKEY
363 476 = ERR_BADCHANMASK
364 477 = ERR_NOCHANMODES
365 478 = ERR_BANLISTFULL
366 481 = ERR_NOPRIVILEGES
367 482 = ERR_CHANOPRIVSNEEDED
368 483 = ERR_CANTKILLSERVER
370 485 = ERR_UNIQOPPRIVSNEEDED
372 501 = ERR_UMODEUNKNOWNFLAG
373 502 = ERR_USERSDONTMATCH
376 instance toInt IRCErrors where
379 ERR_NOSUCHSERVER = 402
380 ERR_NOSUCHCHANNEL = 403
381 ERR_CANNOTSENDTOCHAN = 404
382 ERR_TOOMANYCHANNELS = 405
383 ERR_WASNOSUCHNICK = 406
384 ERR_TOOMANYTARGETS = 407
385 ERR_NOSUCHSERVICE = 408
387 ERR_NORECIPIENT = 411
388 ERR_NOTEXTTOSEND = 412
390 ERR_WILDTOPLEVEL = 414
392 ERR_UNKNOWNCOMMAND = 421
394 ERR_NOADMININFO = 423
396 ERR_NONICKNAMEGIVEN = 431
397 ERR_ERRONEUSNICKNAME = 432
398 ERR_NICKNAMEINUSE = 433
399 ERR_NICKCOLLISION = 436
400 ERR_UNAVAILRESOURCE = 437
401 ERR_USERNOTINCHANNEL = 441
402 ERR_NOTONCHANNEL = 442
403 ERR_USERONCHANNEL = 443
405 ERR_SUMMONDISABLED = 445
406 ERR_USERSDISABLED = 446
407 ERR_NOTREGISTERED = 451
408 ERR_NEEDMOREPARAMS = 461
409 ERR_ALREADYREGISTRED = 462
410 ERR_NOPERMFORHOST = 463
411 ERR_PASSWDMISMATCH = 464
412 ERR_YOUREBANNEDCREEP = 465
413 ERR_YOUWILLBEBANNED = 466
415 ERR_CHANNELISFULL = 471
416 ERR_UNKNOWNMODE = 472
417 ERR_INVITEONLYCHAN = 473
418 ERR_BANNEDFROMCHAN = 474
419 ERR_BADCHANNELKEY = 475
420 ERR_BADCHANMASK = 476
421 ERR_NOCHANMODES = 477
422 ERR_BANLISTFULL = 478
423 ERR_NOPRIVILEGES = 481
424 ERR_CHANOPRIVSNEEDED = 482
425 ERR_CANTKILLSERVER = 483
427 ERR_UNIQOPPRIVSNEEDED = 485
429 ERR_UMODEUNKNOWNFLAG = 501
430 ERR_USERSDONTMATCH = 502