implementation module IRC
-import StdList
-import GenPrint
-import StdOverloaded
-import Data.Maybe
-import Data.Either
-import StdFunc
-import StdString
-import StdChar
-import StdBool
+import StdList, StdTuple, StdOverloaded, StdFunc, StdString, StdChar, StdBool
import _SystemArray
-import Text.Parsers.Simple.Core
-import Text.Parsers.Simple.Chars
-import Data.Tuple
-import Control.Monad
+import GenPrint
+import GenIRC
+
import Control.Applicative
-from Data.Functor import <$>
+import Control.Monad
+import Data.Either
+import Data.Maybe
+import Data.Tuple
+import Text.Parsers.Simple.Chars
+import Text.Parsers.Simple.Core
+import StdDebug
+
+from Data.Functor import <$>
from Data.Func import $
-from Text import class Text(ltrim,indexOf,concat), instance Text String
+from StdMisc import undef, abort
+from Text import class Text(lpad,trim,rtrim,split,indexOf,concat),
+ instance Text String
import qualified Text
-from StdMisc import undef
-import StdDebug
jon :== 'Text'.join
-derive gPrint IRCCommand, IRCReplies, IRCErrors, (,), Maybe, (), Either
-
-//Start = runParser parseMessage $ fromString ":frobnicator!~frobnicat@92.110.128.124 QUIT\r\n"
-//Start = runParser parseMessage $ fromString ":frobnicator!~frobnicat@92.110.128.124 AWAY test\r\n"
-//Start = runParser parseMessage $ fromString ":frobnicator!~frobnicat@92.110.128.124 AWAY :test with spaces\r\n"
-//Start = runParser parseMessage $ fromString ":cherryh.freenode.net NOTICE * :*** Found your hostname\r\n"
-//Start = runParser parseMessage $ fromString ":cherryh.freenode.net QUIT :hoi hoi\r\n"
-Start = parseIRCMessage ":clooglebot!~cloogle@dhcp-077-249-221-037.chello.nl QUIT\r\n"
-
-(<+) infixr 5 :: a b -> String | toString a & toString b
-(<+) a b = toString a +++ toString b
+derive gPrint IRCErrors, IRCReplies, Maybe, Either, IRCUser, IRCNumReply
+
+Start = (map (fmap toString) msgs, msgs)
+where
+ msgs =
+ [ parseIRCMessage ":clooglebot!~cloogle@dhcp-077-249-221-037.chello.nl QUIT\r\n"
+ , parseIRCMessage ":clooglebot!~cloogle QUIT\r\n"
+ , parseIRCMessage ":frobnicator!~frobnicat@92.110.128.124 QUIT\r\n"
+ , parseIRCMessage ":frobnicator!~frobnicat@92.110.128.124 AWAY test\r\n"
+ , parseIRCMessage ":frobnicator!~frobnicat@92.110.128.124 AWAY :test with spaces\r\n"
+ , parseIRCMessage ":cherryh.freenode.net NOTICE * :*** Found your hostname\r\n"
+ , parseIRCMessage ":cherryh.freenode.net QUIT :hoi hoi\r\n"
+ , parseIRCMessage ":cherryh.freenode.net JOIN #cha,#ch-b #twilight\r\n"
+ , parseIRCMessage ":cherryh.freenode.net ISON a b c d e f :g h\r\n"
+ , parseIRCMessage ":wilhelm.freenode.net 001 clooglebot :Welcome to the freenode Internet Relay Chat Network clooglebot\r\n"
+ , parseIRCMessage "PING :orwell.freenode.net\r\n"
+ , parseIRCMessage ":ChanServ!ChanServ@services. MODE #cloogle +o frobnicator\r\n"
+ , parseIRCMessage ":qbot_v01!~qbot@ip-213-124-170-20.ip.prioritytelecom.net PRIVMSG ##chinees :[link] Cloogle - https://cloogle.org"
+ ]
parseIRCMessage :: String -> Either [Error] IRCMessage
parseIRCMessage s = case runParser parsePrefix (fromString s) of
- ([(prefix, rest):_], _) = case parse parseReply rest of
- Left e = case parseCmd rest of
- Left e2 = Left [e2:e]
- Right cmd = Right {IRCMessage | irc_prefix=prefix, irc_command=Right cmd}
- Right repl = Right {IRCMessage | irc_prefix=prefix, irc_command=Left repl}
- (_, es) = Left ["couldn't parse prefix":es]
-
-
+ // Prefix is parsed
+ ([(prefix, rest):_], _)
+ //Try parsing a numeric reply
+ = case parse parseReply rest of
+ //Try a normal command
+ Left e = case parseCmd rest of
+ Left e2 = Left [e2:e]
+ Right cmd = Right {IRCMessage | irc_prefix=prefix, irc_command=Right cmd}
+ Right repl = Right {IRCMessage | irc_prefix=prefix, irc_command=Left repl}
+ // Error parsing prefix
+ (_, es) = Left ["Error parsing prefix"]
+
+//Prefix
+parsePrefix :: Parser Char (Maybe (Either IRCUser String))
+parsePrefix
+ = optional (pToken ':' >>| parseEither parseUser parseHost <* pToken ' ')
+where
+ parseEither :: (Parser a b) (Parser a c) -> Parser a (Either b c)
+ parseEither p q = Left <$> p <|> Right <$> q
+
+ parseUser :: Parser Char IRCUser
+ parseUser = parseNick
+ >>= \nick->optional (pToken '!' >>| parseUsr)
+ >>= \muser->optional (pToken '@' >>| parseHost)
+ >>= \mhost->pure {IRCUser
+ | irc_nick=nick, irc_user=muser, irc_host=mhost}
+
+ parseUsr :: Parser Char String
+ parseUsr = toString <$> pSome (pNoneOf [' ', '@':illegal])
+
+ parseNick :: Parser Char String
+ parseNick = pAlpha
+ >>= \c ->pMany (pAlpha <|> pDigit <|> pOneOf (fromString "_-[]\\`^{}"))
+ >>= \cs->pure (toString [c:cs])
+
+ parseHost :: Parser Char String
+ parseHost = jon "." <$> (pSepBy parseName (pToken '.'))
+ >>= \s->optional (pToken '.') >>= pure o maybe s (\p->s+++toString s)
+ where
+ parseName :: Parser Char String
+ parseName = toString <$> pSome (pAlpha <|> pDigit <|> pOneOf ['-', '/'])
+
+//Parse Cmd
parseCmd :: [Char] -> Either Error IRCCommand
-parseCmd cs
-| not (trace_tn $ toString cs) = undef
-= processParse $ argfun $ 'Text'.split " " $ toString cs
+parseCmd cs = fst $ gIRCParse{|*|} $ argfun $ split " " $ toString cs
where
argfun :: [String] -> [String]
argfun [] = []
argfun [x:xs]
- # x = 'Text'.trim x
- | x.[0] == ':' = ['Text'.join " " [x:xs]]
+ # x = trim x
+ | x.[0] == ':' = [jon " " $ [x % (1, size x):map rtrim xs]]
| otherwise = [x:argfun xs]
- command0 :: IRCCommand [String] -> Either Error IRCCommand
- command0 c [] = Right c
- command0 c x = Left $ toString c +++ " doesn't have arguments"
-
- processParse :: [String] -> Either Error IRCCommand
- processParse [] = Left "Empty list of arguments"
- processParse [cmd:args] = case cmd of
- //"ADMIN" = (Maybe String)
- //"AWAY" = String
- //"CONNECT" = String (Maybe (Int, Maybe String))
- "DIE" = command0 DIE args
- //"ERROR" = String
- //"INFO" = (Maybe String)
- //"INVITE" = String String
- //"ISON" = [String]
- //"JOIN" = [(String, Maybe String)]
- //"KICK" = String String (Maybe String)
- //"KILL" = String String
- //"LINKS" = (Maybe (Maybe String, String))
- //"LIST" = (Maybe ([String], Maybe String))
- //"LUSERS" = (Maybe (String, Maybe String))
- //"MODE" = String String (Maybe String) (Maybe String) (Maybe String)
- //"MOTD" = (Maybe String)
- //"NAMES" = [String]
- //"NICK" = String (Maybe String)
- "NJOIN" = command0 NJOIN args
- //"NOTICE" = String String
- //"OPER" = String String
- //"PART" = [String]
- //"PASS" = String
- //"PING" = String (Maybe String)
- //"PONG" = String (Maybe String)
- //"PRIVMSG" = [String] String
- "QUIT" = case args of
- [_,_:_] = Left $ "QUIT has too many arguments"
- x = Right $ QUIT $ listToMaybe x
- "REHASH" = command0 REHASH args
- "RESTART" = command0 REHASH args
- "SERVER" = command0 REHASH args
- //"SERVICE" = String String String String
- //"SERVLIST" = (Maybe (String, Maybe String))
- //"SQUERY" = String String
- "SQUIRT" = command0 REHASH args
- //"SQUIT" = String String
- //"STATS" = (Maybe (String, Maybe String))
- //"SUMMON" = String (Maybe (String, Maybe String))
- //"TIME" = (Maybe String)
- //"TOPIC" = String (Maybe String)
- //"TRACE" = (Maybe String)
- //"USER" = String String String
- //"USERHOST" = [String]
- //"USERS" = (Maybe String)
- //"VERSION" = (Maybe String)
- //"WALLOPS" = String
- //"WHO" = (Maybe String)
- //"WHOIS" = (Maybe String) [String]
- //"WHOWAS" = (Maybe String) [String]
- _ = Left $ "Unknown command: " +++ cmd
-
-parsePrefix :: Parser Char (Maybe (Either String IRCUser))
-parsePrefix = optional (parseEither parseHost parseUser) <* spaceParser
-
+//Reply
parseReply :: Parser Char IRCNumReply
-parseReply = (toString <$> pSome pDigit)
- >>= \rep->pMiddle
- >>= \recipient->spaceParser >>| (toString <$> pSome (pNoneOf illegal))
- >>= \msg->pure {IRCNumReply|irc_reply=fs rep,irc_recipient=recipient,irc_message=msg}
+parseReply = spaceParser
+ >>| (pMany (pToken '0') >>| pSome pDigit <* spaceParser)
+ >>= \rep->(toString <$> pSome (pNoneOf [' ':illegal]) <* spaceParser)
+ >>= \rec->(toString <$> pSome (pNoneOf illegal))
+ >>= \msg->pure {IRCNumReply
+ | irc_reply = fromInt $ toInt $ toString rep
+ , irc_recipient = rec
+ , irc_message = msg % (if (msg.[0] == ':') 1 0, size msg)
+ }
+ <* pToken '\r' <* pToken '\n'
where
- fs :: String -> IRCReplies
- fs s = fromInt $ toInt s
-
-spaceParser :: Parser Char [Char]
-spaceParser = pMany $ pToken ' '
-
-parseServer :: Parser Char String
-parseServer = pFail
+ spaceParser :: Parser Char [Char]
+ spaceParser = pMany $ pToken ' '
-parseEither :: (Parser a b) (Parser a c) -> Parser a (Either b c)
-parseEither p q = Left <$> p <|> Right <$> q
-
-parseUser :: Parser Char IRCUser
-parseUser = pToken ':' >>| parseNick
- >>= \nick->optional (pToken '!' >>| parseUsr)
- >>= \muser->optional (pToken '@' >>| parseHost)
- >>= \mhost->pure {IRCUser | irc_nick=nick, irc_user=muser, irc_host=mhost}
-
-parseUsr :: Parser Char String
-parseUsr = toString <$> pSome (pNoneOf [' ', '@':illegal])
-
-parseNick :: Parser Char String
-parseNick = pAlpha >>= \c->pMany (pAlpha <|> pDigit <|> pSpecial)
- >>= \cs->pure (toString [c:cs])
-
-pSpecial :: Parser Char Char
-pSpecial = pOneOf ['-', '[', ']', '\\', '\`', '^', '{', '}']
+//Common parsers
+pNoneOf :: [a] -> Parser a a | Eq a
+pNoneOf l = pSatisfy (not o flip isMember l)
-parseHost :: Parser Char String
-parseHost = pToken ':' >>| (concat <$> pSepBy parseName (pToken '.'))
- where
- parseName :: Parser Char String
- parseName = toString <$> pSome (pAlpha <|> pDigit <|> pOneOf ['-'])
+illegal :: [Char]
+illegal = ['\x00','\r','\n']
instance toString IRCNumReply where
- toString m = toInt m.irc_reply <+ " " <+ m.irc_recipient <+ " " <+ formatMSG m.irc_message
+ toString m = lpad (toString $ toInt m.irc_reply) 3 '0' <+ " " <+
+ m.irc_recipient <+ " " <+ concat (gIRCPrint{|*|} m.irc_message)
instance toString IRCMessage where
- toString m = maybe "" (\s->either id ((<+) ":") s <+ " ") m.irc_prefix
+ toString m = maybe "" (\s->either ((<+) ":") id s <+ " ") m.irc_prefix
<+ either toString toString m.irc_command
-
instance toString IRCUser where
toString m = m.irc_nick <+ maybe "" ((<+) "!") m.irc_user
<+ maybe "" ((<+) "@") m.irc_host
-
-cons :: a [a] -> [a]
-cons a as = [a:as]
-
-pMiddle :: Parser Char String
-pMiddle = fmap toString $
- spaceParser >>| liftM2 cons (pNotSatisfy ((==)':')) (pMany $ pNoneOf [' ':illegal])
-
-pTrailing :: Parser Char String
-pTrailing = fmap toString $
- spaceParser >>| pToken ':' >>| pMany (pNoneOf illegal)
-
-pParam :: Parser Char String
-pParam = pMiddle <|> pTrailing
-
-pNoneOf :: [a] -> Parser a a | Eq a
-pNoneOf l = pSatisfy (not o flip isMember l)
-
-pNotSatisfy :: (a -> Bool) -> Parser a a | Eq a
-pNotSatisfy f = pSatisfy (not o f)
-
-pInt :: Parser Char Int
-pInt = toInt o toString <$> (spaceParser >>| pSome pDigit)
-
-illegal :: [Char]
-illegal = ['\x00','\r','\n']
-
instance toString IRCCommand where
- toString r = jon " " (print r) +++ "\r\n"
-
-print :: IRCCommand -> [String]
-print r = case r of
- ADMIN mm = ["ADMIN":maybeToList mm]
- //AWAY String
- //CONNECT String (Maybe (Int, Maybe String))
- //DIE
- //ERROR String
- //INFO (Maybe String)
- //INVITE String String
- //ISON [String]
- JOIN chs = ["JOIN",if (isEmpty chs) "0"
- (jon ", " [jon " " [ch:maybeToList mk]\\(ch, mk)<-chs])]
- //KICK String String (Maybe String)
- //KILL String String
- //LINKS (Maybe (Maybe String, String))
- //LIST (Maybe ([String], Maybe String))
- //LUSERS (Maybe (String, Maybe String))
- //MODE String String (Maybe String) (Maybe String) (Maybe String)
- //MOTD (Maybe String)
- //NAMES [String]
- NICK n ms = ["NICK", n]
- //NJOIN
- //NOTICE String String
- //OPER String String
- //PART [String]
- //PASS String
- PING a mb = ["PING",a:maybeToList mb]
- PONG a mb = ["PONG",a:maybeToList mb]
- PRIVMSG dest msg = ["PRIVMSG",jon "," dest,formatMSG msg]
- QUIT msg = ["QUIT":maybeToList msg]
- //REHASH
- //RESTART
- //SERVER
- //SERVICE String String String String
- //SERVLIST (Maybe (String, Maybe String))
- //SQUERY String String
- //SQUIRT
- //SQUIT String String
- //STATS (Maybe (String, Maybe String))
- //SUMMON String (Maybe (String, Maybe String))
- //TIME (Maybe String)
- //TOPIC String (Maybe String)
- //TRACE (Maybe String)
- USER login mode rn = ["USER", login, mode, "*", ":"+++rn]
- //USERHOST [String]
- //USERS (Maybe String)
- //VERSION (Maybe String)
- //WALLOPS String
- //WHO (Maybe String)
- //WHOIS (Maybe String) [String]
- //WHOWAS (Maybe String) [String]
- _ = [printToString r]
-
-formatMSG :: String -> String
-formatMSG s = if (indexOf " " s > 0 || indexOf " " s > 0) (":" +++ s) s
-
-
+ toString m = jon " " (gIRCPrint{|*|} m) +++ "\r\n"
instance toString IRCReplies where toString r = printToString r
instance toString IRCErrors where toString r = printToString r
+(<+) infixr 5 :: a b -> String | toString a & toString b
+(<+) a b = toString a +++ toString b
+
instance fromInt IRCReplies where
fromInt r = case r of
1 = RPL_WELCOME
393 = RPL_USERS
394 = RPL_ENDOFUSERS
395 = RPL_NOUSERS
- _ = undef
+ _ = RPL_UNKNOWN
instance toInt IRCReplies where
toInt r = case r of
RPL_USERS = 393
RPL_ENDOFUSERS = 394
RPL_NOUSERS = 395
+ RPL_UNKNOWN = 998
instance fromInt IRCErrors where
fromInt r = case r of
491 = ERR_NOOPERHOST
501 = ERR_UMODEUNKNOWNFLAG
502 = ERR_USERSDONTMATCH
+ _ = ERR_UNKNOWN
instance toInt IRCErrors where
toInt r = case r of
ERR_NOOPERHOST = 491
ERR_UMODEUNKNOWNFLAG = 501
ERR_USERSDONTMATCH = 502
+ ERR_UNKNOWN = 999