lux-hammer 0.3.16 → 0.3.20

data/recipes/lib/llm/wrap.rb

@@ -0,0 +1,691 @@
+# frozen_string_literal: true
+
+require 'fileutils'
+require 'io/console'
+require 'pty'
+
+# Run a command in a PTY with the last prompts you typed pinned to the bottom
+# rows of the screen, under a "prompt history" rule.
+#
+# The trick is to lie about the terminal size: the child gets a PTY that is
+# a few rows shorter than the real terminal, so it renders inside that and never
+# draws over the bar - even a full-screen TUI. A scroll region (DECSTBM) keeps
+# normal-buffer scrolling above the bar as well.
+#
+# A "prompt" is whatever you type between Enters. This is keystroke sniffing,
+# so text the program inserts for you (history recall, autocomplete, menu
+# picks) never shows up - that is the price of working with any program at all.
+#
+# Wrapping a program hides it: the child lives on our PTY, in its own session,
+# so anything outside looking at the terminal sees this process and not the
+# program. Terminals that watch what is running in a pane - Herdr naming a tab,
+# or deciding whether an agent is thinking, waiting, or idle - go blind, and the
+# fix is to stop hiding rather than to relay anything: we borrow the child's
+# name (Session#adopt_child_name) and keep the bar from imitating the child's
+# own screen furniture (KeyBuffer::RULE).
+module LlmWrap
+  # Prompts pinned, oldest first, so the newest sits on the very last line -
+  # closest to where you are typing. A labelled rule sits above them, so the
+  # bar is one row taller than the number of prompts.
+  DEFAULT_KEEP ||= 3
+  MAX_KEEP     ||= 20
+
+  # Poll gap used to notice that the child's output has settled, and the
+  # longest we let a stale bar sit while output keeps streaming.
+  IDLE_TICK ||= 0.02
+  MAX_DEFER ||= 0.15
+  CHUNK     ||= 65_536
+
+  # `origin` is how this wrapper was asked for, in the words that would ask for
+  # it again - only the caller knows that, since taking the child's name throws
+  # our own command line away. It is what Handoff writes down; the child's argv
+  # stands in when nobody says.
+  def self.run(argv, keep: DEFAULT_KEEP, origin: argv)
+    keep = keep.to_i.clamp(1, MAX_KEEP)
+    return passthrough(argv) unless $stdin.tty? && $stdout.tty?
+
+    rows, = $stdout.winsize
+    # Leave the child a usable screen, or skip the bar entirely.
+    return passthrough(argv) if rows.to_i < keep + 4
+
+    Session.new(argv, keep, origin).run
+  end
+
+  # No terminal to draw on (piped, scripted, inside another wrapper), so get
+  # out of the way completely rather than half-working.
+  def self.passthrough(argv)
+    exec(*argv)
+  rescue Errno::ENOENT
+    warn "llm wrap: command not found: #{argv.first}"
+    127
+  end
+
+  # Reads the child's line discipline through the PTY master. Used only to tell
+  # a password prompt (echo off, still canonical) apart from a raw-mode TUI
+  # (echo off, canonical off) - see Session#capture?.
+  module Termios
+    # struct termios: c_iflag, c_oflag, c_cflag, c_lflag, ... - we want c_lflag.
+    # Darwin/BSD fields are unsigned long (8b), Linux uses uint32 (4b), and the
+    # ICANON bit differs between them.
+    GETA, LFLAG_AT, LFLAG_PACK, ICANON, ECHO =
+      if RUBY_PLATFORM.match?(/darwin|bsd/)
+        [0x4048_7413, 24, 'Q', 0x100, 0x8] # TIOCGETA
+      else
+        [0x5401, 12, 'L', 0x2, 0x8]        # TCGETS
+      end
+
+    # [echo?, canonical?], or nil when the ioctl is not understood here.
+    def self.state(io)
+      buf = String.new("\0" * 128)
+      io.ioctl(GETA, buf)
+      lflag = buf.unpack1("@#{LFLAG_AT}#{LFLAG_PACK}")
+      [(lflag & ECHO) != 0, (lflag & ICANON) != 0]
+    rescue SystemCallError, IOError, ArgumentError
+      nil
+    end
+  end
+
+  # Taking the child's name costs us our own command line, and that is the only
+  # record of how the pane was started - Herdr's clone-tab reads it back off the
+  # foreground process to reopen the same wrapper. So leave it in a file named
+  # after our pid: whoever can see the process can find the command. One
+  # argument per line, and the file goes away when we do.
+  module Handoff
+    # Read at call time rather than frozen into a constant at load: this is a
+    # long-lived process, and the tests need somewhere else to write.
+    def self.dir
+      state = ENV['XDG_STATE_HOME'].to_s
+      state = File.join(Dir.home, '.local/state') if state.empty?
+      File.join(state, 'llm-wrap')
+    end
+
+    def self.path(pid = Process.pid)
+      File.join(dir, pid.to_s)
+    end
+
+    def self.write(argv)
+      FileUtils.mkdir_p(dir)
+      sweep
+      File.write(path, argv.join("\n"))
+    rescue SystemCallError, IOError
+      nil
+    end
+
+    def self.clear
+      File.unlink(path)
+    rescue SystemCallError
+      nil
+    end
+
+    # A crash leaves its file behind, and pids come round again - a stale entry
+    # would hand clone-tab someone else's command line. Drop the ones whose
+    # process is gone. Signal 0 only asks; EPERM means it is alive and not ours.
+    def self.sweep
+      Dir.children(dir).each do |name|
+        pid = Integer(name, exception: false) or next
+
+        begin
+          Process.kill(0, pid)
+        rescue Errno::ESRCH
+          File.unlink(File.join(dir, name))
+        rescue SystemCallError
+          nil
+        end
+      end
+    rescue SystemCallError, IOError
+      nil
+    end
+  end
+
+  class Session
+    def initialize(argv, keep, origin = argv)
+      @argv     = argv
+      @origin   = origin
+      @bar_rows = keep + 1     # the prompts, plus the rule above them
+      @keys     = KeyBuffer.new(keep)
+      @winch    = false
+      @dirty    = true
+      @drawn_at = 0.0
+      @started  = now
+      @trace    = (File.open(ENV['LLM_WRAP_DEBUG'], 'a') if ENV['LLM_WRAP_DEBUG'].to_s != '')
+    rescue SystemCallError
+      @trace = nil
+    end
+
+    def run
+      @rows, @cols = $stdout.winsize
+      $stdout.sync = true
+
+      @pty_out, @pty_in, @pid = PTY.spawn(*@argv)
+      resize_child
+      adopt_child_name
+      trace('RUN', "#{@argv.inspect} term=#{ENV['TERM'].inspect} #{@rows}x#{@cols} " \
+                   "child=#{child_rows}x#{@cols}")
+      Signal.trap('WINCH') { @winch = true }
+
+      begin
+        enter_screen
+        # raw leaves ISIG off, so Ctrl-C reaches the child as a plain 0x03 byte
+        # and its own line discipline signals it. We must not trap SIGINT here.
+        $stdin.raw { pump }
+      ensure
+        leave_screen
+        Handoff.clear
+      end
+
+      reap
+    rescue Errno::ENOENT
+      warn "llm wrap: command not found: #{@argv.first}"
+      127
+    end
+
+    private
+
+    # Answer to the child's name, so a pane running `llm wrap claude` reads from
+    # the outside as a pane running `claude`.
+    #
+    # Everything that watches a terminal pane identifies what is in it from the
+    # foreground process, and our child is not it: it has its own session on our
+    # PTY and is invisible from out there. Herdr in particular matches argv[0]
+    # against the agent it knows - see `herdr pane process-info` - and with
+    # `ruby` sitting in that slot the pane has no agent, so none of its
+    # detection runs and the tab loses the working spinner and the blocked
+    # marker. The name is all it needs; state it reads off the screen and the
+    # OSC title, both of which already come straight through us.
+    #
+    # This throws our own command line away, which is why Handoff wrote it down
+    # first.
+    def adopt_child_name
+      Handoff.write(@origin)
+      Process.setproctitle(File.basename(@argv.first.to_s))
+    rescue StandardError
+      nil
+    end
+
+    def child_rows
+      [@rows - @bar_rows, 1].max
+    end
+
+    def now
+      Process.clock_gettime(Process::CLOCK_MONOTONIC)
+    end
+
+    # -- screen ------------------------------------------------------------
+
+    def enter_screen
+      emit "\e[2J\e[H"
+      set_region
+      draw_bar
+    end
+
+    def leave_screen
+      emit "\e[r"                              # full screen back
+      emit "\e[#{child_rows + 1};1H\e[J"       # erase the bar, park the cursor
+      emit "\e[?25h"                           # the child may have hidden it
+    end
+
+    def set_region
+      emit "\e[1;#{child_rows}r"
+    end
+
+    def resize_child
+      @pty_out.winsize = [child_rows, @cols]
+    rescue SystemCallError
+      nil
+    end
+
+    # Painted in one write so the cursor never visibly detours through the bar.
+    #
+    # The region is reset before addressing the bar rows and re-asserted after:
+    # that keeps them reachable if the child turned on origin mode, and undoes
+    # any scroll region the child set for itself (a full-screen `ESC[r` from the
+    # child would otherwise let its scrolling eat the bar).
+    #
+    # DECSC/DECRC (ESC7/ESC8) is the only way to get the cursor back without
+    # emulating the child's screen. The terminal has a single save slot, so a
+    # child holding a saved cursor across this point would lose it - drawing
+    # only once output has settled makes that vanishingly rare in practice.
+    def draw_bar
+      out = +"\e7\e[r"
+      @keys.lines(@cols).each_with_index do |line, i|
+        out << "\e[#{child_rows + 1 + i};1H\e[K" << line
+      end
+      out << "\e[1;#{child_rows}r\e8"
+      emit out
+
+      @dirty    = false
+      @drawn_at = now
+    end
+
+    def emit(str)
+      $stdout.write(str)
+    end
+
+    def apply_winch
+      @winch = false
+      rows, cols = $stdout.winsize
+      return if rows.to_i.zero? || cols.to_i.zero?
+
+      @rows, @cols = rows, cols
+      resize_child   # the kernel SIGWINCHes the child off the back of this
+      set_region
+      draw_bar
+    end
+
+    # -- pump --------------------------------------------------------------
+
+    def pump
+      loop do
+        apply_winch if @winch
+
+        ready = IO.select([$stdin, @pty_out], nil, nil, IDLE_TICK)
+
+        unless ready
+          draw_bar if @dirty        # output settled
+          next
+        end
+
+        ready[0].each do |io|
+          if io.equal?($stdin)
+            data = slurp($stdin) or return
+            @pty_in.write(data)
+            taking = capture?
+            @dirty = true if taking && @keys.feed(data)
+            trace_in(data, taking)
+          else
+            data = slurp(@pty_out) or return
+            $stdout.write(data)
+            trace_out(data)
+            # The child cannot address the bar rows, but it can switch to the
+            # alt screen (which starts blank) or scroll a region it set itself.
+            # Repainting after every burst heals both.
+            @dirty = true
+          end
+        end
+
+        # Keep the bar honest while output streams without ever settling.
+        draw_bar if @dirty && now - @drawn_at > MAX_DEFER
+      end
+    rescue Errno::EIO, Errno::EPIPE, EOFError
+      nil
+    end
+
+    def slurp(io)
+      io.readpartial(CHUNK)
+    rescue EOFError, Errno::EIO, IOError
+      nil
+    end
+
+    # Terminals disagree wildly about how they report keys, and the encoding
+    # depends on modes the child turns on and off as it runs. LLM_WRAP_DEBUG=<file>
+    # records both sides raw so the two can be lined up. Off unless asked for -
+    # it necessarily writes down every keystroke.
+    #
+    #   IN  <t> <bytes>  take=.. echo=.. canon=..  buf=<line so far>
+    #   OUT <t> <mode sequences the child just set/reset>
+    def trace(kind, text)
+      return unless @trace
+
+      @trace.write(format("%9.3f %-3s %s\n", now - @started, kind, text))
+      @trace.flush
+    rescue IOError, SystemCallError
+      @trace = nil
+    end
+
+    def trace_in(data, taking)
+      return unless @trace
+
+      echo, canon = Termios.state(@pty_out)
+      trace('IN', "#{data.inspect} take=#{taking} echo=#{echo} canon=#{canon} " \
+                  "buf=#{@keys.pending.inspect}")
+    end
+
+    # Only the mode-setting sequences, not the whole output stream - the render
+    # traffic would bury the log, and it is the modes that decide how the
+    # terminal encodes keys in the first place.
+    def trace_out(data)
+      return unless @trace
+
+      seqs = data.scan(/\e\[\?[\d;]+[hl]|\e\[[<>][\d;]*[a-zA-Z]/)
+      trace('OUT', seqs.uniq.map(&:inspect).join(' ')) unless seqs.empty?
+    end
+
+    # Don't pin what the user cannot see themselves. Echo off while the child
+    # is still canonical is a password prompt (sudo, ssh); echo off with
+    # canonical also off is just a TUI in raw mode, which is the normal case.
+    def capture?
+      echo, canonical = Termios.state(@pty_out)
+      return true if echo.nil?
+
+      echo || !canonical
+    end
+
+    def reap
+      _, status = Process.waitpid2(@pid)
+      status.exitstatus || (128 + status.termsig.to_i)
+    rescue Errno::ECHILD
+      0
+    end
+  end
+
+  # Rebuilds the line you are typing from the raw byte stream on its way to the
+  # child, and keeps the last `keep` submitted lines.
+  #
+  # In raw mode keystrokes usually arrive one at a time, but never rely on it:
+  # reads can split a UTF-8 character or an escape sequence, so bytes are
+  # accumulated in a binary buffer and only decoded when rendered.
+  class KeyBuffer
+    MAX_LEN     ||= 4096
+    NO_PROMPTS  ||= '(nothing typed yet)'
+    LABEL       ||= 'prompt history'
+
+    CR    = 0x0d
+    LF    = 0x0a
+    ESC   = 0x1b
+    TAB   = 0x09
+    BS    = 0x08
+    BEL   = 0x07
+    DEL   = 0x7f
+
+    # Introducers of the escape sequences that carry a string payload:
+    # OSC ], DCS P, SOS X, PM ^, APC _.
+    STRING_INTRO ||= [0x5d, 0x50, 0x58, 0x5e, 0x5f].freeze
+    CTRL_C = 0x03
+    CTRL_U = 0x15
+    CTRL_W = 0x17
+
+    # Modifier bits, as reported by both keyboard protocols (1-based, so the
+    # wire value is this + 1).
+    SHIFT ||= 1
+    ALT   ||= 2
+    CTRL  ||= 4
+    SUPER ||= 8
+
+    # CSI <code>[:<alternates>][;<mods>[:<event>]][;<text codepoints>] u
+    KITTY_KEY ||= /\A\e\[(\d+)(?::\d+(?::\d+)?)?(?:;(\d+)(?::(\d+))?)?(?:;([\d:]+))?u\z/
+    # CSI 27 ; <mods> ; <code> ~   (xterm modifyOtherKeys)
+    XTERM_KEY ||= /\A\e\[27;(\d+);(\d+)~\z/
+
+    # Roughly the double-width ranges of UEAW - enough to keep CJK and emoji
+    # from overflowing the bar without pulling in a character-width gem.
+    WIDE ||= [
+      0x1100..0x115f, 0x2e80..0x303e, 0x3041..0x33ff, 0x3400..0x4dbf,
+      0x4e00..0x9fff, 0xa000..0xa4cf, 0xac00..0xd7a3, 0xf900..0xfaff,
+      0xfe30..0xfe6f, 0xff00..0xff60, 0xffe0..0xffe6,
+      0x1f300..0x1f64f, 0x1f900..0x1f9ff, 0x20000..0x2fffd
+    ].freeze
+    ZERO ||= [0xfe00..0xfe0f, 0x200b..0x200f].freeze
+
+    def initialize(keep)
+      @keep    = keep
+      @prompts = []
+      @buf     = binary
+      @esc     = nil
+      @paste   = false
+    end
+
+    # The line being typed right now, for the debug trace.
+    def pending
+      decode(@buf)
+    end
+
+    # Returns true when the pinned prompts changed.
+    def feed(bytes)
+      changed = false
+      bytes.each_byte { |b| changed = true if @esc ? escape(b) : plain(b) }
+      changed
+    end
+
+    # The whole bar, top to bottom: the labelled rule, then the pinned prompts
+    # oldest first so the newest lands on the last line.
+    def lines(cols)
+      rows = Array.new(@keep) do |slot|
+        age  = @keep - 1 - slot     # 0 is the newest, and it renders last
+        text = @prompts[age]
+
+        if text.nil?
+          age.zero? ? "#{DIM}#{clip(NO_PROMPTS, cols)}#{RESET}" : ''
+        elsif age.zero?
+          "#{CYAN}❯#{RESET} #{clip(text, cols - 2)}"
+        else
+          "#{DIM}❯ #{clip(text, cols - 2)}#{RESET}"
+        end
+      end
+
+      [rule(cols)] + rows
+    end
+
+    private
+
+    CYAN  = "\e[36m"
+    DIM   = "\e[2m"
+    GRAY  = "\e[38;5;245m"
+    RESET = "\e[0m"
+
+    # Dashed, and deliberately not the solid U+2500 "─" it wants to be.
+    #
+    # A pane watcher works out where a program's own furniture is by looking for
+    # the last solid rule on the screen: Herdr hangs its "the prompt box is
+    # here" and "the permission dialog is here" regions off it. Our bar is below
+    # everything, so a solid rule down here becomes the last one and those
+    # regions land on the prompt history instead - detection then sees no prompt
+    # and no dialog, and the pane reads as neither idle nor blocked. Any glyph
+    # that is not box-drawing keeps the bar out of that reckoning; a dashed rule
+    # is also a fair signal that this row is not part of the program above.
+    RULE ||= '┄'
+
+    # A full-width rule with the label centred in it, marking off the bar from
+    # whatever the wrapped program is drawing above it.
+    def rule(cols)
+      return '' if cols <= 0
+
+      label = " #{LABEL} "
+      pad   = cols - width(label)
+      return "#{GRAY}#{RULE * cols}#{RESET}" if pad < 2
+
+      left = pad / 2
+      "#{GRAY}#{RULE * left}#{label}#{RULE * (pad - left)}#{RESET}"
+    end
+
+    def binary
+      String.new(encoding: Encoding::BINARY)
+    end
+
+    def plain(byte)
+      case byte
+      when ESC          then @esc = binary << byte; false
+      when CR, LF       then @paste ? (append("\n"); false) : submit
+      when DEL, BS      then chop_char; false
+      when CTRL_U, CTRL_C then @buf = binary; false
+      when CTRL_W       then drop_word; false
+      when TAB          then false   # completion text is inserted by the app
+      else
+        append_byte(byte) if byte >= 0x20
+        false
+      end
+    end
+
+    # Collect an escape sequence, then decide. Everything is discarded except
+    # the handful of forms that mean "newline, but do not submit" and the
+    # bracketed-paste markers.
+    def escape(byte)
+      @esc << byte
+
+      if @esc.bytesize == 2
+        case byte
+        when 0x5b, 0x4f    then return false       # CSI / SS3: keep collecting
+        when *STRING_INTRO then return false       # OSC / DCS / APC / PM / SOS
+        when CR, LF        then @esc = nil; append("\n"); return false # Option+Enter
+        else @esc = nil; return false
+        end
+      end
+
+      intro = @esc.getbyte(1)
+
+      # String sequences run to BEL or ST (ESC \) rather than a final byte, and
+      # carry a payload of printable text. They are the terminal answering a
+      # question the program asked - Codex queries the fg/bg colour on startup
+      # and gets back "\e]10;rgb:cdcd/d6d6/f4f4\e\\" - so the whole thing is
+      # dropped. Treating them as CSI would spill that payload into the prompt.
+      if STRING_INTRO.include?(intro)
+        @esc = nil if byte == BEL || (byte == 0x5c && @esc.getbyte(-2) == ESC)
+        @esc = nil if @esc && @esc.bytesize > 1024  # unterminated: cut losses
+        return false
+      end
+
+      if intro == 0x4f                             # SS3 is always three bytes
+        @esc = nil
+        return false
+      end
+
+      if byte >= 0x40 && byte <= 0x7e              # CSI final byte
+        seq = @esc
+        @esc = nil
+        return csi(seq)
+      end
+
+      @esc = nil if @esc.bytesize > 32             # runaway guard
+      false
+    end
+
+    # Keys do not always arrive as plain bytes. Claude Code turns on both the
+    # kitty keyboard protocol (CSI > 1 u) and xterm's modifyOtherKeys level 2
+    # (CSI > 4 ; 2 m), under which the terminal reports keys as escape
+    # sequences instead - so "hello" can arrive as five CSI sequences and a
+    # parser that only reads raw bytes sees nothing at all.
+    #
+    # Everything else (arrows, function keys, focus in/out, and the SGR mouse
+    # reports that Claude's motion tracking produces by the hundred) is
+    # discarded: it is not text the user typed.
+    def csi(seq)
+      case seq
+      when "\e[200~" then @paste = true
+      when "\e[201~" then @paste = false
+      when KITTY_KEY                        # CSI code[:alt][;mods[:event]][;text] u
+        m = Regexp.last_match
+        return key(m[1].to_i, m[2].to_i, event: m[3].to_i, text: m[4])
+      when XTERM_KEY                        # CSI 27 ; mods ; code ~
+        m = Regexp.last_match
+        return key(m[2].to_i, m[1].to_i)
+      end
+      false
+    end
+
+    # One decoded keypress. `mods` is the usual 1-based bitfield (1 = none,
+    # +1 shift, +2 alt, +4 ctrl, +8 super); caps/num lock are masked off so a
+    # stuck caps lock does not turn every Enter into a continuation.
+    def key(code, mods, event: 0, text: nil)
+      return false if event == 3                     # key release, not a press
+
+      bits = [mods - 1, 0].max & (SHIFT | ALT | CTRL | SUPER)
+
+      case code
+      when 13, 10 then return bits.zero? ? submit : (append("\n"); false)
+      when 127, 8 then chop_char; return false
+      when 9, 27  then return false                  # Tab, Esc
+      end
+
+      # Ctrl-<key> is a command, never text. Ctrl-V in particular pastes an
+      # image in Claude Code, which puts no typed text on the wire at all.
+      if bits & CTRL != 0
+        case code
+        when 117, 99 then @buf = binary              # ctrl-u, ctrl-c
+        when 119     then drop_word                  # ctrl-w
+        end
+        return false
+      end
+      return false if bits & (ALT | SUPER) != 0
+      return false if code < 0x20
+
+      # The kitty "associated text" field is the literal text of the keypress
+      # and wins when present. Without it, shift lives in the modifiers while
+      # `code` stays the unshifted key, so apply it by hand.
+      if text
+        text.split(':').each { |cp| append(cp.to_i.chr(Encoding::UTF_8)) }
+      else
+        ch = code.chr(Encoding::UTF_8)
+        ch = ch.upcase if bits & SHIFT != 0
+        append(ch)
+      end
+      false
+    rescue RangeError
+      false
+    end
+
+    def append_byte(byte)
+      @buf << byte if @buf.bytesize < MAX_LEN
+    end
+
+    def append(str)
+      @buf << str.b if @buf.bytesize < MAX_LEN
+    end
+
+    # Drop one whole character: back over any UTF-8 continuation bytes first.
+    def chop_char
+      return if @buf.empty?
+
+      i = @buf.bytesize - 1
+      i -= 1 while i.positive? && (@buf.getbyte(i) & 0xc0) == 0x80
+      @buf.slice!(i..)
+    end
+
+    # Readline's unix-word-rubout: eat trailing whitespace, then the word, and
+    # leave the separator before it alone ("one two" -> "one ").
+    def drop_word
+      @buf.sub!(/\s+\z/, '')
+      @buf.sub!(/\S+\z/, '')
+    end
+
+    def submit
+      text = decode(@buf)
+      @buf = binary
+
+      return false if text.empty? || @prompts.first == text
+
+      @prompts.unshift(text)
+      @prompts.pop while @prompts.size > @keep
+      true
+    end
+
+    # A bar row is one line, so a multi-line prompt - a continuation with
+    # Shift/Option+Enter, or a pasted block - is flattened onto it: line breaks
+    # show as a backslash + space, and clip() caps the result to the terminal
+    # width. Nothing is dropped that would still fit.
+    def decode(bytes)
+      text = bytes.dup.force_encoding(Encoding::UTF_8).scrub('').strip
+      return '' if text.empty?
+
+      text.gsub(/[^\S\n]+/, ' ')        # runs of spaces/tabs, newlines kept
+          .gsub(/ ?\n+ ?/) { '\ ' }     # block form: no backslash escaping here
+    end
+
+    def clip(text, cells)
+      return '' if cells <= 0
+      return text if width(text) <= cells
+
+      out  = +''
+      used = 0
+      text.each_char do |ch|
+        w = char_width(ch)
+        break if used + w > cells - 1
+
+        out << ch
+        used += w
+      end
+      out << '…'
+    end
+
+    def width(text)
+      text.each_char.sum { |ch| char_width(ch) }
+    end
+
+    def char_width(char)
+      cp = char.ord
+      return 0 if cp == 0x200d || ZERO.any? { |r| r.cover?(cp) }
+
+      WIDE.any? { |r| r.cover?(cp) } ? 2 : 1
+    rescue RangeError
+      1
+    end
+  end
+end