aboutsummaryrefslogtreecommitdiff
path: root/src/utils/irc/__init__.py
diff options
context:
space:
mode:
authorGravatar jesopo2019-10-28 10:57:19 +0000
committerGravatar jesopo2019-10-28 10:57:19 +0000
commit8983338680e02b2a039711c9e2464e13d9ada33b (patch)
tree10ea3559e0830ed3de06dfc487c09c721391b3b6 /src/utils/irc/__init__.py
parentremove src/utils/irc/protocol.py (diff)
signature
move src/utils/irc/__init__.py to src/utils/irc.py
Diffstat (limited to 'src/utils/irc/__init__.py')
-rw-r--r--src/utils/irc/__init__.py274
1 files changed, 0 insertions, 274 deletions
diff --git a/src/utils/irc/__init__.py b/src/utils/irc/__init__.py
deleted file mode 100644
index b398cb56..00000000
--- a/src/utils/irc/__init__.py
+++ /dev/null
@@ -1,274 +0,0 @@
-import json, string, re, typing, uuid
-from src import utils
-
-ASCII_UPPER = string.ascii_uppercase
-ASCII_LOWER = string.ascii_lowercase
-STRICT_RFC1459_UPPER = ASCII_UPPER+r'\[]'
-STRICT_RFC1459_LOWER = ASCII_LOWER+r'|{}'
-RFC1459_UPPER = STRICT_RFC1459_UPPER+"^"
-RFC1459_LOWER = STRICT_RFC1459_LOWER+"~"
-
-# case mapping lowercase/uppcase logic
-def multi_replace(s: str,
- chars1: typing.Iterable[str],
- chars2: typing.Iterable[str]) -> str:
- for char1, char2 in zip(chars1, chars2):
- s = s.replace(char1, char2)
- return s
-def lower(case_mapping: str, s: str) -> str:
- if case_mapping == "ascii":
- return multi_replace(s, ASCII_UPPER, ASCII_LOWER)
- elif case_mapping == "rfc1459":
- return multi_replace(s, RFC1459_UPPER, RFC1459_LOWER)
- elif case_mapping == "strict-rfc1459":
- return multi_replace(s, STRICT_RFC1459_UPPER, STRICT_RFC1459_LOWER)
- else:
- raise ValueError("unknown casemapping '%s'" % case_mapping)
-
-# compare a string while respecting case mapping
-def equals(case_mapping: str, s1: str, s2: str) -> bool:
- return lower(case_mapping, s1) == lower(case_mapping, s2)
-
-REGEX_COLOR = re.compile("%s(?:(\d{1,2})(?:,(\d{1,2}))?)?" % utils.consts.COLOR)
-
-def color(s: str, foreground: utils.consts.IRCColor,
- background: utils.consts.IRCColor=None) -> str:
- foreground_s = str(foreground.irc).zfill(2)
- background_s = ""
- if background:
- background_s = ",%s" % str(background.irc).zfill(2)
-
- return "%s%s%s%s%s" % (utils.consts.COLOR, foreground_s, background_s, s,
- utils.consts.COLOR)
-
-def bold(s: str) -> str:
- return "%s%s%s" % (utils.consts.BOLD, s, utils.consts.BOLD)
-
-def underline(s: str) -> str:
- return "%s%s%s" % (utils.consts.UNDERLINE, s, utils.consts.UNDERLINE)
-
-def strip_font(s: str) -> str:
- s = s.replace(utils.consts.BOLD, "")
- s = s.replace(utils.consts.ITALIC, "")
- s = REGEX_COLOR.sub("", s)
- s = s.replace(utils.consts.COLOR, "")
- return s
-
-FORMAT_TOKENS = [
- utils.consts.BOLD,
- utils.consts.RESET,
- utils.consts.UNDERLINE
-]
-FORMAT_STRIP = [
- "\x08" # backspace
-]
-def _format_tokens(s: str) -> typing.List[str]:
- is_color = False
- foreground = ""
- background = ""
- is_background = False
- matches = [] # type: typing.List[str]
-
- for i, char in enumerate(s):
- last_char = i == len(s)-1
- if is_color:
- current_color = background if is_background else foreground
- color_finished = True
- if char.isdigit() and len(current_color) < 2:
- if is_background:
- background += char
- else:
- foreground += char
- color_finished = (len(current_color)+1) == 2
- elif char == "," and not is_background:
- is_background = True
- color_finished = False
-
- if not char.isdigit() or last_char or color_finished:
- color = foreground
- if background:
- color += ","+background
-
- matches.append("\x03%s" % color)
- is_color = False
- foreground = ""
- background = ""
- is_background = False
-
- if char == utils.consts.COLOR:
- if is_color:
- matches.append(char)
- else:
- is_color = True
- elif char in FORMAT_TOKENS:
- matches.append(char)
- elif char in FORMAT_STRIP:
- matches.append(char)
- return matches
-
-def _color_match(code: typing.Optional[str], foreground: bool) -> str:
- if not code:
- return ""
- color = utils.consts.COLOR_CODES[int(code)]
- return color.to_ansi(not foreground)
-
-def parse_format(s: str) -> str:
- has_foreground = False
- has_background = False
- bold = False
- underline = False
-
- for token in _format_tokens(s):
- replace = ""
- type = token[0]
-
- if type == utils.consts.COLOR:
- match = REGEX_COLOR.match(token)
-
- if match and (match.group(1) or match.group(2)):
- foreground = _color_match(match.group(1), True)
- background = _color_match(match.group(2), False)
-
- if foreground:
- replace += foreground
- has_foreground = True
- if background:
- replace += background
- has_background = True
- else:
- if has_foreground:
- has_foreground = False
- replace += utils.consts.ANSI_FOREGROUND_RESET
- if has_background:
- has_background = False
- replace += utils.consts.ANSI_BACKGROUND_RESET
- elif type == utils.consts.BOLD:
- if bold:
- replace += utils.consts.ANSI_BOLD_RESET
- else:
- replace += utils.consts.ANSI_BOLD
- bold = not bold
- elif type == utils.consts.RESET:
- replace += utils.consts.ANSI_RESET
- elif type == utils.consts.UNDERLINE:
- if underline:
- replace += utils.consts.ANSI_UNDERLINE_RESET
- else:
- replace += utils.consts.ANSI_UNDERLINE
- underline = not underline
- elif type in FORMAT_STRIP:
- replace = ""
-
- s = s.replace(token, replace, 1)
-
- if has_foreground:
- s += utils.consts.ANSI_FOREGROUND_RESET
- if has_background:
- s += utils.consts.ANSI_BACKGROUND_RESET
- if bold:
- s += utils.consts.ANSI_BOLD_RESET
- if underline:
- s += utils.consts.ANSI_UNDERLINE_RESET
-
- return s
-
-OPT_STR = typing.Optional[str]
-class IRCConnectionParameters(object):
- def __init__(self, id: int, alias: str, hostname: str, port: int,
- password: OPT_STR, tls: bool, bindhost: OPT_STR, nickname: str,
- username: OPT_STR, realname: OPT_STR,
- args: typing.Dict[str, str]={}):
- self.id = id
- self.alias = alias
- self.hostname = hostname
- self.port = port
- self.tls = tls
- self.bindhost = bindhost
- self.password = password
- self.nickname = nickname
- self.username = username
- self.realname = realname
- self.args = args
-
-class CTCPMessage(object):
- def __init__(self, command: str, message: str):
- self.command = command
- self.message = message
-def parse_ctcp(s: str) -> typing.Optional[CTCPMessage]:
- ctcp = s.startswith("\x01")
- if s.startswith("\x01"):
- ctcp_command, sep, ctcp_message = s[1:].partition(" ")
- if ctcp_command.endswith("\x01"):
- ctcp_command = ctcp_command[:-1]
- if ctcp_message.endswith("\x01"):
- ctcp_message = ctcp_message[:-1]
- return CTCPMessage(ctcp_command, ctcp_message)
-
- return None
-
-class Capability(object):
- def __init__(self, ratified_name: typing.Optional[str],
- draft_name: str=None, alias: str=None,
- depends_on: typing.List[str]=None):
- self.alias = alias or ratified_name
- self._caps = set([ratified_name, draft_name])
- self.depends_on = depends_on or []
- self._on_ack_callbacks = [
- ] # type: typing.List[typing.Callable[[], None]]
-
- def available(self, capabilities: typing.Iterable[str]
- ) -> typing.Optional[str]:
- match = list(set(capabilities)&self._caps)
- return match[0] if match else None
-
- def match(self, capability: str) -> typing.Optional[str]:
- cap = list(set([capability])&self._caps)
- return cap[0] if cap else None
-
- def copy(self):
- return Capability(*self._caps, alias=self.alias,
- depends_on=self.depends_on[:])
-
- def on_ack(self, callback: typing.Callable[[], None]):
- self._on_ack_callbacks.append(callback)
- def ack(self):
- for callback in self._on_ack_callbacks:
- callback()
- def nak(self):
- pass
-
-class MessageTag(object):
- def __init__(self, name: typing.Optional[str], draft_name: str=None):
- self._names = set([name, draft_name])
- def get_value(self, tags: typing.Dict[str, str]) -> typing.Optional[str]:
- key = list(set(tags.keys())&self._names)
- return tags[key[0]] if key else None
- def present(self, tags: typing.Dict[str, str]) -> bool:
- return bool(set(tags.keys())&self._names)
- def match(self, tag: str) -> typing.Optional[str]:
- key = list(set([tag])&self._names)
- return key[0] if key else None
-
-class BatchType(object):
- def __init__(self, name: typing.Optional[str], draft_name: str=None):
- self._names = set([name, draft_name])
- def match(self, type: str) -> typing.Optional[str]:
- t = list(set([type])&self._names)
- return t[0] if t else None
-
-def hostmask_match_many(hostmasks: typing.List[str], pattern: str
- ) -> typing.Optional[str]:
- part1_out = []
- for part1 in pattern.split("?"):
- part2_out = []
- for part2 in part1.split("*"):
- part2_out.append(re.escape(part2))
- part1_out.append(".*".join(part2_out))
- pattern_re = re.compile(".".join(part1_out))
- for hostmask in hostmasks:
- if pattern_re.match(hostmask):
- return hostmask
- return None
-
-def hostmask_match(hostmask: str, pattern: str) -> bool:
- return not hostmask_match_many([hostmask], pattern) == None