tui: keystroke right after a bracketed paste is inserted before the pasted text #53160

#!/usr/bin/env python3
"""Usage: python3 repro.py [same-chunk|separate-chunks]"""
import fcntl, os, pty, re, select, signal, struct, subprocess, sys, termios, time

mode = sys.argv[1] if len(sys.argv) > 1 else "same-chunk"
os.makedirs("/tmp/oc-repro", exist_ok=True)

master, slave = pty.openpty()
fcntl.ioctl(master, termios.TIOCSWINSZ, struct.pack("HHHH", 30, 100, 0, 0))
env = dict(os.environ, TERM="xterm-256color",
           XDG_DATA_HOME="/tmp/oc-repro-data", XDG_STATE_HOME="/tmp/oc-repro-state")
proc = subprocess.Popen(
    ["opencode", "--standalone", "--log-level", "error", "/tmp/oc-repro"],
    stdin=slave, stdout=slave, stderr=slave, env=env, cwd="/tmp/oc-repro",
    start_new_session=True)
os.close(slave)
out = bytearray()

def drain(seconds):
    end = time.time() + seconds
    while time.time() < end:
        if select.select([master], [], [], 0.1)[0]:
            try:
                chunk = os.read(master, 65536)
            except OSError:
                return
            if not chunk:
                return
            out.extend(chunk)

try:
    drain(12)  # wait for the TUI
    if mode == "same-chunk":
        os.write(master, b"\x1b[200~AAA\x1b[201~B")
    else:
        os.write(master, b"\x1b[200~AAA\x1b[201~")
        drain(0.8)  # let the paste settle
        os.write(master, b"B")
    drain(1.0)
    fcntl.ioctl(master, termios.TIOCSWINSZ, struct.pack("HHHH", 30, 101, 0, 0))
    drain(0.5)
    fcntl.ioctl(master, termios.TIOCSWINSZ, struct.pack("HHHH", 30, 100, 0, 0))
    drain(1.5)
finally:
    proc.send_signal(signal.SIGKILL)
    proc.wait()
    os.close(master)

def screen_rows(data, rows=30, cols=110):
    grid = [[" "] * cols for _ in range(rows)]
    i, row, col = 0, 1, 1
    while i < len(data):
        b = data[i]
        if b == 0x1B:
            m = re.match(rb"\x1b\[([0-9;?]*)([A-Za-z])", data[i:])
            if m:
                ps = [int(p) for p in m.group(1).replace(b"?", b"").split(b";") if p.isdigit()]
                f = m.group(2)
                if f in (b"H", b"f"):
                    row = ps[0] if len(ps) > 0 and ps[0] else 1
                    col = ps[1] if len(ps) > 1 and ps[1] else 1
                elif f == b"G":
                    col = ps[0] if ps else 1
                elif f == b"A":
                    row = max(1, row - (ps[0] if ps else 1))
                elif f == b"B":
                    row += ps[0] if ps else 1
                elif f == b"C":
                    col += ps[0] if ps else 1
                elif f == b"D":
                    col = max(1, col - (ps[0] if ps else 1))
                elif f == b"K":
                    m0 = ps[0] if ps else 0
                    a, z = (col - 1, cols) if m0 == 0 else (0, col) if m0 == 1 else (0, cols)
                    for c in range(a, z):
                        grid[row - 1][c] = " "
                i += m.end()
                continue
            i += 1
        elif b in (0x0A, 0x0D, 0x08):
            if b == 0x0A: row += 1
            elif b == 0x0D: col = 1
            else: col = max(1, col - 1)
            i += 1
        elif 0x20 <= b < 0x7F:
            if 1 <= row <= rows and 1 <= col <= cols:
                grid[row - 1][col - 1] = chr(b)
            col += 1
            i += 1
        else:
            i += 1
    return ["".join(r).rstrip() for r in grid]

print(f"mode={mode}")
for n, line in enumerate(screen_rows(bytes(out)), 1):
    if "AAA" in line or "BAA" in line:
        print(f"  {n:2}| {line}")