relay-code 0.1.0__py3-none-any.whl

ui/format.py

@@ -0,0 +1,145 @@
+from pathlib import Path
+from typing import Any, Tuple
+import re
+
+HEADLINE_KEYS = ("path", "command", "pattern", "url", "query", "action")
+
+BULKY_KEYS = frozenset({"content", "old_string", "new_string"})
+
+ARG_ORDER = {
+    "read": ["path", "offset", "limit"],
+    "write": ["path", "create_directories", "content"],
+    "edit": ["path", "replace_all", "old_string", "new_string"],
+    "shell": ['command', 'timeout', 'cwd'],
+    "list_dir": ['path', 'inclide_hidden'],
+    "grep": ['path', 'case_insensitive', 'pattern'],
+    "glob": ['path', 'pattern'],
+    "todo": ['action', 'id', 'content'],
+    "memory": ['action', 'key', 'value'],
+}
+
+_EXTENSION_LANGUAGES = {
+    ".py": "python",
+    ".js": "javascript",
+    ".jsx": "jsx",
+    ".ts": "typescript",
+    ".tsx": "tsx",
+    ".json": "json",
+    ".toml": "toml",
+    ".yaml": "yaml",
+    ".yml": "yaml",
+    ".md": "markdown",
+    ".sh": "bash",
+    ".bash": "bash",
+    ".zsh": "bash",
+    ".rs": "rust",
+    ".go": "go",
+    ".java": "java",
+    ".kt": "kotlin",
+    ".swift": "swift",
+    ".c": "c",
+    ".h": "c",
+    ".cpp": "cpp",
+    ".hpp": "cpp",
+    ".css": "css",
+    ".html": "html",
+    ".xml": "xml",
+    ".sql": "sql",
+}
+
+HEADLINE_MAX_WIDTH = 64
+
+
+def format_elapsed(seconds: float) -> str:
+    if seconds < 1:
+        return f"{seconds * 1000:.0f}ms"
+    if seconds < 60:
+        return f"{seconds:.1f}s"
+    minutes, secs = divmod(int(seconds), 60)
+    return f"{minutes}m{secs:02d}s"
+
+
+def guess_language(path: str | None) -> str:
+    if not path:
+        return "text"
+    return _EXTENSION_LANGUAGES.get(Path(path).suffix.lower(), "text")
+
+
+def ordered_args(tool_name: str, args: dict[str, Any]) -> list[Tuple[str, Any]]:
+    preferred = ARG_ORDER.get(tool_name, [])
+    ordered: list[Tuple[str, Any]] = []
+    seen: set[str] = set()
+
+    for key in preferred:
+        if key in args:
+            ordered.append((key, args[key]))
+            seen.add(key)
+
+    ordered.extend((key, args[key]) for key in sorted(args.keys() - seen))
+    return ordered
+
+
+def summarise_value(key: str, value: Any) -> str:
+    if isinstance(value, str) and key in BULKY_KEYS:
+        line_count = len(value.splitlines())
+        byte_count = len(value.encode("utf-8", errors="replace"))
+        return f"{line_count} lines ┈ {byte_count} bytes"
+    
+    if isinstance(value, bool):
+        value = str(value)
+        
+    return str(value)
+
+
+def headline(args: dict[str, Any]) -> tuple[str, str] | None:
+    for key in HEADLINE_KEYS:
+        value = args.get(key)
+        if isinstance(value, str) and value.strip():
+            first_line = value.strip().splitlines()[0]
+            if len(first_line) > HEADLINE_MAX_WIDTH:
+                first_line = f"{first_line[:HEADLINE_MAX_WIDTH - 1]}…"
+            return key, first_line
+    return None
+
+
+def secondary_args(args: dict[str, Any], headline_key: str | None) -> dict[str, Any]:
+    return {
+        key: value
+        for key, value in args.items()
+        if key != headline_key and not (key == "cwd" and value == ".")
+    }
+
+
+def extract_read_code(text: str) -> tuple[int, str] | None:
+    body = text
+    header_match = re.match(
+        r"^Showing lines (\d+)-(\d+) of (\d+)[^\n]*\n\n", text, re.IGNORECASE
+    )
+    if header_match:
+        body = text[header_match.end():]
+
+    code_lines: list[str] = []
+    start_line: int | None = None
+
+    for line in body.splitlines():
+        match = re.match(r"^\s*(\d+)\|(.*)$", line)
+        if not match:
+            return None
+        if start_line is None:
+            start_line = int(match.group(1))
+        code_lines.append(match.group(2))
+
+    if start_line is None:
+        return None
+
+    return start_line, "\n".join(code_lines)
+
+
+def diff_stat(diff: str) -> str:
+    added = removed = 0
+    for line in diff.splitlines():
+        if line.startswith("+") and not line.startswith("+++"):
+            added += 1
+        elif line.startswith("-") and not line.startswith("---"):
+            removed += 1
+    return f"+{added} -{removed}"

ui/logo.py

@@ -0,0 +1,46 @@
+from rich.text import Text
+import math
+
+RELAY_VERSION = "0.2"
+
+RELAY_LOGO = """\
+⣠⣤⣤⡤⠤⢤⣤⣀⡀⠀⠐⠒⡄⠀⡠⠒⠀⠀⢀⣀⣤⠤⠤⣤⣤⣤⡄
+⠈⠻⣿⡤⠤⡏⠀⠉⠙⠲⣄⠀⢰⢠⠃⢀⡤⠞⠋⠉⠈⢹⠤⢼⣿⠏⠀
+⠀⠀⠘⣿⡅⠓⢒⡤⠤⠀⡈⠱⣄⣼⡴⠋⡀⠀⠤⢤⡒⠓⢬⣿⠃⠀⠀
+⠀⠀⠀⠹⣿⣯⣐⢷⣀⣀⢤⡥⢾⣿⠷⢥⠤⣀⣀⣞⣢⣽⡿⠃⠀⠀⠀
+⠀⠀⠀⠀⠈⢙⣿⠝⠀⢁⠔⡨⡺⡿⡕⢔⠀⡈⠐⠹⣟⠋⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⢼⣟⢦⢶⢅⠜⢰⠃⠀⢹⡌⢢⣸⠦⠴⣿⡇⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠘⣿⣇⡬⡌⢀⡟⠀⠀⠀⢷⠀⣧⢧⣵⣿⠂⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠀⠈⢻⠛⠋⠉⠀⠀⠀⠀⠈⠉⠙⢻⡏⠀⠀⠀⠀⠀⠀⠀
+⠀⠀⠀⠀⠀⠀⢰⡿⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⣿⠄⠀⠀⠀⠀⠀⠀"""
+
+_WHITE = (255, 255, 255)
+_SILVER = (168, 174, 186)
+
+_PIXEL = "██"
+_LETTER_GAP = 1
+
+def small_wordmark(justify: str = "center") -> Text:
+    r, g, b = _SILVER
+    return Text(" ".join("relay"), style=f"rgb({r},{g},{b})", justify=justify, no_wrap=True)
+
+
+def gradient_logo(justify: str = "center") -> Text:
+    rows = RELAY_LOGO.split("\n")
+    height = len(rows)
+    width = max((len(row) for row in rows), default=1)
+    cx = (width - 1) / 2
+    cy = (height - 1) / 2
+    max_dist = math.hypot(cx, cy) or 1.0
+
+    text = Text(justify=justify, no_wrap=True)
+    for y, row in enumerate(rows):
+        for x, char in enumerate(row):
+            t = min(math.hypot(x - cx, y - cy) / max_dist, 1.0)
+            r = round(_WHITE[0] + (_SILVER[0] - _WHITE[0]) * t)
+            g = round(_WHITE[1] + (_SILVER[1] - _WHITE[1]) * t)
+            b = round(_WHITE[2] + (_SILVER[2] - _WHITE[2]) * t)
+            text.append(char, style=f"rgb({r},{g},{b})")
+        if y != height - 1:
+            text.append("\n")
+    return text