1 implementation module IRC
17 import Text.Parsers.Simple.Core
18 import Text.Parsers.Simple.Chars
21 import Control.Applicative
22 from Data.Functor import <$>
24 from Data.Func import $
25 from Text import class Text(ltrim,indexOf,concat), instance Text String
27 from StdMisc import undef
32 derive gPrint IRCCommand, IRCReplies, IRCErrors, (,), Maybe, (), Either, IRCMessage, IRCUser, IRCNumReply
34 Start = jon "\n" $ map printToString
35 [ parseIRCMessage ":clooglebot!~cloogle@dhcp-077-249-221-037.chello.nl QUIT\r\n"
36 , parseIRCMessage ":clooglebot!~cloogle QUIT\r\n"
37 , parseIRCMessage ":frobnicator!~frobnicat@92.110.128.124 QUIT\r\n"
38 , parseIRCMessage ":frobnicator!~frobnicat@92.110.128.124 AWAY test\r\n"
39 , parseIRCMessage ":frobnicator!~frobnicat@92.110.128.124 AWAY :test with spaces\r\n"
40 , parseIRCMessage ":cherryh.freenode.net NOTICE * :*** Found your hostname\r\n"
41 , parseIRCMessage ":cherryh.freenode.net QUIT :hoi hoi\r\n"
44 (<+) infixr 5 :: a b -> String | toString a & toString b
45 (<+) a b = toString a +++ toString b
47 parseIRCMessage :: String -> Either [Error] IRCMessage
48 parseIRCMessage s = case runParser parsePrefix (fromString s) of
49 ([(prefix, rest):_], _)
50 = case parse parseReply rest of
51 Left e = case parseCmd rest of
52 Left e2 = Left $ e2 ++ e
53 Right cmd = Right {IRCMessage | irc_prefix=prefix, irc_command=Right cmd}
54 Right repl = Right {IRCMessage | irc_prefix=prefix, irc_command=Left repl}
55 (_, es) = Left ["couldn't parse prefix":es]
57 parsePrefix :: Parser Char (Maybe (Either IRCUser String))
58 parsePrefix = optional (pToken ':' >>| parseEither parseUser parseHost) <* pToken ' '
60 generic gIRCParse a :: [String] -> (Either [Error] a, [String])
61 gIRCParse{|UNIT|} a = (Right UNIT, a)
62 gIRCParse{|String|} [a:as] = (Right a, as)
63 gIRCParse{|String|} [] = (Left ["Expected a string"], [])
64 gIRCParse{|Int|} [a:as] = (Right $ toInt a, as)
65 gIRCParse{|Int|} [] = (Left ["Expected an integer"], [])
66 gIRCParse{|EITHER|} lp rp as = case lp as of
67 (Right a, rest) = (Right $ LEFT a, rest)
68 (Left e1, rest) = case rp as of
69 (Right a, rest) = (Right $ RIGHT a, rest)
70 (Left e2, rest) = (Left $ e1 ++ e2, [])
71 gIRCParse{|OBJECT|} p as = case p as of
72 (Right e, rest) = (Right $ OBJECT e, rest)
73 (Left e, rest) = (Left e, [])
74 gIRCParse{|CONS of d|} p [] = (Left ["Expected a cmd constructor: " +++ d.gcd_name], [])
75 gIRCParse{|CONS of d|} p [a:as]
76 | a <> d.gcd_name = (Left ["Wrong constructor. expected: " +++ d.gcd_name +++ ", got: " +++ a], [])
78 (Right a, rest) = (Right $ CONS a, rest)
79 (Left e, rest) = (Left e, [])
80 gIRCParse{|PAIR|} pl pr as = case pl as of
81 (Right a1, rest) = case pr rest of
82 (Right a2, rest) = (Right $ PAIR a1 a2, rest)
83 (Left e, rest) = (Left e, [])
84 (Left e, rest) = (Left e, [])
85 gIRCParse{|[]|} pl as = plist pl as
87 plist pl as = case pl as of
88 (Right e, rest) = case plist pl rest of
89 (Right es, rest) = (Right [e:es], rest)
90 (Left e, rest) = (Left e, [])
91 (Left e, rest) = (Right [], as)
92 gIRCParse{|Maybe|} pm as = case pm as of
93 (Right a, rest) = (Right $ Just a, rest)
94 (Left e, rest) = (Right Nothing, as)
96 derive gIRCParse (,), (,,), IRCCommand
98 parseCmd :: [Char] -> Either [Error] IRCCommand
99 parseCmd cs = fst $ gIRCParse{|*|} $ argfun $ 'Text'.split " " $ toString cs
100 //= parse cmdParser $ argfun $ 'Text'.split " " $ toString cs
102 argfun :: [String] -> [String]
106 | x.[0] == ':' = [jon " " $ [x:map 'Text'.rtrim xs]]
107 | otherwise = [x:argfun xs]
109 p = pSatisfy (const True)
110 lst = fmap $ 'Text'.split ","
114 nn p f = pToken p >>| f
116 cmdParser :: Parser String IRCCommand
118 (nn "ADMIN" $ fmap ADMIN opt)
119 <|> (nn "AWAY" $ fmap AWAY p)
120 <|> (nn "CONNECT" $ liftM2 CONNECT p (optional $ liftM2 tuple pInt opt))
121 <|> (nn "DIE" $ pure DIE)
122 <|> (nn "ERROR" $ fmap ERROR p)
123 <|> (nn "INFO" $ fmap INFO opt)
124 <|> (nn "INVITE" $ liftM2 INVITE p p)
125 <|> (nn "ISON" $ fmap ISON $ pMany p)
126 // <|> (nn "JOIN" $ fmap JOIN $ lst p >>= \ch->lst p >>= \ks->pure (zip2 ch (ks ++ repeat Nothing)))
127 <|> (nn "KICK" $ liftM3 KICK p p opt)
128 <|> (nn "KILL" $ liftM2 KILL p p)
129 <|> (nn "LINKS" $ fmap LINKS $ optional $ liftM2 tuple opt p)
130 <|> (nn "LIST" $ fmap LIST $ optional $ liftM2 tuple ('Text'.split "," <$> p) opt)
131 <|> (nn "LUSERS" $ fmap LUSERS $ optional $ liftM2 tuple p opt)
132 <|> (nn "MODE" $ liftM5 MODE p p opt opt opt)
133 <|> (nn "MOTD" $ fmap MOTD $ opt)
134 <|> (nn "NAMES" $ fmap NAMES $ lst p)
135 <|> (nn "NICK" $ fmap NAMES $ lst p)
136 //"NICK" = String (Maybe String)
137 //"NJOIN" = command0 NJOIN args
138 //"NOTICE" = String String
139 //"OPER" = String String
142 //"PING" = String (Maybe String)
143 //"PONG" = String (Maybe String)
144 //"PRIVMSG" = [String] String
145 // "QUIT" = case args of
146 // [_,_:_] = Left $ "QUIT has too many arguments"
147 // x = Right $ QUIT $ listToMaybe x
148 // "REHASH" = command0 REHASH args
149 // "RESTART" = command0 REHASH args
150 // "SERVER" = command0 REHASH args
151 //"SERVICE" = String String String String
152 //"SERVLIST" = (Maybe (String, Maybe String))
153 //"SQUERY" = String String
154 // "SQUIRT" = command0 REHASH args
155 //"SQUIT" = String String
156 //"STATS" = (Maybe (String, Maybe String))
157 //"SUMMON" = String (Maybe (String, Maybe String))
158 //"TIME" = (Maybe String)
159 //"TOPIC" = String (Maybe String)
160 //"TRACE" = (Maybe String)
161 //"USER" = String String String
162 //"USERHOST" = [String]
163 //"USERS" = (Maybe String)
164 //"VERSION" = (Maybe String)
166 //"WHO" = (Maybe String)
167 //"WHOIS" = (Maybe String) [String]
168 //"WHOWAS" = (Maybe String) [String]
170 parseReply :: Parser Char IRCNumReply
171 parseReply = (toString <$> pSome pDigit)
173 >>= \recipient->spaceParser >>| (toString <$> pSome (pNoneOf illegal))
174 >>= \msg->pure {IRCNumReply|irc_reply=fs rep,irc_recipient=recipient,irc_message=msg}
176 fs :: String -> IRCReplies
177 fs s = fromInt $ toInt s
179 spaceParser :: Parser Char [Char]
180 spaceParser = pMany $ pToken ' '
182 parseServer :: Parser Char String
185 parseEither :: (Parser a b) (Parser a c) -> Parser a (Either b c)
186 parseEither p q = Left <$> p <|> Right <$> q
188 parseUser :: Parser Char IRCUser
189 parseUser = parseNick
190 >>= \nick->optional (pToken '!' >>| parseUsr)
191 >>= \muser->optional (pToken '@' >>| parseHost)
192 >>= \mhost->pure {IRCUser | irc_nick=nick, irc_user=muser, irc_host=mhost}
194 parseUsr :: Parser Char String
195 parseUsr = toString <$> pSome (pNoneOf [' ', '@':illegal])
197 parseNick :: Parser Char String
198 parseNick = pAlpha >>= \c->pMany (pAlpha <|> pDigit <|> pSpecial)
199 >>= \cs->pure (toString [c:cs])
201 pSpecial :: Parser Char Char
202 pSpecial = pOneOf ['-', '[', ']', '\\', '\`', '^', '{', '}']
204 parseHost :: Parser Char String
205 parseHost = jon "." <$> pSepBy parseName (pToken '.')
207 parseName :: Parser Char String
208 parseName = toString <$> pSome (pAlpha <|> pDigit <|> pOneOf ['-'])
210 instance toString IRCNumReply where
211 toString m = toInt m.irc_reply <+ " " <+ m.irc_recipient <+ " " <+ formatMSG m.irc_message
212 instance toString IRCMessage where
213 toString m = maybe "" (\s->either ((<+) ":") id s <+ " ") m.irc_prefix
214 <+ either toString toString m.irc_command
216 instance toString IRCUser where
217 toString m = m.irc_nick <+ maybe "" ((<+) "!") m.irc_user
218 <+ maybe "" ((<+) "@") m.irc_host
223 pMiddle :: Parser Char String
224 pMiddle = fmap toString $
225 spaceParser >>| liftM2 cons (pNotSatisfy ((==)':')) (pMany $ pNoneOf [' ':illegal])
227 pTrailing :: Parser Char String
228 pTrailing = fmap toString $
229 spaceParser >>| pToken ':' >>| pMany (pNoneOf illegal)
231 pParam :: Parser Char String
232 pParam = pMiddle <|> pTrailing
234 pNoneOf :: [a] -> Parser a a | Eq a
235 pNoneOf l = pSatisfy (not o flip isMember l)
237 pNotSatisfy :: (a -> Bool) -> Parser a a | Eq a
238 pNotSatisfy f = pSatisfy (not o f)
240 pInt :: Parser Char Int
241 pInt = toInt o toString <$> (spaceParser >>| pSome pDigit)
244 illegal = ['\x00','\r','\n']
246 instance toString IRCCommand where
247 toString r = jon " " (print r) +++ "\r\n"
249 print :: IRCCommand -> [String]
251 ADMIN mm = ["ADMIN":maybeToList mm]
253 //CONNECT String (Maybe (Int, Maybe String))
256 //INFO (Maybe String)
257 //INVITE String String
259 JOIN chs = ["JOIN",if (isEmpty chs) "0"
260 (jon ", " [jon " " [ch:maybeToList mk]\\(ch, mk)<-chs])]
261 //KICK String String (Maybe String)
263 //LINKS (Maybe (Maybe String, String))
264 //LIST (Maybe ([String], Maybe String))
265 //LUSERS (Maybe (String, Maybe String))
266 //MODE String String (Maybe String) (Maybe String) (Maybe String)
267 //MOTD (Maybe String)
269 NICK n ms = ["NICK", n]
271 //NOTICE String String
275 PING a mb = ["PING",a:maybeToList mb]
276 PONG a mb = ["PONG",a:maybeToList mb]
277 PRIVMSG dest msg = ["PRIVMSG",jon "," dest,formatMSG msg]
278 QUIT msg = ["QUIT":maybeToList msg]
282 //SERVICE String String String String
283 //SERVLIST (Maybe (String, Maybe String))
284 //SQUERY String String
286 //SQUIT String String
287 //STATS (Maybe (String, Maybe String))
288 //SUMMON String (Maybe (String, Maybe String))
289 //TIME (Maybe String)
290 //TOPIC String (Maybe String)
291 //TRACE (Maybe String)
292 USER login mode rn = ["USER", login, mode, "*", ":"+++rn]
294 //USERS (Maybe String)
295 //VERSION (Maybe String)
298 //WHOIS (Maybe String) [String]
299 //WHOWAS (Maybe String) [String]
300 _ = [printToString r]
302 formatMSG :: String -> String
303 formatMSG s = if (indexOf " " s > 0 || indexOf " " s > 0) (":" +++ s) s
306 instance toString IRCReplies where toString r = printToString r
307 instance toString IRCErrors where toString r = printToString r
309 instance fromInt IRCReplies where
310 fromInt r = case r of
317 201 = RPL_TRACECONNECTING
318 202 = RPL_TRACEHANDSHAKE
319 203 = RPL_TRACEUNKNOWN
320 204 = RPL_TRACEOPERATOR
322 206 = RPL_TRACESERVER
323 207 = RPL_TRACESERVICE
324 208 = RPL_TRACENEWTYPE
326 210 = RPL_TRACERECONNECT
327 211 = RPL_STATSLINKINFO
328 212 = RPL_STATSCOMMANDS
332 235 = RPL_SERVLISTEND
333 242 = RPL_STATSUPTIME
335 251 = RPL_LUSERCLIENT
337 253 = RPL_LUSERUNKNOWN
338 254 = RPL_LUSERCHANNELS
353 312 = RPL_WHOISSERVER
354 313 = RPL_WHOISOPERATOR
359 319 = RPL_WHOISCHANNELS
363 324 = RPL_CHANNELMODEIS
370 347 = RPL_ENDOFINVITELIST
372 349 = RPL_ENDOFEXCEPTLIST
380 368 = RPL_ENDOFBANLIST
381 369 = RPL_ENDOFWHOWAS
389 383 = RPL_YOURESERVICE
397 instance toInt IRCReplies where
405 RPL_TRACECONNECTING = 201
406 RPL_TRACEHANDSHAKE = 202
407 RPL_TRACEUNKNOWN = 203
408 RPL_TRACEOPERATOR = 204
410 RPL_TRACESERVER = 206
411 RPL_TRACESERVICE = 207
412 RPL_TRACENEWTYPE = 208
414 RPL_TRACERECONNECT = 210
415 RPL_STATSLINKINFO = 211
416 RPL_STATSCOMMANDS = 212
420 RPL_SERVLISTEND = 234
421 RPL_STATSUPTIME = 242
423 RPL_LUSERCLIENT = 251
425 RPL_LUSERUNKNOWN = 253
426 RPL_LUSERCHANNELS = 254
441 RPL_WHOISSERVER = 312
442 RPL_WHOISOPERATOR = 313
447 RPL_WHOISCHANNELS = 319
451 RPL_CHANNELMODEIS = 324
458 RPL_ENDOFINVITELIST = 347
460 RPL_ENDOFEXCEPTLIST = 349
468 RPL_ENDOFBANLIST = 367
469 RPL_ENDOFWHOWAS = 369
477 RPL_YOURESERVICE = 383
484 instance fromInt IRCErrors where
485 fromInt r = case r of
487 402 = ERR_NOSUCHSERVER
488 403 = ERR_NOSUCHCHANNEL
489 404 = ERR_CANNOTSENDTOCHAN
490 405 = ERR_TOOMANYCHANNELS
491 406 = ERR_WASNOSUCHNICK
492 407 = ERR_TOOMANYTARGETS
493 408 = ERR_NOSUCHSERVICE
495 411 = ERR_NORECIPIENT
496 412 = ERR_NOTEXTTOSEND
498 414 = ERR_WILDTOPLEVEL
500 421 = ERR_UNKNOWNCOMMAND
502 423 = ERR_NOADMININFO
504 431 = ERR_NONICKNAMEGIVEN
505 432 = ERR_ERRONEUSNICKNAME
506 433 = ERR_NICKNAMEINUSE
507 436 = ERR_NICKCOLLISION
508 437 = ERR_UNAVAILRESOURCE
509 441 = ERR_USERNOTINCHANNEL
510 442 = ERR_NOTONCHANNEL
511 443 = ERR_USERONCHANNEL
513 445 = ERR_SUMMONDISABLED
514 446 = ERR_USERSDISABLED
515 451 = ERR_NOTREGISTERED
516 461 = ERR_NEEDMOREPARAMS
517 462 = ERR_ALREADYREGISTRED
518 463 = ERR_NOPERMFORHOST
519 464 = ERR_PASSWDMISMATCH
520 465 = ERR_YOUREBANNEDCREEP
521 466 = ERR_YOUWILLBEBANNED
523 471 = ERR_CHANNELISFULL
524 472 = ERR_UNKNOWNMODE
525 473 = ERR_INVITEONLYCHAN
526 474 = ERR_BANNEDFROMCHAN
527 475 = ERR_BADCHANNELKEY
528 476 = ERR_BADCHANMASK
529 477 = ERR_NOCHANMODES
530 478 = ERR_BANLISTFULL
531 481 = ERR_NOPRIVILEGES
532 482 = ERR_CHANOPRIVSNEEDED
533 483 = ERR_CANTKILLSERVER
535 485 = ERR_UNIQOPPRIVSNEEDED
537 501 = ERR_UMODEUNKNOWNFLAG
538 502 = ERR_USERSDONTMATCH
540 instance toInt IRCErrors where
543 ERR_NOSUCHSERVER = 402
544 ERR_NOSUCHCHANNEL = 403
545 ERR_CANNOTSENDTOCHAN = 404
546 ERR_TOOMANYCHANNELS = 405
547 ERR_WASNOSUCHNICK = 406
548 ERR_TOOMANYTARGETS = 407
549 ERR_NOSUCHSERVICE = 408
551 ERR_NORECIPIENT = 411
552 ERR_NOTEXTTOSEND = 412
554 ERR_WILDTOPLEVEL = 414
556 ERR_UNKNOWNCOMMAND = 421
558 ERR_NOADMININFO = 423
560 ERR_NONICKNAMEGIVEN = 431
561 ERR_ERRONEUSNICKNAME = 432
562 ERR_NICKNAMEINUSE = 433
563 ERR_NICKCOLLISION = 436
564 ERR_UNAVAILRESOURCE = 437
565 ERR_USERNOTINCHANNEL = 441
566 ERR_NOTONCHANNEL = 442
567 ERR_USERONCHANNEL = 443
569 ERR_SUMMONDISABLED = 445
570 ERR_USERSDISABLED = 446
571 ERR_NOTREGISTERED = 451
572 ERR_NEEDMOREPARAMS = 461
573 ERR_ALREADYREGISTRED = 462
574 ERR_NOPERMFORHOST = 463
575 ERR_PASSWDMISMATCH = 464
576 ERR_YOUREBANNEDCREEP = 465
577 ERR_YOUWILLBEBANNED = 466
579 ERR_CHANNELISFULL = 471
580 ERR_UNKNOWNMODE = 472
581 ERR_INVITEONLYCHAN = 473
582 ERR_BANNEDFROMCHAN = 474
583 ERR_BADCHANNELKEY = 475
584 ERR_BADCHANMASK = 476
585 ERR_NOCHANMODES = 477
586 ERR_BANLISTFULL = 478
587 ERR_NOPRIVILEGES = 481
588 ERR_CHANOPRIVSNEEDED = 482
589 ERR_CANTKILLSERVER = 483
591 ERR_UNIQOPPRIVSNEEDED = 485
593 ERR_UMODEUNKNOWNFLAG = 501
594 ERR_USERSDONTMATCH = 502