tabs and tab overview

scripts/generate-builtin-terminfo.py

@@ -0,0 +1,230 @@
+#!/usr/bin/env python3
+
+import argparse
+import os
+import re
+
+
+class Capability:
+    def __init__(self, name: str, value: bool | int | str):
+        self._name = name
+        self._value = value
+
+    @property
+    def name(self) -> str:
+        return self._name
+
+    @property
+    def value(self) -> bool | int | str:
+        return self._value
+
+    def __lt__(self, other: object) -> bool:
+        if not isinstance(other, Capability):
+            return NotImplemented
+        return self._name < other._name
+
+    def __le__(self, other: object) -> bool:
+        if not isinstance(other, Capability):
+            return NotImplemented
+        return self._name <= other._name
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, Capability):
+            return NotImplemented
+        return self._name == other._name
+
+    def __ne__(self, other: object) -> bool:
+        if not isinstance(other, Capability):
+            return NotImplemented
+        return bool(self._name != other._name)
+
+    def __gt__(self, other: object) -> bool:
+        if not isinstance(other, Capability):
+            return NotImplemented
+        return bool(self._name > other._name)
+
+    def __ge__(self, other: object) -> bool:
+        if not isinstance(other, Capability):
+            return NotImplemented
+        return self._name >= other._name
+
+
+class BoolCapability(Capability):
+    def __init__(self, name: str):
+        super().__init__(name, True)
+
+
+class IntCapability(Capability):
+    pass
+
+
+class StringCapability(Capability):
+    def __init__(self, name: str, value: str):
+        # see terminfo(5) for valid escape sequences
+
+        # Control characters
+        def translate_ctrl_chr(m: re.Match[str]) -> str:
+            ctrl = m.group(1)
+            if ctrl == '?':
+                return '\\x7f'
+            return f'\\x{ord(ctrl) - ord("@"):02x}'
+        value = re.sub(r'\^([@A-Z[\\\\\]^_?])', translate_ctrl_chr, value)
+
+        # Ensure e.g. \E7 (or \e7) doesn’t get translated to “\0337”,
+        # which would be interpreted as octal 337 by the C compiler
+        value = re.sub(r'(\\E|\\e)([0-7])', r'\\033" "\2', value)
+
+        # Replace \E and \e with ESC
+        value = re.sub(r'\\E|\\e', r'\\033', value)
+
+        # Unescape ,:^
+        value = re.sub(r'\\(,|:|\^)', r'\1', value)
+
+        # Replace \s with space
+        value = value.replace('\\s', ' ')
+
+        # Let \\, \n, \r, \t, \b and \f "fall through", to the C string literal
+
+        if re.search(r'\\l', value):
+            raise NotImplementedError('\\l escape sequence')
+
+        super().__init__(name, value)
+
+
+class Fragment:
+    def __init__(self, name: str, description: str):
+        self._name = name
+        self._description = description
+        self._caps = dict[str, Capability]()
+
+    @property
+    def name(self) -> str:
+        return self._name
+
+    @property
+    def description(self) -> str:
+        return self._description
+
+    @property
+    def caps(self) -> dict[str, Capability]:
+        return self._caps
+
+    def add_capability(self, cap: Capability) -> None:
+        assert cap.name not in self._caps
+        self._caps[cap.name] = cap
+
+    def del_capability(self, name: str) -> None:
+        del self._caps[name]
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser()
+    parser.add_argument('source_entry_name')
+    parser.add_argument('source', type=argparse.FileType('r'))
+    parser.add_argument('target_entry_name')
+    parser.add_argument('target', type=argparse.FileType('w'))
+
+    opts = parser.parse_args()
+    source_entry_name = opts.source_entry_name
+    target_entry_name = opts.target_entry_name
+    source = opts.source
+    target = opts.target
+
+    lines = list[str]()
+    for line in source.readlines():
+        line = line.strip()
+        if line.startswith('#'):
+            continue
+        lines.append(line)
+
+    fragments = dict[str, Fragment]()
+    cur_fragment: Fragment | None = None
+
+    for m in re.finditer(
+            r'(?P<name>(?P<entry_name>[-+\w@]+)\|(?P<entry_desc>.+?),)|'
+            r'(?P<bool_cap>(?P<bool_name>\w+),)|'
+            r'(?P<int_cap>(?P<int_name>\w+)#(?P<int_val>(0x)?[0-9a-fA-F]+),)|'
+            r'(?P<str_cap>(?P<str_name>\w+)=(?P<str_val>(.+?)),)',
+            ''.join(lines)):
+
+        if m.group('name') is not None:
+            name = m.group('entry_name')
+            description = m.group('entry_desc')
+
+            assert name not in fragments
+            fragments[name] = Fragment(name, description)
+            cur_fragment = fragments[name]
+
+        elif m.group('bool_cap') is not None:
+            name = m.group('bool_name')
+            assert cur_fragment is not None
+            cur_fragment.add_capability(BoolCapability(name))
+
+        elif m.group('int_cap') is not None:
+            name = m.group('int_name')
+            int_value = int(m.group('int_val'), 0)
+            assert cur_fragment is not None
+            cur_fragment.add_capability(IntCapability(name, int_value))
+
+        elif m.group('str_cap') is not None:
+            name = m.group('str_name')
+            str_value = m.group('str_val')
+            assert cur_fragment is not None
+            cur_fragment.add_capability(StringCapability(name, str_value))
+
+        else:
+            assert False
+
+    # Expand ‘use’ capabilities
+    for frag in fragments.values():
+        for cap in frag.caps.values():
+            if cap.name == 'use':
+                assert isinstance(cap, StringCapability)
+                assert isinstance(cap.value, str)
+
+                use_frag = fragments[cap.value]
+                for use_cap in use_frag.caps.values():
+                    frag.add_capability(use_cap)
+
+
+            frag.del_capability(cap.name)
+            break
+
+    entry = fragments[source_entry_name]
+
+    try:
+        entry.del_capability('RGB')
+    except KeyError:
+        pass
+
+    entry.add_capability(IntCapability('Co', 256))
+    entry.add_capability(StringCapability('TN', target_entry_name))
+    entry.add_capability(StringCapability('name', target_entry_name))
+    entry.add_capability(IntCapability('RGB', 8))  # 8 bits per channel
+    entry.add_capability(StringCapability('query-os-name', os.uname().sysname))
+
+    terminfo_parts = list[str]()
+    for cap in sorted(entry.caps.values()):
+        name = cap.name
+        value = str(cap.value)
+
+        # Escape ‘“‘
+        name = name.replace('"', '\"')
+        value = value.replace('"', '\"')
+
+        terminfo_parts.append(name)
+        if isinstance(cap, BoolCapability):
+            terminfo_parts.append('')
+        else:
+            terminfo_parts.append(value)
+
+    terminfo = '\\0" "'.join(terminfo_parts)
+
+    target.write('#pragma once\n')
+    target.write('\n')
+    target.write(f'static const char terminfo_capabilities[] = "{terminfo}";')
+    target.write('\n')
+
+
+if __name__ == '__main__':
+    main()