ratatui_ruby 0.5.0 → 0.7.0

data/ext/ratatui_ruby/src/widgets/table.rs

@@ -2,8 +2,10 @@
 // SPDX-License-Identifier: AGPL-3.0-or-later
 
 use crate::style::{parse_block, parse_style};
+use crate::text::{parse_line, parse_span};
+use crate::widgets::table_state::RubyTableState;
 use bumpalo::Bump;
-use magnus::{prelude::*, Error, Symbol, Value};
+use magnus::{prelude::*, Error, Symbol, TryConvert, Value};
 use ratatui::{
     layout::{Constraint, Flex, Rect},
     widgets::{Cell, HighlightSpacing, Row, Table, TableState},
@@ -21,7 +23,7 @@ pub fn render(frame: &mut Frame, area: Rect, node: Value) -> Result<(), Error> {
     let widths_val: Value = node.funcall("widths", ())?;
     let widths_array = magnus::RArray::from_value(widths_val)
         .ok_or_else(|| Error::new(ruby.exception_type_error(), "expected array for widths"))?;
-    let highlight_style_val: Value = node.funcall("highlight_style", ())?;
+    let row_highlight_style_val: Value = node.funcall("row_highlight_style", ())?;
     let column_highlight_style_val: Value = node.funcall("column_highlight_style", ())?;
     let cell_highlight_style_val: Value = node.funcall("cell_highlight_style", ())?;
     let highlight_symbol_val: Value = node.funcall("highlight_symbol", ())?;
@@ -72,8 +74,8 @@ pub fn render(frame: &mut Frame, area: Rect, node: Value) -> Result<(), Error> {
         table = table.block(parse_block(block_val, &bump)?);
     }
 
-    if !highlight_style_val.is_nil() {
-        table = table.row_highlight_style(parse_style(highlight_style_val)?);
+    if !row_highlight_style_val.is_nil() {
+        table = table.row_highlight_style(parse_style(row_highlight_style_val)?);
     }
 
     if !column_highlight_style_val.is_nil() {
@@ -110,12 +112,170 @@ pub fn render(frame: &mut Frame, area: Rect, node: Value) -> Result<(), Error> {
         state.select_column(Some(index));
     }
 
+    let offset_val: Value = node.funcall("offset", ())?;
+    if !offset_val.is_nil() {
+        let offset: usize = offset_val.funcall("to_int", ())?;
+        *state.offset_mut() = offset;
+    }
+
     frame.render_stateful_widget(table, area, &mut state);
     Ok(())
 }
 
+/// Renders a Table with an external state object.
+///
+/// This function ignores `selected_row`, `selected_column`, and `offset` from the widget.
+/// The State object is the single source of truth for selection and scroll position.
+pub fn render_stateful(
+    frame: &mut Frame,
+    area: Rect,
+    node: Value,
+    state_wrapper: Value,
+) -> Result<(), Error> {
+    let bump = Bump::new();
+    let ruby = magnus::Ruby::get().unwrap();
+
+    // Extract the RubyTableState wrapper
+    let state: &RubyTableState = TryConvert::try_convert(state_wrapper)?;
+
+    // Parse rows
+    let rows_value: Value = node.funcall("rows", ())?;
+    let rows_array = magnus::RArray::from_value(rows_value)
+        .ok_or_else(|| Error::new(ruby.exception_type_error(), "expected array for rows"))?;
+    let widths_val: Value = node.funcall("widths", ())?;
+    let widths_array = magnus::RArray::from_value(widths_val)
+        .ok_or_else(|| Error::new(ruby.exception_type_error(), "expected array for widths"))?;
+
+    let mut rows = Vec::new();
+    for i in 0..rows_array.len() {
+        let index = isize::try_from(i)
+            .map_err(|e| Error::new(ruby.exception_range_error(), e.to_string()))?;
+        let row_val: Value = rows_array.entry(index)?;
+        rows.push(parse_row(row_val)?);
+    }
+
+    let constraints = parse_constraints(widths_array)?;
+
+    // Build table (ignoring selected_row, selected_column, offset — State is truth)
+    let header_val: Value = node.funcall("header", ())?;
+    let footer_val: Value = node.funcall("footer", ())?;
+    let row_highlight_style_val: Value = node.funcall("row_highlight_style", ())?;
+    let column_highlight_style_val: Value = node.funcall("column_highlight_style", ())?;
+    let cell_highlight_style_val: Value = node.funcall("cell_highlight_style", ())?;
+    let highlight_symbol_val: Value = node.funcall("highlight_symbol", ())?;
+    let block_val: Value = node.funcall("block", ())?;
+    let flex_sym: Symbol = node.funcall("flex", ())?;
+    let highlight_spacing_sym: Symbol = node.funcall("highlight_spacing", ())?;
+    let style_val: Value = node.funcall("style", ())?;
+    let column_spacing_val: Value = node.funcall("column_spacing", ())?;
+
+    let flex = match flex_sym.to_string().as_str() {
+        "start" => Flex::Start,
+        "center" => Flex::Center,
+        "end" => Flex::End,
+        "space_between" => Flex::SpaceBetween,
+        "space_around" => Flex::SpaceAround,
+        "space_evenly" => Flex::SpaceEvenly,
+        _ => Flex::Legacy,
+    };
+
+    let mut table = Table::new(rows, constraints).flex(flex);
+
+    let highlight_spacing = match highlight_spacing_sym.to_string().as_str() {
+        "always" => HighlightSpacing::Always,
+        "never" => HighlightSpacing::Never,
+        _ => HighlightSpacing::WhenSelected,
+    };
+    table = table.highlight_spacing(highlight_spacing);
+
+    if !header_val.is_nil() {
+        table = table.header(parse_row(header_val)?);
+    }
+    if !footer_val.is_nil() {
+        table = table.footer(parse_row(footer_val)?);
+    }
+    if !block_val.is_nil() {
+        table = table.block(parse_block(block_val, &bump)?);
+    }
+    if !row_highlight_style_val.is_nil() {
+        table = table.row_highlight_style(parse_style(row_highlight_style_val)?);
+    }
+    if !column_highlight_style_val.is_nil() {
+        table = table.column_highlight_style(parse_style(column_highlight_style_val)?);
+    }
+    if !cell_highlight_style_val.is_nil() {
+        table = table.cell_highlight_style(parse_style(cell_highlight_style_val)?);
+    }
+    if !highlight_symbol_val.is_nil() {
+        let symbol: String = highlight_symbol_val.funcall("to_s", ())?;
+        table = table.highlight_symbol(symbol);
+    }
+    if !style_val.is_nil() {
+        table = table.style(parse_style(style_val)?);
+    }
+    if !column_spacing_val.is_nil() {
+        let spacing: u16 = column_spacing_val.funcall("to_int", ())?;
+        table = table.column_spacing(spacing);
+    }
+
+    // Borrow the inner TableState, render, and release the borrow immediately
+    {
+        let mut inner_state = state.borrow_mut();
+        frame.render_stateful_widget(table, area, &mut inner_state);
+    }
+
+    Ok(())
+}
+
 fn parse_row(row_val: Value) -> Result<Row<'static>, Error> {
     let ruby = magnus::Ruby::get().unwrap();
+
+    // Check if this is a RatatuiRuby::Row object with cells + style + height + margins
+    let class = row_val.class();
+    // SAFETY: Immediate conversion to owned string avoids GC-unsafe borrowed reference.
+    let class_name = unsafe { class.name() }.into_owned();
+
+    if class_name == "RatatuiRuby::Widgets::Row" {
+        let cells_val: Value = row_val.funcall("cells", ())?;
+        let style_val: Value = row_val.funcall("style", ())?;
+        let height_val: Value = row_val.funcall("height", ())?;
+        let top_margin_val: Value = row_val.funcall("top_margin", ())?;
+        let bottom_margin_val: Value = row_val.funcall("bottom_margin", ())?;
+
+        let cells_array = magnus::RArray::from_value(cells_val).ok_or_else(|| {
+            Error::new(ruby.exception_type_error(), "expected array for Row.cells")
+        })?;
+
+        let mut cells = Vec::new();
+        for i in 0..cells_array.len() {
+            let index = isize::try_from(i)
+                .map_err(|e| Error::new(ruby.exception_range_error(), e.to_string()))?;
+            let entry_val: Value = cells_array.entry(index)?;
+            cells.push(parse_cell(entry_val)?);
+        }
+
+        let mut row = Row::new(cells);
+
+        if !style_val.is_nil() {
+            row = row.style(parse_style(style_val)?);
+        }
+        if !height_val.is_nil() {
+            let h: u16 = height_val.funcall("to_int", ())?;
+            row = row.height(h);
+        }
+        if !top_margin_val.is_nil() {
+            let m: u16 = top_margin_val.funcall("to_int", ())?;
+            row = row.top_margin(m);
+        }
+        if !bottom_margin_val.is_nil() {
+            let m: u16 = bottom_margin_val.funcall("to_int", ())?;
+            row = row.bottom_margin(m);
+        }
+
+        return Ok(row);
+    }
+
+    // Fallback: plain array of cells
     let row_array = magnus::RArray::from_value(row_val)
         .ok_or_else(|| Error::new(ruby.exception_type_error(), "expected array for row"))?;
 
@@ -134,42 +294,39 @@ fn parse_cell(cell_val: Value) -> Result<Cell<'static>, Error> {
     // SAFETY: Immediate conversion to owned string avoids GC-unsafe borrowed reference.
     let class_name = unsafe { class.name() }.into_owned();
 
-    if class_name == "RatatuiRuby::Paragraph" {
+    // Try Text::Line first (contains multiple spans)
+    if class_name.contains("Line") {
+        if let Ok(line) = parse_line(cell_val) {
+            return Ok(Cell::from(line));
+        }
+    }
+
+    // Try Text::Span
+    if class_name.contains("Span") {
+        if let Ok(span) = parse_span(cell_val) {
+            return Ok(Cell::from(ratatui::text::Line::from(vec![span])));
+        }
+    }
+
+    if class_name == "RatatuiRuby::Widgets::Paragraph" {
         let text: String = cell_val.funcall("text", ())?;
         let style_val: Value = cell_val.funcall("style", ())?;
         let cell_style = parse_style(style_val)?;
         Ok(Cell::from(text).style(cell_style))
-    } else if class_name == "RatatuiRuby::Style" {
+    } else if class_name == "RatatuiRuby::Style::Style" {
         Ok(Cell::from("").style(parse_style(cell_val)?))
-    } else if class_name == "RatatuiRuby::Cell" {
-        let symbol: String = cell_val.funcall("char", ())?;
-        let fg_val: Value = cell_val.funcall("fg", ())?;
-        let bg_val: Value = cell_val.funcall("bg", ())?;
-        let modifiers_val: Value = cell_val.funcall("modifiers", ())?;
-
-        let mut style = ratatui::style::Style::default();
-        if !fg_val.is_nil() {
-            if let Some(color) = crate::style::parse_color_value(fg_val)? {
-                style = style.fg(color);
-            }
-        }
-        if !bg_val.is_nil() {
-            if let Some(color) = crate::style::parse_color_value(bg_val)? {
-                style = style.bg(color);
-            }
-        }
-        if let Some(mods_array) = magnus::RArray::from_value(modifiers_val) {
-            let ruby = magnus::Ruby::get().unwrap();
-            for i in 0..mods_array.len() {
-                let index = isize::try_from(i)
-                    .map_err(|e| Error::new(ruby.exception_range_error(), e.to_string()))?;
-                let mod_str: String = mods_array.entry::<String>(index)?;
-                if let Some(modifier) = crate::style::parse_modifier_str(&mod_str) {
-                    style = style.add_modifier(modifier);
-                }
-            }
+    } else if class_name == "RatatuiRuby::Widgets::Cell" {
+        // Widgets::Cell has content (String/Span/Line) and optional style
+        let content_val: Value = cell_val.funcall("content", ())?;
+        let style_val: Value = cell_val.funcall("style", ())?;
+
+        // Recursively parse the content (could be String, Span, or Line)
+        let mut cell = parse_cell(content_val)?;
+
+        if !style_val.is_nil() {
+            cell = cell.style(parse_style(style_val)?);
         }
-        Ok(Cell::from(symbol).style(style))
+        Ok(cell)
     } else {
         let cell_str: String = cell_val.funcall("to_s", ())?;
         Ok(Cell::from(cell_str))

data/ext/ratatui_ruby/src/widgets/table_state.rs

@@ -0,0 +1,121 @@
+// SPDX-FileCopyrightText: 2025 Kerrick Long <me@kerricklong.com>
+// SPDX-License-Identifier: AGPL-3.0-or-later
+
+//! `TableState` wrapper for exposing Ratatui's `TableState` to Ruby.
+//!
+//! This module provides `RubyTableState`, a Magnus-wrapped struct that holds
+//! a `RefCell<TableState>` for interior mutability during stateful rendering.
+//!
+//! # Design
+//!
+//! When using `render_stateful_widget`, the State object is the single source
+//! of truth for selection and offset. Widget properties (`selected_row`,
+//! `selected_column`, `offset`) are ignored in stateful mode.
+
+use magnus::{function, method, prelude::*, Error, Module, Ruby};
+use ratatui::widgets::TableState;
+use std::cell::RefCell;
+
+/// A wrapper around Ratatui's `TableState` exposed to Ruby.
+#[magnus::wrap(class = "RatatuiRuby::TableState")]
+pub struct RubyTableState {
+    inner: RefCell<TableState>,
+}
+
+impl RubyTableState {
+    /// Creates a new `RubyTableState` with optional initial selection.
+    pub fn new(selected: Option<usize>) -> Self {
+        let mut state = TableState::default();
+        if let Some(idx) = selected {
+            state.select(Some(idx));
+        }
+        Self {
+            inner: RefCell::new(state),
+        }
+    }
+
+    /// Sets the selected row index.
+    pub fn select(&self, index: Option<usize>) {
+        self.inner.borrow_mut().select(index);
+    }
+
+    /// Returns the currently selected row index.
+    pub fn selected(&self) -> Option<usize> {
+        self.inner.borrow().selected()
+    }
+
+    /// Sets the selected column index.
+    pub fn select_column(&self, index: Option<usize>) {
+        self.inner.borrow_mut().select_column(index);
+    }
+
+    /// Returns the currently selected column index.
+    pub fn selected_column(&self) -> Option<usize> {
+        self.inner.borrow().selected_column()
+    }
+
+    /// Returns the current scroll offset.
+    pub fn offset(&self) -> usize {
+        self.inner.borrow().offset()
+    }
+
+    /// Scrolls down by the given number of rows.
+    pub fn scroll_down_by(&self, amount: u16) {
+        self.inner.borrow_mut().scroll_down_by(amount);
+    }
+
+    /// Scrolls up by the given number of rows.
+    pub fn scroll_up_by(&self, amount: u16) {
+        self.inner.borrow_mut().scroll_up_by(amount);
+    }
+
+    /// Borrows the inner `TableState` mutably for rendering.
+    pub fn borrow_mut(&self) -> std::cell::RefMut<'_, TableState> {
+        self.inner.borrow_mut()
+    }
+}
+
+/// Registers the `TableState` class with Ruby.
+pub fn register(ruby: &Ruby, module: magnus::RModule) -> Result<(), Error> {
+    let class = module.define_class("TableState", ruby.class_object())?;
+    class.define_singleton_method("new", function!(RubyTableState::new, 1))?;
+    class.define_method("select", method!(RubyTableState::select, 1))?;
+    class.define_method("selected", method!(RubyTableState::selected, 0))?;
+    class.define_method("select_column", method!(RubyTableState::select_column, 1))?;
+    class.define_method(
+        "selected_column",
+        method!(RubyTableState::selected_column, 0),
+    )?;
+    class.define_method("offset", method!(RubyTableState::offset, 0))?;
+    class.define_method("scroll_down_by", method!(RubyTableState::scroll_down_by, 1))?;
+    class.define_method("scroll_up_by", method!(RubyTableState::scroll_up_by, 1))?;
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_new_with_no_selection() {
+        let state = RubyTableState::new(None);
+        assert_eq!(state.selected(), None);
+        assert_eq!(state.selected_column(), None);
+        assert_eq!(state.offset(), 0);
+    }
+
+    #[test]
+    fn test_new_with_selection() {
+        let state = RubyTableState::new(Some(3));
+        assert_eq!(state.selected(), Some(3));
+    }
+
+    #[test]
+    fn test_column_selection() {
+        let state = RubyTableState::new(None);
+        state.select_column(Some(2));
+        assert_eq!(state.selected_column(), Some(2));
+        state.select_column(None);
+        assert_eq!(state.selected_column(), None);
+    }
+}

data/lib/ratatui_ruby/buffer/cell.rb

@@ -0,0 +1,168 @@
+# frozen_string_literal: true
+
+# SPDX-FileCopyrightText: 2025 Kerrick Long <me@kerricklong.com>
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+module RatatuiRuby
+  module Buffer
+    # Represents a single cell in the terminal buffer.
+    #
+    # A terminal grid is made of cells. Each cell contains a character (symbol) and styling (colors, modifiers).
+    # When testing, you often need to verify that a specific cell renders correctly.
+    #
+    # This object encapsulates that state. It provides predicate methods for modifiers, making assertions readable.
+    #
+    # Use it to inspect the visual state of your application in tests.
+    #
+    # === Examples
+    #
+    #   cell = RatatuiRuby.get_cell_at(0, 0)
+    #   cell.char   # => "H"
+    #   cell.fg     # => :red
+    #   cell.bold?  # => true
+    #
+    class Cell
+      # The character displayed in the cell.
+      #
+      # Named to match Ratatui's Cell::symbol() method.
+      attr_reader :symbol
+
+      # Alias for Rubyists who prefer a shorter name.
+      alias char symbol
+
+      # The foreground color of the cell (e.g., :red, :blue, "#ff0000").
+      attr_reader :fg
+
+      # The background color of the cell (e.g., :black, nil).
+      attr_reader :bg
+
+      # The list of active modifiers (e.g., ["bold", "italic"]).
+      attr_reader :modifiers
+
+      # Returns an empty cell (space character, no styles).
+      #
+      # === Example
+      #
+      #   Buffer::Cell.empty # => #<RatatuiRuby::Buffer::Cell char=" ">
+      #
+      def self.empty
+        new(symbol: " ", fg: nil, bg: nil, modifiers: [])
+      end
+
+      # Returns a default cell (alias for empty).
+      #
+      # === Example
+      #
+      #   Buffer::Cell.default # => #<RatatuiRuby::Buffer::Cell char=" ">
+      #
+      def self.default
+        empty
+      end
+
+      # Returns a cell with a specific character and no styles.
+      #
+      # [symbol] String (single character).
+      #
+      # === Example
+      #
+      #   Buffer::Cell.symbol("X") # => #<RatatuiRuby::Buffer::Cell symbol="X">
+      #
+      def self.symbol(symbol)
+        new(symbol:, fg: nil, bg: nil, modifiers: [])
+      end
+
+      # Alias for Rubyists who prefer a shorter name.
+      def self.char(char)
+        symbol(char)
+      end
+
+      # Creates a new Cell.
+      #
+      # [symbol] String (single character). Aliased as <tt>char:</tt>.
+      # [fg] Symbol or String (nullable).
+      # [bg] Symbol or String (nullable).
+      # [modifiers] Array of Strings.
+      def initialize(symbol: nil, char: nil, fg: nil, bg: nil, modifiers: [])
+        @symbol = (symbol || char || " ").freeze
+        @fg = fg&.freeze
+        @bg = bg&.freeze
+        @modifiers = modifiers.map(&:freeze).freeze
+        freeze
+      end
+
+      # Returns true if the cell has the bold modifier.
+      def bold?
+        modifiers.include?("bold")
+      end
+
+      # Returns true if the cell has the dim modifier.
+      def dim?
+        modifiers.include?("dim")
+      end
+
+      # Returns true if the cell has the italic modifier.
+      def italic?
+        modifiers.include?("italic")
+      end
+
+      # Returns true if the cell has the underlined modifier.
+      def underlined?
+        modifiers.include?("underlined")
+      end
+
+      # Returns true if the cell has the slow_blink modifier.
+      def slow_blink?
+        modifiers.include?("slow_blink")
+      end
+
+      # Returns true if the cell has the rapid_blink modifier.
+      def rapid_blink?
+        modifiers.include?("rapid_blink")
+      end
+
+      # Returns true if the cell has the reversed modifier.
+      def reversed?
+        modifiers.include?("reversed")
+      end
+
+      # Returns true if the cell has the hidden modifier.
+      def hidden?
+        modifiers.include?("hidden")
+      end
+
+      # Returns true if the cell has the crossed_out modifier.
+      def crossed_out?
+        modifiers.include?("crossed_out")
+      end
+
+      # Checks equality with another Cell.
+      def ==(other)
+        other.is_a?(Cell) &&
+          char == other.char &&
+          fg == other.fg &&
+          bg == other.bg &&
+          modifiers == other.modifiers
+      end
+
+      # Returns a string representation of the cell.
+      def inspect
+        parts = ["symbol=#{symbol.inspect}"]
+        parts << "fg=#{fg.inspect}" if fg
+        parts << "bg=#{bg.inspect}" if bg
+        parts << "modifiers=#{modifiers.inspect}" unless modifiers.empty?
+        "#<#{self.class} #{parts.join(' ')}>"
+      end
+
+      # Returns the cell's character.
+      def to_s
+        symbol
+      end
+
+      # Support for pattern matching.
+      # Supports both <tt>:symbol</tt> and <tt>:char</tt> keys.
+      def deconstruct_keys(keys)
+        { symbol:, char: symbol, fg:, bg:, modifiers: }
+      end
+    end
+  end
+end

data/lib/ratatui_ruby/buffer.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+# SPDX-FileCopyrightText: 2025 Kerrick Long <me@kerricklong.com>
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+module RatatuiRuby
+  # Buffer primitives for terminal cell inspection.
+  #
+  # This module mirrors +ratatui::buffer+ and contains:
+  # - {Cell} — Single terminal cell (for inspection)
+  module Buffer
+  end
+end
+
+require_relative "buffer/cell"

data/lib/ratatui_ruby/cell.rb

@@ -82,10 +82,10 @@ module RatatuiRuby
     # [bg] Symbol or String (nullable).
     # [modifiers] Array of Strings.
     def initialize(symbol: nil, char: nil, fg: nil, bg: nil, modifiers: [])
-      @symbol = symbol || char || " "
-      @fg = fg
-      @bg = bg
-      @modifiers = modifiers.freeze
+      @symbol = (symbol || char || " ").freeze
+      @fg = fg&.freeze
+      @bg = bg&.freeze
+      @modifiers = modifiers.map(&:freeze).freeze
       freeze
     end
 

data/lib/ratatui_ruby/event/key/character.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+# SPDX-FileCopyrightText: 2025 Kerrick Long <me@kerricklong.com>
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+module RatatuiRuby
+  class Event
+    class Key < Event
+      # Methods for handling printable characters.
+      module Character
+        # Returns true if the key represents a single printable character.
+        #
+        #   RatatuiRuby::Event::Key.new(code: "a").text?       # => true
+        #   RatatuiRuby::Event::Key.new(code: "enter").text?   # => false
+        #   RatatuiRuby::Event::Key.new(code: "space").text?   # => false ("space" is not 1 char, " " is)
+        def text?
+          @code.length == 1
+        end
+
+        # Returns the key as a printable character (if applicable).
+        #
+        # [Printable Characters]
+        #   Returns the character itself (e.g., <tt>"a"</tt>, <tt>"1"</tt>, <tt>" "</tt>).
+        # [Special Keys]
+        #   Returns <tt>nil</tt> (e.g., <tt>"enter"</tt>, <tt>"up"</tt>, <tt>"f1"</tt>).
+        #
+        #   RatatuiRuby::Event::Key.new(code: "a").char      # => "a"
+        #   RatatuiRuby::Event::Key.new(code: "enter").char  # => nil
+        def char
+          text? ? @code : nil
+        end
+      end
+    end
+  end
+end

data/lib/ratatui_ruby/event/key/media.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+# SPDX-FileCopyrightText: 2025 Kerrick Long <me@kerricklong.com>
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+module RatatuiRuby
+  class Event
+    class Key < Event
+      # Methods and logic for media keys.
+      module Media
+        # Returns true if this is a media key.
+        #
+        # Media keys include: play, pause, stop, track controls, volume controls.
+        # These are only available in terminals supporting the Kitty keyboard protocol.
+        #
+        #   event.media? # => true for media_play, media_pause, etc.
+        def media?
+          @kind == :media
+        end
+
+        # Handles media-specific DWIM logic for method_missing.
+        private def match_media_dwim?(key_name)
+          return false unless @kind == :media
+
+          # Allow unprefixed predicate
+          # e.g., pause? returns true for media_pause
+          if @code.start_with?("media_")
+            base_code = @code.delete_prefix("media_")
+            return true if key_name == base_code
+          end
+
+          # Bidirectional media overlaps
+          # e.g., play? and pause? both match media_play_pause
+          return true if @code == "media_play_pause" && (key_name == "play" || key_name == "pause")
+
+          # e.g., play_pause? matches media_play or media_pause
+          return true if key_name == "play_pause" && (@code == "media_play" || @code == "media_pause")
+
+          false
+        end
+      end
+    end
+  end
+end