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
  12 import Text.Parsers.Simple.Core
  13 import Text.Parsers.Simple.Chars
  14 import Control.Monad
  15 import Control.Applicative
  16 from Data.Functor import <$>
  17
  18 from Data.Func import $
  19 from Text import class Text(concat), instance Text String
  20 import qualified Text
  21 from StdMisc import undef
  22
  23 jon :== 'Text'.join
  24
  25 derive gPrint IRCCommands, IRCReplies, IRCErrors, (,), Maybe, (), Either
  26
  27 :: IRCMessage =
  28         { irc_prefix :: Maybe (Either String IRCUser)
  29         , irc_command :: IRCCommands
  30         }
  31
  32 :: IRCUser =
  33         { irc_nick :: String
  34         , irc_user :: Maybe String
  35         , irc_host :: Maybe String
  36         }
  37
  38 //Start = runParser parseMessage $ fromString ":frobnicator!~frobnicat@92.110.128.124 PRIVMSG #cloogle :!query ^_^\r\n"
  39 Start = runParser parseMessage $ fromString ":frobnicator!~frobnicat@92.110.128.124 QUIT\r\n"
  40
  41 (<+) infixr 5 :: a b -> String | toString a & toString b
  42 (<+) a b = toString a +++ toString b
  43
  44 parseIRCMessage :: (String -> Either [Error] IRCMessage)
  45 parseIRCMessage = parse parseMessage o fromString
  46
  47 parseMessage :: Parser Char IRCMessage
  48 parseMessage = optional (parseEither parseHost parseUser) <* spaceParser
  49         >>= \mprefix->parseCommand
  50         <* pToken '\r' <* pToken '\n'
  51         >>= \cmd->pure {IRCMessage | irc_prefix=mprefix, irc_command=cmd}
  52
  53 pCommand :: String -> Parser Char [Char]
  54 pCommand s = pList (map pToken $ fromString s) <* spaceParser
  55
  56 parseCommand :: Parser Char IRCCommands
  57 parseCommand = pFail//pCommand "QUIT" >>| QUIT <$> optional (pure "")
  58
  59
  60 spaceParser :: Parser Char [Char]
  61 spaceParser = pMany $ pToken ' '
  62
  63 parseServer :: Parser Char String
  64 parseServer = pFail
  65
  66 parseEither :: (Parser a b) (Parser a c) -> Parser a (Either b c)
  67 parseEither p q = Left <$> p <|> Right <$> q
  68
  69 parseUser :: Parser Char IRCUser
  70 parseUser = pToken ':' >>| parseNick
  71                 >>= \nick->optional (pToken '!' >>| parseUsr)
  72                 >>= \muser->optional (pToken '@' >>| parseHost)
  73                 >>= \mhost->pure {IRCUser | irc_nick=nick, irc_user=muser, irc_host=mhost}
  74
  75 parseUsr :: Parser Char String
  76 parseUsr = toString <$> pSome (pSatisfy (not o flip isMember [' ', '\x00', '\x0d', '\x0a', '@']))
  77
  78 parseNick :: Parser Char String
  79 parseNick = pAlpha >>= \c->pMany (pAlpha <|> pDigit <|> pSpecial)
  80         >>= \cs->pure (toString [c:cs])
  81
  82 pSpecial :: Parser Char Char
  83 pSpecial = pOneOf ['-', '[', ']', '\\', '\`', '^', '{', '}']
  84
  85 parseHost :: Parser Char String
  86 parseHost = parseName
  87         >>= \nm->pMany (pToken '.' >>| parseName)
  88         >>= \nms->pure (concat [nm:nms])
  89         where
  90                 parseName :: Parser Char String
  91                 parseName = toString <$> pSome (pAlpha <|> pDigit <|> pToken '.')
  92
  93 IRCCommandParser :: Parser Char IRCCommands
  94 IRCCommandParser = pFail
  95
  96 instance toString IRCMessage where
  97         toString m = maybe "" (\s->either id ((<+) ":") s <+ " ") m.irc_prefix <+ m.irc_command
  98
  99 instance toString IRCUser where
 100         toString m = m.irc_nick <+ maybe "" ((<+) "!") m.irc_user
 101                 <+ maybe "" ((<+) "@") m.irc_host
 102
 103 instance toString IRCCommands where
 104         toString r = flip (+++) "\r\n" case r of
 105         //ADMIN (Maybe String)
 106         //AWAY String
 107         //CONNECT String Int (Maybe String)
 108         //DIE
 109         //ERROR String
 110         //INFO (Maybe String)
 111         //INVITE String String
 112         //ISON [String]
 113                 JOIN chs = "JOIN " +++ (if (isEmpty chs) "0"
 114                         (jon ", " [jon " " [ch:maybeToList mk]\\(ch, mk)<-chs]))
 115         //KICK String String (Maybe String)
 116         //KILL String String
 117         //LINKS (Maybe (Maybe String, String))
 118         //LIST [String]
 119         //LUSERS (Maybe (String, Maybe String))
 120         //MODE String
 121         //MOTD (Maybe String)
 122         //NAMES [String]
 123                 NICK n = jon " " ["NICK", n]
 124         //NJOIN
 125         //NOTICE String String
 126         //OPER String String
 127         //PART [String]
 128         //PASS String
 129                 PING a mb = jon " " ["PING",a:maybeToList mb]
 130                 PONG a mb = jon " " ["PONG",a:maybeToList mb]
 131                 PRIVMSG dest msg = jon " " ["PRIVMSG", dest, ":"+++msg]
 132                 QUIT msg = jon " " ["QUIT":maybeToList msg]
 133         //REHASH
 134         //RESTART
 135         //SERVER
 136         //SERVICE String String String String
 137         //SERVLIST (Maybe (String, Maybe String))
 138         //SQUERY String String
 139         //SQUIRT
 140         //SQUIT String String
 141         //STATS (Maybe (String, Maybe String))
 142         //SUMMON String (Maybe (String, Maybe String))
 143         //TIME (Maybe String)
 144         //TOPIC String (Maybe String)
 145         //TRACE (Maybe String)
 146                 USER login mode rn = jon " " ["USER", login, toString mode, "*", ":"+++rn]
 147         //USERHOST [String]
 148         //USERS (Maybe String)
 149         //VERSION (Maybe String)
 150         //WALLOPS String
 151         //WHO (Maybe String)
 152         //WHOIS (Maybe String) [String]
 153         //WHOWAS (Maybe String) [String]
 154                 _ = printToString r
 155
 156
 157 instance toString IRCReplies where toString r = printToString r
 158 instance toString IRCErrors where toString r = printToString r
 159
 160 instance fromInt IRCReplies where
 161         fromInt r = case r of
 162                 1 = RPL_WELCOME
 163                 2 = RPL_YOURHOST
 164                 3 = RPL_CREATED
 165                 4 = RPL_MYINFO
 166                 5 = RPL_BOUNCE
 167                 200 = RPL_TRACELINK
 168                 201 = RPL_TRACECONNECTING
 169                 202 = RPL_TRACEHANDSHAKE
 170                 203 = RPL_TRACEUNKNOWN
 171                 204 = RPL_TRACEOPERATOR
 172                 205 = RPL_TRACEUSER
 173                 206 = RPL_TRACESERVER
 174                 207 = RPL_TRACESERVICE
 175                 208 = RPL_TRACENEWTYPE
 176                 209 = RPL_TRACECLASS
 177                 210 = RPL_TRACERECONNECT
 178                 211 = RPL_STATSLINKINFO
 179                 212 = RPL_STATSCOMMANDS
 180                 219 = RPL_ENDOFSTATS
 181                 221 = RPL_UMODEIS
 182                 234 = RPL_SERVLIST
 183                 235 = RPL_SERVLISTEND
 184                 242 = RPL_STATSUPTIME
 185                 243 = RPL_STATSOLINE
 186                 251 = RPL_LUSERCLIENT
 187                 252 = RPL_LUSEROP
 188                 253 = RPL_LUSERUNKNOWN
 189                 254 = RPL_LUSERCHANNELS
 190                 255 = RPL_LUSERME
 191                 256 = RPL_ADMINME
 192                 257 = RPL_ADMINLOC1
 193                 258 = RPL_ADMINLOC2
 194                 259 = RPL_ADMINEMAIL
 195                 261 = RPL_TRACELOG
 196                 262 = RPL_TRACEEND
 197                 263 = RPL_TRYAGAIN
 198                 301 = RPL_AWAY
 199                 302 = RPL_USERHOST
 200                 303 = RPL_ISON
 201                 304 = RPL_UNAWAY
 202                 305 = RPL_NOWAWAY
 203                 311 = RPL_WHOISUSER
 204                 312 = RPL_WHOISSERVER
 205                 313 = RPL_WHOISOPERATOR
 206                 314 = RPL_WHOWASUSER
 207                 315 = RPL_ENDOFWHO
 208                 317 = RPL_WHOISIDLE
 209                 318 = RPL_ENDOFWHOIS
 210                 319 = RPL_WHOISCHANNELS
 211                 321 = RPL_LISTSTART
 212                 322 = RPL_LIST
 213                 323 = RPL_LISTEND
 214                 324 = RPL_CHANNELMODEIS
 215                 325 = RPL_UNIQOPIS
 216                 331 = RPL_NOTOPIC
 217                 332 = RPL_TOPIC
 218                 341 = RPL_INVITING
 219                 342 = RPL_SUMMONING
 220                 346 = RPL_INVITELIST
 221                 347 = RPL_ENDOFINVITELIST
 222                 348 = RPL_EXCEPTLIST
 223                 349 = RPL_ENDOFEXCEPTLIST
 224                 351 = RPL_VERSION
 225                 352 = RPL_WHOREPLY
 226                 353 = RPL_NAMREPLY
 227                 364 = RPL_LINKS
 228                 365 = RPL_ENDOFLINKS
 229                 366 = RPL_ENDOFNAMES
 230                 367 = RPL_BANLIST
 231                 368 = RPL_ENDOFBANLIST
 232                 369 = RPL_ENDOFWHOWAS
 233                 371 = RPL_INFO
 234                 372 = RPL_MOTD
 235                 374 = RPL_ENDOFINFO
 236                 375 = RPL_MOTDSTART
 237                 376 = RPL_ENDOFMOTD
 238                 381 = RPL_YOUREOPER
 239                 382 = RPL_REHASHING
 240                 383 = RPL_YOURESERVICE
 241                 391 = RPL_TIME
 242                 392 = RPL_USERSSTART
 243                 393 = RPL_USERS
 244                 394 = RPL_ENDOFUSERS
 245                 395 = RPL_NOUSERS
 246                 _ = undef
 247
 248 instance toInt IRCReplies where
 249         toInt r = case r of
 250                 RPL_WELCOME = 1
 251                 RPL_YOURHOST = 2
 252                 RPL_CREATED = 3
 253                 RPL_MYINFO = 4
 254                 RPL_BOUNCE = 5
 255                 RPL_TRACELINK = 200
 256                 RPL_TRACECONNECTING = 201
 257                 RPL_TRACEHANDSHAKE = 202
 258                 RPL_TRACEUNKNOWN = 203
 259                 RPL_TRACEOPERATOR = 204
 260                 RPL_TRACEUSER = 205
 261                 RPL_TRACESERVER = 206
 262                 RPL_TRACESERVICE = 207
 263                 RPL_TRACENEWTYPE = 208
 264                 RPL_TRACECLASS = 209
 265                 RPL_TRACERECONNECT = 210
 266                 RPL_STATSLINKINFO = 211
 267                 RPL_STATSCOMMANDS = 212
 268                 RPL_ENDOFSTATS = 219
 269                 RPL_UMODEIS = 221
 270                 RPL_SERVLIST = 234
 271                 RPL_SERVLISTEND = 234
 272                 RPL_STATSUPTIME = 242
 273                 RPL_STATSOLINE = 243
 274                 RPL_LUSERCLIENT = 251
 275                 RPL_LUSEROP = 252
 276                 RPL_LUSERUNKNOWN = 253
 277                 RPL_LUSERCHANNELS = 254
 278                 RPL_LUSERME = 255
 279                 RPL_ADMINME = 256
 280                 RPL_ADMINLOC1 = 257
 281                 RPL_ADMINLOC2 = 258
 282                 RPL_ADMINEMAIL = 259
 283                 RPL_TRACELOG = 261
 284                 RPL_TRACEEND = 262
 285                 RPL_TRYAGAIN = 263
 286                 RPL_AWAY = 301
 287                 RPL_USERHOST = 302
 288                 RPL_ISON = 303
 289                 RPL_UNAWAY = 304
 290                 RPL_NOWAWAY = 305
 291                 RPL_WHOISUSER = 311
 292                 RPL_WHOISSERVER = 312
 293                 RPL_WHOISOPERATOR = 313
 294                 RPL_WHOWASUSER = 314
 295                 RPL_ENDOFWHO = 315
 296                 RPL_WHOISIDLE = 317
 297                 RPL_ENDOFWHOIS = 318
 298                 RPL_WHOISCHANNELS = 319
 299                 RPL_LISTSTART = 321
 300                 RPL_LIST = 322
 301                 RPL_LISTEND = 323
 302                 RPL_CHANNELMODEIS = 324
 303                 RPL_UNIQOPIS = 325
 304                 RPL_NOTOPIC = 331
 305                 RPL_TOPIC = 332
 306                 RPL_INVITING = 341
 307                 RPL_SUMMONING = 342
 308                 RPL_INVITELIST = 346
 309                 RPL_ENDOFINVITELIST = 347
 310                 RPL_EXCEPTLIST = 348
 311                 RPL_ENDOFEXCEPTLIST = 349
 312                 RPL_VERSION = 351
 313                 RPL_WHOREPLY = 352
 314                 RPL_NAMREPLY = 353
 315                 RPL_LINKS = 364
 316                 RPL_ENDOFLINKS = 365
 317                 RPL_ENDOFNAMES = 366
 318                 RPL_BANLIST = 367
 319                 RPL_ENDOFBANLIST = 367
 320                 RPL_ENDOFWHOWAS = 369
 321                 RPL_INFO = 371
 322                 RPL_MOTD = 372
 323                 RPL_ENDOFINFO = 374
 324                 RPL_MOTDSTART = 375
 325                 RPL_ENDOFMOTD = 376
 326                 RPL_YOUREOPER = 381
 327                 RPL_REHASHING = 382
 328                 RPL_YOURESERVICE = 383
 329                 RPL_TIME = 391
 330                 RPL_USERSSTART = 392
 331                 RPL_USERS = 393
 332                 RPL_ENDOFUSERS = 394
 333                 RPL_NOUSERS = 395
 334
 335 instance fromInt IRCErrors where
 336         fromInt r = case r of
 337                 401 = ERR_NOSUCHNICK
 338                 402 = ERR_NOSUCHSERVER
 339                 403 = ERR_NOSUCHCHANNEL
 340                 404 = ERR_CANNOTSENDTOCHAN
 341                 405 = ERR_TOOMANYCHANNELS
 342                 406 = ERR_WASNOSUCHNICK
 343                 407 = ERR_TOOMANYTARGETS
 344                 408 = ERR_NOSUCHSERVICE
 345                 409 = ERR_NOORIGIN
 346                 411 = ERR_NORECIPIENT
 347                 412 = ERR_NOTEXTTOSEND
 348                 413 = ERR_NOTOPLEVEL
 349                 414 = ERR_WILDTOPLEVEL
 350                 415 = ERR_BADMASK
 351                 421 = ERR_UNKNOWNCOMMAND
 352                 422 = ERR_NOMOTD
 353                 423 = ERR_NOADMININFO
 354                 424 = ERR_FILEERROR
 355                 431 = ERR_NONICKNAMEGIVEN
 356                 432 = ERR_ERRONEUSNICKNAME
 357                 433 = ERR_NICKNAMEINUSE
 358                 436 = ERR_NICKCOLLISION
 359                 437 = ERR_UNAVAILRESOURCE
 360                 441 = ERR_USERNOTINCHANNEL
 361                 442 = ERR_NOTONCHANNEL
 362                 443 = ERR_USERONCHANNEL
 363                 444 = ERR_NOLOGIN
 364                 445 = ERR_SUMMONDISABLED
 365                 446 = ERR_USERSDISABLED
 366                 451 = ERR_NOTREGISTERED
 367                 461 = ERR_NEEDMOREPARAMS
 368                 462 = ERR_ALREADYREGISTRED
 369                 463 = ERR_NOPERMFORHOST
 370                 464 = ERR_PASSWDMISMATCH
 371                 465 = ERR_YOUREBANNEDCREEP
 372                 466 = ERR_YOUWILLBEBANNED
 373                 467 = ERR_KEYSET
 374                 471 = ERR_CHANNELISFULL
 375                 472 = ERR_UNKNOWNMODE
 376                 473 = ERR_INVITEONLYCHAN
 377                 474 = ERR_BANNEDFROMCHAN
 378                 475 = ERR_BADCHANNELKEY
 379                 476 = ERR_BADCHANMASK
 380                 477 = ERR_NOCHANMODES
 381                 478 = ERR_BANLISTFULL
 382                 481 = ERR_NOPRIVILEGES
 383                 482 = ERR_CHANOPRIVSNEEDED
 384                 483 = ERR_CANTKILLSERVER
 385                 484 = ERR_RESTRICTED
 386                 485 = ERR_UNIQOPPRIVSNEEDED
 387                 491 = ERR_NOOPERHOST
 388                 501 = ERR_UMODEUNKNOWNFLAG
 389                 502 = ERR_USERSDONTMATCH
 390
 391 instance toInt IRCErrors where
 392         toInt r = case r of
 393                 ERR_NOSUCHNICK = 401
 394                 ERR_NOSUCHSERVER = 402
 395                 ERR_NOSUCHCHANNEL = 403
 396                 ERR_CANNOTSENDTOCHAN = 404
 397                 ERR_TOOMANYCHANNELS = 405
 398                 ERR_WASNOSUCHNICK = 406
 399                 ERR_TOOMANYTARGETS = 407
 400                 ERR_NOSUCHSERVICE = 408
 401                 ERR_NOORIGIN = 409
 402                 ERR_NORECIPIENT = 411
 403                 ERR_NOTEXTTOSEND = 412
 404                 ERR_NOTOPLEVEL = 413
 405                 ERR_WILDTOPLEVEL = 414
 406                 ERR_BADMASK = 415
 407                 ERR_UNKNOWNCOMMAND = 421
 408                 ERR_NOMOTD = 422
 409                 ERR_NOADMININFO = 423
 410                 ERR_FILEERROR = 424
 411                 ERR_NONICKNAMEGIVEN = 431
 412                 ERR_ERRONEUSNICKNAME = 432
 413                 ERR_NICKNAMEINUSE = 433
 414                 ERR_NICKCOLLISION = 436
 415                 ERR_UNAVAILRESOURCE = 437
 416                 ERR_USERNOTINCHANNEL = 441
 417                 ERR_NOTONCHANNEL = 442
 418                 ERR_USERONCHANNEL = 443
 419                 ERR_NOLOGIN = 444
 420                 ERR_SUMMONDISABLED = 445
 421                 ERR_USERSDISABLED = 446
 422                 ERR_NOTREGISTERED = 451
 423                 ERR_NEEDMOREPARAMS = 461
 424                 ERR_ALREADYREGISTRED = 462
 425                 ERR_NOPERMFORHOST = 463
 426                 ERR_PASSWDMISMATCH = 464
 427                 ERR_YOUREBANNEDCREEP = 465
 428                 ERR_YOUWILLBEBANNED = 466
 429                 ERR_KEYSET = 467
 430                 ERR_CHANNELISFULL = 471
 431                 ERR_UNKNOWNMODE = 472
 432                 ERR_INVITEONLYCHAN = 473
 433                 ERR_BANNEDFROMCHAN = 474
 434                 ERR_BADCHANNELKEY = 475
 435                 ERR_BADCHANMASK = 476
 436                 ERR_NOCHANMODES = 477
 437                 ERR_BANLISTFULL = 478
 438                 ERR_NOPRIVILEGES = 481
 439                 ERR_CHANOPRIVSNEEDED = 482
 440                 ERR_CANTKILLSERVER = 483
 441                 ERR_RESTRICTED = 484
 442                 ERR_UNIQOPPRIVSNEEDED = 485
 443                 ERR_NOOPERHOST = 491
 444                 ERR_UMODEUNKNOWNFLAG = 501
 445                 ERR_USERSDONTMATCH = 502