|
|
- from typing import Any, Callable, List, Optional, Union
-
- from paircd.message import Message
-
-
- def cmd_fn(cmd: str, tmpl: str) -> Callable:
- def fn(*args: List[Any], prefix: Optional[str] = None) -> Message:
- msg = tmpl.format(*args)
- return (
- Message(cmd=cmd, args=[msg], prefix=prefix)
- if prefix is not None
- else Message(cmd=cmd, args=[msg])
- )
-
- return fn
-
-
- def reply_fn(cmd: Union[int, str], tmpl: str) -> Callable:
- def fn(target: str, *args: List[Any]) -> Message:
- msg = f"{target} {tmpl.format(*args)}"
- return Message(cmd=str(cmd), args=[msg])
-
- return fn
-
-
- # Commands
- JOIN = cmd_fn("JOIN", "{0}")
- MODE = cmd_fn("MODE", "{0} {1}")
- PONG = cmd_fn("PONG", ":{0}")
- PRIVMSG = cmd_fn("PRIVMSG", "{0} :{1}")
-
- # Error replies
- ERR_NOSUCHNICK = reply_fn(401, "{0} :No such nick/channel")
- ERR_NOSUCHSERVER = reply_fn(402, "{0} :No such server")
- ERR_NOSUCHCHANNEL = reply_fn(403, "{0} :No such channel")
- ERR_CANNOTSENDTOCHAN = reply_fn(404, "{0} :Cannot send to channel")
- ERR_TOOMANYCHANNELS = reply_fn(405, "{0} :You have joined too many channels")
- ERR_NOTEXTTOSEND = reply_fn(412, ":No text to send")
- ERR_NOTREGISTERED = reply_fn(451, ":You have not registered")
-
- # Command responses
- RPL_WELCOME = reply_fn("001", "Welcome to the Internet Relay Network {0}")
- RPL_YOURHOST = reply_fn("002", "Your host is {0}, running version {1}")
- RPL_CREATED = reply_fn("003", "This server was created {0}")
- RPL_MYINFO = reply_fn("004", "{0} {1} {2} {3}")
- RPL_AWAY = reply_fn(301, "{0} :{1}")
- RPL_UNAWAY = reply_fn(305, ":You are no longer marked as being away")
- RPL_NOWAWAY = reply_fn(306, ":You have been marked as being away")
- RPL_WHOISUSER = reply_fn(311, "{0} {1} {2} * :{3}")
- RPL_ENDOFWHO = reply_fn(315, "{0} :End of /WHO list")
- RPL_ENDOFWHOIS = reply_fn(318, "{0} :End of /WHOIS list")
- RPL_CHANNELMODEIS = reply_fn(324, "{0} {1}{2}")
- RPL_TOPIC = reply_fn(332, "{0} :{1}")
- RPL_WHOREPLY = reply_fn(352, "{0} ~{1} {2} {3} {4} {5} :{6} {7}")
- RPL_NAMREPLY = reply_fn(353, "{0} {1} :{2}")
- RPL_ENDOFNAMES = reply_fn(366, "{0} :End of /NAMES list")
|
