ruby_rich 0.3.1 → 0.4.0

data/lib/ruby_rich/progress_bar.rb

@@ -1,29 +1,247 @@
 module RubyRich
   class ProgressBar
     
-    attr_reader :progress
+    attr_reader :progress, :total, :start_time
 
-    def initialize(total, width: 50, style: :default)
+    # 进度条样式
+    STYLES = {
+      default: { filled: '=', empty: ' ', prefix: '[', suffix: ']' },
+      blocks: { filled: '█', empty: '░', prefix: '[', suffix: ']' },
+      arrows: { filled: '>', empty: '-', prefix: '[', suffix: ']' },
+      dots: { filled: '●', empty: '○', prefix: '(', suffix: ')' },
+      line: { filled: '━', empty: '─', prefix: '│', suffix: '│' },
+      gradient: { filled: ['▁', '▂', '▃', '▄', '▅', '▆', '▇', '█'], empty: ' ', prefix: '[', suffix: ']' }
+    }.freeze
+
+    def initialize(total, width: 50, style: :default, title: nil, show_percentage: true, show_rate: false, show_eta: false)
       @total = total
       @progress = 0
       @width = width
       @style = style
+      @title = title
+      @show_percentage = show_percentage
+      @show_rate = show_rate
+      @show_eta = show_eta
+      @start_time = nil
+      @last_update_time = nil
+      @update_history = []
     end
 
-    def advance(amount)
+    def start
+      @start_time = Time.now
+      @last_update_time = @start_time
+      render
+    end
+
+    def advance(amount = 1)
+      @start_time ||= Time.now
       @progress += amount
-      @progress = @total if @progress > @total
+      @progress = [@progress, @total].min
+      
+      current_time = Time.now
+      @update_history << { time: current_time, progress: @progress }
+      
+      # 保留最近的几个更新用于计算速率
+      @update_history = @update_history.last(10)
+      @last_update_time = current_time
+      
       render
+      puts if completed?
+    end
+
+    def set_progress(value)
+      @start_time ||= Time.now
+      @progress = [[value, 0].max, @total].min
+      
+      current_time = Time.now
+      @update_history << { time: current_time, progress: @progress }
+      @update_history = @update_history.last(10)
+      @last_update_time = current_time
+      
+      render
+      puts if completed?
+    end
+
+    def completed?
+      @progress >= @total
+    end
+
+    def percentage
+      return 0 if @total == 0
+      (@progress.to_f / @total * 100).round(1)
+    end
+
+    def elapsed_time
+      return 0 unless @start_time
+      Time.now - @start_time
+    end
+
+    def rate
+      return 0 if @update_history.length < 2
+      
+      first_update = @update_history.first
+      last_update = @update_history.last
+      
+      time_diff = last_update[:time] - first_update[:time]
+      progress_diff = last_update[:progress] - first_update[:progress]
+      
+      return 0 if time_diff == 0
+      progress_diff / time_diff
+    end
+
+    def eta
+      return 0 if rate == 0 || completed?
+      remaining = @total - @progress
+      remaining / rate
     end
 
     def render
-      percentage = (@progress.to_f / @total * 100).to_i
-      completed_width = (@progress.to_f / @total * @width).to_i
-      incomplete_width = @width - completed_width
+      bar_content = render_bar
+      status_text = render_status
+      
+      output = ""
+      output << "\e[94m#{@title}: \e[0m" if @title
+      output << bar_content
+      output << " #{status_text}" unless status_text.empty?
+      
+      print "\r\e[K#{output}"
+      $stdout.flush
+    end
+
+    def finish(message: nil)
+      if message
+        puts "\r\e[K#{message}"
+      else
+        puts
+      end
+    end
+
+    # 静态方法：创建带回调的进度条
+    def self.with_progress(total, **options)
+      bar = new(total, **options)
+      bar.start
+      
+      begin
+        yield(bar) if block_given?
+      ensure
+        bar.finish
+      end
+      
+      bar
+    end
+
+    # 多进度条管理器
+    class MultiProgress
+      def initialize
+        @bars = []
+        @active = false
+      end
+
+      def add(title, total, **options)
+        bar = ProgressBar.new(total, title: title, **options)
+        @bars << bar
+        bar
+      end
+
+      def start
+        @active = true
+        render_all
+      end
+
+      def render_all
+        return unless @active
+        
+        print "\e[#{@bars.length}A" unless @bars.empty? # 移动光标到顶部
+        
+        @bars.each_with_index do |bar, index|
+          bar_output = render_bar_line(bar)
+          puts "\e[K#{bar_output}"
+        end
+        
+        $stdout.flush
+      end
+
+      def finish_all
+        @active = false
+        puts
+      end
+
+      private
+
+      def render_bar_line(bar)
+        bar_content = bar.send(:render_bar)
+        status_text = bar.send(:render_status)
+        
+        output = ""
+        output << "\e[94m#{bar.instance_variable_get(:@title)}: \e[0m" if bar.instance_variable_get(:@title)
+        output << bar_content
+        output << " #{status_text}" unless status_text.empty?
+        output
+      end
+    end
+
+    private
+
+    def render_bar
+      style_config = STYLES[@style] || STYLES[:default]
+      
+      completed_width = (@progress.to_f / @total * @width).round
+      
+      if @style == :gradient
+        filled_chars = style_config[:filled]
+        filled_full = filled_chars.last * (completed_width / filled_chars.length)
+        filled_partial = filled_chars[completed_width % filled_chars.length] if completed_width % filled_chars.length > 0
+        filled = filled_full + (filled_partial || '')
+        empty = style_config[:empty] * (@width - filled.length)
+      else
+        filled = style_config[:filled] * completed_width
+        empty = style_config[:empty] * (@width - completed_width)
+      end
+      
+      # 添加颜色
+      color_filled = completed? ? "\e[92m" : "\e[96m"  # 完成时绿色，进行中蓝色
+      color_empty = "\e[90m"  # 空白部分灰色
+      color_reset = "\e[0m"
+      
+      "#{style_config[:prefix]}#{color_filled}#{filled}#{color_empty}#{empty}#{color_reset}#{style_config[:suffix]}"
+    end
+
+    def render_status
+      parts = []
+      
+      if @show_percentage
+        parts << "#{percentage}%"
+      end
+      
+      parts << "(#{@progress}/#{@total})"
+      
+      if @show_rate && rate > 0
+        parts << sprintf("%.1f/s", rate)
+      end
+      
+      if @show_eta && eta > 0 && !completed?
+        eta_formatted = format_time(eta)
+        parts << "ETA: #{eta_formatted}"
+      end
+      
+      if @start_time
+        elapsed_formatted = format_time(elapsed_time)
+        parts << "#{elapsed_formatted}"
+      end
+      
+      parts.join(" ")
+    end
 
-      bar = "[#{"=" * completed_width}#{" " * incomplete_width}]"
-      print "\r#{bar} #{percentage}%"
-      puts if @progress == @total
+    def format_time(seconds)
+      if seconds < 60
+        sprintf("%.1fs", seconds)
+      elsif seconds < 3600
+        minutes = seconds / 60
+        sprintf("%.1fm", minutes)
+      else
+        hours = seconds / 3600
+        sprintf("%.1fh", hours)
+      end
     end
   end
-end
+end

data/lib/ruby_rich/status.rb

@@ -0,0 +1,246 @@
+module RubyRich
+  class Status
+    # 状态指示器类型
+    INDICATORS = {
+      # 简单状态
+      success: { symbol: '✅', color: "\e[92m", text: 'Success' },
+      error: { symbol: '❌', color: "\e[91m", text: 'Error' },
+      warning: { symbol: '⚠️', color: "\e[93m", text: 'Warning' },
+      info: { symbol: 'ℹ️', color: "\e[94m", text: 'Info' },
+      
+      # 进度状态
+      pending: { symbol: '⏳', color: "\e[93m", text: 'Pending' },
+      running: { symbol: '🏃', color: "\e[94m", text: 'Running' },
+      completed: { symbol: '✅', color: "\e[92m", text: 'Completed' },
+      failed: { symbol: '💥', color: "\e[91m", text: 'Failed' },
+      
+      # 系统状态
+      online: { symbol: '🟢', color: "\e[92m", text: 'Online' },
+      offline: { symbol: '🔴', color: "\e[91m", text: 'Offline' },
+      maintenance: { symbol: '🔧', color: "\e[93m", text: 'Maintenance' },
+      
+      # 安全状态
+      secure: { symbol: '🔒', color: "\e[92m", text: 'Secure' },
+      insecure: { symbol: '🔓', color: "\e[91m", text: 'Insecure' },
+      
+      # 等级状态
+      low: { symbol: '🔵', color: "\e[94m", text: 'Low' },
+      medium: { symbol: '🟡', color: "\e[93m", text: 'Medium' },
+      high: { symbol: '🔴', color: "\e[91m", text: 'High' },
+      critical: { symbol: '💀', color: "\e[95m", text: 'Critical' }
+    }.freeze
+
+    # 加载动画帧
+    SPINNER_FRAMES = {
+      dots: ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'],
+      line: ['|', '/', '-', '\\'],
+      arrow: ['←', '↖', '↑', '↗', '→', '↘', '↓', '↙'],
+      bounce: ['⠁', '⠂', '⠄', '⠂'],
+      pulse: ['●', '◐', '◑', '◒', '◓', '◔', '◕', '○'],
+      clock: ['🕐', '🕑', '🕒', '🕓', '🕔', '🕕', '🕖', '🕗', '🕘', '🕙', '🕚', '🕛']
+    }.freeze
+
+    def self.indicator(type, text: nil, show_text: true, colorize: true)
+      config = INDICATORS[type.to_sym]
+      return "Unknown status: #{type}" unless config
+      
+      symbol = config[:symbol]
+      color = colorize ? config[:color] : ""
+      reset = colorize ? "\e[0m" : ""
+      status_text = text || config[:text]
+      
+      if show_text
+        "#{color}#{symbol} #{status_text}#{reset}"
+      else
+        "#{color}#{symbol}#{reset}"
+      end
+    end
+
+    def self.spinner(type: :dots, text: 'Loading...', delay: 0.1)
+      frames = SPINNER_FRAMES[type.to_sym] || SPINNER_FRAMES[:dots]
+      
+      Thread.new do
+        frame_index = 0
+        loop do
+          print "\r\e[K\e[94m#{frames[frame_index]}\e[0m #{text}"
+          $stdout.flush
+          sleep delay
+          frame_index = (frame_index + 1) % frames.length
+        end
+      end
+    end
+
+    def self.stop_spinner(final_message: nil)
+      if final_message
+        print "\r\e[K#{final_message}\n"
+      else
+        print "\r\e[K"
+      end
+      $stdout.flush
+    end
+
+    # 静态进度条
+    def self.progress_bar(current, total, width: 30, style: :filled)
+      percentage = (current.to_f / total * 100).round(1)
+      filled_width = (current.to_f / total * width).round
+      
+      case style
+      when :filled
+        filled = '█' * filled_width
+        empty = '░' * (width - filled_width)
+        bar = "#{filled}#{empty}"
+      when :blocks
+        filled = '■' * filled_width
+        empty = '□' * (width - filled_width)
+        bar = "#{filled}#{empty}"
+      when :dots
+        filled = '●' * filled_width
+        empty = '○' * (width - filled_width)
+        bar = "#{filled}#{empty}"
+      else
+        filled = '=' * filled_width
+        empty = '-' * (width - filled_width)
+        bar = "#{filled}#{empty}"
+      end
+      
+      "\e[92m[#{bar}]\e[0m #{percentage}% (#{current}/#{total})"
+    end
+
+    # 状态板
+    class StatusBoard
+      def initialize(width: 60)
+        @width = width
+        @items = []
+      end
+
+      def add_item(label, status, description: nil)
+        @items << {
+          label: label,
+          status: status,
+          description: description
+        }
+        self
+      end
+
+      def render(show_descriptions: true, align_status: :right)
+        lines = []
+        lines << "┌#{'─' * (@width - 2)}┐"
+        
+        @items.each do |item|
+          label = item[:label]
+          status_text = RubyRich::Status.indicator(item[:status])
+          description = item[:description]
+          
+          # 计算实际显示宽度（排除 ANSI 代码）
+          status_display_width = status_text.gsub(/\e\[[0-9;]*m/, '').length
+          
+          case align_status
+          when :right
+            padding = @width - 4 - label.length - status_display_width
+            padding = [padding, 1].max
+            main_line = "│ #{label}#{' ' * padding}#{status_text} │"
+          when :left
+            padding = @width - 4 - label.length - status_display_width
+            padding = [padding, 1].max
+            main_line = "│ #{status_text} #{label}#{' ' * padding}│"
+          else # center
+            total_content = label.length + status_display_width + 1
+            left_padding = [(@width - 2 - total_content) / 2, 1].max
+            right_padding = @width - 2 - total_content - left_padding
+            main_line = "│#{' ' * left_padding}#{label} #{status_text}#{' ' * right_padding}│"
+          end
+          
+          lines << main_line
+          
+          if show_descriptions && description
+            desc_lines = wrap_text(description, @width - 4)
+            desc_lines.each do |desc_line|
+              padding = @width - 4 - desc_line.length
+              lines << "│  \e[90m#{desc_line}#{' ' * padding}\e[0m  │"
+            end
+          end
+        end
+        
+        lines << "└#{'─' * (@width - 2)}┘"
+        lines.join("\n")
+      end
+
+      private
+
+      def wrap_text(text, max_width)
+        words = text.split(' ')
+        lines = []
+        current_line = ''
+        
+        words.each do |word|
+          if (current_line + ' ' + word).length <= max_width
+            current_line += current_line.empty? ? word : ' ' + word
+          else
+            lines << current_line unless current_line.empty?
+            current_line = word
+          end
+        end
+        
+        lines << current_line unless current_line.empty?
+        lines
+      end
+    end
+
+    # 实时状态监控
+    class Monitor
+      def initialize(refresh_rate: 1.0)
+        @refresh_rate = refresh_rate
+        @items = {}
+        @running = false
+      end
+
+      def add_item(key, label, &block)
+        @items[key] = {
+          label: label,
+          block: block
+        }
+        self
+      end
+
+      def start
+        @running = true
+        
+        Thread.new do
+          while @running
+            system('clear')
+            puts render_status
+            sleep @refresh_rate
+          end
+        end
+      end
+
+      def stop
+        @running = false
+      end
+
+      private
+
+      def render_status
+        lines = []
+        lines << "\e[1m\e[96mSystem Status Monitor\e[0m"
+        lines << "─" * 40
+        lines << ""
+        
+        @items.each do |key, item|
+          begin
+            status = item[:block].call
+            status_indicator = RubyRich::Status.indicator(status)
+            lines << "#{item[:label]}: #{status_indicator}"
+          rescue => e
+            error_indicator = RubyRich::Status.indicator(:error, text: "Error: #{e.message}")
+            lines << "#{item[:label]}: #{error_indicator}"
+          end
+        end
+        
+        lines << ""
+        lines << "\e[90mPress Ctrl+C to stop monitoring\e[0m"
+        lines.join("\n")
+      end
+    end
+  end
+end

data/lib/ruby_rich/syntax.rb

@@ -0,0 +1,171 @@
+require 'rouge'
+
+module RubyRich
+  class Syntax
+    # 支持的语言别名映射
+    LANGUAGE_ALIASES = {
+      'rb' => 'ruby',
+      'py' => 'python',
+      'js' => 'javascript',
+      'ts' => 'typescript',
+      'sh' => 'shell',
+      'bash' => 'shell',
+      'zsh' => 'shell',
+      'yml' => 'yaml',
+      'md' => 'markdown'
+    }.freeze
+
+    # 语法高亮主题颜色映射
+    THEME_COLORS = {
+      # Rouge token types to ANSI colors
+      'Comment' => "\e[90m",           # Bright black (gray)
+      'Comment.Single' => "\e[90m",
+      'Comment.Multiline' => "\e[90m",
+      'Comment.Preproc' => "\e[95m",   # Bright magenta
+      
+      'Keyword' => "\e[94m",           # Bright blue
+      'Keyword.Constant' => "\e[94m",
+      'Keyword.Declaration' => "\e[94m",
+      'Keyword.Namespace' => "\e[94m",
+      'Keyword.Pseudo' => "\e[94m",
+      'Keyword.Reserved' => "\e[94m",
+      'Keyword.Type' => "\e[94m",
+      
+      'Literal' => "\e[96m",           # Bright cyan
+      'Literal.Date' => "\e[96m",
+      'Literal.Number' => "\e[93m",    # Bright yellow
+      'Literal.Number.Bin' => "\e[93m",
+      'Literal.Number.Float' => "\e[93m",
+      'Literal.Number.Hex' => "\e[93m",
+      'Literal.Number.Integer' => "\e[93m",
+      'Literal.Number.Oct' => "\e[93m",
+      
+      'Literal.String' => "\e[92m",    # Bright green
+      'Literal.String.Affix' => "\e[92m",
+      'Literal.String.Backtick' => "\e[92m",
+      'Literal.String.Char' => "\e[92m",
+      'Literal.String.Doc' => "\e[92m",
+      'Literal.String.Double' => "\e[92m",
+      'Literal.String.Escape' => "\e[96m",
+      'Literal.String.Heredoc' => "\e[92m",
+      'Literal.String.Interpol' => "\e[96m",
+      'Literal.String.Other' => "\e[92m",
+      'Literal.String.Regex' => "\e[91m",  # Bright red
+      'Literal.String.Single' => "\e[92m",
+      'Literal.String.Symbol' => "\e[95m", # Bright magenta
+      
+      'Name' => "\e[97m",              # Bright white
+      'Name.Attribute' => "\e[93m",    # Bright yellow
+      'Name.Builtin' => "\e[96m",      # Bright cyan
+      'Name.Builtin.Pseudo' => "\e[96m",
+      'Name.Class' => "\e[93m",        # Bright yellow
+      'Name.Constant' => "\e[93m",
+      'Name.Decorator' => "\e[95m",    # Bright magenta
+      'Name.Entity' => "\e[93m",
+      'Name.Exception' => "\e[91m",    # Bright red
+      'Name.Function' => "\e[96m",     # Bright cyan
+      'Name.Property' => "\e[96m",
+      'Name.Label' => "\e[95m",
+      'Name.Namespace' => "\e[93m",
+      'Name.Other' => "\e[97m",
+      'Name.Tag' => "\e[94m",          # Bright blue
+      'Name.Variable' => "\e[97m",
+      'Name.Variable.Class' => "\e[97m",
+      'Name.Variable.Global' => "\e[97m",
+      'Name.Variable.Instance' => "\e[97m",
+      
+      'Operator' => "\e[91m",          # Bright red
+      'Operator.Word' => "\e[94m",     # Bright blue
+      
+      'Punctuation' => "\e[97m",       # Bright white
+      
+      'Error' => "\e[101m\e[97m",      # Red background, white text
+      'Generic.Deleted' => "\e[91m",   # Bright red
+      'Generic.Emph' => "\e[3m",       # Italic
+      'Generic.Error' => "\e[91m",     # Bright red
+      'Generic.Heading' => "\e[1m\e[94m", # Bold bright blue
+      'Generic.Inserted' => "\e[92m",  # Bright green
+      'Generic.Output' => "\e[90m",    # Bright black (gray)
+      'Generic.Prompt' => "\e[1m",     # Bold
+      'Generic.Strong' => "\e[1m",     # Bold
+      'Generic.Subheading' => "\e[95m", # Bright magenta
+      'Generic.Traceback' => "\e[91m"  # Bright red
+    }.freeze
+
+    def self.highlight(code, language = nil, theme: :default)
+      new(theme: theme).highlight(code, language)
+    end
+
+    def initialize(theme: :default)
+      @theme = theme
+    end
+
+    def highlight(code, language = nil)
+      return code if code.nil? || code.empty?
+      
+      # 检测或规范化语言
+      language = detect_language(code) if language.nil?
+      language = normalize_language(language) if language
+      
+      # 如果无法检测语言，返回原始代码
+      return code unless language
+      
+      begin
+        lexer = Rouge::Lexer.find(language)
+        return code unless lexer
+        
+        # 使用自定义格式化器
+        formatter = AnsiFormatter.new
+        formatter.format(lexer.lex(code))
+      rescue => e
+        # 如果高亮失败，返回原始代码
+        code
+      end
+    end
+
+    def list_languages
+      Rouge::Lexer.all.map(&:tag).sort
+    end
+
+    private
+
+    def detect_language(code)
+      # 简单的语言检测启发式
+      return 'ruby' if code.include?('def ') && code.include?('end')
+      return 'python' if code.include?('def ') && code.include?(':')
+      return 'javascript' if code.include?('function') || code.include?('=>')
+      return 'html' if code.include?('<html') || code.include?('<!DOCTYPE')
+      return 'css' if code.match?(/\w+\s*{[^}]*}/)
+      return 'shell' if code.start_with?('#!/bin/') || code.include?('$ ')
+      return 'json' if code.strip.start_with?('{') && code.include?(':')
+      return 'yaml' if code.include?('---') || code.match?(/^\w+:/)
+      
+      nil
+    end
+
+    def normalize_language(language)
+      return nil if language.nil?
+      language = language.to_s.downcase.strip
+      LANGUAGE_ALIASES[language] || language
+    end
+
+    # 自定义 ANSI 格式化器
+    class AnsiFormatter
+      def format(tokens)
+        output = ''
+        tokens.each do |token, value|
+          color_code = THEME_COLORS[token.qualname] || 
+                      THEME_COLORS[token.token_chain.map(&:qualname).find { |t| THEME_COLORS[t] }] ||
+                      ''
+          
+          if color_code.empty?
+            output << value
+          else
+            output << "#{color_code}#{value}\e[0m"
+          end
+        end
+        output
+      end
+    end
+  end
+end