tng 0.2.1

data/lib/tng/ui/theme.rb

@@ -0,0 +1,258 @@
+# 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
+
+      module_function
+
+      # Cache for background detection to avoid multiple escape sequences
+      @background_cache = nil
+
+      def detect_terminal_background
+        # Return cached result if available
+        return @background_cache if @background_cache
+
+        # Get background color and cache the result
+        @background_cache = get_background_color
+        @background_cache
+      end
+
+      def get_background_color
+        return :dark unless $stdout.tty? && $stdin.tty? && interactive_session?
+
+        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 interactive_session?
+        return false if defined?(Rails) && Rails.respond_to?(:application) && Rails.application&.initialized?
+        return false if ENV["BUNDLE_GEMFILE"]
+        return false if $PROGRAM_NAME&.include?("bundle")
+        return false if $PROGRAM_NAME&.include?("rails") && ARGV.any? { |arg| %w[server console runner].include?(arg) }
+        
+        ENV["TNG_CLI"] == "true" || $PROGRAM_NAME&.include?("tng")
+      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