dev'
[cloogle-irc.git] / IRC.icl
1 implementation module IRC
2
3 import StdList
4 import GenPrint
5 import StdOverloaded
6 import Data.Maybe
7 import Data.Either
8 import StdFunc
9 import StdString
10 import StdChar
11 import StdBool
12 import _SystemArray
13
14 import Text.Parsers.Simple.Core
15 import Text.Parsers.Simple.Chars
16 import Data.Tuple
17 import Control.Monad
18 import Control.Applicative
19 from Data.Functor import <$>
20
21 from Data.Func import $
22 from Text import class Text(ltrim,indexOf,concat), instance Text String
23 import qualified Text
24 from StdMisc import undef
25 import StdDebug
26
27 jon :== 'Text'.join
28
29 derive gPrint IRCCommand, IRCReplies, IRCErrors, (,), Maybe, (), Either
30
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"
37
38 (<+) infixr 5 :: a b -> String | toString a & toString b
39 (<+) a b = toString a +++ toString b
40
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
45 Left e2 = Left [e2:e]
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]
49
50
51 parseCmd :: [Char] -> Either Error IRCCommand
52 parseCmd cs
53 | not (trace_tn $ toString cs) = undef
54 = processParse $ argfun $ 'Text'.split " " $ toString cs
55 where
56 argfun :: [String] -> [String]
57 argfun [] = []
58 argfun [x:xs]
59 # x = 'Text'.trim x
60 | x.[0] == ':' = ['Text'.join " " [x:xs]]
61 | otherwise = [x:argfun xs]
62
63 command0 :: IRCCommand [String] -> Either Error IRCCommand
64 command0 c [] = Right c
65 command0 c x = Left $ toString c +++ " doesn't have arguments"
66
67 processParse :: [String] -> Either Error IRCCommand
68 processParse [] = Left "Empty list of arguments"
69 processParse [cmd:args] = case cmd of
70 //"ADMIN" = (Maybe String)
71 //"AWAY" = String
72 //"CONNECT" = String (Maybe (Int, Maybe String))
73 "DIE" = command0 DIE args
74 //"ERROR" = String
75 //"INFO" = (Maybe String)
76 //"INVITE" = String String
77 //"ISON" = [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)
86 //"NAMES" = [String]
87 //"NICK" = String (Maybe String)
88 "NJOIN" = command0 NJOIN args
89 //"NOTICE" = String String
90 //"OPER" = String String
91 //"PART" = [String]
92 //"PASS" = String
93 //"PING" = String (Maybe String)
94 //"PONG" = String (Maybe String)
95 //"PRIVMSG" = [String] String
96 "QUIT" = case args of
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)
116 //"WALLOPS" = String
117 //"WHO" = (Maybe String)
118 //"WHOIS" = (Maybe String) [String]
119 //"WHOWAS" = (Maybe String) [String]
120 _ = Left $ "Unknown command: " +++ cmd
121
122 parsePrefix :: Parser Char (Maybe (Either String IRCUser))
123 parsePrefix = optional (parseEither parseHost parseUser) <* spaceParser
124
125 parseReply :: Parser Char IRCNumReply
126 parseReply = (toString <$> pSome pDigit)
127 >>= \rep->pMiddle
128 >>= \recipient->spaceParser >>| (toString <$> pSome (pNoneOf illegal))
129 >>= \msg->pure {IRCNumReply|irc_reply=fs rep,irc_recipient=recipient,irc_message=msg}
130 where
131 fs :: String -> IRCReplies
132 fs s = fromInt $ toInt s
133
134 spaceParser :: Parser Char [Char]
135 spaceParser = pMany $ pToken ' '
136
137 parseServer :: Parser Char String
138 parseServer = pFail
139
140 parseEither :: (Parser a b) (Parser a c) -> Parser a (Either b c)
141 parseEither p q = Left <$> p <|> Right <$> q
142
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}
148
149 parseUsr :: Parser Char String
150 parseUsr = toString <$> pSome (pNoneOf [' ', '@':illegal])
151
152 parseNick :: Parser Char String
153 parseNick = pAlpha >>= \c->pMany (pAlpha <|> pDigit <|> pSpecial)
154 >>= \cs->pure (toString [c:cs])
155
156 pSpecial :: Parser Char Char
157 pSpecial = pOneOf ['-', '[', ']', '\\', '\`', '^', '{', '}']
158
159 parseHost :: Parser Char String
160 parseHost = pToken ':' >>| (concat <$> pSepBy parseName (pToken '.'))
161 where
162 parseName :: Parser Char String
163 parseName = toString <$> pSome (pAlpha <|> pDigit <|> pOneOf ['-'])
164
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
170
171 instance toString IRCUser where
172 toString m = m.irc_nick <+ maybe "" ((<+) "!") m.irc_user
173 <+ maybe "" ((<+) "@") m.irc_host
174
175 cons :: a [a] -> [a]
176 cons a as = [a:as]
177
178 pMiddle :: Parser Char String
179 pMiddle = fmap toString $
180 spaceParser >>| liftM2 cons (pNotSatisfy ((==)':')) (pMany $ pNoneOf [' ':illegal])
181
182 pTrailing :: Parser Char String
183 pTrailing = fmap toString $
184 spaceParser >>| pToken ':' >>| pMany (pNoneOf illegal)
185
186 pParam :: Parser Char String
187 pParam = pMiddle <|> pTrailing
188
189 pNoneOf :: [a] -> Parser a a | Eq a
190 pNoneOf l = pSatisfy (not o flip isMember l)
191
192 pNotSatisfy :: (a -> Bool) -> Parser a a | Eq a
193 pNotSatisfy f = pSatisfy (not o f)
194
195 pInt :: Parser Char Int
196 pInt = toInt o toString <$> (spaceParser >>| pSome pDigit)
197
198 illegal :: [Char]
199 illegal = ['\x00','\r','\n']
200
201 instance toString IRCCommand where
202 toString r = jon " " (print r) +++ "\r\n"
203
204 print :: IRCCommand -> [String]
205 print r = case r of
206 ADMIN mm = ["ADMIN":maybeToList mm]
207 //AWAY String
208 //CONNECT String (Maybe (Int, Maybe String))
209 //DIE
210 //ERROR String
211 //INFO (Maybe String)
212 //INVITE String String
213 //ISON [String]
214 JOIN chs = ["JOIN",if (isEmpty chs) "0"
215 (jon ", " [jon " " [ch:maybeToList mk]\\(ch, mk)<-chs])]
216 //KICK String String (Maybe String)
217 //KILL String 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)
223 //NAMES [String]
224 NICK n ms = ["NICK", n]
225 //NJOIN
226 //NOTICE String String
227 //OPER String String
228 //PART [String]
229 //PASS 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]
234 //REHASH
235 //RESTART
236 //SERVER
237 //SERVICE String String String String
238 //SERVLIST (Maybe (String, Maybe String))
239 //SQUERY String String
240 //SQUIRT
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]
248 //USERHOST [String]
249 //USERS (Maybe String)
250 //VERSION (Maybe String)
251 //WALLOPS String
252 //WHO (Maybe String)
253 //WHOIS (Maybe String) [String]
254 //WHOWAS (Maybe String) [String]
255 _ = [printToString r]
256
257 formatMSG :: String -> String
258 formatMSG s = if (indexOf " " s > 0 || indexOf " " s > 0) (":" +++ s) s
259
260
261 instance toString IRCReplies where toString r = printToString r
262 instance toString IRCErrors where toString r = printToString r
263
264 instance fromInt IRCReplies where
265 fromInt r = case r of
266 1 = RPL_WELCOME
267 2 = RPL_YOURHOST
268 3 = RPL_CREATED
269 4 = RPL_MYINFO
270 5 = RPL_BOUNCE
271 200 = RPL_TRACELINK
272 201 = RPL_TRACECONNECTING
273 202 = RPL_TRACEHANDSHAKE
274 203 = RPL_TRACEUNKNOWN
275 204 = RPL_TRACEOPERATOR
276 205 = RPL_TRACEUSER
277 206 = RPL_TRACESERVER
278 207 = RPL_TRACESERVICE
279 208 = RPL_TRACENEWTYPE
280 209 = RPL_TRACECLASS
281 210 = RPL_TRACERECONNECT
282 211 = RPL_STATSLINKINFO
283 212 = RPL_STATSCOMMANDS
284 219 = RPL_ENDOFSTATS
285 221 = RPL_UMODEIS
286 234 = RPL_SERVLIST
287 235 = RPL_SERVLISTEND
288 242 = RPL_STATSUPTIME
289 243 = RPL_STATSOLINE
290 251 = RPL_LUSERCLIENT
291 252 = RPL_LUSEROP
292 253 = RPL_LUSERUNKNOWN
293 254 = RPL_LUSERCHANNELS
294 255 = RPL_LUSERME
295 256 = RPL_ADMINME
296 257 = RPL_ADMINLOC1
297 258 = RPL_ADMINLOC2
298 259 = RPL_ADMINEMAIL
299 261 = RPL_TRACELOG
300 262 = RPL_TRACEEND
301 263 = RPL_TRYAGAIN
302 301 = RPL_AWAY
303 302 = RPL_USERHOST
304 303 = RPL_ISON
305 304 = RPL_UNAWAY
306 305 = RPL_NOWAWAY
307 311 = RPL_WHOISUSER
308 312 = RPL_WHOISSERVER
309 313 = RPL_WHOISOPERATOR
310 314 = RPL_WHOWASUSER
311 315 = RPL_ENDOFWHO
312 317 = RPL_WHOISIDLE
313 318 = RPL_ENDOFWHOIS
314 319 = RPL_WHOISCHANNELS
315 321 = RPL_LISTSTART
316 322 = RPL_LIST
317 323 = RPL_LISTEND
318 324 = RPL_CHANNELMODEIS
319 325 = RPL_UNIQOPIS
320 331 = RPL_NOTOPIC
321 332 = RPL_TOPIC
322 341 = RPL_INVITING
323 342 = RPL_SUMMONING
324 346 = RPL_INVITELIST
325 347 = RPL_ENDOFINVITELIST
326 348 = RPL_EXCEPTLIST
327 349 = RPL_ENDOFEXCEPTLIST
328 351 = RPL_VERSION
329 352 = RPL_WHOREPLY
330 353 = RPL_NAMREPLY
331 364 = RPL_LINKS
332 365 = RPL_ENDOFLINKS
333 366 = RPL_ENDOFNAMES
334 367 = RPL_BANLIST
335 368 = RPL_ENDOFBANLIST
336 369 = RPL_ENDOFWHOWAS
337 371 = RPL_INFO
338 372 = RPL_MOTD
339 374 = RPL_ENDOFINFO
340 375 = RPL_MOTDSTART
341 376 = RPL_ENDOFMOTD
342 381 = RPL_YOUREOPER
343 382 = RPL_REHASHING
344 383 = RPL_YOURESERVICE
345 391 = RPL_TIME
346 392 = RPL_USERSSTART
347 393 = RPL_USERS
348 394 = RPL_ENDOFUSERS
349 395 = RPL_NOUSERS
350 _ = undef
351
352 instance toInt IRCReplies where
353 toInt r = case r of
354 RPL_WELCOME = 1
355 RPL_YOURHOST = 2
356 RPL_CREATED = 3
357 RPL_MYINFO = 4
358 RPL_BOUNCE = 5
359 RPL_TRACELINK = 200
360 RPL_TRACECONNECTING = 201
361 RPL_TRACEHANDSHAKE = 202
362 RPL_TRACEUNKNOWN = 203
363 RPL_TRACEOPERATOR = 204
364 RPL_TRACEUSER = 205
365 RPL_TRACESERVER = 206
366 RPL_TRACESERVICE = 207
367 RPL_TRACENEWTYPE = 208
368 RPL_TRACECLASS = 209
369 RPL_TRACERECONNECT = 210
370 RPL_STATSLINKINFO = 211
371 RPL_STATSCOMMANDS = 212
372 RPL_ENDOFSTATS = 219
373 RPL_UMODEIS = 221
374 RPL_SERVLIST = 234
375 RPL_SERVLISTEND = 234
376 RPL_STATSUPTIME = 242
377 RPL_STATSOLINE = 243
378 RPL_LUSERCLIENT = 251
379 RPL_LUSEROP = 252
380 RPL_LUSERUNKNOWN = 253
381 RPL_LUSERCHANNELS = 254
382 RPL_LUSERME = 255
383 RPL_ADMINME = 256
384 RPL_ADMINLOC1 = 257
385 RPL_ADMINLOC2 = 258
386 RPL_ADMINEMAIL = 259
387 RPL_TRACELOG = 261
388 RPL_TRACEEND = 262
389 RPL_TRYAGAIN = 263
390 RPL_AWAY = 301
391 RPL_USERHOST = 302
392 RPL_ISON = 303
393 RPL_UNAWAY = 304
394 RPL_NOWAWAY = 305
395 RPL_WHOISUSER = 311
396 RPL_WHOISSERVER = 312
397 RPL_WHOISOPERATOR = 313
398 RPL_WHOWASUSER = 314
399 RPL_ENDOFWHO = 315
400 RPL_WHOISIDLE = 317
401 RPL_ENDOFWHOIS = 318
402 RPL_WHOISCHANNELS = 319
403 RPL_LISTSTART = 321
404 RPL_LIST = 322
405 RPL_LISTEND = 323
406 RPL_CHANNELMODEIS = 324
407 RPL_UNIQOPIS = 325
408 RPL_NOTOPIC = 331
409 RPL_TOPIC = 332
410 RPL_INVITING = 341
411 RPL_SUMMONING = 342
412 RPL_INVITELIST = 346
413 RPL_ENDOFINVITELIST = 347
414 RPL_EXCEPTLIST = 348
415 RPL_ENDOFEXCEPTLIST = 349
416 RPL_VERSION = 351
417 RPL_WHOREPLY = 352
418 RPL_NAMREPLY = 353
419 RPL_LINKS = 364
420 RPL_ENDOFLINKS = 365
421 RPL_ENDOFNAMES = 366
422 RPL_BANLIST = 367
423 RPL_ENDOFBANLIST = 367
424 RPL_ENDOFWHOWAS = 369
425 RPL_INFO = 371
426 RPL_MOTD = 372
427 RPL_ENDOFINFO = 374
428 RPL_MOTDSTART = 375
429 RPL_ENDOFMOTD = 376
430 RPL_YOUREOPER = 381
431 RPL_REHASHING = 382
432 RPL_YOURESERVICE = 383
433 RPL_TIME = 391
434 RPL_USERSSTART = 392
435 RPL_USERS = 393
436 RPL_ENDOFUSERS = 394
437 RPL_NOUSERS = 395
438
439 instance fromInt IRCErrors where
440 fromInt r = case r of
441 401 = ERR_NOSUCHNICK
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
449 409 = ERR_NOORIGIN
450 411 = ERR_NORECIPIENT
451 412 = ERR_NOTEXTTOSEND
452 413 = ERR_NOTOPLEVEL
453 414 = ERR_WILDTOPLEVEL
454 415 = ERR_BADMASK
455 421 = ERR_UNKNOWNCOMMAND
456 422 = ERR_NOMOTD
457 423 = ERR_NOADMININFO
458 424 = ERR_FILEERROR
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
467 444 = ERR_NOLOGIN
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
477 467 = ERR_KEYSET
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
489 484 = ERR_RESTRICTED
490 485 = ERR_UNIQOPPRIVSNEEDED
491 491 = ERR_NOOPERHOST
492 501 = ERR_UMODEUNKNOWNFLAG
493 502 = ERR_USERSDONTMATCH
494
495 instance toInt IRCErrors where
496 toInt r = case r of
497 ERR_NOSUCHNICK = 401
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
505 ERR_NOORIGIN = 409
506 ERR_NORECIPIENT = 411
507 ERR_NOTEXTTOSEND = 412
508 ERR_NOTOPLEVEL = 413
509 ERR_WILDTOPLEVEL = 414
510 ERR_BADMASK = 415
511 ERR_UNKNOWNCOMMAND = 421
512 ERR_NOMOTD = 422
513 ERR_NOADMININFO = 423
514 ERR_FILEERROR = 424
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
523 ERR_NOLOGIN = 444
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
533 ERR_KEYSET = 467
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
545 ERR_RESTRICTED = 484
546 ERR_UNIQOPPRIVSNEEDED = 485
547 ERR_NOOPERHOST = 491
548 ERR_UMODEUNKNOWNFLAG = 501
549 ERR_USERSDONTMATCH = 502