tng 0.1.3 → 0.1.5

data/lib/tng/ui/show_help.rb

@@ -1,6 +1,7 @@
 require "tty-box"
 require "pastel"
 require "tty-screen"
+require_relative "theme"
 
 class ShowHelp
   attr_reader :pastel
@@ -17,73 +18,107 @@ class ShowHelp
 
   def content
     [
-      @pastel.cyan.bold("TNG - LLM-Powered Rails Test Generator"),
-      @pastel.dim("Version: #{@version}"),
+      @pastel.public_send(Tng::UI::Theme.color(:secondary)).bold("TNG - LLM-Powered Rails Test Generator"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted), "Version: #{@version}"),
       "",
-      @pastel.yellow.bold("Usage:"),
-      @pastel.bright_white("  bundle exec tng") + @pastel.dim("                           # Interactive mode"),
-      @pastel.bright_white("  bundle exec tng --type=TYPE --file=FILE") + @pastel.dim("   # Direct mode"),
-      @pastel.bright_white("  bundle exec tng -t TYPE -f FILE") + @pastel.dim("           # Short aliases"),
-      @pastel.bright_white("  bundle exec tng -t=TYPE f=FILE") + @pastel.dim("            # Mixed format"),
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Usage:"),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  bundle exec tng") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                     "                           # Interactive mode"),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  bundle exec tng --type=TYPE --file=FILE") + @pastel.public_send(
+                            Tng::UI::Theme.color(:muted), "   # Direct mode"
+                          ),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  bundle exec tng -t TYPE -f FILE") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                                     "           # Short aliases"),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  bundle exec tng -t=TYPE f=FILE") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                                    "            # Mixed format"),
       "",
-      @pastel.yellow.bold("Examples:"),
-      @pastel.green("  bundle exec tng -t c -f ping") + @pastel.dim("                    # Controller test"),
-      @pastel.green("  bundle exec tng -t=c f=ping") + @pastel.dim("                     # Mixed format"),
-      @pastel.green("  bundle exec tng -t m -f user") + @pastel.dim("                    # Model test"),
-      @pastel.green("  bundle exec tng --type=controller --file=users_controller"),
-      @pastel.green("  bundle exec tng -t c -f \"admin/users_controller\"") + @pastel.dim(" # Namespaced"),
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Examples:"),
+      @pastel.public_send(Tng::UI::Theme.color(:success),
+                          "  bundle exec tng -t c -f ping") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                                  "                    # Controller test"),
+      @pastel.public_send(Tng::UI::Theme.color(:success),
+                          "  bundle exec tng -t=c f=ping") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                                 "                     # Mixed format"),
+      @pastel.public_send(Tng::UI::Theme.color(:success),
+                          "  bundle exec tng -t m -f user") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                                  "                    # Model test"),
+      @pastel.public_send(Tng::UI::Theme.color(:success),
+                          "  bundle exec tng --type=controller --file=users_controller"),
+      @pastel.public_send(Tng::UI::Theme.color(:success),
+                          "  bundle exec tng -t c -f \"admin/users_controller\"") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                                                        " # Namespaced"),
       "",
-      @pastel.yellow.bold("Options:"),
-      @pastel.bright_white("  -t, --type=TYPE") + @pastel.dim("     Test type (controller, model)"),
-      @pastel.bright_white("  -f, --file=FILE") + @pastel.dim("     File name (without .rb extension)"),
-      @pastel.bright_white("  -m, --method=METHOD") + @pastel.dim(" Method name (for per-method tests)"),
-      @pastel.bright_white("  -h, --help") + @pastel.dim("          Show this help message"),
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Options:"),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  -t, --type=TYPE") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                     "     Test type (controller, model)"),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  -f, --file=FILE") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                     "     File name (without .rb extension)"),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  -m, --method=METHOD") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                         " Method name (for per-method tests)"),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  -h, --help") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                "          Show this help message"),
       "",
-      @pastel.yellow.bold("Type Aliases:"),
-      @pastel.dim("  • Controllers: ") + @pastel.cyan("c, controller"),
-      @pastel.dim("  • Models: ") + @pastel.cyan("m, mo, model"),
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Type Aliases:"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Controllers: ") + @pastel.public_send(Tng::UI::Theme.color(:secondary),
+                                                                                                   "c, controller"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Models: ") + @pastel.public_send(
+                            Tng::UI::Theme.color(:secondary), "m, mo, model"
+                          ),
       "",
-      @pastel.yellow.bold("File Format:"),
-      @pastel.dim("  • Controllers: ") + @pastel.green("users_controller, api/v1/posts_controller"),
-      @pastel.dim("  • Models: ") + @pastel.green("user, blog/post"),
-      @pastel.dim("  • Don't include .rb extension"),
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("File Format:"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Controllers: ") + @pastel.public_send(Tng::UI::Theme.color(:success),
+                                                                                                   "users_controller, api/v1/posts_controller"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Models: ") + @pastel.public_send(
+                            Tng::UI::Theme.color(:success), "user, blog/post"
+                          ),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Don't include .rb extension"),
       "",
-      @pastel.magenta.bold("Pro Tips:"),
-      @pastel.dim("  • Mix formats: ") + @pastel.cyan("-t=c f=ping") + @pastel.dim(" works perfectly"),
-      @pastel.dim("  • Use filtering in interactive mode for large projects"),
-      @pastel.dim("  • All formats are equivalent - use what feels natural"),
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Pro Tips:"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Mix formats: ") + @pastel.public_send(Tng::UI::Theme.color(:secondary),
+                                                                                                   "-t=c f=ping") + @pastel.public_send(
+                                                                                                     Tng::UI::Theme.color(:muted), " works perfectly"
+                                                                                                   ),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Use filtering in interactive mode for large projects"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} All formats are equivalent - use what feels natural"),
       "",
-      @pastel.cyan.bold("Happy testing! ") + @pastel.yellow("★")
+      @pastel.public_send(Tng::UI::Theme.color(:secondary)).bold("Happy testing! ") + @pastel.public_send(
+        Tng::UI::Theme.color(:accent), "★"
+      )
     ].join("\n")
   end
 
   def render
     # Use a reasonable width that matches other UI components
-    box_width = [@terminal_width - 4, 80].min
-    box_width = 60 if box_width < 60
+    box_width = Tng::UI::Theme.calculate_box_width(@terminal_width)
+    box_width = [box_width, 80].min # Allow wider for help content
+    style = Tng::UI::Theme.box_style(:default)
 
     box = TTY::Box.frame(
       title: { top_left: " TNG Help ", bottom_right: " v#{@version} " },
-      style: {
-        fg: :bright_white,
-        border: { fg: :cyan },
-        title: { fg: :bright_cyan }
-      },
-      padding: [1, 2],
+      style: style[:style],
+      padding: style[:padding],
       align: :left,
       width: box_width
     ) do
       content
     end
 
-    center_box(box, box_width)
-  end
-
-  private
-
-  def center_box(box_string, box_width)
-    padding = (@terminal_width - box_width) / 2
-    padding = 0 if padding.negative?
-    box_string.lines.map { |line| (" " * padding) + line.chomp }.join("\n")
+    Tng::UI::Theme.center_box(box, box_width, @terminal_width)
   end
 end

data/lib/tng/ui/system_status_display.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require "pastel"
+require_relative "theme"
 
 class SystemStatusDisplay
   def initialize(pastel, args)
@@ -11,47 +12,117 @@ class SystemStatusDisplay
   def display(status)
     case status[:status]
     when :ok
-      puts @pastel.green("✅ System operational") unless @args[:type] && @args[:file]
+      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
-      puts @pastel.red.bold("❌ Version Mismatch!")
-      puts @pastel.yellow("Your gem version: #{status[:gem_version]}")
-      puts @pastel.yellow("Required version: #{status[:current_version]}")
-      puts
-      puts @pastel.bright_white("Please update your gem:")
-      puts @pastel.cyan("  gem update tng")
-      puts @pastel.dim("  or")
-      puts @pastel.cyan("  bundle update tng")
+      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
-      puts @pastel.red.bold("❌ Endpoint URL Mismatch!")
-      puts @pastel.yellow("Your configured URL: #{status[:user_base_url]}")
-      puts @pastel.yellow("Server's current URL: #{status[:server_base_url]}")
-      puts
-      puts @pastel.bright_white("Please update your configuration:")
-      puts @pastel.cyan("  Edit config/initializers/tng.rb")
-      puts @pastel.cyan("  Set: config.base_url = '#{status[:server_base_url]}'")
-      puts
-      puts @pastel.dim("This ensures you're connecting to the correct endpoint.")
+      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
-      puts @pastel.red.bold("❌ Service Unavailable")
-      puts @pastel.dim(status[:message])
-      puts
-      puts @pastel.yellow("The TNG service is currently down.")
-      puts @pastel.dim("Please try again later or contact support.")
+      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
-      puts @pastel.red.bold("❌ Connection Error")
-      puts @pastel.dim(status[:message])
-      puts
-      puts @pastel.yellow("Unable to connect to TNG service.")
-      puts @pastel.dim("Please check your internet connection and try again.")
+      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
-      puts @pastel.red.bold("❌ Unknown Error")
-      puts @pastel.dim(status[:message] || "An unknown error occurred")
+      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,266 @@
+# 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
+          # Skip background detection if explicitly disabled
+          return :dark if ENV["TNG_DISABLE_BACKGROUND_DETECTION"] == "true"
+          
+          get_background_color
+        end
+      end
+
+      def get_background_color
+        # Don't attempt background detection if not in an interactive terminal
+        # or during Rails startup/loading
+        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?
+        # Skip background detection during Rails startup, bundler operations, etc.
+        return false if defined?(Rails) && Rails.respond_to?(:application) && Rails.application&.initializing?
+        return false if ENV["BUNDLE_GEMFILE"] && !ENV["TNG_FORCE_BACKGROUND_DETECTION"]
+        return false if $PROGRAM_NAME&.include?("bundle")
+        return false if $PROGRAM_NAME&.include?("rails") && ARGV.any? { |arg| %w[server console runner].include?(arg) }
+        
+        # Only allow in truly interactive contexts
+        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