omlish 0.0.0.dev220__py3-none-any.whl → 0.0.0.dev221__py3-none-any.whl

omlish/specs/irc/format/rendering.py

@@ -0,0 +1,150 @@
+import typing as ta
+
+from .consts import MAX_LEN_CLIENT_TAG_DATA
+from .consts import MAX_LEN_SERVER_TAG_DATA
+from .consts import MAX_LEN_TAGS_FROM_CLIENT
+from .errors import BadCharactersError
+from .errors import BadParamError
+from .errors import CommandMissingError
+from .errors import InvalidTagContentError
+from .errors import TagsTooLongError
+from .message import Message
+from .tags import escape_tag_value
+from .tags import validate_tag_name
+from .tags import validate_tag_value
+from .utils import find_utf8_truncation_point
+
+
+def param_requires_trailing(param: str) -> bool:
+    return len(param) == 0 or ' ' in param or param[0] == ':'
+
+
+class RenderedLine(ta.NamedTuple):
+    raw: bytes
+
+    truncated: bool = False
+
+
+def render_line_(
+        msg: Message,
+        *,
+        tag_limit: int | None = None,
+        client_only_tag_data_limit: int | None = None,
+        server_added_tag_data_limit: int | None = None,
+        truncate_len: int | None = None,
+) -> RenderedLine:
+    if not msg.command:
+        raise CommandMissingError
+
+    buf = bytearray()
+    len_regular_tags = len_client_only_tags = 0
+
+    # Write the tags, computing the budgets for client-only tags and regular tags
+    if msg.tags or msg.client_only_tags:
+        buf.append(ord('@'))
+        first_tag = True
+        tag_error = None
+
+        def write_tags(tags: ta.Mapping[str, str]) -> None:
+            nonlocal first_tag, tag_error
+            for tag, val in tags.items():
+                if not (validate_tag_name(tag) and validate_tag_value(val)):
+                    tag_error = InvalidTagContentError
+                if not first_tag:
+                    buf.append(ord(';'))
+                buf.extend(tag.encode('utf-8'))
+                if val:
+                    buf.append(ord('='))
+                    buf.extend(escape_tag_value(val).encode('utf-8'))
+                first_tag = False
+
+        write_tags(msg.tags or {})
+        len_regular_tags = len(buf) - 1
+        write_tags(msg.client_only_tags or {})
+        len_client_only_tags = (len(buf) - 1) - len_regular_tags
+        if len_regular_tags:
+            # Semicolon between regular and client-only tags is not counted
+            len_client_only_tags -= 1
+        buf.append(ord(' '))
+        if tag_error:
+            raise tag_error
+
+    len_tags = len(buf)
+    if tag_limit is not None and len(buf) > tag_limit:
+        raise TagsTooLongError
+    if (
+            (client_only_tag_data_limit is not None and len_client_only_tags > client_only_tag_data_limit) or
+            (server_added_tag_data_limit is not None and len_regular_tags > server_added_tag_data_limit)
+    ):
+        raise TagsTooLongError
+
+    if msg.source:
+        buf.append(ord(':'))
+        buf.extend(msg.source.encode('utf-8'))
+        buf.append(ord(' '))
+
+    buf.extend(msg.command.encode('utf-8'))
+
+    for i, param in enumerate(msg.params):
+        buf.append(ord(' '))
+        requires_trailing = param_requires_trailing(param)
+        last_param = i == len(msg.params) - 1
+
+        if (requires_trailing or msg.force_trailing) and last_param:
+            buf.append(ord(':'))
+        elif requires_trailing and not last_param:
+            raise BadParamError
+
+        buf.extend(param.encode('utf-8'))
+
+    # Truncate if desired; leave 2 bytes over for \r\n:
+    truncated = False
+    if truncate_len is not None and (truncate_len - 2) < (len(buf) - len_tags):
+        truncated = True
+        new_buf_len = len_tags + (truncate_len - 2)
+        buf = buf[:find_utf8_truncation_point(buf, new_buf_len)]
+
+    buf.extend(b'\r\n')
+
+    to_validate = buf[:-2]
+    if any(c in to_validate for c in (b'\x00', b'\r', b'\n')):
+        raise BadCharactersError
+
+    raw = bytes(buf)
+
+    return RenderedLine(
+        raw=raw,
+        truncated=truncated,
+    )
+
+
+def render_line(msg: Message) -> bytes:
+    return render_line_(msg).raw
+
+
+def render_line_strict(
+        msg: Message,
+        from_client: bool,
+        truncate_len: int | None,
+) -> RenderedLine:
+    tag_limit: int | None = None
+    client_only_tag_data_limit: int | None = None
+    server_added_tag_data_limit: int | None = None
+    if from_client:
+        # enforce client max tags:
+        # <client_max>   (4096)  :: '@' <tag_data 4094> ' '
+        tag_limit = MAX_LEN_TAGS_FROM_CLIENT
+    else:
+        # on the server side, enforce separate client-only and server-added tag budgets:
+        # "Servers MUST NOT add tag data exceeding 4094 bytes to messages."
+        # <combined_max> (8191)  :: '@' <tag_data 4094> ';' <tag_data 4094> ' '
+        client_only_tag_data_limit = MAX_LEN_CLIENT_TAG_DATA
+        server_added_tag_data_limit = MAX_LEN_SERVER_TAG_DATA
+
+    return render_line_(
+        msg,
+        tag_limit=tag_limit,
+        client_only_tag_data_limit=client_only_tag_data_limit,
+        server_added_tag_data_limit=server_added_tag_data_limit,
+        truncate_len=truncate_len,
+    )

omlish/specs/irc/format/tags.py

@@ -0,0 +1,99 @@
+from .errors import InvalidTagContentError
+
+
+# Mapping for escaping tag values
+TAG_VAL_TO_ESCAPE = {
+    '\\': '\\\\',
+    ';': '\\:',
+    ' ': '\\s',
+    '\r': '\\r',
+    '\n': '\\n',
+}
+
+
+TAG_ESCAPE_CHAR_LOOKUP_TABLE = {i: chr(i) for i in range(256)}  # Most chars escape to themselves
+
+# These are the exceptions
+TAG_ESCAPE_CHAR_LOOKUP_TABLE.update({
+    ord(':'): ';',
+    ord('s'): ' ',
+    ord('r'): '\r',
+    ord('n'): '\n',
+})
+
+
+def escape_tag_value(in_string: str) -> str:
+    for key, val in TAG_VAL_TO_ESCAPE.items():
+        in_string = in_string.replace(key, val)
+    return in_string
+
+
+def unescape_tag_value(in_string: str) -> str:
+    if '\\' not in in_string:
+        return in_string
+
+    buf = []
+    remainder = in_string
+    while remainder:
+        backslash_pos = remainder.find('\\')
+        if backslash_pos == -1:
+            buf.append(remainder)
+            break
+        elif backslash_pos == len(remainder) - 1:
+            # Trailing backslash, which we strip
+            buf.append(remainder[:-1])
+            break
+
+        buf.append(remainder[:backslash_pos])
+        buf.append(TAG_ESCAPE_CHAR_LOOKUP_TABLE.get(ord(remainder[backslash_pos + 1]), remainder[backslash_pos + 1]))
+        remainder = remainder[backslash_pos + 2:]
+
+    return ''.join(buf)
+
+
+def validate_tag_name(name: str) -> bool:
+    if len(name) == 0:
+        return False
+    if name[0] == '+':
+        name = name[1:]
+    if len(name) == 0:
+        return False
+    # Let's err on the side of leniency here; allow -./ (45-47) in any position
+    for c in name:  # noqa
+        if not (('-' <= c <= '/') or ('0' <= c <= '9') or ('A' <= c <= 'Z') or ('a' <= c <= 'z')):
+            return False
+    return True
+
+
+def validate_tag_value(value: str) -> bool:
+    rt = value.encode('utf-8', 'ignore').decode('utf-8', 'ignore')
+    return value == rt
+
+
+def parse_tags(raw_tags: str) -> dict[str, str]:
+    dct: dict[str, str] = {}
+
+    while raw_tags:
+        tag_end = raw_tags.find(';')
+        if tag_end == -1:
+            tag_pair = raw_tags
+            raw_tags = ''
+        else:
+            tag_pair = raw_tags[:tag_end]
+            raw_tags = raw_tags[tag_end + 1:]
+
+        equals_index = tag_pair.find('=')
+        if equals_index == -1:
+            # Tag with no value
+            tag_name, tag_value = tag_pair, ''
+        else:
+            tag_name, tag_value = tag_pair[:equals_index], tag_pair[equals_index + 1:]
+
+        # "Implementations [...] MUST NOT perform any validation that would reject the message if an invalid tag key
+        # name is used."
+        if validate_tag_name(tag_name):
+            if not validate_tag_value(tag_value):
+                raise InvalidTagContentError
+            dct[tag_name] = unescape_tag_value(tag_value)
+
+    return dct

omlish/specs/irc/format/utils.py

@@ -0,0 +1,27 @@
+import itertools
+import operator
+
+
+def truncate_utf8_safe(string: str, length: int) -> str:
+    return string[:length] \
+        .encode('utf-8', 'ignore') \
+        .decode('utf-8', 'ignore')
+
+
+def find_utf8_truncation_point(buf: bytes | bytearray, length: int) -> int:
+    if len(buf) < length:
+        raise ValueError(buf)
+    cs = itertools.accumulate(
+        (len(c.encode('utf-8')) for c in buf.decode('utf-8')),
+        operator.add,
+        initial=0,
+    )
+    return next(i for i, o in enumerate(cs) if o >= length)
+
+
+def trim_initial_spaces(string: str) -> str:
+    return string.lstrip(' ')
+
+
+def is_ascii(string: str) -> bool:
+    return all(ord(c) < 128 for c in string)

omlish/specs/irc/numerics/formats.py

@@ -0,0 +1,94 @@
+import dataclasses as dc
+import typing as ta
+
+from .... import check
+from .... import lang
+
+
+FormatPart: ta.TypeAlias = ta.Union[str, 'Formats.Optional', 'Formats.Variadic']
+FormatParts: ta.TypeAlias = ta.Sequence[FormatPart]
+
+
+class Formats(lang.Namespace):
+    @dc.dataclass(frozen=True)
+    class Name:
+        name: str
+
+    @dc.dataclass(frozen=True)
+    class Optional:
+        body: FormatParts
+
+    @dc.dataclass(frozen=True)
+    class Variadic:
+        body: FormatParts
+
+    #
+
+    _PARTS_BY_DELIMITERS: ta.Mapping[tuple[str, str], type] = {
+        ('[', ']'): Optional,
+        ('{', '}'): Variadic,
+    }
+
+    _DELIMITERS_BY_PARTS: ta.Mapping[type, tuple[str, str]] = {v: k for k, v in _PARTS_BY_DELIMITERS.items()}
+
+    #
+
+    @staticmethod
+    def split_parts(s: str) -> FormatParts:
+        stk: list[tuple[str, list]] = [('', [])]
+
+        p = 0
+        while p < len(s):
+            n = lang.find_any(s, '{}[]<', p)
+
+            if n < 0:
+                check.state(not stk[-1][0])
+                stk[-1][1].append(s[p:])
+                break
+
+            if n != p:
+                stk[-1][1].append(s[p:n])
+
+            d = s[n]
+            if d == '<':
+                e = s.index('>', n)
+                stk[-1][1].append(Formats.Name(s[n + 1:e]))
+                p = e + 1
+
+            elif d in '{[':
+                stk.append((d, []))
+                p = n + 1
+
+            elif d in '}]':
+                x, l = stk.pop()
+                pc = Formats._PARTS_BY_DELIMITERS[(x, d)]
+                stk[-1][1].append(pc(l))
+                p = n + 1
+
+            else:
+                raise RuntimeError
+
+        _, ret = check.single(stk)
+        return ret
+
+    #
+
+    @staticmethod
+    def render_parts(p: FormatPart | FormatParts) -> ta.Iterator[str]:
+        if isinstance(p, str):
+            yield p
+
+        elif isinstance(p, Formats.Name):
+            yield '<'
+            yield p.name
+            yield '>'
+
+        elif isinstance(p, (Formats.Optional, Formats.Variadic)):
+            l, r = Formats._DELIMITERS_BY_PARTS[type(p)]
+            yield l
+            yield from Formats.render_parts(p.body)
+            yield r
+
+        else:
+            for c in p:
+                yield from Formats.render_parts(c)