From 7ee65f8f8ca9da93174284450db891346ea9daaf Mon Sep 17 00:00:00 2001 From: jesopo Date: Mon, 28 Oct 2019 10:56:33 +0000 Subject: remove src/utils/irc/protocol.py --- src/IRCServer.py | 68 +++++++++++++++++++---------------- src/utils/irc/__init__.py | 1 - src/utils/irc/protocol.py | 90 ----------------------------------------------- 3 files changed, 37 insertions(+), 122 deletions(-) delete mode 100644 src/utils/irc/protocol.py (limited to 'src') diff --git a/src/IRCServer.py b/src/IRCServer.py index 17b0bc13..87ed7f58 100644 --- a/src/IRCServer.py +++ b/src/IRCServer.py @@ -296,14 +296,18 @@ class Server(IRCObject.Object): def send_raw(self, line: str): return self.send(IRCLine.parse_line(line)) + def _line(self, command, args, tags={}): + return IRCLine.ParsedLine(command, + [arg for arg in args if not arg == None], tags=tags) + def send_user(self, username: str, realname: str ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.user(username, realname)) + return self.send(self._line("USER",[username, "0", "*", realname])) def send_nick(self, nickname: str) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.nick(nickname)) + return self.send(self._line("NICK", [nickname])) def send_capibility_ls(self) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.capability_ls()) + return self.send(self._line("CAP", ["LS", "302"])) def send_capability_queue(self): # textwrap works here because in ASCII, all chars are 1 bytes: capabilities = list(self.capability_queue.keys()) @@ -319,13 +323,13 @@ class Server(IRCObject.Object): for capability_batch in capability_batches: self.send_capability_request(capability_batch) - def send_capability_request(self, capability: str + def send_capability_request(self, capabilities: str ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.capability_request(capability)) + return self.send(self._line("CAP", ["REQ", capabilities])) def send_capability_end(self) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.capability_end()) + return self.send(self._line("CAP", ["END"])) def send_authenticate(self, text: str) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.authenticate(text)) + return self.send(self._line("AUTHENTICATE", [text])) def has_capability(self, capability: utils.irc.Capability) -> bool: return bool(self.available_capability(capability)) def has_capability_str(self, capability: str) -> bool: @@ -347,65 +351,67 @@ class Server(IRCObject.Object): self._capabilities_waiting.clear() def send_pass(self, password: str) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.password(password)) + return self.send(self._line("PASS", [password])) - def send_ping(self, nonce: str="hello" + def send_ping(self, token: str="hello" ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.ping(nonce), immediate=True) - def send_pong(self, nonce: str) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.pong(nonce), immediate=True) + return self.send(self._line("PING", [token])) + def send_pong(self, token: str) -> typing.Optional[IRCLine.SentLine]: + return self.send(self._line("PONG", [token])) - def send_join(self, channel_name: str, keys: typing.List[str]=None + def send_join(self, channel_name: str, key: typing.List[str]=None ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.join(channel_name, keys)) + return self.send(self._line("JOIN", [channel_name, key])) def send_joins(self, channel_names: typing.List[str], keys: typing.List[str]=None): - return self.send(utils.irc.protocol.join(",".join(channel_names), - keys)) + return self.send(self._line("JOIN", + [",".join(channel_names)]+(keys or []))) def send_part(self, channel_name: str, reason: str=None ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.part(channel_name, reason)) + return self.send(self._line("PART", [channel_name, reason])) def send_quit(self, reason: str="Leaving" ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.quit(reason)) + return self.send(self._line("QUIT", [reason])) def send_message(self, target: str, message: str, tags: dict={} ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.privmsg(target, message, tags)) + return self.send(self._line("PRIVMSG", [target, message], tags=tags)) def send_notice(self, target: str, message: str, tags: dict={} ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.notice(target, message, tags)) + return self.send(self._line("NOTICE", [target, message], tags=tags)) def send_tagmsg(self, target: str, tags: dict): - return self.send(utils.irc.protocol.tagmsg(target, tags)) + return self.send(self._line("TARGMSG", [], tags=tags)) def send_mode(self, target: str, mode: str=None, args: typing.List[str]=None ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.mode(target, mode, args)) + return self.send(self._line("MODE", [target, mode, args])) def send_topic(self, channel_name: str, topic: str ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.topic(channel_name, topic)) + return self.send(self._line("TOPIC", [channel_name, topic])) def send_kick(self, channel_name: str, target: str, reason: str=None ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.kick(channel_name, target, reason)) + return self.send(self._line("KICK", [channel_name, target, reason])) def send_names(self, channel_name: str) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.names(channel_name)) + return self.send(self._line("NAMES", [channel_name])) def send_list(self, search_for: str=None ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.list(search_for)) + return self.send(self._line("LIST", [search_for])) def send_invite(self, channel_name: str, target: str ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.invite(channel_name, target)) + return self.send(self._line("INVITE", [target, channel_name])) def send_whois(self, target: str) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.whois(target)) + return self.send(self._line("WHOIS", [target])) def send_whowas(self, target: str, amount: int=None, server: str=None ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.whowas(target, amount, server)) + amount_str = str(amount) if not amount == None else None + return self.send(self._line("WHOWAS", [target, amount_str, server])) def send_who(self, filter: str=None) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.who(filter)) + return self.send(self._line("WHO", [filter])) def send_whox(self, mask: str, filter: str, fields: str, label: str=None ) -> typing.Optional[IRCLine.SentLine]: - return self.send(utils.irc.protocol.whox(mask, filter, fields, label)) + flags = "%s%%%s%s" % (filter, fields, ","+label if label else "") + return self.send(self._line("WHO", [mask, flags])) diff --git a/src/utils/irc/__init__.py b/src/utils/irc/__init__.py index fa544c5d..b398cb56 100644 --- a/src/utils/irc/__init__.py +++ b/src/utils/irc/__init__.py @@ -1,6 +1,5 @@ import json, string, re, typing, uuid from src import utils -from . import protocol ASCII_UPPER = string.ascii_uppercase ASCII_LOWER = string.ascii_lowercase diff --git a/src/utils/irc/protocol.py b/src/utils/irc/protocol.py deleted file mode 100644 index ab2f7b5e..00000000 --- a/src/utils/irc/protocol.py +++ /dev/null @@ -1,90 +0,0 @@ -import typing -from src import IRCLine, utils - -def user(username: str, realname: str) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("USER", [username, "0", "*", realname]) -def nick(nickname: str) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("NICK", [nickname]) - -def capability_ls() -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("CAP", ["LS", "302"]) -def capability_request(capability: str) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("CAP", ["REQ", capability]) -def capability_end() -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("CAP", ["END"]) -def authenticate(text: str) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("AUTHENTICATE", [text]) - -def password(password: str) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("PASS", [password]) - -def ping(nonce: str="hello") -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("PING", [nonce]) -def pong(nonce: str="hello") -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("PONG", [nonce]) - -def join(channel_name: str, keys: typing.List[str]=None - ) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("JOIN", [channel_name]+( - keys if keys else [])) -def part(channel_name: str, reason: str=None) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("PART", [channel_name]+( - [reason] if reason else [])) -def quit(reason: str=None) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("QUIT", [reason] if reason else []) - -def privmsg(target: str, message: str, tags: typing.Dict[str, str]={} - ) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("PRIVMSG", [target, message], tags=tags) -def notice(target: str, message: str, tags: typing.Dict[str, str]={} - ) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("NOTICE", [target, message], tags=tags) -def tagmsg(target, tags: dict) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("TAGMSG", [target], tags=tags) - -def mode(target: str, mode: str=None, args: typing.List[str]=None - ) -> IRCLine.ParsedLine: - command_args = [target] - if mode: - command_args.append(mode) - if args: - command_args = command_args+args - return IRCLine.ParsedLine("MODE", command_args) - -def topic(channel_name: str, topic: str) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("TOPIC", [channel_name, topic]) -def kick(channel_name: str, target: str, reason: str=None - ) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("KICK", [channel_name, target]+( - [reason] if reason else [])) -def names(channel_name: str) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("NAMES", [channel_name]) -def list(search_for: str=None) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("LIST", [search_for] if search_for else []) -def invite(channel_name: str, target: str) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("INVITE", [target, channel_name]) - -def whois(target: str) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("WHOIS", [target]) -def whowas(target: str, amount: int=None, server: str=None - ) -> IRCLine.ParsedLine: - command_args = [target] - if amount: - command_args.append(str(amount)) - if server: - command_args.append(server) - return IRCLine.ParsedLine("WHOWAS", command_args) -def who(filter: str=None) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("WHO", [filter] if filter else []) -def whox(mask: str, filter: str, fields: str, label: str=None - ) -> IRCLine.ParsedLine: - flags = "%s%%%s%s" % (filter, fields, ","+label if label else "") - return IRCLine.ParsedLine("WHO", [mask, flags]) - -def batch_start(identifier: str, batch_type: str, tags: typing.Dict[str, str]={} - ) -> IRCLine.ParsedLine: - return IRCLine.ParsedLine("BATCH", ["+%s" % identifier, batch_type], - tags=tags) - -def batch_end(identifier: str, tags: typing.Dict[str, str]={}): - return IRCLine.ParsedLine("BATCH", ["-%s" % identifier], tags=tags) -- cgit v1.3.1-10-gc9f91