frames.py

import re
import json
from PyQt5.QtGui import QTextCursor

# ── Colour helpers ────────────────────────────────────────────────────────────
serverFormat  = '<span style="color:#29F500;">{}</span>'
allFormat     = '<span style="color:gray;">{}</span>'
uiFormat      = '<span style="color:yellow;">{}</span>'
discFormat    = '<span style="color:red;">{}</span>'
warnFormat    = '<span style="color:orange;">{}</span>'
messageFormat = '<span style="color:white;">{}</span>'

# Regex patterns for firmware text lines
_RE_CHAT    = re.compile(r'^\[CHAT\]\[([0-9A-Fa-f]{8})\]:\s*(.*)')
_RE_TX_CHAT = re.compile(r'^TX CHAT:\s*(.*)')
_RE_DROP    = re.compile(r'^@DROP\s+(.*)')
_RE_MET     = re.compile(r'^@MET\s+(\{.*\})')
_RE_SUPPORT = re.compile(r'^@SUPPORTPACK\s+(\{.*\})')
_RE_POLICY  = re.compile(r'^@POLICY\s+(\{.*\})')
_RE_RST     = re.compile(r'^@RST\s+(\{.*\})')
_RE_CRASH   = re.compile(r'^@CRASH\s+(\{.*\})')
_RE_OK      = re.compile(r'^OK\s+(.*)')
_RE_ERR     = re.compile(r'^ERR:\s*(.*)')
_RE_WARN    = re.compile(r'^WARN:\s*(.*)')
# ESP-IDF log:  I (12345) TAG: message  or  W/E/D/V
_RE_ESP_LOG = re.compile(r'^([IWEDV])\s+\((\d+)\)\s+(\w+):\s*(.*)')


class RivrLineHandler:
    """Parse and display a single text line from Rivr firmware serial output."""

    def received_line(self, line: str, ui) -> None:
        line = line.strip()
        if not line or line == '>':
            return

        m = _RE_CHAT.match(line)
        if m:
            node_id, msg = m.group(1), m.group(2)
            ui.tEMonitor.append(
                allFormat.format(f'[CHAT from 0x{node_id}]')
            )
            ui.tEMonitor.append(messageFormat.format(f'  {msg}'))
            ui.tEMonitor.moveCursor(QTextCursor.End)
            # also push to chat-only view if available
            if hasattr(ui, 'tEChat'):
                ui.tEChat.append(
                    allFormat.format(f'[0x{node_id}]')
                )
                ui.tEChat.append(messageFormat.format(f'  {msg}'))
                ui.tEChat.moveCursor(QTextCursor.End)
            return

        m = _RE_TX_CHAT.match(line)
        if m:
            ui.tEMonitor.append(
                uiFormat.format(f'[TX] {m.group(1)}')
            )
            ui.tEMonitor.moveCursor(QTextCursor.End)
            if hasattr(ui, 'tEChat'):
                ui.tEChat.append(uiFormat.format(f'[TX] {m.group(1)}'))
                ui.tEChat.moveCursor(QTextCursor.End)
            return

        m = _RE_DROP.match(line)
        if m:
            ui.tEMonitor.append(discFormat.format(f'[@DROP] {m.group(1)}'))
            ui.tEMonitor.moveCursor(QTextCursor.End)
            return

        m = _RE_MET.match(line)
        if m:
            try:
                d = json.loads(m.group(1))
                # Header line
                ui.tEMonitor.append(allFormat.format(
                    f'── @MET node=0x{d.get("node_id", 0):08X} ──'
                ))
                # Print every key=value pair, 4 per line
                items = [f'{k}={v}' for k, v in d.items() if k != 'node_id']
                row = []
                for item in items:
                    row.append(item)
                    if len(row) == 4:
                        ui.tEMonitor.append(allFormat.format('  ' + '  '.join(row)))
                        row = []
                if row:
                    ui.tEMonitor.append(allFormat.format('  ' + '  '.join(row)))
            except (ValueError, KeyError):
                ui.tEMonitor.append(allFormat.format(f'[@MET] {m.group(1)[:80]}'))
            ui.tEMonitor.moveCursor(QTextCursor.End)
            return

        m = _RE_SUPPORT.match(line)
        if m:
            try:
                d = json.loads(m.group(1))
                summary = (
                    f'[@BOOT] env={d.get("env", "?")} '
                    f'sha={d.get("sha", "?")} '
                    f'role={d.get("role", "?")} '
                    f'radio={d.get("radio", "?")} '
                    f'freq={d.get("freq", "?")}'
                )
                ui.tEMonitor.append(serverFormat.format(summary))
            except (ValueError, KeyError):
                ui.tEMonitor.append(
                    serverFormat.format(f'[@SUPPORTPACK] {m.group(1)[:80]}')
                )
            ui.tEMonitor.moveCursor(QTextCursor.End)
            return

        m = _RE_CRASH.match(line)
        if m:
            ui.tEMonitor.append(discFormat.format(f'[@CRASH] {m.group(1)}'))
            ui.tEMonitor.moveCursor(QTextCursor.End)
            return

        for pat in (_RE_POLICY, _RE_RST):
            m = pat.match(line)
            if m:
                ui.tEMonitor.append(allFormat.format(f'{line[:100]}'))
                ui.tEMonitor.moveCursor(QTextCursor.End)
                return

        m = _RE_ESP_LOG.match(line)
        if m:
            level, ts, tag, msg = m.group(1), m.group(2), m.group(3), m.group(4)
            if level == 'E':
                fmt = discFormat
            elif level == 'W':
                fmt = warnFormat
            else:
                fmt = allFormat
            ui.tEMonitor.append(fmt.format(f'[{level}][{tag}] {msg}'))
            ui.tEMonitor.moveCursor(QTextCursor.End)
            return

        m = _RE_ERR.match(line)
        if m:
            ui.tEMonitor.append(discFormat.format(f'[ERR] {m.group(1)}'))
            ui.tEMonitor.moveCursor(QTextCursor.End)
            return

        m = _RE_WARN.match(line)
        if m:
            ui.tEMonitor.append(warnFormat.format(f'[WARN] {m.group(1)}'))
            ui.tEMonitor.moveCursor(QTextCursor.End)
            return

        m = _RE_OK.match(line)
        if m:
            ui.tEMonitor.append(serverFormat.format(f'[OK] {m.group(1)}'))
            ui.tEMonitor.moveCursor(QTextCursor.End)
            return

        # ── Fallback: show line as-is ─────────────────────────────────────
        ui.tEMonitor.append(allFormat.format(line[:200]))
        ui.tEMonitor.moveCursor(QTextCursor.End)