rich-ruby 1.0.1 → 1.0.2

data/lib/rich/panel.rb

@@ -1,311 +1,318 @@
-# frozen_string_literal: true
-
-require_relative "box"
-require_relative "style"
-require_relative "segment"
-require_relative "cells"
-require_relative "text"
-
-module Rich
-  # A bordered panel container for content
-  class Panel
-    # @return [Object] Content to display
-    attr_reader :content
-
-    # @return [String, nil] Panel title
-    attr_reader :title
-
-    # @return [String, nil] Panel subtitle
-    attr_reader :subtitle
-
-    # @return [Box] Box style
-    attr_reader :box
-
-    # @return [Style, nil] Border style
-    attr_reader :border_style
-
-    # @return [Style, nil] Title style
-    attr_reader :title_style
-
-    # @return [Style, nil] Subtitle style
-    attr_reader :subtitle_style
-
-    # @return [Boolean] Expand to fill width
-    attr_reader :expand
-
-    # @return [Integer] Padding inside panel
-    attr_reader :padding
-
-    # @return [Integer, nil] Fixed width
-    attr_reader :width
-
-    # @return [Symbol] Title alignment
-    attr_reader :title_align
-
-    def initialize(
-      content,
-      title: nil,
-      subtitle: nil,
-      box: Box::ROUNDED,
-      border_style: nil,
-      title_style: nil,
-      subtitle_style: nil,
-      expand: true,
-      padding: 1,
-      width: nil,
-      title_align: :center
-    )
-      @content = content
-      @title = title
-      @subtitle = subtitle
-      @box = box
-      @border_style = border_style.is_a?(String) ? Style.parse(border_style) : border_style
-      @title_style = title_style.is_a?(String) ? Style.parse(title_style) : title_style
-      @subtitle_style = subtitle_style.is_a?(String) ? Style.parse(subtitle_style) : subtitle_style
-      @expand = expand
-      @padding = padding
-      @width = width
-      @title_align = title_align
-    end
-
-    # Render panel to segments
-    # @param max_width [Integer] Maximum width
-    # @return [Array<Segment>]
-    def to_segments(max_width: 80)
-      segments = []
-      inner_width = calculate_inner_width(max_width)
-      content_width = inner_width - @padding * 2
-
-      # Render content to lines
-      content_lines = render_content(content_width)
-
-      # Top border with title
-      segments.concat(render_top_border(inner_width))
-      segments << Segment.new("\n")
-
-      # Padding top
-      @padding.times do
-        segments.concat(render_empty_row(inner_width))
-        segments << Segment.new("\n")
-      end
-
-      # Content rows
-      content_lines.each do |line|
-        segments.concat(render_content_row(line, inner_width))
-        segments << Segment.new("\n")
-      end
-
-      # Padding bottom
-      @padding.times do
-        segments.concat(render_empty_row(inner_width))
-        segments << Segment.new("\n")
-      end
-
-      # Bottom border with subtitle
-      segments.concat(render_bottom_border(inner_width))
-
-      segments
-    end
-
-    # Render panel to string with ANSI codes
-    # @param max_width [Integer] Maximum width
-    # @param color_system [Symbol] Color system
-    # @return [String]
-    def render(max_width: 80, color_system: ColorSystem::TRUECOLOR)
-      Segment.render(to_segments(max_width: max_width), color_system: color_system)
-    end
-
-    # Print panel to console
-    # @param console [Console] Console to print to
-    def print_to(console)
-      rendered = render(max_width: console.width, color_system: console.color_system)
-      console.write(rendered)
-      console.write("\n")
-    end
-
-    class << self
-      # Create a simple panel
-      # @param content [String] Content
-      # @param title [String, nil] Title
-      # @return [Panel]
-      def fit(content, title: nil)
-        new(content, title: title, expand: false)
-      end
-    end
-
-    private
-
-    def calculate_inner_width(max_width)
-      if @width
-        @width - 2 # Subtract borders
-      elsif @expand
-        max_width - 2
-      else
-        content_width = measure_content_width + @padding * 2
-        [content_width, max_width - 2].min
-      end
-    end
-
-    def measure_content_width
-      case @content
-      when String
-        Cells.cell_len(@content)
-      when Text
-        @content.cell_length
-      else
-        Cells.cell_len(@content.to_s)
-      end
-    end
-
-    def render_content(width)
-      case @content
-      when String
-        wrap_text(@content, width)
-      when Text
-        @content.wrap(width).map { |t| t.to_segments }
-      else
-        wrap_text(@content.to_s, width)
-      end
-    end
-
-    def wrap_text(text, width)
-      lines = []
-      text.split("\n").each do |line|
-        if Cells.cell_len(line) <= width
-          lines << [Segment.new(line)]
-        else
-          # Simple wrapping
-          current_line = +""
-          current_width = 0
-
-          line.each_char do |char|
-            char_width = Cells.char_width(char)
-            if current_width + char_width > width
-              lines << [Segment.new(current_line)]
-              current_line = +char
-              current_width = char_width
-            else
-              current_line << char
-              current_width += char_width
-            end
-          end
-
-          lines << [Segment.new(current_line)] unless current_line.empty?
-        end
-      end
-      lines
-    end
-
-    def render_top_border(inner_width)
-      segments = []
-
-      if @title
-        title_text = " #{@title} "
-        title_width = Cells.cell_len(title_text)
-
-        if title_width < inner_width
-          remaining = inner_width - title_width
-          case @title_align
-          when :left
-            left = 2
-            right = remaining - 2
-          when :right
-            left = remaining - 2
-            right = 2
-          else # :center
-            left = remaining / 2
-            right = remaining - left
-          end
-
-          segments << Segment.new(@box.top_left, style: @border_style)
-          segments << Segment.new(@box.horizontal * left, style: @border_style)
-          segments << Segment.new(title_text, style: @title_style || @border_style)
-          segments << Segment.new(@box.horizontal * right, style: @border_style)
-          segments << Segment.new(@box.top_right, style: @border_style)
-        else
-          segments << Segment.new(@box.top_edge(inner_width), style: @border_style)
-        end
-      else
-        segments << Segment.new(@box.top_left, style: @border_style)
-        segments << Segment.new(@box.horizontal * inner_width, style: @border_style)
-        segments << Segment.new(@box.top_right, style: @border_style)
-      end
-
-      segments
-    end
-
-    def render_bottom_border(inner_width)
-      segments = []
-
-      if @subtitle
-        sub_text = " #{@subtitle} "
-        sub_width = Cells.cell_len(sub_text)
-
-        if sub_width < inner_width
-          remaining = inner_width - sub_width
-          case @title_align
-          when :left
-            left = 2
-            right = remaining - 2
-          when :right
-            left = remaining - 2
-            right = 2
-          else # :center
-            left = remaining / 2
-            right = remaining - left
-          end
-
-          segments << Segment.new(@box.bottom_left, style: @border_style)
-          segments << Segment.new(@box.horizontal * left, style: @border_style)
-          segments << Segment.new(sub_text, style: @subtitle_style || @border_style)
-          segments << Segment.new(@box.horizontal * right, style: @border_style)
-          segments << Segment.new(@box.bottom_right, style: @border_style)
-        else
-          segments << Segment.new(@box.bottom_edge(inner_width), style: @border_style)
-        end
-      else
-        segments << Segment.new(@box.bottom_left, style: @border_style)
-        segments << Segment.new(@box.horizontal * inner_width, style: @border_style)
-        segments << Segment.new(@box.bottom_right, style: @border_style)
-      end
-
-      segments
-    end
-
-    def render_content_row(content_segments, inner_width)
-      segments = []
-
-      # Left border
-      segments << Segment.new(@box.vertical, style: @border_style)
-
-      # Left padding
-      segments << Segment.new(" " * @padding)
-
-      # Content
-      content_width = content_segments.sum(&:cell_length)
-      segments.concat(content_segments)
-
-      # Right padding (fill to width)
-      remaining = inner_width - @padding * 2 - content_width
-      segments << Segment.new(" " * [remaining, 0].max)
-
-      # Right padding
-      segments << Segment.new(" " * @padding)
-
-      # Right border
-      segments << Segment.new(@box.vertical, style: @border_style)
-
-      segments
-    end
-
-    def render_empty_row(inner_width)
-      segments = []
-
-      segments << Segment.new(@box.vertical, style: @border_style)
-      segments << Segment.new(" " * inner_width)
-      segments << Segment.new(@box.vertical, style: @border_style)
-
-      segments
-    end
-  end
-end
+# frozen_string_literal: true
+
+require_relative "box"
+require_relative "style"
+require_relative "segment"
+require_relative "cells"
+require_relative "text"
+
+module Rich
+  # A bordered panel container for content
+  class Panel
+    # @return [Object] Content to display
+    attr_reader :content
+
+    # @return [String, nil] Panel title
+    attr_reader :title
+
+    # @return [String, nil] Panel subtitle
+    attr_reader :subtitle
+
+    # @return [Box] Box style
+    attr_reader :box
+
+    # @return [Style, nil] Border style
+    attr_reader :border_style
+
+    # @return [Style, nil] Title style
+    attr_reader :title_style
+
+    # @return [Style, nil] Subtitle style
+    attr_reader :subtitle_style
+
+    # @return [Boolean] Expand to fill width
+    attr_reader :expand
+
+    # @return [Integer] Padding inside panel
+    attr_reader :padding
+
+    # @return [Integer, nil] Fixed width
+    attr_reader :width
+
+    # @return [Symbol] Title alignment
+    attr_reader :title_align
+
+    # @return [Symbol] Subtitle alignment
+    attr_reader :subtitle_align
+
+    def initialize(
+      content,
+      title: nil,
+      subtitle: nil,
+      box: Box::ROUNDED,
+      border_style: nil,
+      title_style: nil,
+      subtitle_style: nil,
+      expand: true,
+      padding: 1,
+      width: nil,
+      title_align: :center,
+      subtitle_align: :center
+    )
+      @content = content
+      @title = title
+      @subtitle = subtitle
+      @box = box
+      @border_style = border_style.is_a?(String) ? Style.parse(border_style) : border_style
+      @title_style = title_style.is_a?(String) ? Style.parse(title_style) : title_style
+      @subtitle_style = subtitle_style.is_a?(String) ? Style.parse(subtitle_style) : subtitle_style
+      @expand = expand
+      @padding = padding
+      @width = width
+      @title_align = title_align
+      @subtitle_align = subtitle_align
+    end
+
+    # Render panel to segments
+    # @param max_width [Integer] Maximum width
+    # @return [Array<Segment>]
+    def to_segments(max_width: 80)
+      segments = []
+      inner_width = calculate_inner_width(max_width)
+      content_width = [inner_width - @padding * 2, 0].max
+
+      # Render content to lines
+      content_lines = render_content(content_width)
+
+      # Top border with title
+      segments.concat(render_top_border(inner_width))
+      segments << Segment.new("\n")
+
+      # Padding top
+      @padding.times do
+        segments.concat(render_empty_row(inner_width))
+        segments << Segment.new("\n")
+      end
+
+      # Content rows
+      content_lines.each do |line|
+        segments.concat(render_content_row(line, inner_width))
+        segments << Segment.new("\n")
+      end
+
+      # Padding bottom
+      @padding.times do
+        segments.concat(render_empty_row(inner_width))
+        segments << Segment.new("\n")
+      end
+
+      # Bottom border with subtitle
+      segments.concat(render_bottom_border(inner_width))
+
+      segments
+    end
+
+    # Render panel to string with ANSI codes
+    # @param max_width [Integer] Maximum width
+    # @param color_system [Symbol] Color system
+    # @return [String]
+    def render(max_width: 80, color_system: ColorSystem::TRUECOLOR)
+      Segment.render(to_segments(max_width: max_width), color_system: color_system)
+    end
+
+    # Print panel to console
+    # @param console [Console] Console to print to
+    def print_to(console)
+      rendered = render(max_width: console.width, color_system: console.color_system)
+      console.write(rendered)
+      console.write("\n")
+    end
+
+    class << self
+      # Create a simple panel
+      # @param content [String] Content
+      # @param title [String, nil] Title
+      # @return [Panel]
+      def fit(content, title: nil)
+        new(content, title: title, expand: false)
+      end
+    end
+
+    private
+
+    def calculate_inner_width(max_width)
+      raw = if @width
+              @width - 2 # Subtract borders
+            elsif @expand
+              max_width - 2
+            else
+              content_width = measure_content_width + @padding * 2
+              [content_width, max_width - 2].min
+            end
+      # Never go negative: a width smaller than the two border columns would
+      # otherwise produce `String * -1` and crash.
+      [raw, 0].max
+    end
+
+    def measure_content_width
+      case @content
+      when String
+        Cells.cell_len(@content)
+      when Text
+        @content.cell_length
+      else
+        Cells.cell_len(@content.to_s)
+      end
+    end
+
+    def render_content(width)
+      case @content
+      when String
+        wrap_text(@content, width)
+      when Text
+        @content.wrap(width).map { |t| t.to_segments }
+      else
+        wrap_text(@content.to_s, width)
+      end
+    end
+
+    def wrap_text(text, width)
+      lines = []
+      text.split("\n").each do |line|
+        if Cells.cell_len(line) <= width
+          lines << [Segment.new(line)]
+        else
+          # Simple wrapping
+          current_line = +""
+          current_width = 0
+
+          line.each_char do |char|
+            char_width = Cells.char_width(char)
+            if current_width + char_width > width
+              lines << [Segment.new(current_line)]
+              current_line = +char
+              current_width = char_width
+            else
+              current_line << char
+              current_width += char_width
+            end
+          end
+
+          lines << [Segment.new(current_line)] unless current_line.empty?
+        end
+      end
+      lines
+    end
+
+    def render_top_border(inner_width)
+      render_label_edge(@title, @title_align, @box.top_left, @box.top_right,
+                        @title_style, inner_width)
+    end
+
+    def render_bottom_border(inner_width)
+      render_label_edge(@subtitle, @subtitle_align, @box.bottom_left, @box.bottom_right,
+                        @subtitle_style, inner_width)
+    end
+
+    # Build a horizontal edge that optionally embeds a centered/left/right
+    # label. The edge always spans exactly inner_width cells between the two
+    # corners, and every fill count is clamped non-negative so a label that is
+    # as wide as (or wider than) the panel can never crash or misalign.
+    def render_label_edge(label, align, left_corner, right_corner, label_style, inner_width)
+      segments = []
+
+      unless label
+        segments << Segment.new(left_corner, style: @border_style)
+        segments << Segment.new(@box.horizontal * inner_width, style: @border_style)
+        segments << Segment.new(right_corner, style: @border_style)
+        return segments
+      end
+
+      text = " #{label} "
+      if Cells.cell_len(text) > inner_width
+        label = truncate_label(label, [inner_width - 2, 0].max)
+        text = " #{label} "
+      end
+      text_width = Cells.cell_len(text)
+      remaining = [inner_width - text_width, 0].max
+
+      case align
+      when :left
+        left = [2, remaining].min
+        right = remaining - left
+      when :right
+        right = [2, remaining].min
+        left = remaining - right
+      else # :center
+        left = remaining / 2
+        right = remaining - left
+      end
+
+      segments << Segment.new(left_corner, style: @border_style)
+      segments << Segment.new(@box.horizontal * left, style: @border_style)
+      segments << Segment.new(text, style: label_style || @border_style)
+      segments << Segment.new(@box.horizontal * right, style: @border_style)
+      segments << Segment.new(right_corner, style: @border_style)
+      segments
+    end
+
+    # Truncate a label to a maximum cell width (used when a title/subtitle is
+    # wider than the panel).
+    def truncate_label(text, max_width)
+      return "" if max_width <= 0
+
+      result = +""
+      current_width = 0
+      text.to_s.each_char do |char|
+        cw = Cells.char_width(char)
+        break if current_width + cw > max_width
+
+        result << char
+        current_width += cw
+      end
+      result
+    end
+
+    def render_content_row(content_segments, inner_width)
+      segments = []
+
+      # Left border
+      segments << Segment.new(@box.vertical, style: @border_style)
+
+      # Left padding
+      segments << Segment.new(" " * @padding)
+
+      # Content (cropped so it can never exceed the content area and push the
+      # right border out of alignment)
+      content_area = [inner_width - @padding * 2, 0].max
+      content_width = content_segments.sum(&:cell_length)
+      if content_width > content_area
+        content_segments = Segment.crop_line(content_segments, content_area)
+        content_width = content_segments.sum(&:cell_length)
+      end
+      segments.concat(content_segments)
+
+      # Right padding (fill to width)
+      remaining = inner_width - @padding * 2 - content_width
+      segments << Segment.new(" " * [remaining, 0].max)
+
+      # Right padding
+      segments << Segment.new(" " * @padding)
+
+      # Right border
+      segments << Segment.new(@box.vertical, style: @border_style)
+
+      segments
+    end
+
+    def render_empty_row(inner_width)
+      segments = []
+
+      segments << Segment.new(@box.vertical, style: @border_style)
+      segments << Segment.new(" " * inner_width)
+      segments << Segment.new(@box.vertical, style: @border_style)
+
+      segments
+    end
+  end
+end