pindo 5.13.12 → 5.13.13

data/lib/pindo/module/task/output/multi_line_output_manager.rb

@@ -0,0 +1,380 @@
+require 'thread'
+require 'fileutils'
+require_relative 'multi_line_task_display'
+require_relative '../../../base/output_sink'
+
+module Pindo
+  module TaskSystem
+    # 多行输出管理器
+    # 管理所有任务的输出（终端显示 + 日志文件）
+    # 实现 OutputSink 接口
+    class MultiLineOutputManager < OutputSink
+      # 初始化输出管理器
+      # @param log_dir [String] 日志目录路径
+      # @param max_lines_per_task [Integer] 每个任务显示的最大行数
+      # @param max_recent_completed [Integer] 显示的最近完成任务数
+      # @param auto_adjust [Boolean] 是否自动适应终端大小
+      def initialize(
+        log_dir: './pindo_logs',
+        max_lines_per_task: 5,
+        max_recent_completed: 3,
+        auto_adjust: true
+      )
+        @mutex = Mutex.new
+        @log_dir = log_dir
+        @max_lines_per_task = max_lines_per_task
+        @max_recent_completed = max_recent_completed
+        @auto_adjust = auto_adjust
+
+        # 任务显示器映射
+        @task_displays = {}  # task_id => MultiLineTaskDisplay
+
+        # 日志文件映射
+        @task_loggers = {}   # task_id => File
+
+        # 终端状态
+        @terminal_lines = 0  # 当前占用的终端行数
+        @last_render_time = Time.now
+        @min_render_interval = 0.1  # 最小刷新间隔（秒）
+
+        # 创建日志目录
+        FileUtils.mkdir_p(@log_dir)
+
+        # 隐藏光标（更流畅的动画）
+        STDOUT.print "\e[?25l" if supports_ansi?
+
+        # 注册退出时恢复光标
+        at_exit { STDOUT.print "\e[?25h" if supports_ansi? }
+      end
+
+      # ============================================
+      # 任务注册
+      # ============================================
+
+      # 注册任务到输出管理器
+      # @param task [Object] 任务对象（需要有 id, name, task_key 属性）
+      def register_task(task)
+        @mutex.synchronize do
+          task_id = task.id
+
+          # 创建多行显示器
+          display = MultiLineTaskDisplay.new(
+            task_id,
+            task.name,
+            max_lines: @max_lines_per_task
+          )
+          @task_displays[task_id] = display
+
+          # 创建日志文件
+          log_file_name = "#{task.task_key}_#{task_id}.log"
+          log_file_path = File.join(@log_dir, log_file_name)
+          @task_loggers[task_id] = File.open(log_file_path, 'w')
+
+          # 渲染终端
+          render_terminal
+        end
+      end
+
+      # ============================================
+      # 任务生命周期
+      # ============================================
+
+      # 任务开始
+      # @param task_id [String] 任务 ID
+      def start_task(task_id)
+        @mutex.synchronize do
+          display = @task_displays[task_id]
+          display&.add_line("任务开始")
+
+          write_log(task_id, "=" * 60)
+          write_log(task_id, "任务开始")
+          write_log(task_id, "=" * 60)
+
+          render_terminal
+        end
+      end
+
+      # 任务成功
+      # @param task_id [String] 任务 ID
+      # @param message [String, nil] 成功消息
+      # @param execution_time [Float, nil] 执行时间（秒）
+      def success_task(task_id, message = nil, execution_time: nil)
+        @mutex.synchronize do
+          display = @task_displays[task_id]
+
+          msg = if execution_time
+                  message ? "#{message} (耗时 #{execution_time.round(2)}s)" : "完成 (耗时 #{execution_time.round(2)}s)"
+                else
+                  message || "完成"
+                end
+
+          display&.mark_success(msg)
+
+          write_log(task_id, "=" * 60)
+          write_log(task_id, "任务成功: #{msg}")
+          write_log(task_id, "=" * 60)
+
+          close_log(task_id)
+          render_terminal
+        end
+      end
+
+      # 任务失败
+      # @param task_id [String] 任务 ID
+      # @param error_message [String] 错误消息
+      def error_task(task_id, error_message)
+        @mutex.synchronize do
+          display = @task_displays[task_id]
+          display&.mark_error(error_message)
+
+          write_log(task_id, "=" * 60)
+          write_log(task_id, "任务失败: #{error_message}")
+          write_log(task_id, "=" * 60)
+
+          close_log(task_id)
+          render_terminal
+        end
+      end
+
+      # ============================================
+      # 实现 OutputSink 接口
+      # ============================================
+
+      # 更新进度（终端可见 + 日志记录）
+      # @param task_id [String] 任务 ID
+      # @param message [String] 进度消息
+      def update_progress(task_id, message)
+        @mutex.synchronize do
+          display = @task_displays[task_id]
+          display&.add_line(message)
+
+          write_log(task_id, "[PROGRESS] #{message}")
+          render_terminal
+        end
+      end
+
+      # 更新当前行（不滚动）
+      # @param task_id [String] 任务 ID
+      # @param message [String] 更新消息
+      def update_current_line(task_id, message)
+        @mutex.synchronize do
+          display = @task_displays[task_id]
+          display&.update_last_line(message)
+
+          write_log(task_id, "[UPDATE] #{message}")
+          render_terminal
+        end
+      end
+
+      # 记录日志（更新终端 + 写文件）
+      # @param task_id [String] 任务 ID
+      # @param message [String] 日志消息
+      def log_info(task_id, message)
+        @mutex.synchronize do
+          display = @task_displays[task_id]
+          display&.add_line(message)
+          
+          write_log(task_id, "[INFO] #{message}")
+          render_terminal
+        end
+      end
+
+      # 记录警告
+      # @param task_id [String] 任务 ID
+      # @param message [String] 警告消息
+      def log_warning(task_id, message)
+        @mutex.synchronize do
+          display = @task_displays[task_id]
+          display&.add_line("\e[33m#{message}\e[0m") # 黄色高亮
+
+          write_log(task_id, "[WARN] #{message}")
+          render_terminal
+        end
+      end
+
+      # 记录错误
+      # @param task_id [String] 任务 ID
+      # @param message [String] 错误消息
+      def log_error(task_id, message)
+        @mutex.synchronize do
+          display = @task_displays[task_id]
+          display&.add_line("\e[31m#{message}\e[0m") # 红色高亮
+
+          write_log(task_id, "[ERROR] #{message}")
+          render_terminal
+        end
+      end
+
+      # 记录详细日志
+      # @param task_id [String] 任务 ID
+      # @param message [String] 详细日志
+      def log_detail(task_id, message)
+        @mutex.synchronize do
+          # 只有非空消息才更新显示
+          unless message.strip.empty?
+            display = @task_displays[task_id]
+            # 为了避免刷屏，普通日志使用 update_last_line 还是 add_line?
+            # 考虑到 log_detail 可能是一行行输出，使用 add_line 更像滚动的日志
+            display&.add_line(message)
+          end
+
+          write_log(task_id, message)
+          render_terminal
+        end
+      end
+
+      # ============================================
+      # 终端渲染（核心显示逻辑）
+      # ============================================
+
+      private
+
+      # 渲染终端显示
+      def render_terminal
+        # 限制刷新频率
+        now = Time.now
+        return if now - @last_render_time < @min_render_interval
+
+        # 如果不支持 ANSI，跳过渲染
+        # 注意：这里使用 STDOUT 检查，因为 $stdout 可能已被重定向
+        return unless supports_ansi?
+
+        # 获取终端尺寸
+        terminal_height = get_terminal_height
+        terminal_width = get_terminal_width
+        available_lines = terminal_height - 5
+
+        # 清除旧内容
+        clear_terminal
+
+        # 分类任务
+        running_tasks = @task_displays.select { |_, d| d.status == MultiLineTaskDisplay::STATUS_RUNNING }
+        completed_tasks = @task_displays.select { |_, d|
+          d.status == MultiLineTaskDisplay::STATUS_SUCCESS || d.status == MultiLineTaskDisplay::STATUS_ERROR
+        }
+        pending_tasks = @task_displays.select { |_, d| d.status == MultiLineTaskDisplay::STATUS_PENDING }
+
+        # 标题
+        total_tasks = @task_displays.size
+        completed_count = completed_tasks.size
+        STDOUT.puts "\e[1;36m⣾ 执行任务中... (#{running_tasks.size}/#{total_tasks} 运行中, #{completed_count}/#{total_tasks} 已完成)\e[0m"
+        used_lines = 1
+
+        # 1. 正在运行的任务（优先级最高）
+        if running_tasks.any?
+          STDOUT.puts ""
+          STDOUT.puts "\e[2m━━━━━━━━━━━━━━━━ 正在运行 ━━━━━━━━━━━━━━━━\e[0m"
+          used_lines += 2
+
+          running_tasks.each do |task_id, display|
+            # 自动适应终端大小
+            lines_for_this_task = if @auto_adjust && (used_lines + display.line_count > available_lines)
+                                    [available_lines - used_lines, 2].max
+                                  else
+                                    @max_lines_per_task
+                                  end
+
+            STDOUT.puts ""
+            # 传递 max_width 防止换行
+            STDOUT.puts display.render(max_lines: lines_for_this_task, max_width: terminal_width)
+            used_lines += display.line_count + 1
+
+            break if used_lines >= available_lines
+          end
+        end
+
+        # 2. 最近完成的任务（如果空间允许）
+        if completed_tasks.any? && (used_lines + 10 < available_lines)
+          STDOUT.puts ""
+          STDOUT.puts "\e[2m━━━━━━━━━━━━━━━━ 最近完成 ━━━━━━━━━━━━━━━━\e[0m"
+          used_lines += 2
+
+          recent_completed = completed_tasks.values.last(@max_recent_completed)
+          recent_completed.each do |display|
+            STDOUT.puts ""
+            # 完成的任务只显示 1 行，同样需要防止换行
+            STDOUT.puts display.render(max_lines: 1, max_width: terminal_width)
+            used_lines += 2
+
+            break if used_lines >= available_lines
+          end
+        end
+
+        # 3. 其他任务（折叠显示）
+        if pending_tasks.any? || completed_tasks.size > @max_recent_completed
+          STDOUT.puts ""
+          STDOUT.puts "\e[2m━━━━━━━━━━━━━━━━ 其他 ━━━━━━━━━━━━━━━━\e[0m"
+
+          other_completed = [completed_tasks.size - @max_recent_completed, 0].max
+          summary = "  已完成: #{other_completed}  等待中: #{pending_tasks.size}"
+          STDOUT.puts summary
+          used_lines += 3
+        end
+
+        @terminal_lines = used_lines
+        @last_render_time = now
+      end
+
+      # 清除终端内容
+      def clear_terminal
+        if @terminal_lines > 0
+          STDOUT.print "\e[#{@terminal_lines}A"  # 光标上移
+          STDOUT.print "\e[J"                     # 清除到屏幕底部
+        end
+      end
+
+      # 获取终端高度
+      # @return [Integer] 终端行数
+      def get_terminal_height
+        require 'io/console'
+        $stdout.winsize[0]
+      rescue
+        40  # 默认 40 行
+      end
+
+      # 获取终端宽度
+      # @return [Integer] 终端列数
+      def get_terminal_width
+        require 'io/console'
+        $stdout.winsize[1]
+      rescue
+        80  # 默认 80 列
+      end
+
+      # 检查是否支持 ANSI 转义序列
+      # @return [Boolean] 是否支持 ANSI
+      def supports_ansi?
+        return false unless $stdout.tty?
+        return false if ENV['TERM'] == 'dumb'
+        return false if ENV['PINDO_NO_SPINNER'] == '1'
+        true
+      end
+
+      # ============================================
+      # 日志文件操作
+      # ============================================
+
+      # 写入日志
+      # @param task_id [String] 任务 ID
+      # @param message [String] 日志消息
+      def write_log(task_id, message)
+        logger = @task_loggers[task_id]
+        return unless logger
+
+        timestamp = Time.now.strftime('%H:%M:%S')
+        logger.puts("[#{timestamp}] #{message}")
+        logger.flush
+      end
+
+      # 关闭日志文件
+      # @param task_id [String] 任务 ID
+      def close_log(task_id)
+        logger = @task_loggers[task_id]
+        return unless logger
+
+        logger.close
+        @task_loggers.delete(task_id)
+      end
+    end
+  end
+end

data/lib/pindo/module/task/output/multi_line_task_display.rb

@@ -0,0 +1,185 @@
+require 'thread'
+
+module Pindo
+  module TaskSystem
+    # 单任务多行显示器
+    # 管理单个任务的多行滚动显示，采用 FIFO 队列机制
+    class MultiLineTaskDisplay
+      attr_reader :task_id, :task_name, :status, :lines
+
+      # 任务状态枚举
+      STATUS_RUNNING = :running
+      STATUS_SUCCESS = :success
+      STATUS_ERROR = :error
+      STATUS_PENDING = :pending
+
+      # 初始化显示器
+      # @param task_id [String] 任务 ID
+      # @param task_name [String] 任务名称
+      # @param max_lines [Integer] 最大显示行数（默认 5）
+      def initialize(task_id, task_name, max_lines: 5)
+        @task_id = task_id
+        # 防止任务名称中有换行符破坏显示
+        @task_name = task_name.to_s.gsub("\n", " ")
+        @max_lines = max_lines
+        @status = STATUS_RUNNING
+        @lines = []  # 滚动缓冲区（FIFO 队列）
+        @mutex = Mutex.new
+      end
+
+      # 添加新行（滚动）
+      # 新行添加到队列尾部，如果超过 max_lines，则移除最旧的行
+      # 支持多行消息自动拆分
+      # @param message [String] 消息内容
+      def add_line(message)
+        @mutex.synchronize do
+          timestamp = Time.now.strftime('%H:%M:%S')
+          
+          # 处理包含换行符的消息，确保每一行都独立计数
+          str_message = message.to_s
+          
+          # 如果消息为空，至少添加一行空行（或者忽略，这里选择忽略空消息以保持整洁）
+          return if str_message.empty?
+          
+          sub_messages = str_message.split("\n")
+          if sub_messages.empty?
+             # 处理仅包含换行符的情况
+             sub_messages = [""] 
+          end
+
+          sub_messages.each do |msg|
+            line = "#{timestamp} #{msg}"
+            @lines << line
+            
+            # 保持最多 max_lines 行（FIFO）
+            @lines.shift if @lines.size > @max_lines
+          end
+        end
+      end
+
+      # 更新最后一行（不滚动，用于进度百分比）
+      # 为了保证行数计算准确，这里强行将换行符替换为空格
+      # @param message [String] 更新消息
+      def update_last_line(message)
+        @mutex.synchronize do
+          timestamp = Time.now.strftime('%H:%M:%S')
+          
+          # 替换换行符，保证只占用一行
+          clean_message = message.to_s.gsub("\n", " ")
+          line = "#{timestamp} #{clean_message}"
+
+          if @lines.empty?
+            @lines << line
+          else
+            @lines[-1] = line
+          end
+        end
+      end
+
+      # 标记任务成功
+      # @param message [String, nil] 成功消息
+      def mark_success(message = nil)
+        @mutex.synchronize do
+          @status = STATUS_SUCCESS
+          add_line_unsafe(message) if message
+        end
+      end
+
+      # 标记任务失败
+      # @param message [String] 错误消息
+      def mark_error(message)
+        @mutex.synchronize do
+          @status = STATUS_ERROR
+          add_line_unsafe("错误: #{message}")
+        end
+      end
+
+      # 渲染为终端字符串
+      # @param max_lines [Integer, nil] 最大显示行数（可选覆盖默认值）
+      # @param max_width [Integer, nil] 最大显示宽度（可选，用于截断长行以防止换行破坏布局）
+      # @return [String] 终端显示字符串
+      def render(max_lines: nil, max_width: nil)
+        @mutex.synchronize do
+          lines_output = []
+
+          # 状态图标和颜色
+          icon = case @status
+                 when STATUS_RUNNING then '⣾'
+                 when STATUS_SUCCESS then '✔'
+                 when STATUS_ERROR then '✖'
+                 when STATUS_PENDING then '⊙'
+                 else '?'
+                 end
+
+          color_code = case @status
+                       when STATUS_SUCCESS then "\e[32m"  # 绿色
+                       when STATUS_ERROR then "\e[31m"    # 红色
+                       when STATUS_PENDING then "\e[33m"  # 黄色
+                       else "\e[34m"                      # 蓝色（运行中）
+                       end
+
+          # 任务名称行
+          # 截断任务名称（保留一些空间给图标）
+          task_line = "  #{color_code}#{icon} #{@task_name}\e[0m"
+          lines_output << task_line
+
+          # 消息行（树形结构）
+          display_lines = max_lines ? @lines.last(max_lines) : @lines
+          display_lines.each_with_index do |line, index|
+            prefix = if index == display_lines.size - 1
+                       '└─'  # 最后一行
+                     else
+                       '├─'  # 中间行
+                     end
+            
+            full_line = "    #{prefix} #{line}"
+            
+            # 如果指定了最大宽度，进行截断
+            if max_width && max_width > 0
+              # 简单截断防止换行
+              # 注意：这里假设全角字符占2宽度的情况未被完美处理，但对于防止严重错乱已足够
+              if full_line.length > max_width
+                 # 保留最后 3 个字符为 "..."
+                 full_line = full_line[0, max_width - 3] + "..."
+              end
+            end
+
+            lines_output << full_line
+          end
+
+          lines_output.join("\n")
+        end
+      end
+
+      # 计算占用的行数
+      # @return [Integer] 占用的终端行数
+      def line_count
+        @mutex.synchronize do
+          1 + @lines.size  # 任务名称行 + 消息行
+        end
+      end
+
+      private
+
+      # 内部方法：添加新行（不加锁，由调用者保证线程安全）
+      # 支持多行消息拆分
+      def add_line_unsafe(message)
+        timestamp = Time.now.strftime('%H:%M:%S')
+        
+        str_message = message.to_s
+        return if str_message.empty?
+        
+        sub_messages = str_message.split("\n")
+        if sub_messages.empty?
+             sub_messages = [""]
+        end
+
+        sub_messages.each do |msg|
+            line = "#{timestamp} #{msg}"
+            @lines << line
+            @lines.shift if @lines.size > @max_lines
+        end
+      end
+    end
+  end
+end

data/lib/pindo/module/task/output/stdout_redirector.rb

@@ -0,0 +1,95 @@
+module Pindo
+  module TaskSystem
+    # Stdout 重定向器
+    # 用于在任务执行期间捕获标准输出和标准错误，将其路由到输出管理器
+    class StdoutRedirector
+      # 为任务重定向 stdout/stderr
+      # @param task_id [String] 任务 ID
+      # @param output_manager [OutputSink] 输出管理器
+      # @yield 执行任务的代码块
+      def self.redirect_for_task(task_id, output_manager)
+        # 保存原始 stdout/stderr
+        original_stdout = $stdout
+        original_stderr = $stderr
+
+        begin
+          # 重定向到自定义 IO
+          $stdout = CustomIO.new(task_id, output_manager, :stdout)
+          $stderr = CustomIO.new(task_id, output_manager, :stderr)
+
+          yield
+        ensure
+          # 恢复原始 stdout/stderr
+          $stdout = original_stdout
+          $stderr = original_stderr
+        end
+      end
+    end
+
+    # 自定义 IO 对象
+    # 拦截所有 puts/print 输出，将其路由到输出管理器
+    class CustomIO
+      # 初始化自定义 IO
+      # @param task_id [String] 任务 ID
+      # @param output_manager [OutputSink] 输出管理器
+      # @param stream_type [Symbol] 流类型（:stdout 或 :stderr）
+      def initialize(task_id, output_manager, stream_type)
+        @task_id = task_id
+        @output_manager = output_manager
+        @stream_type = stream_type
+      end
+
+      # 写入数据
+      # @param message [String] 消息内容
+      # @return [Integer] 写入的字节数
+      def write(message)
+        # 捕获输出，写入日志文件（跳过空行）
+        unless message.strip.empty?
+          # 根据流类型选择日志级别
+          if @stream_type == :stderr
+            @output_manager.log_error(@task_id, message.chomp)
+          else
+            @output_manager.log_detail(@task_id, message.chomp)
+          end
+        end
+        message.length
+      end
+
+      # puts 方法
+      # @param args [Array] 参数列表
+      def puts(*args)
+        if args.empty?
+          write("\n")
+        else
+          args.each { |arg| write("#{arg}\n") }
+        end
+        nil
+      end
+
+      # print 方法
+      # @param args [Array] 参数列表
+      def print(*args)
+        write(args.join)
+        nil
+      end
+
+      # flush 方法（无操作）
+      def flush
+        # 不需要处理
+      end
+
+      # 委托其他方法给 STDOUT
+      def method_missing(method, *args, &block)
+        STDOUT.send(method, *args, &block)
+      end
+
+      # 响应方法检查
+      # @param method [Symbol] 方法名
+      # @param include_private [Boolean] 是否包含私有方法
+      # @return [Boolean] 是否响应该方法
+      def respond_to_missing?(method, include_private = false)
+        STDOUT.respond_to?(method, include_private) || super
+      end
+    end
+  end
+end