IRC.icl

   1 implementation module IRC
   2
   3 import StdList, StdTuple, StdOverloaded, StdFunc, StdString, StdChar, StdBool
   4 import _SystemArray
   5
   6 import GenPrint
   7 import GenIRC
   8
   9 import Control.Applicative
  10 import Control.Monad
  11 import Data.Either
  12 import Data.Maybe
  13 import Data.Tuple
  14 import Text.Parsers.Simple.Chars
  15 import Text.Parsers.Simple.Core
  16
  17 import StdDebug
  18
  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),
  23         instance Text String
  24 import qualified Text
  25
  26 jon :== 'Text'.join
  27
  28 derive gPrint IRCErrors, IRCReplies, Maybe, Either, IRCUser, IRCNumReply
  29
  30 Start = (map (fmap toString) msgs, msgs)
  31 where
  32         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"
  46                 ]
  47
  48 parseIRCMessage :: String -> Either [Error] IRCMessage
  49 parseIRCMessage s = case runParser parsePrefix (fromString s) of
  50         // Prefix is parsed
  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
  56                                 Left e2 = Left [e2:e]
  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"]
  61
  62 //Prefix
  63 parsePrefix :: Parser Char (Maybe (Either IRCUser String))
  64 parsePrefix
  65         = optional (pToken ':' >>| parseEither parseUser parseHost <* pToken ' ')
  66 where
  67         parseEither :: (Parser a b) (Parser a c) -> Parser a (Either b c)
  68         parseEither p q = Left <$> p <|> Right <$> q
  69
  70         parseUser :: Parser Char IRCUser
  71         parseUser = parseNick
  72                         >>= \nick->optional (pToken '!' >>| parseUsr)
  73                         >>= \muser->optional (pToken '@' >>| parseHost)
  74                         >>= \mhost->pure {IRCUser
  75                                 | irc_nick=nick, irc_user=muser, irc_host=mhost}
  76
  77         parseUsr :: Parser Char String
  78         parseUsr = toString <$> pSome (pNoneOf [' ', '@':illegal])
  79
  80         parseNick :: Parser Char String
  81         parseNick = pAlpha
  82                 >>= \c ->pMany (pAlpha <|> pDigit <|> pOneOf (fromString "_-[]\\`^{}"))
  83                 >>= \cs->pure (toString [c:cs])
  84
  85         parseHost :: Parser Char String
  86         parseHost = jon "." <$> (pSepBy parseName (pToken '.'))
  87                 >>= \s->optional (pToken '.') >>= pure o maybe s (\p->s+++toString s)
  88                 where
  89                         parseName :: Parser Char String
  90                         parseName = toString <$> pSome (pAlpha <|> pDigit <|> pOneOf ['-', '/'])
  91
  92 //Parse Cmd
  93 parseCmd :: [Char] -> Either Error IRCCommand
  94 parseCmd cs = fst $ gIRCParse{|*|} $ argfun $ split " " $ toString cs
  95         where
  96                 argfun :: [String] -> [String]
  97                 argfun [] = []
  98                 argfun [x:xs]
  99                 # x = trim x
 100                 | x.[0] == ':' = [jon " " $ [x % (1, size x):map rtrim xs]]
 101                 | otherwise = [x:argfun xs]
 102
 103 //Reply
 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)
 113                 }
 114         <* pToken '\r' <* pToken '\n'
 115         where
 116                 spaceParser :: Parser Char [Char]
 117                 spaceParser = pMany $ pToken ' '
 118
 119 //Common parsers
 120 pNoneOf :: [a] -> Parser a a | Eq a
 121 pNoneOf l = pSatisfy (not o flip isMember l)
 122
 123 illegal :: [Char]
 124 illegal = ['\x00','\r','\n']
 125
 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
 139
 140 (<+) infixr 5 :: a b -> String | toString a & toString b
 141 (<+) a b = toString a +++ toString b
 142
 143 instance fromInt IRCReplies where
 144         fromInt r = case r of
 145                 1 = RPL_WELCOME
 146                 2 = RPL_YOURHOST
 147                 3 = RPL_CREATED
 148                 4 = RPL_MYINFO
 149                 5 = RPL_BOUNCE
 150                 200 = RPL_TRACELINK
 151                 201 = RPL_TRACECONNECTING
 152                 202 = RPL_TRACEHANDSHAKE
 153                 203 = RPL_TRACEUNKNOWN
 154                 204 = RPL_TRACEOPERATOR
 155                 205 = RPL_TRACEUSER
 156                 206 = RPL_TRACESERVER
 157                 207 = RPL_TRACESERVICE
 158                 208 = RPL_TRACENEWTYPE
 159                 209 = RPL_TRACECLASS
 160                 210 = RPL_TRACERECONNECT
 161                 211 = RPL_STATSLINKINFO
 162                 212 = RPL_STATSCOMMANDS
 163                 219 = RPL_ENDOFSTATS
 164                 221 = RPL_UMODEIS
 165                 234 = RPL_SERVLIST
 166                 235 = RPL_SERVLISTEND
 167                 242 = RPL_STATSUPTIME
 168                 243 = RPL_STATSOLINE
 169                 251 = RPL_LUSERCLIENT
 170                 252 = RPL_LUSEROP
 171                 253 = RPL_LUSERUNKNOWN
 172                 254 = RPL_LUSERCHANNELS
 173                 255 = RPL_LUSERME
 174                 256 = RPL_ADMINME
 175                 257 = RPL_ADMINLOC1
 176                 258 = RPL_ADMINLOC2
 177                 259 = RPL_ADMINEMAIL
 178                 261 = RPL_TRACELOG
 179                 262 = RPL_TRACEEND
 180                 263 = RPL_TRYAGAIN
 181                 301 = RPL_AWAY
 182                 302 = RPL_USERHOST
 183                 303 = RPL_ISON
 184                 304 = RPL_UNAWAY
 185                 305 = RPL_NOWAWAY
 186                 311 = RPL_WHOISUSER
 187                 312 = RPL_WHOISSERVER
 188                 313 = RPL_WHOISOPERATOR
 189                 314 = RPL_WHOWASUSER
 190                 315 = RPL_ENDOFWHO
 191                 317 = RPL_WHOISIDLE
 192                 318 = RPL_ENDOFWHOIS
 193                 319 = RPL_WHOISCHANNELS
 194                 321 = RPL_LISTSTART
 195                 322 = RPL_LIST
 196                 323 = RPL_LISTEND
 197                 324 = RPL_CHANNELMODEIS
 198                 325 = RPL_UNIQOPIS
 199                 331 = RPL_NOTOPIC
 200                 332 = RPL_TOPIC
 201                 341 = RPL_INVITING
 202                 342 = RPL_SUMMONING
 203                 346 = RPL_INVITELIST
 204                 347 = RPL_ENDOFINVITELIST
 205                 348 = RPL_EXCEPTLIST
 206                 349 = RPL_ENDOFEXCEPTLIST
 207                 351 = RPL_VERSION
 208                 352 = RPL_WHOREPLY
 209                 353 = RPL_NAMREPLY
 210                 364 = RPL_LINKS
 211                 365 = RPL_ENDOFLINKS
 212                 366 = RPL_ENDOFNAMES
 213                 367 = RPL_BANLIST
 214                 368 = RPL_ENDOFBANLIST
 215                 369 = RPL_ENDOFWHOWAS
 216                 371 = RPL_INFO
 217                 372 = RPL_MOTD
 218                 374 = RPL_ENDOFINFO
 219                 375 = RPL_MOTDSTART
 220                 376 = RPL_ENDOFMOTD
 221                 381 = RPL_YOUREOPER
 222                 382 = RPL_REHASHING
 223                 383 = RPL_YOURESERVICE
 224                 391 = RPL_TIME
 225                 392 = RPL_USERSSTART
 226                 393 = RPL_USERS
 227                 394 = RPL_ENDOFUSERS
 228                 395 = RPL_NOUSERS
 229                 _ = RPL_UNKNOWN
 230
 231 instance toInt IRCReplies where
 232         toInt r = case r of
 233                 RPL_WELCOME = 1
 234                 RPL_YOURHOST = 2
 235                 RPL_CREATED = 3
 236                 RPL_MYINFO = 4
 237                 RPL_BOUNCE = 5
 238                 RPL_TRACELINK = 200
 239                 RPL_TRACECONNECTING = 201
 240                 RPL_TRACEHANDSHAKE = 202
 241                 RPL_TRACEUNKNOWN = 203
 242                 RPL_TRACEOPERATOR = 204
 243                 RPL_TRACEUSER = 205
 244                 RPL_TRACESERVER = 206
 245                 RPL_TRACESERVICE = 207
 246                 RPL_TRACENEWTYPE = 208
 247                 RPL_TRACECLASS = 209
 248                 RPL_TRACERECONNECT = 210
 249                 RPL_STATSLINKINFO = 211
 250                 RPL_STATSCOMMANDS = 212
 251                 RPL_ENDOFSTATS = 219
 252                 RPL_UMODEIS = 221
 253                 RPL_SERVLIST = 234
 254                 RPL_SERVLISTEND = 234
 255                 RPL_STATSUPTIME = 242
 256                 RPL_STATSOLINE = 243
 257                 RPL_LUSERCLIENT = 251
 258                 RPL_LUSEROP = 252
 259                 RPL_LUSERUNKNOWN = 253
 260                 RPL_LUSERCHANNELS = 254
 261                 RPL_LUSERME = 255
 262                 RPL_ADMINME = 256
 263                 RPL_ADMINLOC1 = 257
 264                 RPL_ADMINLOC2 = 258
 265                 RPL_ADMINEMAIL = 259
 266                 RPL_TRACELOG = 261
 267                 RPL_TRACEEND = 262
 268                 RPL_TRYAGAIN = 263
 269                 RPL_AWAY = 301
 270                 RPL_USERHOST = 302
 271                 RPL_ISON = 303
 272                 RPL_UNAWAY = 304
 273                 RPL_NOWAWAY = 305
 274                 RPL_WHOISUSER = 311
 275                 RPL_WHOISSERVER = 312
 276                 RPL_WHOISOPERATOR = 313
 277                 RPL_WHOWASUSER = 314
 278                 RPL_ENDOFWHO = 315
 279                 RPL_WHOISIDLE = 317
 280                 RPL_ENDOFWHOIS = 318
 281                 RPL_WHOISCHANNELS = 319
 282                 RPL_LISTSTART = 321
 283                 RPL_LIST = 322
 284                 RPL_LISTEND = 323
 285                 RPL_CHANNELMODEIS = 324
 286                 RPL_UNIQOPIS = 325
 287                 RPL_NOTOPIC = 331
 288                 RPL_TOPIC = 332
 289                 RPL_INVITING = 341
 290                 RPL_SUMMONING = 342
 291                 RPL_INVITELIST = 346
 292                 RPL_ENDOFINVITELIST = 347
 293                 RPL_EXCEPTLIST = 348
 294                 RPL_ENDOFEXCEPTLIST = 349
 295                 RPL_VERSION = 351
 296                 RPL_WHOREPLY = 352
 297                 RPL_NAMREPLY = 353
 298                 RPL_LINKS = 364
 299                 RPL_ENDOFLINKS = 365
 300                 RPL_ENDOFNAMES = 366
 301                 RPL_BANLIST = 367
 302                 RPL_ENDOFBANLIST = 367
 303                 RPL_ENDOFWHOWAS = 369
 304                 RPL_INFO = 371
 305                 RPL_MOTD = 372
 306                 RPL_ENDOFINFO = 374
 307                 RPL_MOTDSTART = 375
 308                 RPL_ENDOFMOTD = 376
 309                 RPL_YOUREOPER = 381
 310                 RPL_REHASHING = 382
 311                 RPL_YOURESERVICE = 383
 312                 RPL_TIME = 391
 313                 RPL_USERSSTART = 392
 314                 RPL_USERS = 393
 315                 RPL_ENDOFUSERS = 394
 316                 RPL_NOUSERS = 395
 317                 RPL_UNKNOWN = 998
 318
 319 instance fromInt IRCErrors where
 320         fromInt r = case r of
 321                 401 = ERR_NOSUCHNICK
 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
 329                 409 = ERR_NOORIGIN
 330                 411 = ERR_NORECIPIENT
 331                 412 = ERR_NOTEXTTOSEND
 332                 413 = ERR_NOTOPLEVEL
 333                 414 = ERR_WILDTOPLEVEL
 334                 415 = ERR_BADMASK
 335                 421 = ERR_UNKNOWNCOMMAND
 336                 422 = ERR_NOMOTD
 337                 423 = ERR_NOADMININFO
 338                 424 = ERR_FILEERROR
 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
 347                 444 = ERR_NOLOGIN
 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
 357                 467 = ERR_KEYSET
 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
 369                 484 = ERR_RESTRICTED
 370                 485 = ERR_UNIQOPPRIVSNEEDED
 371                 491 = ERR_NOOPERHOST
 372                 501 = ERR_UMODEUNKNOWNFLAG
 373                 502 = ERR_USERSDONTMATCH
 374                 _ = ERR_UNKNOWN
 375
 376 instance toInt IRCErrors where
 377         toInt r = case r of
 378                 ERR_NOSUCHNICK = 401
 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
 386                 ERR_NOORIGIN = 409
 387                 ERR_NORECIPIENT = 411
 388                 ERR_NOTEXTTOSEND = 412
 389                 ERR_NOTOPLEVEL = 413
 390                 ERR_WILDTOPLEVEL = 414
 391                 ERR_BADMASK = 415
 392                 ERR_UNKNOWNCOMMAND = 421
 393                 ERR_NOMOTD = 422
 394                 ERR_NOADMININFO = 423
 395                 ERR_FILEERROR = 424
 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
 404                 ERR_NOLOGIN = 444
 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
 414                 ERR_KEYSET = 467
 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
 426                 ERR_RESTRICTED = 484
 427                 ERR_UNIQOPPRIVSNEEDED = 485
 428                 ERR_NOOPERHOST = 491
 429                 ERR_UMODEUNKNOWNFLAG = 501
 430                 ERR_USERSDONTMATCH = 502
 431                 ERR_UNKNOWN = 999