tng 0.1.4

data/lib/tng/ui/system_status_display.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+require "pastel"
+require_relative "theme"
+
+class SystemStatusDisplay
+  def initialize(pastel, args)
+    @pastel = pastel
+    @args = args
+  end
+
+  def display(status)
+    case status[:status]
+    when :ok
+      unless @args[:type] && @args[:file]
+        puts @pastel.public_send(Tng::UI::Theme.color(:success),
+                                 "#{Tng::UI::Theme.icon(:success)} System operational")
+      end
+      true
+    when :version_mismatch
+      display_status_message(
+        title: "Version Mismatch",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [
+          "Your gem version: #{status[:gem_version]}",
+          "Required version: #{status[:current_version]}",
+          "",
+          "Quick fix:",
+          "• gem update tng",
+          "• bundle update tng"
+        ]
+      )
+      false
+    when :base_url_mismatch
+      display_status_message(
+        title: "URL Mismatch",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [
+          "Your URL: #{status[:user_base_url]}",
+          "Server URL: #{status[:server_base_url]}",
+          "",
+          "Quick fix:",
+          "• Edit config/initializers/tng.rb",
+          "• Set: config.base_url = '#{status[:server_base_url]}'"
+        ]
+      )
+      false
+    when :service_down
+      display_status_message(
+        title: "Service Unavailable",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [
+          status[:message] || "TNG service is currently down",
+          "",
+          "Please try again later or contact support"
+        ]
+      )
+      false
+    when :error
+      display_status_message(
+        title: "Connection Error",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [
+          status[:message] || "Unable to connect to TNG service",
+          "",
+          "Please check your internet connection and try again"
+        ]
+      )
+      false
+    else
+      display_status_message(
+        title: "Unknown Error",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [status[:message] || "An unknown error occurred"]
+      )
+      false
+    end
+  end
+
+  private
+
+  def display_status_message(title:, icon:, color:, content:)
+    require "tty-box"
+    require "tty-screen"
+
+    terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+
+    formatted_content = content.map do |line|
+      case line
+      when /^•/
+        @pastel.public_send(Tng::UI::Theme.color(:muted), line)
+      when /^Quick fix:/
+        @pastel.public_send(Tng::UI::Theme.color(:secondary), line)
+      when ""
+        line
+      else
+        @pastel.public_send(Tng::UI::Theme.color(:warning), line)
+      end
+    end.join("\n")
+
+    box_width = Tng::UI::Theme.calculate_box_width(terminal_width)
+
+    status_box = TTY::Box.frame(
+      title: { top_left: " #{title} " },
+      style: {
+        fg: :bright_white,
+        border: { fg: color },
+        title: { fg: color == Tng::UI::Theme.color(:error) ? :bright_red : color }
+      },
+      padding: [1, 2],
+      width: box_width
+    ) do
+      formatted_content
+    end
+
+    puts Tng::UI::Theme.center_box(status_box, box_width, terminal_width)
+    puts
+  end
+end

data/lib/tng/ui/theme.rb

@@ -0,0 +1,248 @@
+# frozen_string_literal: true
+
+module Tng
+  module UI
+    module Theme
+      # Icons - consistent across all UI components
+      ICONS = {
+        # Status icons
+        success: "✅",
+        error: "❌",
+        warning: "⚠️",
+        info: "ℹ️",
+
+        # Action icons
+        rocket: "🚀",
+        wave: "👋",
+        stats: "📊",
+        config: "📋",
+        heart: "❤️",
+        lightbulb: "💡",
+
+        # Navigation icons
+        back: "⬅️ ", # Added space after back arrow
+        marker: "▶",
+        bullet: "•"
+      }.freeze
+
+      # Colors - terminal-agnostic color scheme
+      COLORS = {
+        # Primary status colors
+        success: :green,
+        error: :red,
+        warning: :yellow,
+        info: :cyan,
+
+        # Text hierarchy colors
+        primary: :bright_white,
+        secondary: :cyan,
+        accent: :yellow,
+        muted: :dim,
+
+        # Border colors for boxes
+        border_success: :green,
+        border_error: :red,
+        border_warning: :yellow,
+        border_info: :cyan,
+        border_primary: :cyan
+      }.freeze
+
+      # Box styling constants
+      BOX_STYLES = {
+        default: {
+          style: {
+            border: {
+              type: :light,
+              top_left: "┌",
+              top_right: "┐",
+              bottom_left: "└",
+              bottom_right: "┘",
+              top: "─",
+              bottom: "─",
+              left: "│",
+              right: "│"
+            }
+          },
+          padding: [1, 2],
+          max_width: 100
+        },
+        success: {
+          style: {
+            border: {
+              type: :light,
+              top_left: "┌",
+              top_right: "┐",
+              bottom_left: "└",
+              bottom_right: "┘",
+              top: "─",
+              bottom: "─",
+              left: "│",
+              right: "│"
+            },
+            fg: :green
+          },
+          padding: [1, 2],
+          max_width: 100
+        },
+        warning: {
+          style: {
+            border: {
+              type: :light,
+              top_left: "┌",
+              top_right: "┐",
+              bottom_left: "└",
+              bottom_right: "┘",
+              top: "─",
+              bottom: "─",
+              left: "│",
+              right: "│"
+            },
+            fg: :yellow
+          },
+          padding: [1, 2],
+          max_width: 100
+        }
+      }.freeze
+
+      # Helper methods for consistent styling
+      module_function
+
+      def detect_terminal_background
+        # Manual override for TNG (highest priority)
+        case ENV["TNG_TERMINAL_THEME"]&.downcase
+        when "light"
+          :light
+        when "dark"
+          :dark
+        else
+          get_background_color
+        end
+      end
+
+      def get_background_color
+        print "\e]11;?\e\\"
+        $stdout.flush
+
+        require "timeout"
+        begin
+          response = ""
+          Timeout.timeout(0.1) do
+            response = $stdin.read_nonblock(100)
+          end
+
+          case response
+          when %r{rgb:([0-9a-f]{4})/([0-9a-f]{4})/([0-9a-f]{4})}i
+            r, g, b = [::Regexp.last_match(1), ::Regexp.last_match(2), ::Regexp.last_match(3)].map do |hex|
+              hex.to_i(16) >> 8
+            end
+            brightness = (r * 0.299 + g * 0.587 + b * 0.114)
+            return brightness > 128 ? :light : :dark
+          when /rgb:([0-9a-f]{2})([0-9a-f]{2})([0-9a-f]{2})/i
+            r, g, b = [::Regexp.last_match(1), ::Regexp.last_match(2), ::Regexp.last_match(3)].map do |hex|
+              hex.to_i(16)
+            end
+            brightness = (r * 0.299 + g * 0.587 + b * 0.114)
+            return brightness > 128 ? :light : :dark
+          end
+        rescue IO::WaitReadable, Errno::EAGAIN
+          # No data available
+        rescue Timeout::Error, StandardError
+          # Fall back to environment detection
+        end
+
+        # Enhanced fallback detection
+        case ENV["TERM_PROGRAM"]
+        when "iTerm.app"
+          ENV["ITERM_PROFILE"]&.downcase&.include?("light") ? :light : :dark
+        when "Apple_Terminal"
+          :light  # Terminal.app defaults to light
+        when "vscode"
+          :dark   # VS Code integrated terminal usually dark
+        else
+          # Check for common dark theme indicators
+          if ENV["COLORTERM"] == "truecolor" || ENV["TERM"]&.include?("256")
+            :dark
+          else
+            :light
+          end
+        end
+      end
+
+      def adaptive_color(color_key)
+        base_color = base_color(color_key)
+
+        # Adjust colors based on terminal background for optimal visibility
+        if detect_terminal_background == :light
+          # Light terminal adjustments
+          case base_color
+          when :dim
+            :black
+          when :bright_white
+            :black  # White text invisible on light backgrounds
+          when :cyan
+            :blue   # Cyan can be hard to read on light backgrounds
+          when :yellow
+            :red    # Yellow invisible on light backgrounds
+          else
+            base_color
+          end
+        else
+          # Dark terminal adjustments - optimized for macOS Terminal
+          case base_color
+          when :dim
+            # macOS Terminal renders :white better than :bright_white
+            :white
+          when :black
+            :white # Black text invisible on dark backgrounds
+          when :cyan
+            # macOS Terminal cyan can be hard to read, use blue
+            :blue
+          when :yellow
+            # macOS Terminal yellow can be faded, use brighter alternatives
+            :red
+          when :bright_white
+            # Force to regular white for better macOS Terminal compatibility
+            :white
+          else
+            base_color
+          end
+        end
+      end
+
+      def icon(name)
+        ICONS[name] || ""
+      end
+
+      def color(name)
+        adaptive_color(name)
+      end
+
+      def base_color(name)
+        COLORS[name] || :white
+      end
+
+      def box_style(name)
+        BOX_STYLES[name] || BOX_STYLES[:default]
+      end
+
+      def calculate_box_width(terminal_width)
+        [terminal_width - 4, BOX_STYLES[:default][:max_width]].min
+      end
+
+      def center_box(box_content, box_width, terminal_width)
+        lines = box_content.split("\n")
+        padding = (terminal_width - box_width) / 2
+        padding = 0 if padding.negative?
+        lines.map { |line| (" " * padding) + line }.join("\n")
+      end
+
+      def center_text(text, terminal_width)
+        # Remove ANSI color codes for length calculation
+        clean_text = text.gsub(/\e\[[0-9;]*m/, "")
+        padding = (terminal_width - clean_text.length) / 2
+        padding = 0 if padding.negative?
+        (" " * padding) + text
+      end
+    end
+  end
+end

data/lib/tng/ui/user_stats_display.rb

@@ -0,0 +1,160 @@
+# frozen_string_literal: true
+
+require "tty-box"
+require "tty-progressbar"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class UserStatsDisplay
+  def initialize(pastel, prompt)
+    @pastel = pastel
+    @prompt = prompt
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def display(stats_data)
+    header = @pastel.public_send(Tng::UI::Theme.color(:primary)).bold("#{Tng::UI::Theme.icon(:stats)} User Statistics")
+    puts Tng::UI::Theme.center_text(header, @terminal_width)
+
+    if stats_data
+      display_stats_box(stats_data)
+      display_usage_progress(stats_data)
+      display_usage_info(stats_data)
+    else
+      display_error
+    end
+
+    puts
+    tip_msg = @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                  "#{Tng::UI::Theme.icon(:lightbulb)} Tip: Contact support if you need more test generations")
+    puts Tng::UI::Theme.center_text(tip_msg, @terminal_width)
+    @prompt.keypress(Tng::UI::Theme.center_text(
+                       @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                           "Press any key to continue..."), @terminal_width
+                     ))
+  end
+
+  def display_stats_box(stats_data)
+    stats_content = [
+      @pastel.public_send(Tng::UI::Theme.color(:secondary)).bold("Account Information"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:primary), "Test Runs: ") + @pastel.public_send(Tng::UI::Theme.color(:success)).bold(stats_data["runs"].to_s),
+      @pastel.public_send(Tng::UI::Theme.color(:primary), "Max Runs: ") + @pastel.public_send(Tng::UI::Theme.color(:warning)).bold(stats_data["max_runs"].to_s),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "Gem Version: ") + @pastel.public_send(Tng::UI::Theme.color(:accent),
+                                                                 stats_data["gem_version"] || "N/A"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:muted), "Request ID: #{stats_data["request_id"]}")
+    ].join("\n")
+
+    box_width = Tng::UI::Theme.calculate_box_width(@terminal_width)
+    style = Tng::UI::Theme.box_style(:success)
+
+    box = TTY::Box.frame(
+      title: { top_left: " Your TNG Stats " },
+      style: style[:style],
+      padding: style[:padding],
+      align: :left,
+      width: box_width
+    ) { stats_content }
+
+    puts Tng::UI::Theme.center_box(box, box_width, @terminal_width)
+  end
+
+  def display_usage_progress(stats_data)
+    runs = stats_data["runs"]
+    max_runs = stats_data["max_runs"]
+
+    usage_percent = max_runs > 0 ? (runs.to_f / max_runs * 100).round(1) : 0
+
+    bar_color = determine_bar_color(usage_percent)
+
+    puts
+    puts Tng::UI::Theme.center_text(@pastel.public_send(Tng::UI::Theme.color(:primary)).bold("Usage Overview"),
+                                    @terminal_width)
+    puts
+
+    progress_width = 40
+    padding = (@terminal_width - progress_width - 20) / 2
+    padding = 0 if padding.negative?
+
+    bar_format = "#{" " * padding}Usage: [:bar] :percent (:current/:total)"
+
+    bar = TTY::ProgressBar.new(bar_format,
+                               total: max_runs,
+                               width: progress_width,
+                               complete: bar_color,
+                               incomplete: @pastel.public_send(Tng::UI::Theme.color(:muted), "░"),
+                               head: bar_color)
+
+    current_progress = 0
+    while current_progress < runs
+      step = [runs / 10, 1].max
+      current_progress = [current_progress + step, runs].min
+      bar.current = current_progress
+      bar.render
+      sleep(0.05) if runs > 10
+    end
+
+    bar.current = runs
+    bar.render
+
+    puts
+
+    status_msg = case usage_percent
+                 when 0..50
+                   @pastel.public_send(Tng::UI::Theme.color(:success),
+                                       "#{Tng::UI::Theme.icon(:success)} Good usage - plenty of runs remaining")
+                 when 51..80
+                   @pastel.public_send(Tng::UI::Theme.color(:warning),
+                                       "#{Tng::UI::Theme.icon(:warning)} Moderate usage - consider monitoring")
+                 when 81..95
+                   @pastel.public_send(Tng::UI::Theme.color(:info),
+                                       "#{Tng::UI::Theme.icon(:warning)} High usage - approaching limit")
+                 else
+                   @pastel.public_send(Tng::UI::Theme.color(:error),
+                                       "#{Tng::UI::Theme.icon(:error)} Limit reached - contact support for more runs")
+                 end
+
+    puts Tng::UI::Theme.center_text(status_msg, @terminal_width)
+  end
+
+  def determine_bar_color(usage_percent)
+    case usage_percent
+    when 0..50
+      @pastel.public_send(Tng::UI::Theme.color(:success), "█")
+    when 51..80
+      @pastel.public_send(Tng::UI::Theme.color(:warning), "█")
+    when 81..95
+      @pastel.public_send(Tng::UI::Theme.color(:info), "█")
+    else
+      @pastel.public_send(Tng::UI::Theme.color(:error), "█")
+    end
+  end
+
+  def display_usage_info(stats_data)
+    usage_remaining = stats_data["max_runs"] - stats_data["runs"]
+    usage_msg = if usage_remaining.positive?
+                  @pastel.public_send(Tng::UI::Theme.color(:success),
+                                      "#{Tng::UI::Theme.icon(:success)} You have #{usage_remaining} test generations remaining")
+                else
+                  @pastel.public_send(Tng::UI::Theme.color(:error),
+                                      "#{Tng::UI::Theme.icon(:warning)} You have reached your test generation limit")
+                end
+    puts Tng::UI::Theme.center_text(usage_msg, @terminal_width)
+  end
+
+  def display_error
+    error_msg = @pastel.public_send(Tng::UI::Theme.color(:error)).bold("#{Tng::UI::Theme.icon(:error)} Failed to fetch user statistics")
+    puts Tng::UI::Theme.center_text(error_msg, @terminal_width)
+    puts Tng::UI::Theme.center_text(
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "Please check your API key and internet connection"), @terminal_width
+    )
+  end
+end